fabricatio 0.2.0.dev18__cp312-cp312-win_amd64.whl → 0.2.0.dev20__cp312-cp312-win_amd64.whl

fabricatio/__init__.py

@@ -11,6 +11,7 @@ from fabricatio.models.task import Task
 from fabricatio.models.tool import ToolBox
 from fabricatio.models.utils import Message, Messages
 from fabricatio.parser import Capture, CodeBlockCapture, JsonCapture, PythonCapture
+from fabricatio.toolboxes import arithmetic_toolbox, basic_toolboxes, fs_toolbox, task_toolbox
 
 __all__ = [
     "Action",
@@ -25,8 +26,12 @@ __all__ = [
     "Task",
     "ToolBox",
     "WorkFlow",
+    "arithmetic_toolbox",
+    "basic_toolboxes",
     "env",
+    "fs_toolbox",
     "logger",
     "magika",
+    "task_toolbox",
     "template_manager",
 ]

fabricatio/_rust.pyi

@@ -2,40 +2,30 @@ from pathlib import Path
 from typing import Any, Dict, List, Optional
 
 class TemplateManager:
-    def __init__(self, template_dirs: List[Path], suffix: Optional[str] = None) -> None:
+    """TemplateManager class for managing handlebars templates."""
+    def __init__(
+        self, template_dirs: List[Path], suffix: Optional[str] = None, active_loading: Optional[bool] = None
+    ) -> None:
         """Initialize the template manager.
 
         Args:
             template_dirs (List[Path]): A list of paths to directories containing templates.
-            suffix (str, optional): The suffix of template files. None means 'hbs' suffix .
+            suffix (str, optional): The suffix of template files. None means 'hbs' suffix.
+            active_loading (bool, optional): Whether to enable active loading of templates.
         """
 
     @property
     def template_count(self) -> int:
         """Get the number of templates discovered."""
 
-    @property
-    def templates(self) -> List[str]:
-        """Get a list of template names."""
-
-    def get_template(self, name: str) -> str:
-        """Get a template by name.
-
-        Args:
-            name (str): The name of the template to retrieve.
-
-        Returns:
-            str: The template content.
-        """
-
-    def get_template_source(self, name: str) -> str:
+    def get_template_source(self, name: str) -> Optional[str]:
         """Get the source path of a template by name.
 
         Args:
             name (str): The name of the template to retrieve.
 
         Returns:
-            str: The source path of the template.
+            Optional[str]: The source path of the template.
         """
 
     def discover_templates(self) -> None:
@@ -51,3 +41,13 @@ class TemplateManager:
         Returns:
             str: The rendered template.
         """
+
+def blake3_hash(content: bytes) -> str:
+    """Calculate the BLAKE3 hash of the given data.
+
+    Args:
+        content (bytes): The data to hash.
+
+    Returns:
+        str: The BLAKE3 hash of the data.
+    """

fabricatio/_rust_instances.py

@@ -1,4 +1,8 @@
 from fabricatio._rust import TemplateManager
 from fabricatio.config import configs
 
-template_manager = TemplateManager(template_dirs=configs.templates.template_dir, suffix=configs.templates.template_suffix)
+template_manager = TemplateManager(
+    template_dirs=configs.templates.template_dir,
+    suffix=configs.templates.template_suffix,
+    active_loading=configs.templates.active_loading,
+)

fabricatio/actions/communication.py

@@ -12,4 +12,4 @@ class Talk(Action):
 
     async def _execute(self, task_input: Task[str], **_) -> str:
         """Execute the action."""
-        return await self.aask(task_input.briefing, system_message=task_input.generate_prompt())
+        return await self.aask(task_input.briefing, system_message=task_input.dependencies_prompt())

fabricatio/config.py

@@ -63,7 +63,7 @@ class LLMConfig(BaseModel):
     temperature: NonNegativeFloat = Field(default=1.0)
     """The temperature of the LLM model. Controls randomness in generation. Set to 1.0 as per request."""
 
-    stop_sign: str | List[str] = Field(default=("\n\n", "User:"))
+    stop_sign: str | List[str] = Field(default=("\n\n\n", "User:"))
     """The stop sign of the LLM model. No default stop sign specified."""
 
     top_p: NonNegativeFloat = Field(default=0.35)
@@ -115,6 +115,15 @@ class DebugConfig(BaseModel):
     log_file: FilePath = Field(default=rf"{ROAMING_DIR}\fabricatio.log")
     """The log file of the application."""
 
+    rotation: int = Field(default=1)
+    """The rotation of the log file. in weeks."""
+
+    retention: int = Field(default=2)
+    """The retention of the log file. in weeks."""
+
+    streaming_visible: bool = Field(default=False)
+    """Whether to print the llm output when streaming."""
+
 
 class TemplateConfig(BaseModel):
     """Template configuration class."""
@@ -124,10 +133,30 @@ class TemplateConfig(BaseModel):
         default_factory=lambda: [DirectoryPath(r".\templates"), DirectoryPath(rf"{ROAMING_DIR}\templates")]
     )
     """The directory containing the templates."""
+    active_loading: bool = Field(default=False)
+    """Whether to enable active loading of templates."""
 
     template_suffix: str = Field(default="hbs", frozen=True)
     """The suffix of the templates."""
 
+    propose_task_template: str = Field(default="propose_task")
+    """The name of the propose task template which will be used to propose a task."""
+
+    draft_tool_usage_code_template: str = Field(default="draft_tool_usage_code")
+    """The name of the draft tool usage code template which will be used to draft tool usage code."""
+
+    make_choice_template: str = Field(default="make_choice")
+    """The name of the make choice template which will be used to make a choice."""
+
+    make_judgment_template: str = Field(default="make_judgment")
+    """The name of the make judgment template which will be used to make a judgment."""
+
+    dependencies_template: str = Field(default="dependencies")
+    """The name of the dependencies template which will be used to manage dependencies."""
+
+    task_briefing_template: str = Field(default="task_briefing")
+    """The name of the task briefing template which will be used to brief a task."""
+
 
 class MagikaConfig(BaseModel):
     """Magika configuration class."""

fabricatio/fs/curd.py

@@ -5,10 +5,11 @@ import subprocess
 from pathlib import Path
 from typing import Union
 
-from fabricatio.decorators import depend_on_external_cmd
+from fabricatio.decorators import confirm_to_execute, depend_on_external_cmd
 from fabricatio.journal import logger
 
 
+@confirm_to_execute
 def copy_file(src: Union[str, Path], dst: Union[str, Path]) -> None:
     """Copy a file from source to destination.
 
@@ -28,6 +29,7 @@ def copy_file(src: Union[str, Path], dst: Union[str, Path]) -> None:
         raise
 
 
+@confirm_to_execute
 def move_file(src: Union[str, Path], dst: Union[str, Path]) -> None:
     """Move a file from source to destination.
 
@@ -47,6 +49,7 @@ def move_file(src: Union[str, Path], dst: Union[str, Path]) -> None:
         raise
 
 
+@confirm_to_execute
 def delete_file(file_path: Union[str, Path]) -> None:
     """Delete a file.
 
@@ -65,6 +68,7 @@ def delete_file(file_path: Union[str, Path]) -> None:
         raise
 
 
+@confirm_to_execute
 def create_directory(dir_path: Union[str, Path], parents: bool = True, exist_ok: bool = True) -> None:
     """Create a directory.
 
@@ -81,6 +85,7 @@ def create_directory(dir_path: Union[str, Path], parents: bool = True, exist_ok:
         raise
 
 
+@confirm_to_execute
 @depend_on_external_cmd(
     "erd",
     "Please install `erd` using `cargo install erdtree` or `scoop install erdtree`.",
@@ -92,6 +97,7 @@ def tree(dir_path: Union[str, Path]) -> str:
     return subprocess.check_output(("erd", dir_path.as_posix()), encoding="utf-8")  # noqa: S603
 
 
+@confirm_to_execute
 def delete_directory(dir_path: Union[str, Path]) -> None:
     """Delete a directory and its contents.
 

fabricatio/fs/readers.py

@@ -1,7 +1,24 @@
 """Filesystem readers for Fabricatio."""
 
+from pathlib import Path
+
 from magika import Magika
 
 from fabricatio.config import configs
 
 magika = Magika(model_dir=configs.magika.model_dir)
+
+
+def safe_text_read(path: Path) -> str:
+    """Safely read the text from a file.
+
+    Args:
+        path (Path): The path to the file.
+
+    Returns:
+        str: The text from the file.
+    """
+    try:
+        return path.read_text(encoding="utf-8")
+    except (UnicodeDecodeError, IsADirectoryError, FileNotFoundError):
+        return ""

fabricatio/journal.py

@@ -3,17 +3,22 @@
 import sys
 
 from loguru import logger
-from rich import traceback
+from rich import pretty, traceback
 
 from fabricatio.config import configs
 
+pretty.install()
 traceback.install()
 logger.remove()
 logger.add(
-    configs.debug.log_file, level=configs.debug.log_level, rotation="1 weeks", retention="1 month", compression="zip"
+    configs.debug.log_file,
+    level=configs.debug.log_level,
+    rotation=f"{configs.debug.rotation} weeks",
+    retention=f"{configs.debug.retention} weeks",
 )
 logger.add(sys.stderr, level=configs.debug.log_level)
 
+
 if __name__ == "__main__":
     logger.debug("This is a trace message.")
     logger.info("This is an information message.")

fabricatio/models/action.py

@@ -3,17 +3,17 @@
 import traceback
 from abc import abstractmethod
 from asyncio import Queue
-from typing import Any, Dict, Self, Tuple, Type, Unpack
+from typing import Any, Dict, Self, Tuple, Type, Union, Unpack
 
 from fabricatio.journal import logger
-from fabricatio.models.advanced import ProposeTask
+from fabricatio.models.advanced import 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(ProposeTask, ToolBoxUsage):
+class Action(HandleTask, ProposeTask):
     """Class that represents an action to be executed in a workflow."""
 
     personality: str = Field(default="")
@@ -52,7 +52,7 @@ class Action(ProposeTask, ToolBoxUsage):
         return f"# The action you are going to perform: \n{super().briefing}"
 
 
-class WorkFlow[A: Type[Action] | Action](WithBriefing, ToolBoxUsage):
+class WorkFlow[A: Union[Type[Action], Action]](WithBriefing, ToolBoxUsage):
     """Class that represents a workflow to be executed in a task."""
 
     _context: Queue[Dict[str, Any]] = PrivateAttr(default_factory=lambda: Queue(maxsize=1))
@@ -102,26 +102,31 @@ class WorkFlow[A: Type[Action] | Action](WithBriefing, ToolBoxUsage):
             task: The task to be served.
         """
         await task.start()
-        await self._init_context()
+        await self._init_context(task)
         current_action = None
         try:
             for step in self._instances:
                 logger.debug(f"Executing step: {step.name}")
-                cxt = await self._context.get()
-                modified_ctx = await step.act(cxt)
+                modified_ctx = await step.act(await self._context.get())
                 await self._context.put(modified_ctx)
                 current_action = step.name
             logger.info(f"Finished executing workflow: {self.name}")
-            await task.finish((await self._context.get()).get(self.task_output_key, None))
+            final_ctx = await self._context.get()
+            if self.task_output_key not in final_ctx:
+                logger.warning(
+                    f"Task output key: {self.task_output_key} not found in the context, None will be returned. You can check if `Action.output_key` is set the same as `WorkFlow.task_output_key`."
+                )
+
+            await task.finish(final_ctx.get(self.task_output_key, None))
         except RuntimeError as e:
             logger.error(f"Error during task: {current_action} execution: {e}")  # Log the exception
             logger.error(traceback.format_exc())  # Add this line to log the traceback
             await task.fail()  # Mark the task as failed
 
-    async def _init_context(self) -> None:
+    async def _init_context[T](self, task: Task[T]) -> None:
         """Initialize the context dictionary for workflow execution."""
         logger.debug(f"Initializing context for workflow: {self.name}")
-        await self._context.put({self.task_input_key: None, **dict(self.extra_init_context)})
+        await self._context.put({self.task_input_key: task, **dict(self.extra_init_context)})
 
     def steps_fallback_to_self(self) -> Self:
         """Set the fallback for each step to the workflow itself."""

fabricatio/models/advanced.py

@@ -16,15 +16,15 @@ from loguru import logger
 from pydantic import PositiveInt, ValidationError
 
 
-class ProposeTask(LLMUsage, WithBriefing):
+class ProposeTask(WithBriefing, LLMUsage):
     """A class that proposes a task based on a prompt."""
 
-    async def propose(
+    async def propose[T](
         self,
         prompt: str,
         max_validations: PositiveInt = 2,
         **kwargs: Unpack[LLMKwargs],
-    ) -> Task:
+    ) -> Task[T]:
         """Asynchronously proposes a task based on a given prompt and parameters.
 
         Parameters:
@@ -52,7 +52,7 @@ class ProposeTask(LLMUsage, WithBriefing):
 
         template_data = {"prompt": prompt, "json_example": Task.json_example()}
         return await self.aask_validate(
-            question=template_manager.render_template("propose_task", template_data),
+            question=template_manager.render_template(configs.templates.propose_task_template, template_data),
             validator=_validate_json,
             system_message=f"# your personal briefing: \n{self.briefing}",
             max_validations=max_validations,
@@ -86,7 +86,7 @@ class HandleTask(WithBriefing, ToolBoxUsage):
 
         return await self.aask_validate(
             question=template_manager.render_template(
-                "draft_tool_usage_code",
+                configs.templates.draft_tool_usage_code_template,
                 {
                     "tool_module_name": configs.toolbox.tool_module_name,
                     "task": task.briefing,

fabricatio/models/events.py

@@ -1,11 +1,11 @@
 """The module containing the Event class."""
 
-from typing import List, Self
+from typing import List, Self, Union
 
 from fabricatio.config import configs
 from pydantic import BaseModel, ConfigDict, Field
 
-type EventLike = str | List[str] | Self
+type EventLike = Union[str, List[str], "Event"]
 
 
 class Event(BaseModel):

fabricatio/models/generic.py

@@ -4,7 +4,10 @@ from pathlib import Path
 from typing import List, Self
 
 import orjson
-from fabricatio.fs.readers import magika
+from fabricatio._rust import blake3_hash
+from fabricatio._rust_instances import template_manager
+from fabricatio.config import configs
+from fabricatio.fs.readers import magika, safe_text_read
 from pydantic import (
     BaseModel,
     ConfigDict,
@@ -96,15 +99,26 @@ class WithDependency(Base):
             self.dependencies.remove(Path(d).as_posix())
         return self
 
-    def generate_prompt(self) -> str:
+    @property
+    def dependencies_prompt(self) -> str:
         """Generate a prompt for the task based on the file dependencies.
 
         Returns:
             str: The generated prompt for the task.
         """
-        contents = [Path(d).read_text("utf-8") for d in self.dependencies]
-        recognized = [magika.identify_path(c) for c in contents]
-        out = ""
-        for r, p, c in zip(recognized, self.dependencies, contents, strict=False):
-            out += f"---\n\n> {p}\n```{r.dl.ct_label}\n{c}\n```\n\n"
-        return out
+        return template_manager.render_template(
+            configs.templates.dependencies_template,
+            {
+                (pth := Path(p)).name: {
+                    "path": pth.as_posix(),
+                    "exists": pth.exists(),
+                    "description": (identity := magika.identify_path(pth)).output.description,
+                    "size": f"{pth.stat().st_size / (1024 * 1024) if pth.exists() and pth.is_file() else 0:.3f} MB",
+                    "content": (text := safe_text_read(pth)),
+                    "lines": len(text.splitlines()),
+                    "language": identity.output.ct_label,
+                    "checksum": blake3_hash(pth.read_bytes()) if pth.exists() and pth.is_file() else "unknown",
+                }
+                for p in self.dependencies
+            },
+        )

fabricatio/models/role.py

@@ -1,6 +1,6 @@
 """Module that contains the Role class."""
 
-from typing import Any, Set
+from typing import Any, Self, Set
 
 from fabricatio.core import env
 from fabricatio.journal import logger
@@ -22,8 +22,22 @@ class Role(ProposeTask, ToolBoxUsage):
     toolboxes: Set[ToolBox] = Field(default=basic_toolboxes)
 
     def model_post_init(self, __context: Any) -> None:
+        """Register the workflows in the role to the event bus."""
+        self.resolve_configuration().register_workflows()
+
+    def register_workflows(self) -> Self:
         """Register the workflows in the role to the event bus."""
         for event, workflow in self.registry.items():
+            logger.debug(
+                f"Registering workflow: `{workflow.name}` for event: `{Event.instantiate_from(event).collapse()}`"
+            )
+            env.on(event, workflow.serve)
+        return self
+
+    def resolve_configuration(self) -> Self:
+        """Resolve the configuration of the role."""
+        for workflow in self.registry.values():
+            logger.debug(f"Resolving config for workflow: `{workflow.name}`")
             (
                 workflow.fallback_to(self)
                 .steps_fallback_to_self()
@@ -32,7 +46,4 @@ class Role(ProposeTask, ToolBoxUsage):
                 .steps_supply_tools_from_self()
             )
 
-            logger.debug(
-                f"Registering workflow: {workflow.name} for event: {event.collapse() if isinstance(event, Event) else event}"
-            )
-            env.on(event, workflow.serve)
+        return self

fabricatio/models/task.py

@@ -7,6 +7,8 @@ from asyncio import Queue
 from enum import Enum
 from typing import Any, List, Optional, Self
 
+from fabricatio._rust_instances import template_manager
+from fabricatio.config import configs
 from fabricatio.core import env
 from fabricatio.journal import logger
 from fabricatio.models.events import Event, EventLike
@@ -72,13 +74,28 @@ class Task[T](WithBriefing, WithJsonExample, WithDependency):
         """Move the task to a new namespace.
 
         Args:
-            new_namespace (List[str]): The new namespace to move the task to.
+            new_namespace (EventLike): The new namespace to move the task to.
 
         Returns:
             Task: The moved instance of the `Task` class.
         """
-        self.namespace = new_namespace
+        logger.debug(f"Moving task `{self.name}` to `{new_namespace}`")
         self._namespace.clear().concat(new_namespace)
+        self.namespace = self._namespace.segments
+        return self
+
+    def nested_move_to(self, new_parent_namespace: EventLike) -> Self:
+        """Move the task to a new namespace by nesting it under the new parent namespace.
+
+        Args:
+            new_parent_namespace (EventLike): The new parent namespace to move the task to.
+
+        Returns:
+            Task: The nested moved instance of the `Task` class.
+        """
+        logger.debug(f"Nested moving task `{self.name}` to `{new_parent_namespace}`")
+        self._namespace.clear().concat(new_parent_namespace).concat(self.namespace)
+        self.namespace = self._namespace.segments
         return self
 
     @classmethod
@@ -199,7 +216,7 @@ class Task[T](WithBriefing, WithJsonExample, WithDependency):
         Returns:
             Task: The running instance of the `Task` class.
         """
-        logger.info(f"Starting task {self.name}")
+        logger.info(f"Starting task `{self.name}`")
         self._status = TaskStatus.Running
         await env.emit_async(self.running_label, self)
         return self
@@ -210,6 +227,7 @@ class Task[T](WithBriefing, WithJsonExample, WithDependency):
         Returns:
             Task: The cancelled instance of the `Task` class.
         """
+        logger.info(f"Cancelling task `{self.name}`")
         self._status = TaskStatus.Cancelled
         await env.emit_async(self.cancelled_label, self)
         return self
@@ -220,7 +238,7 @@ class Task[T](WithBriefing, WithJsonExample, WithDependency):
         Returns:
             Task: The failed instance of the `Task` class.
         """
-        logger.error(f"Task {self.name} failed")
+        logger.info(f"Failing task `{self.name}`")
         self._status = TaskStatus.Failed
         await env.emit_async(self.failed_label, self)
         return self
@@ -231,8 +249,8 @@ class Task[T](WithBriefing, WithJsonExample, WithDependency):
         Returns:
             Task: The published instance of the `Task` class
         """
-        logger.info(f"Publishing task {self.name}")
-        await env.emit_async(self.pending_label, self)
+        logger.info(f"Publishing task `{(label := self.pending_label)}`")
+        await env.emit_async(label, self)
         return self
 
     async def delegate(self) -> T:
@@ -241,8 +259,8 @@ class Task[T](WithBriefing, WithJsonExample, WithDependency):
         Returns:
             T: The output of the task
         """
-        logger.info(f"Delegating task {self.name}")
-        await env.emit_async(self.pending_label, self)
+        logger.info(f"Delegating task `{(label := self.pending_label)}`")
+        await env.emit_async(label, self)
         return await self.get_output()
 
     @property
@@ -252,4 +270,7 @@ class Task[T](WithBriefing, WithJsonExample, WithDependency):
         Returns:
             str: The briefing of the task.
         """
-        return f"{super().briefing}\n{self.goal}"
+        return template_manager.render_template(
+            configs.templates.task_briefing_template,
+            self.model_dump(include={"name", "description", "dependencies", "goal"}),
+        )

fabricatio/models/usages.py

@@ -2,6 +2,7 @@
 
 from typing import Callable, Dict, Iterable, List, Optional, Self, Set, Union, Unpack
 
+import asyncstdlib
 import litellm
 import orjson
 from fabricatio._rust_instances import template_manager
@@ -13,7 +14,13 @@ from fabricatio.models.task import Task
 from fabricatio.models.tool import Tool, ToolBox
 from fabricatio.models.utils import Messages
 from fabricatio.parser import JsonCapture
-from litellm.types.utils import Choices, ModelResponse, StreamingChoices
+from litellm import stream_chunk_builder
+from litellm.types.utils import (
+    Choices,
+    ModelResponse,
+    StreamingChoices,
+)
+from litellm.utils import CustomStreamWrapper
 from pydantic import Field, HttpUrl, NonNegativeFloat, NonNegativeInt, PositiveInt, SecretStr
 
 
@@ -58,7 +65,7 @@ class LLMUsage(Base):
         messages: List[Dict[str, str]],
         n: PositiveInt | None = None,
         **kwargs: Unpack[LLMKwargs],
-    ) -> ModelResponse:
+    ) -> ModelResponse | CustomStreamWrapper:
         """Asynchronously queries the language model to generate a response based on the provided messages and parameters.
 
         Args:
@@ -105,13 +112,23 @@ class LLMUsage(Base):
         Returns:
             List[Choices | StreamingChoices]: A list of choices or streaming choices from the model response.
         """
-        return (
-            await self.aquery(
-                messages=Messages().add_system_message(system_message).add_user_message(question),
-                n=n,
-                **kwargs,
-            )
-        ).choices
+        resp = await self.aquery(
+            messages=Messages().add_system_message(system_message).add_user_message(question),
+            n=n,
+            **kwargs,
+        )
+        if isinstance(resp, ModelResponse):
+            return resp.choices
+        if isinstance(resp, CustomStreamWrapper):
+            if configs.debug.streaming_visible:
+                chunks = []
+                async for chunk in resp:
+                    chunks.append(chunk)
+                    print(chunk.choices[0].delta.content or "", end="")  # noqa: T201
+                return stream_chunk_builder(chunks).choices
+            return stream_chunk_builder(await asyncstdlib.list()).choices
+        logger.critical(err := f"Unexpected response type: {type(resp)}")
+        raise ValueError(err)
 
     async def aask(
         self,
@@ -137,10 +154,8 @@ class LLMUsage(Base):
                     system_message=system_message,
                     **kwargs,
                 )
-            )
-            .pop()
-            .message.content
-        )
+            ).pop()
+        ).message.content
 
     async def aask_validate[T](
         self,
@@ -165,7 +180,7 @@ class LLMUsage(Base):
         Raises:
             ValueError: If the response fails to validate after the maximum number of attempts.
         """
-        for _ in range(max_validations):
+        for i in range(max_validations):
             if (
                 response := await self.aask(
                     question=question,
@@ -173,7 +188,10 @@ class LLMUsage(Base):
                     **kwargs,
                 )
             ) and (validated := validator(response)):
+                logger.debug(f"Successfully validated the response at {i}th attempt. response: \n{response}")
                 return validated
+            logger.debug(f"Failed to validate the response at {i}th attempt. response: \n{response}")
+        logger.error(f"Failed to validate the response after {max_validations} attempts.")
         raise ValueError("Failed to validate the response.")
 
     async def achoose[T: WithBriefing](
@@ -204,7 +222,7 @@ class LLMUsage(Base):
             - Relies on `aask_validate` to implement retry mechanisms with validation.
         """
         prompt = template_manager.render_template(
-            "make_choice",
+            configs.templates.make_choice_template,
             {
                 "instruction": instruction,
                 "options": [m.model_dump(include={"name", "briefing"}) for m in choices],
@@ -264,7 +282,8 @@ class LLMUsage(Base):
 
         return await self.aask_validate(
             question=template_manager.render_template(
-                "make_judgment", {"prompt": prompt, "affirm_case": affirm_case, "deny_case": deny_case}
+                configs.templates.make_judgment_template,
+                {"prompt": prompt, "affirm_case": affirm_case, "deny_case": deny_case},
             ),
             validator=_validate,
             max_validations=max_validations,

fabricatio/parser.py

@@ -85,4 +85,5 @@ class Capture(BaseModel):
 
 JsonCapture = Capture.capture_code_block("json")
 PythonCapture = Capture.capture_code_block("python")
-CodeBlockCapture = Capture.capture_code_block("")
+MarkdownCapture = Capture.capture_code_block("markdown")
+CodeBlockCapture = Capture(pattern="```.*?\n(.*?)\n```")

fabricatio/toolboxes/__init__.py

@@ -4,6 +4,7 @@ from typing import Set
 
 from fabricatio.models.tool import ToolBox
 from fabricatio.toolboxes.arithmetic import arithmetic_toolbox
+from fabricatio.toolboxes.fs import fs_toolbox
 from fabricatio.toolboxes.task import task_toolbox
 
 basic_toolboxes: Set[ToolBox] = {task_toolbox, arithmetic_toolbox}
@@ -11,5 +12,6 @@ basic_toolboxes: Set[ToolBox] = {task_toolbox, arithmetic_toolbox}
 __all__ = [
     "arithmetic_toolbox",
     "basic_toolboxes",
+    "fs_toolbox",
     "task_toolbox",
 ]

{fabricatio-0.2.0.dev18.dist-info → fabricatio-0.2.0.dev20.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fabricatio
-Version: 0.2.0.dev18
+Version: 0.2.0.dev20
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: 3.12
@@ -10,6 +10,7 @@ Classifier: Framework :: Pydantic :: 2
 Classifier: Typing :: Typed
 Requires-Dist: appdirs>=1.4.4
 Requires-Dist: asyncio>=3.4.3
+Requires-Dist: asyncstdlib>=3.13.0
 Requires-Dist: code2prompt
 Requires-Dist: gitpython>=3.1.44
 Requires-Dist: litellm>=1.60.0
@@ -47,26 +48,28 @@ Fabricatio is a powerful framework designed to facilitate the creation and manag
 
 - [Installation](#installation)
 - [Usage](#usage)
-    - [Defining a Task](#defining-a-task)
-    - [Creating an Action](#creating-an-action)
-    - [Assigning a Role](#assigning-a-role)
-    - [Logging](#logging)
+  - [Defining a Task](#defining-a-task)
+  - [Creating an Action](#creating-an-action)
+  - [Assigning a Role](#assigning-a-role)
+  - [Logging](#logging)
 - [Configuration](#configuration)
-    - [LLM Configuration](#llm-configuration)
-    - [Debug Configuration](#debug-configuration)
+  - [LLM Configuration](#llm-configuration)
+  - [Debug Configuration](#debug-configuration)
 - [Examples](#examples)
-    - [Simple Task Example](#simple-task-example)
-    - [Complex Workflow Example](#complex-workflow-example)
+  - [Simple Task Example](#simple-task-example)
+  - [Complex Workflow Example](#complex-workflow-example)
 - [Contributing](#contributing)
 - [License](#license)
 
 ## Installation
+
 To install Fabricatio, you can use pip:
 
 ```bash
 pip install fabricatio
 ```
 
+
 Alternatively, you can clone the repository and install it manually:
 
 ```bash
@@ -230,4 +233,110 @@ Contributions to Fabricatio are welcome! Please submit a pull request with your
 ## License
 
 Fabricatio is licensed under the MIT License. See the [LICENSE](LICENSE) file for more information.
+
+---
+
+### Additional Features and Modules
+
+#### Advanced Models and Functionalities
+
+The `advanced.py` module provides advanced models and functionalities for handling complex tasks and workflows.
+
+```python
+from fabricatio.models.advanced import ProposeTask, HandleTask
+
+class ProposeTaskExample(ProposeTask):
+    pass
+
+class HandleTaskExample(HandleTask):
+    pass
+```
+
+
+#### Toolboxes
+
+Fabricatio includes various toolboxes for different types of operations. For example, the `arithmetic.py` toolbox provides arithmetic operations.
+
+```python
+from fabricatio.toolboxes.arithmetic import add, subtract, multiply, divide
+
+result = add(1, 2)
+print(result)  # Output: 3
+```
+
+
+#### File System Operations
+
+The `fs.py` toolbox offers tools for file system operations such as copying, moving, deleting files, and creating directories.
+
+```python
+from fabricatio.toolboxes.fs import copy_file, move_file, delete_file, create_directory
+
+copy_file("source.txt", "destination.txt")
+move_file("old_location.txt", "new_location.txt")
+delete_file("file_to_delete.txt")
+create_directory("new_directory")
+```
+
+
+#### Logging Setup
+
+The logging setup in Fabricatio is handled by the `journal.py` module, which configures Loguru for logging.
+
+```python
+from fabricatio.journal import logger
+
+logger.debug("This is a debug message.")
+logger.info("This is an info message.")
+logger.success("This is a success message.")
+logger.warning("This is a warning message.")
+logger.error("This is an error message.")
+logger.critical("This is a critical message.")
+```
+
+
+#### Configuration Management
+
+The configuration management in Fabricatio is handled by the `config.py` module, which uses Pydantic for defining and validating configurations.
+
+```python
+from fabricatio.config import Settings, LLMConfig, DebugConfig
+
+settings = Settings()
+llm_config = LLMConfig(api_endpoint="https://api.example.com")
+debug_config = DebugConfig(log_level="DEBUG", log_file="fabricatio.log")
+```
+
+
+#### Testing
+
+Fabricatio includes a suite of test cases to ensure the stability and correctness of the codebase. The tests are located in the `tests` directory and cover various modules and functionalities.
+
+```python
+# Example of a test case for the config module
+import pytest
+from fabricatio.config import DebugConfig
+
+def test_debug_config_initialization():
+    temp_log_file = "fabricatio.log"
+    debug_config = DebugConfig(log_level="DEBUG", log_file=temp_log_file)
+    assert debug_config.log_level == "DEBUG"
+    assert str(debug_config.log_file) == temp_log_file
+```
+
+
+---
+
+### Conclusion
+
+Fabricatio is a versatile and powerful framework for managing tasks, actions, and workflows. It provides a robust set of tools and features to facilitate task automation and orchestration. Whether you're building a simple script or a complex application, Fabricatio has the capabilities to meet your needs.
+
+For more detailed information and examples, please refer to the [official documentation](https://fabricatio.readthedocs.io).
+
+---
+
+If you have any questions or need further assistance, feel free to reach out to the community or open an issue on the GitHub repository.
+
+Happy coding!
+
 

fabricatio-0.2.0.dev20.dist-info/RECORD

@@ -0,0 +1,35 @@
+fabricatio-0.2.0.dev20.dist-info/METADATA,sha256=YliY1ygZl8El9BvZ9DCsdSUUwteSNDhX3i-vyXJ6yXo,9664
+fabricatio-0.2.0.dev20.dist-info/WHEEL,sha256=tpW5AN9B-9qsM9WW2FXG2r193YXiqexDadpKp0A2daI,96
+fabricatio-0.2.0.dev20.dist-info/licenses/LICENSE,sha256=do7J7EiCGbq0QPbMAL_FqLYufXpHnCnXBOuqVPwSV8Y,1088
+fabricatio/actions/communication.py,sha256=uNZY0YBJzWNPWjoKeX5Gks9MI4k9uOwjTuvkYzfm6ws,488
+fabricatio/actions/transmission.py,sha256=fU-xL8fDG3oRDD9x7Q94OU2Sb9G6xQfrj5IMlMYsgiM,1240
+fabricatio/actions/__init__.py,sha256=eFmFVPQvtNgFynIXBVr3eP-vWQDWCPng60YY5LXvZgg,115
+fabricatio/config.py,sha256=FsmqfYJBwn_i2luiMFd1K04OOGMU3em9hBrdQprsNtk,10166
+fabricatio/core.py,sha256=yQK2ZrbPYDJOaNDp0Bky3muTkB-ZaQ1ld_Qfflm2dY0,5938
+fabricatio/decorators.py,sha256=XCnFmBoQMBB2rrJe58Kg8UP3lAyWwH9cGL7myh4FJNA,2630
+fabricatio/fs/curd.py,sha256=wBBAxZqTNdjLL5y0b61_YoAT9X0cQDhokrk96muqiA4,3569
+fabricatio/fs/readers.py,sha256=eDL9QhKEd6xfJBZaiwraObpGJWWpUd9NVYHoIvWgVqY,551
+fabricatio/fs/__init__.py,sha256=lWcKYg0v3mv2LnnSegOQaTtlVDODU0vtw_s6iKU5IqQ,122
+fabricatio/journal.py,sha256=siqimKF0M_QaaOCMxtjr_BJVNyUIAQWILzE9Q4T6-7c,781
+fabricatio/models/action.py,sha256=3UWSyLjJEqGcvw8QcrbgIuXzNaxfboZgcA_5IACkbhQ,5658
+fabricatio/models/advanced.py,sha256=VJJTWItMC-fDBdpJ7wx2tAlNq9NsXceexAb8AHZNL1E,4725
+fabricatio/models/events.py,sha256=mrihNEFgQ5o7qFWja1z_qX8dnaTLwPBoJdVlzxQV5oM,2719
+fabricatio/models/generic.py,sha256=_RR3Z20AjY7L9FbXVL49Is7lf1qdMEVX9kjnF7HAGyQ,4108
+fabricatio/models/kwargs_types.py,sha256=lSZAxOnhFdQwRkm-NrbJVMSyBbfdeuVNx807LvJpEOo,901
+fabricatio/models/role.py,sha256=Uqkqp06J3dSvrniGouonRA1ZRWjj8bKUGQwhuDq3b3A,1856
+fabricatio/models/task.py,sha256=8xBHQv1GifvSFRQSPQqUBBA33wCDhM9qIvP2CG1WG04,9215
+fabricatio/models/tool.py,sha256=mD0rlG7hpMxTd_pI6JfRxAVsqPaXGdw1VLThAlbrqFE,6385
+fabricatio/models/usages.py,sha256=ZXjVq4QvBqWMbybY41knDStUgYC-MOvuPnMl26qvGtA,21787
+fabricatio/models/utils.py,sha256=i_kpcQpct04mQFk1nbcVGV-pl1YThWu4Qk3wbewzKkc,2535
+fabricatio/parser.py,sha256=R4dKANSacrt6StUWtlHyVVvFLivlXmdGD3v6HI62xh8,3358
+fabricatio/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+fabricatio/toolboxes/arithmetic.py,sha256=WLqhY-Pikv11Y_0SGajwZx3WhsLNpHKf9drzAqOf_nY,1369
+fabricatio/toolboxes/fs.py,sha256=q5weqzAPv4RddC73blEeYYMTeXoIZjHuK55Us1yVlj0,455
+fabricatio/toolboxes/task.py,sha256=kU4a501awIDV7GwNDuSlK3_Ym-5OhCp5sS-insTmUmQ,269
+fabricatio/toolboxes/__init__.py,sha256=b13KmASO8q5fBLwew964fn9oH86ER5g-S1PgA4fZ_xs,482
+fabricatio/_rust.pyi,sha256=0wCqtwWkVxxoqprvk8T27T8QYKIAKHS7xgsmdMNjQKc,1756
+fabricatio/_rust_instances.py,sha256=dl0-yZ4UvT5g20tQgnPJpmqtkjFGXNG_YK4eLfi_ugQ,279
+fabricatio/__init__.py,sha256=opIrN8lGyT-h2If4Qez0bRuWBa3uIT9GsM9CZy7_XJ0,1100
+fabricatio/_rust.cp312-win_amd64.pyd,sha256=fT-yxY85rLQmuH7LJvj0aDrrwx1AKzfeWjfseDsRifo,1256960
+fabricatio-0.2.0.dev20.data/scripts/tdown.exe,sha256=Mvfseba7QyujspdgUXx8VTzBIgGuMlpoVgcRzwrK2sM,3379200
+fabricatio-0.2.0.dev20.dist-info/RECORD,,

fabricatio-0.2.0.dev18.dist-info/RECORD

@@ -1,35 +0,0 @@
-fabricatio-0.2.0.dev18.dist-info/METADATA,sha256=iasqYxUndNOxKQXzDKo53YvvgDA9l9QujIHcGx_8h30,6386
-fabricatio-0.2.0.dev18.dist-info/WHEEL,sha256=tpW5AN9B-9qsM9WW2FXG2r193YXiqexDadpKp0A2daI,96
-fabricatio-0.2.0.dev18.dist-info/licenses/LICENSE,sha256=do7J7EiCGbq0QPbMAL_FqLYufXpHnCnXBOuqVPwSV8Y,1088
-fabricatio/actions/communication.py,sha256=wyc1DfhVwrwadhYG_auI9ykGXM0fwVYNwxpp91sSfiU,484
-fabricatio/actions/transmission.py,sha256=fU-xL8fDG3oRDD9x7Q94OU2Sb9G6xQfrj5IMlMYsgiM,1240
-fabricatio/actions/__init__.py,sha256=eFmFVPQvtNgFynIXBVr3eP-vWQDWCPng60YY5LXvZgg,115
-fabricatio/config.py,sha256=qT_ArV1m5CVF6No-WaUmB-Jgg_tAZ1e50A4_43x5DgE,8802
-fabricatio/core.py,sha256=yQK2ZrbPYDJOaNDp0Bky3muTkB-ZaQ1ld_Qfflm2dY0,5938
-fabricatio/decorators.py,sha256=XCnFmBoQMBB2rrJe58Kg8UP3lAyWwH9cGL7myh4FJNA,2630
-fabricatio/fs/curd.py,sha256=DulsGzat7jcR08WpP8yS1-VcDXEaaBeqnO_IZwn5kOY,3423
-fabricatio/fs/readers.py,sha256=lwAoOGafusBWyTZeN11oXJc224nPZrfafVHMmGSajkY,167
-fabricatio/fs/__init__.py,sha256=lWcKYg0v3mv2LnnSegOQaTtlVDODU0vtw_s6iKU5IqQ,122
-fabricatio/journal.py,sha256=z5K5waad9xmGr1hGrqSgFDRH3wiDQ5Oqfe0o98DaM-k,707
-fabricatio/models/action.py,sha256=YbfCZc5pWzfSsOATMaoPNwQH8uR7AOdjzqo289XVnrE,5301
-fabricatio/models/advanced.py,sha256=DTezOSrd9lni2Xjccu3BOhEdxGW3N0lsHnJIbtqLr34,4669
-fabricatio/models/events.py,sha256=p9uXXQNWmDcDvGuiCzeVudMC1cdwB0zSdf5FX8mEBnY,2704
-fabricatio/models/generic.py,sha256=bxH0TidKhDN8n8LOCHjPx3ViIclztHGao1N6YM1IGuQ,3425
-fabricatio/models/kwargs_types.py,sha256=lSZAxOnhFdQwRkm-NrbJVMSyBbfdeuVNx807LvJpEOo,901
-fabricatio/models/role.py,sha256=O0hMOGpSXFLPx2A5XhRis7BtPebG-EgT9tY9P5-8p4Y,1417
-fabricatio/models/task.py,sha256=vVOGwnT-oaIIgjemEa0riMMz_CpWdCasie2NSIRvTKM,8188
-fabricatio/models/tool.py,sha256=mD0rlG7hpMxTd_pI6JfRxAVsqPaXGdw1VLThAlbrqFE,6385
-fabricatio/models/usages.py,sha256=SHY8DHNO8poTFGRmKc4AWXvKzLXS540qw8SBWDCjSbg,20719
-fabricatio/models/utils.py,sha256=i_kpcQpct04mQFk1nbcVGV-pl1YThWu4Qk3wbewzKkc,2535
-fabricatio/parser.py,sha256=cpJtH711Lzl1JmwwJ1Jj3ZCzZxIblwrkl18QXLay2fE,3293
-fabricatio/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-fabricatio/toolboxes/arithmetic.py,sha256=WLqhY-Pikv11Y_0SGajwZx3WhsLNpHKf9drzAqOf_nY,1369
-fabricatio/toolboxes/fs.py,sha256=q5weqzAPv4RddC73blEeYYMTeXoIZjHuK55Us1yVlj0,455
-fabricatio/toolboxes/task.py,sha256=kU4a501awIDV7GwNDuSlK3_Ym-5OhCp5sS-insTmUmQ,269
-fabricatio/toolboxes/__init__.py,sha256=xWObAUPasJjBw0pY9V_XTA9fTavdQkCZSYi7-CTfOqs,415
-fabricatio/_rust.pyi,sha256=rZd_143JcGrLtRF-r5H1rErp8jQxH74n8ItA5FrohT0,1639
-fabricatio/_rust_instances.py,sha256=P5yKVZ9M9CC4ryI1G2rq4mHL96iu6DW_85RMTX8onpA,211
-fabricatio/__init__.py,sha256=C9r6OVyMBb8IqwERNUq8lKDLe4BqN7fiu-O4TsXZ5xU,913
-fabricatio/_rust.cp312-win_amd64.pyd,sha256=zcOGopnAj1DXcPzdR7LV8NYWoumcl5jBeIZ3zC9EwXg,1132032
-fabricatio-0.2.0.dev18.data/scripts/tdown.exe,sha256=-wy8nh24fWDJ_M6NpAPqzQPJ2rIoStthYT1uFy4vCfU,3383808
-fabricatio-0.2.0.dev18.dist-info/RECORD,,