From 3576dee75a1623aa2385b6afe8b922ad5affca26 Mon Sep 17 00:00:00 2001
From: Heiner Lohaus <hlohaus@users.noreply.github.com>
Date: Wed, 6 Dec 2023 09:35:36 +0100
Subject: Add selenium to dockerfile Load model and provider list in gui Remove
 needs_auth in HuggingChat Add default model and login url in gui

---
 g4f/gui/server/backend.py | 44 +++++++++++++++++++++++++-------------------
 1 file changed, 25 insertions(+), 19 deletions(-)

(limited to 'g4f/gui/server/backend.py')

diff --git a/g4f/gui/server/backend.py b/g4f/gui/server/backend.py
index 3d7bfedc..03363201 100644
--- a/g4f/gui/server/backend.py
+++ b/g4f/gui/server/backend.py
@@ -3,9 +3,8 @@ import g4f
 from flask      import request
 from .internet  import search
 from .config    import special_instructions
-from .provider  import get_provider
 
-g4f.logging = True
+g4f.debug.logging = True
 
 class Backend_Api:
     def __init__(self, app) -> None:
@@ -15,6 +14,10 @@ class Backend_Api:
                 'function': self.models,
                 'methods' : ['GET']
             },
+            '/backend-api/v2/providers': {
+                'function': self.providers,
+                'methods' : ['GET']
+            },
             '/backend-api/v2/conversation': {
                 'function': self._conversation,
                 'methods': ['POST']
@@ -37,6 +40,9 @@ class Backend_Api:
     def models(self):
         return g4f._all_models
     
+    def providers(self):
+        return [provider.__name__ for provider in g4f.Provider.__providers__ if provider.working]
+    
     def _gen_title(self):
         return {
             'title': ''
@@ -47,26 +53,26 @@ class Backend_Api:
             #jailbreak       = request.json['jailbreak']
             #internet_access = request.json['meta']['content']['internet_access']
             #conversation    = request.json['meta']['content']['conversation']
-            prompt          = request.json['meta']['content']['parts']
-            model           = request.json['model']
-            provider        = request.json.get('provider').split('g4f.Provider.')[1]
-
-            messages = prompt
-            print(messages)
+            messages = request.json['meta']['content']['parts']
+            model = request.json.get('model')
+            model = model if model else g4f.models.default
+            provider = request.json.get('provider', 'Auto').replace('g4f.Provider.', '')
+            provider = provider if provider != "Auto" else None
+            if provider != None:
+                provider = g4f.Provider.ProviderUtils.convert.get(provider)
 
-            def stream():
-                yield from g4f.ChatCompletion.create(
-                    model=model,
-                    provider=get_provider(provider),
-                    messages=messages,
-                    stream=True,
-                ) if provider else g4f.ChatCompletion.create(
-                    model=model, messages=messages, stream=True
-                )
+            response = g4f.ChatCompletion.create(
+                model=model,
+                provider=provider,
+                messages=messages,
+                stream=True,
+                ignore_stream_and_auth=True
+            )
 
-            return self.app.response_class(stream(), mimetype='text/event-stream')
+            return self.app.response_class(response, mimetype='text/event-stream')
 
-        except Exception as e:    
+        except Exception as e:
+            print(e)
             return {
                 'code'   : 'G4F_ERROR',
                 '_action': '_ask',
-- 
cgit v1.2.3