From 2df2d6b0cf63fa90b0fe5160a02dade7d161f667 Mon Sep 17 00:00:00 2001 From: Heiner Lohaus Date: Tue, 14 Jan 2025 17:07:39 +0100 Subject: Read FinishReason and Usage from Gemini API Add "Custom Provider": Set API Url in the settings Remove Discord link from result, add them to attr: Jmuz Fix Bug: File content are added to the prompt Changed response from /v1/models API Disable Pizzagpt Provider --- g4f/api/__init__.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) (limited to 'g4f/api/__init__.py') diff --git a/g4f/api/__init__.py b/g4f/api/__init__.py index d4a3bd76..374d7c64 100644 --- a/g4f/api/__init__.py +++ b/g4f/api/__init__.py @@ -215,12 +215,16 @@ class Api: HTTP_200_OK: {"model": List[ModelResponseModel]}, }) async def models(): - return [{ - 'id': model_id, - 'object': 'model', - 'created': 0, - 'owned_by': model.base_provider - } for model_id, model in g4f.models.ModelUtils.convert.items()] + return { + "object": "list", + "data": [{ + "id": model_id, + "object": "model", + "created": 0, + "owned_by": model.base_provider, + "image": isinstance(model, g4f.models.ImageModel), + } for model_id, model in g4f.models.ModelUtils.convert.items()] + } @self.app.get("/v1/models/{model_name}", responses={ HTTP_200_OK: {"model": ModelResponseModel}, -- cgit v1.2.3