grasp_agents 0.3.10__tar.gz → 0.3.11__tar.gz

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

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

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

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

{grasp_agents-0.3.10 → 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: