phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

utils.validate_environment to handle OLLAMA_API_BASE env

Browse source
This commit is contained in:
Mikhail Khludnev 2024-03-02 13:41:59 +03:00
parent 45544d44c0
commit 2686d0368e
2 changed files with 20 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

17
litellm/tests/test_utils.py
View file

@ -1,4 +1,6 @@
import sys, os
import sys
from unittest import mock
from dotenv import load_dotenv
import copy
@ -214,8 +216,19 @@ def test_validate_environment_empty_model():
raise Exception()
# test_validate_environment_empty_model()
@mock.patch.dict(os.environ, {"OLLAMA_API_BASE": "foo"}, clear=True)
def test_validate_environment_ollama():
for provider in ["ollama", "ollama_chat"]:
kv = validate_environment(provider+"/mistral")
assert kv["keys_in_environment"]
assert kv["missing_keys"] == []
@mock.patch.dict(os.environ, {}, clear=True)
def test_validate_environment_ollama_failed():
for provider in ["ollama", "ollama_chat"]:
kv = validate_environment(provider+"/mistral")
assert not kv["keys_in_environment"]
assert kv["missing_keys"] == ["OLLAMA_API_BASE"]
def test_function_to_dict():
print("testing function to dict for get current weather")