diff --git a/sigmund/model/__init__.py b/sigmund/model/__init__.py index 717fa7a..5937a79 100644 --- 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)