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

Update utils.py (#6468)

Browse source 
Fixed missing keys
This commit is contained in:
vibhanshu-ob 2024-10-29 21:36:23 +05:30 • committed by GitHub
parent e16f780b7c
commit 134bd2cebb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/utils.py
View file

@ -5361,7 +5361,7 @@ def validate_environment( # noqa: PLR0915
if "CODESTRAL_API_KEY" in os.environ: if "CODESTRAL_API_KEY" in os.environ:
keys_in_environment = True keys_in_environment = True
else: else:
missing_keys.append("GROQ_API_KEY") missing_keys.append("CODESTRAL_API_KEY")
elif custom_llm_provider == "deepseek": elif custom_llm_provider == "deepseek":
if "DEEPSEEK_API_KEY" in os.environ: if "DEEPSEEK_API_KEY" in os.environ:
keys_in_environment = True keys_in_environment = True
@ -5451,7 +5451,7 @@ def validate_environment( # noqa: PLR0915
if "VERTEXAI_PROJECT" in os.environ and "VERTEXAI_LOCATION" in os.environ: if "VERTEXAI_PROJECT" in os.environ and "VERTEXAI_LOCATION" in os.environ:
keys_in_environment = True keys_in_environment = True
else: else:
missing_keys.extend(["VERTEXAI_PROJECT", "VERTEXAI_PROJECT"]) missing_keys.extend(["VERTEXAI_PROJECT", "VERTEXAI_LOCATION"])
## huggingface ## huggingface
elif model in litellm.huggingface_models: elif model in litellm.huggingface_models:
if "HUGGINGFACE_API_KEY" in os.environ: if "HUGGINGFACE_API_KEY" in os.environ: