From 66f6d3e9583f56bf67ea5d836c296ee18d806c16 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Thu, 5 Oct 2023 23:07:26 -0700
Subject: [PATCH] fix(test_completion.py): fix testing for claude

---
 litellm/tests/test_completion.py | 28 +---------------------------
 1 file changed, 1 insertion(+), 27 deletions(-)

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index c2c139d05bc..95434c0e763 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -38,7 +38,7 @@ def test_completion_custom_provider_model_name():
 
 def test_completion_claude():
     litellm.set_verbose = True
-    litellm.anthropic_config(max_tokens_to_sample=200, metadata={"user_id": "1224"})
+    litellm.AnthropicConfig(max_tokens_to_sample=200, metadata={"user_id": "1224"})
     try:
         # test without max tokens
         response = completion(
@@ -52,32 +52,6 @@ def test_completion_claude():
 
 # test_completion_claude()
 
-def test_completion_claude_max_tokens():
-    try:
-        litellm.set_verbose = True
-        # Redirect stdout
-        old_stdout = sys.stdout
-        sys.stdout = new_stdout = io.StringIO()
-
-        # test setting max tokens for claude-2
-        user_message = "tell me everything about YC - be verbose"
-        messages = [{"content": user_message, "role": "user"}]
-        litellm.set_verbose = True
-        response = completion(
-            model="claude-instant-1", messages=messages, max_tokens=1200
-        )
-        # Restore stdout
-        sys.stdout = old_stdout
-        output = new_stdout.getvalue().strip()
-        # Add any assertions here to check the response
-        if "LiteLLM.Anthropic: Max Tokens Set" not in output:
-            raise Exception("Required log message not found!")
-        print(f"response: {response}")
-        litellm.set_verbose = False
-    except Exception as e:
-        pytest.fail(f"Error occurred: {e}")
-# test_completion_claude_max_tokens()
-
 # def test_completion_oobabooga():
 #     try:
 #         response = completion(