fabricatio 0.2.0.dev8__cp312-cp312-win_amd64.whl → 0.2.0.dev10__cp312-cp312-win_amd64.whl

fabricatio/__init__.py

@@ -1,5 +1,7 @@
 """Fabricatio is a Python library for building llm app using event-based agent structure."""
 
+from fabricatio._rust import TemplateManager
+from fabricatio.config import configs
 from fabricatio.core import env
 from fabricatio.fs import magika
 from fabricatio.journal import logger
@@ -10,7 +12,8 @@ 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
+
+template_manager = TemplateManager(configs.code2prompt.template_dir)
 
 __all__ = [
     "Action",
@@ -28,5 +31,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/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-0.2.0.dev8.dist-info → fabricatio-0.2.0.dev10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fabricatio
-Version: 0.2.0.dev8
+Version: 0.2.0.dev10
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: Implementation :: CPython

{fabricatio-0.2.0.dev8.dist-info → fabricatio-0.2.0.dev10.dist-info}/RECORD

@@ -1,11 +1,11 @@
-fabricatio-0.2.0.dev8.dist-info/METADATA,sha256=b1LUrgHhnGE_K7-nEeHU7G9hDet4hQtTufFOTSvvGtE,5970
-fabricatio-0.2.0.dev8.dist-info/WHEEL,sha256=tpW5AN9B-9qsM9WW2FXG2r193YXiqexDadpKp0A2daI,96
-fabricatio-0.2.0.dev8.dist-info/licenses/LICENSE,sha256=do7J7EiCGbq0QPbMAL_FqLYufXpHnCnXBOuqVPwSV8Y,1088
+fabricatio-0.2.0.dev10.dist-info/METADATA,sha256=dNWLPv66ZRGHyczv0HrHrih837C8eg8MrWnZa6aPKSk,5971
+fabricatio-0.2.0.dev10.dist-info/WHEEL,sha256=tpW5AN9B-9qsM9WW2FXG2r193YXiqexDadpKp0A2daI,96
+fabricatio-0.2.0.dev10.dist-info/licenses/LICENSE,sha256=do7J7EiCGbq0QPbMAL_FqLYufXpHnCnXBOuqVPwSV8Y,1088
 fabricatio/actions/communication.py,sha256=tmsr3H_w-V-b2WxLEyWByGuwSCLgHIHTdHYAgHrdUxc,425
 fabricatio/actions/transmission.py,sha256=PedZ6XsflKdT5ikzaqWr_6h8jci0kekAHfwygzKBUns,1188
 fabricatio/actions/__init__.py,sha256=eFmFVPQvtNgFynIXBVr3eP-vWQDWCPng60YY5LXvZgg,115
-fabricatio/config.py,sha256=FspRwp_gpms_b1rnTgGm5TBkBmG1avEapcxO4Jq6gdc,8041
-fabricatio/core.py,sha256=aaQDGz7spEeJtI2SVSXox4jnMrkLfzEfQDh7mIgoQmQ,5866
+fabricatio/config.py,sha256=wArRP1n3QIRwGjZOgAezdYwMYqhsrxz3D_biXAZjB28,8057
+fabricatio/core.py,sha256=yQK2ZrbPYDJOaNDp0Bky3muTkB-ZaQ1ld_Qfflm2dY0,5938
 fabricatio/decorators.py,sha256=Qsyb-_cDwtXY5yhyLrYZEAlrHh5ZuEooQ8vEOUAxj8k,1855
 fabricatio/fs/readers.py,sha256=mw0VUH3P7Wk0SMlcQm2yOfjEz5C3mQ_kjduAjecaxgY,123
 fabricatio/fs/__init__.py,sha256=lWcKYg0v3mv2LnnSegOQaTtlVDODU0vtw_s6iKU5IqQ,122
@@ -19,11 +19,10 @@ fabricatio/models/tool.py,sha256=3htDwksf4k6JfYRY3RYirjvcpZUqwj5BNB7DO0nGBN0,339
 fabricatio/models/utils.py,sha256=i_kpcQpct04mQFk1nbcVGV-pl1YThWu4Qk3wbewzKkc,2535
 fabricatio/parser.py,sha256=eIdpGBUKHAAWaWEu3NP_7zwgsxHoXIMVaHUAlSjQ6ko,2424
 fabricatio/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-fabricatio/templates.py,sha256=rmt1cf66wDTjK_LSeATv60eWX2jYO97uAX2yB_Zwtxw,1615
 fabricatio/toolboxes/task.py,sha256=xgyPetm2R_HlQwpzE8YPnBN7QOYLd0-T8E6QPZG1PPQ,204
 fabricatio/toolboxes/__init__.py,sha256=bjefmPd7wBaWhbZzdMPXvrjMTeRzlUh_Dev2PUAc124,158
-fabricatio/_rust.pyi,sha256=FZ9noKJQrdhxq0S9LJYy7BvZnShnDrupDeOv-PpliI4,39
-fabricatio/__init__.py,sha256=eznZiLDVe3pw0NwX36Rari-DK0zxDTpa_wwAmWGnto8,909
-fabricatio/_rust.cp312-win_amd64.pyd,sha256=7cdkQwcA3f7pP_cEV9c9xMH1evvJjENbCO3LH6v8Vbw,174592
-fabricatio-0.2.0.dev8.data/scripts/tdown.exe,sha256=g8NT3yY6J7JdS9F23yQ0Ocem13amG1_2ZWup9PYAkH0,3380736
-fabricatio-0.2.0.dev8.dist-info/RECORD,,
+fabricatio/_rust.pyi,sha256=IHNv9SHdjve24PBWhdRGCqWYdo2tSAkxYR9CddHhzX8,1540
+fabricatio/__init__.py,sha256=fop9UtHY_FGoqjtiHHREXFqO5vUrbc9vnm0wgLgZC8I,1012
+fabricatio/_rust.cp312-win_amd64.pyd,sha256=-TbjfckvcCu7xmWpbgRQGpr8Y6vModoVFOjWWX6s5x8,1123840
+fabricatio-0.2.0.dev10.data/scripts/tdown.exe,sha256=5pJQDhralWahKr5r84HeLW02DyG2K2kXidUQFzw9h38,3387904
+fabricatio-0.2.0.dev10.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()