From 42c4a4315a1f768f522508aa3cdc44b391f9a1a8 Mon Sep 17 00:00:00 2001
From: Debaditya Banerji <77636021+devAdityaa@users.noreply.github.com>
Date: Sat, 13 Jan 2024 20:28:12 +0530
Subject: Added support for more models: FreeChatgpt (#1456)

* Improved FreeChatgpt, added support for more models

* Added new model: Gemini Pro
---
 g4f/Provider/FreeChatgpt.py | 77 +++++++++++++++++++++------------------------
 1 file changed, 35 insertions(+), 42 deletions(-)

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

diff --git a/g4f/Provider/FreeChatgpt.py b/g4f/Provider/FreeChatgpt.py
index 05a0016e..82fc3ef4 100644
--- a/g4f/Provider/FreeChatgpt.py
+++ b/g4f/Provider/FreeChatgpt.py
@@ -7,6 +7,15 @@ from ..typing import AsyncResult, Messages
 from .base_provider import AsyncGeneratorProvider
 
 
+
+models = {
+     "claude-v1":"claude-2.1",
+     "claude-v2":"claude-2.0",
+     "gpt_35_turbo":"gpt-3.5-turbo-1106",
+     "gpt-4":"gpt-4",
+     "gemini-pro":"google-gemini-pro"
+}
+
 class FreeChatgpt(AsyncGeneratorProvider):
     url = "https://free.chatgpt.org.uk"
     working = True
@@ -22,47 +31,31 @@ class FreeChatgpt(AsyncGeneratorProvider):
         proxy: str = None,
         **kwargs
     ) -> AsyncResult:
-        if not model:
-            model = "gpt-3.5-turbo"
+        model = models[model] if model in models else "gpt-3.5-turbo-1106"
         headers = {
-            "User-Agent": "Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:109.0) Gecko/20100101 Firefox/119.0",
-            "Accept": "application/json, text/event-stream",
-            "Accept-Language": "de,en-US;q=0.7,en;q=0.3",
-            "Accept-Encoding": "gzip, deflate, br",
-            "Content-Type": "application/json",
-            "Referer": "https://free.chatgpt.org.uk/",
-            "x-requested-with": "XMLHttpRequest",
-            "Origin": "https://free.chatgpt.org.uk",
-            "Sec-Fetch-Dest": "empty",
-            "Sec-Fetch-Mode": "cors",
-            "Sec-Fetch-Site": "same-origin",
-            "Connection": "keep-alive",
-            "Alt-Used": "free.chatgpt.org.uk",
-            "Pragma": "no-cache",
-            "Cache-Control": "no-cache",
-            "TE": "trailers",
-        }
+    "Accept": "application/json, text/event-stream",
+    "Content-Type":"application/json",
+    "Accept-Encoding": "gzip, deflate, br",
+    "Accept-Language": "en-US,en;q=0.5",
+    "Host":"free.chatgpt.org.uk",
+    "Referer":f"{cls.url}/",
+    "Origin":f"{cls.url}",
+    "Sec-Fetch-Dest": "empty",
+    "Sec-Fetch-Mode": "cors",
+    "Sec-Fetch-Site": "same-origin",
+    "User-Agent": "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36", 
+}
         async with ClientSession(headers=headers) as session:
-            data = {
-                "messages": messages,
-                "isAzure": False,
-                "azureApiVersion": "2023-08-01-preview",
-                "stream": True,
-                "model": model,
-                "temperature": 0.5,
-                "presence_penalty": 0,
-                "frequency_penalty": 0,
-                "top_p": 1,
-                "baseUrl": "/api/openai",
-                "maxIterations": 10,
-                "returnIntermediateSteps": True,
-                "useTools": ["web-search", "calculator", "web-browser"],
-                **kwargs
-            }
-            async with session.post(f"{cls.url}/api/langchain/tool/agent/nodejs", json=data, proxy=proxy) as response:
-                response.raise_for_status()
-                async for line in response.content:
-                    if line.startswith(b"data: "):
-                        data = json.loads(line[6:])
-                        if data["isSuccess"] and not data["isToolMessage"]:
-                            yield data["message"]
\ No newline at end of file
+            data = {"messages":messages,"stream":True,"model":model,"temperature":0.5,"presence_penalty":0,"frequency_penalty":0,"top_p":1}
+            async with session.post(f'{cls.url}/api/openai/v1/chat/completions',json=data) as result:
+                async for chunk in result.content:
+                    
+                    line = chunk.decode()
+                    if line.startswith("data: [DONE]"):
+                            break
+                    elif line.startswith("data: "):
+                         line = json.loads(line[6:])
+                         if(line["choices"]==[]):
+                              continue
+                         if(line["choices"][0]["delta"].get("content") and line["choices"][0]["delta"]["content"]!=None):
+                              yield line["choices"][0]["delta"]["content"]
\ No newline at end of file
-- 
cgit v1.2.3