lionagi 0.8.3__tar.gz → 0.8.4__tar.gz

{lionagi-0.8.3 → lionagi-0.8.4}/PKG-INFO

@@ -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 → lionagi-0.8.4}/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-0.8.3 → lionagi-0.8.4}/lionagi/operatives/forms/flow.py

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

{lionagi-0.8.3 → lionagi-0.8.4}/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-0.8.3 → lionagi-0.8.4}/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-0.8.4/lionagi/version.py

@@ -0,0 +1 @@
+__version__ = "0.8.4"

{lionagi-0.8.3 → lionagi-0.8.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "lionagi"
-version = "0.8.3"
+version = "0.8.4"
 description = "An Intelligence Operating System."
 authors = [
     { name = "HaiyangLi", email = "quantocean.li@gmail.com" },

{lionagi-0.8.3 → lionagi-0.8.4}/uv.lock

@@ -895,7 +895,7 @@ wheels = [
 
 [[package]]
 name = "lionagi"
-version = "0.8.3"
+version = "0.8.4"
 source = { editable = "." }
 dependencies = [
     { name = "aiocache" },

lionagi-0.8.3/lionagi/version.py

@@ -1 +0,0 @@
-__version__ = "0.8.3"