lionagi 0.2.7__py3-none-any.whl → 0.2.8__py3-none-any.whl

lionagi/core/generic/graph.py

@@ -2,18 +2,17 @@ import contextlib
 from collections import deque
 from typing import Any
 
-from lionagi.libs.ln_convert import to_list
+from lionagi.core.collections import Pile, pile
 from lionagi.core.collections.abc import (
-    Condition,
     Actionable,
-    LionTypeError,
+    Condition,
     ItemNotFoundError,
     LionIDable,
+    LionTypeError,
 )
-from lionagi.core.collections import pile, Pile
-
 from lionagi.core.generic.edge import Edge
 from lionagi.core.generic.node import Node
+from lionagi.libs.ln_convert import to_list
 
 
 class Graph(Node):
@@ -211,8 +210,8 @@ class Graph(Node):
         SysUtil.check_import("networkx")
         SysUtil.check_import("matplotlib", "pyplot")
 
-        import networkx as nx
         import matplotlib.pyplot as plt
+        import networkx as nx
 
         g = self.to_networkx(**kwargs)
         pos = nx.spring_layout(g)

lionagi/core/generic/node.py

@@ -8,12 +8,12 @@ Includes functionality for managing relationships, such as adding,
 modifying, and removing edges, and querying related nodes and connections.
 """
 
-from pydantic import Field
-from pandas import Series
 from typing import Callable
 
-from lionagi.libs.ln_convert import to_list
+from pandas import Series
+from pydantic import Field
 
+from lionagi.core.collections import Pile, pile
 from lionagi.core.collections.abc import (
     Component,
     Condition,
@@ -21,8 +21,8 @@ from lionagi.core.collections.abc import (
     RelationError,
     get_lion_id,
 )
-from lionagi.core.collections import pile, Pile
 from lionagi.core.generic.edge import Edge
+from lionagi.libs.ln_convert import to_list
 
 
 class Node(Component, Relatable):

lionagi/core/generic/tree.py

@@ -1,10 +1,11 @@
 """This module provides tree structure."""
 
 from pydantic import Field
+
 from lionagi.core.collections.abc import Condition
 from lionagi.core.collections.util import to_list_type
-from lionagi.core.generic.tree_node import TreeNode
 from lionagi.core.generic.graph import Graph
+from lionagi.core.generic.tree_node import TreeNode
 
 
 class Tree(Graph):

lionagi/core/generic/tree_node.py

@@ -1,5 +1,7 @@
 from enum import Enum
+
 from pydantic import Field
+
 from lionagi.core.collections.abc import Condition
 from lionagi.core.collections.util import to_list_type
 from lionagi.core.generic.node import Node

lionagi/core/mail/__init__.py

@@ -3,7 +3,6 @@ from .mail_manager import MailManager
 from .package import Package
 from .start_mail import StartMail
 
-
 __all__ = [
     "Mail",
     "MailManager",

lionagi/core/mail/mail.py

@@ -1,5 +1,6 @@
 from lionagi.core.collections.abc import Element, Field, Sendable
-from .package import PackageCategory, Package
+
+from .package import Package, PackageCategory
 
 
 class Mail(Element, Sendable):

lionagi/core/mail/mail_manager.py

@@ -1,11 +1,13 @@
+import asyncio
 from collections import deque
+
 from pydantic import Field
-from lionagi.libs import AsyncUtil
-from lionagi.core.collections.abc import Executable, Element
-from lionagi.core.collections import Exchange
-from lionagi.core.collections.util import to_list_type, get_lion_id
+
+from lionagi.core.collections import Exchange, Pile, pile
+from lionagi.core.collections.abc import Element, Executable
+from lionagi.core.collections.util import get_lion_id, to_list_type
+
 from .mail import Mail, Package
-from lionagi.core.collections import Pile, pile
 
 
 class MailManager(Element, Executable):
@@ -178,4 +180,4 @@ class MailManager(Element, Executable):
         while not self.execute_stop:
             self.collect_all()
             self.send_all()
-            await AsyncUtil.sleep(refresh_time)
+            await asyncio.sleep(refresh_time)

lionagi/core/mail/package.py

@@ -1,6 +1,8 @@
 from enum import Enum
 from typing import Any
+
 from pydantic import field_validator
+
 from lionagi.core.collections.abc import Element, Field
 
 

lionagi/core/mail/start_mail.py

@@ -1,8 +1,8 @@
-from collections import deque
 from pydantic import Field
+
+from lionagi.core.collections import Exchange
 from lionagi.core.generic.node import Node
 from lionagi.core.mail.mail import Mail, Package
-from lionagi.core.collections import Exchange
 
 
 class StartMail(Node):

lionagi/core/message/__init__.py

@@ -1,12 +1,11 @@
-from .message import RoledMessage, MessageRole
-from .system import System
-from .instruction import Instruction
-from .assistant_response import AssistantResponse
 from .action_request import ActionRequest
 from .action_response import ActionResponse
+from .assistant_response import AssistantResponse
+from .instruction import Instruction
+from .message import MessageRole, RoledMessage
+from .system import System
 from .util import create_message
 
-
 __all__ = [
     "RoledMessage",
     "MessageRole",

lionagi/core/message/action_request.py

@@ -1,23 +1,9 @@
-"""
-Copyright 2024 HaiyangLi
-
-Licensed under the Apache License, Version 2.0 (the "License");
-you may not use this file except in compliance with the License.
-You may obtain a copy of the License at
-
-    http://www.apache.org/licenses/LICENSE-2.0
-
-Unless required by applicable law or agreed to in writing, software
-distributed under the License is distributed on an "AS IS" BASIS,
-WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-See the License for the specific language governing permissions and
-limitations under the License.
-"""
-
 import inspect
+
+from lionfuncs import to_dict
 from pydantic import Field
-from lionagi.libs import convert, ParseUtil
-from .message import RoledMessage, MessageRole
+
+from .message import MessageRole, RoledMessage
 
 
 class ActionRequest(RoledMessage):
@@ -111,21 +97,9 @@ class ActionRequest(RoledMessage):
 
 
 def _prepare_arguments(arguments):
-    """
-    Prepares the arguments for the action request.
-
-    Args:
-        arguments (Any): The arguments to be prepared.
-
-    Returns:
-        dict: The prepared arguments.
-
-    Raises:
-        ValueError: If the arguments are invalid.
-    """
     if not isinstance(arguments, dict):
         try:
-            arguments = ParseUtil.fuzzy_parse_json(convert.to_str(arguments))
+            arguments = to_dict(str(arguments), fuzzy_parse=True)
         except Exception as e:
             raise ValueError(f"Invalid arguments: {e}") from e
     if isinstance(arguments, dict):

lionagi/core/message/action_response.py

@@ -1,23 +1,9 @@
-"""
-Copyright 2024 HaiyangLi
-
-Licensed under the Apache License, Version 2.0 (the "License");
-you may not use this file except in compliance with the License.
-You may obtain a copy of the License at
-
-    http://www.apache.org/licenses/LICENSE-2.0
-
-Unless required by applicable law or agreed to in writing, software
-distributed under the License is distributed on an "AS IS" BASIS,
-WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-See the License for the specific language governing permissions and
-limitations under the License.
-"""
-
 from typing import Any
+
 from pydantic import Field
-from .message import RoledMessage, MessageRole
+
 from .action_request import ActionRequest
+from .message import MessageRole, RoledMessage
 
 
 # action response must correlates to a specific action request
@@ -95,12 +81,6 @@ class ActionResponse(RoledMessage):
         action_request.action_response = self.ln_id
 
     def _to_dict(self):
-        """
-        Converts the action response to a dictionary.
-
-        Returns:
-            dict: A dictionary representation of the action response.
-        """
         return {
             "function": self.function,
             "arguments": self.arguments,

lionagi/core/message/assistant_response.py

@@ -1,21 +1,6 @@
-"""
-Copyright 2024 HaiyangLi
-
-Licensed under the Apache License, Version 2.0 (the "License");
-you may not use this file except in compliance with the License.
-You may obtain a copy of the License at
-
-    http://www.apache.org/licenses/LICENSE-2.0
-
-Unless required by applicable law or agreed to in writing, software
-distributed under the License is distributed on an "AS IS" BASIS,
-WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-See the License for the specific language governing permissions and
-limitations under the License.
-"""
-
 from typing import Any
-from .message import RoledMessage, MessageRole
+
+from .message import MessageRole, RoledMessage
 
 
 class AssistantResponse(RoledMessage):

lionagi/core/message/instruction.py

@@ -1,22 +1,6 @@
-"""
-Copyright 2024 HaiyangLi
-
-Licensed under the Apache License, Version 2.0 (the "License");
-you may not use this file except in compliance with the License.
-You may obtain a copy of the License at
-
-    http://www.apache.org/licenses/LICENSE-2.0
-
-Unless required by applicable law or agreed to in writing, software
-distributed under the License is distributed on an "AS IS" BASIS,
-WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-See the License for the specific language governing permissions and
-limitations under the License.
-"""
-
-from lionagi.core.collections.abc import LionIDable, SYSTEM_FIELDS
+from lionagi.core.collections.abc import SYSTEM_FIELDS, LionIDable
+from lionagi.core.message.message import MessageRole, RoledMessage
 from lionagi.core.report.form import Form
-from lionagi.core.message.message import RoledMessage, MessageRole
 
 
 class Instruction(RoledMessage):
@@ -108,13 +92,6 @@ class Instruction(RoledMessage):
         return text_msg
 
     def _add_context(self, context: dict | str | None = None, **kwargs):
-        """
-        Adds context to the instruction message.
-
-        Args:
-            context (dict or str, optional): Additional context to be added.
-            **kwargs: Additional context fields to be added.
-        """
         if "context" not in self.content:
             self.content["context"] = {}
         if isinstance(context, dict):
@@ -123,12 +100,6 @@ class Instruction(RoledMessage):
             self.content["context"]["additional_context"] = context
 
     def _update_requested_fields(self, requested_fields: dict):
-        """
-        Updates the requested fields in the instruction message.
-
-        Args:
-            requested_fields (dict): The fields requested in the instruction.
-        """
         if "context" not in self.content:
             self.content["context"] = {}
             self.content["context"]["requested_fields"] = {}
@@ -137,14 +108,6 @@ class Instruction(RoledMessage):
     def _initiate_content(
         self, context, requested_fields, images, image_detail, **kwargs
     ):
-        """
-        Processes context and requested fields to update the message content.
-
-        Args:
-            context (dict or str, optional): Additional context for the instruction.
-            requested_fields (dict, optional): Fields requested in the instruction.
-            **kwargs: Additional context fields to be added.
-        """
         if context:
             context = {"context": context} if not isinstance(context, dict) else context
             if (
@@ -188,15 +151,6 @@ class Instruction(RoledMessage):
 
     @staticmethod
     def _format_requested_fields(requested_fields):
-        """
-        Formats the requested fields into a JSON-parseable response format.
-
-        Args:
-            requested_fields (dict): The fields requested in the instruction.
-
-        Returns:
-            dict: The formatted requested fields.
-        """
         format_ = f"""
         MUST RETURN JSON-PARSEABLE RESPONSE ENCLOSED BY JSON CODE BLOCKS. ----
         ```json

lionagi/core/message/message.py

@@ -1,21 +1,6 @@
-"""
-Copyright 2024 HaiyangLi
-
-Licensed under the Apache License, Version 2.0 (the "License");
-you may not use this file except in compliance with the License.
-You may obtain a copy of the License at
-
-    http://www.apache.org/licenses/LICENSE-2.0
-
-Unless required by applicable law or agreed to in writing, software
-distributed under the License is distributed on an "AS IS" BASIS,
-WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-See the License for the specific language governing permissions and
-limitations under the License.
-"""
-
 from enum import Enum
-from lionagi.core.collections.abc import Sendable, Field
+
+from lionagi.core.collections.abc import Field, Sendable
 from lionagi.core.generic.node import Node
 
 

lionagi/core/message/system.py

@@ -1,22 +1,7 @@
-"""
-Copyright 2024 HaiyangLi
-
-Licensed under the Apache License, Version 2.0 (the "License");
-you may not use this file except in compliance with the License.
-You may obtain a copy of the License at
-
-    http://www.apache.org/licenses/LICENSE-2.0
-
-Unless required by applicable law or agreed to in writing, software
-distributed under the License is distributed on an "AS IS" BASIS,
-WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-See the License for the specific language governing permissions and
-limitations under the License.
-"""
-
 from typing import Any
+
 from ..collections.abc import Field
-from .message import RoledMessage, MessageRole
+from .message import MessageRole, RoledMessage
 
 
 class System(RoledMessage):

lionagi/core/message/util.py

@@ -1,33 +1,14 @@
-"""
-Copyright 2024 HaiyangLi
-
-Licensed under the Apache License, Version 2.0 (the "License");
-you may not use this file except in compliance with the License.
-You may obtain a copy of the License at
-
-    http://www.apache.org/licenses/LICENSE-2.0
-
-Unless required by applicable law or agreed to in writing, software
-distributed under the License is distributed on an "AS IS" BASIS,
-WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-See the License for the specific language governing permissions and
-limitations under the License.
-"""
-
-import re
 import json
-import contextlib
+import re
 
-from lionagi.libs import ParseUtil
-from lionagi.libs.ln_convert import strip_lower, to_dict
-from lionagi.libs.ln_nested import nget
+from lionfuncs import nget, to_dict
 
-from .message import RoledMessage
-from .system import System
-from .instruction import Instruction
-from .assistant_response import AssistantResponse
 from .action_request import ActionRequest
 from .action_response import ActionResponse
+from .assistant_response import AssistantResponse
+from .instruction import Instruction
+from .message import RoledMessage
+from .system import System
 
 
 def create_message(
@@ -170,7 +151,7 @@ def _parse_action_request(response):
     message = to_dict(response) if not isinstance(response, dict) else response
     content_ = None
 
-    if strip_lower(nget(message, ["content"])) == "none":
+    if str(nget(message, ["content"])).strip().lower() == "none":
         content_ = _handle_action_request(message)
 
     elif nget(message, ["content", "tool_uses"], None):

lionagi/core/report/base.py

@@ -1,32 +1,10 @@
-"""
-Copyright 2024 HaiyangLi
-
-Licensed under the Apache License, Version 2.0 (the "License");
-you may not use this file except in compliance with the License.
-You may obtain a copy of the License at
-
-    http://www.apache.org/licenses/LICENSE-2.0
-
-Unless required by applicable law or agreed to in writing, software
-distributed under the License is distributed on an "AS IS" BASIS,
-WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-See the License for the specific language governing permissions and
-limitations under the License.
-"""
-
-"""
-This module defines the BaseForm class, a foundation for form handling in
-applications. It provides common functionalities for form operations like
-initialization, validation, and state management. Extend this class to
-implement specific form behaviors and configurations.
-"""
-
-from abc import abstractmethod
-from typing import Any, List, Dict
 import contextlib
+from abc import abstractmethod
+from typing import Any, Dict, List
+
 from lionagi.core.collections.abc import Component, Field
+
 from ..collections.util import to_list_type
-from lionagi.libs.ln_convert import to_str
 
 
 class BaseForm(Component):

lionagi/core/report/form.py

@@ -1,31 +1,10 @@
-"""
-Copyright 2024 HaiyangLi
-
-Licensed under the Apache License, Version 2.0 (the "License");
-you may not use this file except in compliance with the License.
-You may obtain a copy of the License at
-
-    http://www.apache.org/licenses/LICENSE-2.0
-
-Unless required by applicable law or agreed to in writing, software
-distributed under the License is distributed on an "AS IS" BASIS,
-WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-See the License for the specific language governing permissions and
-limitations under the License.
-"""
-
-"""
-This module extends the BaseForm class to implement the Form class, which
-dynamically manages form operations based on specific assignments. It provides
-functionalities for initializing fields, filling forms with data, and
-validating the readiness of forms for further processing.
-"""
-
-from typing import Dict, Any
-from lionagi.libs.ln_convert import to_readable_dict
+from typing import Any, Dict
+
+from lionfuncs import as_readable_json
+
 from lionagi.core.collections.abc import SYSTEM_FIELDS
-from lionagi.core.report.util import get_input_output_fields
 from lionagi.core.report.base import BaseForm
+from lionagi.core.report.util import get_input_output_fields
 
 
 class Form(BaseForm):
@@ -214,7 +193,7 @@ class Form(BaseForm):
         Args:
             fields (optional): Specific fields to display. Defaults to None.
         """
-        from IPython.display import display, Markdown
+        from IPython.display import Markdown, display
 
         fields = fields or self.work_fields
 
@@ -224,7 +203,7 @@ class Form(BaseForm):
 
         for k, v in fields.items():
             if isinstance(v, dict):
-                v = to_readable_dict(v)
+                v = as_readable_json(v)
             if len(str(v)) > 50:
                 display(Markdown(f"**{k}**: \n {v}"))
             else:

lionagi/core/report/report.py

@@ -1,33 +1,10 @@
-"""
-Copyright 2024 HaiyangLi
-
-Licensed under the Apache License, Version 2.0 (the "License");
-you may not use this file except in compliance with the License.
-You may obtain a copy of the License at
-
-    http://www.apache.org/licenses/LICENSE-2.0
-
-Unless required by applicable law or agreed to in writing, software
-distributed under the License is distributed on an "AS IS" BASIS,
-WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-See the License for the specific language governing permissions and
-limitations under the License.
-"""
-
-"""
-This module introduces the Report class, an extension of the BaseForm class
-designed to manage and synchronize a collection of Form instances based on
-specific assignments. The Report class handles the creation and updating of
-forms, ensuring each is properly configured according to the report's
-requirements.
-"""
-
 from typing import Any, Type
-from lionagi.core.collections.abc import Field
+
 from lionagi.core.collections import Pile, pile
-from lionagi.core.report.util import get_input_output_fields
+from lionagi.core.collections.abc import Field
 from lionagi.core.report.base import BaseForm
 from lionagi.core.report.form import Form
+from lionagi.core.report.util import get_input_output_fields
 
 
 class Report(BaseForm):

lionagi/core/report/util.py

@@ -1,19 +1,4 @@
-from lionagi.libs.ln_convert import strip_lower
-
-
 def get_input_output_fields(str_: str) -> list[list[str]]:
-    """
-    Parses an assignment string to extract input and output fields.
-
-    Args:
-        str_ (str): The assignment string in the format 'inputs -> outputs'.
-
-    Returns:
-        list[list[str]]: A list containing two lists - one for input fields and one for requested fields.
-
-    Raises:
-        ValueError: If the assignment string is None or if it does not contain '->' indicating invalid format.
-    """
     if str_ is None:
         return [], []
 
@@ -22,7 +7,7 @@ def get_input_output_fields(str_: str) -> list[list[str]]:
 
     inputs, outputs = str_.split("->")
 
-    input_fields = [strip_lower(i) for i in inputs.split(",")]
-    requested_fields = [strip_lower(o) for o in outputs.split(",")]
+    input_fields = [str(i).strip().lower() for i in inputs.split(",")]
+    requested_fields = [str(o).strip().lower() for o in outputs.split(",")]
 
     return input_fields, requested_fields

lionagi/core/rule/_default.py

@@ -1,10 +1,11 @@
 from enum import Enum
+
+from .action import ActionRequestRule
+from .boolean import BooleanRule
 from .choice import ChoiceRule
 from .mapping import MappingRule
 from .number import NumberRule
-from .boolean import BooleanRule
 from .string import StringRule
-from .action import ActionRequestRule
 
 
 class DEFAULT_RULES(Enum):

lionagi/core/rule/action.py

@@ -1,23 +1,8 @@
-"""
-Copyright 2024 HaiyangLi
-
-Licensed under the Apache License, Version 2.0 (the "License");
-you may not use this file except in compliance with the License.
-You may obtain a copy of the License at
-
-    http://www.apache.org/licenses/LICENSE-2.0
+from enum import Enum
 
-Unless required by applicable law or agreed to in writing, software
-distributed under the License is distributed on an "AS IS" BASIS,
-WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-See the License for the specific language governing permissions and
-limitations under the License.
-"""
+from lionfuncs import to_dict, to_list
 
-from enum import Enum
 from lionagi.core.collections.abc import ActionError
-from lionagi.libs import ParseUtil
-from lionagi.libs.ln_convert import to_list, to_dict
 from lionagi.core.rule.mapping import MappingRule
 
 
@@ -83,10 +68,10 @@ class ActionRequestRule(MappingRule):
         """
         corrected = []
         if isinstance(value, str):
-            value = ParseUtil.fuzzy_parse_json(value)
+            value = to_dict(value, fuzzy_parse=True)
 
         try:
-            value = to_list(value)
+            value = to_list(value, dropna=True, flatten=True)
             for i in value:
                 i = to_dict(i)
                 if list(i.keys()) >= ["function", "arguments"]: