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

(feat) claude-3 test fixes

Browse source
This commit is contained in:
ishaan-jaff 2024-03-04 07:53:06 -08:00
parent 25bbb73ce3
commit d179ae376e
3 changed files with 21 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
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_to_sample=200, metadata={"user_id": "1224"})
litellm.AnthropicConfig(max_tokens=200, metadata={"user_id": "1224"})
messages = [
{
"role": "system",
@ -67,7 +67,7 @@ def test_completion_claude():
try:
# test without max tokens
response = completion(
model="claude-instant-1",
model="claude-instant-1.2",
messages=messages,
request_timeout=10,
)