From 9069bfb645594ea78eeffbc689d5c86c37d45da9 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Mon, 4 Mar 2024 21:27:48 -0800
Subject: [PATCH] test(test_completion.py): fix claude test

---
 litellm/tests/test_completion.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index 36ca7b8b03..b61a865d26 100644
--- 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)