agno 2.3.2__py3-none-any.whl → 2.3.4__py3-none-any.whl

agno/models/cometapi/cometapi.py

@@ -1,9 +1,10 @@
 from dataclasses import dataclass, field
 from os import getenv
-from typing import List, Optional
+from typing import Any, Dict, List, Optional
 
 import httpx
 
+from agno.exceptions import ModelProviderError
 from agno.models.openai.like import OpenAILike
 from agno.utils.log import log_debug
 
@@ -26,6 +27,23 @@ class CometAPI(OpenAILike):
     api_key: Optional[str] = field(default_factory=lambda: getenv("COMETAPI_KEY"))
     base_url: str = "https://api.cometapi.com/v1"
 
+    def _get_client_params(self) -> Dict[str, Any]:
+        """
+        Returns client parameters for API requests, checking for COMETAPI_KEY.
+
+        Returns:
+            Dict[str, Any]: A dictionary of client parameters for API requests.
+        """
+        if not self.api_key:
+            self.api_key = getenv("COMETAPI_KEY")
+            if not self.api_key:
+                raise ModelProviderError(
+                    message="COMETAPI_KEY not set. Please set the COMETAPI_KEY environment variable.",
+                    model_name=self.name,
+                    model_id=self.id,
+                )
+        return super()._get_client_params()
+
     def get_available_models(self) -> List[str]:
         """
         Fetch available chat models from CometAPI, filtering out non-chat models.

agno/models/deepinfra/deepinfra.py

@@ -1,7 +1,8 @@
 from dataclasses import dataclass, field
 from os import getenv
-from typing import Optional
+from typing import Any, Dict, Optional
 
+from agno.exceptions import ModelProviderError
 from agno.models.openai.like import OpenAILike
 
 
@@ -26,3 +27,20 @@ class DeepInfra(OpenAILike):
     base_url: str = "https://api.deepinfra.com/v1/openai"
 
     supports_native_structured_outputs: bool = False
+
+    def _get_client_params(self) -> Dict[str, Any]:
+        """
+        Returns client parameters for API requests, checking for DEEPINFRA_API_KEY.
+
+        Returns:
+            Dict[str, Any]: A dictionary of client parameters for API requests.
+        """
+        if not self.api_key:
+            self.api_key = getenv("DEEPINFRA_API_KEY")
+            if not self.api_key:
+                raise ModelProviderError(
+                    message="DEEPINFRA_API_KEY not set. Please set the DEEPINFRA_API_KEY environment variable.",
+                    model_name=self.name,
+                    model_id=self.id,
+                )
+        return super()._get_client_params()

agno/models/fireworks/fireworks.py

@@ -1,7 +1,8 @@
 from dataclasses import dataclass, field
 from os import getenv
-from typing import Optional
+from typing import Any, Dict, Optional
 
+from agno.exceptions import ModelProviderError
 from agno.models.openai import OpenAILike
 
 
@@ -24,3 +25,20 @@ class Fireworks(OpenAILike):
 
     api_key: Optional[str] = field(default_factory=lambda: getenv("FIREWORKS_API_KEY"))
     base_url: str = "https://api.fireworks.ai/inference/v1"
+
+    def _get_client_params(self) -> Dict[str, Any]:
+        """
+        Returns client parameters for API requests, checking for FIREWORKS_API_KEY.
+
+        Returns:
+            Dict[str, Any]: A dictionary of client parameters for API requests.
+        """
+        if not self.api_key:
+            self.api_key = getenv("FIREWORKS_API_KEY")
+            if not self.api_key:
+                raise ModelProviderError(
+                    message="FIREWORKS_API_KEY not set. Please set the FIREWORKS_API_KEY environment variable.",
+                    model_name=self.name,
+                    model_id=self.id,
+                )
+        return super()._get_client_params()