feat: 100k model support for Claude

ciuzaak · May 12, 2023 · 9ed5990 · 9ed5990
1 parent 131b817
commit 9ed5990
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/utils/claude_utils.py b/utils/claude_utils.py
@@ -15,8 +15,19 @@ def reset(self):
         self.prompt = ''
 
     def change_model(self, model):
-        valid_models = {'claude-v1', 'claude-v1.0', 'claude-v1.2',
-                        'claude-v1.3', 'claude-instant-v1', 'claude-instant-v1.0'}
+        valid_models = {
+            'claude-v1',
+            'claude-v1-100k',
+            'claude-instant-v1',
+            'claude-instant-v1-100k',
+            'claude-v1.3',
+            'claude-v1.3-100k',
+            'claude-v1.2',
+            'claude-v1.0',
+            'claude-instant-v1.1',
+            'claude-instant-v1.1-100k',
+            'claude-instant-v1.0'
+        }
         if model in valid_models:
             self.model = model
             return True