scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

test(test_completion.py): fix claude test

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-04 21:27:48 -08:00
parent f277b204a3
commit 9069bfb645
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/tests/test_completion.py
View file

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