pt
/
gpt4free
mirror of https://github.com/xtekky/gpt4free.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1243 from AndPim4912/main

This commit is contained in:
Tekky 2023-11-13 09:57:50 +00:00 committed by GitHub
parent ed008dcfe6 5a6d59e171
commit 82f3cdc762
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
g4f/api/__init__.py
View File

@ -1,3 +1,6 @@
import ast
import logging
from fastapi import FastAPI, Response, Request from fastapi import FastAPI, Response, Request
from fastapi.responses import StreamingResponse from fastapi.responses import StreamingResponse
from typing import List, Union, Any, Dict, AnyStr from typing import List, Union, Any, Dict, AnyStr
@ -68,14 +71,20 @@ class Api:
'stream': False, 'stream': False,
} }
item_data.update(item or {}) # item contains byte keys, and dict.get suppresses error
item_data.update({key.decode('utf-8') if isinstance(key, bytes) else key: str(value) for key, value in (item or {}).items()})
# messages is str, need dict
if isinstance(item_data.get('messages'), str):
item_data['messages'] = ast.literal_eval(item_data.get('messages'))
model = item_data.get('model') model = item_data.get('model')
stream = item_data.get('stream') stream = item_data.get('stream')
messages = item_data.get('messages') messages = item_data.get('messages')
try: try:
response = g4f.ChatCompletion.create(model=model, stream=stream, messages=messages) response = g4f.ChatCompletion.create(model=model, stream=stream, messages=messages)
except: except Exception as e:
logging.exception(e)
return Response(content=json.dumps({"error": "An error occurred while generating the response."}, indent=4), media_type="application/json") return Response(content=json.dumps({"error": "An error occurred while generating the response."}, indent=4), media_type="application/json")
completion_id = ''.join(random.choices(string.ascii_letters + string.digits, k=28)) completion_id = ''.join(random.choices(string.ascii_letters + string.digits, k=28))