fabricatio 0.2.0.dev9__cp312-cp312-manylinux_2_34_x86_64.whl → 0.2.0.dev11__cp312-cp312-manylinux_2_34_x86_64.whl

fabricatio/__init__.py

@@ -1,5 +1,6 @@
 """Fabricatio is a Python library for building llm app using event-based agent structure."""
 
+from fabricatio._rust_instances import template_manager
 from fabricatio.core import env
 from fabricatio.fs import magika
 from fabricatio.journal import logger
@@ -10,7 +11,6 @@ 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.templates import templates_manager
 
 __all__ = [
     "Action",
@@ -28,5 +28,5 @@ __all__ = [
     "env",
     "logger",
     "magika",
-    "templates_manager",
+    "template_manager"
 ]

fabricatio/_rust.pyi

@@ -1 +1,53 @@
-def download_templates() -> None: ...
+from pathlib import Path
+from typing import Dict, List, Any
+
+
+class TemplateManager:
+    def __init__(self, template_dirs: List[Path]) -> None:
+        """
+        Initialize the template manager.
+        Args:
+            template_dirs (List[Path]): A list of paths to directories containing 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:
+        """
+        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.
+        """
+
+    def discover_templates(self) -> None:
+        """Discover templates in the specified directories."""
+
+    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, Any]): The data to pass to the template.
+
+        Returns:
+            str: The rendered template.
+        """

fabricatio/_rust_instances.py

@@ -0,0 +1,4 @@
+from fabricatio._rust import TemplateManager
+from fabricatio.config import configs
+
+template_manager = TemplateManager(configs.code2prompt.template_dir)

fabricatio/config.py

@@ -1,6 +1,5 @@
 """Configuration module for the Fabricatio application."""
 
-from pathlib import Path
 from typing import List, Literal, Optional
 
 from appdirs import user_config_dir
@@ -122,7 +121,7 @@ class Code2PromptConfig(BaseModel):
 
     model_config = ConfigDict(use_attribute_docstrings=True)
     template_dir: List[DirectoryPath] = Field(
-        default_factory=lambda: [Path(r".\templates"), Path(rf"{ROAMING_DIR}\templates")]
+        default_factory=lambda: [DirectoryPath(r".\templates"), DirectoryPath(rf"{ROAMING_DIR}\templates")]
     )
     """The directory containing the templates for code2prompt."""
 
@@ -176,12 +175,12 @@ class Settings(BaseSettings):
 
     @classmethod
     def settings_customise_sources(
-        cls,
-        settings_cls: type[BaseSettings],
-        init_settings: PydanticBaseSettingsSource,
-        env_settings: PydanticBaseSettingsSource,
-        dotenv_settings: PydanticBaseSettingsSource,
-        file_secret_settings: PydanticBaseSettingsSource,
+            cls,
+            settings_cls: type[BaseSettings],
+            init_settings: PydanticBaseSettingsSource,
+            env_settings: PydanticBaseSettingsSource,
+            dotenv_settings: PydanticBaseSettingsSource,
+            file_secret_settings: PydanticBaseSettingsSource,
     ) -> tuple[PydanticBaseSettingsSource, ...]:
         """Customize settings sources.
 

fabricatio/core.py

@@ -38,11 +38,11 @@ class Env(BaseModel):
 
     @overload
     def on[**P, R](
-        self,
-        event: str | Event,
-        func: Optional[Callable[P, R]] = None,
-        /,
-        ttl: int = -1,
+            self,
+            event: str | Event,
+            func: Optional[Callable[P, R]] = None,
+            /,
+            ttl: int = -1,
     ) -> Callable[[Callable[P, R]], Callable[P, R]]:
         """
         Registers an event listener with a specific function that listens indefinitely or for a specified number of times.
@@ -58,11 +58,11 @@ class Env(BaseModel):
         ...
 
     def on[**P, R](
-        self,
-        event: str | Event,
-        func: Optional[Callable[P, R]] = None,
-        /,
-        ttl=-1,
+            self,
+            event: str | Event,
+            func: Optional[Callable[P, R]] = None,
+            /,
+            ttl=-1,
     ) -> Callable[[Callable[P, R]], Callable[P, R]] | Self:
         """Registers an event listener with a specific function that listens indefinitely or for a specified number of times.
 
@@ -84,8 +84,8 @@ class Env(BaseModel):
 
     @overload
     def once[**P, R](
-        self,
-        event: str | Event,
+            self,
+            event: str | Event,
     ) -> Callable[[Callable[P, R]], Callable[P, R]]:
         """
         Registers an event listener that listens only once.
@@ -100,9 +100,9 @@ class Env(BaseModel):
 
     @overload
     def once[**P, R](
-        self,
-        event: str | Event,
-        func: Callable[[Callable[P, R]], Callable[P, R]],
+            self,
+            event: str | Event,
+            func: Callable[[Callable[P, R]], Callable[P, R]],
     ) -> Self:
         """
         Registers an event listener with a specific function that listens only once.
@@ -117,9 +117,9 @@ class Env(BaseModel):
         ...
 
     def once[**P, R](
-        self,
-        event: str | Event,
-        func: Optional[Callable[P, R]] = None,
+            self,
+            event: str | Event,
+            func: Optional[Callable[P, R]] = None,
     ) -> Callable[[Callable[P, R]], Callable[P, R]] | Self:
         """Registers an event listener with a specific function that listens only once.
 

fabricatio/models/task.py

@@ -7,6 +7,7 @@ from asyncio import Queue
 from enum import Enum
 from typing import Any, List, Optional, Self
 
+from fabricatio._rust_instances import template_manager
 from fabricatio.core import env
 from fabricatio.journal import logger
 from fabricatio.models.events import Event, EventLike
@@ -273,11 +274,12 @@ class ProposeTask(LLMUsage, WithBriefing):
                 logger.error(f"Failed to parse task from JSON: {e}")
                 return None
 
+        template_data = {
+            "prompt": prompt,
+            "json_example": Task.json_example()
+        }
         return await self.aask_validate(
-            f"{prompt} \n\nBased on requirement above, "
-            f"you need to construct a task to satisfy that requirement in JSON format "
-            f"written like this: \n\n```json\n{Task.json_example()}\n```\n\n"
-            f"No extra explanation needed. ",
+            template_manager.render_template("propose_task", template_data),
             _validate_json,
             system_message=f"# your personal briefing: \n{self.briefing}",
         )

{fabricatio-0.2.0.dev9.dist-info → fabricatio-0.2.0.dev11.dist-info}/METADATA

@@ -1,7 +1,8 @@
 Metadata-Version: 2.4
 Name: fabricatio
-Version: 0.2.0.dev9
+Version: 0.2.0.dev11
 Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Framework :: AsyncIO
@@ -23,7 +24,7 @@ Requires-Dist: regex>=2024.11.6
 Requires-Dist: rich>=13.9.4
 License-File: LICENSE
 Summary: A LLM multi-agent framework.
-Keywords: ai,agents,multi-agent,llm
+Keywords: ai,agents,multi-agent,llm,pyo3
 Author-email: Whth <zettainspector@foxmail.com>
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM

{fabricatio-0.2.0.dev9.dist-info → fabricatio-0.2.0.dev11.dist-info}/RECORD

@@ -1,29 +1,29 @@
-fabricatio-0.2.0.dev9.dist-info/METADATA,sha256=I7x3HxGkHVT7mHN5FsLmHIkuFQ5R1XMdAH-k0V_rtOA,5777
-fabricatio-0.2.0.dev9.dist-info/WHEEL,sha256=RIvmwLDYujv60MYBx2jxyP4vdn1DD7X0kBgz1TQvZuc,108
-fabricatio-0.2.0.dev9.dist-info/licenses/LICENSE,sha256=yDZaTLnOi03bi3Dk6f5IjhLUc5old2yOsihHWU0z-i0,1067
+fabricatio-0.2.0.dev11.dist-info/METADATA,sha256=5XdMpp40bDU_ulMm0r4EfK6wTERGc1Uiz56lzpIMCPM,5824
+fabricatio-0.2.0.dev11.dist-info/WHEEL,sha256=RIvmwLDYujv60MYBx2jxyP4vdn1DD7X0kBgz1TQvZuc,108
+fabricatio-0.2.0.dev11.dist-info/licenses/LICENSE,sha256=yDZaTLnOi03bi3Dk6f5IjhLUc5old2yOsihHWU0z-i0,1067
 fabricatio/fs/__init__.py,sha256=bYE9r8uR0dtknzbg_YaGv_6Wwa27ntkQt0Tl7Kb3HFI,117
 fabricatio/fs/readers.py,sha256=E219Pef7gknltkHjL9iZXwjI841_EPFpNvDG9KBUG3g,118
 fabricatio/toolboxes/__init__.py,sha256=8gd8yPhhQBO0A37w1zqVLfbOG-pyCPrveEtO0MX042A,151
 fabricatio/toolboxes/task.py,sha256=G_nNYT8Sy5ll6YmXES458MUmgJnDSBNwlozyJewdnzY,200
-fabricatio/__init__.py,sha256=EVT4wvmRYDmYMTME_ZWb6U39Kbt5grAtMxfaH3faNXc,877
-fabricatio/core.py,sha256=r7qjG-ACFGgOsiPjWStlNWb0oRnIVvO5MEfMoagiVT8,5699
-fabricatio/config.py,sha256=434wpWNzzmBnLxsj07l7xp7YTUc1--SzdOhQIXE0tuw,7835
+fabricatio/__init__.py,sha256=5z3QgwOGRzH7EIxKNWsZ_A6Ytfcu23nl-r6prZ-Z4cw,880
+fabricatio/core.py,sha256=apwXgI94DCWpGujGlsmXsTZQvJOQMB9llmuUo7ohd-4,5771
+fabricatio/config.py,sha256=Or_PzOzXy_4Rqje1r788sID4bxILXhNC3r7eelcZJy0,7852
 fabricatio/decorators.py,sha256=Y2F5JuN3K5Pas_JPBLQrB_Np0s1U8k2lPvrdtCHtcRE,1799
 fabricatio/journal.py,sha256=u6YTKrD9EoebbR8guvAe5zSqWV1nS-BGyKxZR_Sw2ik,684
-fabricatio/templates.py,sha256=wXEdFlWSP5SAFivDEr8nkvPkcjGTipdpKkwzJcRHANc,1574
 fabricatio/models/generic.py,sha256=1S5guUWnHSpoyNi1D1t4q14e6YNIovI4Goc3kizHd8M,15698
 fabricatio/models/action.py,sha256=PNcm4t34fSaI03ECIOZ8ezvB6PlZkNhJtZFEhtU0Gec,4916
-fabricatio/models/task.py,sha256=W66bRf6au6ysS_ypZ3i4W1mL-WhotfvbhbjX-3CrBj4,9146
+fabricatio/models/task.py,sha256=Li66mMYM4xF3vGH0FHNZv02YQ2KhcU67CSTTcqTrGVU,9124
 fabricatio/models/role.py,sha256=1vhpHRKJ6f7PDBLl6jYRZbxTJyrVhNN3FEEh_HAwU3o,990
 fabricatio/models/tool.py,sha256=jzH34UHIgqEwYdXOQ3Jz_TN0cyj_jLTVBVs0qGl9tQU,3299
 fabricatio/models/events.py,sha256=S0E7l5iQoJYdUn0pSSziHdXqPzIvvLy8ifVPQxXeAho,2570
 fabricatio/models/utils.py,sha256=u-eR_v-o0BSFSmCprP3mOyAO-WOyg_P68jjzLjyit_w,2457
 fabricatio/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+fabricatio/_rust_instances.py,sha256=c3aB95D7qKrA1AxDHD94lYinklAmK5EL6_DDJUEzPB0,153
 fabricatio/actions/__init__.py,sha256=eLa_5ACZ-FqdrLtOfCHk5nQBxzhIs1kgMIXWmkm2P8Y,110
 fabricatio/actions/communication.py,sha256=4hNzyv5_j4HU291dDuBmBEb66VihQPI9SLRVZrxWu1Y,412
 fabricatio/actions/transmission.py,sha256=gLLKoi4keL64uaEkHDxwVrBGciheupQrO2fW3GkjfEw,1156
-fabricatio/_rust.pyi,sha256=yvIFCtXH7p5F-1OA4k1diSTlhpDAShdOoFzRxh9LLtY,38
+fabricatio/_rust.pyi,sha256=EwMPoCZFOCYZ5h_9PZLvO3YrgBEG0Hwae0ux0hcBabM,1487
 fabricatio/parser.py,sha256=2wFvkyAIlFyB_dj8txr6shc-0ik5cz0mUy_Yj_GrqcI,2355
-fabricatio/_rust.cpython-312-x86_64-linux-gnu.so,sha256=zlsqjtyM4uvxykIZOW5JG1RUVCbhC7nlpxzaUgUdB2E,402344
-fabricatio-0.2.0.dev9.data/scripts/tdown,sha256=o5p_FBwrVaRyVry-j3_mYbFLQg5wWcUC4_i-_P7eda8,4527456
-fabricatio-0.2.0.dev9.dist-info/RECORD,,
+fabricatio/_rust.cpython-312-x86_64-linux-gnu.so,sha256=zSQ7Y-Hp7TDAzKenBEEWirLdaVturuUyGbNn1oXtRYs,1191136
+fabricatio-0.2.0.dev11.data/scripts/tdown,sha256=jyItwd3z9dBoOT6EEA3niK0e7AXhhBnL87-T0IBRkYk,4540840
+fabricatio-0.2.0.dev11.dist-info/RECORD,,

fabricatio/templates.py

@@ -1,41 +0,0 @@
-"""A module that manages templates for code generation."""
-
-from typing import Any, Dict, List, Self
-
-from pydantic import BaseModel, ConfigDict, DirectoryPath, Field, FilePath, PrivateAttr
-
-from fabricatio.config import configs
-from fabricatio.journal import logger
-
-
-class TemplateManager(BaseModel):
-    """A class that manages templates for code generation."""
-
-    model_config = ConfigDict(use_attribute_docstrings=True)
-    templates_dir: List[DirectoryPath] = Field(default_factory=lambda: list(configs.code2prompt.template_dir))
-    """The directories containing the templates. first element has the highest override priority."""
-    _discovered_templates: Dict[str, FilePath] = PrivateAttr(default_factory=dict)
-
-    def model_post_init(self, __context: Any) -> None:
-        """Post-initialization method for the model."""
-        self.discover_templates()
-
-    def discover_templates(self) -> Self:
-        """Discover the templates in the template directories."""
-        discovered = [
-            f
-            for d in self.templates_dir[::-1]
-            for f in d.rglob(f"*{configs.code2prompt.template_suffix}", case_sensitive=False)
-            if f.is_file()
-        ]
-
-        self._discovered_templates = {f.stem: f for f in discovered}
-        logger.info(f"Discovered {len(self._discovered_templates)} templates.")
-        return self
-
-    def get_template(self, name: str) -> FilePath | None:
-        """Get the template with the specified name."""
-        return self._discovered_templates.get(name, None)
-
-
-templates_manager = TemplateManager()