Add claude-3-5-sonnet-20240620, refs #7

Also fix broken test
simonw · Jun 20, 2024 · 8dbe06a · 8dbe06a
1 parent 8911982
commit 8dbe06a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/llm_claude_3.py b/llm_claude_3.py
@@ -10,6 +10,7 @@ def register_models(register):
     register(ClaudeMessages("claude-3-opus-20240229"), aliases=("claude-3-opus",))
     register(ClaudeMessages("claude-3-sonnet-20240229"), aliases=("claude-3-sonnet",))
     register(ClaudeMessages("claude-3-haiku-20240307"), aliases=("claude-3-haiku",))
+    register(ClaudeMessages("claude-3-5-sonnet-20240620"), aliases=("claude-3-sonnet-3.5",))
 
 
 class ClaudeOptions(llm.Options):

diff --git a/tests/test_claude_3.py b/tests/test_claude_3.py
@@ -17,5 +17,5 @@ def test_prompt():
         "stop_reason": "end_turn",
         "stop_sequence": None,
         "type": "message",
-        "usage": {"input_tokens": 17, "output_tokens": 1},
+        "usage": {"input_tokens": 17, "output_tokens": 15},
     }