(test) add gpt-4-turbo

This commit is contained in:
ishaan-jaff 2023-11-06 13:32:53 -08:00
parent f7990525ff
commit 8d9e2574cf

View file

@ -110,6 +110,18 @@ def test_completion_with_litellm_call_id():
pytest.fail(f"Error occurred: {e}")
# test_completion_with_litellm_call_id()
def test_completion_gpt4_turbo():
try:
response = completion(
model="gpt-4-1106-preview",
messages=messages,
max_tokens=10,
)
print(response)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_gpt4_turbo()
def test_completion_perplexity_api():
try:
# litellm.set_verbose=True