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

lionagi/integrations/bridge/llamaindex_/llama_pack.py

@@ -20,7 +20,9 @@ class LlamaPack:
         return pack(*args, **kwargs)
 
     @staticmethod
-    def stock_market_pack(pack_path="./stock_market_data_pack", args=[], **kwargs):
+    def stock_market_pack(
+        pack_path="./stock_market_data_pack", args=[], **kwargs
+    ):
         name_ = "StockMarketDataQueryEnginePack"
         return LlamaPack.build(name_, pack_path, args, **kwargs)
 
@@ -32,7 +34,9 @@ class LlamaPack:
         return LlamaPack.build(name_, pack_path, args, **kwargs)
 
     @staticmethod
-    def rag_evaluator_pack(pack_path="./rag_evaluator_pack", args=[], **kwargs):
+    def rag_evaluator_pack(
+        pack_path="./rag_evaluator_pack", args=[], **kwargs
+    ):
         name_ = "RagEvaluatorPack"
         return LlamaPack.build(name_, pack_path, args, **kwargs)
 
@@ -49,17 +53,23 @@ class LlamaPack:
         return LlamaPack.build(name_, pack_path, args, **kwargs)
 
     @staticmethod
-    def resume_screener_pack(pack_path="./resume_screener_pack", args=[], **kwargs):
+    def resume_screener_pack(
+        pack_path="./resume_screener_pack", args=[], **kwargs
+    ):
         name_ = "ResumeScreenerPack"
         return LlamaPack.build(name_, pack_path, args, **kwargs)
 
     @staticmethod
-    def ragatouille_retriever_pack(pack_path="./ragatouille_pack", args=[], **kwargs):
+    def ragatouille_retriever_pack(
+        pack_path="./ragatouille_pack", args=[], **kwargs
+    ):
         name_ = "RAGatouilleRetrieverPack"
         return LlamaPack.build(name_, pack_path, args, **kwargs)
 
     @staticmethod
-    def chain_of_table_pack(pack_path="./chain_of_table_pack", args=[], **kwargs):
+    def chain_of_table_pack(
+        pack_path="./chain_of_table_pack", args=[], **kwargs
+    ):
         name_ = "ChainOfTablePack"
         return LlamaPack.build(name_, pack_path, args, **kwargs)
 
@@ -152,7 +162,9 @@ class LlamaPack:
         return LlamaPack.build(name_, pack_path, args, **kwargs)
 
     @staticmethod
-    def llama_guard_moderator_pack(pack_path="./llamaguard_pack", args=[], **kwargs):
+    def llama_guard_moderator_pack(
+        pack_path="./llamaguard_pack", args=[], **kwargs
+    ):
         name_ = "LlamaGuardModeratorPack"
         return LlamaPack.build(name_, pack_path, args, **kwargs)
 
@@ -178,7 +190,9 @@ class LlamaPack:
         return LlamaPack.build(name_, pack_path, args, **kwargs)
 
     @staticmethod
-    def multi_tenancy_rag_pack(pack_path="./multitenancy_rag_pack", args=[], **kwargs):
+    def multi_tenancy_rag_pack(
+        pack_path="./multitenancy_rag_pack", args=[], **kwargs
+    ):
         name_ = "MultiTenancyRAGPack"
         return LlamaPack.build(name_, pack_path, args, **kwargs)
 
@@ -188,12 +202,16 @@ class LlamaPack:
         return LlamaPack.build(name_, pack_path, args, **kwargs)
 
     @staticmethod
-    def snowflake_query_engine_pack(pack_path="./snowflake_pack", args=[], **kwargs):
+    def snowflake_query_engine_pack(
+        pack_path="./snowflake_pack", args=[], **kwargs
+    ):
         name_ = "SnowflakeQueryEnginePack"
         return LlamaPack.build(name_, pack_path, args, **kwargs)
 
     @staticmethod
-    def agent_search_retriever_pack(pack_path="./agent_search_pack", args=[], **kwargs):
+    def agent_search_retriever_pack(
+        pack_path="./agent_search_pack", args=[], **kwargs
+    ):
         name_ = "AgentSearchRetrieverPack"
         return LlamaPack.build(name_, pack_path, args, **kwargs)
 
@@ -203,12 +221,16 @@ class LlamaPack:
         return LlamaPack.build(name_, pack_path, args, **kwargs)
 
     @staticmethod
