add util functions to get models, providers and endpoints

README.md

@@ -59,6 +59,14 @@ unify.set_provider("deepinfra")
 ### Supported Models
 The list of supported models and providers is available in [the platform](https://unify.ai/hub).
 
+You can also get this information directly in Python using `list_models()`, `list_providers()` and `list_endpoints()`.
+
+```python
+models = unify.list_models()
+providers = unify.list_providers("mistral-7b-instruct-v0.1")
+endpoints = unify.list_endpoints("mistral-7b-instruct-v0.1")
+```
+
 ### API Key
 You can get an API Key from [the Unify console](https://console.unify.ai/)
 

unify/__init__.py

@@ -2,3 +2,4 @@
 
 from unify.clients import AsyncUnify, Unify  # noqa: F403
 from unify.chat import ChatBot  # noqa: F403
+from unify.utils import list_endpoints, list_models, list_providers

unify/utils.py

@@ -1,4 +1,6 @@
 import os
+import requests
+import json
 from typing import Optional, Tuple
 
 from unify.exceptions import UnifyError
@@ -16,6 +18,21 @@
     "tks-per-sec",
 ]
 
+_base_url = "https://api.unify.ai/v0"
+
+def _res_to_list(response):
+    return json.loads(response.text)
+
+def list_models():
+    return _res_to_list(requests.get(_base_url + "/models"))
+
+def list_endpoints(model: str):
+    url = _base_url + "/endpoints_of"
+    return _res_to_list(requests.get(url, params={"model": model}))
+
+def list_providers(model: str):
+    url = _base_url + "/providers_of"
+    return _res_to_list(requests.get(url, params={"model": model}))
 
 def _validate_api_key(api_key: Optional[str]) -> str:
     if api_key is None: