forked from phoenix/litellm-mirror
Update litellm/tests/test_completion.py
This commit is contained in:
parent
6a5ebea383
commit
351264ce1c
1 changed files with 2 additions and 2 deletions
|
@ -12,7 +12,7 @@ messages = [{ "content": user_message,"role": "user"}]
|
|||
|
||||
def test_completion_openai():
|
||||
try:
|
||||
response = completion(model="gpt-3.5-turbo", messages=messages)
|
||||
response = completion(model="updated-model", messages=messages)
|
||||
# Add any assertions here to check the response
|
||||
print(response)
|
||||
except Exception as e:
|
||||
|
@ -80,7 +80,7 @@ def test_completion_azure():
|
|||
|
||||
def test_completion_claude():
|
||||
try:
|
||||
response = completion(model="claude-instant-1", messages=messages)
|
||||
response = completion(model="updated-model", messages=messages)
|
||||
# Add any assertions here to check the response
|
||||
print(response)
|
||||
except Exception as e:
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue