lionagi 0.0.201__py3-none-any.whl → 0.0.204__py3-none-any.whl

{lionagi-0.0.201.dist-info → lionagi-0.0.204.dist-info}/RECORD

@@ -1,10 +1,10 @@
 lionagi/__init__.py,sha256=pRm9tnfYQnj_ZCan3xr3HQO7tbUeqwRhSBtwOvXJ-o8,886
-lionagi/version.py,sha256=ktNwZ1SXW3TrDDzpm1Ws6572EdHKpyaNGPUy7n6dcf4,24
+lionagi/version.py,sha256=DFD_QxPb5C0pITp5fF5NUF2YLCrCu9FEfCgCY7YVy4I,24
 lionagi/_services/__init__.py,sha256=zU5sxmSI9-Jtp_WsI-Zsb6hmT8y5zF9YtJ7XikAjnbs,60
-lionagi/_services/anthropic.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/_services/anthropic.py,sha256=pLebbnr2H1A41bzXrJrU7yQbZY35swKrSi4mktXoIyk,3195
 lionagi/_services/anyscale.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/_services/azure.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
-lionagi/_services/base_service.py,sha256=tNP3C48Rp0rsydNGPvDOIY5X9EXgaizFQ5nd-30TVyo,17321
+lionagi/_services/base_service.py,sha256=bbPQ9xTaY5jxrHe6vW_PeyRkoxVaxWmpWlCqQxSfRI8,17314
 lionagi/_services/bedrock.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/_services/everlyai.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/_services/gemini.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -20,23 +20,38 @@ lionagi/_services/openrouter.py,sha256=MuuwoT2ro9FmY7O1jzCenRrL2YfiYUMM8-0kKoGZH
 lionagi/_services/perplexity.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/_services/predibase.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/_services/rungpt.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lionagi/_services/services.py,sha256=RyR7NUVnVryzDKyOvQZNXK444x_iQcc88Bba6NjTy5Y,2609
+lionagi/_services/services.py,sha256=kY2TpT98pDS_qCrCO4H1YBlDYNDqekx5S9rWQdWm5Ck,4007
+lionagi/_services/transformers.py,sha256=tHlo9QVV5ycB2xUEKNRf-b665o_21fPC6c4ycH3Hjk4,1444
 lionagi/_services/vllm.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/_services/xinference.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/agents/planner.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/agents/prompter.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/agents/scorer.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/agents/summarizer.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/agents/validator.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
 lionagi/bridge/__init__.py,sha256=YDKi-pniFEknEXTSEvX6yEe4Y69f0kLvbjvEQ0TdrTU,575
 lionagi/bridge/langchain.py,sha256=NuuUOJ5oV0l6ae4EeXxaeB82nJ6FDJHVC4AFBoIqiDs,5726
 lionagi/bridge/llama_index.py,sha256=oggpASmfVA6IhfgBOJS8CWJrQb3zrGaDApUQ7MWT8OM,6652
 lionagi/configs/__init__.py,sha256=QOd4Rs7vjIpNWvIocxWQeU-q-MPRC-AOxh-gM-eBJ2o,142
-lionagi/configs/oai_configs.py,sha256=bBQLDynTXKBd1EOQ0NTrKdpMHr9CSTrxg_aL5J0wms8,2645
-lionagi/configs/openrouter_configs.py,sha256=BA7hqZTc6ctdb2bMu9UXN4wJpCoDI-KyYi8kV_SVIDs,1268
-lionagi/core/__init__.py,sha256=fDUke-5rbliBX1iRCeOZv4C1H_XjAEogHBGtLzbk5mw,301
-lionagi/core/branch.py,sha256=bxvt4ouOvefpyU-UyzASV-3iUV2gWzv2sbLE6vS1fig,6955
-lionagi/core/conversation.py,sha256=6Mo_T8YlM4ELs0gEmQzRs4HnwL3LwuAHUlVEOBC35Zc,11946
-lionagi/core/flow.py,sha256=DMcCkViv5wWFhQ8RjbcA9pAQoCHo1etV6O9ORdDrXHc,191
-lionagi/core/flow_util.py,sha256=OoQ2-ktkpQs9f1m1VI1pucUeq75Mx1aKqz8KdINMt8M,2083
-lionagi/core/instruction_set.py,sha256=-q_tOiLYBFKSNbryoZLOSd3i43WGD-m2p-nObnadu1g,5687
-lionagi/core/messages.py,sha256=tKIXLyzhqRK2ReXwTrgWp2EjkN161RNFHlr8CGPUaWk,8147
-lionagi/core/sessions.py,sha256=3CMUMQhlpaolkEhQZZHfc0kr4j3EtANGIZl4ShiSCTY,18440
+lionagi/configs/oai_configs.py,sha256=Q2ESc5QiMprnRc_w7SeMlaTYUWl_Y4SEzZSE4iOkz4Q,2646
+lionagi/configs/openrouter_configs.py,sha256=IBQHqb8mo4Jb3kYAm_7NOHSKRPwSdGbPpDJoiwHxLYw,1269
+lionagi/core/__init__.py,sha256=6uDjq1WCWBahNiCpzGaUCJe7GDg6lwUD_cGNT86GhwM,200
+lionagi/core/core_util.py,sha256=80fmVywc0UC9FklXpQr5_dHdtdHdosv5Zxwy9tC-Ufg,2339
+lionagi/core/branch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/core/branch/branch.py,sha256=w3GZAws8_qgY17XOn6svmO4ECQqhNQ_rWMzXiwBQVO8,23403
+lionagi/core/branch/branch_manager.py,sha256=kLVYUYVUmFQ2CPV34vfVRrkR6fhlizoAr3Dw2ilMX6M,4904
+lionagi/core/branch/cluster.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
+lionagi/core/branch/conversation.py,sha256=q05EwDouTAH5eMDCp2wwRHK0ipbm0MYFngk8R6vOTB0,18664
+lionagi/core/flow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/core/flow/flow.py,sha256=Fd6xVZKgjjpr-rcVAjvuMGP04OxNK4LVkbB9VnEP21k,813
+lionagi/core/flow/flow_util.py,sha256=OoQ2-ktkpQs9f1m1VI1pucUeq75Mx1aKqz8KdINMt8M,2083
+lionagi/core/instruction_set/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/core/instruction_set/instruction_set.py,sha256=bxrxPxLJdaenvZ2CMaiucNB4fZ_5AWVz49MYs3mG2G8,13682
+lionagi/core/messages/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/core/messages/messages.py,sha256=bokGS8hu4wQxa-4Gsf2Eg7Q99ANccjjixZpE-DC0P7g,6533
+lionagi/core/sessions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/core/sessions/session.py,sha256=-dSlIIzHmD2dGd3zu4m_U-UE0BKraKYyt1H_DAzpngk,17694
 lionagi/datastores/__init__.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
 lionagi/datastores/chroma.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
 lionagi/datastores/deeplake.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
@@ -49,12 +64,15 @@ lionagi/loaders/__init__.py,sha256=vOOwHkdI0yIA_jV_pNyOrdkS5Ghs0k4od75S1U60jJE,4
 lionagi/loaders/chunker.py,sha256=UY6GrC8qC0MLRaHiSfgG5HMnrtWSTuIvaRPhYfdm9ak,6438
 lionagi/loaders/load_util.py,sha256=4fzhMk3H6OvcQcViUwlLPSTMpcY4alfdel16lJgXz8Y,8358
 lionagi/loaders/reader.py,sha256=xI1uxw9qGJ_rWev_s3vtW8Ep9YaK-15q7ts-Jy61PGg,4625
+lionagi/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/models/base_model.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/models/imodel.py,sha256=VultpAAZ5PBT6_Cps-LjgZqHEyuF-km6eVvT-sEPEgo,1491
 lionagi/schema/__init__.py,sha256=XRWXK9sztycoIMCTzwLEVMxxc1wgWKNUDRbWTpn5Ie0,208
 lionagi/schema/async_queue.py,sha256=e_wFyDvCeKTxW6MAIU6Q3jxw24uEQuahaZwNDzZMB4k,5674
 lionagi/schema/base_condition.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
 lionagi/schema/base_node.py,sha256=Ol4Y6gn81Dm8FJb3fdKVkJlq4R-OAGIf891exnVv45I,13211
 lionagi/schema/base_tool.py,sha256=8LR-MYOGkv9zCHd8tWry296OZNYlbTpYeO1xd-oQcOM,1254
-lionagi/schema/data_logger.py,sha256=By4YWr5HhW49EDmEia7I8iu37mxqRl0dpqkbBlbt6OM,8191
+lionagi/schema/data_logger.py,sha256=2rNPBYF29_2xgv2m226BzKdTfFIVU1YlzQ0PK8zLAvY,4313
 lionagi/schema/data_node.py,sha256=k80mv5DCqAHrGNlmaIHAkCA7JF4dkaRaRDh_oku4kUY,2272
 lionagi/schema/status_tracker.py,sha256=6otnTSMrH5jM0PUDiDeK3zG0VOSKfNBDzNN6Bts2tmA,1236
 lionagi/structures/__init__.py,sha256=wMPekT2vbWwUkJ5aW5o-lzJC9Fzhta6RHDiFPTNUm_0,120
@@ -64,28 +82,25 @@ lionagi/structures/structure.py,sha256=YyL3LxgeQWgkWDhACRxo8GoDn0IjE7idh7B0r5U3m
 lionagi/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/tests/test_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/tests/test_utils/test_api_util.py,sha256=7Zyc0J1glZrIWI1HrTRSRhzw8jaUW1L2vVLFAlUhI4g,9721
-lionagi/tests/test_utils/test_call_util.py,sha256=rTW-yDDzLjTHyE3jRxHUXT0d3Iicn9iA2f-IIF5-fZk,25096
+lionagi/tests/test_utils/test_call_util.py,sha256=7xmfFaWvniMQfaOyfwasA2enJQVuSlcAwc8gUyAR_7k,26277
 lionagi/tests/test_utils/test_encrypt_util.py,sha256=hlkbFjQs2jodr8fgtPli6f1MO3doQbTcsZfzGKVrG5k,12652
 lionagi/tests/test_utils/test_io_util.py,sha256=cFZCT6EikVeuXB13w-UbtO3YceCHBO5RlNXxGICqg_U,11002
 lionagi/tests/test_utils/test_nested_util.py,sha256=Z1boHufhjZryw51qW2lABOnnyJ1snAFp26KKzzzD8Bs,12612
 lionagi/tests/test_utils/test_sys_util.py,sha256=TDCkzll-JLa6NuBbN_-ay5Rw9KTa_HcSHHAq62RVwGI,13545
 lionagi/tools/__init__.py,sha256=ZEck-ReP5Co05nAA2gUXTpKoDN2QZqrL7DvU9Z09gqg,69
-lionagi/tools/planner.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
-lionagi/tools/prompter.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
-lionagi/tools/scorer.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
-lionagi/tools/summarizer.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
-lionagi/tools/tool_manager.py,sha256=znpDGeg7m71CCGFUkLCPmXIFj8jQG9t1n1Gcj4Aqgz0,9555
+lionagi/tools/tool_manager.py,sha256=4yCjMVOrKo4PoEMv0xF6wMirpj20qhLx4J7op0v5h2w,6543
 lionagi/tools/tool_util.py,sha256=5ln7lnqC_rjhKDDwef10xrBrjP1yLzsQvphllD5crec,9252
-lionagi/tools/validator.py,sha256=chG3GNX2BBDTWIuSVfZUJ_YF_ZVBSoel2d_AN0OChS0,6
-lionagi/utils/__init__.py,sha256=V3yj15tXS4ghXT0Du5l34RyIeIQzZCB1nBwWYydA32I,1081
+lionagi/utils/__init__.py,sha256=FJAqYyb19Mp6kiietyjtp0WLlW2857-HAHl7OxqCc5Y,954
 lionagi/utils/api_util.py,sha256=YV-DKPgR4lCKO_riroSKyq6mO-8vz-SilRo_gWtg4Zg,15282
-lionagi/utils/call_util.py,sha256=Cmjx7HqAkANLfjfBIbNoIwzFhrilMqk-k0ishytQCCg,31469
+lionagi/utils/call_util.py,sha256=G3K8dkEZ9AehrBdbKHK6Xtr1pZZWxrtqhQEu5xkItuo,32919
 lionagi/utils/encrypt_util.py,sha256=iZjZdXVvl0lw4Yw_YNzIWriM3F2qKtzai7PgSQ1TExc,9316
-lionagi/utils/io_util.py,sha256=aCZn1-cIV1ceD5dleuWTXAOU6oK_pioVf_nA0ria4k0,6358
-lionagi/utils/nested_util.py,sha256=x2L8umSwcAZpCHdsUbalmgUj6vuJul0r0WR68-hvCEM,34071
-lionagi/utils/sys_util.py,sha256=2IxpAv43jVa49gK5ZYJgmF2Ec63LY-CffT6XpfvNWwM,10322
-lionagi-0.0.201.dist-info/LICENSE,sha256=TBnSyG8fs_tMRtK805GzA1cIyExleKyzoN_kuVxT9IY,11358
-lionagi-0.0.201.dist-info/METADATA,sha256=17XyutuFOoL7zKU1nP4V7sg_Y9ySDnKsRhwG-pw4efQ,17882
-lionagi-0.0.201.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-lionagi-0.0.201.dist-info/top_level.txt,sha256=szvch_d2jE1Lu9ZIKsl26Ll6BGfYfbOgt5lm-UpFSo4,8
-lionagi-0.0.201.dist-info/RECORD,,
+lionagi/utils/io_util.py,sha256=xoVsq8sP5JGsosuC80Kad3GkGjm8Qm0OLYyTw-U5ru8,6455
+lionagi/utils/nested_util.py,sha256=67j-ySQtuMGxtjnC-Ty2mwQgqp2g1gZhXRy1MulUu1U,26656
+lionagi/utils/pd_util.py,sha256=ShLdNRJI-U2nN9TmZEGPdRXHzFMfrmw-sTpUbxNWr1w,2274
+lionagi/utils/sys_util.py,sha256=iZQu3HvYLl-12mmZ0kk4lX-3FnkRh-EAMAefPt_6P7k,10893
+lionagi/utils/url_util.py,sha256=fu1uRFwSR9D3dO1nfSYVNRD1b1BZVClcbpgF7tA_U4s,1864
+lionagi-0.0.204.dist-info/LICENSE,sha256=TBnSyG8fs_tMRtK805GzA1cIyExleKyzoN_kuVxT9IY,11358
+lionagi-0.0.204.dist-info/METADATA,sha256=kxTcrwAWtkK8cRY0CAxJaOZ8lBIVQgFcy-8HT92mTuc,17894
+lionagi-0.0.204.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+lionagi-0.0.204.dist-info/top_level.txt,sha256=szvch_d2jE1Lu9ZIKsl26Ll6BGfYfbOgt5lm-UpFSo4,8
+lionagi-0.0.204.dist-info/RECORD,,

lionagi/core/branch.py

@@ -1,193 +0,0 @@
-from copy import deepcopy
-from typing import Any, Dict, Union
-from datetime import datetime
-from pandas import DataFrame
-
-from ..tools.tool_manager import ToolManager
-from .messages import Message
-from .conversation import Conversation
-
-
-class Branch(Conversation):
-    """
-    Branch of a conversation with message tracking and tool management.
-
-    This class extends the Conversation class, managing a branch of conversation
-    with the ability to track messages, maintain instruction sets, and manage tools
-    using a ToolManager.
-
-    Args:
-        init_message: The initial message or messages of the conversation branch.
-        instruction_sets: A set of instructions to be used in the conversation.
-            Defaults to None, which creates an empty dict.
-        tool_manager: A ToolManager instance to manage tools. Defaults to None,
-            which creates a new ToolManager instance.
-
-    Raises:
-        ValueError: If the input `init_message` is not a DataFrame or Message instance.
-
-    Attributes:
-        instruction_sets: Instruction sets associated with the conversation branch.
-        tool_manager: Tool manager for the branch.
-        messages: DataFrame containing the conversation messages.
-        system_message: The system message of the conversation branch.
-
-    Examples:
-        # Creating a Branch with an initial message
-        initial_message = Message(role="user", content="Hi there!")
-        branch = Branch(init_message=initial_message)
-
-        # Creating a Branch with initial messages as DataFrame
-        messages_df = DataFrame({
-            "role": ["user", "system"],
-            "content": ["Hi there!", "Hello!"]
-        })
-        branch = Branch(init_message=messages_df)
-    """
-
-    def __init__(self, init_message: Union[Message, DataFrame],
-                 instruction_sets: dict = None,
-                 tool_manager: ToolManager = None) -> None:
-        super().__init__()
-        self.instruction_sets = instruction_sets if instruction_sets else {}
-        self.tool_manager = tool_manager if tool_manager else ToolManager()
-
-        if isinstance(init_message, DataFrame):
-            self.messages = init_message
-        elif isinstance(init_message, Message):
-            self.add_message(init_message)
-        else:
-            raise ValueError('Please input a valid init_message: DataFrame or Message')
-        self.system_message = self.messages.loc[0, 'content']
-
-    def change_system_message(self, system: Message):
-        """
-        Change the system message to the provided message.
-
-        Args:
-            system: A Message object representing the new system message.
-        """
-        message_dict = system.to_dict()
-        message_dict['timestamp'] = datetime.now()
-        self.messages.loc[0] = message_dict
-        self.system_message = self.messages.loc[0, 'content']
-
-    def add_instruction_set(self, name, instruction_set):
-        """
-        Add an instruction set to the conversation branch.
-
-        Args:
-            name: The name of the instruction set.
-            instruction_set: The instruction set to be added.
-        """
-        self.instruction_sets[name] = instruction_set
-
-    def remove_instruction_set(self, name):
-        """
-        Remove an instruction set from the conversation branch by name.
-
-        Args:
-            name: The name of the instruction set to be removed.
-
-        Returns:
-            The removed instruction set if it exists, else None.
-        """
-        return self.instruction_sets.pop(name)
-
-    def register_tools(self, tools):
-        """
-        Register a list of tools to the ToolManager.
-
-        Args:
-            tools: A single tool or a list of tools to be registered.
-        """
-        if not isinstance(tools, list):
-            tools = [tools]
-        self.tool_manager.register_tools(tools=tools)
-
-    def delete_tool(self, name):
-        """
-        Delete a tool from the ToolManager by name.
-
-        Args:
-            name: The name of the tool to be deleted.
-
-        Returns:
-            True if the tool was successfully deleted, False otherwise.
-        """
-        if name in self.tool_manager.registry:
-            self.tool_manager.registry.pop(name)
-            return True
-        return False
-
-    def clone(self):
-        """
-        Create a deep copy of the current branch.
-
-        Returns:
-            A new Branch instance that is a deep copy of the current branch.
-        """
-        cloned = Branch(self.messages.copy())
-        cloned.instruction_sets = deepcopy(self.instruction_sets)
-        cloned.tool_manager = ToolManager()
-        cloned.tool_manager.registry = deepcopy(self.tool_manager.registry)
-        return cloned
-
-    def merge(self, branch, update=True):
-        """
-        Merge another Branch into the current one.
-
-        Args:
-            branch: The Branch to merge into the current one.
-            update: Whether to update the current Branch. If False, only adds
-                non-existing items from the other Branch.
-        """
-        message_copy = branch.messages.copy()
-        branch_system = message_copy.loc[0]
-        message_copy.drop(0, inplace=True)
-        self.messages = self.messages.merge(message_copy, how='outer')
-        if update:
-            self.instruction_sets.update(branch.instruction_sets)
-            self.tool_manager.registry.update(branch.tool_manager.registry)
-            self.messages.loc[0] = branch_system
-        else:
-            for key, value in branch.instruction_sets.items():
-                if key not in self.instruction_sets:
-                    self.instruction_sets[key] = value
-
-            for key, value in branch.tool_manager.registry.items():
-                if key not in self.tool_manager.registry:
-                    self.tool_manager.registry[key] = value
-
-    def report(self) -> Dict[str, Any]:
-        """
-        Generate a report about the conversation branch.
-
-        Returns:
-            A dictionary containing information about total messages, a summary by role,
-            instruction sets, registered tools, and the messages themselves.
-        """
-
-        return {
-            "total_messages": len(self.messages),
-            "summary_by_role": self.message_counts(),
-            "instruction_sets": self.instruction_sets,
-            "registered_tools": self.tool_manager.registry,
-            "messages": [
-                msg.to_dict() for _, msg in self.messages.iterrows()
-            ],
-        }
-
-    def to_chatcompletion_message(self):
-        """
-        Convert the conversation branch to a list of messages for chat completion.
-
-        Returns:
-            A list of dictionaries with 'name' or 'role' and 'content' from the conversation messages.
-        """
-        message = []
-        for _, row in self.messages.iterrows():
-            out = {"role": row['role'], "content": row['content']}
-            message.append(out)
-        return message
-    

lionagi/core/conversation.py

@@ -1,341 +0,0 @@
-import json
-from datetime import datetime
-from typing import Optional, Dict, Any
-import pandas as pd
-
-from .messages import Message
-
-class Conversation:
-    """
-    Manages the flow and archival of a conversation using Pandas DataFrame.
-
-    Attributes:
-        messages (pd.DataFrame): A DataFrame to store messages.
-        dir (Optional[str]): The directory path for saving files, if provided.
-
-    Examples:
-        >>> conv = Conversation()
-        >>> msg = Message(node_id="1", role="user", name="Alice", content="Hi there!")
-        >>> conv.add_message(msg)
-        >>> conv.get_message_by_role("user")
-        >>> conv.last_message()
-        >>> conv.search_message("Hi")
-        >>> conv.message_counts()
-    """
-
-    def __init__(self, dir: Optional[str] = None) -> None:
-        self.messages = pd.DataFrame(
-            columns=["node_id", "role", "name", "timestamp", "content"]
-        )
-        self.dir = dir
-
-    def add_message(self, message: Message) -> None:
-        """
-        Adds a new message to the DataFrame.
-
-        Args:
-            message (Message): The message object to be added.
-
-        Examples:
-            >>> conv = Conversation()
-            >>> msg = Message(node_id="1", role="user", name="Alice", content="Hi there!")
-            >>> conv.add_message(msg)
-        """
-        message_dict = message.to_dict()
-        if isinstance(message_dict['content'], dict):
-            message_dict['content'] = json.dumps(message_dict['content'])
-        message_dict['timestamp'] = datetime.now()
-        self.messages.loc[len(self.messages)] = message_dict
-
-    def get_message_by_role(self, role: str) -> pd.DataFrame:
-        """
-        Retrieves messages sent by a specific role.
-
-        Args:
-            role (str): The role to filter messages by.
-
-        Returns:
-            pd.DataFrame: DataFrame of messages matching the role.
-
-        Examples:
-            >>> conv = Conversation()
-            >>> conv.get_message_by_role("user")
-        """
-        return self.messages[self.messages["role"] == role]
-
-    def last_message(self):
-        """
-        Retrieves the last message from the conversation.
-
-        Returns:
-            Message: The last message object, if the DataFrame is not empty.
-
-        Examples:
-            >>> conv = Conversation()
-            >>> conv.last_message()
-        """
-        if not self.messages.empty:
-            return Message(**self.messages.iloc[-1].to_dict())
-
-    def last_message_by_role(self, role: str) -> Optional[Message]:
-        """
-        Retrieves the last message of a specific role.
-
-        Args:
-            role (str): The role to get messages by.
-
-        Returns:
-            Optional[Message]: The last message of the given role, if exists.
-
-        Examples:
-            >>> conv = Conversation()
-            >>> conv.last_message_by_role("user")
-        """
-        filtered_messages = self.messages[
-            self.messages["role"] == role
-        ]
-        if not filtered_messages.empty:
-            return Message(**filtered_messages.iloc[-1].to_dict())
-        return None
-
-    def search_message(self, keyword: str, case_sensitive: bool = False) -> pd.DataFrame:
-        """
-        Searches for messages containing a specific keyword.
-
-        Args:
-            keyword (str): The keyword to search for.
-            case_sensitive (bool, optional): Whether the search should be case-sensitive. Defaults to False.
-
-        Returns:
-            pd.DataFrame: DataFrame of messages containing the keyword.
-
-        Examples:
-            >>> conv = Conversation()
-            >>> conv.search_message("Hi")
-            >>> conv.search_message("Hi", case_sensitive=True)
-        """
-        if not case_sensitive:
-            keyword = keyword.lower()
-            return self.messages[
-                self.messages["content"].str.lower().str.contains(keyword)
-            ]
-        return self.messages[self.messages["content"].str.contains(keyword)]
-
-    def delete_message(self, message_id: str) -> bool:
-        """
-        Deletes a message by its unique identifier.
-
-        Args:
-            message_id (str): The unique identifier of the message.
-
-        Returns:
-            bool: True if a message was deleted, False otherwise.
-
-        Examples:
-            >>> conv = Conversation()
-            >>> conv.delete_message("1")
-        """
-        initial_length = len(self.messages)
-        self.messages = self.messages[self.messages["node_id"] != message_id]
-        return len(self.messages) < initial_length
-
-    def update_message_content(self, message_id: str, new_content: str) -> bool:
-        """
-        Updates the content of a specific message by its unique identifier.
-
-        Args:
-            message_id (str): The unique identifier of the message.
-            new_content (str): The new content for the message.
-
-        Returns:
-            bool: True if the content was updated, False otherwise.
-
-        Examples:
-            >>> conv = Conversation()
-            >>> conv.update_message_content("1", "New content")
-        """
-        index = self.messages.index[self.messages["id_"] == message_id].tolist()
-        if index:
-            self.messages.at[index[0], "content"] = new_content
-            return True
-        return False
-
-    def get_message_by_id(self, message_id: str) -> Optional[Message]:
-        """
-        Retrieves a message by its unique identifier.
-
-        Args:
-            message_id (str): The unique identifier of the message.
-
-        Returns:
-            Optional[Message]: The message if found, None otherwise.
-
-        Examples:
-            >>> conv = Conversation()
-            >>> conv.get_message_by_id("1")
-        """
-        message_df = self.messages[self.messages["id_"] == message_id]
-        if not message_df.empty:
-            return Message.from_json(message_df.iloc[0].to_dict())
-        return None
-
-    def message_counts(self, use_name=False) -> Dict[str, int]:
-        """
-        Provides a summary of message counts in the conversation, optionally grouped by sender's name.
-
-        Args:
-            use_name (bool, optional): If True, groups by sender's name instead of role. Defaults to False.
-
-        Returns:
-            Dict[str, int]: A dictionary with roles or names as keys and their associated message counts as values.
-
-        Examples:
-            >>> conv = Conversation()
-            >>> conv.message_counts()  # Group by role
-            >>> conv.message_counts(use_name=True)  # Group by sender's name
-        """
-        messages = self.messages['name'] if use_name else self.messages['role']
-        result = messages.value_counts().to_dict()
-        result['total'] = len(self.messages)
-        return result
-
-    def report(self) -> Dict[str, Any]:
-        """
-        Generates a detailed report of the conversation.
-
-        Returns:
-            Dict[str, Any]: A dictionary with various details about the conversation.
-        """
-        return {
-            "total_messages": len(self.messages),
-            "summary_by_role": self.message_counts(),
-            "messages": [
-                msg.to_dict() for _, msg in self.messages.iterrows()
-            ],
-        }
-
-    def history(
-        self, start_date: Optional[datetime] = None, end_date: Optional[datetime] = None
-    ) -> pd.DataFrame:
-        """
-        Returns a history of the conversation, optionally filtered by date range.
-
-        Args:
-            start_date (Optional[datetime], optional): The start date of the date range filter. Defaults to None.
-            end_date (Optional[datetime], optional): The end date of the date range filter. Defaults to None.
-
-        Returns:
-            pd.DataFrame: A DataFrame containing the filtered conversation history.
-        """
-        if isinstance(start_date, str):
-            start_date = datetime.strptime(start_date, '%Y-%m-%d')
-        if isinstance(end_date, str):
-            end_date = datetime.strptime(end_date, '%Y-%m-%d')
-        if start_date and end_date:
-            return self.messages[
-                (self.messages["timestamp"].dt.date >= start_date.date())
-                & (self.messages["timestamp"].dt.date <= end_date.date())
-            ]
-        elif start_date:
-            return self.messages[(self.messages["timestamp"].dt.date >= start_date.date())]
-        elif end_date:
-            return self.messages[(self.messages["timestamp"].dt.date <= end_date.date())]
-        return self.messages
-
-    def replace_keyword(self, keyword: str, replacement: str, case_sensitive: bool = False) -> None:
-        """
-        Replaces a keyword in all messages with a replacement.
-
-        Args:
-            keyword (str): The keyword to replace.
-            replacement (str): The text to replace the keyword with.
-            case_sensitive (bool, optional): Whether the replacement should be case-sensitive. Defaults to False.
-        """
-        if not case_sensitive:
-            self.messages["content"] = self.messages["content"].str.replace(
-                keyword, replacement, case=False
-            )
-        else:
-            self.messages["content"] = self.messages["content"].str.replace(
-                keyword, replacement
-            )
-
-    def clone(self) -> 'Conversation':
-        """
-        Creates a deep copy of the conversation.
-
-        Returns:
-            Conversation: A new Conversation instance with the same messages.
-        """
-        cloned = Conversation(self.dir)
-        cloned.messages = self.messages.copy()
-        return cloned
-
-    def merge(self, other: 'Conversation') -> None:
-        """
-        Merges another conversation into this one.
-
-        Args:
-            other (Conversation): Another Conversation instance to merge with this one.
-        """
-        self.messages = self.messages.merge(other.messages, how='outer')
-
-    def rollback(self, steps: int) -> None:
-        """
-        Rollbacks the conversation to a previous state by the given number of steps.
-
-        Args:
-            steps (int): The number of steps to rollback.
-
-        Raises:
-            ValueError: If steps are negative or exceed the current number of messages.
-        """
-        if steps < 0 or steps > len(self.messages):
-            raise ValueError("Steps must be a non-negative integer less than or equal to the number of messages.")
-        self.messages = self.messages[:-steps].reset_index(drop=True)
-
-    def reset(self) -> None:
-        """Clears all messages in the DataFrame."""
-        self.messages = pd.DataFrame(columns=self.messages.columns)
-
-    def to_csv(self, filepath: str, **kwargs) -> None:
-        """
-        Saves the conversation to a CSV file.
-
-        Args:
-            filepath (str): The path to the file where the conversation should be saved.
-        """
-        self.messages.to_csv(filepath, **kwargs)
-
-    def from_csv(self, filepath: str, **kwargs) -> None:
-        """
-        Loads messages from a CSV file into the conversation.
-
-        Args:
-            filepath (str): The path to the file from which to load the conversation.
-        """
-        self.messages = pd.read_csv(filepath, **kwargs)
-
-    def to_json(self, filepath: str) -> None:
-        """
-        Saves the entire conversation as a JSON file.
-
-        Args:
-            filepath (str): The file path where the conversation will be saved.
-        """
-        self.messages.to_json(
-            filepath, orient="records", lines=True, date_format="iso"
-        )
-
-    def from_json(self, filepath: str) -> None:
-        """
-        Loads messages from a JSON file into the conversation.
-
-        Args:
-            filepath (str): The file path from which to load the conversation.
-        """
-        self.reset()
-        self.messages = pd.read_json(filepath, orient="records", lines=True)
-
-    # def append(self, other: 'Conversation'):
-    #     self.messages = pd.concat([self.messages, other.messages]).reset_index(drop=True)

lionagi/core/flow.py

@@ -1,8 +0,0 @@
-# from ..structures import Relationship, Structure
-# from .sessions import Session
-# from .flow_util import run_session
-
-# # represents structured sessions
-
-# class Flow(Structure):
-#     ...