(test) claude-instant-1

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

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,
)