From fd41925ccf136f45bd2b670c378650cf32007d94 Mon Sep 17 00:00:00 2001 From: vincent Date: Sun, 12 Apr 2026 18:11:08 +0000 Subject: [PATCH] fix(py): forward api_key kwargs to litellm validate_environment and check_valid_key --- py_zerox/pyzerox/models/modellitellm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/py_zerox/pyzerox/models/modellitellm.py b/py_zerox/pyzerox/models/modellitellm.py index e7b2d204..6bd2f596 100644 --- a/py_zerox/pyzerox/models/modellitellm.py +++ b/py_zerox/pyzerox/models/modellitellm.py @@ -52,7 +52,7 @@ def system_prompt(self, prompt: str) -> None: ## custom method on top of BaseModel def validate_environment(self) -> None: """Validates the environment variables required for the model.""" - env_config = litellm.validate_environment(model=self.model) + env_config = litellm.validate_environment(model=self.model, api_key=self.kwargs.get("api_key")) if not env_config["keys_in_environment"]: raise MissingEnvironmentVariables(extra_info=env_config) @@ -64,7 +64,7 @@ def validate_model(self) -> None: def validate_access(self) -> None: """Validates access to the model -> if environment variables are set correctly with correct values.""" - if not litellm.check_valid_key(model=self.model,api_key=None): + if not litellm.check_valid_key(model=self.model, api_key=self.kwargs.get("api_key")): raise ModelAccessError(extra_info={"model": self.model})