From 76956cce521f88e7222fa623bdf875b15cf84987 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 29 Sep 2023 15:37:05 -0700 Subject: [PATCH] update claude testing --- litellm/tests/test_completion.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index e2b29c2da1..760ffa2d92 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -784,13 +784,15 @@ def test_completion_bedrock_claude(): response = completion( model="bedrock/anthropic.claude-instant-v1", messages=messages, + max_tokens=10, + temperature=0.1, logger_fn=logger_fn ) # Add any assertions here to check the response print(response) except Exception as e: pytest.fail(f"Error occurred: {e}") -# test_completion_bedrock_claude() +test_completion_bedrock_claude() def test_completion_bedrock_claude_stream(): print("calling claude") @@ -807,7 +809,7 @@ def test_completion_bedrock_claude_stream(): print(chunk) except Exception as e: pytest.fail(f"Error occurred: {e}") -test_completion_bedrock_claude_stream() +# test_completion_bedrock_claude_stream() def test_completion_bedrock_ai21(): try: