From 7181f2897be05de1c42fbeb651a952e84f2be82c Mon Sep 17 00:00:00 2001
From: Commenter123321 <36051603+Commenter123321@users.noreply.github.com>
Date: Fri, 29 Sep 2023 16:21:18 +0200
Subject: Fix Aivvm and add new models in models.py

---
 g4f/models.py | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

(limited to 'g4f/models.py')

diff --git a/g4f/models.py b/g4f/models.py
index 7c2d6822..74d938b4 100644
--- a/g4f/models.py
+++ b/g4f/models.py
@@ -59,7 +59,7 @@ gpt_4 = Model(
     name          = 'gpt-4',
     base_provider = 'openai',
     best_provider = RetryProvider([
-        Myshell, AItianhuSpace,
+        Myshell, AItianhuSpace, Aivvm
     ])
 )
 
@@ -149,7 +149,7 @@ code_davinci_002 = Model(
 gpt_35_turbo_16k = Model(
     name          = 'gpt-3.5-turbo-16k',
     base_provider = 'openai',
-    best_provider = Vercel)
+    best_provider = Aivvm)
 
 gpt_35_turbo_16k_0613 = Model(
     name          = 'gpt-3.5-turbo-16k-0613',
@@ -166,7 +166,17 @@ gpt_35_turbo_0613 = Model(
 gpt_4_0613 = Model(
     name          = 'gpt-4-0613',
     base_provider = 'openai',
-    best_provider = Vercel)
+    best_provider = Aivvm)
+
+gpt_4_32k = Model(
+    name          = 'gpt-4-32k',
+    base_provider = 'openai',
+    best_provider = Aivvm)
+
+gpt_4_32k_0613 = Model(
+    name          = 'gpt-4-32k-0613',
+    base_provider = 'openai',
+    best_provider = Aivvm)
 
 text_ada_001 = Model(
     name          = 'text-ada-001',
-- 
cgit v1.2.3