lionagi 0.8.3__py3-none-any.whl → 0.8.4__py3-none-any.whl

lionagi/operatives/action/tool.py

@@ -12,7 +12,6 @@ import inspect
 from collections.abc import Callable
 from typing import Any, TypeAlias
 
-from openai import BaseModel
 from pydantic import Field, field_validator, model_validator
 from typing_extensions import Self
 

lionagi/operatives/forms/flow.py

@@ -1,5 +1,4 @@
 # forms/flow.py
-from typing import List
 
 from pydantic import BaseModel, ConfigDict, Field
 

lionagi/operatives/forms/form.py

@@ -1,6 +1,6 @@
 # forms/form.py
 
-from typing import Any, Optional
+from typing import Any
 
 from pydantic import ConfigDict, Field, model_validator
 from typing_extensions import Self

lionagi/session/branch.py

@@ -345,6 +345,20 @@ class Branch(Element, Communicatable, Relational):
 
         return branch_clone
 
+    def register_tools(
+        self, tools: FuncTool | list[FuncTool], update: bool = False
+    ):
+        """
+        Registers one or more tools in the ActionManager.
+
+        Args:
+            tools (FuncTool | list[FuncTool]):
+                A single tool or a list of tools to register.
+            update (bool, optional):
+                If `True`, updates existing tools with the same name.
+        """
+        self._action_manager.register_tools(tools, update=update)
+
     # -------------------------------------------------------------------------
     # Conversion / Serialization
     # -------------------------------------------------------------------------
@@ -532,6 +546,33 @@ class Branch(Element, Communicatable, Relational):
         for key in self.mailbox.pending_ins:
             self.receive(key)
 
+    def connect(
+        self,
+        name: str,
+        imodel: iModel,
+        request_options: type[BaseModel],
+        description: str = None,
+        update: bool = False,
+    ):
+        if not update and name in self.tools:
+            raise ValueError(f"Tool with name '{name}' already exists.")
+
+        async def _connect(**kwargs):
+            """connect to an api endpoint"""
+            api_call = await imodel.invoke(**kwargs)
+            self._log_manager.log(Log.create(api_call))
+            return api_call.response
+
+        _connect.__name__ = name
+        if description:
+            _connect.__doc__ = description
+
+        tool = Tool(
+            func_callable=_connect,
+            request_options=request_options,
+        )
+        self._action_manager.register_tools(tool, update=update)
+
     # -------------------------------------------------------------------------
     # Dictionary Conversion
     # -------------------------------------------------------------------------

lionagi/version.py

@@ -1 +1 @@
-__version__ = "0.8.3"
+__version__ = "0.8.4"

{lionagi-0.8.3.dist-info → lionagi-0.8.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lionagi
-Version: 0.8.3
+Version: 0.8.4
 Summary: An Intelligence Operating System.
 Author-email: HaiyangLi <quantocean.li@gmail.com>
 License:            Apache License

{lionagi-0.8.3.dist-info → lionagi-0.8.4.dist-info}/RECORD

@@ -4,7 +4,7 @@ lionagi/_errors.py,sha256=wNKdnVQvE_CHEstK7htrrj334RA_vbGcIds-3pUiRkc,455
 lionagi/_types.py,sha256=9g7iytvSj3UjZxD-jL06_fxuNfgZyWT3Qnp0XYp1wQU,63
 lionagi/settings.py,sha256=k9zRJXv57TveyfHO3Vr9VGiKrSwlRUUVKt5zf6v9RU4,1627
 lionagi/utils.py,sha256=QbF4E1PG-BaRcEVH3kJIYCJVNq-oRNoTxjda5k8NYW4,73177
-lionagi/version.py,sha256=otnwfmvJLUamPajTPUaIekiO9mA-2HPi-_h_E0N-uOQ,22
+lionagi/version.py,sha256=jhHEJFZWhkQDemoZMomBYq-RNrKXknYzUaeIU9A6XsI,22
 lionagi/libs/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/libs/parse.py,sha256=tpEbmIRGuHhLCJlUlm6fjmqm_Z6XJLAXGNFHNuk422I,1011
 lionagi/libs/file/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
@@ -85,12 +85,12 @@ lionagi/operatives/action/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQp
 lionagi/operatives/action/function_calling.py,sha256=dFNldEuV77quaM2M4VGXPQRhh3yGCAA5WucyipRu13M,5067
 lionagi/operatives/action/manager.py,sha256=FXfWhQMSFE5qJNZPpgdz4ePvthLafZfnmak2PImCrsc,8824
 lionagi/operatives/action/request_response_model.py,sha256=mCyKub_WoEttJ_mqLhGoOoPVBQHOhr7sswy_jN6-620,3378
-lionagi/operatives/action/tool.py,sha256=G63P1Q9Kmq6wtK3OJy7G8lqzLo6AsmLa7Yk4fGgida4,5613
+lionagi/operatives/action/tool.py,sha256=39Z2beUUdG0ngoK1r2ynmX4yIiDIyHv_dxb4X53In24,5584
 lionagi/operatives/action/utils.py,sha256=vUe7Aysuzbg16rAfe2Ttp5QUz5_L6mMedBVAWzGAHwk,4330
 lionagi/operatives/forms/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/operatives/forms/base.py,sha256=hitr0eKk7Yts2VfpuBGl0YxpClMJEpsy7xWodhg2AhQ,2720
-lionagi/operatives/forms/flow.py,sha256=JRHY_t8yZNN2p_spPvoFG1hD--2YVtAiA8h3hrK7Vt0,2556
-lionagi/operatives/forms/form.py,sha256=GJTE_PrlHT_bgWf-zOQaIFdEkx1CUDTzgAyRofPXoOE,2934
+lionagi/operatives/forms/flow.py,sha256=Em0AIiSQT5r45wKjoq57J_wuDZ9fecKZxf6FzmmamQM,2532
+lionagi/operatives/forms/form.py,sha256=j5Mf9DCblJYtN_uCgBhXzE4IkrZHWc7jyzog3tXqeW4,2924
 lionagi/operatives/forms/report.py,sha256=ZQsM5nDnmF3WdCSurDUBTMI-PEy8yTnRA_uN6tyRneA,1488
 lionagi/operatives/instruct/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/operatives/instruct/base.py,sha256=Vw9u3BGzjNRGcPfW1ROr2uCZcZsjrm4WU6491P6ecCI,1895
@@ -187,9 +187,9 @@ lionagi/service/providers/openrouter_/chat_completions.py,sha256=MRf4ZbMCgzNIL4g
 lionagi/service/providers/perplexity_/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/service/providers/perplexity_/chat_completions.py,sha256=SsDbrtXwQsR4Yu2VMU43KfeS86QWI8UTNhDth5lNWNs,1055
 lionagi/session/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
-lionagi/session/branch.py,sha256=JvErd9YUuGdWyLj37rtKOteSqV0ltn9lg0R2G8GO40c,62539
+lionagi/session/branch.py,sha256=ZLItnjW66__85s0brxhq1LZisjDoVS4HBJSHUpFaxsg,63834
 lionagi/session/session.py,sha256=po6C7PnM0iu_ISHUo4PBzzQ61HFOgcsAUfPoO--eLak,8987
-lionagi-0.8.3.dist-info/METADATA,sha256=uWD1Rcqe1Qknq4Fcroot2pVTsFNyPxM0UrbyZZaa8NY,22819
-lionagi-0.8.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lionagi-0.8.3.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
-lionagi-0.8.3.dist-info/RECORD,,
+lionagi-0.8.4.dist-info/METADATA,sha256=MHgfVuup5176MSrXlfX2B_tQUaiKNdGAjtITrJVLCWY,22819
+lionagi-0.8.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lionagi-0.8.4.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
+lionagi-0.8.4.dist-info/RECORD,,