Spaces:
Running
Running
File size: 1,744 Bytes
57d4e72 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 | from __future__ import annotations
from helpers.api import ApiHandler, Request
from plugins._oauth.helpers.providers import CODEX_PROVIDER_ID, get_provider
class Models(ApiHandler):
async def process(self, input: dict, request: Request) -> dict:
del request
raw_provider_id = _provider_id(input)
try:
provider = get_provider(raw_provider_id)
model_catalog = getattr(provider, "model_catalog", None)
if callable(model_catalog):
catalog = model_catalog()
models = [
str(item.get("slug") or item.get("id") or "")
for item in catalog
if isinstance(item, dict) and (item.get("slug") or item.get("id"))
]
else:
catalog = []
models = provider.models()
return {
"ok": True,
"provider_id": provider.provider_id,
"models": models,
"model_metadata": catalog,
}
except Exception as exc:
return {
"ok": False,
"provider_id": _provider_id_label(raw_provider_id),
"error": str(exc),
"models": [],
}
def _provider_id(input: dict) -> object:
if "provider_id" not in input or input.get("provider_id") is None:
return CODEX_PROVIDER_ID
value = input.get("provider_id")
if isinstance(value, str) and not value.strip():
return CODEX_PROVIDER_ID
return value
def _provider_id_label(value: object) -> str:
if value is None:
return CODEX_PROVIDER_ID
text = str(value).strip()
return text or CODEX_PROVIDER_ID
|