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

test(test_completion.py-+-test_streaming.py): add ollama endpoint to ci/cd pipeline

Browse source
This commit is contained in:
Krrish Dholakia 2023-12-22 12:21:33 +05:30
parent 57607f111a
commit eb2d13e2fb
3 changed files with 47 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

18
litellm/tests/test_completion.py
View file

@ -571,6 +571,22 @@ def test_completion_openai_litellm_key():
# test_completion_openai_litellm_key()
def test_completion_ollama_hosted():
try:
litellm.set_verbose = True
response = completion(
model="ollama/phi",
messages=messages,
max_tokens=10,
api_base="https://test-ollama-endpoint.onrender.com"
)
# Add any assertions here to check the response
print(response)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_ollama_hosted()
def test_completion_openrouter1():
try:
response = completion(
@ -626,7 +642,7 @@ def test_completion_anyscale_with_functions():
print(response)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_completion_anyscale_with_functions()
# test_completion_anyscale_with_functions()
def test_completion_azure_key_completion_arg():
# this tests if we can pass api_key to completion, when it's not in the env