gpt4free

公开

关注 0 Fork 0 Star 0

XFEstudio/gpt4free

fix: add model count to provider response and clean up type assignment

2f29284b

hlohaus <hlohaus@users.noreply.github.com>

提交于 2026-06-11 04:50

父提交 c7d231a

1 个文件 +2 -1

Modified g4f/api/__init__.py +2 -1

@@ -381,7 +381,7 @@ class Api:

381 381 "vision": provider in vision_models,

382 382 "audio": provider in audio_models,

383 383 "video": provider in video_models,

384 "type": "image" if provider in image_models else "chat",

384 "type": "image" if provider in image_models else "chat"

385 385 }]

386 386 }

387 387 return ErrorResponse.from_message(str(e), 404)

@@ -403,6 +403,7 @@ class Api:

403 403

                    "audio": (model.get("id") if isinstance(model, dict) else model) in getattr(provider, "audio_models", []),

404 404

                    "video": (model.get("id") if isinstance(model, dict) else model) in getattr(provider, "video_models", []),

405 405

                    "type": "image" if (model.get("id") if isinstance(model, dict) else model) in getattr(provider, "image_models", []) else "chat",

406 "count": provider.models_count.get(model, 0),

406 407 **(model if isinstance(model, dict) else {})

407 408 } for model in (models.values() if isinstance(models, dict) else models)]

408 409 }