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/models.py | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

(limited to 'g4f/models.py')

diff --git a/g4f/models.py b/g4f/models.py
index 21e85c72..90d9f444 100644
--- a/g4f/models.py
+++ b/g4f/models.py
@@ -69,8 +69,9 @@ gpt_35_long = Model(
 gpt_35_turbo = Model(
     name          = 'gpt-3.5-turbo',
     base_provider = 'openai',
-    best_provider=RetryProvider([
-        GptGo, You, 
+    best_provider=RetryProvider([ 
+        FreeChatgpt,
+        GptGo, You,
         GptForLove, ChatBase,
         Chatgpt4Online,
     ])
@@ -80,7 +81,7 @@ gpt_4 = Model(
     name          = 'gpt-4',
     base_provider = 'openai',
     best_provider = RetryProvider([
-        Bing, Phind, Liaobots
+        Bing, FreeChatgpt, Phind, Liaobots, 
     ])
 )
 
@@ -158,12 +159,12 @@ claude_instant_v1 = Model(
 claude_v1 = Model(
     name          = 'claude-v1',
     base_provider = 'anthropic',
-    best_provider = Vercel)
+    best_provider = RetryProvider([FreeChatgpt,Vercel]))
 
 claude_v2 = Model(
     name          = 'claude-v2',
     base_provider = 'anthropic',
-    best_provider = Vercel)
+    best_provider = RetryProvider([FreeChatgpt,Vercel]))
 
 command_light_nightly = Model(
     name          = 'command-light-nightly',
@@ -245,6 +246,13 @@ gpt_4_32k_0613 = Model(
     best_provider = gpt_4.best_provider
 )
 
+#Gemini
+gemini_pro = Model(
+    name          = 'gemini-pro',
+    base_provider = 'google',
+    best_provider = FreeChatgpt
+)
+
 text_ada_001 = Model(
     name          = 'text-ada-001',
     base_provider = 'openai',
@@ -318,6 +326,8 @@ class ModelUtils:
         'mistral-7b': mistral_7b,
         'openchat_3.5': openchat_35,
         
+        # Gemini Pro
+        'gemini-pro': gemini_pro,
         # Bard
         'palm2'       : palm,
         'palm'        : palm,
-- 
cgit v1.2.3