grasp_agents 0.2.11__py3-none-any.whl → 0.3.1__py3-none-any.whl

grasp_agents/workflow/looped_agent.py

@@ -1,134 +0,0 @@
-from collections.abc import Sequence
-from itertools import pairwise
-from logging import getLogger
-from typing import Any, ClassVar, Generic, Protocol, TypeVar, cast, final
-
-from ..agent_message_pool import AgentMessage, AgentMessagePool
-from ..comm_agent import CommunicatingAgent
-from ..run_context import CtxT, RunContextWrapper
-from ..typing.io import AgentID, AgentState, InT, OutT
-from .workflow_agent import WorkflowAgent
-
-logger = getLogger(__name__)
-
-_EH_OutT = TypeVar("_EH_OutT", contravariant=True)  # noqa: PLC0105
-
-
-class ExitWorkflowLoopHandler(Protocol[_EH_OutT, CtxT]):
-    def __call__(
-        self,
-        output_message: AgentMessage[_EH_OutT, Any],
-        ctx: RunContextWrapper[CtxT] | None,
-        **kwargs: Any,
-    ) -> bool: ...
-
-
-class LoopedWorkflowAgent(WorkflowAgent[InT, OutT, CtxT], Generic[InT, OutT, CtxT]):
-    _generic_arg_to_instance_attr_map: ClassVar[dict[int, str]] = {
-        0: "_in_type",
-        1: "_out_type",
-    }
-
-    def __init__(
-        self,
-        agent_id: AgentID,
-        subagents: Sequence[CommunicatingAgent[Any, Any, Any, CtxT]],
-        exit_agent: CommunicatingAgent[Any, OutT, Any, CtxT],
-        message_pool: AgentMessagePool[CtxT] | None = None,
-        recipient_ids: list[AgentID] | None = None,
-        dynamic_routing: bool = False,
-        max_iterations: int = 10,
-        **kwargs: Any,  # noqa: ARG002
-    ) -> None:
-        super().__init__(
-            subagents=subagents,
-            agent_id=agent_id,
-            start_agent=subagents[0],
-            end_agent=exit_agent,
-            message_pool=message_pool,
-            recipient_ids=recipient_ids,
-            dynamic_routing=dynamic_routing,
-        )
-
-        for prev_agent, agent in pairwise(subagents):
-            if prev_agent.out_type != agent.in_type:
-                raise ValueError(
-                    f"Output type {prev_agent.out_type} of agent {prev_agent.agent_id} "
-                    f"does not match input type {agent.in_type} of agent "
-                    f"{agent.agent_id}"
-                )
-        if subagents[-1].out_type != subagents[0].in_type:
-            raise ValueError(
-                f"Looped workflow's last agent output type {subagents[-1].out_type} "
-                f"does not match first agent input type {subagents[0].in_type}"
-            )
-
-        self._max_iterations = max_iterations
-
-        self._exit_workflow_loop_impl: ExitWorkflowLoopHandler[OutT, CtxT] | None = None
-
-    @property
-    def max_iterations(self) -> int:
-        return self._max_iterations
-
-    def exit_workflow_loop_handler(
-        self, func: ExitWorkflowLoopHandler[OutT, CtxT]
-    ) -> ExitWorkflowLoopHandler[OutT, CtxT]:
-        self._exit_workflow_loop_impl = func
-
-        return func
-
-    def _exit_workflow_loop(
-        self,
-        output_message: AgentMessage[OutT, Any],
-        *,
-        ctx: RunContextWrapper[CtxT] | None = None,
-        **kwargs: Any,
-    ) -> bool:
-        if self._exit_workflow_loop_impl:
-            return self._exit_workflow_loop_impl(output_message, ctx=ctx, **kwargs)
-
-        return False
-
-    @final
-    async def run(
-        self,
-        chat_inputs: 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 = in_message
-        num_iterations = 0
-        exit_message: AgentMessage[OutT, Any] | None = None
-
-        while True:
-            for subagent in self.subagents:
-                agent_message = await subagent.run(
-                    chat_inputs=chat_inputs,
-                    in_args=in_args,
-                    in_message=agent_message,
-                    entry_point=entry_point,
-                    forbid_state_change=forbid_state_change,
-                    ctx=ctx,
-                    **kwargs,
-                )
-
-                if subagent is self._end_agent:
-                    num_iterations += 1
-                    exit_message = cast("AgentMessage[OutT, AgentState]", agent_message)
-                    if self._exit_workflow_loop(exit_message, ctx=ctx):
-                        return exit_message
-                    if num_iterations >= self._max_iterations:
-                        logger.info(
-                            f"Max iterations reached ({self._max_iterations}). Exiting loop."
-                        )
-                        return exit_message
-
-                chat_inputs = None
-                in_args = None
-                entry_point = False

grasp_agents/workflow/sequential_agent.py

@@ -1,72 +0,0 @@
-from collections.abc import Sequence
-from itertools import pairwise
-from typing import Any, ClassVar, Generic, cast, final
-
-from ..agent_message_pool import AgentMessage, AgentMessagePool
-from ..comm_agent import CommunicatingAgent
-from ..run_context import CtxT, RunContextWrapper
-from ..typing.io import AgentID, InT, OutT
-from .workflow_agent import WorkflowAgent
-
-
-class SequentialWorkflowAgent(WorkflowAgent[InT, OutT, CtxT], Generic[InT, OutT, CtxT]):
-    _generic_arg_to_instance_attr_map: ClassVar[dict[int, str]] = {
-        0: "_in_type",
-        1: "_out_type",
-    }
-
-    def __init__(
-        self,
-        agent_id: AgentID,
-        subagents: Sequence[CommunicatingAgent[Any, Any, Any, CtxT]],
-        message_pool: AgentMessagePool[CtxT] | None = None,
-        recipient_ids: list[AgentID] | None = None,
-        dynamic_routing: bool = False,
-        **kwargs: Any,  # noqa: ARG002
-    ) -> None:
-        super().__init__(
-            subagents=subagents,
-            start_agent=subagents[0],
-            end_agent=subagents[-1],
-            agent_id=agent_id,
-            message_pool=message_pool,
-            recipient_ids=recipient_ids,
-            dynamic_routing=dynamic_routing,
-        )
-
-        for prev_agent, agent in pairwise(subagents):
-            if prev_agent.out_type != agent.in_type:
-                raise ValueError(
-                    f"Output type {prev_agent.out_type} of agent {prev_agent.agent_id} "
-                    f"does not match input type {agent.in_type} of agent "
-                    f"{agent.agent_id}"
-                )
-
-    @final
-    async def run(
-        self,
-        chat_inputs: 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 = in_message
-        for subagent in self.subagents:
-            agent_message = await subagent.run(
-                chat_inputs=chat_inputs,
-                in_args=in_args,
-                in_message=agent_message,
-                entry_point=entry_point,
-                forbid_state_change=forbid_state_change,
-                ctx=ctx,
-                **kwargs,
-            )
-            chat_inputs = None
-            in_args = None
-            entry_point = False
-
-        return cast("AgentMessage[OutT, Any]", agent_message)

grasp_agents/workflow/workflow_agent.py

@@ -1,88 +0,0 @@
-from abc import ABC, abstractmethod
-from collections.abc import Sequence
-from typing import Any, ClassVar, Generic
-
-from ..agent_message_pool import AgentMessage, AgentMessagePool
-from ..comm_agent import CommunicatingAgent
-from ..run_context import CtxT, RunContextWrapper
-from ..typing.io import AgentID, InT, OutT
-
-
-class WorkflowAgent(
-    CommunicatingAgent[InT, OutT, Any, CtxT], ABC, Generic[InT, OutT, CtxT]
-):
-    _generic_arg_to_instance_attr_map: ClassVar[dict[int, str]] = {
-        0: "_in_type",
-        1: "_out_type",
-    }
-
-    def __init__(
-        self,
-        agent_id: AgentID,
-        subagents: Sequence[CommunicatingAgent[Any, Any, Any, CtxT]],
-        start_agent: CommunicatingAgent[InT, Any, Any, CtxT],
-        end_agent: CommunicatingAgent[Any, OutT, Any, CtxT],
-        message_pool: AgentMessagePool[CtxT] | None = None,
-        recipient_ids: list[AgentID] | None = None,
-        dynamic_routing: bool = False,
-        **kwargs: Any,  # noqa: ARG002
-    ) -> None:
-        super().__init__(
-            agent_id=agent_id,
-            message_pool=message_pool,
-            recipient_ids=recipient_ids,
-            dynamic_routing=dynamic_routing,
-        )
-
-        if len(subagents) < 2:
-            raise ValueError("At least two steps are required")
-        if start_agent not in subagents:
-            raise ValueError("Start agent must be in the subagents list")
-        if end_agent not in subagents:
-            raise ValueError("End agent must be in the subagents list")
-
-        if start_agent.in_type != self.in_type:
-            raise ValueError(
-                f"Start agent's input type {start_agent.in_type} does not "
-                f"match workflow's input type {self._in_type}"
-            )
-        if end_agent.out_type != self.out_type:
-            raise ValueError(
-                f"End agent's output type {end_agent.out_type} does not "
-                f"match workflow's output type {self._out_type}"
-            )
-
-        self._subagents = subagents
-        self._start_agent = start_agent
-        self._end_agent = end_agent
-
-        for subagent in subagents:
-            assert not subagent.recipient_ids, (
-                "Subagents must not have recipient_ids set."
-            )
-
-    @property
-    def subagents(self) -> Sequence[CommunicatingAgent[Any, Any, Any, CtxT]]:
-        return self._subagents
-
-    @property
-    def start_agent(self) -> CommunicatingAgent[InT, Any, Any, CtxT]:
-        return self._start_agent
-
-    @property
-    def end_agent(self) -> CommunicatingAgent[Any, OutT, Any, CtxT]:
-        return self._end_agent
-
-    @abstractmethod
-    async def run(
-        self,
-        chat_inputs: 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,
-        **generation_kwargs: Any,
-    ) -> AgentMessage[OutT, Any]:
-        pass

grasp_agents-0.2.11.dist-info/RECORD

@@ -1,46 +0,0 @@
-grasp_agents/__init__.py,sha256=WPNUUFbucwli6oWIwxbckz0zpY4W2LnND0o7melYZOw,979
-grasp_agents/agent_message.py,sha256=eJV5n44t8EIE6M3jl48Ld7pmaW9dDhBX_FWm_u9yGWE,877
-grasp_agents/agent_message_pool.py,sha256=OKTXNEo9LAJTQJkzxmJ3TQgWw7WJKOzrKCJjeHpln6o,3158
-grasp_agents/base_agent.py,sha256=6WHBS17KqbiCVSREOiIJXlHfNrYb0AKvs0dEShogQGE,1363
-grasp_agents/cloud_llm.py,sha256=QqfnqLtInbekeT8R-sWMTrPiMwazyPaGoQ9K2wuh6gE,13336
-grasp_agents/comm_agent.py,sha256=U2R942X5yjPr27j9soUdueXtAzvduI7iON9QgoSDBAo,7271
-grasp_agents/costs_dict.yaml,sha256=2MFNWtkv5W5WSCcv1Cj13B1iQLVv5Ot9pS_KW2Gu2DA,2510
-grasp_agents/generics_utils.py,sha256=kw4Odte6Nvl4c9U7-mKPgXCavWZXo009zYDHAA0BR3g,6234
-grasp_agents/grasp_logging.py,sha256=H1GYhXdQvVkmauFDZ-KDwvVmPQHZUUm9sRqX_ObK2xI,1111
-grasp_agents/http_client.py,sha256=KZva2MjJjuI5ohUeU8RdTAImUnQYaqBrV2jDH8smbJw,738
-grasp_agents/llm.py,sha256=YClNxN9GUGaFHXhTU72z1AqW_Y726OC7kyVRYCnfhZ8,3682
-grasp_agents/llm_agent.py,sha256=UvmcvuLl8EWe51isSw9Oa9lxA9gQBJngRXj8LUMAd4w,16097
-grasp_agents/llm_agent_state.py,sha256=L54zUW5nAT-ubvEB7XNAQ84ExOgRlUFzc-Q49mUXUT0,2390
-grasp_agents/memory.py,sha256=X1YtVX8XxP5KnGPMW8BqjID8QK4hTG2obxoyhnnZ4pU,5575
-grasp_agents/printer.py,sha256=ZENcTITCcgSizvcUXnIQiFl_NcVHg-801Z-sbT0D8rg,5030
-grasp_agents/prompt_builder.py,sha256=A4z8B5nIfgn_cHnZCNyYjwA904UDiOaCyBc5VPu8168,8867
-grasp_agents/run_context.py,sha256=OpTdo32-WPD7XwE16LQphh2C7yrrkZ3C4Ia80pWeBDQ,2192
-grasp_agents/tool_orchestrator.py,sha256=XgKCOytPs4y4SBpS5i4eogCB428XaDnNIB3VxzJC_k0,6095
-grasp_agents/usage_tracker.py,sha256=4gy0XtfIBAjQHblEFpQuPelmxMGDvE7vw4c8Ccr1msk,3471
-grasp_agents/utils.py,sha256=KzoInW0sq-pwwUtgjtYMW8b0ivBH6MR0Zxv6Kqq3k3M,4510
-grasp_agents/openai/__init__.py,sha256=P7oa4tdJ2JKJv16frRi1C09FnrMayuOYomu5l95uayE,3196
-grasp_agents/openai/completion_converters.py,sha256=gt06DO2jrMsgJGHPIhoTHeM3R0yHqUjS0_VfaC6Xu4I,2146
-grasp_agents/openai/content_converters.py,sha256=6GI0D7xJalzsiawAJOyCUzTJTo0NQdpv87YKmfN0LYQ,2631
-grasp_agents/openai/converters.py,sha256=DBXBxow9oRG6pc8inpZBLiuUqHzVfpscmHFpN9bAdvc,5276
-grasp_agents/openai/message_converters.py,sha256=Gol5lFDu3eSagDsvWi_ql0QYcAT3-_5YCbU75xwuBK0,4823
-grasp_agents/openai/openai_llm.py,sha256=JZdM0YerpDJW1IX7BW8lawgGdUHUW1ka4JSEqKxXuUQ,6075
-grasp_agents/openai/tool_converters.py,sha256=72DmyNW9WylesRH-m5fi7H2Wu1U9gUXqMQ-UdX_M7mU,1068
-grasp_agents/rate_limiting/__init__.py,sha256=KRgtF_E7R3YfA2cpYcFcZ7wycV0pWVJ0xRQC7YhiIEQ,158
-grasp_agents/rate_limiting/rate_limiter_chunked.py,sha256=BPgkUXvhmZhTpZs2T6uujNFuxH_kYHiISuf6_-eNhUc,5544
-grasp_agents/rate_limiting/types.py,sha256=PbnNhEAcYedQdIpPJWud8HUVcxa_xZS2RDZu4c5jr40,1003
-grasp_agents/rate_limiting/utils.py,sha256=oEDWDNHYMUdxOOG49PlAJochkZq8nnVBCo6JxPc1iSo,2007
-grasp_agents/typing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-grasp_agents/typing/completion.py,sha256=_KDLx3Gtz7o-pEZrvAFgCZwDmkr2oQkxrL-2LSXHHsw,657
-grasp_agents/typing/content.py,sha256=VdnHdW8PHDCtX_ffvcwQz-7ypPUQNSGqHa3txFE_72Y,3676
-grasp_agents/typing/converters.py,sha256=yORIljRsVoKz7oj38pHLD6luIelM1RcYL_PqG_D4nWM,3086
-grasp_agents/typing/io.py,sha256=uxSvbD05UK5nIhPfDvXIoGuU6xRMW4USZq_4IgBeGCY,609
-grasp_agents/typing/message.py,sha256=DC24XMcUPG1MHDNZKT67yKWyaMTQNF5B0yDiUg8b54Q,3833
-grasp_agents/typing/tool.py,sha256=e0pTMnRcpMpGNVQ8muE9wnh7LdIgh92AqXDo9hMDxf0,1960
-grasp_agents/workflow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-grasp_agents/workflow/looped_agent.py,sha256=f7R-yVq4cP8zMTlvAF0GCpD0Izkyz5jUvUrxaXycusw,4710
-grasp_agents/workflow/sequential_agent.py,sha256=yU3X28rqZo-xVdkbEETb9pBCBG7v1GK7oxtbl5xLdCo,2526
-grasp_agents/workflow/workflow_agent.py,sha256=T1DpSipIkcrC_WJKR6Ho_cSHmVltCcF6Ve1F7isPRp0,3031
-grasp_agents-0.2.11.dist-info/METADATA,sha256=xXvakAAM5nycFuJf70buNFqQ5BMNyMbI7rdFZHHwphc,7188
-grasp_agents-0.2.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-grasp_agents-0.2.11.dist-info/licenses/LICENSE.md,sha256=-nNNdWqGB8gJ2O-peFQ2Irshv5tW5pHKyTcYkwvH7CE,1201
-grasp_agents-0.2.11.dist-info/RECORD,,