lionagi 0.6.0__py3-none-any.whl → 0.6.1__py3-none-any.whl

lionagi/__init__.py

@@ -2,6 +2,7 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
+from lionagi.operations import types as op
 from lionagi.operatives import types as ops_types
 from lionagi.protocols import types as types
 from lionagi.service.imodel import iModel
@@ -17,5 +18,6 @@ __all__ = (
     "LiteiModel",
     "types",
     "ops_types",
+    "op",
     "__version__",
 )

lionagi/operations/plan/plan.py

@@ -27,9 +27,9 @@ from .prompt import EXPANSION_PROMPT, PLAN_PROMPT
 class PlanOperation(BaseModel):
     """
     Stores all relevant outcomes for a multi-step Plan:
-      * initial: The result of the initial plan prompt
-      * plan: A list of plan steps (Instruct objects) generated from the initial planning
-      * execute: Any responses from executing those plan steps
+        * initial: The result of the initial plan prompt
+        * plan: A list of plan steps (Instruct objects) generated from the initial planning
+        * execute: Any responses from executing those plan steps
     """
 
     initial: Any

lionagi/operations/select/select.py

@@ -8,8 +8,8 @@ from typing import Any
 
 from pydantic import BaseModel, Field
 
-from lionagi import Branch
 from lionagi.operatives.types import Instruct
+from lionagi.session.branch import Branch
 
 from .prompt import PROMPT
 from .utils import parse_selection, parse_to_representation

lionagi/operations/strategies/params.py

@@ -10,9 +10,25 @@ from lionagi.operatives.instruct.instruct import (
     LIST_INSTRUCT_FIELD_MODEL,
     Instruct,
 )
-from lionagi.protocols.types import FieldModel, SchemaModel
+from lionagi.operatives.types import FieldModel, SchemaModel
 from lionagi.session.session import Branch, Session
-from lionagi.utils import RCallParams
+
+
+class RCallParams(SchemaModel):
+    """Parameters for remote function calls."""
+
+    timeout: float = Field(
+        default=60, description="Timeout for remote function call"
+    )
+    max_retries: int = Field(
+        default=3, description="Maximum number of retries"
+    )
+    retry_delay: float = Field(
+        default=0.5, description="Delay between retries"
+    )
+    retry_backoff: float = Field(
+        default=2, description="Backoff factor for retry delay"
+    )
 
 
 class StrategyParams(SchemaModel):

lionagi/operations/types.py

@@ -2,6 +2,12 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-from .brainstorm import *
-from .plan import *
-from .select import *
+from .brainstorm.brainstorm import brainstorm
+from .plan.plan import plan
+from .select.select import select
+
+__all__ = (
+    "brainstorm",
+    "plan",
+    "select",
+)

lionagi/operatives/action/function_calling.py

@@ -7,7 +7,7 @@ from typing import Any, Self
 
 from pydantic import Field, model_validator
 
-from lionagi.protocols.types import Event, EventStatus
+from lionagi.protocols.generic.event import Event, EventStatus
 from lionagi.utils import is_coro_func
 
 from .tool import Tool

lionagi/operatives/action/manager.py

@@ -4,13 +4,10 @@
 
 from typing import Any
 
-from lionagi.protocols.types import (
-    ActionRequest,
-    EventStatus,
-    Execution,
-    Log,
-    Manager,
-)
+from lionagi.protocols._concepts import Manager
+from lionagi.protocols.generic.event import EventStatus, Execution
+from lionagi.protocols.generic.log import Log
+from lionagi.protocols.messages.action_request import ActionRequest
 from lionagi.utils import to_list
 
 from .function_calling import FunctionCalling

lionagi/operatives/action/tool.py

@@ -10,7 +10,7 @@ from pydantic import Field, field_validator, model_validator
 
 from lionagi.libs.schema.function_to_schema import function_to_schema
 from lionagi.libs.validate.common_field_validators import validate_callable
-from lionagi.protocols.types import Element
+from lionagi.protocols.generic.element import Element
 
 __all__ = (
     "Tool",

lionagi/protocols/adapters/adapter.py

@@ -0,0 +1,95 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
+import logging
+from typing import Any, Protocol, TypeVar, runtime_checkable
+
+from typing_extensions import get_protocol_members
+
+T = TypeVar("T")
+
+__all__ = (
+    "Adapter",
+    "ADAPTER_MEMBERS",
+    "AdapterRegistry",
+)
+
+
+@runtime_checkable
+class Adapter(Protocol):
+
+    obj_key: str
+
+    @classmethod
+    def from_obj(
+        cls,
+        subj_cls: type[T],
+        obj: Any,
+        /,
+        *,
+        many: bool,
+        **kwargs,
+    ) -> dict | list[dict]: ...
+
+    @classmethod
+    def to_obj(
+        cls,
+        subj: T,
+        /,
+        *,
+        many: bool,
+        **kwargs,
+    ) -> Any: ...
+
+
+ADAPTER_MEMBERS = get_protocol_members(Adapter)  # duck typing
+
+
+class AdapterRegistry:
+
+    _adapters: dict[str, Adapter] = {}
+
+    @classmethod
+    def list_adapters(cls) -> list[tuple[str | type, ...]]:
+        return list(cls._adapters.keys())
+
+    @classmethod
+    def register(cls, adapter: type[Adapter]) -> None:
+        for member in ADAPTER_MEMBERS:
+            if not hasattr(adapter, member):
+                _str = getattr(adapter, "obj_key", None) or repr(adapter)
+                _str = _str[:50] if len(_str) > 50 else _str
+                raise AttributeError(
+                    f"Adapter {_str} missing required methods."
+                )
+
+        if isinstance(adapter, type):
+            cls._adapters[adapter.obj_key] = adapter()
+        else:
+            cls._adapters[adapter.obj_key] = adapter
+
+    @classmethod
+    def get(cls, obj_key: type | str) -> Adapter:
+        try:
+            return cls._adapters[obj_key]
+        except Exception as e:
+            logging.error(f"Error getting adapter for {obj_key}. Error: {e}")
+
+    @classmethod
+    def adapt_from(
+        cls, subj_cls: type[T], obj: Any, obj_key: type | str, **kwargs
+    ) -> dict | list[dict]:
+        try:
+            return cls.get(obj_key).from_obj(subj_cls, obj, **kwargs)
+        except Exception as e:
+            logging.error(f"Error adapting data from {obj_key}. Error: {e}")
+            raise e
+
+    @classmethod
+    def adapt_to(cls, subj: T, obj_key: type | str, **kwargs) -> Any:
+        try:
+            return cls.get(obj_key).to_obj(subj, **kwargs)
+        except Exception as e:
+            logging.error(f"Error adapting data to {obj_key}. Error: {e}")
+            raise e

lionagi/protocols/adapters/json_adapter.py

@@ -0,0 +1,101 @@
+import json
+import logging
+from pathlib import Path
+
+from lionagi.protocols._concepts import Collective
+
+from .adapter import Adapter, T
+
+
+class JsonAdapter(Adapter):
+
+    obj_key = "json"
+
+    @classmethod
+    def from_obj(
+        cls,
+        subj_cls: type[T],
+        obj: str,
+        /,
+        *,
+        many: bool = False,
+        **kwargs,
+    ) -> dict | list[dict]:
+        """
+        kwargs for json.loads(s, **kwargs)
+        """
+        result = json.loads(obj, **kwargs)
+        if many:
+            return result if isinstance(result, list) else [result]
+        return (
+            result[0]
+            if isinstance(result, list) and len(result) > 0
+            else result
+        )
+
+    @classmethod
+    def to_obj(
+        cls,
+        subj: T,
+        *,
+        many: bool = False,
+        **kwargs,
+    ):
+        """
+        kwargs for json.dumps(obj, **kwargs)
+        """
+        if many:
+            if isinstance(subj, Collective):
+                return json.dumps([i.to_dict() for i in subj], **kwargs)
+            return json.dumps([subj.to_dict()], **kwargs)
+        return json.dumps(subj.to_dict(), **kwargs)
+
+
+class JsonFileAdapter(Adapter):
+
+    obj_key = ".json"
+
+    @classmethod
+    def from_obj(
+        cls,
+        subj_cls: type[T],
+        obj: str | Path,
+        /,
+        *,
+        many: bool = False,
+        **kwargs,
+    ) -> dict | list[dict]:
+        """
+        kwargs for json.load(fp, **kwargs)
+        """
+        with open(obj) as f:
+            result = json.load(f, **kwargs)
+        if many:
+            return result if isinstance(result, list) else [result]
+        return (
+            result[0]
+            if isinstance(result, list) and len(result) > 0
+            else result
+        )
+
+    @classmethod
+    def to_obj(
+        cls,
+        subj: T,
+        /,
+        *,
+        fp: str | Path,
+        many: bool = False,
+        **kwargs,
+    ):
+        """
+        kwargs for json.dump(obj, fp, **kwargs)
+        """
+        if many:
+            if isinstance(subj, Collective):
+                json.dump([i.to_dict() for i in subj], fp=fp, **kwargs)
+                return
+            json.dump([subj.to_dict()], fp=fp, **kwargs)
+            return
+        json.dump(subj.to_dict(), fp=fp, **kwargs)
+        logging.info(f"Successfully saved data to {fp}")

lionagi/protocols/adapters/pandas_/csv_adapter.py

@@ -0,0 +1,50 @@
+import logging
+from pathlib import Path
+
+import pandas as pd
+
+from lionagi.protocols._concepts import Collective
+
+from ..adapter import Adapter, T
+
+
+class CSVFileAdapter(Adapter):
+    obj_key = ".csv"
+
+    @classmethod
+    def from_obj(
+        cls,
+        subj_cls: type[T],
+        obj: str | Path,
+        /,
+        *,
+        many: bool = False,
+        **kwargs,
+    ) -> list[dict]:
+        """kwargs for pd.read_csv"""
+        df: pd.DataFrame = pd.read_csv(obj, **kwargs)
+        dicts_ = df.to_dict(orient="records")
+        return dicts_[0] if not many else dicts_
+
+    @classmethod
+    def to_obj(
+        cls,
+        subj: T,
+        /,
+        *,
+        fp: str | Path,
+        many: bool = False,
+        **kwargs,
+    ):
+        """kwargs for pd.DataFrame.to_csv"""
+        kwargs["index"] = False
+        if many:
+            if isinstance(subj, Collective):
+                pd.DataFrame([i.to_dict() for i in subj]).to_csv(fp, **kwargs)
+                logging.info(f"Successfully saved data to {fp}")
+                return
+            pd.DataFrame([subj.to_dict()]).to_csv(fp, **kwargs)
+            logging.info(f"Successfully saved data to {fp}")
+            return
+        pd.DataFrame([subj.to_dict()]).to_csv(fp, **kwargs)
+        logging.info(f"Successfully saved data to {fp}")

lionagi/protocols/adapters/pandas_/excel_adapter.py

@@ -0,0 +1,52 @@
+import logging
+from pathlib import Path
+
+import pandas as pd
+
+from lionagi.protocols._concepts import Collective
+
+from ..adapter import Adapter, T
+
+
+class ExcelFileAdapter(Adapter):
+    obj_key = ".xlsx"
+
+    @classmethod
+    def from_obj(
+        cls,
+        subj_cls: type[T],
+        obj: str | Path,
+        /,
+        *,
+        many: bool = False,
+        **kwargs,
+    ) -> list[dict]:
+        """kwargs for pd.read_csv"""
+        df: pd.DataFrame = pd.read_excel(obj, **kwargs)
+        dicts_ = df.to_dict(orient="records")
+        return dicts_[0] if not many else dicts_
+
+    @classmethod
+    def to_obj(
+        cls,
+        subj: T,
+        /,
+        *,
+        fp: str | Path,
+        many: bool = False,
+        **kwargs,
+    ):
+        """kwargs for pd.DataFrame.to_csv"""
+        kwargs["index"] = False
+        if many:
+            if isinstance(subj, Collective):
+                pd.DataFrame([i.to_dict() for i in subj]).to_excel(
+                    fp, **kwargs
+                )
+                logging.info(f"Successfully saved data to {fp}")
+                return
+            pd.DataFrame([subj.to_dict()]).to_excel(fp, **kwargs)
+            logging.info(f"Successfully saved data to {fp}")
+            return
+        pd.DataFrame([subj.to_dict()]).to_excel(fp, **kwargs)
+        logging.info(f"Successfully saved data to {fp}")

lionagi/protocols/adapters/pandas_/pd_dataframe_adapter.py

@@ -0,0 +1,31 @@
+from datetime import datetime
+
+import pandas as pd
+
+from ..adapter import Adapter, T
+
+
+class PandasDataFrameAdapter(Adapter):
+
+    obj_key = "pd_dataframe"
+    alias = ("pandas_dataframe", "pd.DataFrame", "pd_dataframe")
+
+    @classmethod
+    def from_obj(
+        cls, subj_cls: type[T], obj: pd.DataFrame, /, **kwargs
+    ) -> list[dict]:
+        """kwargs for pd.DataFrame.to_dict"""
+        return obj.to_dict(orient="records", **kwargs)
+
+    @classmethod
+    def to_obj(cls, subj: list[T], /, **kwargs) -> pd.DataFrame:
+        """kwargs for pd.DataFrame"""
+        out_ = []
+        for i in subj:
+            _dict = i.to_dict()
+            _dict["created_at"] = datetime.fromtimestamp(_dict["created_at"])
+            out_.append(_dict)
+        df = pd.DataFrame(out_, **kwargs)
+        if "created_at" in df.columns:
+            df["created_at"] = pd.to_datetime(df["created_at"])
+        return df

lionagi/protocols/adapters/pandas_/pd_series_adapter.py

@@ -0,0 +1,17 @@
+import pandas as pd
+
+from ..adapter import Adapter, T
+
+
+class PandasSeriesAdapter(Adapter):
+
+    obj_key = "pd_series"
+    alias = ("pandas_series", "pd.series", "pd_series")
+
+    @classmethod
+    def from_obj(cls, subj_cls: type[T], obj: pd.Series, /, **kwargs) -> dict:
+        return obj.to_dict(**kwargs)
+
+    @classmethod
+    def to_obj(cls, subj: T, /, **kwargs) -> pd.Series:
+        return pd.Series(subj.to_dict(), **kwargs)

lionagi/protocols/adapters/types.py

@@ -0,0 +1,18 @@
+from .adapter import ADAPTER_MEMBERS, Adapter, AdapterRegistry
+from .json_adapter import JsonAdapter, JsonFileAdapter
+from .pandas_.csv_adapter import CSVFileAdapter
+from .pandas_.excel_adapter import ExcelFileAdapter
+from .pandas_.pd_dataframe_adapter import PandasDataFrameAdapter
+from .pandas_.pd_series_adapter import PandasSeriesAdapter
+
+__all__ = (
+    "Adapter",
+    "AdapterRegistry",
+    "ADAPTER_MEMBERS",
+    "JsonAdapter",
+    "JsonFileAdapter",
+    "CSVFileAdapter",
+    "PandasSeriesAdapter",
+    "PandasDataFrameAdapter",
+    "ExcelFileAdapter",
+)

lionagi/protocols/generic/pile.py

@@ -27,8 +27,12 @@ from typing_extensions import override
 from lionagi._errors import ItemExistsError, ItemNotFoundError
 from lionagi.utils import UNDEFINED, is_same_dtype, to_list
 
-from .._adapter import Adapter, AdapterRegistry, PileAdapterRegistry
 from .._concepts import Observable
+from ..adapters.adapter import Adapter, AdapterRegistry
+from ..adapters.json_adapter import JsonAdapter, JsonFileAdapter
+from ..adapters.pandas_.csv_adapter import CSVFileAdapter
+from ..adapters.pandas_.excel_adapter import ExcelFileAdapter
+from ..adapters.pandas_.pd_dataframe_adapter import PandasDataFrameAdapter
 from .element import ID, Collective, E, Element, IDType, validate_order
 from .progression import Progression
 
@@ -36,6 +40,23 @@ D = TypeVar("D")
 T = TypeVar("T", bound=E)
 
 
+PILE_DEFAULT_ADAPTERS = (
+    JsonAdapter,
+    JsonFileAdapter,
+    CSVFileAdapter,
+    ExcelFileAdapter,
+    PandasDataFrameAdapter,
+)
+
+
+class PileAdapterRegistry(AdapterRegistry):
+    pass
+
+
+for i in PILE_DEFAULT_ADAPTERS:
+    PileAdapterRegistry.register(i)
+
+
 __all__ = (
     "Pile",
     "pile",

lionagi/protocols/graph/node.py

@@ -9,10 +9,26 @@ from pydantic import field_validator
 
 from lionagi._class_registry import LION_CLASS_REGISTRY
 
-from .._adapter import AdapterRegistry, NodeAdapterRegistry
 from .._concepts import Relational
+from ..adapters.adapter import AdapterRegistry
+from ..adapters.json_adapter import JsonAdapter, JsonFileAdapter
+from ..adapters.pandas_.pd_series_adapter import PandasSeriesAdapter
 from ..generic.element import Element
 
+NODE_DEFAULT_ADAPTERS = (
+    JsonAdapter,
+    JsonFileAdapter,
+    PandasSeriesAdapter,
+)
+
+
+class NodeAdapterRegistry(AdapterRegistry):
+    pass
+
+
+for i in NODE_DEFAULT_ADAPTERS:
+    NodeAdapterRegistry.register(i)
+
 __all__ = ("Node",)
 
 

lionagi/protocols/types.py

@@ -2,7 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-from ._adapter import Adapter, AdapterRegistry
 from ._concepts import (
     Collective,
     Communicatable,
@@ -14,6 +13,7 @@ from ._concepts import (
     Relational,
     Sendable,
 )
+from .adapters.adapter import Adapter, AdapterRegistry
 from .generic.element import ID, Element, IDError, IDType, validate_order
 from .generic.event import Event, EventStatus, Execution
 from .generic.log import Log, LogManager, LogManagerConfig
@@ -89,8 +89,8 @@ __all__ = (
     "MessageField",
     "MESSAGE_FIELDS",
     "validate_sender_recipient",
-    "Adapter",
-    "AdapterRegistry",
     "MessageManager",
     "to_list_type",
+    "Adapter",
+    "AdapterRegistry",
 )

lionagi/service/__init__.py

@@ -1,14 +1 @@
-# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
-#
-# SPDX-License-Identifier: Apache-2.0
-
-from .endpoints.base import APICalling, EndPoint
-from .imodel import iModel
-from .manager import iModelManager
-
-__all__ = (
-    "iModel",
-    "iModelManager",
-    "EndPoint",
-    "APICalling",
-)
+from .types import *

lionagi/service/endpoints/base.py

@@ -12,7 +12,7 @@ from aiocache import cached
 from pydantic import BaseModel, ConfigDict, Field
 
 from lionagi._errors import ExecutionError, RateLimitError
-from lionagi.protocols.types import Event, EventStatus
+from lionagi.protocols.generic.event import Event, EventStatus
 from lionagi.settings import Settings
 
 from .token_calculator import TokenCalculator

lionagi/service/endpoints/rate_limited_processor.py

@@ -8,7 +8,8 @@ from typing import Self
 
 from typing_extensions import override
 
-from ...protocols.types import Executor, Processor
+from lionagi.protocols.generic.processor import Executor, Processor
+
 from .base import APICalling
 
 __all__ = (

lionagi/service/manager.py

@@ -2,9 +2,9 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
+from lionagi.protocols._concepts import Manager
 from lionagi.utils import is_same_dtype
 
-from ..protocols._concepts import Manager
 from .endpoints.chat_completion import ChatCompletionEndPoint
 from .imodel import iModel
 

lionagi/service/types.py

@@ -0,0 +1,18 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from .endpoints.base import APICalling, EndPoint
+from .endpoints.rate_limited_processor import RateLimitedAPIExecutor
+from .endpoints.token_calculator import TokenCalculator
+from .imodel import iModel
+from .manager import iModelManager
+
+__all__ = (
+    "APICalling",
+    "EndPoint",
+    "RateLimitedAPIExecutor",
+    "TokenCalculator",
+    "iModel",
+    "iModelManager",
+)

lionagi/version.py

@@ -1 +1 @@
-__version__ = "0.6.0"
+__version__ = "0.6.1"

{lionagi-0.6.0.dist-info → lionagi-0.6.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lionagi
-Version: 0.6.0
+Version: 0.6.1
 Summary: An AGentic Intelligence Operating System.
 Author-email: HaiyangLi <quantocean.li@gmail.com>
 License:            Apache License
@@ -217,13 +217,13 @@ Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
-Requires-Python: >=3.11
+Requires-Python: >=3.10
 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: pillow>=10.0.0
 Requires-Dist: pydantic>=2.0.0
 Requires-Dist: python-dotenv>=1.0.1
 Description-Content-Type: text/markdown
@@ -232,7 +232,7 @@ Description-Content-Type: text/markdown
 ![PyPI - Downloads](https://img.shields.io/pypi/dm/lionagi?color=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)
+[Documentation](https://lion-agi.github.io/lionagi/) | [Discord](https://discord.gg/aqSJ2v46vu) | [PyPI](https://pypi.org/project/lionagi/) | [Roadmap](https://trello.com/b/3seomsrI/lionagi)
 
 # LION Framework
 ### Language InterOperable Network - The Future of Controlled AI Operations

{lionagi-0.6.0.dist-info → lionagi-0.6.1.dist-info}/RECORD

@@ -1,9 +1,9 @@
-lionagi/__init__.py,sha256=l4FOohHV6hKyCsbrbUabbXbu-yHSGNO_0wmRSPT_QT4,488
+lionagi/__init__.py,sha256=sBS47lQGuXNAdfSoVLW8szKbCfDWrfAceMMUVYNK3IU,541
 lionagi/_class_registry.py,sha256=dutMsw-FQNqVV5gGH-NEIv90uBkSr8fERJ_x3swbb-s,3112
 lionagi/_errors.py,sha256=wNKdnVQvE_CHEstK7htrrj334RA_vbGcIds-3pUiRkc,455
 lionagi/settings.py,sha256=k9zRJXv57TveyfHO3Vr9VGiKrSwlRUUVKt5zf6v9RU4,1627
 lionagi/utils.py,sha256=hBlGmi8Px_ng3v4_D_k8xNpW3X93u01930lkZIHUQ-Y,73019
-lionagi/version.py,sha256=cID1jLnC_vj48GgMN6Yb1FA3JsQ95zNmCHmRYE8TFhY,22
+lionagi/version.py,sha256=baAcEjLSYFIeNZF51tOMmA_zAMhN8HvKael-UU-Ruec,22
 lionagi/libs/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/libs/parse.py,sha256=tpEbmIRGuHhLCJlUlm6fjmqm_Z6XJLAXGNFHNuk422I,1011
 lionagi/libs/compress/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
@@ -43,24 +43,24 @@ lionagi/libs/validate/fuzzy_validate_mapping.py,sha256=SQqyxgrAgJ5zBKxIAnulWsZXb
 lionagi/libs/validate/string_similarity.py,sha256=7x8D4LZCMNJGz2ZCeKmct7Nc4i8m_ORoHIeGvkeMZOA,8733
 lionagi/libs/validate/validate_boolean.py,sha256=h3d7Dn7asJokBozWaKxaV_3Y6vUWBc0-zfNJjTQ9Bo8,3614
 lionagi/operations/__init__.py,sha256=_TvmWRotAQILyE3JWls7uY1luV7GAwwddtVdKnvjgIk,280
-lionagi/operations/types.py,sha256=_1HkP-xiWxh7l5dmqCnC_H40Unx_Nj11O1n9a1_WD80,181
+lionagi/operations/types.py,sha256=_TvmWRotAQILyE3JWls7uY1luV7GAwwddtVdKnvjgIk,280
 lionagi/operations/utils.py,sha256=Twy6L_UFt9JqJFRYuKKTKVZIXsePidNl5ipcYcCbesI,1220
 lionagi/operations/brainstorm/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/operations/brainstorm/brainstorm.py,sha256=88Cq2IBrUs5du9q_8rf_VmhLYBfGKJv4rCeDTcJ07_g,17196
 lionagi/operations/brainstorm/prompt.py,sha256=f-Eh6pO606dT2TrX9BFv_einRDpYwFi6Gep9Strd1cM,610
 lionagi/operations/plan/__init__.py,sha256=AFkAmOJBTqPlYuqFRRn7rCvIw3CGh9XXH_22cNWbfig,156
-lionagi/operations/plan/plan.py,sha256=4DFBY6XSbTle96qCfx2QyN3ekTyZjrMar14eAl1zlAg,15327
+lionagi/operations/plan/plan.py,sha256=Uxysob3evbjhXWS_wseO3Mb_A90p1wk3ralJ17rHFCk,15333
 lionagi/operations/plan/prompt.py,sha256=ig4JjJR5gV-lXPRVbiaJuL9qeoahM_afYvWphpY1lWA,993
 lionagi/operations/select/__init__.py,sha256=an28I5IbKj0waICk5SjYnz3WmkQLpIARO0yNUu9VWNE,162
 lionagi/operations/select/prompt.py,sha256=2OJQE8EKPj9nubP_drp6BYyF7J8RiDs1BEkv0D-HLn8,299
-lionagi/operations/select/select.py,sha256=NYHfrLUOLklVuvKCObHlmSj42R7ye8ftNS70vPPnOrQ,3151
+lionagi/operations/select/select.py,sha256=B_XZcdmW9mmVGFZGnfCn89TxZvAMlS27YcD9RQ96f40,3166
 lionagi/operations/select/utils.py,sha256=hYOSGx7vMvjGHibcFOUuCfTXFO2IzzIUwQy82PQtluQ,3251
 lionagi/operations/strategies/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/operations/strategies/base.py,sha256=cfZXUZYPypW-hFZJj7HDtTPc-x99XB6dO_S5os1srTk,1820
 lionagi/operations/strategies/concurrent.py,sha256=bLLL90LYywkxGlMY8XXGsZ1GRZrtB_8lsfkqspLgQ6o,2721
 lionagi/operations/strategies/concurrent_chunk.py,sha256=9qwYoqO0LO1BMIyxU-tc_9ppBK0J94cnWs3e7wx7PR0,1678
 lionagi/operations/strategies/concurrent_sequential_chunk.py,sha256=5Zw9fO1YWSj81w6IjOdEPzHE-4TxIdui-ljjrAsbEEI,3658
-lionagi/operations/strategies/params.py,sha256=BMGclThbRYnUmmcFSh9JAng12dykBl_Ud_5VHLp18g0,4966
+lionagi/operations/strategies/params.py,sha256=XNCD8sQJCm7xk0yGFKSQ14FuUNV1f8wLfNNJg_NYsn8,5407
 lionagi/operations/strategies/sequential.py,sha256=nSMtYy10P4W8U2SoARfKCBDN6y8_WoubKjoORUnewjY,949
 lionagi/operations/strategies/sequential_chunk.py,sha256=bs_0zZJjJpYdmEEWx3mwh3it5ErQYfZr-ImahyKEy8Y,3161
 lionagi/operations/strategies/sequential_concurrent_chunk.py,sha256=GJurnTuu6U7FOPR9mZxkjoJMIiQbgvRGt5iZZtmB2mY,3559
@@ -71,10 +71,10 @@ lionagi/operatives/operative.py,sha256=WZeg1XHS8iuzs4I3-va9FS9tOWu0XZ7xnOVzHthvE
 lionagi/operatives/step.py,sha256=DevwisZc2q88ynUiiSu7VBEY2A_G4Q5iRLrVgVLHNJU,9843
 lionagi/operatives/types.py,sha256=8krpGIeJL2GkO580ePOuAZfGc7vUVPIanemoA77tbVY,1697
 lionagi/operatives/action/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
-lionagi/operatives/action/function_calling.py,sha256=d07l39EqIsLOvWUcxWo1UjWu8VDfbjhGKJVA_In8tjs,4594
-lionagi/operatives/action/manager.py,sha256=XFFl93xaeErX4lZz6060nscsG4OnYPz7eM4WoJ14n4g,7813
+lionagi/operatives/action/function_calling.py,sha256=VHUq8qebCAAdaWcPJQCZNI-P6zC_P1kp9o-d7nhZf_0,4602
+lionagi/operatives/action/manager.py,sha256=H50L9fv6kpQ87vW4nEIfP1JHZ5gX4fQs0uPDeRpEYEw,7929
 lionagi/operatives/action/request_response_model.py,sha256=7X_ZkcgmQ0D77qyP8_39RdEAKASCjEPWH4ey83jIUKc,2326
-lionagi/operatives/action/tool.py,sha256=Wxmh5v0pH4qjYGXn7w-RCGh756ARhgGfq7mjcn2EIic,4267
+lionagi/operatives/action/tool.py,sha256=fUwlzmi6kuuwHU6xA8Soo_bWOgD-I7TBLtJ-WJ1OBLk,4277
 lionagi/operatives/action/utils.py,sha256=vUe7Aysuzbg16rAfe2Ttp5QUz5_L6mMedBVAWzGAHwk,4330
 lionagi/operatives/forms/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/operatives/forms/base.py,sha256=ALPREgOkRcY2wtRG0JpTXztlohkRjyCiwoleQfGCpSg,7207
@@ -95,20 +95,28 @@ lionagi/operatives/models/note.py,sha256=8TS0Zi0eSo1fmVoqBMWB6ChSrXKb36ozAM70o7P
 lionagi/operatives/models/operable_model.py,sha256=mnTIAXJHJecXjqKNZWSch5w6x3U3Fl_9mUK4RudnM64,15573
 lionagi/operatives/models/schema_model.py,sha256=-BeCwW_1Rle--w-f7MajbOYH7t_8SPWw0_qK0fpTsRg,666
 lionagi/protocols/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
-lionagi/protocols/_adapter.py,sha256=UPz6LlEb1fAc53Cq8QJLpVngFlPqi2JnMUqqi2FxkPM,5875
 lionagi/protocols/_concepts.py,sha256=aIV3cIuKxRcEFdROemlFR6mMAwqiYv-rjkZQ4rHBfXo,1554
-lionagi/protocols/types.py,sha256=of761-J1RDZEZXIPxEXVz2PRg_KuajfMrjzhZGwk2o4,2057
+lionagi/protocols/types.py,sha256=M-cWgRGl8vaU776MuX-QSQVmaI1rbXmzyW9tfPHzGYc,2065
+lionagi/protocols/adapters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/protocols/adapters/adapter.py,sha256=CyAZ4IR6S4FSmKMk_3YzyE1Ny8TJjeYc4h1Z710WL6M,2417
+lionagi/protocols/adapters/json_adapter.py,sha256=Jl0pyu2r-vZ5PVfqBzp3eAl9no9MQwg2ln5ty3UDI4s,2337
+lionagi/protocols/adapters/types.py,sha256=qYCbxjWNLVIUOosC3N8oJHP8tDJ9kjsnaYuUh_h-70c,563
+lionagi/protocols/adapters/pandas_/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lionagi/protocols/adapters/pandas_/csv_adapter.py,sha256=iKBX1XXRh85O4595pvxKwxv5A-80OYBhzQoaAeF6MgI,1307
+lionagi/protocols/adapters/pandas_/excel_adapter.py,sha256=XoPD16Kus6lPNUEsy8JZDi5JMlwSsB6VVOQU3QIOoIs,1356
+lionagi/protocols/adapters/pandas_/pd_dataframe_adapter.py,sha256=3JQFTN7xTovPet98cOxx2mCq1O7OXZnXZEZv6m06dtc,908
+lionagi/protocols/adapters/pandas_/pd_series_adapter.py,sha256=f5mj4HSL6G1R02RaRRxQ06bfxble19da8BtH3WD889M,434
 lionagi/protocols/generic/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/protocols/generic/element.py,sha256=Pi-YHinDDab7sIYNo4Caj-gJGBidcIAEN1Bpl1D100Q,14161
 lionagi/protocols/generic/event.py,sha256=SjR9N4Egr5_oqOBSKVsxQjsn6MlqNcwf48bee-qIT6Y,4879
 lionagi/protocols/generic/log.py,sha256=xi8dRKwxtxVYU8T_E4wYJE4lCQzkERgAUARcAN7ZngI,7441
-lionagi/protocols/generic/pile.py,sha256=TpPZVqlc09-mTdMuEUKfG8znqO9U7ahWi-wsP6tMP-s,30999
+lionagi/protocols/generic/pile.py,sha256=kXzN9ZHmmR__tOItwvx9xcB9IDxJYYHIkHOtDX0wsFE,31509
 lionagi/protocols/generic/processor.py,sha256=4Gkie1DxE0U-uZAdNBTuTibUlyeEGm_OyVlMXilCEm8,10115
 lionagi/protocols/generic/progression.py,sha256=3PjIBlPoj7jahy75ERbo9vHKVNU7fFl4be5ETNzphJU,15160
 lionagi/protocols/graph/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/protocols/graph/edge.py,sha256=LOESNQc3aVNKXZrr5ZrRCK5biTWkyxGtTQ7GxJFyCx8,5221
 lionagi/protocols/graph/graph.py,sha256=bRlZot_6eN4K3CYxgFnfehnQHDGJTQ3OPqITaz8ETiU,10055
-lionagi/protocols/graph/node.py,sha256=s9TmFIXY2yL9Vc8nlzTfaM9ZXP8qboqcF5pvfgG0HA0,3435
+lionagi/protocols/graph/node.py,sha256=5VStUHSV0Qjv6DtwJPCzod2stveixnsErJR09uLLYhc,3773
 lionagi/protocols/mail/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/protocols/mail/exchange.py,sha256=bT-JUU9CXDdUn6GlAjB1Pc4KD7J8Ncm7s-xbmFKI45g,4066
 lionagi/protocols/mail/mail.py,sha256=4O4R0ak7dFar80wjj1DK64SPgRhYAR7qKjv_tSDbkXA,625
@@ -131,14 +139,15 @@ lionagi/protocols/messages/templates/assistant_response.jinja2,sha256=oKOX4myBy7
 lionagi/protocols/messages/templates/instruction_message.jinja2,sha256=L-ptw5OHxdL1KVKVhLOn0KAFw6IEFI9QHZxiFuovzhk,1382
 lionagi/protocols/messages/templates/system_message.jinja2,sha256=JRKJ0aFpYfaXSFouKc_N4unZ35C3yZTOWhIrIdCB5qk,215
 lionagi/protocols/messages/templates/tool_schemas.jinja2,sha256=ozIaSDCRjIAhLyA8VM6S-YqS0w2NcctALSwx4LjDwII,126
-lionagi/service/__init__.py,sha256=1v6f3RxLaZmSRj3uCEahSg0plWsxV8l_UisLbUOXoAQ,308
+lionagi/service/__init__.py,sha256=DMGXIqPsmut9H5GT0ZeSzQIzYzzPwI-2gLXydpbwiV8,21
 lionagi/service/imodel.py,sha256=-FudX8C_wmc7ByUv-pILrybtQd4E2gfE51C5l0IR0ho,10815
-lionagi/service/manager.py,sha256=qM_URzibfWvtqcunTJ0jwrAPR9X3R4XomSecKseAD_4,1429
+lionagi/service/manager.py,sha256=MKSYBkg23s7YhZy5GEFdnpspEnhPVfFhpkpoJe20D7k,1435
+lionagi/service/types.py,sha256=v9SAn5-GTmds4Mar13Or_VFrRHCinBK99dmeDUd-QNk,486
 lionagi/service/endpoints/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
-lionagi/service/endpoints/base.py,sha256=K6VM4wn4Lk59tzdbsC_Q263lmKMqJFJNbrJ0wKQrH1g,18267
+lionagi/service/endpoints/base.py,sha256=RbBBQIbtNtkP1LT9U3ZrhL35SfmVIpaq-AGW8d538k4,18275
 lionagi/service/endpoints/chat_completion.py,sha256=9ltSQaKPH43WdEDW32_-f5x07I9hOU8g-T_PAG-nYsQ,2529
 lionagi/service/endpoints/match_endpoint.py,sha256=n7F9NoTXfUBL29HrDcFLF5AXYR8pcx_IumQ7BiJXC-w,1740
-lionagi/service/endpoints/rate_limited_processor.py,sha256=mBSf4d58Dss45NMam6r8nvwxjUKw6CjXcq7IpUqkDV8,4663
+lionagi/service/endpoints/rate_limited_processor.py,sha256=GkWK9XR3XNwn6qKUif6HMeqMJtO1E2Hq_tqGYai_wp8,4681
 lionagi/service/endpoints/token_calculator.py,sha256=MflqImGUr_1jh465hB7cUAaIPICBkjirvre1fWGXLrA,6161
 lionagi/service/providers/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/service/providers/anthropic_/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
@@ -154,7 +163,7 @@ lionagi/service/providers/perplexity_/chat_completions.py,sha256=SsDbrtXwQsR4Yu2
 lionagi/session/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/session/branch.py,sha256=-lu63-Eua1pgo9RVoiS54G4_JtmNC__ZEaegRtKFPro,47448
 lionagi/session/session.py,sha256=A2PCG1BD1noMLtCJD3C_H7r-0GUQ_ru2szOhF1pOCtY,8976
-lionagi-0.6.0.dist-info/METADATA,sha256=THShEYoEI8tUGFhUjDZXSqF6esJrx86Alx2cCSu2-5M,22771
-lionagi-0.6.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lionagi-0.6.0.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
-lionagi-0.6.0.dist-info/RECORD,,
+lionagi-0.6.1.dist-info/METADATA,sha256=_KZh-djuuzJ-mwnUvpVUJ7TGmw_ghQnj0_QvqG8nkFY,22776
+lionagi-0.6.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lionagi-0.6.1.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
+lionagi-0.6.1.dist-info/RECORD,,

lionagi/protocols/_adapter.py

@@ -1,224 +0,0 @@
-# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
-#
-# SPDX-License-Identifier: Apache-2.0
-
-import json
-import logging
-from datetime import datetime
-from pathlib import Path
-from typing import Any, Protocol, TypeVar, runtime_checkable
-
-import pandas as pd
-from typing_extensions import get_protocol_members
-
-T = TypeVar("T")
-
-
-@runtime_checkable
-class Adapter(Protocol):
-
-    obj_key: str
-
-    @classmethod
-    def from_obj(
-        cls, subj_cls: type[T], obj: Any, /, **kwargs
-    ) -> dict | list[dict]: ...
-
-    @classmethod
-    def to_obj(cls, subj: T, /, **kwargs) -> Any: ...
-
-
-adapter_members = get_protocol_members(Adapter)  # duck typing
-
-
-class AdapterRegistry:
-
-    _adapters: dict[str, Adapter] = {}
-
-    @classmethod
-    def list_adapters(cls) -> list[tuple[str | type, ...]]:
-        return list(cls._adapters.keys())
-
-    @classmethod
-    def register(cls, adapter: type[Adapter]) -> None:
-        for member in adapter_members:
-            if not hasattr(adapter, member):
-                _str = getattr(adapter, "obj_key", None) or repr(adapter)
-                _str = _str[:50] if len(_str) > 50 else _str
-                raise AttributeError(
-                    f"Adapter {_str} missing required methods."
-                )
-
-        if isinstance(adapter, type):
-            cls._adapters[adapter.obj_key] = adapter()
-        else:
-            cls._adapters[adapter.obj_key] = adapter
-
-    @classmethod
-    def get(cls, obj_key: type | str) -> Adapter:
-        try:
-            return cls._adapters[obj_key]
-        except Exception as e:
-            logging.error(f"Error getting adapter for {obj_key}. Error: {e}")
-
-    @classmethod
-    def adapt_from(
-        cls, subj_cls: type[T], obj: Any, obj_key: type | str, **kwargs
-    ) -> dict | list[dict]:
-        try:
-            return cls.get(obj_key).from_obj(subj_cls, obj, **kwargs)
-        except Exception as e:
-            logging.error(f"Error adapting data from {obj_key}. Error: {e}")
-            raise e
-
-    @classmethod
-    def adapt_to(cls, subj: T, obj_key: type | str, **kwargs) -> Any:
-        try:
-            return cls.get(obj_key).to_obj(subj, **kwargs)
-        except Exception as e:
-            logging.error(f"Error adapting data to {obj_key}. Error: {e}")
-            raise e
-
-
-class JsonAdapter(Adapter):
-
-    obj_key = "json"
-
-    @classmethod
-    def from_obj(cls, subj_cls: type[T], obj: str, /) -> dict:
-        return json.loads(obj)
-
-    @classmethod
-    def to_obj(cls, subj: T) -> str:
-        return json.dumps(subj.to_dict())
-
-
-class JsonFileAdapter(Adapter):
-
-    obj_key = ".json"
-
-    @classmethod
-    def from_obj(cls, subj_cls: type[T], obj: str | Path, /) -> dict:
-        with open(obj) as f:
-            return json.load(f)
-
-    @classmethod
-    def to_obj(
-        cls,
-        subj: T,
-        /,
-        fp: str | Path,
-    ) -> None:
-        with open(fp, "w") as f:
-            json.dump(subj.to_dict(), f)
-        logging.info(f"Successfully saved data to {fp}")
-
-
-class PandasSeriesAdapter(Adapter):
-
-    obj_key = "pd_series"
-    alias = ("pandas_series", "pd.series", "pd_series")
-
-    @classmethod
-    def from_obj(cls, subj_cls: type[T], obj: pd.Series, /, **kwargs) -> dict:
-        return obj.to_dict(**kwargs)
-
-    @classmethod
-    def to_obj(cls, subj: T, /, **kwargs) -> pd.Series:
-        return pd.Series(subj.to_dict(), **kwargs)
-
-
-class PandasDataFrameAdapter(Adapter):
-
-    obj_key = "pd_dataframe"
-    alias = ("pandas_dataframe", "pd.DataFrame", "pd_dataframe")
-
-    @classmethod
-    def from_obj(
-        cls, subj_cls: type[T], obj: pd.DataFrame, /, **kwargs
-    ) -> list[dict]:
-        """kwargs for pd.DataFrame.to_dict"""
-        return obj.to_dict(orient="records", **kwargs)
-
-    @classmethod
-    def to_obj(cls, subj: list[T], /, **kwargs) -> pd.DataFrame:
-        """kwargs for pd.DataFrame"""
-        out_ = []
-        for i in subj:
-            _dict = i.to_dict()
-            _dict["created_at"] = datetime.fromtimestamp(_dict["created_at"])
-            out_.append(_dict)
-        df = pd.DataFrame(out_, **kwargs)
-        if "created_at" in df.columns:
-            df["created_at"] = pd.to_datetime(df["created_at"])
-        return df
-
-
-class CSVFileAdapter(Adapter):
-
-    obj_key = ".csv"
-    alias = (".csv", "csv_file", "csv")
-
-    @classmethod
-    def from_obj(
-        cls, subj_cls: type[T], obj: str | Path, /, **kwargs
-    ) -> list[dict]:
-        """kwargs for pd.read_csv"""
-        df = pd.read_csv(obj, **kwargs)
-        return df.to_dict(orient="records")
-
-    @classmethod
-    def to_obj(
-        cls,
-        subj: list[T],
-        /,
-        fp: str | Path,
-        **kwargs,
-    ) -> None:
-        """kwargs for pd.DataFrame.to_csv"""
-        kwargs["index"] = False
-        pd.DataFrame([i.to_dict() for i in subj]).to_csv(fp, **kwargs)
-        logging.info(f"Successfully saved data to {fp}")
-
-
-class ExcelFileAdapter(Adapter):
-
-    obj_key = ".xlsx"
-    alias = (".xlsx", "excel_file", "excel", "xlsx", "xls", ".xls")
-
-    @classmethod
-    def from_obj(
-        cls, subj_cls: type[T], obj: str | Path, /, **kwargs
-    ) -> list[dict]:
-        return pd.read_excel(obj, **kwargs).to_dict(orient="records")
-
-    @classmethod
-    def to_obj(cls, subj: list[T], /, fp: str | Path, **kwargs) -> None:
-        kwargs["index"] = False
-        pd.DataFrame([i.to_dict() for i in subj]).to_excel(fp, **kwargs)
-        logging.info(f"Saved {subj.class_name()} to {fp}")
-
-
-NODE_ADAPTERS = [
-    JsonAdapter,
-    JsonFileAdapter,
-    PandasSeriesAdapter,
-]
-
-
-class NodeAdapterRegistry(AdapterRegistry):
-    _adapters = {k.obj_key: k() for k in NODE_ADAPTERS}
-
-
-PILE_ADAPTERS = [
-    JsonAdapter,
-    JsonFileAdapter,
-    PandasDataFrameAdapter,
-    CSVFileAdapter,
-    ExcelFileAdapter,
-]
-
-
-class PileAdapterRegistry(AdapterRegistry):
-
-    _adapters = {k.obj_key: k() for k in PILE_ADAPTERS}