fabricatio 0.1.2__py3-none-any.whl → 0.1.3__py3-none-any.whl

fabricatio/actions/__init__.py

@@ -0,0 +1,5 @@
+"""module for actions."""
+
+from fabricatio.actions.transmission import SendTask
+
+__all__ = ["SendTask"]

fabricatio/actions/transmission.py

@@ -0,0 +1,16 @@
+from typing import List
+
+from fabricatio.journal import logger
+from fabricatio.models.action import Action
+from fabricatio.models.task import Task
+
+
+class SendTask(Action):
+    """Action that sends a task to a user."""
+
+    name: str = "send_task"
+
+    async def _execute(self, send_targets: List[str], send_task: Task, **_) -> None:
+        logger.info(f"Sending task {send_task.name} to {send_targets}")
+        for target in send_targets:
+            await send_task.publish(target)

fabricatio/models/events.py

@@ -50,8 +50,10 @@ class Event(BaseModel):
         self.segments.clear()
         return self
 
-    def concat(self, event: Self) -> Self:
+    def concat(self, event: Self | str) -> Self:
         """Concatenate another event to this event."""
+        if isinstance(event, str):
+            event = Event.from_string(event)
         self.segments.extend(event.segments)
         return self
 

fabricatio/models/task.py

@@ -1,4 +1,4 @@
-"""This module defines the Task class, which represents a task with a status and output.
+"""This module defines the `Task` class, which represents a task with a status and output.
 
 It includes methods to manage the task's lifecycle, such as starting, finishing, cancelling, and failing the task.
 """
@@ -12,11 +12,20 @@ from pydantic import Field, PrivateAttr
 from fabricatio.config import configs
 from fabricatio.core import env
 from fabricatio.journal import logger
+from fabricatio.models.events import Event
 from fabricatio.models.generic import WithBriefing, WithJsonExample
 
 
 class TaskStatus(Enum):
-    """Enum that represents the status of a task."""
+    """An enumeration representing the status of a task.
+
+    Attributes:
+        Pending: The task is pending.
+        Running: The task is currently running.
+        Finished: The task has been successfully completed.
+        Failed: The task has failed.
+        Cancelled: The task has been cancelled.
+    """
 
     Pending = "pending"
     Running = "running"
@@ -26,13 +35,11 @@ class TaskStatus(Enum):
 
 
 class Task[T](WithBriefing, WithJsonExample):
-    """Class that represents a task with a status and output.
+    """A class representing a task with a status and output.
 
     Attributes:
         name (str): The name of the task.
         description (str): The description of the task.
-        _output (Queue): The output queue of the task.
-        _status (TaskStatus): The status of the task.
         goal (str): The goal of the task.
     """
 
@@ -47,6 +54,7 @@ class Task[T](WithBriefing, WithJsonExample):
 
     _output: Queue = PrivateAttr(default_factory=lambda: Queue(maxsize=1))
     """The output queue of the task."""
+
     _status: TaskStatus = PrivateAttr(default=TaskStatus.Pending)
     """The status of the task."""
 
@@ -60,7 +68,7 @@ class Task[T](WithBriefing, WithJsonExample):
             description (str): The description of the task.
 
         Returns:
-            Self: A new instance of the Task class.
+            Task: A new instance of the `Task` class.
         """
         return cls(name=name, goal=goal, description=description)
 
@@ -68,11 +76,11 @@ class Task[T](WithBriefing, WithJsonExample):
         """Update the goal and description of the task.
 
         Args:
-            goal (Optional[str]): The new goal of the task.
-            description (Optional[str]): The new description of the task.
+            goal (str, optional): The new goal of the task.
+            description (str, optional): The new description of the task.
 
         Returns:
-            Self: The updated instance of the Task class.
+            Task: The updated instance of the `Task` class.
         """
         if goal:
             self.goal = goal
@@ -152,7 +160,7 @@ class Task[T](WithBriefing, WithJsonExample):
             output (T): The output of the task.
 
         Returns:
-            Self: The finished instance of the Task class.
+            Task: The finished instance of the `Task` class.
         """
         logger.info(f"Finishing task {self.name}")
         self._status = TaskStatus.Finished
@@ -166,7 +174,7 @@ class Task[T](WithBriefing, WithJsonExample):
         """Mark the task as running.
 
         Returns:
-            Self: The running instance of the Task class.
+            Task: The running instance of the `Task` class.
         """
         logger.info(f"Starting task {self.name}")
         self._status = TaskStatus.Running
@@ -177,7 +185,7 @@ class Task[T](WithBriefing, WithJsonExample):
         """Mark the task as cancelled.
 
         Returns:
-            Self: The cancelled instance of the Task class.
+            Task: The cancelled instance of the `Task` class.
         """
         self._status = TaskStatus.Cancelled
         await env.emit_async(self.cancelled_label, self)
@@ -187,31 +195,43 @@ class Task[T](WithBriefing, WithJsonExample):
         """Mark the task as failed.
 
         Returns:
-            Self: The failed instance of the Task class.
+            Task: The failed instance of the `Task` class.
         """
         logger.error(f"Task {self.name} failed")
         self._status = TaskStatus.Failed
         await env.emit_async(self.failed_label, self)
         return self
 
