lionagi 0.2.4__tar.gz → 0.2.5__tar.gz

{lionagi-0.2.4 → lionagi-0.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lionagi
-Version: 0.2.4
+Version: 0.2.5
 Summary: Towards automated general intelligence.
 Author: HaiyangLi
 Author-email: Haiyang Li <ocean@lionagi.ai>
@@ -237,7 +237,7 @@ Requires-Dist: boto3>=1.34.131
 ### an AGentic Intelligence Operating System
 
 ```
-pip install lionagi==0.2.3
+pip install lionagi==0.2.5
 ```
 
 **Powerful Intelligent Workflow Automation**

{lionagi-0.2.4 → lionagi-0.2.5}/README.md

@@ -8,7 +8,7 @@
 ### an AGentic Intelligence Operating System
 
 ```
-pip install lionagi==0.2.3
+pip install lionagi==0.2.5
 ```
 
 **Powerful Intelligent Workflow Automation**

{lionagi-0.2.4 → lionagi-0.2.5}/lionagi/core/engine/branch_engine.py

@@ -57,7 +57,10 @@ class BranchExecutor(Branch, BaseExecutor):
                     await self._process_condition(mail)
                 elif mail.category == "end":
                     self._process_end(mail)
-            if key in self.mailbox.pending_ins and self.mailbox.pending_ins.get(key, Pile()).size() == 0:
+            if (
+                key in self.mailbox.pending_ins
+                and self.mailbox.pending_ins.get(key, Pile()).size() == 0
+            ):
                 self.mailbox.pending_ins.pop(key)
 
     async def execute(self, refresh_time=1) -> None:

{lionagi-0.2.4 → lionagi-0.2.5}/lionagi/core/generic/graph.py

@@ -202,7 +202,9 @@ class Graph(Node):
 
         return g
 
-    def display(self, node_label="class_name", edge_label="label", draw_kwargs={}, **kwargs):
+    def display(
+        self, node_label="class_name", edge_label="label", draw_kwargs={}, **kwargs
+    ):
         """Display the graph using NetworkX and Matplotlib."""
         from lionagi.libs import SysUtil
 
@@ -224,7 +226,7 @@ class Graph(Node):
             node_color="orange",
             alpha=0.9,
             labels=nx.get_node_attributes(g, node_label),
-            **draw_kwargs
+            **draw_kwargs,
         )
 
         labels = nx.get_edge_attributes(g, edge_label)

{lionagi-0.2.4 → lionagi-0.2.5}/lionagi/core/generic/node.py

@@ -124,7 +124,7 @@ class Node(Component, Relatable):
         label: str | None = None,
         bundle: bool = False,
         edge_class: Callable = Edge,
-        **kwargs
+        **kwargs,
     ) -> None:
         """
         Establish directed relationship from this node to another.
@@ -150,7 +150,7 @@ class Node(Component, Relatable):
             condition=condition,
             bundle=bundle,
             label=label,
-            **kwargs
+            **kwargs,
         )
 
         self.relations[direction].include(edge)

{lionagi-0.2.4 → lionagi-0.2.5}/lionagi/core/session/directive_mixin.py

@@ -248,7 +248,7 @@ class DirectiveMixin:
                 form.action_response.update(_dict)
 
             return form
-        
+
         form = await _directive.direct(
             instruction=instruction,
             context=context,
@@ -283,5 +283,5 @@ class DirectiveMixin:
             if not hasattr(form, "action_response"):
                 form.append_to_request("action_response", {})
             form.action_response.update(_dict)
-        
+
         return form

{lionagi-0.2.4 → lionagi-0.2.5}/lionagi/core/unit/unit_mixin.py

@@ -658,7 +658,7 @@ class DirectiveMixin(ABC):
         if allow_action and not tools:
             tools = True
 
-        tool_schema=None
+        tool_schema = None
         if tools:
             tool_schema = branch.tool_manager.get_tool_schema(tools)
 

{lionagi-0.2.4 → lionagi-0.2.5}/lionagi/core/work/work_edge.py

@@ -21,19 +21,19 @@ class WorkEdge(Edge, Progressable):
             other than "from_work" and "from_result".
         associated_worker (Worker): The worker to which this WorkEdge belongs.
     """
+
     convert_function: Callable = Field(
         ...,
-        description="Function to transform the result of the previous work into parameters for the next work."
+        description="Function to transform the result of the previous work into parameters for the next work.",
     )
 
     convert_function_kwargs: dict = Field(
         {},
-        description="parameters for the worklink function other than \"from_work\" and \"from_result\""
+        description='parameters for the worklink function other than "from_work" and "from_result"',
     )
 
     associated_worker: Worker = Field(
-        ...,
-        description="The worker to which this WorkEdge belongs."
+        ..., description="The worker to which this WorkEdge belongs."
     )
 
     @field_validator("convert_function", mode="before")
@@ -81,8 +81,10 @@ class WorkEdge(Edge, Progressable):
             StopIteration: If the task has no available steps left to proceed.
         """
         if task.available_steps == 0:
-            task.status_note = ("Task stopped proceeding further as all available steps have been used up, "
-                                "but the task has not yet reached completion.")
+            task.status_note = (
+                "Task stopped proceeding further as all available steps have been used up, "
+                "but the task has not yet reached completion."
+            )
             return
         func_signature = inspect.signature(self.convert_function)
         kwargs = self.convert_function_kwargs.copy()

{lionagi-0.2.4 → lionagi-0.2.5}/lionagi/core/work/work_function.py

@@ -13,6 +13,7 @@ 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 asyncio
 import logging
 
@@ -33,7 +34,13 @@ class WorkFunction:
     """
 
     def __init__(
-        self, assignment, function, retry_kwargs=None, guidance=None, capacity=10, refresh_time=1
+        self,
+        assignment,
+        function,
+        retry_kwargs=None,
+        guidance=None,
+        capacity=10,
+        refresh_time=1,
     ):
         """
         Initializes a WorkFunction instance.
@@ -69,11 +76,11 @@ class WorkFunction:
     @property
     def execution_mode(self):
         """
-         Gets the execution mode of the work function's queue.
+        Gets the execution mode of the work function's queue.
 
-         Returns:
-             bool: The execution mode of the work function's queue.
-         """
+        Returns:
+            bool: The execution mode of the work function's queue.
+        """
         return self.worklog.queue.execution_mode
 
     def is_progressable(self):
@@ -121,4 +128,4 @@ class WorkFunction:
         """
         Stops the execution of the work function's queue.
         """
-        await self.worklog.stop()
+        await self.worklog.stop()

{lionagi-0.2.4 → lionagi-0.2.5}/lionagi/core/work/work_function_node.py

@@ -21,7 +21,16 @@ class WorkFunctionNode(WorkFunction, Node):
         refresh_time (int): The time interval to refresh the work log queue.
     """
 
-    def __init__(self, assignment, function, retry_kwargs=None, guidance=None, capacity=10, refresh_time=1, **kwargs):
+    def __init__(
+        self,
+        assignment,
+        function,
+        retry_kwargs=None,
+        guidance=None,
+        capacity=10,
+        refresh_time=1,
+        **kwargs,
+    ):
         """
         Initializes a WorkFunctionNode instance.
 
@@ -35,10 +44,12 @@ class WorkFunctionNode(WorkFunction, Node):
             **kwargs: Additional keyword arguments for the Node initialization.
         """
         Node.__init__(self, **kwargs)
-        WorkFunction.__init__(self,
-                              assignment=assignment,
-                              function=function,
-                              retry_kwargs=retry_kwargs,
-                              guidance=guidance,
-                              capacity=capacity,
-                              refresh_time=refresh_time)
+        WorkFunction.__init__(
+            self,
+            assignment=assignment,
+            function=function,
+            retry_kwargs=retry_kwargs,
+            guidance=guidance,
+            capacity=capacity,
+            refresh_time=refresh_time,
+        )

{lionagi-0.2.4 → lionagi-0.2.5}/lionagi/core/work/work_queue.py

@@ -90,11 +90,11 @@ class WorkQueue:
 
     async def execute(self):
         """
-            Continuously executes the process method at a specified refresh interval.
+        Continuously executes the process method at a specified refresh interval.
 
-            Args:
-                refresh_time (int, optional): The time in seconds to wait between
-                    successive calls to `process`. Defaults to 1.
+        Args:
+            refresh_time (int, optional): The time in seconds to wait between
+                successive calls to `process`. Defaults to 1.
         """
         self.execution_mode = True
         self._stop_event.clear()

{lionagi-0.2.4 → lionagi-0.2.5}/lionagi/core/work/work_task.py

@@ -20,39 +20,24 @@ class WorkTask(Component):
         current_work (Work | None): The current work in progress.
         post_processing (Callable | None): The post-processing function to be executed after the entire task is successfully completed.
     """
-    name: str | None = Field(
-        None,
-        description="Name of the task"
-    )
+
+    name: str | None = Field(None, description="Name of the task")
 
     status: WorkStatus = Field(
-        WorkStatus.PENDING,
-        description="The current status of the task"
+        WorkStatus.PENDING, description="The current status of the task"
     )
 
-    status_note: str = Field(
-        None,
-        description="Note for tasks current status"
-    )
+    status_note: str = Field(None, description="Note for tasks current status")
 
-    work_history: list[Work] = Field(
-        [],
-        description="List of works processed"
-    )
+    work_history: list[Work] = Field([], description="List of works processed")
 
-    max_steps: int | None = Field(
-        10,
-        description="Maximum number of works allowed"
-    )
+    max_steps: int | None = Field(10, description="Maximum number of works allowed")
 
-    current_work: Work | None = Field(
-        None,
-        description="The current work in progress"
-    )
+    current_work: Work | None = Field(None, description="The current work in progress")
 
     post_processing: Callable | None = Field(
         None,
-        description="The post-processing function to be executed after the entire task has been successfully completed."
+        description="The post-processing function to be executed after the entire task has been successfully completed.",
     )
 
     @field_validator("max_steps", mode="before")
@@ -108,7 +93,12 @@ class WorkTask(Component):
         Returns:
             WorkTask: A new instance of WorkTask with the same attributes.
         """
-        new_worktask = WorkTask(name=self.name, status=self.status, max_steps=self.max_steps, current_work=self.current_work)
+        new_worktask = WorkTask(
+            name=self.name,
+            status=self.status,
+            max_steps=self.max_steps,
+            current_work=self.current_work,
+        )
         for work in self.work_history:
             new_worktask.work_history.append(work)
         return new_worktask

{lionagi-0.2.4 → lionagi-0.2.5}/lionagi/core/work/worker.py

@@ -93,7 +93,9 @@ class Worker(ABC):
 
         """
         if form_key not in self.forms.keys():
-            raise ValueError(f"Unable to change default form. Key {form_key} does not exist.")
+            raise ValueError(
+                f"Unable to change default form. Key {form_key} does not exist."
+            )
         self.default_form = self.forms[form_key]
 
     def _get_decorated_functions(self, decorator_attr, name_only=True):
@@ -108,7 +110,9 @@ class Worker(ABC):
             list: List of decorated function names or tuples containing function details.
         """
         decorated_functions = []
-        for name, func in inspect.getmembers(self.__class__, predicate=inspect.isfunction):
+        for name, func in inspect.getmembers(
+            self.__class__, predicate=inspect.isfunction
+        ):
             if hasattr(func, decorator_attr):
                 if name_only:
                     decorated_functions.append(name)
@@ -121,11 +125,18 @@ class Worker(ABC):
         """
         Validates worklink functions to ensure they have the required parameters.
         """
-        worklink_decorated_function = self._get_decorated_functions(decorator_attr="_worklink_decorator_params", name_only=False)
+        worklink_decorated_function = self._get_decorated_functions(
+            decorator_attr="_worklink_decorator_params", name_only=False
+        )
         for func_name, func, _ in worklink_decorated_function:
             func_signature = inspect.signature(func)
-            if "from_work" not in func_signature.parameters and "from_result" not in func_signature.parameters:
-                raise ValueError(f"Either \"from_work\" or \"from_result\" must be a parameter in function {func_name}")
+            if (
+                "from_work" not in func_signature.parameters
+                and "from_result" not in func_signature.parameters
+            ):
+                raise ValueError(
+                    f'Either "from_work" or "from_result" must be a parameter in function {func_name}'
+                )
 
     def construct_all_work_functions(self):
         """
@@ -133,7 +144,9 @@ class Worker(ABC):
         """
         if getattr(self, "work_functions", None) is None:
             self.work_functions = {}
-        work_decorated_function = self._get_decorated_functions(decorator_attr="_work_decorator_params", name_only=False)
+        work_decorated_function = self._get_decorated_functions(
+            decorator_attr="_work_decorator_params", name_only=False
+        )
         for func_name, func, dec_params in work_decorated_function:
             if func_name not in self.work_functions:
                 self.work_functions[func_name] = WorkFunction(**dec_params)
@@ -174,7 +187,7 @@ class Worker(ABC):
                 retry_kwargs=retry_kwargs or {},
                 guidance=guidance or function.__doc__,
                 capacity=capacity,
-                refresh_time=refresh_time
+                refresh_time=refresh_time,
             )
 
         work_func: WorkFunction = self.work_functions[function.__name__]
@@ -183,7 +196,9 @@ class Worker(ABC):
         if form_param_key:
             func_signature = inspect.signature(function)
             if form_param_key not in func_signature.parameters:
-                raise KeyError(f"Failed to locate form. \"{form_param_key}\" is not defined in the function.")
+                raise KeyError(
+                    f'Failed to locate form. "{form_param_key}" is not defined in the function.'
+                )
             if "self" in func_signature.parameters:
                 bound_args = func_signature.bind(None, *args, **kwargs)
             else:
@@ -199,14 +214,18 @@ class Worker(ABC):
             form = self.forms.get(form_key) or self.default_form
 
             if form:
-                subform = form.__class__(assignment=work_func.assignment, task=work_func.guidance)
+                subform = form.__class__(
+                    assignment=work_func.assignment, task=work_func.guidance
+                )
                 for k in subform.input_fields:
                     v = getattr(form, k, None)
                     setattr(subform, k, v)
                 subform.origin = form
                 kwargs = {"form": subform} | kwargs
             else:
-                raise ValueError(f"Cannot locate form in Worker's forms and default_form is not available.")
+                raise ValueError(
+                    f"Cannot locate form in Worker's forms and default_form is not available."
+                )
 
         task = work_func.perform(self, *args, **kwargs)
         work = Work(async_task=task, async_task_name=work_func.name)
@@ -221,7 +240,7 @@ def work(
     guidance=None,
     retry_kwargs=None,
     timeout=10,
-    refresh_time=1
+    refresh_time=1,
 ):
     """
     Decorator to mark a method as a work function.
@@ -271,23 +290,22 @@ def work(
                 refresh_time=refresh_time,
                 **kwargs,
             )
-        wrapper._work_decorator_params = {"assignment": assignment,
-                                          "function": func,
-                                          "retry_kwargs": retry_kwargs,
-                                          "guidance": guidance,
-                                          "capacity": capacity,
-                                          "refresh_time": refresh_time}
+
+        wrapper._work_decorator_params = {
+            "assignment": assignment,
+            "function": func,
+            "retry_kwargs": retry_kwargs,
+            "guidance": guidance,
+            "capacity": capacity,
+            "refresh_time": refresh_time,
+        }
 
         return wrapper
 
     return decorator
 
 
-def worklink(
-        from_: str,
-        to_: str,
-        auto_schedule: bool = True
-):
+def worklink(from_: str, to_: str, auto_schedule: bool = True):
     """
     Decorator to create a link between two work functions.
 
@@ -299,26 +317,31 @@ def worklink(
     Returns:
         Callable: The decorated function.
     """
+
     def decorator(func):
         @wraps(func)
         async def wrapper(
-            self: Worker,
-            *args,
-            func=func,
-            from_=from_,
-            to_=to_,
-            **kwargs
+            self: Worker, *args, func=func, from_=from_, to_=to_, **kwargs
         ):
             if not inspect.iscoroutinefunction(func):
                 raise TypeError(f"{func.__name__} must be an asynchronous function")
 
-            work_funcs = self._get_decorated_functions(decorator_attr="_work_decorator_params")
+            work_funcs = self._get_decorated_functions(
+                decorator_attr="_work_decorator_params"
+            )
             if from_ not in work_funcs or to_ not in work_funcs:
-                raise ValueError("Invalid link. 'from_' and 'to_' must be the name of work decorated functions.")
+                raise ValueError(
+                    "Invalid link. 'from_' and 'to_' must be the name of work decorated functions."
+                )
 
             func_signature = inspect.signature(func)
-            if "from_work" not in func_signature.parameters and "from_result" not in func_signature.parameters:
-                raise ValueError(f"Either \"from_work\" or \"from_result\" must be a parameter in function {func.__name__}")
+            if (
+                "from_work" not in func_signature.parameters
+                and "from_result" not in func_signature.parameters
+            ):
+                raise ValueError(
+                    f'Either "from_work" or "from_result" must be a parameter in function {func.__name__}'
+                )
 
             if "self" in func_signature.parameters:
                 bound_args = func_signature.bind(None, *args, **kwargs)
@@ -331,10 +354,14 @@ def worklink(
 
             if from_work := arguments.get("from_work"):
                 if not isinstance(from_work, Work):
-                    raise ValueError("Invalid type for from_work. Only work objects are accepted.")
+                    raise ValueError(
+                        "Invalid type for from_work. Only work objects are accepted."
+                    )
                 if from_work.async_task_name != from_:
-                    raise ValueError(f"Invalid work object in from_work. "
-                                     f"async_task_name \"{from_work.async_task_name}\" does not match from_ \"{from_}\"")
+                    raise ValueError(
+                        f"Invalid work object in from_work. "
+                        f'async_task_name "{from_work.async_task_name}" does not match from_ "{from_}"'
+                    )
 
             next_params = await func(self, *args, **kwargs)
             to_work_func = getattr(self, to_)
@@ -347,7 +374,9 @@ def worklink(
                 if wrapper.auto_schedule:
                     return await to_work_func(**next_params)
             elif isinstance(next_params, tuple) and len(next_params) == 2:
-                if isinstance(next_params[0], list) and isinstance(next_params[1], dict):
+                if isinstance(next_params[0], list) and isinstance(
+                    next_params[1], dict
+                ):
                     if wrapper.auto_schedule:
                         return await to_work_func(*next_params[0], **next_params[1])
                 else:
@@ -358,9 +387,7 @@ def worklink(
             return next_params
 
         wrapper.auto_schedule = auto_schedule
-        wrapper._worklink_decorator_params = {"func": func,
-                                              "from_": from_,
-                                              "to_": to_}
+        wrapper._worklink_decorator_params = {"func": func, "from_": from_, "to_": to_}
 
         return wrapper
 

{lionagi-0.2.4 → lionagi-0.2.5}/lionagi/core/work/worker_engine.py

@@ -41,7 +41,15 @@ class WorkerEngine:
         self.refresh_time = refresh_time
         self._stop_event = asyncio.Event()
 
-    async def add_task(self, *args, task_function: str, task_name=None, task_max_steps=10, task_post_processing=None, **kwargs):
+    async def add_task(
+        self,
+        *args,
+        task_function: str,
+        task_name=None,
+        task_max_steps=10,
+        task_post_processing=None,
+        **kwargs,
+    ):
         """
         Adds a new task to the task queue.
 
@@ -56,7 +64,11 @@ class WorkerEngine:
         Returns:
             WorkTask: The newly created task.
         """
-        task = WorkTask(name=task_name, max_steps=task_max_steps, post_processing=task_post_processing)
+        task = WorkTask(
+            name=task_name,
+            max_steps=task_max_steps,
+            post_processing=task_post_processing,
+        )
         self.tasks.append(task)
         function = getattr(self.worker, task_function)
         work = await function(*args, **kwargs)
@@ -66,7 +78,7 @@ class WorkerEngine:
 
     async def activate_work_queues(self):
         """
-                Activates the work queues for all work functions.
+        Activates the work queues for all work functions.
         """
         for work_function in self.worker.work_functions.values():
             if not work_function.worklog.queue.execution_mode:
@@ -141,11 +153,12 @@ class WorkerEngine:
         Executes tasks continuously until stopped.
         """
         self._stop_event.clear()
-        
+
         async def execute_lasting_inner():
             while not self.stopped:
                 await self.execute(stop_queue=False)
                 await asyncio.sleep(self.refresh_time)
+
         asyncio.create_task(execute_lasting_inner())
 
     def _construct_work_functions(self):
@@ -154,26 +167,38 @@ class WorkerEngine:
         """
         if getattr(self.worker, "work_functions", None) is None:
             self.worker.work_functions = {}
-        work_decorated_function = self.worker._get_decorated_functions(decorator_attr="_work_decorator_params",
-                                                                       name_only=False)
+        work_decorated_function = self.worker._get_decorated_functions(
+            decorator_attr="_work_decorator_params", name_only=False
+        )
         for func_name, func, dec_params in work_decorated_function:
             if func_name not in self.worker.work_functions:
                 self.worker.work_functions[func_name] = WorkFunctionNode(**dec_params)
                 self.worker_graph.add_node(self.worker.work_functions[func_name])
             else:
-                if not isinstance(self.worker.work_functions[func_name], WorkFunctionNode):
-                    raise TypeError(f"WorkFunction {func_name} already exists but is not a WorkFunctionNode. "
-                                    f"If you would like to use it in WorkerEngine, please convert it to a "
-                                    f"WorkFunctionNode, or initiate a new worker, or pop it from work_function dict")
+                if not isinstance(
+                    self.worker.work_functions[func_name], WorkFunctionNode
+                ):
+                    raise TypeError(
+                        f"WorkFunction {func_name} already exists but is not a WorkFunctionNode. "
+                        f"If you would like to use it in WorkerEngine, please convert it to a "
+                        f"WorkFunctionNode, or initiate a new worker, or pop it from work_function dict"
+                    )
 
     def _construct_workedges(self):
         """
         Constructs work edges for the worker graph.
         """
-        worklink_decorated_function = self.worker._get_decorated_functions(decorator_attr="_worklink_decorator_params",
-                                                                           name_only=False)
+        worklink_decorated_function = self.worker._get_decorated_functions(
+            decorator_attr="_worklink_decorator_params", name_only=False
+        )
 
         for func_name, func, dec_params in worklink_decorated_function:
             head = self.worker.work_functions[dec_params["from_"]]
             tail = self.worker.work_functions[dec_params["to_"]]
-            self.worker_graph.add_edge(head=head, tail=tail, convert_function=func, associated_worker=self.worker, edge_class=WorkEdge)
+            self.worker_graph.add_edge(
+                head=head,
+                tail=tail,
+                convert_function=func,
+                associated_worker=self.worker,
+                edge_class=WorkEdge,
+            )

{lionagi-0.2.4 → lionagi-0.2.5}/lionagi/integrations/config/oai_configs.py

@@ -4,7 +4,7 @@ API_key_schema = ("OPENAI_API_KEY",)
 
 # ChatCompletion
 oai_chat_llmconfig = {
-    "model": "gpt-4o",
+    "model": "gpt-4o-2024-08-06",
     "frequency_penalty": 0,
     "max_tokens": None,
     "n": 1,

{lionagi-0.2.4 → lionagi-0.2.5}/lionagi/integrations/provider/litellm.py

@@ -27,8 +27,9 @@ class LiteLLMService(BaseService):
         SysUtil.check_import("litellm")
 
         import litellm
+
         litellm.drop_params = True
-        
+
         self.acompletion = litellm.acompletion
         self.model = model
         self.kwargs = kwargs
@@ -45,9 +46,7 @@ class LiteLLMService(BaseService):
         kwargs["model"] = self.model or kwargs.get("model")
 
         try:
-            completion = await self.acompletion(
-                messages=messages, **kwargs
-            )
+            completion = await self.acompletion(messages=messages, **kwargs)
             return payload, completion.model_dump()
         except Exception as e:
             self.status_tracker.num_tasks_failed += 1