Improve compatibility of openai-node (#1966)

* 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

View File

@ -120,7 +120,10 @@ class Api:
'created': 0,
'owned_by': model.base_provider
} 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}")
async def model_info(model_name: str):

View File

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

View File

@ -14,7 +14,7 @@ else:
SHA256 = NewType('sha_256_hash', str)
CreateResult = Iterator[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]
ImageType = Union[str, bytes, IO, Image, None]