phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-28 04:04:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update claude testing

Browse source
This commit is contained in:
ishaan-jaff 2023-09-29 15:37:05 -07:00
parent aaa5ac0488
commit f24fe309e2
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_completion.py
View file

@ -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: