diff options
author | Heiner Lohaus <hlohaus@users.noreply.github.com> | 2023-12-11 00:56:06 +0100 |
---|---|---|
committer | Heiner Lohaus <hlohaus@users.noreply.github.com> | 2023-12-11 00:56:06 +0100 |
commit | d743ee0c26f00916233d751f2f375be414487d49 (patch) | |
tree | 4377351363d186fefdf7d1eed27a4fde39f173a3 /g4f/gui | |
parent | Improve error handling in gui (diff) | |
download | gpt4free-d743ee0c26f00916233d751f2f375be414487d49.tar gpt4free-d743ee0c26f00916233d751f2f375be414487d49.tar.gz gpt4free-d743ee0c26f00916233d751f2f375be414487d49.tar.bz2 gpt4free-d743ee0c26f00916233d751f2f375be414487d49.tar.lz gpt4free-d743ee0c26f00916233d751f2f375be414487d49.tar.xz gpt4free-d743ee0c26f00916233d751f2f375be414487d49.tar.zst gpt4free-d743ee0c26f00916233d751f2f375be414487d49.zip |
Diffstat (limited to 'g4f/gui')
-rw-r--r-- | g4f/gui/server/backend.py | 18 |
1 files changed, 8 insertions, 10 deletions
diff --git a/g4f/gui/server/backend.py b/g4f/gui/server/backend.py index 9dc5dbb4..e4669699 100644 --- a/g4f/gui/server/backend.py +++ b/g4f/gui/server/backend.py @@ -4,8 +4,9 @@ from g4f.Provider import __providers__ import json from flask import request, Flask from .internet import get_search_message +from g4f import debug -g4f.debug.logging = True +debug.logging = True class Backend_Api: def __init__(self, app: Flask) -> None: @@ -52,8 +53,8 @@ class Backend_Api: def version(self): return { - "version": g4f.get_version(), - "lastet_version": g4f.get_lastet_version(), + "version": debug.get_version(), + "lastet_version": debug.get_lastet_version(), } def _gen_title(self): @@ -68,28 +69,25 @@ class Backend_Api: messages[-1]["content"] = get_search_message(messages[-1]["content"]) model = request.json.get('model') model = model if model else g4f.models.default - provider = request.json.get('provider').replace('g4f.Provider.', '') + provider = request.json.get('provider', '').replace('g4f.Provider.', '') provider = provider if provider and provider != "Auto" else None - if provider != None: - provider = g4f.Provider.ProviderUtils.convert.get(provider) def try_response(): try: - response = g4f.ChatCompletion.create( + yield from g4f.ChatCompletion.create( model=model, provider=provider, messages=messages, stream=True, ignore_stream_and_auth=True ) - yield from response except Exception as e: print(e) yield json.dumps({ 'code' : 'G4F_ERROR', '_action': '_ask', 'success': False, - 'error' : f'an error occurred {str(e)}' - }) + 'error' : f'{e.__class__.__name__}: {e}' + }) return self.app.response_class(try_response(), mimetype='text/event-stream')
\ No newline at end of file |