fabricatio 0.2.0.dev19__cp312-cp312-manylinux_2_34_x86_64.whl → 0.2.0.dev20__cp312-cp312-manylinux_2_34_x86_64.whl

fabricatio/_rust.pyi

@@ -1,5 +1,5 @@
 from pathlib import Path
-from typing import Dict, List, Optional
+from typing import Any, Dict, List, Optional
 
 class TemplateManager:
     """TemplateManager class for managing handlebars templates."""
@@ -31,12 +31,12 @@ class TemplateManager:
     def discover_templates(self) -> None:
         """Discover templates in the specified directories."""
 
-    def render_template(self, name: str, data: Dict[str, str]) -> str:
+    def render_template(self, name: str, data: Dict[str, Any]) -> str:
         """Render a template with the given name and data.
 
         Args:
             name (str): The name of the template to render.
-            data (Dict[str, str]): The data to pass to the template.
+            data (Dict[str, Any]): The data to pass to the template.
 
         Returns:
             str: The rendered template.

fabricatio/config.py

@@ -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."""
@@ -145,6 +154,9 @@ class TemplateConfig(BaseModel):
     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/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

@@ -111,7 +111,13 @@ class WorkFlow[A: Union[Type[Action], Action]](WithBriefing, ToolBoxUsage):
                 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

fabricatio/models/generic.py

@@ -109,7 +109,8 @@ class WithDependency(Base):
         return template_manager.render_template(
             configs.templates.dependencies_template,
             {
-                (pth := Path(p)).as_posix(): {
+                (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",

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
@@ -214,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
@@ -225,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
@@ -235,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
@@ -267,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,

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fabricatio
-Version: 0.2.0.dev19
+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
@@ -231,3 +234,109 @@ Contributions to Fabricatio are welcome! Please submit a pull request with your
 
 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.dev19.dist-info → fabricatio-0.2.0.dev20.dist-info}/RECORD

@@ -1,6 +1,6 @@
-fabricatio-0.2.0.dev19.dist-info/METADATA,sha256=LxgCQTsmmr_eLZhnCRbH0tjOYgb0Or1qmF5CokmoB5I,6193
-fabricatio-0.2.0.dev19.dist-info/WHEEL,sha256=RIvmwLDYujv60MYBx2jxyP4vdn1DD7X0kBgz1TQvZuc,108
-fabricatio-0.2.0.dev19.dist-info/licenses/LICENSE,sha256=yDZaTLnOi03bi3Dk6f5IjhLUc5old2yOsihHWU0z-i0,1067
+fabricatio-0.2.0.dev20.dist-info/METADATA,sha256=8bFpdHeec5FaYeoDfkXRwqZ5TURu2M8C7TG_BxzWgyk,9363
+fabricatio-0.2.0.dev20.dist-info/WHEEL,sha256=RIvmwLDYujv60MYBx2jxyP4vdn1DD7X0kBgz1TQvZuc,108
+fabricatio-0.2.0.dev20.dist-info/licenses/LICENSE,sha256=yDZaTLnOi03bi3Dk6f5IjhLUc5old2yOsihHWU0z-i0,1067
 fabricatio/fs/__init__.py,sha256=bYE9r8uR0dtknzbg_YaGv_6Wwa27ntkQt0Tl7Kb3HFI,117
 fabricatio/fs/readers.py,sha256=Q-rqgXAsNcVu6WGFTvK70gZFyi088C3cuBdZOhn1oRI,527
 fabricatio/fs/curd.py,sha256=n7Kj93Us201n0LMCKCJJFRDAeGBV9GrAoQgVGcljY_A,3453
@@ -10,26 +10,26 @@ fabricatio/toolboxes/task.py,sha256=9J3W-48CCEUh5PFz0XEC47Hv23Ugn1BEg-J_woQE1UA,
 fabricatio/toolboxes/arithmetic.py,sha256=sSTPkKI6-mb278DwQKFO9jKyzc9kCx45xNH7V6bGBpE,1307
 fabricatio/__init__.py,sha256=h0FgwSAdI2yhxYiuQi46uFvzqkK5LfJLt5XsHZ9rWIo,1063
 fabricatio/core.py,sha256=apwXgI94DCWpGujGlsmXsTZQvJOQMB9llmuUo7ohd-4,5771
-fabricatio/config.py,sha256=uMmnUCR5Tph7pdBfiUHYaDuIYbmj6GXZc8Q9WvH1nsY,9467
+fabricatio/config.py,sha256=9xwT_RBOuSKseYZSHv3Dpe6khVK4ikdEvStaF4bzwe8,9906
 fabricatio/decorators.py,sha256=PQ_oItynybZ-snc3PYkPXgWZd4sVubRJoCIHmqRE6jA,2558
-fabricatio/journal.py,sha256=u6YTKrD9EoebbR8guvAe5zSqWV1nS-BGyKxZR_Sw2ik,684
-fabricatio/models/generic.py,sha256=gfNmspLLBn6Pxk6hBp9Hr4Ux-cXTfvDab08rHD3E888,3946
+fabricatio/journal.py,sha256=bzxZay48ZWI0VIkkDXm4Wc_Cc9lBQYa2VGx3Hxy_PtA,753
+fabricatio/models/generic.py,sha256=Lbjcr98nkcAUsfCvZQyeBEe8WE7uKh-zrR5DBwJf9Zk,3984
 fabricatio/models/kwargs_types.py,sha256=INQIZiCOfGVkRbPmUMp-ZYtSkPwZmgFowid-D3A3TpE,875
-fabricatio/models/action.py,sha256=MShGT54kNEEIuDnSe9X4PC91OERBBiaBVoGWyIhwjuE,5186
-fabricatio/models/task.py,sha256=WHNl0Pm9p1LbIEKky_fy8bHAFmG5CyYn4lk0puzwvr4,8638
+fabricatio/models/action.py,sha256=YJbPd4IVb9pcJXVWIsqnFNMU__KUIZbgKqEQG4Dtjqk,5519
+fabricatio/models/task.py,sha256=-qYGdO5MT9grl1N12l9_va-1PdHSq2QSgDdyf4k5Hvo,8939
 fabricatio/models/role.py,sha256=NaQ6cSrBmY4w-l_PSgefcCzjVANEzfKhH_xYdWwsGMg,1807
 fabricatio/models/tool.py,sha256=pChgISB38r_Htkjd9XtJ_ph3iQKniP_Mzs7_u63wIm0,6213
 fabricatio/models/events.py,sha256=TkMSJYSzRgurbfY2knWIc8gYw7rmaWqB7KeYzy5jeWU,2637
 fabricatio/models/utils.py,sha256=u-eR_v-o0BSFSmCprP3mOyAO-WOyg_P68jjzLjyit_w,2457
-fabricatio/models/usages.py,sha256=nfj3BfkAf2cOOQ22oykQCOdc_-PhsWasXo3Fgejtfws,20633
+fabricatio/models/usages.py,sha256=mCtvBePALA97Fgtdu1G5YlmR-u1MbbkIjSOiRqiA-xo,21312
 fabricatio/models/advanced.py,sha256=UfvJBY31EXpKh9M1OLaFNHQiS3btVyptrglmS-Lx2x8,4606
 fabricatio/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 fabricatio/_rust_instances.py,sha256=JAtO-vL8ihvduf1SHLNf0w7ZSVGCJeIv6zZ9Ekyy1hY,271
 fabricatio/actions/__init__.py,sha256=eLa_5ACZ-FqdrLtOfCHk5nQBxzhIs1kgMIXWmkm2P8Y,110
 fabricatio/actions/communication.py,sha256=Z266NQMB9C7-IDUhXXLT9-M3YKTvB_cRUfccmWYgDtg,473
 fabricatio/actions/transmission.py,sha256=aXbrnVDj-eBtoWTiWvc6KHFV25jKwrCsZvnfFrmkjls,1206
-fabricatio/_rust.pyi,sha256=neXQlLUmyKbGDoy7rUAXFFFvRtSAQq0VWAVLlZMkGWU,1698
+fabricatio/_rust.pyi,sha256=clhcURuiB9zlFo4m3VyoWQ8Xs4tvg6KNHXpF-ok9h4o,1703
 fabricatio/parser.py,sha256=D_b59qn39Qz5oQhTZPRAmhESV4ccvG0tFaYg9AXFFO4,3269
-fabricatio/_rust.cpython-312-x86_64-linux-gnu.so,sha256=p90jrJqzR3Br0aznLrLbl61jgcGNRSsf6jae9jWaQns,1339520
-fabricatio-0.2.0.dev19.data/scripts/tdown,sha256=18GNlTpfgbgPDOseiSRge9he9lJ7wp8Rs-YuBqsm--c,4553776
-fabricatio-0.2.0.dev19.dist-info/RECORD,,
+fabricatio/_rust.cpython-312-x86_64-linux-gnu.so,sha256=fEQZrSros6itNJCaBM_U-gm8BK1yik33Tkr6hxT-wmc,1337080
+fabricatio-0.2.0.dev20.data/scripts/tdown,sha256=mAia7kc6d5zFaN5HJv5JUJJ1WKTk6rdR9-DxtV4Q44k,4554560
+fabricatio-0.2.0.dev20.dist-info/RECORD,,