From 8964d15dc36338e994285552e2b49435428ca50b Mon Sep 17 00:00:00 2001
From: abc <98614666+xtekky@users.noreply.github.com>
Date: Tue, 10 Oct 2023 15:48:56 +0100
Subject: ~ | g4f `v-0.1.5.8`

---
 g4f/Provider/Aivvm.py        |  2 +-
 g4f/__init__.py              |  2 +-
 g4f/gui/client/js/chat.v2.js | 11 +----------
 g4f/gui/server/backend.py    |  5 +++--
 4 files changed, 6 insertions(+), 14 deletions(-)

(limited to 'g4f')

diff --git a/g4f/Provider/Aivvm.py b/g4f/Provider/Aivvm.py
index 05f12320..a9eede67 100644
--- a/g4f/Provider/Aivvm.py
+++ b/g4f/Provider/Aivvm.py
@@ -65,7 +65,7 @@ class Aivvm(BaseProvider):
         response = requests.post("https://chat.aivvm.com/api/chat", headers=headers, data=data, stream=True)
         response.raise_for_status()
 
-        for chunk in response.iter_content():
+        for chunk in response.iter_content(chunk_size=4096):
             try:
                 yield chunk.decode("utf-8")
             except UnicodeDecodeError:
diff --git a/g4f/__init__.py b/g4f/__init__.py
index fa20f6eb..dc248fde 100644
--- a/g4f/__init__.py
+++ b/g4f/__init__.py
@@ -5,7 +5,7 @@ from .Provider import BaseProvider
 from .typing import Messages, CreateResult, Union
 from .debug import logging
 
-version = '0.1.5.7'
+version = '0.1.5.8'
 version_check = True
 
 def check_pypi_version() -> None:
diff --git a/g4f/gui/client/js/chat.v2.js b/g4f/gui/client/js/chat.v2.js
index 24f2e45b..f052b7e6 100644
--- a/g4f/gui/client/js/chat.v2.js
+++ b/g4f/gui/client/js/chat.v2.js
@@ -144,10 +144,6 @@ const ask_gpt = async (message) => {
 
             chunk = new TextDecoder().decode(value);
 
-            if (chunk.includes('<form id="challenge-form" action="/backend-api/v2/conversation?"')) {
-                chunk = `cloudflare token expired, please refresh the page.`;
-            }
-
             text += chunk;
 
             document.getElementById(`gpt_${window.token}`).innerHTML =
@@ -160,12 +156,7 @@ const ask_gpt = async (message) => {
             message_box.scrollTo({ top: message_box.scrollHeight, behavior: "auto" });
         }
 
-        // if text contains :
-        if (text.includes("instead. Maintaining this website and API costs a lot of money")) {
-            document.getElementById(`gpt_${window.token}`).innerHTML = "An error occured, please reload / refresh cache and try again or use a differnet browser";
-        }
-
-        if (text.includes(`anerroroccuredmf`)) {
+        if (text.includes(`G4F_ERROR`)) {
             document.getElementById(`gpt_${window.token}`).innerHTML = "An error occured, please try again, if the problem persists, please reload / refresh cache or use a differnet browser";
         }
 
diff --git a/g4f/gui/server/backend.py b/g4f/gui/server/backend.py
index a76ca12b..7b5e9f0d 100644
--- a/g4f/gui/server/backend.py
+++ b/g4f/gui/server/backend.py
@@ -33,7 +33,7 @@ class Backend_Api:
             conversation    = request.json['meta']['content']['conversation']
             prompt          = request.json['meta']['content']['parts'][0]
             model           = request.json['model']
-            provider        = request.json.get('provider').split("g4f.Provider.")[1]
+            provider        = request.json.get('provider').split('g4f.Provider.')[1]
             
             messages = special_instructions[jailbreak] + conversation + search(internet_access, prompt) + [prompt]
             
@@ -52,6 +52,7 @@ class Backend_Api:
 
         except Exception as e:    
             return {
+                'code'   : 'G4F_ERROR',
                 '_action': '_ask',
                 'success': False,
-                "error": f"an error occured {str(e)}"}, 400
\ No newline at end of file
+                'error'  : f'an error occured {str(e)}'}, 400
\ No newline at end of file
-- 
cgit v1.2.3