-    def chroma_autoretrieval_pack(pack_path="./chroma_pack", args=[], **kwargs):
+    def chroma_autoretrieval_pack(
+        pack_path="./chroma_pack", args=[], **kwargs
+    ):
         name_ = "ChromaAutoretrievalPack"
         return LlamaPack.build(name_, pack_path, args, **kwargs)
 
     @staticmethod
-    def arize_phoenix_query_engine_pack(pack_path="./arize_pack", args=[], **kwargs):
+    def arize_phoenix_query_engine_pack(
+        pack_path="./arize_pack", args=[], **kwargs
+    ):
         name_ = "ArizePhoenixQueryEnginePack"
         return LlamaPack.build(name_, pack_path, args, **kwargs)
 
@@ -227,6 +249,8 @@ class LlamaPack:
         return LlamaPack.build(name_, pack_path, args, **kwargs)
 
     @staticmethod
-    def weaviate_retry_engine_pack(pack_path="./weaviate_pack", args=[], **kwargs):
+    def weaviate_retry_engine_pack(
+        pack_path="./weaviate_pack", args=[], **kwargs
+    ):
         name_ = "WeaviateRetryEnginePack"
         return LlamaPack.build(name_, pack_path, args, **kwargs)

lionagi/integrations/bridge/llamaindex_/node_parser.py

@@ -33,7 +33,9 @@ def get_llama_index_node_parser(node_parser: Any):
         pip_name="llama-index",
     )
 
-    if not isinstance(node_parser, str) and not issubclass(node_parser, NodeParser):
+    if not isinstance(node_parser, str) and not issubclass(
+        node_parser, NodeParser
+    ):
         raise TypeError("node_parser must be a string or NodeParser.")
 
     if isinstance(node_parser, str):
@@ -41,11 +43,14 @@ def get_llama_index_node_parser(node_parser: Any):
             SysUtil.check_import("tree_sitter_languages")
 
         try:
-            node_module = import_module("llama_index.core", module_name="node_parser")
+            node_module = import_module(
+                "llama_index.core", module_name="node_parser"
+            )
             return getattr(node_module, node_parser)
         except Exception as e:
             raise AttributeError(
-                f"llama_index_core has no such attribute:" f" {node_parser}, Error: {e}"
+                f"llama_index_core has no such attribute:"
+                f" {node_parser}, Error: {e}"
             ) from e
 
     elif isinstance(node_parser, NodeParser):

lionagi/integrations/bridge/llamaindex_/reader.py

@@ -40,7 +40,9 @@ def get_llama_index_reader(reader: Any | str = None) -> Any:
     ]:
         return SimpleDirectoryReader
 
-    if not isinstance(reader, str) and not issubclass(reader, BasePydanticReader):
+    if not isinstance(reader, str) and not issubclass(
+        reader, BasePydanticReader
+    ):
         raise TypeError(f"reader must be a string or BasePydanticReader.")
 
     if isinstance(reader, str):

lionagi/integrations/bridge/llamaindex_/textnode.py

@@ -3,7 +3,9 @@ from typing import Any, TypeVar
 from lionagi.libs.sys_util import SysUtil
 
 
-def to_llama_index_node(lion_node, node_type: Any = None, **kwargs: Any) -> Any:
+def to_llama_index_node(
+    lion_node, node_type: Any = None, **kwargs: Any
+) -> Any:
     """
     Converts a Lion node to a Llama Index node of a specified type.
 
@@ -44,10 +46,14 @@ def to_llama_index_node(lion_node, node_type: Any = None, **kwargs: Any) -> Any:
             if isinstance(node_type, str) and hasattr(
                 llama_index.core.schema, node_type
             ):
-                return getattr(llama_index.core.schema, node_type).from_dict(_dict)
+                return getattr(llama_index.core.schema, node_type).from_dict(
+                    _dict
+                )
             elif issubclass(node_type, BaseNode):
                 return node_type.from_dict(_dict)
             else:
-                raise AttributeError(f"Invalid llama-index node type: {node_type}")
+                raise AttributeError(
+                    f"Invalid llama-index node type: {node_type}"
+                )
         except Exception as e:
             raise AttributeError(f"Error: {e}")

lionagi/integrations/bridge/pydantic_/pydantic_bridge.py

@@ -1 +1,7 @@
-from pydantic import AliasChoices, BaseModel, Field, ValidationError, field_serializer
+from pydantic import (
+    AliasChoices,
+    BaseModel,
+    Field,
+    ValidationError,
+    field_serializer,
+)

lionagi/integrations/bridge/transformers_/install_.py

@@ -34,4 +34,6 @@ def install_transformers():
             "transformers is required. Would you like to install it now? (y/n): "
         )
         if in_ == "y":
-            SysUtil.install_import(package_name="transformers", import_name="pipeline")
+            SysUtil.install_import(
+                package_name="transformers", import_name="pipeline"
+            )

lionagi/integrations/chunker/chunk.py

@@ -1,4 +1,5 @@
-from typing import Callable, Union
+from collections.abc import Callable
+from typing import Union
 
 from lionagi.core.collections import pile
 from lionagi.core.generic import Node
@@ -56,7 +57,9 @@ def text_chunker(documents, args, kwargs):
         func_call.lcall(chunks, lambda chunk: chunk.pop("ln_id"))
         return [Node.from_obj({**chunk}) for chunk in chunks]
 
-    a = to_list([chunk_node(doc) for doc in documents], flatten=True, dropna=True)
+    a = to_list(
+        [chunk_node(doc) for doc in documents], flatten=True, dropna=True
+    )
     return pile(a)
 
 

lionagi/integrations/loader/load.py

@@ -1,4 +1,4 @@
-from typing import Callable
+from collections.abc import Callable
 
 from lionagi.core.collections import pile
 from lionagi.core.generic import Node
@@ -141,7 +141,9 @@ def _plain_reader(reader, reader_args, reader_kwargs):
         ) from e
 
 
-def _langchain_reader(reader, reader_args, reader_kwargs, to_lion: bool | Callable):
+def _langchain_reader(
+    reader, reader_args, reader_kwargs, to_lion: bool | Callable
+):
     """
     Reads data using a Langchain reader.
 
@@ -157,7 +159,9 @@ def _langchain_reader(reader, reader_args, reader_kwargs, to_lion: bool | Callab
     Example usage:
         >>> nodes = _langchain_reader('langchain_reader', ['arg1'], {'key': 'value'}, True)
     """
-    nodes = LangchainBridge.langchain_loader(reader, reader_args, reader_kwargs)
+    nodes = LangchainBridge.langchain_loader(
+        reader, reader_args, reader_kwargs
+    )
     if isinstance(to_lion, bool) and to_lion is True:
         return pile([Node.from_langchain(i) for i in nodes])
 

lionagi/integrations/loader/load_util.py

@@ -24,7 +24,7 @@ class ChunkerType(str, Enum):
 
 def dir_to_path(
     dir: str, ext: str, recursive: bool = False, flatten: bool = True
-) -> List[Path]:
+) -> list[Path]:
     """
     Generates a list of file paths from a directory with the given file extension.
 
@@ -53,16 +53,18 @@ def dir_to_path(
             func_call.lcall(ext, _dir_to_path, flatten=True), flatten=flatten
         )
     except:
-        raise ValueError("Invalid directory or extension, please check the path")
+        raise ValueError(
+            "Invalid directory or extension, please check the path"
+        )
 
 
 def dir_to_nodes(
     dir_: str,
-    ext: Union[List[str], str],
+    ext: list[str] | str,
     recursive: bool = False,
     flatten: bool = True,
     clean_text: bool = True,
-) -> List[Node]:
+) -> list[Node]:
     """
     Converts directory contents into Node objects based on specified file extensions.
 
@@ -84,12 +86,16 @@ def dir_to_nodes(
         # converting them into Node objects.
     """
 
-    path_list = dir_to_path(dir=dir_, ext=ext, recursive=recursive, flatten=flatten)
+    path_list = dir_to_path(
+        dir=dir_, ext=ext, recursive=recursive, flatten=flatten
+    )
     files_info = func_call.lcall(path_list, read_text, clean=clean_text)
-    return func_call.lcall(files_info, lambda x: Node(content=x[0], metadata=x[1]))
+    return func_call.lcall(
+        files_info, lambda x: Node(content=x[0], metadata=x[1])
+    )
 
 
-def read_text(filepath: str, clean: bool = True) -> Tuple[str, dict]:
+def read_text(filepath: str, clean: bool = True) -> tuple[str, dict]:
     """
     Reads text from a file and optionally cleans it, returning the content and metadata.
 
@@ -115,9 +121,15 @@ def read_text(filepath: str, clean: bool = True) -> Tuple[str, dict]:
 
         file = filepath
         size = os.path.getsize(filepath)
-        creation_date = datetime.fromtimestamp(os.path.getctime(filepath)).date()
-        modified_date = datetime.fromtimestamp(os.path.getmtime(filepath)).date()
-        last_accessed_date = datetime.fromtimestamp(os.path.getatime(filepath)).date()
+        creation_date = datetime.fromtimestamp(
+            os.path.getctime(filepath)
+        ).date()
+        modified_date = datetime.fromtimestamp(
+            os.path.getmtime(filepath)
+        ).date()
+        last_accessed_date = datetime.fromtimestamp(
+            os.path.getatime(filepath)
+        ).date()
         return {
             "file": convert.to_str(file),
             "size": size,
@@ -127,7 +139,7 @@ def read_text(filepath: str, clean: bool = True) -> Tuple[str, dict]:
         }
 
     try:
-        with open(filepath, "r") as f:
+        with open(filepath) as f:
             content = f.read()
             if clean:
                 # Define characters to replace and their replacements
@@ -141,19 +153,22 @@ def read_text(filepath: str, clean: bool = True) -> Tuple[str, dict]:
 
 
 def _file_to_chunks(
-    input: Dict[str, Any],
+    input: dict[str, Any],
     field: str = "content",
     chunk_size: int = 1500,
     overlap: float = 0.1,
     threshold: int = 200,
-) -> List[Dict[str, Any]]:
+) -> list[dict[str, Any]]:
     try:
         out = {key: value for key, value in input.items() if key != field} | {
             "chunk_overlap": overlap,
             "chunk_threshold": threshold,
         }
         chunks = TokenizeUtil.chunk_by_chars(
-            input[field], chunk_size=chunk_size, overlap=overlap, threshold=threshold
+            input[field],
+            chunk_size=chunk_size,
+            overlap=overlap,
+            threshold=threshold,
         )
         logs = []
         for i, chunk in enumerate(chunks):
@@ -168,7 +183,9 @@ def _file_to_chunks(
         return logs
 
     except Exception as e:
-        raise ValueError(f"An error occurred while chunking the file. {e}") from e
+        raise ValueError(
+            f"An error occurred while chunking the file. {e}"
+        ) from e
 
 
 # needs doing TODO
@@ -184,7 +201,9 @@ def file_to_chunks(
     #    verbose=True,
     #    timestamp=True,
     #    logger=None,
-    return convert.to_list(func_call.lcall(input, chunk_func, **kwargs), flatten=True)
+    return convert.to_list(
+        func_call.lcall(input, chunk_func, **kwargs), flatten=True
+    )
 
 
 def _datanode_parser(nodes, parser):

lionagi/integrations/provider/oai.py

@@ -76,7 +76,9 @@ class OpenAIService(BaseService):
         self.active_endpoint = []
         self.allowed_kwargs = allowed_kwargs
 
-    async def serve(self, input_, endpoint="chat/completions", method="post", **kwargs):
+    async def serve(
+        self, input_, endpoint="chat/completions", method="post", **kwargs
+    ):
         """
         Serves the input using the specified endpoint and method.
 
@@ -137,14 +139,18 @@ class OpenAIService(BaseService):
                     _content = []
                     for i in content:
                         if "text" in i:
-                            _content.append({"type": "text", "text": str(i["text"])})
+                            _content.append(
+                                {"type": "text", "text": str(i["text"])}
+                            )
                         elif "image_url" in i:
                             _content.append(
                                 {
                                     "type": "image_url",
                                     "image_url": {
                                         "url": f"{i['image_url'].get('url')}",
-                                        "detail": i["image_url"].get("detail", "low"),
+                                        "detail": i["image_url"].get(
+                                            "detail", "low"
+                                        ),
                                     },
                                 }
                             )
@@ -158,7 +164,10 @@ class OpenAIService(BaseService):
         )
         try:
             completion = await self.call_api(
-                payload, "chat/completions", "post", required_tokens=required_tokens
+                payload,
+                "chat/completions",
+                "post",
+                required_tokens=required_tokens,
             )
             return payload, completion
         except Exception as e:

lionagi/integrations/provider/openrouter.py

@@ -50,7 +50,9 @@ class OpenRouterService(BaseService):
         self.active_endpoint = []
         self.allowed_kwargs = allowed_kwargs
 
-    async def serve(self, input_, endpoint="chat/completions", method="post", **kwargs):
+    async def serve(
+        self, input_, endpoint="chat/completions", method="post", **kwargs
+    ):
         """
         Serves the input using the specified endpoint and method.
 
@@ -111,14 +113,18 @@ class OpenRouterService(BaseService):
                     _content = []
                     for i in content:
                         if "text" in i:
-                            _content.append({"type": "text", "text": str(i["text"])})
+                            _content.append(
+                                {"type": "text", "text": str(i["text"])}
+                            )
                         elif "image_url" in i:
                             _content.append(
                                 {
                                     "type": "image_url",
                                     "image_url": {
                                         "url": f"{i['image_url'].get('url')}",
-                                        "detail": i["image_url"].get("detail", "low"),
+                                        "detail": i["image_url"].get(
+                                            "detail", "low"
+                                        ),
                                     },
                                 }
                             )
@@ -132,7 +138,10 @@ class OpenRouterService(BaseService):
         )
         try:
             completion = await self.call_api(
-                payload, "chat/completions", "post", required_tokens=required_tokens
+                payload,
+                "chat/completions",
+                "post",
+                required_tokens=required_tokens,
             )
             return payload, completion
         except Exception as e:

lionagi/integrations/provider/services.py

@@ -68,7 +68,9 @@ class Services:
                 asynchronous operations.
         """
 
-        from lionagi.integrations.provider.transformers import TransformersService
+        from lionagi.integrations.provider.transformers import (
+            TransformersService,
+        )
 
         return TransformersService(**kwargs)
 

lionagi/integrations/provider/transformers.py

@@ -46,8 +46,8 @@ class TransformersService(BaseService):
     def __init__(
         self,
         task: str = None,
-        model: Union[str, Any] = None,
-        config: Union[str, Dict, Any] = None,
+        model: str | Any = None,
+        config: str | dict | Any = None,
         device="cpu",
         **kwargs,
     ):
@@ -83,7 +83,9 @@ class TransformersService(BaseService):
     async def serve_chat(self, messages, **kwargs):
         if self.task:
             if self.task != "conversational":
-                raise ValueError(f"Invalid transformers pipeline task: {self.task}.")
+                raise ValueError(
+                    f"Invalid transformers pipeline task: {self.task}."
+                )
 
         payload = {"messages": messages}
         config = {}

lionagi/integrations/storage/neo4j.py

@@ -1,6 +1,9 @@
 from neo4j import AsyncGraphDatabase
 
-from lionagi.integrations.storage.storage_util import output_edge_list, output_node_list
+from lionagi.integrations.storage.storage_util import (
+    output_edge_list,
+    output_node_list,
+)
 
 
 class Neo4j:
@@ -45,7 +48,9 @@ class Neo4j:
             SET n.timestamp = $timestamp
             SET n.name = $name
             """
-        await tx.run(query, ln_id=node["ln_id"], timestamp=node["timestamp"], name=name)
+        await tx.run(
+            query, ln_id=node["ln_id"], timestamp=node["timestamp"], name=name
+        )
         # heads=node['head_nodes'],
         # nodes=node['nodes'],
         # edges=node['edges'])
@@ -247,7 +252,9 @@ class Neo4j:
             MERGE (n:EdgeCondition:LionNode {className: $className})
             SET n.code = $code
             """
-        await tx.run(query, className=condCls["class_name"], code=condCls["class"])
+        await tx.run(
+            query, className=condCls["class_name"], code=condCls["class"]
+        )
 
     async def add_node(self, tx, node_dict, structure_name):
         """
@@ -278,7 +285,10 @@ class Neo4j:
             elif node == "Tool":
                 [await self.add_tool_node(tx, i) for i in node_list]
             elif node == "DirectiveSelection":
-                [await self.add_directiveSelection_node(tx, i) for i in node_list]
+                [
+                    await self.add_directiveSelection_node(tx, i)
+                    for i in node_list
+                ]
             elif node == "BaseAgent":
                 [await self.add_baseAgent_node(tx, i) for i in node_list]
             else:
@@ -382,7 +392,9 @@ class Neo4j:
                     await self.check_structure_name_constraint(tx)
                     await tx.commit()
                 except Exception as e:
-                    raise ValueError(f"transaction rolled back due to exception: {e}")
+                    raise ValueError(
+                        f"transaction rolled back due to exception: {e}"
+                    )
                 finally:
                     await tx.close()
 
@@ -395,7 +407,9 @@ class Neo4j:
                     await self.add_head_edge(tx, structure)
                     await tx.commit()
                 except Exception as e:
-                    raise ValueError(f"transaction rolled back due to exception: {e}")
+                    raise ValueError(
+                        f"transaction rolled back due to exception: {e}"
+                    )
                 finally:
                     await tx.close()
 
@@ -566,7 +580,9 @@ class Neo4j:
             else:
                 raise ValueError(f"Structure id {structure_id} is invalid")
 
-    async def get_heads(self, structure_name: str = None, structure_id: str = None):
+    async def get_heads(
+        self, structure_name: str = None, structure_id: str = None
+    ):
         """
         Asynchronously retrieves the head nodes associated with a given structure in the graph.
 

lionagi/integrations/storage/storage_util.py

@@ -58,7 +58,9 @@ def output_node_list(structure):
         elif isinstance(node, BaseAgent):
             node_output["structure_id"] = node.structure.ln_id
             node_output["output_parser"] = (
-                inspect.getsource(node.output_parser) if node.output_parser else None
+                inspect.getsource(node.output_parser)
+                if node.output_parser
+                else None
             )
         else:
             raise ValueError("Not supported node type detected")
@@ -94,7 +96,9 @@ def output_edge_list(structure):
         }
         if edge.condition:
             cls_name = edge.condition.__class__.__qualname__
-            cond = json.dumps({"class": cls_name, "args": edge.condition.model_dump()})
+            cond = json.dumps(
+                {"class": cls_name, "args": edge.condition.model_dump()}
+            )
             cls = edge.string_condition()
             if cls is not None and cls_name not in edge_cls_dict:
                 edge_cls_dict.update({cls_name: cls})
@@ -184,7 +188,9 @@ class ParseNode:
         Returns:
             System: An instantiated System node filled with properties from info_dict.
         """
-        info_dict["system"] = json.loads(info_dict.pop("content"))["system_info"]
+        info_dict["system"] = json.loads(info_dict.pop("content"))[
+            "system_info"
+        ]
         node = System.from_obj(info_dict)
         return node
 
@@ -199,7 +205,9 @@ class ParseNode:
         Returns:
             Instruction: An instantiated Instruction node filled with properties from info_dict.
         """
-        info_dict["instruction"] = json.loads(info_dict.pop("content"))["instruction"]
+        info_dict["instruction"] = json.loads(info_dict.pop("content"))[
+            "instruction"
+        ]
         node = Instruction.from_obj(info_dict)
         return node
 
@@ -218,10 +226,14 @@ class ParseNode:
         node.directive = info_dict["directive"]
         if "directive_kwargs" in info_dict:
             if info_dict["directive_kwargs"]:
-                node.directive_kwargs = json.loads(info_dict["directive_kwargs"])
+                node.directive_kwargs = json.loads(
+                    info_dict["directive_kwargs"]
+                )
         elif "directiveKwargs" in info_dict:
             if info_dict["directiveKwargs"]:
-                node.directive_kwargs = json.loads(info_dict["directiveKwargs"])
+                node.directive_kwargs = json.loads(
+                    info_dict["directiveKwargs"]
+                )
         return node
 
     @staticmethod
@@ -239,7 +251,11 @@ class ParseNode:
             ValueError: If unsafe code is detected in the function definition.
         """
         func_code = info_dict["function"]
-        if "import os" in func_code or "__" in func_code or "import sys" in func_code:
+        if (
+            "import os" in func_code
+            or "__" in func_code
+            or "import sys" in func_code
+        ):
             raise ValueError(
                 "Unsafe code detected in Tool function. Please double check or implement explicitly"
             )

lionagi/integrations/storage/structure_excel.py

@@ -10,7 +10,9 @@ from lionagi.core.executor.graph_executor import GraphExecutor
 from lionagi.integrations.storage.storage_util import ParseNode
 
 
-def excel_reload(structure_name=None, structure_id=None, dir="structure_storage"):
+def excel_reload(
+    structure_name=None, structure_id=None, dir="structure_storage"
+):
     """
     Loads a structure from an Excel file into a StructureExecutor instance.
 
@@ -58,7 +60,10 @@ class StructureExcel:
     default_agent_executable: BaseExecutor = InstructionMapEngine()
 
     def __init__(
-        self, structure_name=None, structure_id=None, file_path="structure_storage"
+        self,
+        structure_name=None,
+        structure_id=None,
+        file_path="structure_storage",
     ):
         """
         Initializes the StructureExcel class with specified parameters.