grasp_agents 0.5.12__tar.gz → 0.5.14__tar.gz

{grasp_agents-0.5.12 → grasp_agents-0.5.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: grasp_agents
-Version: 0.5.12
+Version: 0.5.14
 Summary: Grasp Agents Library
 License-File: LICENSE.md
 Requires-Python: <4,>=3.11.4

{grasp_agents-0.5.12 → grasp_agents-0.5.14}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "grasp_agents"
-version = "0.5.12"
+version = "0.5.14"
 description = "Grasp Agents Library"
 readme = "README.md"
 requires-python = ">=3.11.4,<4"

{grasp_agents-0.5.12 → grasp_agents-0.5.14}/src/grasp_agents/litellm/lite_llm.py

@@ -74,23 +74,22 @@ class LiteLLM(CloudLLM[LiteLLMSettings, LiteLLMConverters]):
         )
 
         _api_provider = self.api_provider
-
-        if self.model_name in litellm.get_valid_models():  # type: ignore[no-untyped-call]
+        try:
             _, provider_name, _, _ = litellm.get_llm_provider(self.model_name)  # type: ignore[no-untyped-call]
             _api_provider = APIProvider(name=provider_name)
-        elif self.api_provider is not None:
-            self._lite_llm_completion_params["api_key"] = self.api_provider.get(
-                "api_key"
-            )
-            self._lite_llm_completion_params["api_base"] = self.api_provider.get(
-                "api_base"
-            )
-        elif self.api_provider is None:
-            raise ValueError(
-                f"Model '{self.model_name}' is not supported by LiteLLM and no API provider "
-                "was specified. Please provide a valid API provider or use a different "
-                "model."
-            )
+        except Exception as exc:
+            if self.api_provider is not None:
+                self._lite_llm_completion_params["api_key"] = self.api_provider.get(
+                    "api_key"
+                )
+                self._lite_llm_completion_params["api_base"] = self.api_provider.get(
+                    "api_base"
+                )
+            else:
+                raise ValueError(
+                    f"Failed to retrieve a LiteLLM supported API provider for model "
+                    f"'{self.model_name}' and no custom API provider was specified."
+                ) from exc
 
         if self.llm_settings is not None:
             stream_options = self.llm_settings.get("stream_options") or {}

{grasp_agents-0.5.12 → grasp_agents-0.5.14}/src/grasp_agents/processors/base_processor.py

@@ -47,6 +47,7 @@ def with_retry(func: F) -> F:
     async def wrapper(
         self: "BaseProcessor[Any, Any, Any, Any]", *args: Any, **kwargs: Any
     ) -> Packet[Any]:
+        none_packet = Packet(payloads=[None], sender=self.name)
         call_id = kwargs.get("call_id", "unknown")
         for n_attempt in range(self.max_retries + 1):
             try:
@@ -61,12 +62,12 @@ def with_retry(func: F) -> F:
                     else:
                         logger.warning(f"{err_message} after retrying:\n{err}")
                     # raise ProcRunError(proc_name=self.name, call_id=call_id) from err
-                    return None  # type: ignore[return]
+                    return none_packet
 
                 logger.warning(f"{err_message} (retry attempt {n_attempt + 1}):\n{err}")
         # This part should not be reachable due to the raise in the loop
         # raise ProcRunError(proc_name=self.name, call_id=call_id)
-        return None  # type: ignore[return]
+        return none_packet
 
     return cast("F", wrapper)