Update Claude models

open-cogsci · Jan 10, 2025 · 0c459fa · 0c459fa
1 parent 3a15074
commit 0c459fa
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/sigmund/model/__init__.py b/sigmund/model/__init__.py
@@ -23,10 +23,13 @@ def model(sigmund, model, **kwargs):
         return AnthropicModel(sigmund, 'claude-3-sonnet-20240229', **kwargs)
     if model == 'claude-3.5-sonnet':
         from ._anthropic_model import AnthropicModel
-        return AnthropicModel(sigmund, 'claude-3-5-sonnet-20240620', **kwargs)
+        return AnthropicModel(sigmund, 'claude-3-5-sonnet-latest', **kwargs)
     if model == 'claude-3-haiku':
         from ._anthropic_model import AnthropicModel
         return AnthropicModel(sigmund, 'claude-3-haiku-20240307', **kwargs)
+    if model == 'claude-3.5-haiku':
+        from ._anthropic_model import AnthropicModel
+        return AnthropicModel(sigmund, 'claude-3-5-haiku-latest', **kwargs)
     if 'mistral' in model:
         from ._mistral_model import MistralModel
         return MistralModel(sigmund, model, **kwargs)