diff options
Diffstat (limited to 'g4f/gui/server')
| -rw-r--r-- | g4f/gui/server/backend.py | 25 | ||||
| -rw-r--r-- | g4f/gui/server/internet.py | 2 |
2 files changed, 11 insertions, 16 deletions
diff --git a/g4f/gui/server/backend.py b/g4f/gui/server/backend.py index 00f2a827..67f13de4 100644 --- a/g4f/gui/server/backend.py +++ b/g4f/gui/server/backend.py @@ -1,19 +1,14 @@ import logging -import g4f +import json +from flask import request, Flask +from g4f import debug, version, models +from g4f import _all_models, get_last_provider, ChatCompletion from g4f.Provider import __providers__ +from g4f.Provider.bing.create_images import patch_provider +from .internet import get_search_message -import json -from flask import request, Flask -from .internet import get_search_message -from g4f import debug, version -from g4f.base_provider import ProviderType debug.logging = True -def patch_provider(provider: ProviderType): - from g4f.Provider import CreateImagesProvider - from g4f.Provider.bing.create_images import create_completion - return CreateImagesProvider(provider, create_completion) - class Backend_Api: def __init__(self, app: Flask) -> None: self.app: Flask = app @@ -50,7 +45,7 @@ class Backend_Api: return 'ok', 200 def models(self): - return g4f._all_models + return _all_models def providers(self): return [ @@ -74,7 +69,7 @@ class Backend_Api: if request.json.get('internet_access'): messages[-1]["content"] = get_search_message(messages[-1]["content"]) model = request.json.get('model') - model = model if model else g4f.models.default + model = model if model else models.default provider = request.json.get('provider', '').replace('g4f.Provider.', '') provider = provider if provider and provider != "Auto" else None patch = patch_provider if request.json.get('patch_provider') else None @@ -82,7 +77,7 @@ class Backend_Api: def try_response(): try: first = True - for chunk in g4f.ChatCompletion.create( + for chunk in ChatCompletion.create( model=model, provider=provider, messages=messages, @@ -94,7 +89,7 @@ class Backend_Api: first = False yield json.dumps({ 'type' : 'provider', - 'provider': g4f.get_last_provider(True) + 'provider': get_last_provider(True) }) + "\n" yield json.dumps({ 'type' : 'content', diff --git a/g4f/gui/server/internet.py b/g4f/gui/server/internet.py index 97f842d7..6c2e3a89 100644 --- a/g4f/gui/server/internet.py +++ b/g4f/gui/server/internet.py @@ -145,5 +145,5 @@ User request: """ return message except Exception as e: - print("Couldn't search DuckDuckGo:", e) + print("Couldn't do web search:", e) return prompt |