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

feat(utils.py): accept 'api_key' as param for validate_environment

Browse source 
Closes https://github.com/BerriAI/litellm/issues/4375
This commit is contained in:
Krrish Dholakia 2024-07-11 12:02:23 -07:00
parent 2163434ff3
commit 1ba3fcc3fb
2 changed files with 18 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/tests/test_utils.py
View file

@ -258,6 +258,13 @@ def test_validate_environment_empty_model():
raise Exception()
def test_validate_environment_api_key():
response_obj = validate_environment(model="gpt-3.5-turbo", api_key="sk-my-test-key")
assert (
response_obj["keys_in_environment"] is True
), f"Missing keys={response_obj['missing_keys']}"
@mock.patch.dict(os.environ, {"OLLAMA_API_BASE": "foo"}, clear=True)
def test_validate_environment_ollama():
for provider in ["ollama", "ollama_chat"]: