fabricatio 0.3.14.dev4__cp313-cp313-win_amd64.whl → 0.3.14.dev7__cp313-cp313-win_amd64.whl

fabricatio/actions/article_rag.py

@@ -1,10 +1,10 @@
 """A module for writing articles using RAG (Retrieval-Augmented Generation) capabilities."""
 
 from asyncio import gather
-from pathlib import Path
-from typing import ClassVar, List, Optional
 
+from pathlib import Path
 from pydantic import Field, PositiveInt
+from typing import ClassVar, List, Optional
 
 from fabricatio.capabilities.advanced_rag import AdvancedRAG
 from fabricatio.capabilities.censor import Censor
@@ -21,10 +21,7 @@ from fabricatio.models.extra.rule import RuleSet
 from fabricatio.models.kwargs_types import ChooseKwargs, LLMKwargs
 from fabricatio.rust import (
     BibManager,
-    convert_all_block_tex,
-    convert_all_inline_tex,
-    convert_to_block_formula,
-    convert_to_inline_formula,
+    convert_all_tex_math,
     fix_misplaced_labels,
 )
 from fabricatio.utils import ok
@@ -78,11 +75,11 @@ class WriteArticleContentRAG(Action, Extract, AdvancedRAG):
     tei_endpoint: Optional[str] = None
 
     async def _execute(
-        self,
-        article_outline: ArticleOutline,
-        collection_name: Optional[str] = None,
-        supervisor: Optional[bool] = None,
-        **cxt,
+            self,
+            article_outline: ArticleOutline,
+            collection_name: Optional[str] = None,
+            supervisor: Optional[bool] = None,
+            **cxt,
     ) -> Article:
         article = Article.from_outline(article_outline).update_ref(article_outline)
         self.target_collection = collection_name or self.safe_target_collection
@@ -103,12 +100,12 @@ class WriteArticleContentRAG(Action, Extract, AdvancedRAG):
         "questionary", "`questionary` is required for supervisor mode, please install it by `fabricatio[qa]`"
     )
     async def _supervisor_inner(
-        self,
-        article: Article,
-        article_outline: ArticleOutline,
-        chap: ArticleChapter,
-        sec: ArticleSection,
-        subsec: ArticleSubsection,
+            self,
+            article: Article,
+            article_outline: ArticleOutline,
+            chap: ArticleChapter,
+            sec: ArticleSection,
+            subsec: ArticleSubsection,
     ) -> ArticleSubsection:
         from questionary import confirm, text
         from rich import print as r_print
@@ -129,20 +126,19 @@ class WriteArticleContentRAG(Action, Extract, AdvancedRAG):
                 raw_paras = edt
 
             raw_paras = fix_misplaced_labels(raw_paras)
-            raw_paras = convert_all_inline_tex(raw_paras)
-            raw_paras = convert_all_block_tex(raw_paras)
+            raw_paras = convert_all_tex_math(raw_paras)
 
             r_print(raw_paras)
 
         return await self.extract_new_subsec(subsec, raw_paras, cm)
 
     async def _inner(
-        self,
-        article: Article,
-        article_outline: ArticleOutline,
-        chap: ArticleChapter,
-        sec: ArticleSection,
-        subsec: ArticleSubsection,
+            self,
+            article: Article,
+            article_outline: ArticleOutline,
+            chap: ArticleChapter,
+            sec: ArticleSection,
+            subsec: ArticleSubsection,
     ) -> ArticleSubsection:
         cm = CitationManager()
 
@@ -153,13 +149,12 @@ class WriteArticleContentRAG(Action, Extract, AdvancedRAG):
         raw_paras = "\n".join(p for p in raw_paras.splitlines() if p and not p.endswith("**") and not p.startswith("#"))
 
         raw_paras = fix_misplaced_labels(raw_paras)
