tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-13 20:23:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

(test) claude-instant-1

Browse Source
This commit is contained in:
ishaan-jaff
2024-03-04 08:32:13 -08:00
parent 3e8cecc03b
commit 9bac163e4e
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_provider_specific_config.py
+3 -3
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,
)
litellm/tests/test_router.py
+1 -1
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,
},
}
litellm/tests/test_router_timeout.py
+1 -1
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,