test(test_completion.py): fix claude test

2024-03-04 21:27:48 -08:00 · 2024-03-04 21:27:48 -08:00 · 9069bfb645
commit 9069bfb645
parent f277b204a3
1 changed files with 2 additions and 1 deletions
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@ -56,7 +56,7 @@ def test_completion_custom_provider_model_name():
 def test_completion_claude():
    litellm.set_verbose = True
    litellm.cache = None
-    litellm.AnthropicConfig(max_tokens=200, metadata={"user_id": "1224"})
+    litellm.AnthropicTextConfig(max_tokens=200, metadata={"user_id": "1224"})
    messages = [
        {
            "role": "system",
@ -70,6 +70,7 @@ def test_completion_claude():
            model="claude-instant-1.2",
            messages=messages,
            request_timeout=10,
+            max_tokens=10,
        )
        # Add any assertions, here to check response args
        print(response)