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

2
litellm/tests/test_router.py
View file

@ -933,7 +933,7 @@ def test_router_anthropic_key_dynamic():
{ {
"model_name": "anthropic-claude", "model_name": "anthropic-claude",
"litellm_params": { "litellm_params": {
"model": "claude-instant-1", "model": "claude-instant-1.2",
"api_key": anthropic_api_key, "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", "model_name": "anthropic-claude-instant-1.2",
"litellm_params": { "litellm_params": {
"model": "claude-instant-1", "model": "claude-instant-1.2",
"api_key": "os.environ/ANTHROPIC_API_KEY", "api_key": "os.environ/ANTHROPIC_API_KEY",
}, },
"tpm": 20000, "tpm": 20000,