From 6ef282de3a3245acbfecd08ae48dba85ff91d031 Mon Sep 17 00:00:00 2001
From: H Lohaus <hlohaus@users.noreply.github.com>
Date: Tue, 12 Mar 2024 02:06:06 +0100
Subject: Remove all not working provider (#1679)

Fix many providers
Add selenium-wire to requierments
---
 g4f/Provider/Vercel.py | 22 +++++++++++-----------
 1 file changed, 11 insertions(+), 11 deletions(-)

(limited to 'g4f/Provider/Vercel.py')

diff --git a/g4f/Provider/Vercel.py b/g4f/Provider/Vercel.py
index e10aa232..f8faeeaf 100644
--- a/g4f/Provider/Vercel.py
+++ b/g4f/Provider/Vercel.py
@@ -8,17 +8,18 @@ try:
 except ImportError:
     has_requirements = False
 
-from ..typing       import Messages, TypedDict, CreateResult, Any
+from ..typing       import Messages, CreateResult
 from .base_provider import AbstractProvider
-from ..errors       import MissingRequirementsError
+from ..requests     import raise_for_status
+from ..errors       import MissingRequirementsError, RateLimitError, ResponseStatusError
 
 class Vercel(AbstractProvider):
     url = 'https://chat.vercel.ai'
     working = True
-    supports_message_history = True 
+    supports_message_history = True
+    supports_system_message  = True
     supports_gpt_35_turbo    = True
     supports_stream          = True
-    supports_gpt_4           = False
     
     @staticmethod
     def create_completion(
@@ -26,6 +27,7 @@ class Vercel(AbstractProvider):
         messages: Messages,
         stream: bool,
         proxy: str = None,
+        max_retries: int = 6,
         **kwargs
     ) -> CreateResult:
         if not has_requirements:
@@ -54,19 +56,17 @@ class Vercel(AbstractProvider):
             'messages': messages,
             'id'      : f'{os.urandom(3).hex()}a',
         }
-        
-        max_retries  = kwargs.get('max_retries', 6)
+        response = None
         for _ in range(max_retries):
             response = requests.post('https://chat.vercel.ai/api/chat', 
                                     headers=headers, json=json_data, stream=True, proxies={"https": proxy})
-            try:
-                response.raise_for_status()
-            except:
+            if not response.ok:
                 continue
             for token in response.iter_content(chunk_size=None):
                 yield token.decode()
             break
-        
+        raise_for_status(response)
+
 def get_anti_bot_token() -> str:
     headers = {
         'authority': 'sdk.vercel.ai',
@@ -92,7 +92,7 @@ def get_anti_bot_token() -> str:
 
     js_script = '''const globalThis={marker:"mark"};String.prototype.fontcolor=function(){return `<font>${this}</font>`};
         return (%s)(%s)''' % (raw_data['c'], raw_data['a'])
-        
+
     sec_list = [execjs.compile(js_script).call('')[0], [], "sentinel"]
 
     raw_token = json.dumps({'r': sec_list, 't': raw_data['t']}, 
-- 
cgit v1.2.3