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

(test) claude-instant-1

Browse source
This commit is contained in:
ishaan-jaff 2024-03-04 08:32:13 -08:00
parent 351c648dc0
commit b5fcc89682
3 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_provider_specific_config.py
View file

@ -53,7 +53,7 @@ def claude_test_completion():
try:
# OVERRIDE WITH DYNAMIC MAX TOKENS
response_1 = litellm.completion(
model="claude-instant-1",
model="claude-instant-1.2",
messages=[{"content": "Hello, how are you?", "role": "user"}],
max_tokens=10,
)
@ -63,7 +63,7 @@ def claude_test_completion():
# USE CONFIG TOKENS
response_2 = litellm.completion(
model="claude-instant-1",
model="claude-instant-1.2",
messages=[{"content": "Hello, how are you?", "role": "user"}],
)
# Add any assertions here to check the response
@ -74,7 +74,7 @@ def claude_test_completion():
try:
response_3 = litellm.completion(
model="claude-instant-1",
model="claude-instant-1.2",
messages=[{"content": "Hello, how are you?", "role": "user"}],
n=2,
)

2
litellm/tests/test_router.py
View file

@ -933,7 +933,7 @@ def test_router_anthropic_key_dynamic():
{
"model_name": "anthropic-claude",
"litellm_params": {
"model": "claude-instant-1",
"model": "claude-instant-1.2",
"api_key": anthropic_api_key,
},
}

2
litellm/tests/test_router_timeout.py
View file

@ -35,7 +35,7 @@ def test_router_timeouts():
{
"model_name": "anthropic-claude-instant-1.2",
"litellm_params": {
"model": "claude-instant-1",
"model": "claude-instant-1.2",
"api_key": "os.environ/ANTHROPIC_API_KEY",
},
"tpm": 20000,