grasp_agents 0.3.9__tar.gz → 0.3.11__tar.gz

{grasp_agents-0.3.9 → grasp_agents-0.3.11}/PKG-INFO

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

{grasp_agents-0.3.9 → grasp_agents-0.3.11}/pyproject.toml

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

{grasp_agents-0.3.9 → grasp_agents-0.3.11}/src/grasp_agents/cloud_llm.py

@@ -21,7 +21,11 @@ from .llm import LLM, ConvertT_co, LLMSettings, SettingsT_co
 from .message_history import MessageHistory
 from .rate_limiting.rate_limiter_chunked import RateLimiterC, limit_rate_chunked
 from .typing.completion import Completion
-from .typing.completion_chunk import CompletionChunk, combine_completion_chunks
+from .typing.completion_chunk import (
+    CompletionChoice,
+    CompletionChunk,
+    combine_completion_chunks,
+)
 from .typing.events import CompletionChunkEvent, CompletionEvent
 from .typing.message import AssistantMessage, Messages
 from .typing.tool import BaseTool, ToolChoice
@@ -61,12 +65,12 @@ PROVIDERS: dict[APIProvider, APIProviderInfo] = {
 }
 
 
-def retry_error_callback(retry_state: RetryCallState) -> None:
+def retry_error_callback(retry_state: RetryCallState) -> Completion:
     assert retry_state.outcome is not None
     exception = retry_state.outcome.exception()
     if exception:
         if retry_state.attempt_number == 1:
-            logger.error(
+            logger.warning(
                 f"CloudLLM completion request failed:\n{exception}",
                 exc_info=exception,
             )
@@ -75,6 +79,12 @@ def retry_error_callback(retry_state: RetryCallState) -> None:
                 f"CloudLLM completion request failed after retrying:\n{exception}",
                 exc_info=exception,
             )
+    failed_message = AssistantMessage(content=None, refusal=str(exception))
+
+    return Completion(
+        model="",
+        choices=[CompletionChoice(message=failed_message, finish_reason=None, index=0)],
+    )
 
 
 def retry_before_callback(retry_state: RetryCallState) -> None:

{grasp_agents-0.3.9 → grasp_agents-0.3.11}/src/grasp_agents/comm_processor.py

@@ -90,13 +90,14 @@ class CommProcessor(
         *,
         in_packet: Packet[InT_contra] | None = None,
         in_args: InT_contra | Sequence[InT_contra] | None = None,
-        forgetful: bool = True,
+        forgetful: bool = False,
         ctx: RunContext[CtxT] | None = None,
     ) -> Packet[OutT_co]:
         out_packet = await super().run(
             chat_inputs=chat_inputs,
             in_packet=in_packet,
             in_args=in_args,
+            forgetful=forgetful,
             ctx=ctx,
         )
         recipients = self._validate_routing(out_packet.payloads)
@@ -115,7 +116,7 @@ class CommProcessor(
         *,
         in_packet: Packet[InT_contra] | None = None,
         in_args: InT_contra | Sequence[InT_contra] | None = None,
-        forgetful: bool = True,
+        forgetful: bool = False,
         ctx: RunContext[CtxT] | None = None,
     ) -> AsyncIterator[Event[Any]]:
         out_packet: Packet[OutT_co] | None = None
@@ -123,6 +124,7 @@ class CommProcessor(
             chat_inputs=chat_inputs,
             in_packet=in_packet,
             in_args=in_args,
+            forgetful=forgetful,
             ctx=ctx,
         ):
             if isinstance(event, PacketEvent):