grasp_agents 0.2.4__tar.gz → 0.2.5__tar.gz

{grasp_agents-0.2.4 → grasp_agents-0.2.5}/PKG-INFO

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

{grasp_agents-0.2.4 → grasp_agents-0.2.5}/pyproject.toml

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

{grasp_agents-0.2.4 → grasp_agents-0.2.5}/src/grasp_agents/comm_agent.py

@@ -51,7 +51,7 @@ class CommunicatingAgent(
         self._in_type: type[InT]
         super().__init__(agent_id=agent_id, **kwargs)
 
-        self._rcv_args_type_adapter: TypeAdapter[InT] = TypeAdapter(self._in_type)
+        self._in_args_type_adapter: TypeAdapter[InT] = TypeAdapter(self._in_type)
         self.recipient_ids = recipient_ids or []
 
         self._message_pool = message_pool or AgentMessagePool()
@@ -102,7 +102,7 @@ class CommunicatingAgent(
         chat_inputs: Any | None = None,
         *,
         ctx: RunContextWrapper[CtxT] | None = None,
-        rcv_message: AgentMessage[InT, AgentState] | None = None,
+        in_message: AgentMessage[InT, AgentState] | None = None,
         entry_point: bool = False,
         forbid_state_change: bool = False,
         **kwargs: Any,
@@ -113,7 +113,7 @@ class CommunicatingAgent(
         self, ctx: RunContextWrapper[CtxT] | None = None, **run_kwargs: Any
     ) -> None:
         output_message = await self.run(
-            ctx=ctx, rcv_message=None, entry_point=True, **run_kwargs
+            ctx=ctx, in_message=None, entry_point=True, **run_kwargs
         )
         await self.post_message(output_message)
 
@@ -140,8 +140,8 @@ class CommunicatingAgent(
         ctx: RunContextWrapper[CtxT] | None = None,
         **run_kwargs: Any,
     ) -> None:
-        rcv_message = cast("AgentMessage[InT, AgentState]", message)
-        out_message = await self.run(ctx=ctx, rcv_message=rcv_message, **run_kwargs)
+        in_message = cast("AgentMessage[InT, AgentState]", message)
+        out_message = await self.run(ctx=ctx, in_message=in_message, **run_kwargs)
 
         if self._exit_condition(output_message=out_message, ctx=ctx):
             await self._message_pool.stop_all()
@@ -199,14 +199,14 @@ class CommunicatingAgent(
                 inp: InT,
                 ctx: RunContextWrapper[CtxT] | None = None,
             ) -> OutT:
-                rcv_args = in_type.model_validate(inp)
-                rcv_message = AgentMessage[in_type, AgentState](
-                    payloads=[rcv_args],
+                in_args = in_type.model_validate(inp)
+                in_message = AgentMessage[in_type, AgentState](
+                    payloads=[in_args],
                     sender_id="<tool_user>",
                     recipient_ids=[agent_instance.agent_id],
                 )
                 agent_result = await agent_instance.run(
-                    rcv_message=rcv_message,
+                    in_message=in_message,
                     entry_point=False,
                     forbid_state_change=True,
                     ctx=ctx,

{grasp_agents-0.2.4 → grasp_agents-0.2.5}/src/grasp_agents/llm_agent.py

@@ -52,7 +52,7 @@ class ParseOutputHandler(Protocol[InT, OutT, CtxT]):
         self,
         conversation: Conversation,
         *,
-        rcv_args: InT | None,
+        in_args: InT | None,
         batch_idx: int,
         ctx: RunContextWrapper[CtxT] | None,
     ) -> OutT: ...
@@ -74,8 +74,8 @@ class LLMAgent(
         # LLM
         llm: LLM[LLMSettings, Converters],
         # Input prompt template (combines user and received arguments)
-        inp_prompt: LLMPrompt | None = None,
-        inp_prompt_path: str | Path | None = None,
+        in_prompt: LLMPrompt | None = None,
+        in_prompt_path: str | Path | None = None,
         # System prompt template
         sys_prompt: LLMPrompt | None = None,
         sys_prompt_path: str | Path | None = None,
@@ -119,13 +119,13 @@ class LLMAgent(
 
         # Prompt builder
         sys_prompt = get_prompt(prompt_text=sys_prompt, prompt_path=sys_prompt_path)
-        inp_prompt = get_prompt(prompt_text=inp_prompt, prompt_path=inp_prompt_path)
+        in_prompt = get_prompt(prompt_text=in_prompt, prompt_path=in_prompt_path)
         self._prompt_builder: PromptBuilder[InT, CtxT] = PromptBuilder[
             self.in_type, CtxT
         ](
             agent_id=self._agent_id,
             sys_prompt=sys_prompt,
-            inp_prompt=inp_prompt,
+            in_prompt=in_prompt,
             sys_args_schema=sys_args_schema,
             usr_args_schema=usr_args_schema,
         )
@@ -159,14 +159,14 @@ class LLMAgent(
         return self._prompt_builder.sys_prompt
 
     @property
-    def inp_prompt(self) -> LLMPrompt | None:
-        return self._prompt_builder.inp_prompt
+    def in_prompt(self) -> LLMPrompt | None:
+        return self._prompt_builder.in_prompt
 
     def _parse_output(
         self,
         conversation: Conversation,
         *,
-        rcv_args: InT | None = None,
+        in_args: InT | None = None,
         batch_idx: int = 0,
         ctx: RunContextWrapper[CtxT] | None = None,
     ) -> OutT:
@@ -180,7 +180,7 @@ class LLMAgent(
 
             return self._parse_output_impl(
                 conversation=conversation,
-                rcv_args=rcv_args,
+                in_args=in_args,
                 batch_idx=batch_idx,
                 ctx=ctx,
             )
@@ -194,21 +194,21 @@ class LLMAgent(
     @staticmethod
     def _validate_run_inputs(
         chat_inputs: LLMPrompt | Sequence[str | ImageData] | None = None,
-        rcv_args: InT | Sequence[InT] | None = None,
-        rcv_message: AgentMessage[InT, AgentState] | None = None,
+        in_args: InT | Sequence[InT] | None = None,
+        in_message: AgentMessage[InT, AgentState] | None = None,
         entry_point: bool = False,
     ) -> None:
         multiple_inputs_err_message = (
-            "Only one of chat_inputs, rcv_args, or rcv_message must be provided."
+            "Only one of chat_inputs, in_args, or in_message must be provided."
         )
-        if chat_inputs is not None and rcv_args is not None:
+        if chat_inputs is not None and in_args is not None:
             raise ValueError(multiple_inputs_err_message)
-        if chat_inputs is not None and rcv_message is not None:
+        if chat_inputs is not None and in_message is not None:
             raise ValueError(multiple_inputs_err_message)
-        if rcv_args is not None and rcv_message is not None:
+        if in_args is not None and in_message is not None:
             raise ValueError(multiple_inputs_err_message)
 
-        if entry_point and rcv_message is not None:
+        if entry_point and in_message is not None:
             raise ValueError(
                 "Entry point agent cannot receive messages from other agents."
             )
@@ -218,8 +218,8 @@ class LLMAgent(
         self,
         chat_inputs: LLMPrompt | Sequence[str | ImageData] | None = None,
         *,
-        rcv_message: AgentMessage[InT, AgentState] | None = None,
-        rcv_args: InT | Sequence[InT] | None = None,
+        in_message: AgentMessage[InT, AgentState] | None = None,
+        in_args: InT | Sequence[InT] | None = None,
         entry_point: bool = False,
         ctx: RunContextWrapper[CtxT] | None = None,
         forbid_state_change: bool = False,
@@ -236,8 +236,8 @@ class LLMAgent(
 
         self._validate_run_inputs(
             chat_inputs=chat_inputs,
-            rcv_args=rcv_args,
-            rcv_message=rcv_message,
+            in_args=in_args,
+            in_message=in_message,
             entry_point=entry_point,
         )
 
@@ -249,12 +249,12 @@ class LLMAgent(
         # 2. Set agent state
 
         cur_state = self.state.model_copy(deep=True)
-        rcv_state = rcv_message.sender_state if rcv_message else None
+        in_state = in_message.sender_state if in_message else None
         prev_mh_len = len(cur_state.message_history)
 
-        state = LLMAgentState.from_cur_and_rcv_states(
+        state = LLMAgentState.from_cur_and_in_states(
             cur_state=cur_state,
-            rcv_state=rcv_state,
+            in_state=in_state,
             sys_prompt=formatted_sys_prompt,
             strategy=self.set_state_strategy,
             set_agent_state_impl=self._set_agent_state_impl,
@@ -264,16 +264,16 @@ class LLMAgent(
         self._print_sys_msg(state=state, prev_mh_len=prev_mh_len, ctx=ctx)
 
         # 3. Make and add user messages (can be empty)
-        _rcv_args_batch: Sequence[InT] | None = None
-        if rcv_message is not None:
-            _rcv_args_batch = rcv_message.payloads
-        elif rcv_args is not None:
-            _rcv_args_batch = rcv_args if isinstance(rcv_args, Sequence) else [rcv_args]  # type: ignore[assignment]
+        _in_args_batch: Sequence[InT] | None = None
+        if in_message is not None:
+            _in_args_batch = in_message.payloads
+        elif in_args is not None:
+            _in_args_batch = in_args if isinstance(in_args, Sequence) else [in_args]  # type: ignore[assignment]
 
         user_message_batch = self._prompt_builder.make_user_messages(
             chat_inputs=chat_inputs,
             usr_args=usr_args,
-            rcv_args_batch=_rcv_args_batch,
+            in_args_batch=_in_args_batch,
             entry_point=entry_point,
             ctx=ctx,
         )
@@ -291,14 +291,14 @@ class LLMAgent(
 
         # 5. Parse outputs
         batch_size = state.message_history.batch_size
-        rcv_args_batch = rcv_message.payloads if rcv_message else batch_size * [None]
+        in_args_batch = in_message.payloads if in_message else batch_size * [None]
         val_output_batch = [
             self._out_type_adapter.validate_python(
-                self._parse_output(conversation=conv, rcv_args=rcv_args, ctx=ctx)
+                self._parse_output(conversation=conv, in_args=in_args, ctx=ctx)
             )
-            for conv, rcv_args in zip(
+            for conv, in_args in zip(
                 state.message_history.batched_conversations,
-                rcv_args_batch,
+                in_args_batch,
                 strict=False,
             )
         ]
@@ -313,10 +313,10 @@ class LLMAgent(
                 recipient_ids=recipient_ids,
                 chat_inputs=chat_inputs,
                 sys_prompt=self.sys_prompt,
-                inp_prompt=self.inp_prompt,
+                in_prompt=self.in_prompt,
                 sys_args=sys_args,
                 usr_args=usr_args,
-                rcv_args=(rcv_message.payloads if rcv_message is not None else None),
+                in_args=(in_message.payloads if in_message is not None else None),
                 outputs=val_output_batch,
                 state=state,
             )
@@ -369,10 +369,10 @@ class LLMAgent(
 
         return func
 
-    def format_inp_args_handler(
+    def format_in_args_handler(
         self, func: FormatInputArgsHandler[InT, CtxT]
     ) -> FormatInputArgsHandler[InT, CtxT]:
-        self._prompt_builder.format_inp_args_impl = func
+        self._prompt_builder.format_in_args_impl = func
 
         return func
 
@@ -411,8 +411,8 @@ class LLMAgent(
         if cur_cls._format_sys_args is not base_cls._format_sys_args:  # noqa: SLF001
             self._prompt_builder.format_sys_args_impl = self._format_sys_args
 
-        if cur_cls._format_inp_args is not base_cls._format_inp_args:  # noqa: SLF001
-            self._prompt_builder.format_inp_args_impl = self._format_inp_args
+        if cur_cls._format_in_args is not base_cls._format_in_args:  # noqa: SLF001
+            self._prompt_builder.format_in_args_impl = self._format_in_args
 
         if cur_cls._set_agent_state is not base_cls._set_agent_state:  # noqa: SLF001
             self._set_agent_state_impl = self._set_agent_state
@@ -438,23 +438,23 @@ class LLMAgent(
             "if it's intended to be used as the system arguments formatter."
         )
 
-    def _format_inp_args(
+    def _format_in_args(
         self,
         *,
         usr_args: LLMPromptArgs,
-        rcv_args: InT,
+        in_args: InT,
         batch_idx: int = 0,
         ctx: RunContextWrapper[CtxT] | None = None,
     ) -> LLMFormattedArgs:
         raise NotImplementedError(
-            "LLMAgent._format_inp_args must be overridden by a subclass"
+            "LLMAgent._format_in_args must be overridden by a subclass"
         )
 
     def _set_agent_state(
         self,
         cur_state: LLMAgentState,
         *,
-        rcv_state: AgentState | None,
+        in_state: AgentState | None,
         sys_prompt: LLMPrompt | None,
         ctx: RunContextWrapper[Any] | None,
     ) -> LLMAgentState:

{grasp_agents-0.2.4 → grasp_agents-0.2.5}/src/grasp_agents/llm_agent_state.py

@@ -15,7 +15,7 @@ class SetAgentState(Protocol):
         self,
         cur_state: "LLMAgentState",
         *,
-        rcv_state: AgentState | None,
+        in_state: AgentState | None,
         sys_prompt: LLMPrompt | None,
         ctx: RunContextWrapper[Any] | None,
     ) -> "LLMAgentState": ...
@@ -29,11 +29,11 @@ class LLMAgentState(AgentState):
         return self.message_history.batch_size
 
     @classmethod
-    def from_cur_and_rcv_states(
+    def from_cur_and_in_states(
         cls,
         cur_state: "LLMAgentState",
         *,
-        rcv_state: Optional["AgentState"] = None,
+        in_state: Optional["AgentState"] = None,
         sys_prompt: LLMPrompt | None = None,
         strategy: SetAgentStateStrategy = "from_sender",
         set_agent_state_impl: SetAgentState | None = None,
@@ -50,13 +50,13 @@ class LLMAgentState(AgentState):
             upd_mh.reset(sys_prompt)
 
         elif strategy == "from_sender":
-            rcv_mh = (
-                rcv_state.message_history
-                if rcv_state and isinstance(rcv_state, "LLMAgentState")
+            in_mh = (
+                in_state.message_history
+                if in_state and isinstance(in_state, "LLMAgentState")
                 else None
             )
-            if rcv_mh:
-                upd_mh = deepcopy(rcv_mh)
+            if in_mh:
+                in_mh = deepcopy(in_mh)
             else:
                 upd_mh.reset(sys_prompt)
 
@@ -66,7 +66,7 @@ class LLMAgentState(AgentState):
             )
             return set_agent_state_impl(
                 cur_state=cur_state,
-                rcv_state=rcv_state,
+                in_state=in_state,
                 sys_prompt=sys_prompt,
                 ctx=ctx,
             )

{grasp_agents-0.2.4 → grasp_agents-0.2.5}/src/grasp_agents/prompt_builder.py

@@ -35,7 +35,7 @@ class FormatInputArgsHandler(Protocol[InT, CtxT]):
         self,
         *,
         usr_args: LLMPromptArgs,
-        rcv_args: InT,
+        in_args: InT,
         batch_idx: int,
         ctx: RunContextWrapper[CtxT] | None,
     ) -> LLMFormattedArgs: ...
@@ -48,7 +48,7 @@ class PromptBuilder(AutoInstanceAttributesMixin, Generic[InT, CtxT]):
         self,
         agent_id: str,
         sys_prompt: LLMPrompt | None,
-        inp_prompt: LLMPrompt | None,
+        in_prompt: LLMPrompt | None,
         sys_args_schema: type[LLMPromptArgs],
         usr_args_schema: type[LLMPromptArgs],
     ):
@@ -57,13 +57,13 @@ class PromptBuilder(AutoInstanceAttributesMixin, Generic[InT, CtxT]):
 
         self._agent_id = agent_id
         self.sys_prompt = sys_prompt
-        self.inp_prompt = inp_prompt
+        self.in_prompt = in_prompt
         self.sys_args_schema = sys_args_schema
         self.usr_args_schema = usr_args_schema
         self.format_sys_args_impl: FormatSystemArgsHandler[CtxT] | None = None
-        self.format_inp_args_impl: FormatInputArgsHandler[InT, CtxT] | None = None
+        self.format_in_args_impl: FormatInputArgsHandler[InT, CtxT] | None = None
 
-        self._rcv_args_type_adapter: TypeAdapter[InT] = TypeAdapter(self._in_type)
+        self._in_args_type_adapter: TypeAdapter[InT] = TypeAdapter(self._in_type)
 
     def _format_sys_args(
         self,
@@ -75,31 +75,31 @@ class PromptBuilder(AutoInstanceAttributesMixin, Generic[InT, CtxT]):
 
         return sys_args.model_dump(exclude_unset=True)
 
-    def _format_inp_args(
+    def _format_in_args(
         self,
         *,
         usr_args: LLMPromptArgs,
-        rcv_args: InT,
+        in_args: InT,
         batch_idx: int = 0,
         ctx: RunContextWrapper[CtxT] | None = None,
     ) -> LLMFormattedArgs:
-        if self.format_inp_args_impl:
-            return self.format_inp_args_impl(
-                usr_args=usr_args, rcv_args=rcv_args, batch_idx=batch_idx, ctx=ctx
+        if self.format_in_args_impl:
+            return self.format_in_args_impl(
+                usr_args=usr_args, in_args=in_args, batch_idx=batch_idx, ctx=ctx
             )
 
-        if not isinstance(rcv_args, BaseModel) and rcv_args is not None:
+        if not isinstance(in_args, BaseModel) and in_args is not None:
             raise TypeError(
                 "Cannot apply default formatting to non-BaseModel received arguments."
             )
 
         usr_args_ = usr_args
-        rcv_args_ = DummySchema() if rcv_args is None else rcv_args
+        in_args_ = DummySchema() if in_args is None else in_args
 
         usr_args_dump = usr_args_.model_dump(exclude_unset=True)
-        rcv_args_dump = rcv_args_.model_dump(exclude={"selected_recipient_ids"})
+        in_args_dump = in_args_.model_dump(exclude={"selected_recipient_ids"})
 
-        return usr_args_dump | rcv_args_dump
+        return usr_args_dump | in_args_dump
 
     def make_sys_prompt(
         self,
@@ -122,13 +122,13 @@ class PromptBuilder(AutoInstanceAttributesMixin, Generic[InT, CtxT]):
     ) -> Sequence[UserMessage]:
         return [UserMessage.from_content_parts(content_parts, model_id=self._agent_id)]
 
-    def _usr_messages_from_rcv_args(
-        self, rcv_args_batch: Sequence[InT]
+    def _usr_messages_from_in_args(
+        self, in_args_batch: Sequence[InT]
     ) -> Sequence[UserMessage]:
         return [
             UserMessage.from_text(
-                self._rcv_args_type_adapter.dump_json(
-                    rcv,
+                self._in_args_type_adapter.dump_json(
+                    inp,
                     exclude_unset=True,
                     indent=2,
                     exclude={"selected_recipient_ids"},
@@ -136,46 +136,46 @@ class PromptBuilder(AutoInstanceAttributesMixin, Generic[InT, CtxT]):
                 ).decode("utf-8"),
                 model_id=self._agent_id,
             )
-            for rcv in rcv_args_batch
+            for inp in in_args_batch
         ]
 
     def _usr_messages_from_prompt_template(
         self,
-        inp_prompt: LLMPrompt,
+        in_prompt: LLMPrompt,
         usr_args: UserRunArgs | None = None,
-        rcv_args_batch: Sequence[InT] | None = None,
+        in_args_batch: Sequence[InT] | None = None,
         ctx: RunContextWrapper[CtxT] | None = None,
     ) -> Sequence[UserMessage]:
-        usr_args_batch_, rcv_args_batch_ = self._make_batched(usr_args, rcv_args_batch)
+        usr_args_batch_, in_args_batch_ = self._make_batched(usr_args, in_args_batch)
 
         val_usr_args_batch_ = [
             self.usr_args_schema.model_validate(u) for u in usr_args_batch_
         ]
-        val_rcv_args_batch_ = [
-            self._rcv_args_type_adapter.validate_python(rcv) for rcv in rcv_args_batch_
+        val_in_args_batch_ = [
+            self._in_args_type_adapter.validate_python(inp) for inp in in_args_batch_
         ]
 
-        formatted_inp_args_batch = [
-            self._format_inp_args(
-                usr_args=val_usr_args, rcv_args=val_rcv_args, batch_idx=i, ctx=ctx
+        formatted_in_args_batch = [
+            self._format_in_args(
+                usr_args=val_usr_args, in_args=val_in_args, batch_idx=i, ctx=ctx
             )
-            for i, (val_usr_args, val_rcv_args) in enumerate(
-                zip(val_usr_args_batch_, val_rcv_args_batch_, strict=False)
+            for i, (val_usr_args, val_in_args) in enumerate(
+                zip(val_usr_args_batch_, val_in_args_batch_, strict=False)
             )
         ]
 
         return [
             UserMessage.from_formatted_prompt(
-                prompt_template=inp_prompt, prompt_args=inp_args
+                prompt_template=in_prompt, prompt_args=in_args
             )
-            for inp_args in formatted_inp_args_batch
+            for in_args in formatted_in_args_batch
         ]
 
     def make_user_messages(
         self,
         chat_inputs: LLMPrompt | Sequence[str | ImageData] | None = None,
         usr_args: UserRunArgs | None = None,
-        rcv_args_batch: Sequence[InT] | None = None,
+        in_args_batch: Sequence[InT] | None = None,
         entry_point: bool = False,
         ctx: RunContextWrapper[CtxT] | None = None,
     ) -> Sequence[UserMessage]:
@@ -197,15 +197,20 @@ class PromptBuilder(AutoInstanceAttributesMixin, Generic[InT, CtxT]):
                 return self._usr_messages_from_content_parts(chat_inputs)
 
         # 2) No input prompt template + received args → raw JSON messages
-        if self.inp_prompt is None and rcv_args_batch:
-            return self._usr_messages_from_rcv_args(rcv_args_batch)
+        if self.in_prompt is None and in_args_batch:
+            return self._usr_messages_from_in_args(in_args_batch)
 
         # 3) Input prompt template + any args → batch & format
-        if self.inp_prompt is not None:
+        if self.in_prompt is not None:
+            if in_args_batch and not isinstance(in_args_batch[0], BaseModel):
+                raise TypeError(
+                    "Cannot use the input prompt template with "
+                    "non-BaseModel received arguments."
+                )
             return self._usr_messages_from_prompt_template(
-                inp_prompt=self.inp_prompt,
+                in_prompt=self.in_prompt,
                 usr_args=usr_args,
-                rcv_args_batch=rcv_args_batch,
+                in_args_batch=in_args_batch,
                 ctx=ctx,
             )
 
@@ -214,19 +219,19 @@ class PromptBuilder(AutoInstanceAttributesMixin, Generic[InT, CtxT]):
     def _make_batched(
         self,
         usr_args: UserRunArgs | None = None,
-        rcv_args_batch: Sequence[InT] | None = None,
+        in_args_batch: Sequence[InT] | None = None,
     ) -> tuple[Sequence[LLMPromptArgs | DummySchema], Sequence[InT | DummySchema]]:
         usr_args_batch_ = (
             usr_args if isinstance(usr_args, list) else [usr_args or DummySchema()]
         )
-        rcv_args_batch_ = rcv_args_batch or [DummySchema()]
+        in_args_batch_ = in_args_batch or [DummySchema()]
 
         # Broadcast singleton → match lengths
-        if len(usr_args_batch_) == 1 and len(rcv_args_batch_) > 1:
-            usr_args_batch_ = [deepcopy(usr_args_batch_[0]) for _ in rcv_args_batch_]
-        if len(rcv_args_batch_) == 1 and len(usr_args_batch_) > 1:
-            rcv_args_batch_ = [deepcopy(rcv_args_batch_[0]) for _ in usr_args_batch_]
-        if len(usr_args_batch_) != len(rcv_args_batch_):
+        if len(usr_args_batch_) == 1 and len(in_args_batch_) > 1:
+            usr_args_batch_ = [deepcopy(usr_args_batch_[0]) for _ in in_args_batch_]
+        if len(in_args_batch_) == 1 and len(usr_args_batch_) > 1:
+            in_args_batch_ = [deepcopy(in_args_batch_[0]) for _ in usr_args_batch_]
+        if len(usr_args_batch_) != len(in_args_batch_):
             raise ValueError("User args and received args must have the same length")
 
-        return usr_args_batch_, rcv_args_batch_
+        return usr_args_batch_, in_args_batch_

{grasp_agents-0.2.4 → grasp_agents-0.2.5}/src/grasp_agents/run_context.py

@@ -34,10 +34,10 @@ class InteractionRecord(BaseModel, Generic[InT, OutT, StateT]):
     state: StateT
     chat_inputs: LLMPrompt | Sequence[str | ImageData] | None = None
     sys_prompt: LLMPrompt | None = None
-    inp_prompt: LLMPrompt | None = None
+    in_prompt: LLMPrompt | None = None
     sys_args: SystemRunArgs | None = None
     usr_args: UserRunArgs | None = None
-    rcv_args: Sequence[InT] | None = None
+    in_args: Sequence[InT] | None = None
     outputs: Sequence[OutT]
 
     model_config = ConfigDict(extra="forbid", frozen=True)

{grasp_agents-0.2.4 → grasp_agents-0.2.5}/src/grasp_agents/workflow/looped_agent.py

@@ -81,14 +81,14 @@ class LoopedWorkflowAgent(WorkflowAgent[InT, OutT, CtxT], Generic[InT, OutT, Ctx
         self,
         chat_inputs: Any | None = None,
         *,
-        rcv_args: InT | Sequence[InT] | None = None,
-        rcv_message: AgentMessage[InT, Any] | None = None,
+        in_args: InT | Sequence[InT] | None = None,
+        in_message: AgentMessage[InT, Any] | None = None,
         ctx: RunContextWrapper[CtxT] | None = None,
         entry_point: bool = False,
         forbid_state_change: bool = False,
         **kwargs: Any,
     ) -> AgentMessage[OutT, AgentState]:
-        agent_message = rcv_message
+        agent_message = in_message
         num_iterations = 0
         exit_message: AgentMessage[OutT, Any] | None = None
 
@@ -96,8 +96,8 @@ class LoopedWorkflowAgent(WorkflowAgent[InT, OutT, CtxT], Generic[InT, OutT, Ctx
             for subagent in self.subagents:
                 agent_message = await subagent.run(
                     chat_inputs=chat_inputs,
-                    rcv_args=rcv_args,
-                    rcv_message=agent_message,
+                    in_args=in_args,
+                    in_message=agent_message,
                     entry_point=entry_point,
                     forbid_state_change=forbid_state_change,
                     ctx=ctx,
@@ -116,5 +116,5 @@ class LoopedWorkflowAgent(WorkflowAgent[InT, OutT, CtxT], Generic[InT, OutT, Ctx
                         return exit_message
 
                 chat_inputs = None
-                rcv_args = None
+                in_args = None
                 entry_point = False

{grasp_agents-0.2.4 → grasp_agents-0.2.5}/src/grasp_agents/workflow/sequential_agent.py

@@ -38,26 +38,26 @@ class SequentialWorkflowAgent(WorkflowAgent[InT, OutT, CtxT], Generic[InT, OutT,
         self,
         chat_inputs: Any | None = None,
         *,
-        rcv_args: InT | Sequence[InT] | None = None,
-        rcv_message: AgentMessage[InT, Any] | None = None,
+        in_args: InT | Sequence[InT] | None = None,
+        in_message: AgentMessage[InT, Any] | None = None,
         ctx: RunContextWrapper[CtxT] | None = None,
         entry_point: bool = False,
         forbid_state_change: bool = False,
         **kwargs: Any,
     ) -> AgentMessage[OutT, Any]:
-        agent_message = rcv_message
+        agent_message = in_message
         for subagent in self.subagents:
             agent_message = await subagent.run(
                 chat_inputs=chat_inputs,
-                rcv_args=rcv_args,
-                rcv_message=agent_message,
+                in_args=in_args,
+                in_message=agent_message,
                 entry_point=entry_point,
                 forbid_state_change=forbid_state_change,
                 ctx=ctx,
                 **kwargs,
             )
             chat_inputs = None
-            rcv_args = None
+            in_args = None
             entry_point = False
 
         return cast("AgentMessage[OutT, Any]", agent_message)

{grasp_agents-0.2.4 → grasp_agents-0.2.5}/src/grasp_agents/workflow/workflow_agent.py

@@ -63,8 +63,8 @@ class WorkflowAgent(
         self,
         chat_inputs: Any | None = None,
         *,
-        rcv_args: InT | Sequence[InT] | None = None,
-        rcv_message: AgentMessage[InT, Any] | None = None,
+        in_args: InT | Sequence[InT] | None = None,
+        in_message: AgentMessage[InT, Any] | None = None,
         ctx: RunContextWrapper[CtxT] | None = None,
         entry_point: bool = False,
         forbid_state_change: bool = False,