fabricatio 0.2.1.dev2__tar.gz → 0.2.1.dev3__tar.gz

{fabricatio-0.2.1.dev2 → fabricatio-0.2.1.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fabricatio
-Version: 0.2.1.dev2
+Version: 0.2.1.dev3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: 3.12

fabricatio-0.2.1.dev3/examples/make_a_rating/rating.py

@@ -0,0 +1,84 @@
+"""Example of proposing a task to a role."""
+
+import asyncio
+from typing import Dict, List, Set, Unpack
+
+import orjson
+from fabricatio import Action, JsonCapture, Role, WorkFlow, logger
+from fabricatio.models.events import Event
+from fabricatio.models.task import Task
+
+
+class Rate(Action):
+    """Rate the task."""
+
+    name: str = "rate"
+    output_key: str = "task_output"
+
+    async def _execute(self, to_rate: List[str], rate_topic: str, dimensions: Set[str], **_) -> [Dict[str, float]]:
+        """Rate the task."""
+        return await asyncio.gather(
+            *[
+                self.rate(
+                    target,
+                    rate_topic,
+                    dimensions,
+                )
+                for target in to_rate
+            ]
+        )
+
+
+class WhatToRate(Action):
+    """Figure out what to rate."""
+
+    name: str = "figure out what to rate"
+
+    output_key: str = "to_rate"
+
+    async def _execute(self, task_input: Task, rate_topic: str, **cxt: Unpack) -> List[str]:
+        def _validate(resp: str) -> List[str] | None:
+            if (
+                (cap := JsonCapture.convert_with(resp, orjson.loads)) is not None
+                and isinstance(cap, list)
+                and all(isinstance(i, str) for i in cap)
+            ):
+                return cap
+            return None
+
+        return await self.aask_validate(
+            f"This is task briefing:\n{task_input.briefing}\n\n"
+            f"We are talking about {rate_topic}. you need to extract targets to rate into a the JSON array\n"
+            f"The response SHALL be a JSON array of strings within the codeblock\n"
+            f"# Example\n"
+            f'```json\n["this is a target to rate", "this is another target to rate"]\n```',
+            _validate,
+        )
+
+
+async def main() -> None:
+    """Main function."""
+    role = Role(
+        name="TaskRater",
+        description="A role that can rate tasks.",
+        registry={
+            Event.instantiate_from("rate_food").push_wildcard().push("pending"): WorkFlow(
+                name="Rate food",
+                steps=(WhatToRate, Rate),
+                extra_init_context={
+                    "rate_topic": "If this food is cheap and delicious",
+                    "dimensions": {"taste", "price", "quality", "safety", "healthiness", "freshness"},
+                },
+            ),
+        },
+    )
+    task = await role.propose(
+        "rate for rotten apple, ripen banana, fresh orange, giga-burger, smelly pizza with flies on it, and a boiling instant coffee",
+    )
+    rating = await task.move_to("rate_food").delegate()
+
+    logger.success(f"Result: \n{rating}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

{fabricatio-0.2.1.dev2 → fabricatio-0.2.1.dev3}/examples/minor/hello_fabricatio.py

@@ -5,7 +5,7 @@ from typing import Any
 
 from fabricatio import Action, Role, Task, WorkFlow, logger
 
-task = Task(name="say hello", goal="say hello", description="say hello to the world")
+task = Task(name="say hello", goal=["say hello"], description="say hello to the world")
 
 
 class Hello(Action):

{fabricatio-0.2.1.dev2 → fabricatio-0.2.1.dev3}/examples/propose_task/propose.py

@@ -5,7 +5,7 @@ from typing import Any
 
 from fabricatio import Action, Role, Task, WorkFlow, logger
 
-task = Task(name="say hello", goal="say hello", description="say hello to the world")
+task = Task(name="say hello", goal=["say hello"], description="say hello to the world")
 
 
 class Talk(Action):

{fabricatio-0.2.1.dev2 → fabricatio-0.2.1.dev3}/examples/simple_chat/chat.py

@@ -5,7 +5,7 @@ from typing import Any
 
 from fabricatio import Action, Role, Task, WorkFlow, logger
 
-task = Task(name="say hello", goal="say hello", description="say hello to the world")
+task = Task(name="say hello", goal=["say hello"], description="say hello to the world")
 
 
 class Talk(Action):

{fabricatio-0.2.1.dev2 → fabricatio-0.2.1.dev3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "fabricatio"
-version = "0.2.1-dev2"
+version = "0.2.1-dev3"
 description = "A LLM multi-agent framework."
 readme = "README.md"
 license = { file = "LICENSE" }

{fabricatio-0.2.1.dev2 → fabricatio-0.2.1.dev3}/python/fabricatio/config.py

@@ -163,6 +163,9 @@ class TemplateConfig(BaseModel):
     draft_rating_manual_template: str = Field(default="draft_rating_manual")
     """The name of the draft rating manual template which will be used to draft rating manual."""
 
+    draft_rating_dimensions_template: str = Field(default="draft_rating_dimensions")
+    """The name of the draft rating dimensions template which will be used to draft rating dimensions."""
+
 
 class MagikaConfig(BaseModel):
     """Magika configuration class."""

{fabricatio-0.2.1.dev2 → fabricatio-0.2.1.dev3}/python/fabricatio/models/action.py

@@ -6,14 +6,14 @@ from asyncio import Queue
 from typing import Any, Dict, Self, Tuple, Type, Union, Unpack
 
 from fabricatio.journal import logger
-from fabricatio.models.advanced import HandleTask, ProposeTask
+from fabricatio.models.advanced import GiveRating, HandleTask, ProposeTask
 from fabricatio.models.generic import WithBriefing
 from fabricatio.models.task import Task
 from fabricatio.models.usages import ToolBoxUsage
 from pydantic import Field, PrivateAttr
 
 
-class Action(HandleTask, ProposeTask):
+class Action(HandleTask, ProposeTask, GiveRating):
     """Class that represents an action to be executed in a workflow."""
 
     personality: str = Field(default="")

{fabricatio-0.2.1.dev2 → fabricatio-0.2.1.dev3}/python/fabricatio/models/advanced.py

@@ -13,7 +13,7 @@ from fabricatio.models.tool import Tool, ToolExecutor
 from fabricatio.models.usages import LLMUsage, ToolBoxUsage
 from fabricatio.parser import JsonCapture, PythonCapture
 from loguru import logger
-from pydantic import ValidationError
+from pydantic import NonNegativeInt, ValidationError
 
 
 class ProposeTask(WithBriefing, LLMUsage):
@@ -242,3 +242,48 @@ class GiveRating(WithBriefing, LLMUsage):
             system_message=f"# your personal briefing: \n{self.briefing}",
             **kwargs,
         )
+
+    async def draft_rating_dimensions(
+        self,
+        topic: str,
+        dimensions_count: NonNegativeInt = 0,
+        examples: Optional[List[str]] = None,
+        **kwargs: Unpack[ValidateKwargs],
+    ) -> Set[str]:
+        """Drafts rating dimensions based on a topic.
+
+        Args:
+            topic: The topic for the rating dimensions.
+            dimensions_count: The number of dimensions to draft, 0 means no limit.
+            examples: A list of examples which is rated based on the rating dimensions.
+            **kwargs: Additional keyword arguments for the LLM usage.
+
+        Returns:
+            A set of rating dimensions.
+        """
+
+        def _validator(response: str) -> Set[str] | None:
+            if (
+                (json_data := JsonCapture.convert_with(response, orjson.loads)) is not None
+                and isinstance(json_data, list)
+                and all(isinstance(v, str) for v in json_data)
+                and (dimensions_count == 0 or len(json_data) == dimensions_count)
+            ):
+                return set(json_data)
+            return None
+
+        return await self.aask_validate(
+            question=(
+                template_manager.render_template(
+                    configs.templates.draft_rating_dimensions_template,
+                    {
+                        "topic": topic,
+                        "examples": examples,
+                        "dimensions_count": dimensions_count,
+                    },
+                )
+            ),
+            validator=_validator,
+            system_message=f"# your personal briefing: \n{self.briefing}",
+            **kwargs,
+        )

{fabricatio-0.2.1.dev2 → fabricatio-0.2.1.dev3}/python/fabricatio/models/role.py

@@ -5,14 +5,13 @@ from typing import Any, Self, Set
 from fabricatio.core import env
 from fabricatio.journal import logger
 from fabricatio.models.action import WorkFlow
-from fabricatio.models.advanced import ProposeTask
+from fabricatio.models.advanced import GiveRating, HandleTask, ProposeTask
 from fabricatio.models.events import Event
 from fabricatio.models.tool import ToolBox
-from fabricatio.models.usages import ToolBoxUsage
 from pydantic import Field
 
 
-class Role(ProposeTask, ToolBoxUsage):
+class Role(ProposeTask, HandleTask, GiveRating):
     """Class that represents a role with a registry of events and workflows."""
 
     registry: dict[Event | str, WorkFlow] = Field(...)

{fabricatio-0.2.1.dev2 → fabricatio-0.2.1.dev3}/python/fabricatio/models/task.py

@@ -57,6 +57,9 @@ class Task[T](WithBriefing, WithJsonExample, WithDependency):
     namespace: List[str] = Field(default_factory=list)
     """The namespace of the task, a list of namespace segment, as string."""
 
+    dependencies: List[str] = Field(default_factory=list)
+    """A list of file paths, These file are needed to read or write to meet a specific requirement of this task."""
+
     _output: Queue = PrivateAttr(default_factory=lambda: Queue(maxsize=1))
     """The output queue of the task."""
 

fabricatio-0.2.1.dev3/templates/built-in/draft_rating_dimensions.hbs

@@ -0,0 +1,24 @@
+# Task
+Please draft rating dimensions for the topic: "{{ topic }}".
+Return the dimensions{{#if dimensions_count}} with length of {{ dimensions_count }}{{/if}} in JSON format as a list of strings within code block.
+
+
+# Example:
+returning dimensions{{#if dimensions_count}} with length of 3{{/if}} for the topic "Nice icecream"
+
+----- Start of response example -----
+```json
+[
+    "outlook",
+    "taste",
+    "texture"
+]
+```
+----- End of response example -----
+
+
+# Warning
+- Please ensure that the dimensions are clear and concise.
+- The response SHALL be returned as a JSON object within the codeblock.
+- No additional Explanation is needed.
+{{#if dimensions_count }}- You must return exactly {{ dimensions_count }} dimensions.{{/if}}

fabricatio-0.2.1.dev3/templates/built-in/draft_rating_manual.hbs

@@ -0,0 +1,26 @@
+# Task
+Please draft a rating manual for the following topic and dimensions, which will be used to rate the entity based.
+
+Topic: {{ topic }}
+Dimensions: {{ dimensions }}
+
+Return the rating manual as a JSON object within the codeblock where the keys are the dimensions and the values are the criteria for each dimension.
+
+
+# Example:
+
+topic: Nice icecream
+----- Start of response example -----
+```json
+{
+"outlook": "Appearance of the icecream, how it looks. High score for an ice cream with a beautiful appearance such as vibrant color combinations and topped with fresh fruits or chocolate pieces; low score for an ice cream that looks unappealing, possibly showing signs of melting or covered in ice crystals.",
+"taste": "How the icecream tastes. High score for delicious ice cream with rich and authentic flavors, like vanilla with natural vanilla bean aroma and moderate sweetness; low score for bland or overly sweet ice cream with noticeable artificial flavoring or not matching its claimed flavor.",
+"texture": "How the icecream feels. High score for smooth texture without icy bits, melts smoothly in the mouth providing a creamy experience; low score for rough texture containing many ice crystals, feeling like it has been frozen for too long lacking softness."
+}
+```
+----- end of response example -----
+
+# Warning
+- Please ensure that the rating manual is clear and concise, and that the criteria for each dimension are well-defined.
+- The response SHALL be returned as a JSON object within the codeblock.
+- No additional Explanation is needed.

{fabricatio-0.2.1.dev2 → fabricatio-0.2.1.dev3}/uv.lock

@@ -216,7 +216,7 @@ wheels = [
 
 [[package]]
 name = "fabricatio"
-version = "0.2.1.dev2"
+version = "0.2.1.dev3"
 source = { editable = "." }
 dependencies = [
     { name = "appdirs" },

fabricatio-0.2.1.dev2/templates/built-in/draft_rating_manual.hbs

@@ -1,26 +0,0 @@
-# Task
-Please draft a rating manual for the following topic and dimensions, which will be used to rate the entity based.
-
-Topic: {{ topic }}
-Dimensions: {{ dimensions }}
-
-Return the rating manual as a JSON object within the codeblock where the keys are the dimensions and the values are the criteria for each dimension.
-
-
-# Example:
-
-topic: Nice icecream
------ Start of response example -----
-```json
-{
-  "outlook": "Appearance of the icecream, how it looks, high score for a nice looking icecream",
-  "taste": "How the icecream tastes, high score for a delicious icecream",
-  "texture": "How the icecream feels, high score for a smooth icecream",
-}
-```
------ end of response example -----
-
-# Warning
-- Please ensure that the rating manual is clear and concise, and that the criteria for each dimension are well-defined.
-- The response SHALL be returned as a JSON object within the codeblock.
-- No additional Explanation is needed.