grasp_agents 0.3.11__py3-none-any.whl → 0.4.0__py3-none-any.whl

grasp_agents/message_history.py

@@ -1,140 +0,0 @@
-import logging
-from collections.abc import Iterator, Sequence
-from copy import deepcopy
-
-from .typing.io import LLMPrompt
-from .typing.message import Message, Messages, SystemMessage
-
-logger = logging.getLogger(__name__)
-
-
-class MessageHistory:
-    def __init__(self, sys_prompt: LLMPrompt | None = None) -> None:
-        self._sys_prompt = sys_prompt
-        self._conversations: list[Messages]
-        self.reset()
-
-    @property
-    def sys_prompt(self) -> LLMPrompt | None:
-        return self._sys_prompt
-
-    def add_message_batch(self, message_batch: Sequence[Message]) -> None:
-        """
-        Adds a batch of messages to the current batched conversations.
-        This method verifies that the size of the input message batch matches
-        the expected batch size (self.batch_size).
-        If there is a mismatch, the method adjusts by duplicating either
-        the message or the conversation as necessary:
-
-        - If the message batch contains exactly one message and
-            self.batch_size > 1, the single message is duplicated to match
-            the batch size.
-        - If the message batch contains multiple messages but
-            self.batch_size == 1, the entire conversation is duplicated to
-            accommodate each message in the batch.
-        - If the message batch size does not match self.batch_size and none of
-            the above adjustments apply, a ValueError is raised.
-
-        Afterwards, each message in the batch is appended to its corresponding
-        conversation in the batched conversations.
-
-        Args:
-            message_batch: A sequence of Message objects
-                representing the batch of messages to be added. Must align with
-                or be adjusted to match the current batch size.
-
-        Raises:
-            ValueError: If the message batch size does not match the current
-                batch size and cannot be automatically adjusted.
-
-        """
-        message_batch_size = len(message_batch)
-
-        if message_batch_size == 1 and self.batch_size > 1:
-            logger.info(
-                "Message batch size is 1, current batch size is "
-                f"{self.batch_size}: duplicating the message to match the "
-                "current batch size"
-            )
-            message_batch = self._duplicate_message_to_current_batch_size(message_batch)
-            message_batch_size = self.batch_size
-        elif message_batch_size > 1 and self.batch_size == 1:
-            logger.info(
-                f"Message batch size is {len(message_batch)}, current batch "
-                "size is 1: duplicating the conversation to match the message "
-                "batch size"
-            )
-            self._duplicate_conversation_to_target_batch_size(message_batch_size)
-        elif message_batch_size != self.batch_size:
-            raise ValueError(
-                f"Message batch size {message_batch_size} does not match "
-                f"current batch size {self.batch_size}"
-            )
-
-        for batch_id in range(message_batch_size):
-            self._conversations[batch_id].append(message_batch[batch_id])
-
-    def add_message_batches(self, message_batches: Sequence[Sequence[Message]]) -> None:
-        for message_batch in message_batches:
-            self.add_message_batch(message_batch)
-
-    def add_message(self, message: Message) -> None:
-        for conversation in self._conversations:
-            conversation.append(message)
-
-    def add_message_list(self, message_list: Sequence[Message]) -> None:
-        for message in message_list:
-            self.add_message(message)
-
-    def __len__(self) -> int:
-        return len(self._conversations[0])
-
-    def __repr__(self) -> str:
-        return f"{self.__class__.__name__}(len={len(self)}; bs={self.batch_size})"
-
-    def __getitem__(self, idx: int) -> tuple[Message, ...]:
-        return tuple(conversation[idx] for conversation in self._conversations)
-
-    def __iter__(self) -> Iterator[tuple[Message, ...]]:
-        for idx in range(len(self)):
-            yield tuple(conversation[idx] for conversation in self._conversations)
-
-    def _duplicate_message_to_current_batch_size(
-        self, message_batch: Sequence[Message]
-    ) -> Sequence[Message]:
-        assert len(message_batch) == 1, (
-            "Message batch size must be 1 to duplicate to current batch size"
-        )
-
-        return [deepcopy(message_batch[0]) for _ in range(self.batch_size)]
-
-    def _duplicate_conversation_to_target_batch_size(
-        self, target_batch_size: int
-    ) -> None:
-        assert self.batch_size == 1, "Batch size must be 1 to duplicate conversation"
-        self._conversations = [
-            deepcopy(self._conversations[0]) for _ in range(target_batch_size)
-        ]
-
-    @property
-    def conversations(self) -> list[Messages]:
-        return self._conversations
-
-    @property
-    def batch_size(self) -> int:
-        return len(self._conversations)
-
-    def reset(
-        self, sys_prompt: LLMPrompt | None = None, *, batch_size: int = 1
-    ) -> None:
-        if sys_prompt is not None:
-            self._sys_prompt = sys_prompt
-
-        conv: Messages = []
-        if self._sys_prompt is not None:
-            conv.append(SystemMessage(content=self._sys_prompt))
-
-        self._conversations = [deepcopy(conv) for _ in range(batch_size)]
-
-    def erase(self) -> None:
-        self._conversations = [[]]

