lionagi 0.5.4__tar.gz → 0.6.0__tar.gz

{lionagi-0.5.4 → lionagi-0.6.0}/.env.example

@@ -1,6 +1,6 @@
-
 # model providers
 # OPENAI_API_KEY=
 # ANTHROPIC_API_KEY=
 # PERPLEXITY_API_KEY=
 # GROQ_API_KEY=
+# OPENROUTER_API_KEY=

{lionagi-0.5.4 → lionagi-0.6.0}/.github/workflows/ci.yml

@@ -11,7 +11,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ['3.11', '3.12', '3.13']
+        python-version: ['3.10', '3.11', '3.12', '3.13']
     steps:
     - uses: actions/checkout@v4
     - name: Set up Python ${{ matrix.python-version }}

{lionagi-0.5.4 → lionagi-0.6.0}/.github/workflows/release.yml

@@ -12,7 +12,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ['3.11', '3.12', '3.13']
+        python-version: ['3.10', '3.11', '3.12', '3.13']
     steps:
     - uses: actions/checkout@v4
     - name: Set up Python ${{ matrix.python-version }}

{lionagi-0.5.4 → lionagi-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lionagi
-Version: 0.5.4
+Version: 0.6.0
 Summary: An AGentic Intelligence Operating System.
 Author-email: HaiyangLi <quantocean.li@gmail.com>
 License:            Apache License
@@ -218,18 +218,19 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Python: >=3.11
-Requires-Dist: aiohttp>=3.11.10
-Requires-Dist: jinja2>=3.1.4
+Requires-Dist: aiocache>=0.12.0
+Requires-Dist: aiohttp>=3.11.0
+Requires-Dist: jinja2>=3.1.0
 Requires-Dist: litellm>=1.55.3
 Requires-Dist: pandas>=2.0.0
 Requires-Dist: pillow>=11.0.0
-Requires-Dist: pydantic>=2.10.3
+Requires-Dist: pydantic>=2.0.0
 Requires-Dist: python-dotenv>=1.0.1
 Description-Content-Type: text/markdown
 
 ![PyPI - Version](https://img.shields.io/pypi/v/lionagi?labelColor=233476aa&color=231fc935)
 ![PyPI - Downloads](https://img.shields.io/pypi/dm/lionagi?color=blue)
-![Python Version](https://img.shields.io/badge/python-3.11%2B-blue)
+![Python Version](https://img.shields.io/badge/python-3.10%2B-blue)
 
 [Documentation](https://ocean-lion.com/Welcome) | [Discord](https://discord.gg/aqSJ2v46vu) | [PyPI](https://pypi.org/project/lionagi/) | [Roadmap](https://trello.com/b/3seomsrI/lionagi)
 
@@ -541,7 +542,7 @@ Join our [Discord community](https://discord.gg/aqSJ2v46vu) to:
   author = {Haiyang Li},
   month = {12},
   year = {2023},
-  title = {LionAGI: Intelligent Integration Framework},
+  title = {LionAGI: Towards Automated General Intelligence},
   url = {https://github.com/lion-agi/lionagi},
 }
 ```

{lionagi-0.5.4 → lionagi-0.6.0}/README.md

@@ -1,6 +1,6 @@
 ![PyPI - Version](https://img.shields.io/pypi/v/lionagi?labelColor=233476aa&color=231fc935)
 ![PyPI - Downloads](https://img.shields.io/pypi/dm/lionagi?color=blue)
-![Python Version](https://img.shields.io/badge/python-3.11%2B-blue)
+![Python Version](https://img.shields.io/badge/python-3.10%2B-blue)
 
 [Documentation](https://ocean-lion.com/Welcome) | [Discord](https://discord.gg/aqSJ2v46vu) | [PyPI](https://pypi.org/project/lionagi/) | [Roadmap](https://trello.com/b/3seomsrI/lionagi)
 
@@ -312,7 +312,7 @@ Join our [Discord community](https://discord.gg/aqSJ2v46vu) to:
   author = {Haiyang Li},
   month = {12},
   year = {2023},
-  title = {LionAGI: Intelligent Integration Framework},
+  title = {LionAGI: Towards Automated General Intelligence},
   url = {https://github.com/lion-agi/lionagi},
 }
 ```

lionagi-0.6.0/dev_tools/count_code_base_lines.py

@@ -0,0 +1,113 @@
+import os
+from collections import defaultdict
+
+
+def analyze_file(file_path):
+    with open(file_path, encoding="utf-8") as file:
+        lines = file.readlines()
+
+    stats = {
+        "total_lines": len(lines),
+        "code_lines": 0,
+        "comment_lines": 0,
+        "blank_lines": 0,
+        "docstring_lines": 0,
+    }
+
+    in_docstring = False
+    docstring_delimiter = None
+
+    for line in lines:
+        stripped = line.strip()
+
+        if not stripped:
+            stats["blank_lines"] += 1
+            continue
+
+        if in_docstring:
+            stats["docstring_lines"] += 1
+            if stripped.endswith(docstring_delimiter):
+                in_docstring = False
+            continue
+
+        if stripped.startswith("'''") or stripped.startswith('"""'):
+            stats["docstring_lines"] += 1
+            in_docstring = True
+            docstring_delimiter = stripped[0] * 3
+            if stripped.endswith(docstring_delimiter) and len(stripped) > 3:
+                in_docstring = False
+            continue
+
+        if stripped.startswith("#"):
+            stats["comment_lines"] += 1
+        else:
+            stats["code_lines"] += 1
+
+    return stats
+
+
+def analyze_package(package_path):
+    total_stats = defaultdict(lambda: defaultdict(int))
+    file_stats = {}
+
+    for root, _, files in os.walk(package_path):
+        for file in files:
+            if file.endswith((".py", ".pyx", ".pyd")):
+                file_path = os.path.join(root, file)
+                relative_path = os.path.relpath(file_path, package_path)
+                stats = analyze_file(file_path)
+                file_stats[relative_path] = stats
+
+                file_type = "Python"
+                if file.endswith(".pyx"):
+                    file_type = "Cython"
+                elif file.endswith(".pyd"):
+                    file_type = "Python DLL"
+
+                for key, value in stats.items():
+                    total_stats[file_type][key] += value
+
+    return total_stats, file_stats
+
+
+def print_analysis(total_stats, file_stats):
+    print("Package Analysis:")
+    print("=================")
+
+    grand_total = defaultdict(int)
+    for file_type, stats in total_stats.items():
+        print(f"\n{file_type} Files:")
+        print(f"  Total Lines: {stats['total_lines']}")
+        print(f"  Code Lines: {stats['code_lines']}")
+        print(f"  Comment Lines: {stats['comment_lines']}")
+        print(f"  Docstring Lines: {stats['docstring_lines']}")
+        print(f"  Blank Lines: {stats['blank_lines']}")
+
+        for key, value in stats.items():
+            grand_total[key] += value
+
+    print("\nGrand Total:")
+    print(f"  Total Lines: {grand_total['total_lines']}")
+    print(f"  Code Lines: {grand_total['code_lines']}")
+    print(f"  Comment Lines: {grand_total['comment_lines']}")
+    print(f"  Docstring Lines: {grand_total['docstring_lines']}")
+    print(f"  Blank Lines: {grand_total['blank_lines']}")
+
+    print("\nTop 10 Files by Code Lines:")
+    sorted_files = sorted(
+        file_stats.items(), key=lambda x: x[1]["code_lines"], reverse=True
+    )
+    for file, stats in sorted_files[:10]:
+        print(f"  {file}: {stats['code_lines']} code lines")
+
+
+if __name__ == "__main__":
+    print("Lion Analysis")
+    package_path = "./lionagi"  # Replace with your package path
+    total_stats, file_stats = analyze_package(package_path)
+    print_analysis(total_stats, file_stats)
+
+    print("\nDev Tool Analysis")
+    package_path = "./tests"  # Replace with your package path
+    total_stats, file_stats = analyze_package(package_path)
+    print_analysis(total_stats, file_stats)

lionagi-0.6.0/lionagi/__init__.py

@@ -0,0 +1,21 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from lionagi.operatives import types as ops_types
+from lionagi.protocols import types as types
+from lionagi.service.imodel import iModel
+from lionagi.session.session import Branch, Session
+from lionagi.version import __version__
+
+LiteiModel = iModel
+
+__all__ = (
+    "Session",
+    "Branch",
+    "iModel",
+    "LiteiModel",
+    "types",
+    "ops_types",
+    "__version__",
+)

{lionagi-0.5.4/lionagi/core → lionagi-0.6.0/lionagi}/_class_registry.py

@@ -2,23 +2,53 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
+import ast
+import importlib.util
 import os
 from typing import TypeVar
 
-from lionagi.libs.utils import get_class_file_registry, get_class_objects
-
 T = TypeVar("T")
 LION_CLASS_REGISTRY: dict[str, type[T]] = {}
 LION_CLASS_FILE_REGISTRY: dict[str, str] = {}
 
 pattern_list = [
-    "lionagi/core/generic",
-    "lionagi/core/communication",
-    "lionagi/core/action",
-    "lionagi/core/session",
-    "lionagi/core/forms",
+    "lionagi/protocols/generic",
+    "lionagi/protocols/graph",
+    "lionagi/protocols/messages",
 ]
 
+__all__ = (
+    "get_class",
+    "LION_CLASS_REGISTRY",
+)
+
+
+def get_file_classes(file_path):
+    with open(file_path) as file:
+        file_content = file.read()
+
+    tree = ast.parse(file_content)
+
+    class_file_dict = {}
+    for node in tree.body:
+        if isinstance(node, ast.ClassDef):
+            class_file_dict[node.name] = file_path
+
+    return class_file_dict
+
+
+def get_class_file_registry(folder_path, pattern_list):
+    class_file_registry = {}
+    for root, _, files in os.walk(folder_path):
+        for file in files:
+            if file.endswith(".py"):
+                if any(pattern in root for pattern in pattern_list):
+                    class_file_dict = get_file_classes(
+                        os.path.join(root, file)
+                    )
+                    class_file_registry.update(class_file_dict)
+    return class_file_registry
+
 
 if not LION_CLASS_FILE_REGISTRY:
     script_path = os.path.abspath(__file__)
@@ -29,6 +59,20 @@ if not LION_CLASS_FILE_REGISTRY:
     )
 
 
+def get_class_objects(file_path):
+    class_objects = {}
+    spec = importlib.util.spec_from_file_location("module.name", file_path)
+    module = importlib.util.module_from_spec(spec)
+    spec.loader.exec_module(module)
+
+    for class_name in dir(module):
+        obj = getattr(module, class_name)
+        if isinstance(obj, type):
+            class_objects[class_name] = obj
+
+    return class_objects
+
+
 def get_class(class_name: str) -> type:
     """
     Retrieve a class by name from the registry or dynamically import it.
@@ -64,6 +108,3 @@ def get_class(class_name: str) -> type:
         return found_class_dict[class_name]
     except Exception as e:
         raise ValueError(f"Unable to find class {class_name}: {e}")
-
-
-# File: lionagi/core/util/class_registry_util.py

lionagi-0.6.0/lionagi/_errors.py

@@ -0,0 +1,35 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
+
+class LionError(Exception):
+    pass
+
+
+class ItemNotFoundError(LionError):
+    pass
+
+
+class ItemExistsError(LionError):
+    pass
+
+
+class IDError(LionError):
+    pass
+
+
+class RelationError(LionError):
+    pass
+
+
+class RateLimitError(LionError):
+    pass
+
+
+class OperationError(LionError):
+    pass
+
+
+class ExecutionError(LionError):
+    pass

{lionagi-0.5.4 → lionagi-0.6.0}/lionagi/libs/compress/models.py

@@ -1,8 +1,12 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 import numpy as np
 from pydantic import BaseModel
 
-from lionagi.core.generic import Log
-from lionagi.core.typing import Note
+from lionagi.operatives.models.note import Note
+from lionagi.protocols.generic.log import Log
 
 
 class PerplexityTokenScore(BaseModel):

{lionagi-0.5.4 → lionagi-0.6.0}/lionagi/libs/compress/utils.py

@@ -1,22 +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.
-"""
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
 
 import asyncio
 
-from lionagi.integrations.litellm_.imodel import iModel
+from lionagi.service.imodel import iModel
 
 from .models import PerplexityScores
 

{lionagi-0.5.4 → lionagi-0.6.0}/lionagi/libs/file/chunk.py

@@ -1,3 +1,7 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 import math
 from collections.abc import Callable
 from typing import Any, Literal

{lionagi-0.5.4 → lionagi-0.6.0}/lionagi/libs/file/file_ops.py

@@ -1,3 +1,7 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 import logging
 from pathlib import Path
 from shutil import copy2

{lionagi-0.5.4 → lionagi-0.6.0}/lionagi/libs/file/params.py

@@ -1,3 +1,7 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 from collections.abc import Callable
 from pathlib import Path
 from typing import Any, Literal
@@ -45,47 +49,6 @@ class ChunkContentParams(BaseModel):
     )
 
 
-class CreatePathParams(BaseModel):
-    """Parameters for creating file paths with various options.
-
-    This model defines parameters used by the create_path function to generate
-    file paths with support for timestamps, random hashes, and existence checks.
-    """
-
-    directory: Path | str = Field(
-        description="Directory where the file will be created"
-    )
-    filename: str = Field(description="Base name of the file to create")
-    extension: str | None = Field(
-        default=None,
-        description="File extension (with or without leading dot)",
-    )
-    timestamp: bool = Field(
-        default=False, description="If True, adds a timestamp to the filename"
-    )
-    dir_exist_ok: bool = Field(
-        default=True,
-        description="If True, doesn't raise an error if the directory exists",
-    )
-    file_exist_ok: bool = Field(
-        default=False,
-        description="If True, allows overwriting of existing files",
-    )
-    time_prefix: bool = Field(
-        default=False,
-        description="If True, adds timestamp as prefix instead of suffix",
-    )
-    timestamp_format: str | None = Field(
-        default=None,
-        description="Custom format for the timestamp (strftime format)",
-    )
-    random_hash_digits: int = Field(
-        default=0,
-        ge=0,
-        description="Number of random hash digits to append to filename",
-    )
-
-
 class DirToFilesParams(BaseModel):
     """Parameters for processing directories and collecting file paths.
 

{lionagi-0.5.4 → lionagi-0.6.0}/lionagi/libs/file/process.py

@@ -1,3 +1,7 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 import logging
 from collections.abc import Callable
 from concurrent.futures import ThreadPoolExecutor, as_completed

{lionagi-0.5.4 → lionagi-0.6.0}/lionagi/libs/file/save.py

@@ -1,9 +1,13 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 import json
 import logging
 from pathlib import Path
 from typing import Any
 
-from .path import create_path
+from lionagi.utils import create_path
 
 
 def save_to_file(

{lionagi-0.5.4/lionagi/libs/parse/flatten → lionagi-0.6.0/lionagi/libs/nested}/flatten.py

@@ -1,3 +1,7 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 from collections import deque
 from collections.abc import Mapping, Sequence
 from typing import Any, Literal, TypeVar, overload

{lionagi-0.5.4/lionagi/libs/parse → lionagi-0.6.0/lionagi/libs}/nested/nfilter.py

@@ -1,3 +1,7 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 from collections.abc import Callable
 from typing import Any
 

{lionagi-0.5.4/lionagi/libs/parse → lionagi-0.6.0/lionagi/libs}/nested/nget.py

@@ -1,6 +1,11 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 from typing import Any
 
-from ...constants import UNDEFINED
+from lionagi.utils import UNDEFINED
+
 from .utils import get_target_container
 
 

{lionagi-0.5.4/lionagi/libs/parse → lionagi-0.6.0/lionagi/libs}/nested/ninsert.py

@@ -1,6 +1,10 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 from typing import Any
 
-from ..type_convert.to_list import to_list
+from lionagi.utils import to_list
 
 
 def ninsert(

{lionagi-0.5.4/lionagi/libs/parse → lionagi-0.6.0/lionagi/libs}/nested/nmerge.py

@@ -1,3 +1,7 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 from collections import defaultdict
 from collections.abc import Callable, Sequence
 from itertools import chain

{lionagi-0.5.4/lionagi/libs/parse → lionagi-0.6.0/lionagi/libs}/nested/npop.py

@@ -1,8 +1,11 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 from collections.abc import Sequence
 from typing import Any
 
-from ...constants import UNDEFINED
-from ..type_convert.to_list import to_list
+from lionagi.utils import UNDEFINED, to_list
 
 
 def npop(

{lionagi-0.5.4/lionagi/libs/parse → lionagi-0.6.0/lionagi/libs}/nested/nset.py

@@ -1,7 +1,12 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 from collections.abc import Sequence
 from typing import Any
 
-from ..type_convert.to_list import to_list
+from lionagi.utils import to_list
+
 from .utils import ensure_list_index
 
 

{lionagi-0.5.4/lionagi/libs/parse/flatten → lionagi-0.6.0/lionagi/libs/nested}/unflatten.py

@@ -1,3 +1,7 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 from typing import Any
 
 

{lionagi-0.5.4/lionagi/libs/parse → lionagi-0.6.0/lionagi/libs}/nested/utils.py

@@ -1,6 +1,10 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 from typing import Any
 
-from ...constants import UNDEFINED
+from lionagi.utils import UNDEFINED
 
 
 def is_homogeneous(

{lionagi-0.5.4 → lionagi-0.6.0}/lionagi/libs/package/imports.py

@@ -1,10 +1,14 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 import importlib.metadata
 import importlib.util
 import logging
 import subprocess
 from typing import Any
 
-from ..utils import run_pip_command
+from lionagi.utils import run_package_manager_command
 
 
 def check_import(
@@ -135,7 +139,7 @@ def install_import(
     except ImportError:
         logging.info(f"Installing {pip_name}...")
         try:
-            run_pip_command(["install", pip_name])
+            run_package_manager_command(["install", pip_name])
             return import_module(
                 package_name=package_name,
                 module_name=module_name,

{lionagi-0.5.4 → lionagi-0.6.0}/lionagi/libs/package/management.py

@@ -1,8 +1,12 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 import importlib.metadata
 import logging
 import subprocess
 
-from ..utils import run_pip_command
+from lionagi.utils import run_package_manager_command
 
 
 def list_installed_packages() -> list[str]:
@@ -33,7 +37,7 @@ def uninstall_package(package_name: str) -> None:
         subprocess.CalledProcessError: If the uninstallation fails.
     """
     try:
-        run_pip_command(["uninstall", package_name, "-y"])
+        run_package_manager_command(["uninstall", package_name, "-y"])
         logging.info(f"Successfully uninstalled {package_name}.")
     except subprocess.CalledProcessError as e:
         logging.error(f"Failed to uninstall {package_name}. Error: {e}")
@@ -51,7 +55,7 @@ def update_package(package_name: str) -> None:
         subprocess.CalledProcessError: If the update fails.
     """
     try:
-        run_pip_command(["install", "--upgrade", package_name])
+        run_package_manager_command(["install", "--upgrade", package_name])
         logging.info(f"Successfully updated {package_name}.")
     except subprocess.CalledProcessError as e:
         logging.error(f"Failed to update {package_name}. Error: {e}")

{lionagi-0.5.4 → lionagi-0.6.0}/lionagi/libs/package/params.py

@@ -1,3 +1,7 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 from pydantic import BaseModel, Field
 
 

{lionagi-0.5.4 → lionagi-0.6.0}/lionagi/libs/package/system.py

@@ -1,3 +1,7 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 import platform