zishuo/gpt4free
1
0
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2025-11-02 20:34:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add headers / styling from in UI @kqlio67

Browse Source
This commit is contained in:
hlohaus
2025-03-31 15:21:30 +02:00
parent 21268f0988
commit 7a75782af3
6 changed files with 204 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
g4f/gui/server/backend_api.py
View File

@@ -18,6 +18,7 @@ from hashlib import sha256
from ...client.service import convert_to_provider
from ...providers.asyncio import to_sync_generator
from ...providers.response import FinishReason
from ...client.helper import filter_markdown
from ...tools.files import supports_filename, get_streaming, get_bucket_dir, get_buckets
from ...tools.run_tools import iter_run_tools
@@ -79,16 +80,12 @@ class Backend_Api(Api):
@app.route('/backend-api/v2/models', methods=['GET'])
def jsonify_models(**kwargs):
response = get_demo_models() if app.demo else self.get_models(**kwargs)
if isinstance(response, list):
return jsonify(response)
return response
return jsonify(response)
@app.route('/backend-api/v2/models/<provider>', methods=['GET'])
def jsonify_provider_models(**kwargs):
response = self.get_provider_models(**kwargs)
if isinstance(response, list):
return jsonify(response)
return response
return jsonify(response)
@app.route('/backend-api/v2/providers', methods=['GET'])
def jsonify_providers(**kwargs):
@@ -259,9 +256,11 @@ class Backend_Api(Api):
else:
response = iter_run_tools(ChatCompletion.create, **parameters)
cache_dir.mkdir(parents=True, exist_ok=True)
copy_response = [chunk for chunk in response]
with cache_file.open("w") as f:
for chunk in response:
for chunk in copy_response:
f.write(str(chunk))
response = copy_response
else:
response = iter_run_tools(ChatCompletion.create, **parameters)
@@ -269,7 +268,9 @@ class Backend_Api(Api):
return Response(filter_markdown("".join([str(chunk) for chunk in response]), do_filter_markdown), mimetype='text/plain')
def cast_str():
for chunk in response:
if not isinstance(chunk, Exception):
if isinstance(chunk, FinishReason):
yield f"[{chunk.reason}]" if chunk.reason != "stop" else ""
elif not isinstance(chunk, Exception):
yield str(chunk)
return Response(cast_str(), mimetype='text/plain')
except Exception as e: