phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(test) test seed, response format, for gpt-3.5-turbo

Browse source
This commit is contained in:
ishaan-jaff 2023-11-17 14:00:42 -08:00
parent 7abb65d53f
commit f026ed4772
1 changed files with 11 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

15
litellm/tests/test_completion.py
View file

@ -434,10 +434,17 @@ def test_completion_openai_with_optional_params():
try:
litellm.set_verbose = True
response = completion(
model="gpt-3.5-turbo",
messages=messages,
model="gpt-3.5-turbo-1106",
messages=[
{
"role": "user",
"content": "respond in valid, json - what is the day"
}
],
temperature=0.5,
top_p=0.1,
seed=12,
response_format={ "type": "json_object" }
)
# Add any assertions here to check the response
print(response)
@ -446,7 +453,7 @@ def test_completion_openai_with_optional_params():
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_openai_with_optional_params()
test_completion_openai_with_optional_params()
def test_completion_openai_litellm_key():
try:
@ -494,7 +501,7 @@ def test_completion_openrouter1():
print(response)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_completion_openrouter1()
# test_completion_openrouter1()
def test_completion_hf_model_no_provider():
try: