lionagi 0.2.4__py3-none-any.whl → 0.2.6__py3-none-any.whl

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/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/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/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

lionagi/libs/ln_parse.py

@@ -694,11 +694,11 @@ class StringMatch:
 
         if isinstance(out_, str):
             # first try to parse it straight as a fuzzy json
-            
+
             try:
                 out_ = ParseUtil.fuzzy_parse_json(out_)
                 return StringMatch.correct_dict_keys(keys, out_)
-            
+
             except:
                 try:
                     out_ = ParseUtil.md_to_json(out_)

lionagi/tests/api/aws/conftest.py

@@ -1,28 +1,25 @@
-import pytest
-from api.apicore import _connect as _connect_
-from api.apicore import _config as _config_
+# import pytest
+# from api.apicore import _connect as _connect_
+# from api.apicore import _config as _config_
 
 
-"""
-Setup the testing construct to ease future testing cases
-"""
+# """
+# Setup the testing construct to ease future testing cases
+# """
 
 
-@pytest.fixture()
-def config():
-    _config = _config_.ConfigSingleton()
-    _config.config["MOCK"] = True  # turn on mock mode
-    yield _config
+# @pytest.fixture()
+# def config():
+#     _config = _config_.ConfigSingleton()
+#     _config.config["MOCK"] = True  # turn on mock mode
+#     yield _config
 
 
-@pytest.fixture
-def test_s3_conn(config):
-    yield _connect_.get_object("AWSS3")
-
-
-@pytest.fixture
-def test_ec2_conn(config):
-    yield _connect_.get_object("AWSEC2")
-
+# @pytest.fixture
+# def test_s3_conn(config):
+#     yield _connect_.get_object("AWSS3")
 
 
+# @pytest.fixture
+# def test_ec2_conn(config):
+#     yield _connect_.get_object("AWSEC2")

lionagi/tests/api/aws/test_aws_s3.py

@@ -1,7 +1,6 @@
-from moto import mock_aws
+# from moto import mock_aws
 
 
-@mock_aws
-def test_awss3_list_bucket_pass(test_s3_conn):
-    assert test_s3_conn.list_bucket_names() == []
-
+# @mock_aws
+# def test_awss3_list_bucket_pass(test_s3_conn):
+#     assert test_s3_conn.list_bucket_names() == []

lionagi/tests/test_core/generic/test_structure.py

@@ -1,4 +1,4 @@
-#TODO
+# TODO
 # import unittest
 # from lionagi.core.generic.structure import *
 #
@@ -191,4 +191,4 @@
 #         self.assertEqual(edge.label, "test_label")
 #
 # if __name__ == "__main__":
-#     unittest.main()
+#     unittest.main()

lionagi/tests/test_core/graph/test_graph.py

@@ -1,4 +1,4 @@
-#TODO
+# TODO
 # import unittest
 # from unittest.mock import MagicMock, patch
 # from lionagi.core.graph.graph import Graph

lionagi/tests/test_core/graph/test_tree.py

@@ -1,4 +1,4 @@
-#TODO
+# TODO
 # import unittest
 # from lionagi.core.graph.tree import TreeNode, Tree
 #

lionagi/tests/test_core/mail/test_mail.py

@@ -11,7 +11,7 @@ More to be added
 
 def trim_timestamp_to_day(timestamp_str):
     dt = datetime.fromisoformat(timestamp_str)
-    return dt.strftime('%Y-%m-%d')
+    return dt.strftime("%Y-%m-%d")
 
 
 class MockPackage(Package):
@@ -22,7 +22,9 @@ class MockPackage(Package):
 def mail():
     """Fixture to create a Mail instance with a MockPackage."""
     package = MockPackage()
-    mail_instance = Mail(timestamp=datetime.today().strftime('%Y-%m-%d'), package=package)
+    mail_instance = Mail(
+        timestamp=datetime.today().strftime("%Y-%m-%d"), package=package
+    )
     return mail_instance
 
 
@@ -31,9 +33,6 @@ def test_mail_category(mail):
     assert mail.category is None
 
 
-
-
-
 # from lionagi.core.generic.mail import *
 #
 # import unittest

lionagi/tests/test_core/test_structure/test_base_structure.py

@@ -1,4 +1,4 @@
-#TODO
+# TODO
 
 # import unittest
 # from unittest.mock import MagicMock, patch

lionagi/tests/test_core/test_structure/test_graph.py

@@ -1,4 +1,4 @@
-#TODO
+# TODO
 # import unittest
 # from unittest.mock import MagicMock, patch
 # from lionagi.core.generic import BaseNode, Edge

lionagi/tests/test_core/test_structure/test_tree.py

@@ -1,4 +1,4 @@
-#TODO
+# TODO
 # import unittest
 # from lionagi.new.schema.todo.tree import Tree
 # from lionagi.core.generic import TreeNode

lionagi/version.py

@@ -1 +1 @@
-__version__ = "0.2.4"
+__version__ = "0.2.6"

{lionagi-0.2.4.dist-info → lionagi-0.2.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lionagi
-Version: 0.2.4
+Version: 0.2.6
 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**