xtekky/gpt4free
1
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2024-11-26 09:57:24 +03:00
Code Issues Packages Projects Releases Wiki Activity

Improve compatibility of openai-node (#1966)

Browse Source 
* improve compatibility of openai-node

* improve compatibility with OpenAI's models api

* allow image upload
This commit is contained in:
碧舞すみほ 2024-05-18 01:43:17 +09:00 committed by GitHub
parent 0332c0c0dd
commit 1d66c57a65
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
g4f/api/__init__.py
View File

@ -120,7 +120,10 @@ class Api:
'created': 0, 'created': 0,
'owned_by': model.base_provider 'owned_by': model.base_provider
} for model_id, model in model_list.items()] } for model_id, model in model_list.items()]
return JSONResponse(model_list) return JSONResponse({
"object": "list",
"data": model_list,
})
@self.app.get("/v1/models/{model_name}") @self.app.get("/v1/models/{model_name}")
async def model_info(model_name: str): async def model_info(model_name: str):

2
g4f/client/stubs.py
View File

@ -78,12 +78,14 @@ class ChatCompletionDelta(Model):
def __init__(self, content: Union[str, None]): def __init__(self, content: Union[str, None]):
if content is not None: if content is not None:
self.content = content self.content = content
self.role = "assistant"
def to_json(self): def to_json(self):
return self.__dict__ return self.__dict__
class ChatCompletionDeltaChoice(Model): class ChatCompletionDeltaChoice(Model):
def __init__(self, delta: ChatCompletionDelta, finish_reason: Union[str, None]): def __init__(self, delta: ChatCompletionDelta, finish_reason: Union[str, None]):
self.index = 0
self.delta = delta self.delta = delta
self.finish_reason = finish_reason self.finish_reason = finish_reason

2
g4f/typing.py
View File

@ -14,7 +14,7 @@ else:
SHA256 = NewType('sha_256_hash', str) SHA256 = NewType('sha_256_hash', str)
CreateResult = Iterator[str] CreateResult = Iterator[str]
AsyncResult = AsyncIterator[str] AsyncResult = AsyncIterator[str]
Messages = List[Dict[str, str]] Messages = List[Dict[str, Union[str,List[Dict[str,Union[str,Dict[str,str]]]]]]]
Cookies = Dict[str, str] Cookies = Dict[str, str]
ImageType = Union[str, bytes, IO, Image, None] ImageType = Union[str, bytes, IO, Image, None]