grasp_agents/workflow/parallel_processor.py

@@ -1,95 +0,0 @@
-import asyncio
-from collections.abc import Sequence
-from copy import deepcopy
-from typing import Any, Generic, cast
-
-from ..comm_processor import CommProcessor
-from ..packet import Packet
-from ..packet_pool import PacketPool
-from ..processor import Processor
-from ..run_context import CtxT
-from ..typing.io import InT_contra, OutT_co, ProcName
-
-
-class ParallelCommProcessor(
-    CommProcessor[InT_contra, OutT_co, Any, CtxT],
-    Generic[InT_contra, OutT_co, CtxT],
-):
-    def __init__(
-        self,
-        name: ProcName,
-        processor_type: type[Processor[InT_contra, OutT_co, Any, CtxT]],
-        packet_pool: PacketPool[CtxT] | None = None,
-        recipients: Sequence[ProcName] | None = None,
-        **subproc_init_kwargs: Any,
-    ) -> None:
-        self._processor_type = processor_type
-        self._subproc_init_kwargs = deepcopy(subproc_init_kwargs)
-
-        # NOTE: If the processor is an LLMAgent, the parallel subprocessors will share
-        # the same LLM and tools instances. Make sure their state is managed correctly.
-
-        super().__init__(name=name, packet_pool=packet_pool, recipients=recipients)
-
-    @property
-    def processor_type(self) -> type[Processor[InT_contra, OutT_co, Any, CtxT]]:
-        return self._processor_type
-
-    async def _run_subprocessor(
-        self, in_args: InT_contra, name_suffix: str, **subproc_run_kwargs: Any
-    ) -> Packet[OutT_co]:
-        subproc_name = f"{self.name}_{name_suffix}"
-        subproc = self._processor_type(name=subproc_name, **self._subproc_init_kwargs)
-
-        return await subproc.run(in_args=in_args, **subproc_run_kwargs)
-
-    def _validate_par_inputs(
-        self,
-        chat_inputs: Any | None,
-        in_packet: Packet[InT_contra] | None,
-        in_args: InT_contra | Sequence[InT_contra] | None,
-    ) -> Sequence[InT_contra]:
-        if chat_inputs is not None:
-            raise ValueError(
-                "chat_inputs are not supported in ParallelCommProcessor. "
-                "Use in_packet or in_args."
-            )
-        if in_packet is not None:
-            if not in_packet.payloads:
-                raise ValueError(
-                    "ParallelCommProcessor requires at least one input payload in "
-                    "in_packet."
-                )
-            return in_packet.payloads
-        if in_args is not None:
-            if not isinstance(in_args, Sequence) or not in_args:
-                raise ValueError("in_args must be a non-empty sequence of input data.")
-            return cast("Sequence[InT_contra]", in_args)
-        raise ValueError(
-            "ParallelCommProcessor requires either in_packet or in_args to be provided."
-        )
-
-    async def run(
-        self,
-        chat_inputs: Any | None = None,
-        *,
-        in_packet: Packet[InT_contra] | None = None,
-        in_args: InT_contra | Sequence[InT_contra] | None = None,
-        **subproc_run_kwargs: Any,
-    ) -> Packet[OutT_co]:
-        par_inputs = self._validate_par_inputs(
-            chat_inputs=chat_inputs, in_packet=in_packet, in_args=in_args
-        )
-        tasks = [
-            self._run_subprocessor(
-                in_args=inp, name_suffix=str(n), **subproc_run_kwargs
-            )
-            for n, inp in enumerate(par_inputs)
-        ]
-        out_packets = await asyncio.gather(*tasks)
-
-        return Packet(
-            payloads=[out_packet.payloads[0] for out_packet in out_packets],
-            sender=self.name,
-            recipients=(self.recipients or []),
-        )

