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

(test) ollama json mode

Browse source
This commit is contained in:
ishaan-jaff 2023-12-25 14:00:56 +05:30
parent b3a98d8400
commit b87d630b0a
1 changed files with 11 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/tests/test_ollama.py
View file

@ -35,3 +35,13 @@ def test_get_ollama_model():
pytest.fail(f"Error occurred: {e}")
# test_get_ollama_model()
def test_ollama_json_mode():
# assert that format: json gets passed as is to ollama
try:
converted_params = get_optional_params(custom_llm_provider="ollama", model="llama2", format = "json", temperature=0.5)
print("Converted params", converted_params)
assert converted_params == {'temperature': 0.5, 'format': 'json'}, f"{converted_params} != {'temperature': 0.5, 'format': 'json'}"
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_ollama_json_mode()