-    async def publish(self) -> Self:
-        """Publish the task to the environment.
+    async def publish(self, event_namespace: Event | str = "") -> Self:
+        """Publish the task with an optional event namespace.
+
+        Args:
+            event_namespace (Event | str, optional): The event namespace to use. Defaults to an empty string.
 
         Returns:
-            Self: The published instance of the Task class.
+            Task: The published instance of the `Task` class.
         """
+        if isinstance(event_namespace, str):
+            event_namespace = Event.from_string(event_namespace)
+
         logger.info(f"Publishing task {self.name}")
-        await env.emit_async(self.pending_label, self)
+        await env.emit_async(event_namespace.concat(self.pending_label).collapse(), self)
         return self
 
-    async def delegate(self) -> T:
-        """Delegate the task to the environment.
+    async def delegate(self, event_namespace: Event | str = "") -> T:
+        """Delegate the task with an optional event namespace and return the output.
+
+        Args:
+            event_namespace (Event | str, optional): The event namespace to use. Defaults to an empty string.
 
         Returns:
             T: The output of the task.
         """
+        if isinstance(event_namespace, str):
+            event_namespace = Event.from_string(event_namespace)
+
         logger.info(f"Delegating task {self.name}")
-        await env.emit_async(self.pending_label, self)
+        await env.emit_async(event_namespace.concat(self.pending_label).collapse(), self)
         return await self.get_output()
 
     @property

{fabricatio-0.1.2.dist-info → fabricatio-0.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fabricatio
-Version: 0.1.2
+Version: 0.1.3
 Summary: A LLM multi-agent framework.
 Author-email: Whth <zettainspector@foxmail.com>
 License: MIT License

{fabricatio-0.1.2.dist-info → fabricatio-0.1.3.dist-info}/RECORD

@@ -4,16 +4,18 @@ fabricatio/core.py,sha256=B6KBIfBRF023HF0UUaUprEkQd6sT7G_pexGXQ9btJnE,5788
 fabricatio/journal.py,sha256=CW9HePtgTiboOyPTExq9GjG5BseZcbc-S6lxDXrpmv0,667
 fabricatio/parser.py,sha256=On_YUCvOuA0FA_NtDVNJqKp7KEO_sUE89oO_WnkEhQ4,2314
 fabricatio/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+fabricatio/actions/__init__.py,sha256=n3lwq9FPNtvfLu2L1pX4UkwiPITU7luk-b4aMJyjIC8,109
+fabricatio/actions/transmission.py,sha256=Azog4ItVk7aASdYzTwzyckzYG2hDFSXctnA7qp-Qlq0,502
 fabricatio/models/action.py,sha256=M-12dc-nQiNJU6Y9j-dr4Ef3642vRvzHlzxekBepzaU,3358
-fabricatio/models/events.py,sha256=0p42QmNDzmC76DhMwW1H_Mlg15MQ_XjEqkCJc8UkIB8,2055
+fabricatio/models/events.py,sha256=PlavKOD94Q9Q9iPoQPkf9HJdznBtzuSUrPtUoVtkQxU,2143
 fabricatio/models/generic.py,sha256=Sxpx0BO0t85YF5Lwks6F165N6TJsDe7xym28dQG5Mqs,17681
 fabricatio/models/role.py,sha256=jdabuYRXwgvpYoNwvazygDiZHGGQApUIIKltniu78O8,2151
-fabricatio/models/task.py,sha256=0oQeGQ6Rvd_x6ZM5ImtcN2vr0ojFmF6EiWBAMOjledI,6865
+fabricatio/models/task.py,sha256=8IXT192t5EXVHAdHj45DJNdBhdA2xmh36S_vKyEw6y0,7757
 fabricatio/models/tool.py,sha256=UkEp1Nzbl5wZX21q_Z2VkpiJmVDSdoGDzINQniO8hSY,3536
 fabricatio/models/utils.py,sha256=2mgXla9_K3dnRrz6hIKzmltTYPmvDk0MBjjEBkCXTdg,2474
 fabricatio/toolboxes/__init__.py,sha256=bjefmPd7wBaWhbZzdMPXvrjMTeRzlUh_Dev2PUAc124,158
 fabricatio/toolboxes/task.py,sha256=xgyPetm2R_HlQwpzE8YPnBN7QOYLd0-T8E6QPZG1PPQ,204
-fabricatio-0.1.2.dist-info/METADATA,sha256=JoHDaag_cV_OOthFdXHGJyBKlJpyUg4C30lBOvAVA_U,3797
-fabricatio-0.1.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-fabricatio-0.1.2.dist-info/licenses/LICENSE,sha256=do7J7EiCGbq0QPbMAL_FqLYufXpHnCnXBOuqVPwSV8Y,1088
-fabricatio-0.1.2.dist-info/RECORD,,
+fabricatio-0.1.3.dist-info/METADATA,sha256=pDZPikewIva_s6riDaxhlOmvd92ls_JoS_VqETPwDdM,3797
+fabricatio-0.1.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+fabricatio-0.1.3.dist-info/licenses/LICENSE,sha256=do7J7EiCGbq0QPbMAL_FqLYufXpHnCnXBOuqVPwSV8Y,1088
+fabricatio-0.1.3.dist-info/RECORD,,