lionagi 0.1.0__tar.gz → 0.1.2__tar.gz

{lionagi-0.1.0/lionagi.egg-info → lionagi-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lionagi
-Version: 0.1.0
+Version: 0.1.2
 Summary: Towards automated general intelligence.
 Author: HaiyangLi
 Author-email: Haiyang Li <ocean@lionagi.ai>

{lionagi-0.1.0 → lionagi-0.1.2}/lionagi/core/agent/base_agent.py

@@ -19,7 +19,7 @@ class BaseAgent(Node):
 
     def __init__(
         self,
-        structure: StructureExecutor,
+        structure: BaseExecutor,
         executable: BaseExecutor,
         output_parser=None,
         **kwargs,
@@ -33,7 +33,7 @@ class BaseAgent(Node):
             output_parser: A function for parsing the agent's output (optional).
         """
         super().__init__()
-        self.structure: StructureExecutor = structure
+        self.structure: BaseExecutor = structure
         self.executable: BaseExecutor = executable
         for v, k in kwargs.items():
             executable.__setattr__(v, k)
@@ -87,4 +87,3 @@ class BaseAgent(Node):
 
         if self.output_parser:
             return self.output_parser(self)
-        

{lionagi-0.1.0 → lionagi-0.1.2}/lionagi/core/branch/base.py

@@ -650,4 +650,4 @@ class BaseBranch(BaseNode, ABC):
         messages = self.messages["sender"] if use_sender else self.messages["role"]
         result = messages.value_counts().to_dict()
         result["total"] = len(self.messages)
-        return result
+        return result

{lionagi-0.1.0 → lionagi-0.1.2}/lionagi/core/branch/branch.py

@@ -470,4 +470,5 @@ class Branch(BaseBranch, BranchFlowMixin):
             }:
                 return True
         except Exception:
-            return False
+            return False
+        return False

{lionagi-0.1.0 → lionagi-0.1.2}/lionagi/core/branch/flow_mixin.py

@@ -93,4 +93,4 @@ class BranchFlowMixin(ABC):
             output_prompt=output_prompt,
             out=out,
             **kwargs,
-        )
+        )

{lionagi-0.1.0 → lionagi-0.1.2}/lionagi/core/branch/util.py

@@ -320,4 +320,4 @@ class MessageUtil:
             else:
                 with contextlib.suppress(Exception):
                     answers.append(nested.nget(content, ["system_info"]))
-        return "\n".join(answers)
+        return "\n".join(answers)

{lionagi-0.1.0 → lionagi-0.1.2}/lionagi/core/execute/base_executor.py

@@ -18,7 +18,7 @@ class BaseExecutor(BaseComponent, ABC):
     execute_stop: bool = Field(
         False, description="A flag indicating whether to stop execution."
     )
-    context: dict | str | None = Field(
+    context: dict | str | list | None = Field(
         None, description="The context buffer for the next instruction."
     )
     execution_responses: list = Field(
@@ -28,9 +28,6 @@ class BaseExecutor(BaseComponent, ABC):
     verbose: bool = Field(
         True, description="A flag indicating whether to provide verbose output."
     )
-    execute_stop: bool = Field(
-        False, description="A flag indicating whether to stop execution."
-    )
 
     def send(self, recipient_id: str, category: str, package: Any) -> None:
         """

{lionagi-0.1.0 → lionagi-0.1.2}/lionagi/core/execute/branch_executor.py

@@ -11,6 +11,10 @@ from lionagi.core.execute.base_executor import BaseExecutor
 class BranchExecutor(Branch, BaseExecutor):
 
     async def forward(self) -> None:
+        """
+        Forwards the execution by processing all pending incoming mails in each branch. Depending on the category of the mail,
+        it processes starts, nodes, node lists, conditions, or ends, accordingly executing different functions.
+        """
         for key in list(self.pending_ins.keys()):
             while self.pending_ins[key]:
                 mail = self.pending_ins[key].popleft()
@@ -26,11 +30,27 @@ class BranchExecutor(Branch, BaseExecutor):
                     self._process_end(mail)
 
     async def execute(self, refresh_time=1) -> None:
+        """
+        Executes the forward process repeatedly at specified time intervals until execution is instructed to stop.
+
+        Args:
+            refresh_time (int): The interval, in seconds, at which the forward method is called repeatedly.
+        """
         while not self.execute_stop:
             await self.forward()
             await AsyncUtil.sleep(refresh_time)
 
     async def _process_node(self, mail: BaseMail):
+        """
+        Processes a single node based on the node type specified in the mail's package. It handles different types of nodes such as System,
+        Instruction, ActionNode, and generic nodes through separate processes.
+
+        Args:
+            mail (BaseMail): The mail containing the node to be processed along with associated details.
+
+        Raises:
+            ValueError: If an invalid mail is encountered or the process encounters errors.
+        """
         if isinstance(mail.package["package"], System):
             self._system_process(mail.package["package"], verbose=self.verbose)
             self.send(
@@ -74,11 +94,26 @@ class BranchExecutor(Branch, BaseExecutor):
                 raise ValueError(f"Invalid mail to process. Mail:{mail}")
 
     def _process_node_list(self, mail: BaseMail):
+        """
+        Processes a list of nodes provided in the mail, but currently only sends an end signal as multiple path selection is not supported.
+
+        Args:
+            mail (BaseMail): The mail containing a list of nodes to be processed.
+
+        Raises:
+            ValueError: When trying to process multiple paths which is currently unsupported.
+        """
         self.send(mail.sender_id, "end", {"request_source": self.id_, "package": "end"})
         self.execute_stop = True
         raise ValueError("Multiple path selection is currently not supported")
 
     def _process_condition(self, mail: BaseMail):
+        """
+        Processes a condition associated with an edge based on the mail's package, setting up the result of the condition check.
+
+        Args:
+            mail (BaseMail): The mail containing the condition to be processed.
+        """
         relationship: Edge = mail.package["package"]
         check_result = relationship.condition(self)
         back_mail = {
@@ -93,6 +128,14 @@ class BranchExecutor(Branch, BaseExecutor):
         )
 
     def _system_process(self, system: System, verbose=True, context_verbose=False):
+        """
+        Processes a system node, possibly displaying its content and context if verbose is enabled.
+
+        Args:
+            system (System): The system node to process.
+            verbose (bool): Flag to enable verbose output.
+            context_verbose (bool): Flag to enable verbose output specifically for context.
+        """
         from lionagi.libs import SysUtil
 
         SysUtil.check_import("IPython")
@@ -111,6 +154,14 @@ class BranchExecutor(Branch, BaseExecutor):
     async def _instruction_process(
         self, instruction: Instruction, verbose=True, **kwargs
     ):
+        """
+        Processes an instruction node, possibly displaying its content if verbose is enabled, and handling any additional keyword arguments.
+
+        Args:
+            instruction (Instruction): The instruction node to process.
+            verbose (bool): Flag to enable verbose output.
+            **kwargs: Additional keyword arguments that might affect how instructions are processed.
+        """
         from lionagi.libs import SysUtil
 
         SysUtil.check_import("IPython")
@@ -146,6 +197,13 @@ class BranchExecutor(Branch, BaseExecutor):
         self.execution_responses.append(result)
 
     async def _action_process(self, action: ActionNode, verbose=True):
+        """
+        Processes an action node, executing the defined action along with any tools specified within the node.
+
+        Args:
+            action (ActionNode): The action node to process.
+            verbose (bool): Flag to enable verbose output of the action results.
+        """
         from lionagi.libs import SysUtil
 
         SysUtil.check_import("IPython")
@@ -196,7 +254,7 @@ class BranchExecutor(Branch, BaseExecutor):
             agent: The agent to process.
             verbose (bool): A flag indicating whether to provide verbose output (default: True).
         """
-        context = self.responses
+        context = list(self.messages["content"])
         if verbose:
             print("*****************************************************")
         result = await agent.execute(context)
@@ -204,8 +262,13 @@ class BranchExecutor(Branch, BaseExecutor):
         if verbose:
             print("*****************************************************")
 
-        self.context = result
-        self.responses.append(result)
+        from pandas import DataFrame
+
+        if isinstance(result, DataFrame):
+            self.context = list(result["content"])
+        else:
+            self.context = result
+        self.execution_responses.append(result)
 
     def _process_start(self, mail):
         """

{lionagi-0.1.0 → lionagi-0.1.2}/lionagi/core/execute/instruction_map_executor.py

@@ -10,6 +10,18 @@ from lionagi.core.execute.branch_executor import BranchExecutor
 
 
 class InstructionMapExecutor(BaseExecutor):
+    """
+    Manages the execution of a mapped set of instructions across multiple branches within an executable structure.
+
+    Attributes:
+        branches (dict[str, BranchExecutor]): A dictionary of branch executors managing individual instruction flows.
+        structure_id (str): The identifier for the structure within which these branches operate.
+        mail_transfer (MailTransfer): Handles the transfer of mail between branches and other components.
+        branch_kwargs (dict): Keyword arguments used for initializing branches.
+        num_end_branches (int): Tracks the number of branches that have completed execution.
+        mail_manager (MailManager): Manages the distribution and collection of mails across branches.
+    """
+
     branches: dict[str, BranchExecutor] = Field(
         default_factory=dict, description="The branches of the instruction mapping."
     )
@@ -27,10 +39,20 @@ class InstructionMapExecutor(BaseExecutor):
     )
 
     def __init__(self, **kwargs):
+        """
+        Initializes an InstructionMapExecutor with the given parameters.
+
+        Args:
+            **kwargs: Arbitrary keyword arguments passed to the base executor and used for initializing branch executors.
+        """
         super().__init__(**kwargs)
         self.mail_manager = MailManager([self.mail_transfer])
 
     def transfer_ins(self):
+        """
+        Processes incoming mails, directing them appropriately based on their categories, and handles the initial setup
+        of branches or the routing of node and condition mails.
+        """
         for key in list(self.pending_ins.keys()):
             while self.pending_ins[key]:
                 mail: BaseMail = self.pending_ins[key].popleft()
@@ -48,6 +70,10 @@ class InstructionMapExecutor(BaseExecutor):
                     self.mail_transfer.pending_outs.append(mail)
 
     def transfer_outs(self):
+        """
+        Processes outgoing mails from the central mail transfer, handling end-of-execution notifications and routing
+        other mails to appropriate recipients.
+        """
         for key in list(self.mail_transfer.pending_ins.keys()):
             while self.mail_transfer.pending_ins[key]:
                 mail: BaseMail = self.mail_transfer.pending_ins[key].popleft()
@@ -66,6 +92,12 @@ class InstructionMapExecutor(BaseExecutor):
                     self.pending_outs.append(mail)
 
     def _process_start(self, start_mail: BaseMail):
+        """
+        Processes a start mail to initialize a new branch executor and configures it based on the mail's package content.
+
+        Args:
+            start_mail (BaseMail): The mail initiating the start of a new branch execution.
+        """
         branch = BranchExecutor(verbose=self.verbose, **self.branch_kwargs)
         branch.context = start_mail.package["context"]
         self.branches[branch.id_] = branch
@@ -80,6 +112,13 @@ class InstructionMapExecutor(BaseExecutor):
         self.pending_outs.append(mail)
 
     def _process_node_list(self, nl_mail: BaseMail):
+        """
+        Processes a node list mail, setting up new branches or propagating the execution context based on the node list
+        provided in the mail.
+
+        Args:
+            nl_mail (BaseMail): The mail containing a list of nodes to be processed in subsequent branches.
+        """
         source_branch_id = nl_mail.package["request_source"]
         node_list = nl_mail.package["package"]
         shared_context = self.branches[source_branch_id].context
@@ -115,6 +154,9 @@ class InstructionMapExecutor(BaseExecutor):
             self.mail_transfer.pending_outs.append(node_mail)
 
     async def forward(self):
+        """
+        Forwards the execution by processing all incoming and outgoing mails and advancing the state of all active branches.
+        """
         self.transfer_ins()
         self.transfer_outs()
         self.mail_manager.collect_all()
@@ -126,6 +168,12 @@ class InstructionMapExecutor(BaseExecutor):
         return
 
     async def execute(self, refresh_time=1):
+        """
+        Continuously executes the forward process at specified intervals until instructed to stop.
+
+        Args:
+            refresh_time (int): The time in seconds between execution cycles.
+        """
         while not self.execute_stop:
             await self.forward()
             await asyncio.sleep(refresh_time)