Add ApiAirforce provider

This commit is contained in:
hlohaus 2025-08-10 22:28:11 +02:00
parent 480f5caec8
commit 05148d2b30
6 changed files with 33 additions and 18 deletions

View file

@ -47,7 +47,7 @@ class OpenaiTemplate(AsyncGeneratorProvider, ProviderModelMixin, RaiseErrorMixin
raise_for_status(response)
data = response.json()
data = data.get("data") if isinstance(data, dict) else data
cls.image_models = [model.get("name") if cls.use_model_names else model.get("id", model.get("name")) for model in data if model.get("image") or model.get("type") == "image"]
cls.image_models = [model.get("name") if cls.use_model_names else model.get("id", model.get("name")) for model in data if model.get("image") or model.get("type") == "image" or model.get("supports_images")]
cls.vision_models = cls.vision_models.copy()
cls.vision_models += [model.get("name") if cls.use_model_names else model.get("id", model.get("name")) for model in data if model.get("vision")]
cls.models = [model.get("name") if cls.use_model_names else model.get("id", model.get("name")) for model in data]