Convert get_api_key method into a property: _api_key

langchain-ai · Nov 1, 2024 · c80052b · c80052b
1 parent cd72d57
commit c80052b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/libs/community/langchain_community/chat_models/litellm.py b/libs/community/langchain_community/chat_models/litellm.py
@@ -266,7 +266,8 @@ def _default_params(self) -> Dict[str, Any]:
             **self.model_kwargs,
         }
 
-    def get_api_key(self) -> str:
+    @property
+    def _api_key(self) -> str:
         return (
             self.openai_api_key
             or self.azure_api_key
@@ -286,7 +287,7 @@ def _client_params(self) -> Dict[str, Any]:
         if self.model_name is not None:
             set_model_value = self.model_name
         self.client.api_base = self.api_base
-        self.client.api_key = self.get_api_key()
+        self.client.api_key = self._api_key
         self.client.organization = self.organization
         creds: Dict[str, Any] = {
             "model": set_model_value,