-        raw_paras = convert_all_inline_tex(raw_paras)
-        raw_paras = convert_all_block_tex(raw_paras)
+        raw_paras = convert_all_tex_math(raw_paras)
 
         return await self.extract_new_subsec(subsec, raw_paras, cm)
 
     async def extract_new_subsec(
-        self, subsec: ArticleSubsection, raw_paras: str, cm: CitationManager
+            self, subsec: ArticleSubsection, raw_paras: str, cm: CitationManager
     ) -> ArticleSubsection:
         """Extract the new subsec."""
         new_subsec = ok(
@@ -182,14 +177,14 @@ class WriteArticleContentRAG(Action, Extract, AdvancedRAG):
         return subsec
 
     async def write_raw(
-        self,
-        article: Article,
-        article_outline: ArticleOutline,
-        chap: ArticleChapter,
-        sec: ArticleSection,
-        subsec: ArticleSubsection,
-        cm: CitationManager,
-        extra_instruction: str = "",
+            self,
+            article: Article,
+            article_outline: ArticleOutline,
+            chap: ArticleChapter,
+            sec: ArticleSection,
+            subsec: ArticleSubsection,
+            cm: CitationManager,
+            extra_instruction: str = "",
     ) -> str:
         """Write the raw paragraphs of the subsec."""
         return await self.aask(
@@ -205,14 +200,14 @@ class WriteArticleContentRAG(Action, Extract, AdvancedRAG):
         )
 
     async def search_database(
-        self,
-        article: Article,
-        article_outline: ArticleOutline,
-        chap: ArticleChapter,
-        sec: ArticleSection,
-        subsec: ArticleSubsection,
-        cm: CitationManager,
-        extra_instruction: str = "",
+            self,
+            article: Article,
+            article_outline: ArticleOutline,
+            chap: ArticleChapter,
+            sec: ArticleSection,
+            subsec: ArticleSubsection,
+            cm: CitationManager,
+            extra_instruction: str = "",
     ) -> None:
         """Search database for related references."""
         search_req = (
@@ -261,8 +256,6 @@ class ArticleConsultRAG(Action, AdvancedRAG):
         from questionary import confirm, text
         from rich import print as r_print
 
-        from fabricatio.rust import convert_all_block_tex, convert_all_inline_tex, fix_misplaced_labels
-
         self.target_collection = collection_name or self.safe_target_collection
 
         cm = CitationManager()
@@ -272,8 +265,7 @@ class ArticleConsultRAG(Action, AdvancedRAG):
             if await confirm("Empty the cm?").ask_async():
                 cm.empty()
 
-            req = convert_to_block_formula(req)
-            req = convert_to_inline_formula(req)
+            req = convert_all_tex_math(req)
 
             await self.clued_search(
                 req,
@@ -289,8 +281,7 @@ class ArticleConsultRAG(Action, AdvancedRAG):
             ret = await self.aask(f"{cm.as_prompt()}\n{self.req}\n{req}")
 
             ret = fix_misplaced_labels(ret)
-            ret = convert_all_inline_tex(ret)
-            ret = convert_all_block_tex(ret)
+            ret = convert_all_tex_math(ret)
             ret = cm.apply(ret)
 
             r_print(ret)
@@ -321,12 +312,12 @@ class TweakArticleRAG(Action, RAG, Censor):
     """The limit of references to be retrieved"""
 
     async def _execute(
-        self,
-        article: Article,
-        collection_name: str = "article_essence",
-        twk_rag_ruleset: Optional[RuleSet] = None,
-        parallel: bool = False,
-        **cxt,
+            self,
+            article: Article,
+            collection_name: str = "article_essence",
+            twk_rag_ruleset: Optional[RuleSet] = None,
+            parallel: bool = False,
+            **cxt,
     ) -> Article:
         """Write an article based on the provided outline.
 
@@ -381,10 +372,10 @@ class TweakArticleRAG(Action, RAG, Censor):
             subsec,
             ruleset=ruleset,
             reference=f"{'\n\n'.join(d.display() for d in await self.aretrieve(refind_q, document_model=ArticleEssence, max_accepted=self.ref_limit))}\n\n"
-            f"You can use Reference above to rewrite the `{subsec.__class__.__name__}`.\n"
-            f"You should Always use `{subsec.language}` as written language, "
-            f"which is the original language of the `{subsec.title}`. "
-            f"since rewrite a `{subsec.__class__.__name__}` in a different language is usually a bad choice",
+                      f"You can use Reference above to rewrite the `{subsec.__class__.__name__}`.\n"
+                      f"You should Always use `{subsec.language}` as written language, "
+                      f"which is the original language of the `{subsec.title}`. "
+                      f"since rewrite a `{subsec.__class__.__name__}` in a different language is usually a bad choice",
         )
 
 
@@ -399,12 +390,12 @@ class ChunkArticle(Action):
     """The maximum overlapping rate between chunks."""
 
     async def _execute(
-        self,
-        article_path: str | Path,
-        bib_manager: BibManager,
-        max_chunk_size: Optional[int] = None,
-        max_overlapping_rate: Optional[float] = None,
-        **_,
+            self,
+            article_path: str | Path,
+            bib_manager: BibManager,
+            max_chunk_size: Optional[int] = None,
+            max_overlapping_rate: Optional[float] = None,
+            **_,
     ) -> List[ArticleChunk]:
         return ArticleChunk.from_file(
             article_path,

fabricatio/capabilities/rag.py

@@ -51,16 +51,16 @@ class RAG(EmbeddingUsage, ABC):
         return self._client
 
     def init_client(
-        self,
-        milvus_uri: Optional[str] = None,
-        milvus_token: Optional[str] = None,
-        milvus_timeout: Optional[float] = None,
+            self,
+            milvus_uri: Optional[str] = None,
+            milvus_token: Optional[str] = None,
+            milvus_timeout: Optional[float] = None,
     ) -> Self:
         """Initialize the Milvus client."""
         self._client = create_client(
             uri=milvus_uri or ok(self.milvus_uri or CONFIG.rag.milvus_uri),
             token=milvus_token
-            or (token.get_secret_value() if (token := (self.milvus_token or CONFIG.rag.milvus_token)) else ""),
+                  or (token.get_secret_value() if (token := (self.milvus_token or CONFIG.rag.milvus_token)) else ""),
             timeout=milvus_timeout or self.milvus_timeout or CONFIG.rag.milvus_timeout,
         )
         return self
@@ -74,7 +74,7 @@ class RAG(EmbeddingUsage, ABC):
         return self
 
     def view(
-        self, collection_name: Optional[str], create: bool = False, **kwargs: Unpack[CollectionConfigKwargs]
+            self, collection_name: Optional[str], create: bool = False, **kwargs: Unpack[CollectionConfigKwargs]
     ) -> Self:
         """View the specified collection.
 
@@ -116,7 +116,7 @@ class RAG(EmbeddingUsage, ABC):
         return ok(self.target_collection, "No collection is being viewed. Have you called `self.view()`?")
 
     async def add_document[D: MilvusDataBase](
-        self, data: List[D] | D, collection_name: Optional[str] = None, flush: bool = False
+            self, data: List[D] | D, collection_name: Optional[str] = None, flush: bool = False
     ) -> Self:
         """Adds a document to the specified collection.
 
@@ -143,15 +143,15 @@ class RAG(EmbeddingUsage, ABC):
         return self
 
     async def afetch_document[D: MilvusDataBase](
-        self,
-        query: List[str],
-        document_model: Type[D],
-        collection_name: Optional[str] = None,
-        similarity_threshold: float = 0.37,
-        result_per_query: int = 10,
-        tei_endpoint: Optional[str] = None,
-        reranker_threshold: float = 0.7,
-        filter_expr: str = "",
+            self,
+            query: List[str],
+            document_model: Type[D],
+            collection_name: Optional[str] = None,
+            similarity_threshold: float = 0.37,
+            result_per_query: int = 10,
+            tei_endpoint: Optional[str] = None,
+            reranker_threshold: float = 0.7,
+            filter_expr: str = "",
     ) -> List[D]:
         """Asynchronously fetches documents from a Milvus database based on input vectors.
 
@@ -192,7 +192,8 @@ class RAG(EmbeddingUsage, ABC):
                 retrieved_id.update(res["id"] for res in g)
                 if not models:
                     continue
-                rank_scores = await reranker.arerank(q, [m.prepare_vectorization() for m in models], truncate=True)
+                rank_scores = await reranker.arerank(q, [m.prepare_vectorization() for m in models], truncate=True,
+                                                     truncation_direction="Left")
                 raw_result.extend((models[idx], scr) for (idx, scr) in rank_scores if scr > reranker_threshold)
 
             raw_result_sorted = sorted(raw_result, key=lambda x: x[1], reverse=True)
@@ -214,11 +215,11 @@ class RAG(EmbeddingUsage, ABC):
         return document_model.from_sequence(resp)
 
     async def aretrieve[D: MilvusDataBase](
-        self,
-        query: List[str] | str,
-        document_model: Type[D],
-        max_accepted: int = 20,
-        **kwargs: Unpack[FetchKwargs],
+            self,
+            query: List[str] | str,
+            document_model: Type[D],
+            max_accepted: int = 20,
+            **kwargs: Unpack[FetchKwargs],
     ) -> List[D]:
         """Retrieve data from the collection.
 
@@ -235,15 +236,15 @@ class RAG(EmbeddingUsage, ABC):
             query = [query]
 
         return (
-            await self.afetch_document(
-                query=query,
-                document_model=document_model,
-                **kwargs,
-            )
-        )[:max_accepted]
+                   await self.afetch_document(
+                       query=query,
+                       document_model=document_model,
+                       **kwargs,
+                   )
+               )[:max_accepted]
 
     async def arefined_query(
-        self, question: List[str] | str, **kwargs: Unpack[ChooseKwargs[Optional[List[str]]]]
+            self, question: List[str] | str, **kwargs: Unpack[ChooseKwargs[Optional[List[str]]]]
     ) -> Optional[List[str]]:
         """Refines the given question using a template.
 

fabricatio/models/extra/aricle_rag.py

@@ -1,8 +1,13 @@
 """A Module containing the article rag models."""
 
-import re
 from itertools import groupby
+
+import re
+from dataclasses import dataclass, field
+from more_itertools.more import first
+from more_itertools.recipes import flatten, unique
 from pathlib import Path
+from pydantic import Field
 from typing import ClassVar, Dict, List, Optional, Self, Unpack
 
 from fabricatio.fs import safe_text_read
@@ -12,9 +17,6 @@ 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):
@@ -68,7 +70,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):
@@ -85,9 +87,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.")
@@ -165,10 +167,11 @@ class ArticleChunk(MilvusDataBase):
         return self
 
 
+@dataclass
 class CitationManager(AsPrompt):
     """Citation manager."""
 
-    article_chunks: List[ArticleChunk] = Field(default_factory=list)
+    article_chunks: List[ArticleChunk] = field(default_factory=list)
     """Article chunks."""
 
     pat: str = r"(\[\[([\d\s,-]*)]])"
@@ -179,7 +182,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

@@ -3,6 +3,7 @@
 from abc import ABC
 from enum import StrEnum
 from pathlib import Path
+from pydantic import Field
 from typing import ClassVar, Generator, List, Optional, Self, Tuple, Type
 
 from fabricatio.capabilities.persist import PersistentAble
@@ -21,9 +22,8 @@ from fabricatio.models.generic import (
     Titled,
     WordCount,
 )
-from fabricatio.rust import extract_body, inplace_update, split_out_metadata, to_metadata, word_count
+from fabricatio.rust import extract_body, replace_thesis_body, split_out_metadata, to_metadata, word_count
 from fabricatio.utils import fallback_kwargs, ok
-from pydantic import Field
 
 ARTICLE_WRAPPER = "// =-=-=-=-=-=-=-=-=-="
 
@@ -275,7 +275,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.
 
@@ -405,7 +405,7 @@ class ArticleBase[T: ChapterBase](FinalizedDumpAble, AsPrompt, FromTypstCode, To
         """Update the article file."""
         file = Path(file)
         string = safe_text_read(file)
-        if updated := inplace_update(string, ARTICLE_WRAPPER, self.to_typst_code()):
+        if updated := replace_thesis_body(string, ARTICLE_WRAPPER, self.to_typst_code()):
             dump_text(file, updated)
             logger.success(f"Successfully updated {file.as_posix()}.")
         else:

fabricatio/models/extra/article_main.py

@@ -1,15 +1,7 @@
 """ArticleBase and ArticleSubsection classes for managing hierarchical document components."""
 
-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 typing import ClassVar, Dict, Generator, List, Self, Tuple, Type, override
 
 from fabricatio.capabilities.persist import PersistentAble
 from fabricatio.decorators import precheck_package
@@ -27,6 +19,12 @@ from fabricatio.models.extra.article_outline import (
     ArticleSubsectionOutline,
 )
 from fabricatio.models.generic import Described, SequencePatch, SketchedAble, WithRef, WordCount
+from fabricatio.rust import (
+    convert_all_tex_math,
+    fix_misplaced_labels,
+    split_out_metadata,
+    word_count,
+)
 
 PARAGRAPH_SEP = "// - - -"
 
@@ -156,15 +154,13 @@ class Article(
         if descriptions:
             for a in self.iter_dfs():
                 a.description = fix_misplaced_labels(a.description)
-                a.description = convert_all_inline_tex(a.description)
-                a.description = convert_all_block_tex(a.description)
+                a.description = convert_all_tex_math(a.description)
 
         if paragraphs:
             for _, _, subsec in self.iter_subsections():
                 for p in subsec.paragraphs:
                     p.content = fix_misplaced_labels(p.content)
-                    p.content = convert_all_inline_tex(p.content)
-                    p.content = convert_all_block_tex(p.content)
+                    p.content = convert_all_tex_math(p.content)
         return self
 
     @override

fabricatio/models/tool.py

@@ -3,18 +3,18 @@
 This module provides classes for defining tools and toolboxes, which can be used to manage and execute callable functions
 with additional functionalities such as logging, execution info, and briefing.
 """
+
 from importlib.machinery import ModuleSpec
 from importlib.util import module_from_spec
 from inspect import iscoroutinefunction, signature
 from types import CodeType, ModuleType
 from typing import Any, Callable, Dict, List, Optional, Self, cast, overload
 
-from fabricatio.rust import CONFIG
-from pydantic import Field
-
 from fabricatio.decorators import logging_execution_info, use_temp_module
 from fabricatio.journal import logger
 from fabricatio.models.generic import Base, WithBriefing
+from fabricatio.rust import CONFIG
+from pydantic import Field
 
 
 class Tool[**P, R](WithBriefing):

fabricatio/rust.pyi

@@ -12,9 +12,9 @@ Key Features:
 """
 
 from enum import StrEnum
+from pydantic import JsonValue
 from typing import Any, Dict, List, Literal, Optional, Self, Tuple, Union, overload
 
-from pydantic import JsonValue
 
 class TemplateManager:
     """Template rendering engine using Handlebars templates.
@@ -47,8 +47,10 @@ class TemplateManager:
 
     @overload
     def render_template(self, name: str, data: Dict[str, Any]) -> str: ...
+
     @overload
     def render_template(self, name: str, data: List[Dict[str, Any]]) -> List[str]: ...
+
     def render_template(self, name: str, data: Dict[str, Any] | List[Dict[str, Any]]) -> str | List[str]:
         """Render a template with context data.
 
@@ -65,8 +67,10 @@ class TemplateManager:
 
     @overload
     def render_template_raw(self, template: str, data: Dict[str, Any]) -> str: ...
+
     @overload
     def render_template_raw(self, template: str, data: List[Dict[str, Any]]) -> List[str]: ...
+
     def render_template_raw(self, template: str, data: Dict[str, Any] | List[Dict[str, Any]]) -> str | List[str]:
         """Render a template with context data.
 
@@ -78,6 +82,7 @@ class TemplateManager:
             Rendered template content as string or list of strings
         """
 
+
 class BibManager:
     """BibTeX bibliography manager for parsing and querying citation data."""
 
@@ -186,6 +191,7 @@ class BibManager:
             Field value if found, None otherwise
         """
 
+
 def blake3_hash(content: bytes) -> str:
     """Calculate the BLAKE3 cryptographic hash of data.
 
@@ -196,9 +202,11 @@ def blake3_hash(content: bytes) -> str:
         Hex-encoded BLAKE3 hash string
     """
 
+
 def detect_language(string: str) -> str:
     """Detect the language of a given string."""
 
+
 def split_word_bounds(string: str) -> List[str]:
     """Split the string into words based on word boundaries.
 
@@ -209,6 +217,7 @@ def split_word_bounds(string: str) -> List[str]:
         A list of words extracted from the string.
     """
 
+
 def split_sentence_bounds(string: str) -> List[str]:
     """Split the string into sentences based on sentence boundaries.
 
@@ -219,6 +228,7 @@ def split_sentence_bounds(string: str) -> List[str]:
         A list of sentences extracted from the string.
     """
 
+
 def split_into_chunks(string: str, max_chunk_size: int, max_overlapping_rate: float = 0.3) -> List[str]:
     """Split the string into chunks of a specified size.
 
@@ -231,6 +241,7 @@ def split_into_chunks(string: str, max_chunk_size: int, max_overlapping_rate: fl
         A list of chunks extracted from the string.
     """
 
+
 def word_count(string: str) -> int:
     """Count the number of words in the string.
 
@@ -241,51 +252,67 @@ def word_count(string: str) -> int:
         The number of words in the string.
     """
 
+
 def is_chinese(string: str) -> bool:
     """Check if the given string is in Chinese."""
 
+
 def is_english(string: str) -> bool:
     """Check if the given string is in English."""
 
+
 def is_japanese(string: str) -> bool:
     """Check if the given string is in Japanese."""
 
+
 def is_korean(string: str) -> bool:
     """Check if the given string is in Korean."""
 
+
 def is_arabic(string: str) -> bool:
     """Check if the given string is in Arabic."""
 
+
 def is_russian(string: str) -> bool:
     """Check if the given string is in Russian."""
 
+
 def is_german(string: str) -> bool:
     """Check if the given string is in German."""
 
+
 def is_french(string: str) -> bool:
     """Check if the given string is in French."""
 
+
 def is_hindi(string: str) -> bool:
     """Check if the given string is in Hindi."""
 
+
 def is_italian(string: str) -> bool:
     """Check if the given string is in Italian."""
 
+
 def is_dutch(string: str) -> bool:
     """Check if the given string is in Dutch."""
 
+
 def is_portuguese(string: str) -> bool:
     """Check if the given string is in Portuguese."""
 
+
 def is_swedish(string: str) -> bool:
     """Check if the given string is in Swedish."""
 
+
 def is_turkish(string: str) -> bool:
     """Check if the given string is in Turkish."""
 
+
 def is_vietnamese(string: str) -> bool:
     """Check if the given string is in Vietnamese."""
 
+
 def tex_to_typst(string: str) -> str:
     """Convert TeX to Typst.
 
@@ -296,26 +323,20 @@ def tex_to_typst(string: str) -> str:
         The converted Typst string.
     """
 
-def convert_all_inline_tex(string: str) -> str:
-    """Convert all inline TeX code in the string.
-
-    Args:
-        string: The input string containing inline TeX code wrapped in $code$.
 
-    Returns:
-        The converted string with inline TeX code replaced.
-    """
+def convert_all_tex_math(string: str) -> str:
+    r"""Unified function to convert all supported TeX math expressions in a string to Typst format.
 
-def convert_all_block_tex(string: str) -> str:
-    """Convert all block TeX code in the string.
+    Handles $...$, $$...$$, \\(...\\), and \\[...\\]
 
     Args:
-        string: The input string containing block TeX code wrapped in $$code$$.
+        string: The input string containing TeX math expressions.
 
     Returns:
-        The converted string with block TeX code replaced.
+        The string with TeX math expressions converted to Typst format.
     """
 
+
 def fix_misplaced_labels(string: str) -> str:
     """A func to fix labels in a string.
 
@@ -326,8 +347,9 @@ def fix_misplaced_labels(string: str) -> str:
         The fixed string with labels properly placed.
     """
 
+
 def comment(string: str) -> str:
-    """Add comment to the string.
+    r"""Add comment to the string.
 
     Args:
         string: The input string to which comments will be added.
@@ -336,6 +358,7 @@ def comment(string: str) -> str:
         The string with each line prefixed by '// '.
     """
 
+
 def uncomment(string: str) -> str:
     """Remove comment from the string.
 
@@ -346,6 +369,7 @@ def uncomment(string: str) -> str:
         The string with comments (lines starting with '// ' or '//') removed.
     """
 
+
 def split_out_metadata(string: str) -> Tuple[Optional[JsonValue], str]:
     """Split out metadata from a string.
 
@@ -356,6 +380,7 @@ def split_out_metadata(string: str) -> Tuple[Optional[JsonValue], str]:
         A tuple containing the metadata as a Python object (if parseable) and the remaining string.
     """
 
+
 def to_metadata(data: JsonValue) -> str:
     """Convert a Python object to a YAML string.
 
@@ -366,13 +391,8 @@ def to_metadata(data: JsonValue) -> str:
         The YAML string representation of the input data.
     """
 
-def convert_to_inline_formula(string: str) -> str:
-    r"""Convert `$...$` to inline formula `\(...\)` and trim spaces."""
 
-def convert_to_block_formula(string: str) -> str:
-    r"""Convert `$$...$$` to block formula `\[...\]` and trim spaces."""
-
-def inplace_update(string: str, wrapper: str, new_body: str) -> Optional[str]:
+def replace_thesis_body(string: str, wrapper: str, new_body: str) -> Optional[str]:
     """Replace content between wrapper strings.
 
     Args:
@@ -385,6 +405,7 @@ def inplace_update(string: str, wrapper: str, new_body: str) -> Optional[str]:
 
     """
 
+
 def extract_body(string: str, wrapper: str) -> Optional[str]:
     """Extract the content between two occurrences of a wrapper string.
 
@@ -396,6 +417,7 @@ def extract_body(string: str, wrapper: str) -> Optional[str]:
         The content between the first two occurrences of the wrapper string if found, otherwise None.
     """
 
+
 class LLMConfig:
     """LLM configuration structure.
 
@@ -447,6 +469,7 @@ class LLMConfig:
     frequency_penalty: Optional[float]
     """Penalizes new tokens based on their frequency in text so far (-2.0-2.0)."""
 
+
 class EmbeddingConfig:
     """Embedding configuration structure."""
 
@@ -471,6 +494,7 @@ class EmbeddingConfig:
     api_key: Optional[SecretStr]
     """The API key."""
 
+
 class RagConfig:
     """RAG (Retrieval Augmented Generation) configuration structure."""
 
@@ -486,12 +510,14 @@ class RagConfig:
     milvus_dimensions: Optional[int]
     """The dimensions for Milvus vectors."""
 
+
 class DebugConfig:
     """Debug configuration structure."""
 
     log_level: Optional[str]
     """The logging level to use."""
 
+
 class TemplateManagerConfig:
     """Template manager configuration structure."""
 
@@ -504,6 +530,7 @@ class TemplateManagerConfig:
     template_suffix: Optional[str]
     """The suffix of the templates."""
 
+
 class TemplateConfig:
     """Template configuration structure."""
 
@@ -588,6 +615,7 @@ class TemplateConfig:
     chap_summary_template: str
     """The name of the chap summary template which will be used to generate a chapter summary."""
 
+
 class RoutingConfig:
     """Routing configuration structure for controlling request dispatching behavior."""
 
@@ -603,6 +631,7 @@ class RoutingConfig:
     cooldown_time: Optional[int]
     """Time to cooldown a deployment after failure in seconds."""
 
+
 class GeneralConfig:
     """General configuration structure for application-wide settings."""
 
@@ -612,6 +641,7 @@ class GeneralConfig:
     use_json_repair: bool
     """Whether to automatically repair malformed JSON."""
 
+
 class ToolBoxConfig:
     """Configuration for toolbox functionality."""
 
@@ -621,6 +651,7 @@ class ToolBoxConfig:
     data_module_name: str
     """The name of the module containing the data."""
 
+
 class PymitterConfig:
     """Pymitter configuration structure for controlling event emission and listener behavior."""
 
@@ -633,6 +664,7 @@ class PymitterConfig:
     max_listeners: int
     """The maximum number of listeners per event. -1 means unlimited."""
 
+
 class Config:
     """Configuration structure containing all system components."""
 
@@ -666,17 +698,22 @@ class Config:
     pymitter: PymitterConfig
     """Pymitter configuration."""
 
+
 CONFIG: Config
 
+
 class SecretStr:
     """A string that should not be exposed."""
 
     def __init__(self, source: str) -> None: ...
+
     def get_secret_value(self) -> str:
         """Expose the secret string."""
 
+
 TEMPLATE_MANAGER: TemplateManager
 
+
 class Event:
     """Event class that represents a hierarchical event with segments.
 
@@ -788,9 +825,12 @@ class Event:
         """
 
     def __hash__(self) -> int: ...
+
     def __eq__(self, other: object) -> bool: ...
+
     def __ne__(self, other: object) -> bool: ...
 
+
 class TaskStatus(StrEnum, str):
     """Enumeration of possible task statuses."""
 
@@ -809,6 +849,7 @@ class TaskStatus(StrEnum, str):
     Cancelled: TaskStatus
     """Task has been cancelled."""
 
+
 class TEIClient:
     """Client for TEI reranking service.
 
@@ -824,11 +865,11 @@ class TEIClient:
         """
 
     async def arerank(
-        self,
-        query: str,
-        texts: List[str],
-        truncate: bool = False,
-        truncation_direction: Literal["Left", "Right"] = "Left",
+            self,
+            query: str,
+            texts: List[str],
+            truncate: bool = False,
+            truncation_direction: Literal["Left", "Right"] = "Left",
     ) -> List[Tuple[int, float]]:
         """Rerank texts based on relevance to query.
 

{fabricatio-0.3.14.dev4.dist-info → fabricatio-0.3.14.dev7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fabricatio
-Version: 0.3.14.dev4
+Version: 0.3.14.dev7
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: 3.12

{fabricatio-0.3.14.dev4.dist-info → fabricatio-0.3.14.dev7.dist-info}/RECORD

@@ -1,13 +1,17 @@
-fabricatio-0.3.14.dev4.dist-info/METADATA,sha256=RQu1-d7pqsIWpqH7Ts6YVBMl91x2LaSOvvhxozZhrk4,5116
-fabricatio-0.3.14.dev4.dist-info/WHEEL,sha256=6gg8M7kMeWmDzkhJuJzWLhBYdt7OGDF81wlSJjN85_E,96
-fabricatio-0.3.14.dev4.dist-info/licenses/LICENSE,sha256=do7J7EiCGbq0QPbMAL_FqLYufXpHnCnXBOuqVPwSV8Y,1088
+fabricatio-0.3.14.dev7.data/scripts/tdown.exe,sha256=lHV09r-4UOiWlzHyfYV6iq1iXdXP8FbwtMwjGiPKaqU,3450880
+fabricatio-0.3.14.dev7.data/scripts/ttm.exe,sha256=JYT1OR7umACb3xXHjQZ9JzoILYX-atkSZ-ENFBo4Gcw,2561536
+fabricatio-0.3.14.dev7.dist-info/METADATA,sha256=btu_bPIYHocy_FeY8gF0BZ1qx60Td8eujW1SGyEQOr8,5116
+fabricatio-0.3.14.dev7.dist-info/WHEEL,sha256=aR9tSjQsZDIcSIX9Ihx1Vyf9toUnjsKopg6qQPYNdi8,96
+fabricatio-0.3.14.dev7.dist-info/licenses/LICENSE,sha256=do7J7EiCGbq0QPbMAL_FqLYufXpHnCnXBOuqVPwSV8Y,1088
+fabricatio/__init__.py,sha256=w7ObFg6ud4pQuC1DhVyQI9x9dtp05QrcJAEk643iJmc,761
+fabricatio/actions/__init__.py,sha256=wVENCFtpVb1rLFxoOFJt9-8smLWXuJV7IwA8P3EfFz4,48
 fabricatio/actions/article.py,sha256=TPS2fOqCymKv2hK2c_WmMRMKNBkvN8M91QkB9ar8-bg,12507
-fabricatio/actions/article_rag.py,sha256=e1fVh7Jph2zVD0bRAmK2dJ0BVkSEvF-FPfxUKujkn6s,18407
+fabricatio/actions/article_rag.py,sha256=81668J2A_v00ODc01T-4FBEAtla6SB55VwSDO7DuRR4,18213
 fabricatio/actions/fs.py,sha256=gJR14U4ln35nt8Z7OWLVAZpqGaLnED-r1Yi-lX22tkI,959
 fabricatio/actions/output.py,sha256=jZL72D5uFobKNiVFapnVxBcjSNqGEThYNlCUKQvZwz8,9935
 fabricatio/actions/rag.py,sha256=vgCzIfbSd3_vL3QxB12PY4h12V9Pe3sZRsWme0KC6X8,3583
 fabricatio/actions/rules.py,sha256=dkvCgNDjt2KSO1VgPRsxT4YBmIIMeetZb5tiz-slYkU,3640
-fabricatio/actions/__init__.py,sha256=wVENCFtpVb1rLFxoOFJt9-8smLWXuJV7IwA8P3EfFz4,48
+fabricatio/capabilities/__init__.py,sha256=v1cHRHIJ2gxyqMLNCs6ERVcCakSasZNYzmMI4lqAcls,57
 fabricatio/capabilities/advanced_judge.py,sha256=jQ_Gsn6L8EKb6KQi3j0G0GSUz2j8D2220C1hIhrAeU8,682
 fabricatio/capabilities/advanced_rag.py,sha256=DYh-imLQkjVOgKd__OEbwGzqwNeTtX_6NBGx_bCiFs8,2539
 fabricatio/capabilities/censor.py,sha256=e0tHll4J_-TT8-Vn1OZ1innVZbJfx55oyGtDoEI99r8,4745
@@ -16,49 +20,45 @@ fabricatio/capabilities/correct.py,sha256=-JR8ZUAtagmNXepVyY679MBUyFCZwtKPjv8dAN
 fabricatio/capabilities/extract.py,sha256=E7CLZflWzJ6C6DVLEWysYZ_48g_-F93gZJVU56k2-XA,2523
 fabricatio/capabilities/persist.py,sha256=9XnKoeZ62YjXVDpYnkbDFf62B_Mz46WVsq1dTr2Wvvc,3421
 fabricatio/capabilities/propose.py,sha256=v8OiUHc8GU7Jg1zAUghYhrI-AKgmBeUvQMo22ZAOddw,2030
-fabricatio/capabilities/rag.py,sha256=D5rULrQxPmp4kVLP_jBE4yal1v9N68XOgBdJqGvVHpU,10979
+fabricatio/capabilities/rag.py,sha256=lwFrC96tL3uJ4keJ6n46vrvrdF6bARg1Yn6y6pQn7VQ,11194
 fabricatio/capabilities/rating.py,sha256=cm-s2YJMYcS36mR9b7XNwRQ1x0h0uWxLHCapoAORv8I,17815
 fabricatio/capabilities/review.py,sha256=l06BYcQzPi7VKmWdplj9L6WvZEscZqW1Wx9OhR-UnNw,5061
 fabricatio/capabilities/task.py,sha256=-b92cGi7b3B30kOSS-90_H6BjA0VF_cjc1BzPbO5MkI,4401
-fabricatio/capabilities/__init__.py,sha256=v1cHRHIJ2gxyqMLNCs6ERVcCakSasZNYzmMI4lqAcls,57
 fabricatio/decorators.py,sha256=OohwKgc5dUjybv70D-J2lA0C9zjUuq8-gzU5O8JPl8w,8962
 fabricatio/emitter.py,sha256=n4vH6E7lcT57qVve_3hUAdfvj0mQUDkWu6iU5aNztB8,6341
+fabricatio/fs/__init__.py,sha256=USoMI_HcIr3Yc77_JQYYsXrsplYPXtFTaNB9YgFfC4s,713
 fabricatio/fs/curd.py,sha256=652nHulbJ3gwt0Z3nywtPMmjhEyglDvEfc3p7ieJNNA,4777
 fabricatio/fs/readers.py,sha256=UXvcJO3UCsxHu9PPkg34Yh55Zi-miv61jD_wZQJgKRs,1751
-fabricatio/fs/__init__.py,sha256=USoMI_HcIr3Yc77_JQYYsXrsplYPXtFTaNB9YgFfC4s,713
 fabricatio/journal.py,sha256=mnbdB1Dw-mhEKIgWlPKn7W07ssg-6dmxMXilIGQMFV8,216
 fabricatio/models/action.py,sha256=RhjHaEJILiCZux5hzxSZVt_7Evcu3TnFHNuJN8rzgq8,10052
 fabricatio/models/adv_kwargs_types.py,sha256=IBV3ZcsNLvvEjO_2hBpYg_wLSpNKaMx6Ndam3qXJCw8,2097
+fabricatio/models/extra/__init__.py,sha256=XlYnS_2B9nhLhtQkjE7rvvfPmAAtXVdNi9bSDAR-Ge8,54
 fabricatio/models/extra/advanced_judge.py,sha256=INUl_41C8jkausDekkjnEmTwNfLCJ23TwFjq2cM23Cw,1092
-fabricatio/models/extra/aricle_rag.py,sha256=fTxlQyrzyl9bLCC5Zreb71TKaJ7xiHqqyR62HXr2unQ,11935
-fabricatio/models/extra/article_base.py,sha256=Hm78qqqkZpeLlt979cgGiam3LE4CussLv0nQkNC9mW8,16769
+fabricatio/models/extra/aricle_rag.py,sha256=K2ceihjIJFfkE82ykNH8hwTLDbXfFRZAXhz1NkdVZSA,12011
+fabricatio/models/extra/article_base.py,sha256=1azdvRfT9lKiPT3xVTlP26FfzJOTwMlueJQDee3TgOU,16783
 fabricatio/models/extra/article_essence.py,sha256=z3Qz6xVsB9k-K-c4Y2CoKzxZrXaUd4oyt2Mb6hGDYdg,2793
-fabricatio/models/extra/article_main.py,sha256=S6eUUBuzBzRcuFe8hPhaFvL7UhyO1gaUCyJkNbJIgz8,11278
+fabricatio/models/extra/article_main.py,sha256=7fH_pVjIDtaz4Gw885FU1m29puKgm3nHoowrVeWdMIE,11106
 fabricatio/models/extra/article_outline.py,sha256=P0T-1DGCzoNmQ3iQVwSmOul0nwS6qLgr0FF8jDdD7F0,1673
 fabricatio/models/extra/article_proposal.py,sha256=OQIKoJkmJv0ogYVk7eGK_TOtANVYcBPA_HeV1nuG0Vo,1909
 fabricatio/models/extra/patches.py,sha256=_WNCxtYzzsVfUxI16vu4IqsLahLYRHdbQN9er9tqhC0,997
 fabricatio/models/extra/problem.py,sha256=8tTU-3giFHOi5j7NJsvH__JJyYcaGrcfsRnkzQNm0Ew,7216
 fabricatio/models/extra/rag.py,sha256=C7ptZCuGJmT8WikjpF9KhZ0Bw-VicdB-s8EqEAgMLKE,3967
 fabricatio/models/extra/rule.py,sha256=WKahNiaIp8s_l2r_FG21F_PP3_hgNm4hfSVCSFyfoBE,2669
-fabricatio/models/extra/__init__.py,sha256=XlYnS_2B9nhLhtQkjE7rvvfPmAAtXVdNi9bSDAR-Ge8,54
 fabricatio/models/generic.py,sha256=OJrYClooL2XnyalWTyyLgorycA1d_JNW8VqOYNDJdXc,27873
 fabricatio/models/kwargs_types.py,sha256=Ik8-Oi_NmwfkvC9B8K4NsoZc_vSWV85xKCSthA1Xv_k,3403
 fabricatio/models/role.py,sha256=b3zg96YKDsMBqa7SIe9LQHc-IVs2fGWqoQeRQYQIl4o,3856
 fabricatio/models/task.py,sha256=XZ1l1P-iS02ZF9P8cXv8gEfJKBa17PFPNJ1SbhyhT4Q,11033
-fabricatio/models/tool.py,sha256=K7XYG_DnlM5IfFabe4LwEZ3DtlSCcf5LZOKbeDWBH14,12419
+fabricatio/models/tool.py,sha256=q2wDtZAebWMZlsFifgmJq8N3XvAhVNMb0aUIKkdruGc,12419
 fabricatio/models/usages.py,sha256=q2jLqa0vJ7ho9ZUkC-2uPuFpK8uClBLIS6TEEYHUotY,33041
 fabricatio/parser.py,sha256=dYFri9pDlsiwVpEJ-a5jmVU2nFuKN3uBHC8VsVpdEm8,4642
 fabricatio/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-fabricatio/rust.pyi,sha256=dlssKYR2RrS9UQb9bc3y6XP7a9Clq1Qr40dzMW_vNB0,26256
+fabricatio/rust.cp313-win_amd64.pyd,sha256=fMFlamARxZ3hNaXJGCkFN6C2BYuPgZI1poi3QE9JIys,7830016
+fabricatio/rust.pyi,sha256=PK9KfYSUto6nZ2KaCmF5sEVWCNizdtgOLhJhc1E7w20,25962
+fabricatio/toolboxes/__init__.py,sha256=KBJi5OG_pExscdlM7Bnt_UF43j4I3Lv6G71kPVu4KQU,395
 fabricatio/toolboxes/arithmetic.py,sha256=WLqhY-Pikv11Y_0SGajwZx3WhsLNpHKf9drzAqOf_nY,1369
 fabricatio/toolboxes/fs.py,sha256=l4L1CVxJmjw9Ld2XUpIlWfV0_Fu_2Og6d3E13I-S4aE,736
-fabricatio/toolboxes/__init__.py,sha256=KBJi5OG_pExscdlM7Bnt_UF43j4I3Lv6G71kPVu4KQU,395
 fabricatio/utils.py,sha256=WYhFB4tHk6jKmjZgAsYhRmg1ZvBjn4X2y4n7yz25HjE,5454
+fabricatio/workflows/__init__.py,sha256=5ScFSTA-bvhCesj3U9Mnmi6Law6N1fmh5UKyh58L3u8,51
 fabricatio/workflows/articles.py,sha256=ObYTFUqLUk_CzdmmnX6S7APfxcGmPFqnFr9pdjU7Z4Y,969
 fabricatio/workflows/rag.py,sha256=-YYp2tlE9Vtfgpg6ROpu6QVO8j8yVSPa6yDzlN3qVxs,520
-fabricatio/workflows/__init__.py,sha256=5ScFSTA-bvhCesj3U9Mnmi6Law6N1fmh5UKyh58L3u8,51
-fabricatio/__init__.py,sha256=w7ObFg6ud4pQuC1DhVyQI9x9dtp05QrcJAEk643iJmc,761
-fabricatio/rust.cp313-win_amd64.pyd,sha256=-Xps_ojYfsRw7s4AWzvPcftvF2bwbhluP_KXKTJAMns,7819776
-fabricatio-0.3.14.dev4.data/scripts/tdown.exe,sha256=rcQpG_q-9J97Dg_hgHJhUiMdfXBvSwbbrOXHYpx8xoo,3448320
-fabricatio-0.3.14.dev4.data/scripts/ttm.exe,sha256=PkRspfDbHwEkKoBkgftRPu_GGCBOCsuuv61IwV4lJTw,2555392
-fabricatio-0.3.14.dev4.dist-info/RECORD,,
+fabricatio-0.3.14.dev7.dist-info/RECORD,,

{fabricatio-0.3.14.dev4.dist-info → fabricatio-0.3.14.dev7.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: maturin (1.8.3)
+Generator: maturin (1.8.4)
 Root-Is-Purelib: false
 Tag: cp313-cp313-win_amd64