fabricatio 0.2.13.dev3__cp312-cp312-win_amd64.whl → 0.3.13__cp312-cp312-win_amd64.whl

fabricatio/capabilities/review.py

@@ -2,14 +2,14 @@
 
 from typing import Dict, Optional, Set, Unpack
 
+from fabricatio.rust import CONFIG, TEMPLATE_MANAGER
+
 from fabricatio.capabilities.propose import Propose
 from fabricatio.capabilities.rating import Rating
-from fabricatio.config import configs
 from fabricatio.models.extra.problem import Improvement
 from fabricatio.models.generic import Display, WithBriefing
 from fabricatio.models.kwargs_types import ReviewKwargs, ValidateKwargs
 from fabricatio.models.task import Task
-from fabricatio.rust_instances import TEMPLATE_MANAGER
 from fabricatio.utils import ok
 
 
@@ -41,12 +41,12 @@ class Review(Rating, Propose):
         return await self.review_obj(task, **kwargs)
 
     async def review_string(
-        self,
-        input_text: str,
-        topic: str,
-        criteria: Optional[Set[str]] = None,
-        rating_manual: Optional[Dict[str, str]] = None,
-        **kwargs: Unpack[ValidateKwargs[Improvement]],
+            self,
+            input_text: str,
+            topic: str,
+            criteria: Optional[Set[str]] = None,
+            rating_manual: Optional[Dict[str, str]] = None,
+            **kwargs: Unpack[ValidateKwargs[Improvement]],
     ) -> Optional[Improvement]:
         """Review a string based on specified topic and criteria.
 
@@ -70,7 +70,7 @@ class Review(Rating, Propose):
             # this `default` is the default for the `propose` method
             default = kwargs.pop("default")
 
-        criteria = ok(criteria or (await self.draft_rating_criteria(topic, **kwargs))," No criteria could be use.")
+        criteria = ok(criteria or (await self.draft_rating_criteria(topic, **kwargs)), " No criteria could be use.")
         manual = rating_manual or await self.draft_rating_manual(topic, criteria, **kwargs)
 
         if default is not None:
@@ -78,13 +78,14 @@ class Review(Rating, Propose):
         return await self.propose(
             Improvement,
             TEMPLATE_MANAGER.render_template(
-                configs.templates.review_string_template,
+                CONFIG.templates.review_string_template,
                 {"text": input_text, "topic": topic, "criteria_manual": manual},
             ),
             **kwargs,
         )
 
-    async def review_obj[M: (Display, WithBriefing)](self, obj: M, **kwargs: Unpack[ReviewKwargs[Improvement]]) -> Optional[Improvement]:
+    async def review_obj[M: (Display, WithBriefing)](self, obj: M, **kwargs: Unpack[ReviewKwargs[Improvement]]) -> \
+            Optional[Improvement]:
         """Review an object that implements Display or WithBriefing interface.
 
         This method extracts displayable text from the object and performs a review

fabricatio/capabilities/task.py

@@ -4,16 +4,15 @@ from types import CodeType
 from typing import Any, Dict, List, Optional, Tuple, Unpack
 
 import ujson
+from fabricatio.rust import CONFIG, TEMPLATE_MANAGER
 
 from fabricatio.capabilities.propose import Propose
-from fabricatio.config import configs
 from fabricatio.journal import logger
 from fabricatio.models.kwargs_types import ChooseKwargs, ValidateKwargs
 from fabricatio.models.task import Task
 from fabricatio.models.tool import Tool, ToolExecutor
 from fabricatio.models.usages import ToolBoxUsage
 from fabricatio.parser import JsonCapture, PythonCapture
-from fabricatio.rust_instances import TEMPLATE_MANAGER
 
 
 class ProposeTask(Propose):
