From e9467297369dd4f429c084bd1d034f3b1b63e7c2 Mon Sep 17 00:00:00 2001
From: abc <98614666+xtekky@users.noreply.github.com>
Date: Sun, 16 Jul 2023 20:31:51 +0100
Subject: minor fixes

---
 g4f/models.py | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

(limited to 'g4f/models.py')

diff --git a/g4f/models.py b/g4f/models.py
index ecf18e6d..95be4849 100644
--- a/g4f/models.py
+++ b/g4f/models.py
@@ -152,11 +152,42 @@ class Model:
         name: str = 'llama-13b'
         base_provider: str = 'huggingface'
         best_provider: Provider.Provider = Provider.H2o
+        
+    class gpt_35_turbo_16k:
+        name: str = 'gpt-3.5-turbo-16k'
+        base_provider: str = 'openai'
+        best_provider: Provider.Provider = Provider.EasyChat
+        
+    class gpt_35_turbo_0613:
+        name: str = 'gpt-3.5-turbo-0613'
+        base_provider: str = 'openai'
+        best_provider: Provider.Provider = Provider.EasyChat
+        
+    class gpt_35_turbo_16k_0613:
+        name: str = 'gpt-3.5-turbo-16k-0613'
+        base_provider: str = 'openai'
+        best_provider: Provider.Provider = Provider.EasyChat
+        
+    class gpt_4_32k:
+        name: str = 'gpt-4-32k'
+        base_provider: str = 'openai'
+        best_provider =  None
+        
+    class gpt_4_0613:
+        name: str = 'gpt-4-0613'
+        base_provider: str = 'openai'
+        best_provider = None
     
 class ModelUtils:
     convert: dict = {
         'gpt-3.5-turbo': Model.gpt_35_turbo,
+        'gpt-3.6-turbo-16k': Model.gpt_35_turbo_16k,
+        'gpt-3.5-turbo-0613': Model.gpt_35_turbo_0613,
+        'gpt-3.5-turbo-16k-0613': Model.gpt_35_turbo_16k_0613,
+        
         'gpt-4': Model.gpt_4,
+        'gpt-4-32k': Model.gpt_4_32k,
+        'gpt-4-0613': Model.gpt_4_0613,
         
         'claude-instant-v1-100k': Model.claude_instant_v1_100k,
         'claude-v1-100k': Model.claude_v1_100k,
-- 
cgit v1.2.3