grasp_agents 0.3.1__tar.gz → 0.3.3__tar.gz

{grasp_agents-0.3.1 → grasp_agents-0.3.3}/PKG-INFO

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

{grasp_agents-0.3.1 → grasp_agents-0.3.3}/pyproject.toml

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

{grasp_agents-0.3.1 → grasp_agents-0.3.3}/src/grasp_agents/llm_agent.py

@@ -192,7 +192,7 @@ class LLMAgent(
 
         # 1. Make system prompt (can be None)
 
-        formatted_sys_prompt = self._prompt_builder.make_sys_prompt(
+        formatted_sys_prompt = self._prompt_builder.make_system_prompt(
             sys_args=sys_args, ctx=ctx
         )
 
@@ -313,17 +313,17 @@ class LLMAgent(
 
     # -- Decorators for custom implementations --
 
-    def make_sys_prompt(
+    def make_system_prompt(
         self, func: MakeSystemPromptHandler[CtxT]
     ) -> MakeSystemPromptHandler[CtxT]:
-        self._prompt_builder.make_sys_prompt_impl = func
+        self._prompt_builder.make_system_prompt_impl = func
 
         return func
 
-    def make_in_content(
+    def make_input_content(
         self, func: MakeInputContentHandler[InT_contra, CtxT]
     ) -> MakeInputContentHandler[InT_contra, CtxT]:
-        self._prompt_builder.make_in_content_impl = func
+        self._prompt_builder.make_input_content_impl = func
 
         return func
 
@@ -359,30 +359,28 @@ class LLMAgent(
         cur_cls = type(self)
         base_cls = LLMAgent[Any, Any, Any]
 
-        if cur_cls._make_sys_prompt_fn is not base_cls._make_sys_prompt_fn:  # noqa: SLF001
-            self._prompt_builder.make_sys_prompt_impl = self._make_sys_prompt_fn
+        if cur_cls._make_system_prompt is not base_cls._make_system_prompt:  # noqa: SLF001
+            self._prompt_builder.make_system_prompt_impl = self._make_system_prompt
 
-        if cur_cls._make_in_content_fn is not base_cls._make_in_content_fn:  # noqa: SLF001
-            self._prompt_builder.make_in_content_impl = self._make_in_content_fn
+        if cur_cls._make_input_content is not base_cls._make_input_content:  # noqa: SLF001
+            self._prompt_builder.make_input_content_impl = self._make_input_content
 
-        if cur_cls._set_memory_fn is not base_cls._set_memory_fn:  # noqa: SLF001
-            self._set_memory_impl = self._set_memory_fn
+        if cur_cls._set_memory is not base_cls._set_memory:  # noqa: SLF001
+            self._set_memory_impl = self._set_memory
 
-        if cur_cls._manage_memory_fn is not base_cls._manage_memory_fn:  # noqa: SLF001
-            self._policy_executor.manage_memory_impl = self._manage_memory_fn
+        if cur_cls._manage_memory is not base_cls._manage_memory:  # noqa: SLF001
+            self._policy_executor.manage_memory_impl = self._manage_memory
 
         if (
-            cur_cls._exit_tool_call_loop_fn is not base_cls._exit_tool_call_loop_fn  # noqa: SLF001
+            cur_cls._exit_tool_call_loop is not base_cls._exit_tool_call_loop  # noqa: SLF001
         ):
-            self._policy_executor.exit_tool_call_loop_impl = (
-                self._exit_tool_call_loop_fn
-            )
+            self._policy_executor.exit_tool_call_loop_impl = self._exit_tool_call_loop
 
         self._parse_output_impl: (
             ParseOutputHandler[InT_contra, OutT_co, CtxT] | None
         ) = None
 
-    def _make_sys_prompt_fn(
+    def _make_system_prompt(
         self, sys_args: LLMPromptArgs | None, *, ctx: RunContext[CtxT] | None = None
     ) -> str:
         raise NotImplementedError(
@@ -390,7 +388,7 @@ class LLMAgent(
             "if it's intended to be used as the system arguments formatter."
         )
 
-    def _make_in_content_fn(
+    def _make_input_content(
         self,
         *,
         in_args: InT_contra | None = None,
@@ -402,10 +400,10 @@ class LLMAgent(
             "LLMAgent._format_in_args must be overridden by a subclass"
         )
 
-    def _set_memory_fn(
+    def _set_memory(
         self,
         prev_memory: LLMAgentMemory,
-        in_args: InT_contra | Sequence[InT_contra] | None = None,
+        in_args: Sequence[InT_contra] | None = None,
         sys_prompt: LLMPrompt | None = None,
         ctx: RunContext[Any] | None = None,
     ) -> LLMAgentMemory:
@@ -413,7 +411,7 @@ class LLMAgent(
             "LLMAgent._set_memory must be overridden by a subclass"
         )
 
-    def _exit_tool_call_loop_fn(
+    def _exit_tool_call_loop(
         self,
         conversation: Messages,
         *,
@@ -424,7 +422,7 @@ class LLMAgent(
             "LLMAgent._exit_tool_call_loop must be overridden by a subclass"
         )
 
-    def _manage_memory_fn(
+    def _manage_memory(
         self,
         memory: LLMAgentMemory,
         *,

{grasp_agents-0.3.1 → grasp_agents-0.3.3}/src/grasp_agents/llm_policy_executor.py

@@ -28,6 +28,9 @@ from .typing.tool import BaseTool, NamedToolChoice, ToolCall, ToolChoice
 logger = getLogger(__name__)
 
 
+FINAL_ANSWER_TOOL_NAME = "final_answer"
+
+
 _FinalAnswerT = TypeVar("_FinalAnswerT")
 
 
@@ -102,7 +105,7 @@ class LLMPolicyExecutor(AutoInstanceAttributesMixin, Generic[_FinalAnswerT, CtxT
     def max_turns(self) -> int:
         return self._max_turns
 
-    def _exit_tool_call_loop_fn(
+    def _exit_tool_call_loop(
         self,
         conversation: Messages,
         *,
@@ -119,7 +122,7 @@ class LLMPolicyExecutor(AutoInstanceAttributesMixin, Generic[_FinalAnswerT, CtxT
 
         return not bool(conversation[-1].tool_calls)
 
-    def _manage_memory_fn(
+    def _manage_memory(
         self,
         memory: LLMAgentMemory,
         *,
@@ -318,7 +321,7 @@ class LLMPolicyExecutor(AutoInstanceAttributesMixin, Generic[_FinalAnswerT, CtxT
 
             # When final_answer_tool_name is None, we use exit_tool_call_loop_impl
             # to determine whether to exit the loop.
-            if self._final_answer_tool_name is None and self._exit_tool_call_loop_fn(
+            if self._final_answer_tool_name is None and self._exit_tool_call_loop(
                 conversation, ctx=ctx, num_turns=turns
             ):
                 return gen_message
@@ -353,7 +356,7 @@ class LLMPolicyExecutor(AutoInstanceAttributesMixin, Generic[_FinalAnswerT, CtxT
                 await self.call_tools(gen_message.tool_calls, memory=memory, ctx=ctx)
 
             # Apply the memory management function if provided.
-            self._manage_memory_fn(memory, ctx=ctx, num_turns=turns)
+            self._manage_memory(memory, ctx=ctx, num_turns=turns)
 
             # 4. Generate the next message based on the updated memory.
             #    In ReAct mode, we set tool_choice to "none" if we just called tools,
@@ -394,7 +397,7 @@ class LLMPolicyExecutor(AutoInstanceAttributesMixin, Generic[_FinalAnswerT, CtxT
         while True:
             conversation = memory.message_history.conversations[0]
 
-            if self._final_answer_tool_name is None and self._exit_tool_call_loop_fn(
+            if self._final_answer_tool_name is None and self._exit_tool_call_loop(
                 conversation, ctx=ctx, num_turns=turns
             ):
                 return
@@ -429,7 +432,7 @@ class LLMPolicyExecutor(AutoInstanceAttributesMixin, Generic[_FinalAnswerT, CtxT
                 ):
                     yield tool_message_event
 
-            self._manage_memory_fn(memory, ctx=ctx, num_turns=turns)
+            self._manage_memory(memory, ctx=ctx, num_turns=turns)
 
             tool_choice = (
                 "none" if (self._react_mode and gen_message.tool_calls) else "required"
@@ -458,11 +461,10 @@ class LLMPolicyExecutor(AutoInstanceAttributesMixin, Generic[_FinalAnswerT, CtxT
             )
 
         class FinalAnswerTool(BaseTool[self._final_answer_type, None, Any]):
-            name: str = "final_answer"
+            name: str = FINAL_ANSWER_TOOL_NAME
             description: str = (
-                "You must use this tool to provide the final answer. "
-                "Do not provide the final answer anywhere else. "
-                "Input arguments correspond to the final answer."
+                "You must call this tool to provide the final answer. "
+                "DO NOT output your answer before calling the tool. "
             )
 
             async def run(

{grasp_agents-0.3.1 → grasp_agents-0.3.3}/src/grasp_agents/processor.py

@@ -1,4 +1,4 @@
-from abc import ABC, abstractmethod
+from abc import ABC
 from collections.abc import AsyncIterator, Sequence
 from typing import Any, ClassVar, Generic, cast, final
 
@@ -20,7 +20,6 @@ class Processor(
         1: "_out_type",
     }
 
-    @abstractmethod
     def __init__(self, name: ProcName, **kwargs: Any) -> None:
         self._in_type: type[InT_contra]
         self._out_type: type[OutT_co]

{grasp_agents-0.3.1 → grasp_agents-0.3.3}/src/grasp_agents/prompt_builder.py

@@ -53,14 +53,14 @@ class PromptBuilder(AutoInstanceAttributesMixin, Generic[InT_contra, CtxT]):
         self.in_prompt_template = in_prompt_template
         self.sys_args_schema = sys_args_schema
         self.usr_args_schema = usr_args_schema
-        self.make_sys_prompt_impl: MakeSystemPromptHandler[CtxT] | None = None
-        self.make_in_content_impl: MakeInputContentHandler[InT_contra, CtxT] | None = (
-            None
-        )
+        self.make_system_prompt_impl: MakeSystemPromptHandler[CtxT] | None = None
+        self.make_input_content_impl: (
+            MakeInputContentHandler[InT_contra, CtxT] | None
+        ) = None
 
         self._in_args_type_adapter: TypeAdapter[InT_contra] = TypeAdapter(self._in_type)
 
-    def make_sys_prompt(
+    def make_system_prompt(
         self, sys_args: LLMPromptArgs | None = None, ctx: RunContext[CtxT] | None = None
     ) -> str | None:
         if self.sys_prompt_template is None:
@@ -76,8 +76,8 @@ class PromptBuilder(AutoInstanceAttributesMixin, Generic[InT_contra, CtxT]):
                     "provided."
                 )
 
-        if self.make_sys_prompt_impl:
-            return self.make_sys_prompt_impl(sys_args=val_sys_args, ctx=ctx)
+        if self.make_system_prompt_impl:
+            return self.make_system_prompt_impl(sys_args=val_sys_args, ctx=ctx)
 
         sys_args_dict = (
             val_sys_args.model_dump(exclude_unset=True) if val_sys_args else {}
@@ -85,7 +85,7 @@ class PromptBuilder(AutoInstanceAttributesMixin, Generic[InT_contra, CtxT]):
 
         return self.sys_prompt_template.format(**sys_args_dict)
 
-    def make_in_content(
+    def make_input_content(
         self,
         *,
         in_args: InT_contra | None,
@@ -97,8 +97,8 @@ class PromptBuilder(AutoInstanceAttributesMixin, Generic[InT_contra, CtxT]):
             in_args=in_args, usr_args=usr_args
         )
 
-        if self.make_in_content_impl:
-            return self.make_in_content_impl(
+        if self.make_input_content_impl:
+            return self.make_input_content_impl(
                 in_args=val_in_args, usr_args=val_usr_args, batch_idx=batch_idx, ctx=ctx
             )
 
@@ -126,7 +126,7 @@ class PromptBuilder(AutoInstanceAttributesMixin, Generic[InT_contra, CtxT]):
             return self._usr_messages_from_content_parts(chat_inputs)
 
         in_content_batch = [
-            self.make_in_content(
+            self.make_input_content(
                 in_args=in_args, usr_args=usr_args, batch_idx=i, ctx=ctx
             )
             for i, in_args in enumerate(in_args_batch or [None])

{grasp_agents-0.3.1 → grasp_agents-0.3.3}/src/grasp_agents/utils.py

@@ -88,6 +88,9 @@ def validate_obj_from_json_or_py_string(
     else:
         _selected_adapter = adapter
 
+    if _selected_adapter._type is str:  # type: ignore[arg-type]
+        return s
+
     try:
         if from_substring:
             parsed = parse_json_or_py_substring(s, return_none_on_failure=True)