grasp_agents 0.5.12__py3-none-any.whl → 0.5.14__py3-none-any.whl

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/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)
 

{grasp_agents-0.5.12.dist-info → grasp_agents-0.5.14.dist-info}/METADATA

@@ -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.dist-info → grasp_agents-0.5.14.dist-info}/RECORD

@@ -22,7 +22,7 @@ grasp_agents/litellm/__init__.py,sha256=wD8RZBYokFDfbS9Cs7nO_zKb3w7RIVwEGj7g2D5C
 grasp_agents/litellm/completion_chunk_converters.py,sha256=J5PPxzoTBqkvKQnCoBxQxJo7Q8Xfl9cbv2GRZox8Cjo,2689
 grasp_agents/litellm/completion_converters.py,sha256=JQ7XvQwwc-biFqVMcRO61SL5VGs_SkUvAhUz1QD7EmU,2516
 grasp_agents/litellm/converters.py,sha256=XjePHii578sXP26Fyhnv0XfwJ3cNTp5PraggTsvcBXo,4778
-grasp_agents/litellm/lite_llm.py,sha256=6Am1rWZ1LWWog8tjSSoN6sFtOqSi4roX4QNC9FLw5Gs,8403
+grasp_agents/litellm/lite_llm.py,sha256=IXSplFiOksN-DjxkhZmB0eEGYJGr06IVR_ktE-2MVr8,8341
 grasp_agents/litellm/message_converters.py,sha256=PsGLIJEcAeEoluHIh-utEufJ_9WeMYzXkwnR-8jyULQ,2037
 grasp_agents/openai/__init__.py,sha256=xaRnblUskiLvypIhMe4NRp9dxCG-gNR7dPiugUbPbhE,4717
 grasp_agents/openai/completion_chunk_converters.py,sha256=3MnMskdlp7ycsggc1ok1XpCHaP4Us2rLYaxImPLw1eI,2573
@@ -32,7 +32,7 @@ grasp_agents/openai/converters.py,sha256=RKOfMbIJmfFQ7ot0RGR6wrdMbR6_L7PB0UZwxwg
 grasp_agents/openai/message_converters.py,sha256=fhSN81uK51EGbLyM2-f0MvPX_UBrMy7SF3JQPo-dkXg,4686
 grasp_agents/openai/openai_llm.py,sha256=DCvoF7MM_OID2UhTkNcBLeEdf8Ej2SQ-6FhgRGe8l1E,9861
 grasp_agents/openai/tool_converters.py,sha256=rNH5t2Wir9nuy8Ei0jaxNuzDaXGqTLmLz3VyrnJhyn0,1196
-grasp_agents/processors/base_processor.py,sha256=2ks_jpeye0XW9k3BcAhD9XO8NctspOV2py9ZAD-dO6c,10942
+grasp_agents/processors/base_processor.py,sha256=OpEbOHNsi3RkVDrWy7sN42UB-0vx0L9_hWGjlA07QsM,10972
 grasp_agents/processors/parallel_processor.py,sha256=BOXRlPaZ-hooz0hHctqiW_5ldR-yDPYjFxuP7fAbZCI,7911
 grasp_agents/processors/processor.py,sha256=35MtYKrKtCZZMhV-U1DXBXtCNbCvZGaiiXo_5a3tI6s,5249
 grasp_agents/rate_limiting/__init__.py,sha256=KRgtF_E7R3YfA2cpYcFcZ7wycV0pWVJ0xRQC7YhiIEQ,158
@@ -52,7 +52,7 @@ grasp_agents/workflow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3h
 grasp_agents/workflow/looped_workflow.py,sha256=WHp9O3Za2sBVfY_BLOdvPvtY20XsjZQaWSO2-oAFvOY,6806
 grasp_agents/workflow/sequential_workflow.py,sha256=e3BIWzy_2novmEWNwIteyMbrzvl1-evHrTBE3r3SpU8,3648
 grasp_agents/workflow/workflow_processor.py,sha256=DwHz70UOTp9dkbtzH9KE5LkGcT1RdHV7Hdiby0Bu9tw,3535
-grasp_agents-0.5.12.dist-info/METADATA,sha256=zfsWyMd6aeCxdXiTGWUM4ZuOnA3xMAbgfeLXKYpNdcI,6633
-grasp_agents-0.5.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-grasp_agents-0.5.12.dist-info/licenses/LICENSE.md,sha256=-nNNdWqGB8gJ2O-peFQ2Irshv5tW5pHKyTcYkwvH7CE,1201
-grasp_agents-0.5.12.dist-info/RECORD,,
+grasp_agents-0.5.14.dist-info/METADATA,sha256=zXIn4AF6QikIjI4DqPjNzzLfCn3gUXft9j1ktMNQu10,6633
+grasp_agents-0.5.14.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+grasp_agents-0.5.14.dist-info/licenses/LICENSE.md,sha256=-nNNdWqGB8gJ2O-peFQ2Irshv5tW5pHKyTcYkwvH7CE,1201
+grasp_agents-0.5.14.dist-info/RECORD,,