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

lionagi/integrations/bridge/__init__.py

@@ -1,4 +1,4 @@
-from .llamaindex_.llama_index_bridge import LlamaIndexBridge
 from .langchain_.langchain_bridge import LangchainBridge
+from .llamaindex_.llama_index_bridge import LlamaIndexBridge
 
 __all__ = ["LlamaIndexBridge", "LangchainBridge"]

lionagi/integrations/bridge/langchain_/documents.py

@@ -1,4 +1,4 @@
-from typing import Union, Callable, List, Dict, Any, TypeVar
+from typing import Any, Callable, Dict, List, TypeVar, Union
 
 from lionagi.libs.sys_util import SysUtil
 

lionagi/integrations/bridge/llamaindex_/node_parser.py

@@ -1,4 +1,5 @@
 from typing import Any
+
 from lionagi.libs.sys_util import SysUtil
 
 
@@ -25,8 +26,8 @@ def get_llama_index_node_parser(node_parser: Any):
     """
 
     SysUtil.check_import("llama_index", pip_name="llama-index")
-    from llama_index.core.node_parser.interface import NodeParser
     import llama_index.core.node_parser
+    from llama_index.core.node_parser.interface import NodeParser
 
     if not isinstance(node_parser, str) and not issubclass(node_parser, NodeParser):
         raise TypeError("node_parser must be a string or NodeParser.")

lionagi/integrations/bridge/llamaindex_/textnode.py

@@ -1,4 +1,5 @@
 from typing import Any, TypeVar
+
 from lionagi.libs.sys_util import SysUtil
 
 
@@ -24,8 +25,8 @@ def to_llama_index_node(lion_node, node_type: Any = None, **kwargs: Any) -> Any:
     """
 
     SysUtil.check_import("llama_index", pip_name="llama-index")
-    from llama_index.core.schema import BaseNode
     import llama_index.core.schema
+    from llama_index.core.schema import BaseNode
 
     node_type = node_type or "TextNode"
 

lionagi/integrations/bridge/pydantic_/pydantic_bridge.py

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

lionagi/integrations/bridge/transformers_/install_.py

@@ -1,4 +1,5 @@
 import subprocess
+
 from lionagi.libs import SysUtil
 
 

lionagi/integrations/chunker/chunk.py

@@ -1,14 +1,13 @@
-from typing import Union, Callable
+from typing import Callable, Union
 
-from lionagi.libs import func_call
-from lionagi.libs.ln_convert import to_list
 from lionagi.core.collections import pile
 from lionagi.core.generic import Node
+from lionagi.libs import func_call
+from lionagi.libs.ln_convert import to_list
+
 from ..bridge.langchain_.langchain_bridge import LangchainBridge
 from ..bridge.llamaindex_.llama_index_bridge import LlamaIndexBridge
-
-
-from ..loader.load_util import ChunkerType, file_to_chunks, _datanode_parser
+from ..loader.load_util import ChunkerType, _datanode_parser, file_to_chunks
 
 
 def datanodes_convert(documents, chunker_type):

lionagi/integrations/loader/load.py

@@ -1,11 +1,11 @@
 from typing import Callable
 
-from lionagi.core.generic import Node
 from lionagi.core.collections import pile
+from lionagi.core.generic import Node
+
 from ..bridge.langchain_.langchain_bridge import LangchainBridge
 from ..bridge.llamaindex_.llama_index_bridge import LlamaIndexBridge
-
-from .load_util import dir_to_nodes, ReaderType, _datanode_parser
+from .load_util import ReaderType, _datanode_parser, dir_to_nodes
 
 
 def text_reader(args, kwargs):

lionagi/integrations/loader/load_util.py

@@ -1,11 +1,11 @@
 # use utils and schema
 from enum import Enum
 from pathlib import Path
-from typing import List, Union, Dict, Any, Tuple
+from typing import Any, Dict, List, Tuple, Union
 
+from lionagi.core.generic import Node
 from lionagi.libs import convert, func_call
 from lionagi.libs.ln_tokenize import TokenizeUtil
-from lionagi.core.generic import Node
 
 
 class ReaderType(str, Enum):

lionagi/integrations/provider/__init__.py

@@ -1,5 +1,4 @@
 from ...libs.ln_api import APIUtil
-
 from .services import Services
 
 __all__ = [

lionagi/integrations/provider/_mapping.py

@@ -1,11 +1,12 @@
+from lionagi.integrations.config.oai_configs import oai_schema
+from lionagi.integrations.config.openrouter_configs import openrouter_schema
+
+from .litellm import LiteLLMService
+from .mlx_service import MLXService
 from .oai import OpenAIService
-from .openrouter import OpenRouterService
 from .ollama import OllamaService
+from .openrouter import OpenRouterService
 from .transformers import TransformersService
-from .litellm import LiteLLMService
-from .mlx_service import MLXService
-from lionagi.integrations.config.oai_configs import oai_schema
-from lionagi.integrations.config.openrouter_configs import openrouter_schema
 
 SERVICE_PROVIDERS_MAPPING = {
     "openai": {

lionagi/integrations/provider/mlx_service.py

@@ -1,8 +1,9 @@
 import re
-from lionagi.libs.sys_util import SysUtil
+
 import lionagi.libs.ln_convert as convert
-from lionagi.libs.ln_api import BaseService
 from lionagi.integrations.config.mlx_configs import model
+from lionagi.libs.ln_api import BaseService
+from lionagi.libs.sys_util import SysUtil
 
 
 class MLXService(BaseService):
@@ -14,7 +15,7 @@ class MLXService(BaseService):
         if model is not None and "olmo" in str(model).lower():
             SysUtil.check_import("olmo", pip_name="ai2-olmo")
 
-        from mlx_lm import load, generate
+        from mlx_lm import generate, load
 
         super().__init__()
 

lionagi/integrations/provider/oai.py

@@ -1,24 +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
-
-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 os import getenv
+
 from lionagi.integrations.config.oai_configs import oai_schema
 from lionagi.libs.ln_api import BaseService, PayloadPackage
 
-
 allowed_kwargs = [
     "model",
     "frequency_penalty",

lionagi/integrations/provider/ollama.py

@@ -1,5 +1,5 @@
-from lionagi.libs.ln_api import BaseService
 from lionagi.integrations.config.ollama_configs import model
+from lionagi.libs.ln_api import BaseService
 
 allowed_kwargs = [
     "model",

lionagi/integrations/provider/openrouter.py

@@ -1,4 +1,5 @@
 from os import getenv
+
 from lionagi.integrations.config.openrouter_configs import openrouter_schema
 from lionagi.libs.ln_api import BaseService, PayloadPackage
 

lionagi/integrations/provider/transformers.py

@@ -1,8 +1,8 @@
-from typing import Union, Dict, Any
 import subprocess
+from typing import Any, Dict, Union
 
-from lionagi.libs.sys_util import SysUtil
 from lionagi.libs.ln_api import BaseService
+from lionagi.libs.sys_util import SysUtil
 
 allowed_kwargs = [
     # "model",

lionagi/integrations/storage/neo4j.py

@@ -1,6 +1,6 @@
 from neo4j import AsyncGraphDatabase
 
-from lionagi.integrations.storage.storage_util import output_node_list, output_edge_list
+from lionagi.integrations.storage.storage_util import output_edge_list, output_node_list
 
 
 class Neo4j:

lionagi/integrations/storage/storage_util.py

@@ -1,12 +1,11 @@
-import json
 import inspect
+import json
 import re
 
-from lionagi.core.message import System, Instruction, RoledMessage
-from lionagi.core.action import Tool, DirectiveSelection, func_to_tool
-from lionagi.core.action import DirectiveSelection
+from lionagi.core.action import DirectiveSelection, Tool, func_to_tool
 from lionagi.core.agent.base_agent import BaseAgent
 from lionagi.core.generic.edge_condition import EdgeCondition
+from lionagi.core.message import Instruction, RoledMessage, System
 
 
 def output_node_list(structure):

lionagi/integrations/storage/structure_excel.py

@@ -1,12 +1,13 @@
-import pandas as pd
 import json
 from pathlib import Path
 
-from lionagi.integrations.storage.storage_util import ParseNode
-from lionagi.core.executor.graph_executor import GraphExecutor
+import pandas as pd
+
 from lionagi.core.agent.base_agent import BaseAgent
-from lionagi.core.executor.base_executor import BaseExecutor
 from lionagi.core.engine.instruction_map_engine import InstructionMapEngine
+from lionagi.core.executor.base_executor import BaseExecutor
+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"):

lionagi/integrations/storage/to_csv.py

@@ -1,8 +1,9 @@
 import zipfile
-import pandas as pd
 from pathlib import Path
 
-from lionagi.integrations.storage.storage_util import output_node_list, output_edge_list
+import pandas as pd
+
+from lionagi.integrations.storage.storage_util import output_edge_list, output_node_list
 
 
 def _output_csv(

lionagi/integrations/storage/to_excel.py

@@ -1,7 +1,7 @@
 import pandas as pd
-from lionagi.libs import SysUtil
 
-from lionagi.integrations.storage.storage_util import output_node_list, output_edge_list
+from lionagi.integrations.storage.storage_util import output_edge_list, output_node_list
+from lionagi.libs import SysUtil
 
 
 def _output_excel(

lionagi/libs/__init__.py

@@ -1,33 +1,29 @@
-from lionagi.libs.sys_util import SysUtil
-from lionagi.libs.ln_async import AsyncUtil
-
 import lionagi.libs.ln_convert as convert
-from lionagi.libs.ln_convert import (
-    to_str,
-    to_list,
-    to_dict,
-    to_df,
-    to_readable_dict,
-    to_num,
-)
 import lionagi.libs.ln_dataframe as dataframe
 import lionagi.libs.ln_func_call as func_call
-from lionagi.libs.ln_func_call import lcall, CallDecorator
 import lionagi.libs.ln_nested as nested
-from lionagi.libs.ln_nested import nget, nset, ninsert
-from lionagi.libs.ln_parse import ParseUtil, StringMatch
-
 from lionagi.libs.ln_api import (
     APIUtil,
-    SimpleRateLimiter,
-    StatusTracker,
     BaseService,
     PayloadPackage,
+    SimpleRateLimiter,
+    StatusTracker,
 )
-
+from lionagi.libs.ln_async import AsyncUtil
+from lionagi.libs.ln_convert import (
+    to_df,
+    to_dict,
+    to_list,
+    to_num,
+    to_readable_dict,
+    to_str,
+)
+from lionagi.libs.ln_func_call import CallDecorator, lcall
 from lionagi.libs.ln_image import ImageUtil
+from lionagi.libs.ln_nested import nget, ninsert, nset
+from lionagi.libs.ln_parse import ParseUtil, StringMatch
 from lionagi.libs.ln_validate import validation_funcs
-
+from lionagi.libs.sys_util import SysUtil
 
 __all__ = [
     "SysUtil",

lionagi/libs/ln_api.py

@@ -1,35 +1,17 @@
-"""
-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 collections.abc import Sequence, Mapping
-
-from abc import ABC
-from dataclasses import dataclass
-
+import asyncio
 import contextlib
 import logging
 import re
-import asyncio
-import aiohttp
+from abc import ABC
+from collections.abc import Mapping, Sequence
+from dataclasses import dataclass
+from typing import Any, Callable, NoReturn, Type
 
-from typing import Any, NoReturn, Type, Callable
+import aiohttp
 
-from lionagi.libs.ln_async import AsyncUtil
-from lionagi.libs.ln_convert import to_dict, strip_lower, to_str
 import lionagi.libs.ln_func_call as func_call
+from lionagi.libs.ln_async import AsyncUtil
+from lionagi.libs.ln_convert import strip_lower, to_dict, to_str
 from lionagi.libs.ln_nested import nget
 
 
@@ -372,6 +354,7 @@ class APIUtil:
                 # Expected token calculation for the given payload and endpoint.
         """
         import tiktoken
+
         from .ln_image import ImageUtil
 
         token_encoding_name = token_encoding_name or "cl100k_base"

lionagi/libs/ln_async.py

@@ -1,10 +1,11 @@
 from __future__ import annotations
 
 import asyncio
-from collections.abc import Mapping, Callable, Coroutine
+import logging
+from collections.abc import Callable, Coroutine, Mapping
 from functools import lru_cache
 from typing import Any
-import logging
+
 import aiocache
 import aiohttp
 

lionagi/libs/ln_convert.py

@@ -1,24 +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.
-"""
-
 import json
 import re
 from functools import singledispatch
-
-from typing import Any, Type, Iterable, Generator
+from typing import Any, Generator, Iterable, Type
 
 import pandas as pd
 from pydantic import BaseModel

lionagi/libs/ln_func_call.py

@@ -1,31 +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.
-"""
-
 from __future__ import annotations
 
-import functools
 import asyncio
+import functools
 import logging
 from concurrent.futures import ThreadPoolExecutor
-
 from typing import Any, Callable, Coroutine
 
-from lionagi.libs.sys_util import SysUtil
-from lionagi.libs.ln_convert import to_list
 from lionagi.libs.ln_async import AsyncUtil
+from lionagi.libs.ln_convert import to_list
+from lionagi.libs.sys_util import SysUtil
 
 
 def lru_cache(*args, **kwargs):

lionagi/libs/ln_image.py

@@ -1,6 +1,8 @@
 import base64
-import numpy as np
 from typing import Optional
+
+import numpy as np
+
 from .sys_util import SysUtil
 
 
@@ -70,6 +72,7 @@ class ImageUtil:
         """
         import base64
         from io import BytesIO
+
         from PIL import Image
 
         # Decode the base64 string to get image data

lionagi/libs/ln_knowledge_graph.py

@@ -1,4 +1,5 @@
 import math
+
 from lionagi.libs import CallDecorator as cd
 
 
@@ -282,11 +283,13 @@ class KGTripletExtractor:
         )
 
         try:
-            from transformers import AutoModelForSeq2SeqLM, AutoTokenizer  # type: ignore
+            from transformers import AutoModelForSeq2SeqLM  # type: ignore
+            from transformers import AutoTokenizer
         except ImportError:
             install_transformers()
-        from transformers import AutoModelForSeq2SeqLM, AutoTokenizer  # type: ignore
         import torch  # type: ignore
+        from transformers import AutoModelForSeq2SeqLM  # type: ignore
+        from transformers import AutoTokenizer
 
         """
         Extract knowledge graph triplets from text and create a KnowledgeBase (KB) containing entities and relations.

lionagi/libs/ln_nested.py

@@ -1,25 +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 collections import defaultdict
 from itertools import chain
-from typing import Any, Generator, Callable
+from typing import Any, Callable, Generator
 
-from lionagi.libs.sys_util import SysUtil
 import lionagi.libs.ln_convert as convert
+from lionagi.libs.sys_util import SysUtil
 
 
 def nset(nested_structure: dict | list, indices: list[int | str], value: Any) -> None:

lionagi/libs/ln_parse.py

@@ -1,27 +1,12 @@
-"""
-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 collections.abc import Callable
-import re
 import inspect
 import itertools
+import re
+from collections.abc import Callable
 from typing import Any
+
 import numpy as np
-import lionagi.libs.ln_convert as convert
 
+import lionagi.libs.ln_convert as convert
 
 md_json_char_map = {"\n": "\\n", "\r": "\\r", "\t": "\\t", '"': '\\"'}
 

lionagi/libs/ln_queue.py

@@ -1,25 +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.
-"""
-
 """
 A class that manages asynchronous task processing with controlled concurrency.
 """
 
-from typing import Any, Callable
 import asyncio
+from typing import Any, Callable
+
 from lionagi.libs import func_call
 
 

lionagi/libs/ln_tokenize.py

@@ -1,5 +1,7 @@
-import tiktoken
 import math
+
+import tiktoken
+
 from .ln_convert import to_str
 from .special_tokens import disallowed_tokens
 

lionagi/libs/ln_validate.py

@@ -1,19 +1,3 @@
-"""
-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 provides functions for validating and fixing field values based on their data types.
 
@@ -22,8 +6,8 @@ including numeric, boolean, string, and enum. It also provides a dictionary `val
 maps data types to their corresponding validation functions.
 """
 
-from .ln_convert import to_str, is_same_dtype, to_list, to_dict, to_num, strip_lower
-from .ln_parse import StringMatch, ParseUtil
+from .ln_convert import is_same_dtype, strip_lower, to_dict, to_list, to_num, to_str
+from .ln_parse import ParseUtil, StringMatch
 
 
 def check_dict_field(x, keys: list[str] | dict, fix_=True, **kwargs):

lionagi/libs/sys_util.py

@@ -1,19 +1,3 @@
-"""
-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 copy
 import importlib
 import logging

lionagi/lions/coder/code_form.py

@@ -1,8 +1,10 @@
 from typing import Any
+
 from pydantic import Field
-from lionagi.libs import ParseUtil, SysUtil
+
 from lionagi.core import Session
 from lionagi.core.form.action_form import ActionForm
+from lionagi.libs import ParseUtil, SysUtil
 
 
 class CodeForm(ActionForm):