Merge pull request #3324 from suptejas/patch-1

fix: incorrect analysis for vertex ai env vars causing confusion
This commit is contained in:
Krish Dholakia 2024-04-27 07:03:01 -07:00 committed by GitHub
commit 4cccd470ab
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -6563,7 +6563,7 @@ def validate_environment(model: Optional[str] = None) -> dict:
if "VERTEXAI_PROJECT" in os.environ and "VERTEXAI_LOCATION" in os.environ:
keys_in_environment = True
else:
missing_keys.extend(["VERTEXAI_PROJECT", "VERTEXAI_PROJECT"])
missing_keys.extend(["VERTEXAI_PROJECT", "VERTEXAI_LOCATION"])
elif custom_llm_provider == "huggingface":
if "HUGGINGFACE_API_KEY" in os.environ:
keys_in_environment = True