@@ -67,10 +66,10 @@ class HandleTask(ToolBoxUsage):
             return None
 
         q = TEMPLATE_MANAGER.render_template(
-            configs.templates.draft_tool_usage_code_template,
+            CONFIG.templates.draft_tool_usage_code_template,
             {
-                "data_module_name": configs.toolbox.data_module_name,
-                "tool_module_name": configs.toolbox.tool_module_name,
+                "data_module_name": CONFIG.toolbox.data_module_name,
+                "tool_module_name": CONFIG.toolbox.tool_module_name,
                 "task": task.briefing,
                 "deps": task.dependencies_prompt,
                 "tools": [{"name": t.name, "briefing": t.briefing} for t in tools],

fabricatio/core.py

@@ -2,12 +2,10 @@
 
 from typing import Callable, Optional, Self, overload
 
+from fabricatio.rust import CONFIG, Event
 from pydantic import BaseModel, ConfigDict, PrivateAttr
 from pymitter import EventEmitter
 
-from fabricatio.config import configs
-from fabricatio.models.events import Event
-
 
 class Env(BaseModel):
     """Environment class that manages event handling using EventEmitter."""
@@ -15,9 +13,9 @@ class Env(BaseModel):
     model_config = ConfigDict(use_attribute_docstrings=True)
     _ee: EventEmitter = PrivateAttr(
         default_factory=lambda: EventEmitter(
-            delimiter=configs.pymitter.delimiter,
-            new_listener=configs.pymitter.new_listener_event,
-            max_listeners=configs.pymitter.max_listeners,
+            delimiter=CONFIG.pymitter.delimiter,
+            new_listener=CONFIG.pymitter.new_listener_event,
+            max_listeners=CONFIG.pymitter.max_listeners,
             wildcard=True,
         )
     )
@@ -38,11 +36,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 +56,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.
 
@@ -83,8 +81,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.
@@ -99,9 +97,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.
@@ -116,9 +114,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/decorators.py

@@ -8,7 +8,8 @@ from shutil import which
 from types import ModuleType
 from typing import Callable, Coroutine, List, Optional
 
-from fabricatio.config import configs
+from fabricatio.rust import CONFIG
+
 from fabricatio.journal import logger
 
 
@@ -24,7 +25,7 @@ def precheck_package[**P, R](package_name: str, msg: str) -> Callable[[Callable[
     """
 
     def _wrapper(
-        func: Callable[P, R] | Callable[P, Coroutine[None, None, R]],
+            func: Callable[P, R] | Callable[P, Coroutine[None, None, R]],
     ) -> Callable[P, R] | Callable[P, Coroutine[None, None, R]]:
         if iscoroutinefunction(func):
 
@@ -48,7 +49,7 @@ def precheck_package[**P, R](package_name: str, msg: str) -> Callable[[Callable[
 
 
 def depend_on_external_cmd[**P, R](
-    bin_name: str, install_tip: Optional[str], homepage: Optional[str] = None
+        bin_name: str, install_tip: Optional[str], homepage: Optional[str] = None
 ) -> Callable[[Callable[P, R]], Callable[P, R]]:
     """Decorator to check for the presence of an external command.
 
@@ -113,7 +114,7 @@ def confirm_to_execute[**P, R](func: Callable[P, R]) -> Callable[P, Optional[R]]
     Returns:
         Callable: A decorator that wraps the function to confirm before execution.
     """
-    if not configs.general.confirm_on_ops:
+    if not CONFIG.general.confirm_on_ops:
         # Skip confirmation if the configuration is set to False
         return func
     from questionary import confirm
@@ -123,8 +124,8 @@ def confirm_to_execute[**P, R](func: Callable[P, R]) -> Callable[P, Optional[R]]
         @wraps(func)
         async def _wrapper(*args: P.args, **kwargs: P.kwargs) -> Optional[R]:
             if await confirm(
-                f"Are you sure to execute function: {func.__name__}{signature(func)} \n📦 Args:{args}\n🔑 Kwargs:{kwargs}\n",
-                instruction="Please input [Yes/No] to proceed (default: Yes):",
+                    f"Are you sure to execute function: {func.__name__}{signature(func)} \n📦 Args:{args}\n🔑 Kwargs:{kwargs}\n",
+                    instruction="Please input [Yes/No] to proceed (default: Yes):",
             ).ask_async():
                 return await func(*args, **kwargs)
             logger.warning(f"Function: {func.__name__}{signature(func)} canceled by user.")
@@ -135,8 +136,8 @@ def confirm_to_execute[**P, R](func: Callable[P, R]) -> Callable[P, Optional[R]]
         @wraps(func)
         def _wrapper(*args: P.args, **kwargs: P.kwargs) -> Optional[R]:
             if confirm(
-                f"Are you sure to execute function: {func.__name__}{signature(func)} \n📦 Args:{args}\n��� Kwargs:{kwargs}\n",
-                instruction="Please input [Yes/No] to proceed (default: Yes):",
+                    f"Are you sure to execute function: {func.__name__}{signature(func)} \n📦 Args:{args}\n��� Kwargs:{kwargs}\n",
+                    instruction="Please input [Yes/No] to proceed (default: Yes):",
             ).ask():
                 return func(*args, **kwargs)
             logger.warning(f"Function: {func.__name__}{signature(func)} canceled by user.")
@@ -218,7 +219,7 @@ def use_temp_module[**P, R](modules: ModuleType | List[ModuleType]) -> Callable[
 
 
 def logging_exec_time[**P, R](
-    func: Callable[P, R] | Callable[P, Coroutine[None, None, R]],
+        func: Callable[P, R] | Callable[P, Coroutine[None, None, R]],
 ) -> Callable[P, R] | Callable[P, Coroutine[None, None, R]]:
     """Decorator to log the execution time of a function.
 
@@ -231,7 +232,6 @@ def logging_exec_time[**P, R](
     from time import time
 
     if iscoroutinefunction(func):
-
         @wraps(func)
         async def _async_wrapper(*args: P.args, **kwargs: P.kwargs) -> R:
             start_time = time()

fabricatio/fs/__init__.py

@@ -1,7 +1,6 @@
 """FileSystem manipulation module for Fabricatio."""
 from importlib.util import find_spec
 
-from fabricatio.config import configs
 from fabricatio.fs.curd import (
     absolute_path,
     copy_file,
@@ -32,5 +31,5 @@ __all__ = [
 if find_spec("magika"):
     from magika import Magika
 
-    MAGIKA = Magika(model_dir=configs.magika.model_dir)
+    MAGIKA = Magika()
     __all__ += ["MAGIKA"]

fabricatio/journal.py

@@ -2,20 +2,13 @@
 
 import sys
 
+from fabricatio.rust import CONFIG
 from loguru import logger
 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=f"{configs.debug.rotation} weeks",
-    retention=f"{configs.debug.retention} weeks",
-)
-logger.add(sys.stderr, level=configs.debug.log_level)
+logger.add(sys.stderr, level=CONFIG.debug.log_level)
 
 __all__ = ["logger"]

fabricatio/models/action.py

@@ -14,12 +14,13 @@ from abc import abstractmethod
 from asyncio import Queue, create_task
 from typing import Any, ClassVar, Dict, Self, Sequence, Tuple, Type, Union, final
 
+from pydantic import Field, PrivateAttr
+
 from fabricatio.journal import logger
 from fabricatio.models.generic import WithBriefing
 from fabricatio.models.task import Task
 from fabricatio.models.usages import ToolBoxUsage
 from fabricatio.utils import override_kwargs
-from pydantic import Field, PrivateAttr
 
 OUTPUT_KEY = "task_output"
 

fabricatio/models/extra/aricle_rag.py

@@ -5,16 +5,17 @@ from itertools import groupby
 from pathlib import Path
 from typing import ClassVar, Dict, List, Optional, Self, Unpack
 
+from fabricatio.rust import BibManager, blake3_hash, split_into_chunks
+from more_itertools.more import first
+from more_itertools.recipes import flatten, unique
+from pydantic import Field
+
 from fabricatio.fs import safe_text_read
 from fabricatio.journal import logger
 from fabricatio.models.extra.rag import MilvusDataBase
 from fabricatio.models.generic import AsPrompt
 from fabricatio.models.kwargs_types import ChunkKwargs
-from fabricatio.rust import BibManager, blake3_hash, split_into_chunks
 from fabricatio.utils import ok, wrapp_in_block
-from more_itertools.more import first
-from more_itertools.recipes import flatten, unique
-from pydantic import Field
 
 
 class ArticleChunk(MilvusDataBase):
@@ -67,7 +68,7 @@ class ArticleChunk(MilvusDataBase):
 
     @classmethod
     def from_file[P: str | Path](
-        cls, path: P | List[P], bib_mgr: BibManager, **kwargs: Unpack[ChunkKwargs]
+            cls, path: P | List[P], bib_mgr: BibManager, **kwargs: Unpack[ChunkKwargs]
     ) -> List[Self]:
         """Load the article chunks from the file."""
         if isinstance(path, list):
@@ -84,9 +85,9 @@ class ArticleChunk(MilvusDataBase):
         title_seg = path.stem.split(" - ").pop()
 
         key = (
-            bib_mgr.get_cite_key_by_title(title_seg)
-            or bib_mgr.get_cite_key_by_title_fuzzy(title_seg)
-            or bib_mgr.get_cite_key_fuzzy(path.stem)
+                bib_mgr.get_cite_key_by_title(title_seg)
+                or bib_mgr.get_cite_key_by_title_fuzzy(title_seg)
+                or bib_mgr.get_cite_key_fuzzy(path.stem)
         )
         if key is None:
             logger.warning(f"no cite key found for {path.as_posix()}, skip.")
@@ -178,7 +179,7 @@ class CitationManager(AsPrompt):
     """Separator for abbreviated citation numbers."""
 
     def update_chunks(
-        self, article_chunks: List[ArticleChunk], set_cite_number: bool = True, dedup: bool = True
+            self, article_chunks: List[ArticleChunk], set_cite_number: bool = True, dedup: bool = True
     ) -> Self:
         """Update article chunks."""
         self.article_chunks.clear()

fabricatio/models/extra/article_base.py

@@ -5,6 +5,9 @@ from enum import StrEnum
 from pathlib import Path
 from typing import ClassVar, Generator, List, Optional, Self, Tuple, Type
 
+from fabricatio.rust import extract_body, inplace_update, split_out_metadata, to_metadata, word_count
+from pydantic import Field
+
 from fabricatio.fs import dump_text, safe_text_read
 from fabricatio.fs.readers import extract_sections
 from fabricatio.journal import logger
@@ -17,14 +20,11 @@ from fabricatio.models.generic import (
     ModelHash,
     PersistentAble,
     ProposedUpdateAble,
-    ResolveUpdateConflict,
     SketchedAble,
     Titled,
     WordCount,
 )
-from fabricatio.rust import extract_body, inplace_update, split_out_metadata, to_metadata, word_count
 from fabricatio.utils import fallback_kwargs, ok
-from pydantic import Field
 
 ARTICLE_WRAPPER = "// =-=-=-=-=-=-=-=-=-="
 
@@ -88,7 +88,6 @@ class ToTypstCode(ArticleMetaData):
 
 
 class ArticleOutlineBase(
-    ResolveUpdateConflict,
     ProposedUpdateAble,
     PersistentAble,
     ModelHash,
@@ -277,7 +276,7 @@ class ArticleBase[T: ChapterBase](FinalizedDumpAble, AsPrompt, FromTypstCode, To
         )
 
     def iter_dfs_rev(
-        self,
+            self,
     ) -> Generator[ArticleOutlineBase, None, None]:
         """Performs a depth-first search (DFS) through the article structure in reverse order.
 
@@ -415,7 +414,7 @@ class ArticleBase[T: ChapterBase](FinalizedDumpAble, AsPrompt, FromTypstCode, To
         return self
 
     @classmethod
-    def from_article_file(cls, file: str | Path, title: str) -> Self:
+    def from_article_file[S: "ArticleBase"](cls: Type[S], file: str | Path, title: str) -> S:
         """Load article from file."""
         file = Path(file)
         string = safe_text_read(file)

fabricatio/models/extra/article_main.py

@@ -2,6 +2,15 @@
 
 from typing import ClassVar, Dict, Generator, List, Self, Tuple, Type, override
 
+from fabricatio.rust import (
+    convert_all_block_tex,
+    convert_all_inline_tex,
+    fix_misplaced_labels,
+    split_out_metadata,
+    word_count,
+)
+from pydantic import Field, NonNegativeInt
+
 from fabricatio.decorators import precheck_package
 from fabricatio.journal import logger
 from fabricatio.models.extra.article_base import (
@@ -17,14 +26,6 @@ from fabricatio.models.extra.article_outline import (
     ArticleSubsectionOutline,
 )
 from fabricatio.models.generic import Described, PersistentAble, SequencePatch, SketchedAble, WithRef, WordCount
-from fabricatio.rust import (
-    convert_all_block_tex,
-    convert_all_inline_tex,
-    fix_misplaced_labels,
-    split_out_metadata,
-    word_count,
-)
-from pydantic import Field, NonNegativeInt
 
 PARAGRAPH_SEP = "\n\n// - - -\n\n"
 
@@ -81,8 +82,8 @@ class ArticleSubsection(SubSectionBase):
         if len(self.paragraphs) == 0:
             summary += f"`{self.__class__.__name__}` titled `{self.title}` have no paragraphs, You should add some!\n"
         if (
-            abs((wc := self.word_count) - self.expected_word_count) / self.expected_word_count
-            > self._max_word_count_deviation
+                abs((wc := self.word_count) - self.expected_word_count) / self.expected_word_count
+                > self._max_word_count_deviation
         ):
             summary += f"`{self.__class__.__name__}` titled `{self.title}` have {wc} words, expected {self.expected_word_count} words!"