(feat) added support for claude-3-5-haiku

ExtensityAI · Nov 5, 2024 · e1b473c · e1b473c
1 parent 6bcf985
commit e1b473c
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/symai/backend/mixin/anthropic.py b/symai/backend/mixin/anthropic.py
@@ -1,6 +1,7 @@
 SUPPORTED_MODELS = [
     # https://docs.anthropic.com/en/docs/about-claude/models
     'claude-3-5-sonnet-latest',
+    'claude-3-5-haiku-latest',
     'claude-3-5-sonnet-20241022',
     'claude-3-5-sonnet-20240620',
     'claude-3-opus-latest',
@@ -22,10 +23,12 @@ def api_max_context_tokens(self):
 
     def api_max_response_tokens(self):
         if self.model == 'claude-3-5-sonnet-latest' or \
-           self.model == 'claude-3-5-sonnet-20240620' or \
            self.model == 'claude-3-5-sonnet-20241022' or \
+           self.model == 'claude-3-5-haiku-latest':
+            return 8_192
+        if self.model == 'claude-3-5-sonnet-20240620' or \
            self.model == 'claude-3-opus-latest' or \
-           self.model == 'claude-3-opus-20240229' or \
+           self.model == 'clade-3-opus-20240229' or \
            self.model == 'claude-3-sonnet-20240229' or \
            self.model == 'claude-3-haiku-20240307':
             return 4_096