grasp_agents 0.5.11__py3-none-any.whl → 0.5.13__py3-none-any.whl

grasp_agents/__init__.py

@@ -6,6 +6,7 @@ from .llm_agent import LLMAgent
 from .llm_agent_memory import LLMAgentMemory
 from .memory import Memory
 from .packet import Packet
+from .printer import Printer, print_event_stream
 from .processors.base_processor import BaseProcessor
 from .processors.parallel_processor import ParallelProcessor
 from .processors.processor import Processor
@@ -33,9 +34,11 @@ __all__ = [
     "Packet",
     "Packet",
     "ParallelProcessor",
+    "Printer",
     "ProcName",
     "Processor",
     "RunContext",
     "SystemMessage",
     "UserMessage",
+    "print_event_stream",
 ]

grasp_agents/cloud_llm.py

@@ -69,6 +69,7 @@ class CloudLLM(LLM[SettingsT_co, ConvertT_co], Generic[SettingsT_co, ConvertT_co
         0  # LLM response retries: try to regenerate to pass validation
     )
     apply_response_schema_via_provider: bool = False
+    apply_tool_call_schema_via_provider: bool = False
     async_http_client: httpx.AsyncClient | None = None
     async_http_client_params: dict[str, Any] | AsyncHTTPClientParams | None = None
 
@@ -79,6 +80,9 @@ class CloudLLM(LLM[SettingsT_co, ConvertT_co], Generic[SettingsT_co, ConvertT_co
                 f"{self.rate_limiter.rpm} RPM"
             )
 
+        if self.apply_response_schema_via_provider:
+            object.__setattr__(self, "apply_tool_call_schema_via_provider", True)
+
         if self.async_http_client is None and self.async_http_client_params is not None:
             object.__setattr__(
                 self,
@@ -99,7 +103,7 @@ class CloudLLM(LLM[SettingsT_co, ConvertT_co], Generic[SettingsT_co, ConvertT_co
         api_tools = None
         api_tool_choice = None
         if tools:
-            strict = True if self.apply_response_schema_via_provider else None
+            strict = True if self.apply_tool_call_schema_via_provider else None
             api_tools = [
                 self.converters.to_tool(t, strict=strict) for t in tools.values()
             ]
@@ -174,8 +178,8 @@ class CloudLLM(LLM[SettingsT_co, ConvertT_co], Generic[SettingsT_co, ConvertT_co
                 response_schema=response_schema,
                 response_schema_by_xml_tag=response_schema_by_xml_tag,
             )
-            if tools is not None:
-                self._validate_tool_calls(completion, tools=tools)
+        if not self.apply_tool_call_schema_via_provider and tools is not None:
+            self._validate_tool_calls(completion, tools=tools)
 
         return completion
 
@@ -207,17 +211,16 @@ class CloudLLM(LLM[SettingsT_co, ConvertT_co], Generic[SettingsT_co, ConvertT_co
 
                 if n_attempt > self.max_response_retries:
                     if n_attempt == 1:
-                        logger.warning(f"\nCloudLLM completion request failed:\n{err}")
+                        logger.warning(f"\nCloudLLM completion failed:\n{err}")
                     if n_attempt > 1:
                         logger.warning(
-                            f"\nCloudLLM completion request failed after retrying:\n{err}"
+                            f"\nCloudLLM completion failed after retrying:\n{err}"
                         )
                     raise err
                     # return make_refusal_completion(self._model_name, err)
 
                 logger.warning(
-                    f"\nCloudLLM completion request failed (retry attempt {n_attempt}):"
-                    f"\n{err}"
+                    f"\nCloudLLM completion failed (retry attempt {n_attempt}):\n{err}"
                 )
 
         return make_refusal_completion(
@@ -281,8 +284,8 @@ class CloudLLM(LLM[SettingsT_co, ConvertT_co], Generic[SettingsT_co, ConvertT_co
                     response_schema=response_schema,
                     response_schema_by_xml_tag=response_schema_by_xml_tag,
                 )
-                if tools is not None:
-                    self._validate_tool_calls(completion, tools=tools)
+            if not self.apply_tool_call_schema_via_provider and tools is not None:
+                self._validate_tool_calls(completion, tools=tools)
 
         return iterator()
 
@@ -326,11 +329,10 @@ class CloudLLM(LLM[SettingsT_co, ConvertT_co], Generic[SettingsT_co, ConvertT_co
                 n_attempt += 1
                 if n_attempt > self.max_response_retries:
                     if n_attempt == 1:
-                        logger.warning(f"\nCloudLLM completion request failed:\n{err}")
+                        logger.warning(f"\nCloudLLM completion failed:\n{err}")
                     if n_attempt > 1:
                         logger.warning(
-                            "\nCloudLLM completion request failed after "
-                            f"retrying:\n{err}"
+                            f"\nCloudLLM completion failed after retrying:\n{err}"
                         )
                         refusal_completion = make_refusal_completion(
                             self.model_name, err
@@ -344,6 +346,5 @@ class CloudLLM(LLM[SettingsT_co, ConvertT_co], Generic[SettingsT_co, ConvertT_co
                     # return
 
                 logger.warning(
-                    "\nCloudLLM completion request failed "
-                    f"(retry attempt {n_attempt}):\n{err}"
+                    f"\nCloudLLM completion failed (retry attempt {n_attempt}):\n{err}"
                 )

grasp_agents/generics_utils.py

@@ -159,7 +159,7 @@ class AutoInstanceAttributesMixin:
                 attr_type = resolved_attr_types[attr_name]
                 # attr_type = None if _attr_type is type(None) else _attr_type
             else:
-                attr_type = Any
+                attr_type = object
 
             if attr_name in pyd_private:
                 pyd_private[attr_name] = attr_type

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 {}
@@ -149,6 +148,9 @@ class LiteLLM(CloudLLM[LiteLLMSettings, LiteLLMConverters]):
         n_choices: int | None = None,
         **api_llm_settings: Any,
     ) -> LiteLLMCompletion:
+        if api_llm_settings and api_llm_settings.get("stream_options"):
+            api_llm_settings.pop("stream_options")
+
         completion = await litellm.acompletion(  # type: ignore[no-untyped-call]
             model=self.model_name,
             messages=api_messages,

grasp_agents/llm_agent.py

@@ -310,12 +310,9 @@ class LLMAgent(
         yield ProcPayloadOutputEvent(data=output, proc_name=self.name, call_id=call_id)
 
     def _print_messages(
-        self,
-        messages: Sequence[Message],
-        ctx: RunContext[CtxT],
-        call_id: str,
+        self, messages: Sequence[Message], ctx: RunContext[CtxT], call_id: str
     ) -> None:
-        if ctx and ctx.printer:
+        if ctx.printer:
             ctx.printer.print_messages(messages, agent_name=self.name, call_id=call_id)
 
     # -- Override these methods in subclasses if needed --

grasp_agents/llm_policy_executor.py

@@ -161,14 +161,11 @@ class LLMPolicyExecutor(Generic[CtxT]):
             response_schema_by_xml_tag=self.response_schema_by_xml_tag,
             tools=self.tools,
             tool_choice=tool_choice,
-            n_choices=1,
             proc_name=self.agent_name,
             call_id=call_id,
         )
         memory.update(completion.messages)
-        self._process_completion(
-            completion, ctx=ctx, call_id=call_id, print_messages=True
-        )
+        self._process_completion(completion, ctx=ctx, call_id=call_id)
 
         return completion.messages[0]
 
@@ -193,7 +190,6 @@ class LLMPolicyExecutor(Generic[CtxT]):
             response_schema_by_xml_tag=self.response_schema_by_xml_tag,
             tools=self.tools,
             tool_choice=tool_choice,
-            n_choices=1,
             proc_name=self.agent_name,
             call_id=call_id,
         )
@@ -212,9 +208,7 @@ class LLMPolicyExecutor(Generic[CtxT]):
 
         memory.update(completion.messages)
 
-        self._process_completion(
-            completion, print_messages=True, ctx=ctx, call_id=call_id
-        )
+        self._process_completion(completion, ctx=ctx, call_id=call_id)
 
     async def call_tools(
         self,
@@ -237,7 +231,7 @@ class LLMPolicyExecutor(Generic[CtxT]):
 
         memory.update(tool_messages)
 
-        if ctx and ctx.printer:
+        if ctx.printer:
             ctx.printer.print_messages(
                 tool_messages, agent_name=self.agent_name, call_id=call_id
             )
@@ -283,7 +277,7 @@ class LLMPolicyExecutor(Generic[CtxT]):
             "Exceeded the maximum number of turns: provide a final answer now!"
         )
         memory.update([user_message])
-        if ctx and ctx.printer:
+        if ctx.printer:
             ctx.printer.print_messages(
                 [user_message], agent_name=self.agent_name, call_id=call_id
             )
@@ -309,7 +303,7 @@ class LLMPolicyExecutor(Generic[CtxT]):
         yield UserMessageEvent(
             proc_name=self.agent_name, call_id=call_id, data=user_message
         )
-        if ctx and ctx.printer:
+        if ctx.printer:
             ctx.printer.print_messages(
                 [user_message], agent_name=self.agent_name, call_id=call_id
             )
@@ -507,12 +501,7 @@ class LLMPolicyExecutor(Generic[CtxT]):
         return FinalAnswerTool()
 
     def _process_completion(
-        self,
-        completion: Completion,
-        *,
-        print_messages: bool = False,
-        ctx: RunContext[CtxT],
-        call_id: str,
+        self, completion: Completion, *, ctx: RunContext[CtxT], call_id: str
     ) -> None:
         ctx.completions[self.agent_name].append(completion)
         ctx.usage_tracker.update(
@@ -520,7 +509,7 @@ class LLMPolicyExecutor(Generic[CtxT]):
             completions=[completion],
             model_name=self.llm.model_name,
         )
-        if ctx.printer and print_messages:
+        if ctx.printer:
             usages = [None] * (len(completion.messages) - 1) + [completion.usage]
             ctx.printer.print_messages(
                 completion.messages,

grasp_agents/openai/openai_llm.py

@@ -172,6 +172,9 @@ class OpenAILLM(CloudLLM[OpenAILLMSettings, OpenAIConverters]):
         response_format = api_response_schema or NOT_GIVEN
         n = n_choices or NOT_GIVEN
 
+        if api_llm_settings and api_llm_settings.get("stream_options"):
+            api_llm_settings.pop("stream_options")
+
         if self.apply_response_schema_via_provider:
             return await self.client.beta.chat.completions.parse(
                 model=self.model_name,

grasp_agents/printer.py

@@ -72,50 +72,61 @@ ColoringMode: TypeAlias = Literal["agent", "role"]
 CompletionBlockType: TypeAlias = Literal["response", "thinking", "tool_call"]
 
 
-class Printer:
-    def __init__(
-        self, color_by: ColoringMode = "role", msg_trunc_len: int = 20000
-    ) -> None:
-        self.color_by = color_by
-        self.msg_trunc_len = msg_trunc_len
-        self._current_message: str = ""
+def stream_colored_text(new_colored_text: str) -> None:
+    sys.stdout.write(new_colored_text)
+    sys.stdout.flush()
 
-    @staticmethod
-    def get_role_color(role: Role) -> Color:
-        return ROLE_TO_COLOR[role]
 
-    @staticmethod
-    def get_agent_color(agent_name: str) -> Color:
+def get_color(
+    agent_name: str = "", role: Role = Role.ASSISTANT, color_by: ColoringMode = "role"
+) -> Color:
+    if color_by == "agent":
         idx = int(
             hashlib.md5(agent_name.encode()).hexdigest(),  # noqa :S324
             16,
         ) % len(AVAILABLE_COLORS)
 
         return AVAILABLE_COLORS[idx]
+    return ROLE_TO_COLOR[role]
+
 
-    @staticmethod
-    def content_to_str(content: Content | str | None, role: Role) -> str:
-        if role == Role.USER and isinstance(content, Content):
-            content_str_parts: list[str] = []
-            for content_part in content.parts:
-                if isinstance(content_part, ContentPartText):
-                    content_str_parts.append(content_part.data.strip(" \n"))
-                elif content_part.data.type == "url":
-                    content_str_parts.append(str(content_part.data.url))
-                elif content_part.data.type == "base64":
-                    content_str_parts.append("<ENCODED_IMAGE>")
-            return "\n".join(content_str_parts)
+def content_to_str(content: Content | str | None, role: Role) -> str:
+    if role == Role.USER and isinstance(content, Content):
+        content_str_parts: list[str] = []
+        for content_part in content.parts:
+            if isinstance(content_part, ContentPartText):
+                content_str_parts.append(content_part.data.strip(" \n"))
+            elif content_part.data.type == "url":
+                content_str_parts.append(str(content_part.data.url))
+            elif content_part.data.type == "base64":
+                content_str_parts.append("<ENCODED_IMAGE>")
+        return "\n".join(content_str_parts)
 
-        assert isinstance(content, str | None)
+    assert isinstance(content, str | None)
 
-        return (content or "").strip(" \n")
+    return (content or "").strip(" \n")
 
-    @staticmethod
-    def truncate_content_str(content_str: str, trunc_len: int = 2000) -> str:
-        if len(content_str) > trunc_len:
-            return content_str[:trunc_len] + "[...]"
 
-        return content_str
+def truncate_content_str(content_str: str, trunc_len: int = 2000) -> str:
+    if len(content_str) > trunc_len:
+        return content_str[:trunc_len] + "[...]"
+
+    return content_str
+
+
+class Printer:
+    def __init__(
+        self,
+        color_by: ColoringMode = "role",
+        msg_trunc_len: int = 20000,
+        output_to: Literal["print", "log"] = "print",
+        logging_level: Literal["info", "debug", "warning", "error"] = "info",
+    ) -> None:
+        self.color_by: ColoringMode = color_by
+        self.msg_trunc_len = msg_trunc_len
+        self._current_message: str = ""
+        self._logging_level = logging_level
+        self._output_to = output_to
 
     def print_message(
         self,
@@ -128,11 +139,8 @@ class Printer:
             raise ValueError(
                 "Usage information can only be printed for AssistantMessage"
             )
-
-        color = (
-            self.get_agent_color(agent_name)
-            if self.color_by == "agent"
-            else self.get_role_color(message.role)
+        color = get_color(
+            agent_name=agent_name, role=message.role, color_by=self.color_by
         )
         log_kwargs = {"extra": {"color": color}}
 
@@ -144,13 +152,13 @@ class Printer:
             out += f"<thinking>\n{thinking}\n</thinking>\n"
 
         # Content
-        content = self.content_to_str(message.content or "", message.role)
+        content = content_to_str(message.content or "", message.role)
         if content:
             try:
                 content = json.dumps(json.loads(content), indent=2)
             except Exception:
                 pass
-            content = self.truncate_content_str(content, trunc_len=self.msg_trunc_len)
+            content = truncate_content_str(content, trunc_len=self.msg_trunc_len)
             if isinstance(message, SystemMessage):
                 out += f"<system>\n{content}\n</system>\n"
             elif isinstance(message, UserMessage):
@@ -176,7 +184,17 @@ class Printer:
 
             out += f"\n------------------------------------\n{usage_str}\n"
 
-        logger.debug(out, **log_kwargs)  # type: ignore
+        if self._output_to == "log":
+            if self._logging_level == "debug":
+                logger.debug(out, **log_kwargs)  # type: ignore
+            elif self._logging_level == "info":
+                logger.info(out, **log_kwargs)  # type: ignore
+            elif self._logging_level == "warning":
+                logger.warning(out, **log_kwargs)  # type: ignore
+            else:
+                logger.error(out, **log_kwargs)  # type: ignore
+        else:
+            stream_colored_text(colored(out + "\n", color))
 
     def print_messages(
         self,
@@ -193,28 +211,101 @@ class Printer:
             )
 
 
-def stream_text(new_text: str, color: Color) -> None:
-    sys.stdout.write(colored(new_text, color))
-    sys.stdout.flush()
-
-
 async def print_event_stream(
     event_generator: AsyncIterator[Event[Any]],
     color_by: ColoringMode = "role",
     trunc_len: int = 10000,
 ) -> AsyncIterator[Event[Any]]:
-    color = Printer.get_role_color(Role.ASSISTANT)
+    def _make_chunk_text(event: CompletionChunkEvent[CompletionChunk]) -> str:
+        color = get_color(
+            agent_name=event.proc_name or "", role=Role.ASSISTANT, color_by=color_by
+        )
+        text = ""
+
+        if isinstance(event, CompletionStartEvent):
+            text += f"\n<{event.proc_name}> [{event.call_id}]\n"
+        elif isinstance(event, ThinkingStartEvent):
+            text += "<thinking>\n"
+        elif isinstance(event, ResponseStartEvent):
+            text += "<response>\n"
+        elif isinstance(event, ToolCallStartEvent):
+            tc = event.data.tool_call
+            text += f"<tool call> {tc.tool_name} [{tc.id}]\n"
+        elif isinstance(event, AnnotationsStartEvent):
+            text += "<annotations>\n"
+
+        # if isinstance(event, CompletionEndEvent):
+        #     text += f"\n</{event.proc_name}>\n"
+        if isinstance(event, ThinkingEndEvent):
+            text += "\n</thinking>\n"
+        elif isinstance(event, ResponseEndEvent):
+            text += "\n</response>\n"
+        elif isinstance(event, ToolCallEndEvent):
+            text += "\n</tool call>\n"
+        elif isinstance(event, AnnotationsEndEvent):
+            text += "\n</annotations>\n"
+
+        if isinstance(event, ThinkingChunkEvent):
+            thinking = event.data.thinking
+            if isinstance(thinking, str):
+                text += thinking
+            else:
+                text = "\n".join(
+                    [block.get("thinking", "[redacted]") for block in thinking]
+                )
 
-    def _get_color(event: Event[Any], role: Role = Role.ASSISTANT) -> Color:
-        if color_by == "agent":
-            return Printer.get_agent_color(event.proc_name or "")
-        return Printer.get_role_color(role)
+        if isinstance(event, ResponseChunkEvent):
+            text += event.data.response
 
-    def _print_packet(
-        event: ProcPacketOutputEvent | WorkflowResultEvent | RunResultEvent,
-    ) -> None:
-        color = _get_color(event, Role.ASSISTANT)
+        if isinstance(event, ToolCallChunkEvent):
+            text += event.data.tool_call.tool_arguments or ""
 
+        if isinstance(event, AnnotationsChunkEvent):
+            text += "\n".join(
+                [
+                    json.dumps(annotation, indent=2)
+                    for annotation in event.data.annotations
+                ]
+            )
+
+        return colored(text, color)
+
+    def _make_message_text(
+        event: MessageEvent[SystemMessage | UserMessage | ToolMessage],
+    ) -> str:
+        message = event.data
+        role = message.role
+        content = content_to_str(message.content, role=role)
+
+        color = get_color(
+            agent_name=event.proc_name or "", role=role, color_by=color_by
+        )
+        text = f"\n<{event.proc_name}> [{event.call_id}]\n"
+
+        if isinstance(event, (SystemMessageEvent, UserMessageEvent)):
+            content = truncate_content_str(content, trunc_len=trunc_len)
+
+        if isinstance(event, SystemMessageEvent):
+            text += f"<system>\n{content}\n</system>\n"
+
+        elif isinstance(event, UserMessageEvent):
+            text += f"<input>\n{content}\n</input>\n"
+
+        elif isinstance(event, ToolMessageEvent):
+            message = event.data
+            try:
+                content = json.dumps(json.loads(content), indent=2)
+            except Exception:
+                pass
+            text += (
+                f"<tool result> [{message.tool_call_id}]\n{content}\n</tool result>\n"
+            )
+
+        return colored(text, color)
+
+    def _make_packet_text(
+        event: ProcPacketOutputEvent | WorkflowResultEvent | RunResultEvent,
+    ) -> str:
         if isinstance(event, WorkflowResultEvent):
             src = "workflow"
         elif isinstance(event, RunResultEvent):
@@ -222,6 +313,9 @@ async def print_event_stream(
         else:
             src = "processor"
 
+        color = get_color(
+            agent_name=event.proc_name or "", role=Role.ASSISTANT, color_by=color_by
+        )
         text = f"\n<{event.proc_name}> [{event.call_id}]\n"
 
         if event.data.payloads:
@@ -237,7 +331,9 @@ async def print_event_stream(
                 text += f"{p_str}\n"
             text += f"</{src} output>\n"
 
-        stream_text(text, color)
+        return colored(text, color)
+
+    # ------ Wrap event generator -------
 
     async for event in event_generator:
         yield event
@@ -245,91 +341,12 @@ async def print_event_stream(
         if isinstance(event, CompletionChunkEvent) and isinstance(
             event.data, CompletionChunk
         ):
-            color = _get_color(event, Role.ASSISTANT)
-
-            text = ""
-
-            if isinstance(event, CompletionStartEvent):
-                text += f"\n<{event.proc_name}> [{event.call_id}]\n"
-            elif isinstance(event, ThinkingStartEvent):
-                text += "<thinking>\n"
-            elif isinstance(event, ResponseStartEvent):
-                text += "<response>\n"
-            elif isinstance(event, ToolCallStartEvent):
-                tc = event.data.tool_call
-                text += f"<tool call> {tc.tool_name} [{tc.id}]\n"
-            elif isinstance(event, AnnotationsStartEvent):
-                text += "<annotations>\n"
-
-            # if isinstance(event, CompletionEndEvent):
-            #     text += f"\n</{event.proc_name}>\n"
-            if isinstance(event, ThinkingEndEvent):
-                text += "\n</thinking>\n"
-            elif isinstance(event, ResponseEndEvent):
-                text += "\n</response>\n"
-            elif isinstance(event, ToolCallEndEvent):
-                text += "\n</tool call>\n"
-            elif isinstance(event, AnnotationsEndEvent):
-                text += "\n</annotations>\n"
-
-            if isinstance(event, ThinkingChunkEvent):
-                thinking = event.data.thinking
-                if isinstance(thinking, str):
-                    text += thinking
-                else:
-                    text = "\n".join(
-                        [block.get("thinking", "[redacted]") for block in thinking]
-                    )
-
-            if isinstance(event, ResponseChunkEvent):
-                text += event.data.response
-
-            if isinstance(event, ToolCallChunkEvent):
-                text += event.data.tool_call.tool_arguments or ""
-
-            if isinstance(event, AnnotationsChunkEvent):
-                text += "\n".join(
-                    [
-                        json.dumps(annotation, indent=2)
-                        for annotation in event.data.annotations
-                    ]
-                )
-
-            stream_text(text, color)
+            stream_colored_text(_make_chunk_text(event))
 
         if isinstance(event, MessageEvent) and not isinstance(event, GenMessageEvent):
-            assert isinstance(event.data, (SystemMessage | UserMessage | ToolMessage))
-
-            message = event.data
-            role = message.role
-            content = Printer.content_to_str(message.content, role=role)
-            color = _get_color(event, role)
-
-            text = f"\n<{event.proc_name}> [{event.call_id}]\n"
-
-            if isinstance(event, (SystemMessageEvent, UserMessageEvent)):
-                content = Printer.truncate_content_str(content, trunc_len=trunc_len)
-
-            if isinstance(event, SystemMessageEvent):
-                text += f"<system>\n{content}\n</system>\n"
-
-            elif isinstance(event, UserMessageEvent):
-                text += f"<input>\n{content}\n</input>\n"
-
-            elif isinstance(event, ToolMessageEvent):
-                message = event.data
-                try:
-                    content = json.dumps(json.loads(content), indent=2)
-                except Exception:
-                    pass
-                text += (
-                    f"<tool result> [{message.tool_call_id}]\n"
-                    f"{content}\n</tool result>\n"
-                )
-
-            stream_text(text, color)
+            stream_colored_text(_make_message_text(event))
 
         if isinstance(
             event, (ProcPacketOutputEvent, WorkflowResultEvent, RunResultEvent)
         ):
-            _print_packet(event)
+            stream_colored_text(_make_packet_text(event))

grasp_agents/processors/base_processor.py

@@ -60,11 +60,13 @@ def with_retry(func: F) -> F:
                         logger.warning(f"{err_message}:\n{err}")
                     else:
                         logger.warning(f"{err_message} after retrying:\n{err}")
-                    raise ProcRunError(proc_name=self.name, call_id=call_id) from err
+                    # raise ProcRunError(proc_name=self.name, call_id=call_id) from err
+                    return None  # type: ignore[return]
 
                 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)
+        # raise ProcRunError(proc_name=self.name, call_id=call_id)
+        return None  # type: ignore[return]
 
     return cast("F", wrapper)
 

grasp_agents/run_context.py

@@ -1,11 +1,11 @@
 from collections import defaultdict
-from typing import Any, Generic, TypeVar
+from typing import Generic, TypeVar
 
 from pydantic import BaseModel, ConfigDict, Field
 
 from grasp_agents.typing.completion import Completion
 
-from .printer import ColoringMode, Printer
+from .printer import Printer
 from .typing.io import ProcName
 from .usage_tracker import UsageTracker
 
@@ -19,13 +19,6 @@ class RunContext(BaseModel, Generic[CtxT]):
         default_factory=lambda: defaultdict(list)
     )
     usage_tracker: UsageTracker = Field(default_factory=UsageTracker)
-
     printer: Printer | None = None
-    log_messages: bool = False
-    color_messages_by: ColoringMode = "role"
-
-    def model_post_init(self, context: Any) -> None:  # noqa: ARG002
-        if self.log_messages:
-            self.printer = Printer(color_by=self.color_messages_by)
 
     model_config = ConfigDict(extra="forbid", arbitrary_types_allowed=True)

{grasp_agents-0.5.11.dist-info → grasp_agents-0.5.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: grasp_agents
-Version: 0.5.11
+Version: 0.5.13
 Summary: Grasp Agents Library
 License-File: LICENSE.md
 Requires-Python: <4,>=3.11.4
@@ -110,24 +110,16 @@ Create a script, e.g., `problem_recommender.py`:
 
 ```python
 import asyncio
-from pathlib import Path
 from typing import Any
 
 from dotenv import load_dotenv
 from pydantic import BaseModel, Field
 
-from grasp_agents.grasp_logging import setup_logging
+from grasp_agents import LLMAgent, BaseTool, RunContext, Printer
 from grasp_agents.litellm import LiteLLM, LiteLLMSettings
-from grasp_agents import LLMAgent, BaseTool, RunContext
-
-load_dotenv()
 
 
-# Configure the logger to output to the console and/or a file
-setup_logging(
-    logs_file_path="grasp_agents_demo.log",
-    logs_config_path=Path().cwd() / "configs/logging/default.yaml",
-)
+load_dotenv()
 
 sys_prompt_react = """
 Your task is to suggest an exciting stats problem to the student. 
@@ -162,7 +154,7 @@ Returns:
 """
 
 
-class AskStudentTool(BaseTool[TeacherQuestion, StudentReply, Any]):
+class AskStudentTool(BaseTool[TeacherQuestion, StudentReply, None]):
     name: str = "ask_student"
     description: str = ask_student_tool_description
 
@@ -176,11 +168,7 @@ class Problem(BaseModel):
 
 teacher = LLMAgent[None, Problem, None](
     name="teacher",
-    llm=LiteLLM(
-        model_name="gpt-4.1",
-        # model_name="claude-sonnet-4-20250514",
-        # llm_settings=LiteLLMSettings(reasoning_effort="low"),
-    ),
+    llm=LiteLLM(model_name="gpt-4.1"),
     tools=[AskStudentTool()],
     react_mode=True,
     final_answer_as_tool_call=True,
@@ -188,7 +176,7 @@ teacher = LLMAgent[None, Problem, None](
 )
 
 async def main():
-    ctx = RunContext[None](log_messages=True)
+    ctx = RunContext[None](printer=Printer())
     out = await teacher.run("start", ctx=ctx)
     print(out.payloads[0])
     print(ctx.usage_tracker.total_usage)

{grasp_agents-0.5.11.dist-info → grasp_agents-0.5.13.dist-info}/RECORD

@@ -1,20 +1,20 @@
-grasp_agents/__init__.py,sha256=Z3a_j2Etiap9H6lvE8-PQP_OIGMUcHNPeJAJO12B8kY,1031
-grasp_agents/cloud_llm.py,sha256=vwI6gpLOsFqN4KtaTOo75xw8t7uRtdVrYGjopEDmQBw,13091
+grasp_agents/__init__.py,sha256=0pRU10xjcpuPdCitYtPK_bJVSUZ89FD4Jsmv1DJ_0GY,1121
+grasp_agents/cloud_llm.py,sha256=NwKr1XwpJP-px9xZDI1rngHuDmpSbtJ61VynLacbtNo,13237
 grasp_agents/costs_dict.yaml,sha256=2MFNWtkv5W5WSCcv1Cj13B1iQLVv5Ot9pS_KW2Gu2DA,2510
 grasp_agents/errors.py,sha256=K-22TCM1Klhsej47Rg5eTqnGiGPaXgKOpdOZZ7cPipw,4633
-grasp_agents/generics_utils.py,sha256=5Pw3I9dlnKC2VGqYKC4ZZUO3Z_vTNT-NPFovNfPkl6I,6542
+grasp_agents/generics_utils.py,sha256=HTX7G8eoylR-zMKz7JDKC-QnDjwFLqMNMjiI8XoGEow,6545
 grasp_agents/grasp_logging.py,sha256=H1GYhXdQvVkmauFDZ-KDwvVmPQHZUUm9sRqX_ObK2xI,1111
 grasp_agents/http_client.py,sha256=Es8NXGDkp4Nem7g24-jW0KFGA9Hp_o2Cv3cOvjup-iU,859
 grasp_agents/llm.py,sha256=IeV2QpR4AldVP3THzSETEnsaDx3DYz5HM6dkikSpy4o,10684
-grasp_agents/llm_agent.py,sha256=F_ou0pfdztqZzd2yU1jZZZVzcyhsLXfE_i0c4y2fZIQ,14123
+grasp_agents/llm_agent.py,sha256=Ig4YUsxPGvC444ZGl1eDIwfy46gmyygOe7mleEq0ZYM,14090
 grasp_agents/llm_agent_memory.py,sha256=XmOT2G8RG5AHd0LR3WuK7VbD-KFFfThmJnuZK2iU3Fs,1856
-grasp_agents/llm_policy_executor.py,sha256=r0UxwjnVzTBQqLlwvZZ_JL0wl6ZebCgxkcz6I4GdmrM,18136
+grasp_agents/llm_policy_executor.py,sha256=AVUuulV4QYQG86hHrBf7ldfThAGx3MPnZJdotnxwO8U,17886
 grasp_agents/memory.py,sha256=keHuNEZNSxHT9FKpMohHOCNi7UAz_oRIc91IQEuzaWE,1162
 grasp_agents/packet.py,sha256=EmE-W4ZSMVZoqClECGFe7OGqrT4FSJ8IVGICrdjtdEY,1462
 grasp_agents/packet_pool.py,sha256=AF7ZMYY1U6ppNLEn6o0R8QXyWmcLQGcju7_TYQpAudg,4443
-grasp_agents/printer.py,sha256=wVNCaR9mbFKyzYdT8YpYD1JQqRqHdLtdfiZrwYxaM6Y,11132
+grasp_agents/printer.py,sha256=9hw4XOoYlj6mZ3hcH04sxbQQ3rCZqixrn0RIwiXwGh8,11637
 grasp_agents/prompt_builder.py,sha256=wNPphkW8RL8501jV4Z7ncsN_sxBDR9Ax7eILLHr-OYg,6110
-grasp_agents/run_context.py,sha256=7qVs0T5rLvINmtlXqOoyy2Hu9xPzuFDbcVR6R93NF-0,951
+grasp_agents/run_context.py,sha256=0kWvOKBzQzx6FbdtDVAoeCOmiRGssN6X4n8YPX_oLBY,687
 grasp_agents/runner.py,sha256=JL2wSKahbPYVd56NRB09cwco43sjhZPI4XYFCZyOXOA,5173
 grasp_agents/usage_tracker.py,sha256=ZQfVUUpG0C89hyPWT_JgXnjQOxoYmumcQ9t-aCfcMo8,3561
 grasp_agents/utils.py,sha256=qKmGBwrQHw1-BgqRLuGTPKGs3J_zbrpk3nxnP1iZBiQ,6152
@@ -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=2XsPB-BbM-Y2xNxsKmO0JOJOD_UYj6ndGMjfLkGPAK4,8279
+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
@@ -30,9 +30,9 @@ grasp_agents/openai/completion_converters.py,sha256=UlDeQSl0AEFUS-QI5e8rrjfmXZoj
 grasp_agents/openai/content_converters.py,sha256=sMsZhoatuL_8t0IdVaGWIVZLB4nyi1ajD61GewQmeY4,2503
 grasp_agents/openai/converters.py,sha256=RKOfMbIJmfFQ7ot0RGR6wrdMbR6_L7PB0UZwxwgM88g,4691
 grasp_agents/openai/message_converters.py,sha256=fhSN81uK51EGbLyM2-f0MvPX_UBrMy7SF3JQPo-dkXg,4686
-grasp_agents/openai/openai_llm.py,sha256=QjxrZ4fM_FX3ncBjehUjWPCCiI62u_W2XDi7nth1WrY,9737
+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=BQ2k8dJY0jTMmidXZdK7JLO2YIQkmkp5boF1fT1o6uQ,10838
+grasp_agents/processors/base_processor.py,sha256=2ks_jpeye0XW9k3BcAhD9XO8NctspOV2py9ZAD-dO6c,10942
 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.11.dist-info/METADATA,sha256=BkVyEN63RzGsCIJCnm5S38EI2ua9NcbPmr3lRCmWPGs,7021
-grasp_agents-0.5.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-grasp_agents-0.5.11.dist-info/licenses/LICENSE.md,sha256=-nNNdWqGB8gJ2O-peFQ2Irshv5tW5pHKyTcYkwvH7CE,1201
-grasp_agents-0.5.11.dist-info/RECORD,,
+grasp_agents-0.5.13.dist-info/METADATA,sha256=3ElnYJXh0BeigwCOvFoy4TxT-znLGjEP1rn6rDJQik0,6633
+grasp_agents-0.5.13.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+grasp_agents-0.5.13.dist-info/licenses/LICENSE.md,sha256=-nNNdWqGB8gJ2O-peFQ2Irshv5tW5pHKyTcYkwvH7CE,1201
+grasp_agents-0.5.13.dist-info/RECORD,,