phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 19:54:13 +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

5
litellm/utils.py
View file

@ -5602,6 +5602,11 @@ def validate_environment(model: Optional[str] = None) -> dict:
else:
missing_keys.append("AWS_ACCESS_KEY_ID")
missing_keys.append("AWS_SECRET_ACCESS_KEY")
elif custom_llm_provider in ["ollama", "ollama_chat"]:
if "OLLAMA_API_BASE" in os.environ:
keys_in_environment = True
else:
missing_keys.append("OLLAMA_API_BASE")
else:
## openai - chatcompletion + text completion
if (