grasp_agents-0.3.11.dist-info/RECORD

@@ -1,51 +0,0 @@
-grasp_agents/__init__.py,sha256=CIsyUasb9HBC3M4olg6ATAwKXtVNmmtpyGJrt7hpZW4,947
-grasp_agents/cloud_llm.py,sha256=IQa2yX9q8zWtB377PHrGepv8F-gjxgyal6olG1uF11Y,14054
-grasp_agents/comm_processor.py,sha256=w1Ix33K_1NALh1eLOwCpQbW5MDZDuP4eXOwEwPCBDJw,6790
-grasp_agents/costs_dict.yaml,sha256=2MFNWtkv5W5WSCcv1Cj13B1iQLVv5Ot9pS_KW2Gu2DA,2510
-grasp_agents/generics_utils.py,sha256=5Pw3I9dlnKC2VGqYKC4ZZUO3Z_vTNT-NPFovNfPkl6I,6542
-grasp_agents/grasp_logging.py,sha256=H1GYhXdQvVkmauFDZ-KDwvVmPQHZUUm9sRqX_ObK2xI,1111
-grasp_agents/http_client.py,sha256=KZva2MjJjuI5ohUeU8RdTAImUnQYaqBrV2jDH8smbJw,738
-grasp_agents/llm.py,sha256=vVjELab9l0mK9bjO-IJebmmXUyUiws2fNjbXVQQNRvs,5392
-grasp_agents/llm_agent.py,sha256=EM_1h1O8Gvs5RbQiEV3KT52ADpPDSpvToBLRn0MRUu0,14471
-grasp_agents/llm_agent_memory.py,sha256=kK-eB-8jMeVeWQDIye9xixdmqEi1a3vqFtOQfX64JF0,1770
-grasp_agents/llm_policy_executor.py,sha256=mm0WGmRPRMTnvXj9bSKkvPflYhgj7GTmP7kxPXRIloA,17633
-grasp_agents/memory.py,sha256=gPkVIIF6dI_xXzarIAw9kSEnSJcfW_teUsWA2JAih94,671
-grasp_agents/message_history.py,sha256=-ZNy3C1z0yQeahjqR0oIoWDMySJ7vPS19jdutibW7OE,5408
-grasp_agents/packet.py,sha256=PZ1EpclniAoLk7z4ieZbWzgYH3JSRgnlTe_WfbJYG_4,707
-grasp_agents/packet_pool.py,sha256=9umHbi5FwuUYYhhodSR-Z-fRR6OYiZyYEzq5d4nZFK4,3036
-grasp_agents/printer.py,sha256=M-gkLrZMg0ll9T1MRmKOQbp65niFB5ZiZbm-tUT_EYw,5464
-grasp_agents/processor.py,sha256=Atd_iTQNd3Nudb4mHqt3a5AQ31QUgpbrt1Fhn1sgpSk,6708
-grasp_agents/prompt_builder.py,sha256=URMT4X8sNhdMPnkJK0YAF0i2xqBpK2-WlBlkmpVGe_Y,8455
-grasp_agents/run_context.py,sha256=9CidWWCKJ8umlhkRGtg_P3JQsRpH0K3_vhUjgVon4Wk,1597
-grasp_agents/usage_tracker.py,sha256=WzBCqfscVH0YaWcz2wL5wyuHRZt4wwEaxjDcqggVjc8,3760
-grasp_agents/utils.py,sha256=WNA5b0IkAsth0XKHEjgTY3PIHdw5L3vEsYPkyDGw8Mw,4741
-grasp_agents/openai/__init__.py,sha256=wpTeew6EjhM6esHCKrEKUpwq0kygMN2QQDxYtmbRG8Y,4201
-grasp_agents/openai/completion_chunk_converters.py,sha256=i-1SvIWhRKtL0K8p5pb3jjACSnyHuJHTCoZovEERpxs,2628
-grasp_agents/openai/completion_converters.py,sha256=vzPEkUOX4l2hobKxZjEk_dyWfzeYesO0DlvWvNVb-Sg,2656
-grasp_agents/openai/content_converters.py,sha256=r1D5uci5x7sbDyl0XN27y-l_jVigCauJruvSdZSnZcc,2510
-grasp_agents/openai/converters.py,sha256=ncscVyPnPMMbyxAfFX3U73lnr_BZU-I89HA5Ld8BuxI,4691
-grasp_agents/openai/message_converters.py,sha256=_fG4vI42rBzoajuC5iYgnUBalg8cQ1ckSt8xFBOuWVY,4111
-grasp_agents/openai/openai_llm.py,sha256=Hm2S2yyrR3cRvH20YHRegoTEYaetbx_USxI8CXIhfro,8091
-grasp_agents/openai/tool_converters.py,sha256=d_7edJbnUhfSs2-F4J20D71UjVJWIslsXwMyac7-v2Q,1246
-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=KrvdFZqOcw5GePdMhDoFuwCkMJ86nqRdz9mT9s2lFv4,2445
-grasp_agents/typing/completion_chunk.py,sha256=b1d2a_8h21Uoif_FCS05XDMKap2nZWYV9G4goLKkoL0,5866
-grasp_agents/typing/content.py,sha256=VdnHdW8PHDCtX_ffvcwQz-7ypPUQNSGqHa3txFE_72Y,3676
-grasp_agents/typing/converters.py,sha256=kHlocHQS8QnduZOzNPbj3aRD8JpvJd53oudYqWdOxKE,2978
-grasp_agents/typing/events.py,sha256=QbrvXnDmXFr9_cSsdqL9f35eQLOfZ2O0h3a6yCRtKwY,2625
-grasp_agents/typing/io.py,sha256=_9G_pfEc41mHCLZ5lJ4N4qn7I-QDSOrdK8EFnQrSP18,315
-grasp_agents/typing/message.py,sha256=jTdN6-wVftfzplRjAz9zEWCpx_wPOBgcaMl0fcnsioU,2415
-grasp_agents/typing/tool.py,sha256=AfC6dsFiNAhM80yI9eh1Qm5U5WzJXsTHogXK-Zn4_cE,2495
-grasp_agents/workflow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-grasp_agents/workflow/looped_workflow.py,sha256=QqXclXYxsW6C8Rxkf3dRaMHi-DfCvCbjvFZr4nBl1u0,4299
-grasp_agents/workflow/parallel_processor.py,sha256=Xyzs2UR_mRe2GFgzzadHOhqgMu3rFjd3GUjvmZimt_k,3505
-grasp_agents/workflow/sequential_workflow.py,sha256=Pl7jl9ZVDu-rC5UMfympEaQN8iG3kZurVF5eIPG62XA,2130
-grasp_agents/workflow/workflow_processor.py,sha256=2-iaDIlgNXgj-ClGbiE3fYfSv-N_qRC49Gf_dF6M_40,2640
-grasp_agents-0.3.11.dist-info/METADATA,sha256=c5IZLKElzLGTU8UdAgpIjKsvqijaJnYJVAgiKOaA4JQ,6807
-grasp_agents-0.3.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-grasp_agents-0.3.11.dist-info/licenses/LICENSE.md,sha256=-nNNdWqGB8gJ2O-peFQ2Irshv5tW5pHKyTcYkwvH7CE,1201
-grasp_agents-0.3.11.dist-info/RECORD,,