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():
|
def test_completion_openai():
|
||||||
try:
|
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
|
# Add any assertions here to check the response
|
||||||
print(response)
|
print(response)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
@ -80,7 +80,7 @@ def test_completion_azure():
|
||||||
|
|
||||||
def test_completion_claude():
|
def test_completion_claude():
|
||||||
try:
|
try:
|
||||||
response = completion(model="claude-instant-1", messages=messages)
|
response = completion(model="updated-model", messages=messages)
|
||||||
# Add any assertions here to check the response
|
# Add any assertions here to check the response
|
||||||
print(response)
|
print(response)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue