phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(fix) claude-instant-1 test

Browse source
This commit is contained in:
ishaan-jaff 2024-03-04 21:39:12 -08:00
parent 5cb854720f
commit 6a28d0f882
2 changed files with 6 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

7
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.AnthropicTextConfig(max_tokens=200, metadata={"user_id": "1224"})
litellm.AnthropicTextConfig(max_tokens_to_sample=200, metadata={"user_id": "1224"})
messages = [
{
"role": "system",
@ -67,10 +67,7 @@ def test_completion_claude():
try:
# test without max tokens
response = completion(
model="claude-instant-1.2",
messages=messages,
request_timeout=10,
max_tokens=10,
model="claude-instant-1", messages=messages, request_timeout=10
)
# Add any assertions, here to check response args
print(response)