fabricatio 0.3.14.dev2__cp312-cp312-manylinux_2_34_x86_64.whl → 0.3.14.dev3__cp312-cp312-manylinux_2_34_x86_64.whl

fabricatio/__init__.py

@@ -1,19 +1,17 @@
 """Fabricatio is a Python library for building llm app using event-based agent structure."""
 
-from fabricatio import actions, capabilities, parser, toolboxes, workflows
+from fabricatio import actions, capabilities, fs, models, parser, toolboxes, utils, workflows
 from fabricatio.journal import logger
-from fabricatio.models import extra
 from fabricatio.models.action import Action, WorkFlow
 from fabricatio.models.role import Role
 from fabricatio.models.task import Task
 from fabricatio.models.tool import ToolBox
-from fabricatio.rust import CONFIG, TEMPLATE_MANAGER, BibManager, Event
+from fabricatio.rust import CONFIG, TEMPLATE_MANAGER, Event
 
 __all__ = [
     "CONFIG",
     "TEMPLATE_MANAGER",
     "Action",
-    "BibManager",
     "Event",
     "Role",
     "Task",
@@ -21,9 +19,11 @@ __all__ = [
     "WorkFlow",
     "actions",
     "capabilities",
-    "extra",
+    "fs",
     "logger",
+    "models",
     "parser",
     "toolboxes",
+    "utils",
     "workflows",
 ]

fabricatio/models/action.py

@@ -125,10 +125,10 @@ class WorkFlow(WithBriefing):
     steps: Sequence[Union[Type[Action], Action]] = Field(frozen=True)
     """The sequence of actions to be executed, can be action classes or instances."""
 
-    task_input_key: ClassVar[str] = Field(default=INPUT_KEY)
+    task_input_key: ClassVar[str] = INPUT_KEY
     """Key used to store the input task in the context dictionary."""
 
-    task_output_key: ClassVar[str] = Field(default=OUTPUT_KEY)
+    task_output_key: ClassVar[str] = OUTPUT_KEY
     """Key used to extract the final result from the context dictionary."""
 
     extra_init_context: Dict[str, Any] = Field(default_factory=dict, frozen=True)

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.capabilities.persist import PersistentAble
 from fabricatio.decorators import precheck_package
 from fabricatio.journal import logger
@@ -18,16 +27,8 @@ from fabricatio.models.extra.article_outline import (
     ArticleSubsectionOutline,
 )
 from fabricatio.models.generic import Described, 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"
+PARAGRAPH_SEP = "// - - -"
 
 
 class Paragraph(SketchedAble, WordCount, Described):
@@ -50,7 +51,7 @@ class Paragraph(SketchedAble, WordCount, Described):
     @classmethod
     def from_content(cls, content: str) -> Self:
         """Create a Paragraph object from the given content."""
-        return cls(elaboration="", aims=[], expected_word_count=word_count(content), content=content)
+        return cls(elaboration="", aims=[], expected_word_count=word_count(content), content=content.strip())
 
     @property
     def exact_wordcount(self) -> int:
@@ -82,8 +83,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!"
 
@@ -103,7 +104,7 @@ class ArticleSubsection(SubSectionBase):
         Returns:
             str: Typst code snippet for rendering.
         """
-        return super().to_typst_code() + PARAGRAPH_SEP.join(p.content for p in self.paragraphs)
+        return super().to_typst_code() + f"\n\n{PARAGRAPH_SEP}\n\n".join(p.content for p in self.paragraphs)
 
     @classmethod
     def from_typst_code(cls, title: str, body: str, **kwargs) -> Self:

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 WithBriefing
-from fabricatio.rust import CONFIG
-from pydantic import BaseModel, ConfigDict, Field
+from fabricatio.models.generic import Base, WithBriefing
 
 
 class Tool[**P, R](WithBriefing):
@@ -181,7 +181,7 @@ class ToolBox(WithBriefing):
         return hash(self.briefing)
 
 
-class ToolExecutor(BaseModel):
+class ToolExecutor(Base):
     """A class representing a tool executor with a sequence of tools to execute.
 
     This class manages a sequence of tools and provides methods to inject tools and data into a module, execute the tools,
@@ -192,7 +192,6 @@ class ToolExecutor(BaseModel):
         data (Dict[str, Any]): The data that could be used when invoking the tools.
     """
 
-    model_config = ConfigDict(use_attribute_docstrings=True)
     candidates: List[Tool] = Field(default_factory=list, frozen=True)
     """The sequence of tools to execute."""
 

fabricatio/models/usages.py

@@ -6,7 +6,6 @@ from asyncio import gather
 from typing import Callable, Dict, Iterable, List, Literal, Optional, Self, Sequence, Set, Union, Unpack, overload
 
 import asyncstdlib
-import litellm
 from fabricatio.decorators import logging_exec_time
 from fabricatio.journal import logger
 from fabricatio.models.generic import ScopedConfig, WithBriefing
@@ -15,7 +14,12 @@ from fabricatio.models.task import Task
 from fabricatio.models.tool import Tool, ToolBox
 from fabricatio.rust import CONFIG, TEMPLATE_MANAGER
 from fabricatio.utils import first_available, ok
-from litellm import RateLimitError, Router, stream_chunk_builder  # pyright: ignore [reportPrivateImportUsage]
+from litellm import (  # pyright: ignore [reportPrivateImportUsage]
+    RateLimitError,
+    Router,
+    aembedding,
+    stream_chunk_builder,
+)
 from litellm.types.router import Deployment, LiteLLM_Params, ModelInfo
 from litellm.types.utils import (
     Choices,
@@ -534,7 +538,7 @@ class EmbeddingUsage(LLMUsage, ABC):
             logger.error(err := f"Input text exceeds maximum sequence length {max_len}, got {length}.")
             raise ValueError(err)
 
-        return await litellm.aembedding(
+        return await aembedding(
             input=input_text,
             caching=caching or self.embedding_caching or CONFIG.embedding.caching,
             dimensions=dimensions or self.embedding_dimensions or CONFIG.embedding.dimensions,

{fabricatio-0.3.14.dev2.dist-info → fabricatio-0.3.14.dev3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fabricatio
-Version: 0.3.14.dev2
+Version: 0.3.14.dev3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: 3.12
@@ -18,17 +18,13 @@ Requires-Dist: pydantic>=2.10.6
 Requires-Dist: pymitter>=1.0.0
 Requires-Dist: rich>=13.9.4
 Requires-Dist: ujson>=5.10.0
-Requires-Dist: fabricatio[calc,ftd,plot,qa,rag,cli] ; extra == 'full'
+Requires-Dist: fabricatio[ftd,qa,rag,cli] ; extra == 'full'
 Requires-Dist: pymilvus>=2.5.4 ; extra == 'rag'
-Requires-Dist: sympy>=1.13.3 ; extra == 'calc'
-Requires-Dist: matplotlib>=3.10.1 ; extra == 'plot'
 Requires-Dist: questionary>=2.1.0 ; extra == 'qa'
 Requires-Dist: magika>=0.6.1 ; extra == 'ftd'
 Requires-Dist: typer-slim[standard]>=0.15.2 ; extra == 'cli'
 Provides-Extra: full
 Provides-Extra: rag
-Provides-Extra: calc
-Provides-Extra: plot
 Provides-Extra: qa
 Provides-Extra: ftd
 Provides-Extra: cli
@@ -36,7 +32,7 @@ License-File: LICENSE
 Summary: A LLM multi-agent framework.
 Keywords: ai,agents,multi-agent,llm,pyo3
 Author-email: Whth <zettainspector@foxmail.com>
-Requires-Python: >=3.12, <3.13
+Requires-Python: >=3.12, <3.14
 Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
 Project-URL: Homepage, https://github.com/Whth/fabricatio
 Project-URL: Repository, https://github.com/Whth/fabricatio

{fabricatio-0.3.14.dev2.dist-info → fabricatio-0.3.14.dev3.dist-info}/RECORD

@@ -1,6 +1,6 @@
-fabricatio-0.3.14.dev2.dist-info/METADATA,sha256=U3sR1Y1drSmUVygYiXGtZl9hJ_mBMlS_HKAeizPrg-Y,5120
-fabricatio-0.3.14.dev2.dist-info/WHEEL,sha256=7FgAcpQES0h1xhfN9Ugve9FTUilU6sRAr1WJ5ph2cuw,108
-fabricatio-0.3.14.dev2.dist-info/licenses/LICENSE,sha256=yDZaTLnOi03bi3Dk6f5IjhLUc5old2yOsihHWU0z-i0,1067
+fabricatio-0.3.14.dev3.dist-info/METADATA,sha256=XzC31PvJH1THOhtpVgvvWm-yLmE7yQzafpevemUEQTU,4969
+fabricatio-0.3.14.dev3.dist-info/WHEEL,sha256=7FgAcpQES0h1xhfN9Ugve9FTUilU6sRAr1WJ5ph2cuw,108
+fabricatio-0.3.14.dev3.dist-info/licenses/LICENSE,sha256=yDZaTLnOi03bi3Dk6f5IjhLUc5old2yOsihHWU0z-i0,1067
 fabricatio/emitter.py,sha256=QpMvs8dTy1zs5iDORFKzA615S3Lb1tm6AQxYBemQGcc,6164
 fabricatio/capabilities/check.py,sha256=eiZZaiX78k-Zt7-Ik43Pn5visXHeOJLk8yLWgtqln40,8379
 fabricatio/capabilities/propose.py,sha256=KqeXaUURJ6O-Ve0ijZYg88rgQYCZEFbuWoqIepI-nQ8,1965
@@ -16,10 +16,10 @@ fabricatio/capabilities/advanced_judge.py,sha256=wTiTyBxkZfOXsmzULOW4nX-QAwFMz9w
 fabricatio/capabilities/review.py,sha256=rxA_qdnJc8ehytL5EnlKo9QJ99stnF-n6YaBFRYLe5I,4947
 fabricatio/capabilities/__init__.py,sha256=skaJ43CqAQaZMH-mCRzF4Fps3x99P2SwJ8vSM9pInX8,56
 fabricatio/parser.py,sha256=3vT5u5SGpzDH4WLJdMwK5CP8RqO4g1MyQUYpiDKDoEo,4528
-fabricatio/models/action.py,sha256=mqQfD90_xa-YEl2JRyQLeSyItJFGcQLMxLTVthJIWG0,9819
+fabricatio/models/action.py,sha256=O8BLh8fRNqde_3PC7OFHBjLTdLRPvy5mtalMqQFaZXs,9789
 fabricatio/models/extra/article_outline.py,sha256=71mgx66KRiXBtdYId4WNkAYp9tJ7OhUqmQyOEe7IRxI,1627
 fabricatio/models/extra/article_essence.py,sha256=lAkfGj4Jqiy3dSmtloVVr2krej76TV1Ky-2Fr6pNE_Q,2692
-fabricatio/models/extra/article_main.py,sha256=uZAEVYDCuda3zFm-3hxLZx3peU-GYxMcCK2kpHp_E-Y,10971
+fabricatio/models/extra/article_main.py,sha256=TeOhU8EDUok2t_lLl1oaUn5IhOmp3oZW_YFOSnybX3s,10993
 fabricatio/models/extra/article_proposal.py,sha256=7OgcsS9ujjSi_06Z1ln4SCDQgrS4xPGrtgc2dv8EzGo,1857
 fabricatio/models/extra/article_base.py,sha256=2auA10qomj8UnT2tunUf0oQnvhrJBKNV7GF6UWA9tjg,16347
 fabricatio/models/extra/rag.py,sha256=fwyEXOECQNe8LPUKGAxEcp9vp7o5356rna-TzGpkvnE,3869
@@ -29,13 +29,13 @@ fabricatio/models/extra/patches.py,sha256=_ghmnlvTZQq7UJyaH77mTZE9abjvxRJ2mgWHUb
 fabricatio/models/extra/advanced_judge.py,sha256=CKPP4Lseb_Ey8Y7i2V9HJfB-mZgCknFdqq7Zo41o6s4,1060
 fabricatio/models/extra/aricle_rag.py,sha256=egUZPmHkzA48IU8s9f6WRhqVMI8B8Uz8Amx-WkLLWGE,11651
 fabricatio/models/extra/__init__.py,sha256=0R9eZsCNu6OV-Xtf15H7FrqhfHTFBFf3fBrcd7ChsJ0,53
-fabricatio/models/usages.py,sha256=LmaF7ajreU3PqFR-EqWavFzzHa6Uvc_RgKq1YulLk-E,32240
+fabricatio/models/usages.py,sha256=bpM-a9i-WpSOh-XL3LiYTa3AxQUd_ckn44lh-uuKM6M,32250
 fabricatio/models/generic.py,sha256=dGap-ckYy7ZX_lXDNxv4d3yM45vdoLDYW4cl49BbCAY,27061
 fabricatio/models/adv_kwargs_types.py,sha256=nmj1D0GVosZxKcdiw-B5vJB04Whr5zh30ZBJntSZUpY,2034
 fabricatio/models/role.py,sha256=tOwzILaTb8QUOddy9RrJRyhfB_pEVv_IiUBRuc6ylH8,3761
 fabricatio/models/task.py,sha256=CdR1Zbf-lZN0jODj9iriTn1X2DxLxjXlvZgy3kEd6lI,10723
 fabricatio/models/kwargs_types.py,sha256=VrzAJaOSlQ-xN5NIIi3k4KpIY0c9beuxcuUnF-mkEEk,3282
-fabricatio/models/tool.py,sha256=_cwAPVD1_u6t8uK4rRhi7upyqpadj_jiKbOOl0baXv0,12174
+fabricatio/models/tool.py,sha256=_vL5aq5BFjclRxbcNkQCmsMtLUikysv-7Og5HRNc6-U,12091
 fabricatio/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 fabricatio/rust.pyi,sha256=IzEpNSt3tgoR3L2xZNs7skdeGW73L72-eizXQCBWZFk,25410
 fabricatio/actions/article.py,sha256=8ea9QZk7m21j5fw6_CO_znZtik9_o71JmX77Po5gyS4,12188
@@ -57,8 +57,8 @@ fabricatio/toolboxes/fs.py,sha256=OQMdeokYxSNVrCZJAweJ0cYiK4k2QuEiNdIbS5IHIV8,70
 fabricatio/toolboxes/__init__.py,sha256=dYm_Gd8XolSU_h4wnkA09dlaLDK146eeFz0CUgPZ8_c,380
 fabricatio/utils.py,sha256=qvl4R8ThuNIIoBJuR1DGEuWYZ7jRFT_8SRx4I_FA8pU,5298
 fabricatio/journal.py,sha256=qZoaPdv17fc_9l2xVZ-ve7dXKmMFJ8MzPa8_vNXMGyE,204
-fabricatio/__init__.py,sha256=JEjsybquOOQZqPxkTo_84OT7XfOLxRhzRUormbR76f4,755
-fabricatio/rust.cpython-312-x86_64-linux-gnu.so,sha256=wptqK5Es3gOyR3A7knxG4l4H0mYIElklmUMSMEANGOY,7912480
-fabricatio-0.3.14.dev2.data/scripts/tdown,sha256=CWdPVEzyPPmxlKvB9i3Wh4iXIqVDy2IFus4M7HkEHaU,4725472
-fabricatio-0.3.14.dev2.data/scripts/ttm,sha256=U7tjKVGiOvvBPJiGyoLxhFBemYeH2tr3R95Tc00Mz5Y,3924088
-fabricatio-0.3.14.dev2.dist-info/RECORD,,
+fabricatio/__init__.py,sha256=pSLe6QL4zQGaZXfhF9KW4fa1D8chqCQm_7yInCP6Kt8,732
+fabricatio/rust.cpython-312-x86_64-linux-gnu.so,sha256=aDA9n89GtZxKC6FRPoQu6d3lF2j-cDo0P9ZEOa9MYvI,7908728
+fabricatio-0.3.14.dev3.data/scripts/tdown,sha256=IC8p9rZ8AzIwf51yUpOFChs_-1SE3Qq_LB1ZK6O2I3s,4726096
+fabricatio-0.3.14.dev3.data/scripts/ttm,sha256=O_3KVKRssOajmBy5mEmY02QHQHH4IDAs5wCvz4GjeME,3925048
+fabricatio-0.3.14.dev3.dist-info/RECORD,,