fabricatio 0.2.0.dev12__cp312-cp312-manylinux_2_34_x86_64.whl → 0.2.0.dev13__cp312-cp312-manylinux_2_34_x86_64.whl

fabricatio/models/generic.py

@@ -17,6 +17,7 @@ from pydantic import (
     Field,
     HttpUrl,
     NonNegativeFloat,
+    NonNegativeInt,
     PositiveInt,
     SecretStr,
 )
@@ -295,6 +296,7 @@ class LLMUsage(Base):
         self,
         instruction: str,
         choices: List[T],
+        k: NonNegativeInt = 0,
         max_validations: PositiveInt = 2,
         system_message: str = "",
         model: str | None = None,
@@ -311,6 +313,7 @@ class LLMUsage(Base):
         Args:
             instruction: The user-provided instruction/question description.
             choices: A list of candidate options, requiring elements to have `name` and `briefing` fields.
+            k: The number of choices to select, 0 means infinite.
             max_validations: Maximum number of validation failures, default is 2.
             system_message: Custom system-level prompt, defaults to an empty string.
             model: The name of the LLM model to use.
@@ -332,14 +335,17 @@ class LLMUsage(Base):
         """
         prompt = template_manager.render_template(
             "make_choice",
-            {"instruction": instruction, "options": [m.model_dump(include={"name", "briefing"}) for m in choices]},
+            {
+                "instruction": instruction,
+                "options": [m.model_dump(include={"name", "briefing"}) for m in choices],
+                "k": k,
+            },
         )
         names = [c.name for c in choices]
 
         def _validate(response: str) -> List[T] | None:
-            cap = JsonCapture.capture(response)
-            ret = orjson.loads(cap)
-            if not isinstance(ret, List):
+            ret = JsonCapture.convert_with(response, orjson.loads)
+            if not isinstance(ret, List) or len(ret) != k:
                 return None
             if any(n not in names for n in ret):
                 return None
@@ -360,6 +366,70 @@ class LLMUsage(Base):
             max_retries=max_retries,
         )
 
+    async def ajudge(
+        self,
+        prompt: str,
+        affirm_case: str = "",
+        deny_case: str = "",
+        max_validations: PositiveInt = 2,
+        system_message: str = "",
+        model: str | None = None,
+        temperature: NonNegativeFloat | None = None,
+        stop: str | List[str] | None = None,
+        top_p: NonNegativeFloat | None = None,
+        max_tokens: PositiveInt | None = None,
+        stream: bool | None = None,
+        timeout: PositiveInt | None = None,
+        max_retries: PositiveInt | None = None,
+    ) -> bool:
+        """Asynchronously judges a prompt using AI validation.
+
+        Args:
+            prompt (str): The input prompt to be judged.
+            affirm_case (str, optional): The affirmative case for the AI model. Defaults to "".
+            deny_case (str, optional): The negative case for the AI model. Defaults to "".
+            max_validations (PositiveInt, optional): Maximum number of validation attempts. Defaults to 2.
+            system_message (str, optional): System message for the AI model. Defaults to "".
+            model (str | None, optional): AI model to use. Defaults to None.
+            temperature (NonNegativeFloat | None, optional): Sampling temperature. Defaults to None.
+            stop (str | List[str] | None, optional): Stop sequences. Defaults to None.
+            top_p (NonNegativeFloat | None, optional): Nucleus sampling parameter. Defaults to None.
+            max_tokens (PositiveInt | None, optional): Maximum number of tokens to generate. Defaults to None.
+            stream (bool | None, optional): Whether to stream the response. Defaults to None.
+            timeout (PositiveInt | None, optional): Timeout in seconds. Defaults to None.
+            max_retries (PositiveInt | None, optional): Maximum number of retries. Defaults to None.
+
+        Returns:
+            bool: The judgment result (True or False) based on the AI's response.
+
+        Notes:
+            The method uses an internal validator to ensure the response is a boolean value.
+            If the response cannot be converted to a boolean, it will return None.
+        """
+
+        def _validate(response: str) -> bool | None:
+            ret = JsonCapture.convert_with(response, orjson.loads)
+            if not isinstance(ret, bool):
+                return None
+            return ret
+
+        return await self.aask_validate(
+            question=template_manager.render_template(
+                "make_judgment", {"prompt": prompt, "affirm_case": affirm_case, "deny_case": deny_case}
+            ),
+            validator=_validate,
+            max_validations=max_validations,
+            system_message=system_message,
+            model=model,
+            temperature=temperature,
+            stop=stop,
+            top_p=top_p,
+            max_tokens=max_tokens,
+            stream=stream,
+            timeout=timeout,
+            max_retries=max_retries,
+        )
+
     def fallback_to(self, other: "LLMUsage") -> Self:
         """Fallback to another instance's attribute values if the current instance's attributes are None.
 

fabricatio/models/task.py

@@ -268,8 +268,8 @@ class ProposeTask(LLMUsage, WithBriefing):
             try:
                 cap = JsonCapture.capture(response)
                 logger.debug(f"Response: \n{response}")
-                logger.info(f"Captured JSON: \n{cap[0]}")
-                return Task.model_validate_json(cap[0] if cap else response)
+                logger.info(f"Captured JSON: \n{cap}")
+                return Task.model_validate_json(cap)
             except ValidationError as e:
                 logger.error(f"Failed to parse task from JSON: {e}")
                 return None

fabricatio/parser.py

@@ -1,11 +1,13 @@
 """A module to parse text using regular expressions."""
 
-from typing import Any, Self, Tuple
+from typing import Any, Callable, Self, Tuple
 
 import regex
 from pydantic import BaseModel, ConfigDict, Field, PositiveInt, PrivateAttr
 from regex import Pattern, compile
 
+from fabricatio.journal import logger
+
 
 class Capture(BaseModel):
     """A class to capture patterns in text using regular expressions.
@@ -32,7 +34,7 @@ class Capture(BaseModel):
         """
         self._compiled = compile(self.pattern, self.flags)
 
-    def capture(self, text: str) -> Tuple[str, ...] | None:
+    def capture(self, text: str) -> Tuple[str, ...] | str | None:
         """Capture the first occurrence of the pattern in the given text.
 
         Args:
@@ -48,7 +50,25 @@ class Capture(BaseModel):
 
         if self.target_groups:
             return tuple(match.group(g) for g in self.target_groups)
-        return (match.group(),)
+        return match.group(1)
+
+    def convert_with[T](self, text: str, convertor: Callable[[Tuple[str, ...]], T] | Callable[[str], T]) -> T | None:
+        """Convert the given text using the pattern.
+
+        Args:
+            text (str): The text to search the pattern in.
+            convertor (Callable[[Tuple[str, ...]], T] | Callable[[str], T]): The function to convert the captured text.
+
+        Returns:
+            str | None: The converted text if the pattern is found, otherwise None.
+        """
+        if cap := self.capture(text) is None:
+            return None
+        try:
+            return convertor(cap)
+        except ValueError as e:
+            logger.error(f"Failed to convert text using convertor: {convertor.__name__}, error: \n{e}")
+            return None
 
     @classmethod
     def capture_code_block(cls, language: str) -> Self:
@@ -60,7 +80,7 @@ class Capture(BaseModel):
         Returns:
             Self: The instance of the class with the captured code block.
         """
-        return cls(pattern=f"```{language}\n(.*?)\n```", target_groups=(1,))
+        return cls(pattern=f"```{language}\n(.*?)\n```")
 
 
 JsonCapture = Capture.capture_code_block("json")

fabricatio/toolboxes/arithmetic.py

@@ -1,3 +1,5 @@
+"""Arithmetic tools for Fabricatio."""
+
 from fabricatio.models.tool import ToolBox
 
 arithmetic_tools = ToolBox(name="ArithmeticToolBox", description="A toolbox for arithmetic operations.")

{fabricatio-0.2.0.dev12.dist-info → fabricatio-0.2.0.dev13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fabricatio
-Version: 0.2.0.dev12
+Version: 0.2.0.dev13
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: 3.12
@@ -28,6 +28,9 @@ 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
+Project-URL: Homepage, https://github.com/Whth/fabricatio
+Project-URL: Repository, https://github.com/Whth/fabricatio
+Project-URL: Issues, https://github.com/Whth/fabricatio/issues
 
 # Fabricatio
 

{fabricatio-0.2.0.dev12.dist-info → fabricatio-0.2.0.dev13.dist-info}/RECORD

@@ -1,19 +1,19 @@
-fabricatio-0.2.0.dev12.dist-info/METADATA,sha256=gWEeKNwVXLETbzTMO9Qd2rkbN_TLH4sdpedRVvW8myc,5824
-fabricatio-0.2.0.dev12.dist-info/WHEEL,sha256=RIvmwLDYujv60MYBx2jxyP4vdn1DD7X0kBgz1TQvZuc,108
-fabricatio-0.2.0.dev12.dist-info/licenses/LICENSE,sha256=yDZaTLnOi03bi3Dk6f5IjhLUc5old2yOsihHWU0z-i0,1067
+fabricatio-0.2.0.dev13.dist-info/METADATA,sha256=Q4r0vbpk3dJdnWiK1pnt8aj19fPelsaKZH1XXtAw68E,6005
+fabricatio-0.2.0.dev13.dist-info/WHEEL,sha256=RIvmwLDYujv60MYBx2jxyP4vdn1DD7X0kBgz1TQvZuc,108
+fabricatio-0.2.0.dev13.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/toolboxes/arithmetic.py,sha256=MbS1-a4uLGqs7ZypH2dtrTa0d2-Rr8ifok-1mrVfVb4,1257
+fabricatio/toolboxes/arithmetic.py,sha256=bG4ngqBrTaj4zHxpMTbK-JQ4Si96UiiHz7fufSQWu-0,1297
 fabricatio/__init__.py,sha256=UE2qARwRgL3T1NgnzK9b4Cx3eksUvNZmN_rf-dycM4E,881
 fabricatio/core.py,sha256=apwXgI94DCWpGujGlsmXsTZQvJOQMB9llmuUo7ohd-4,5771
 fabricatio/config.py,sha256=Or_PzOzXy_4Rqje1r788sID4bxILXhNC3r7eelcZJy0,7852
 fabricatio/decorators.py,sha256=UgW8_pV_RUm0a69co1POnKvJHh9Qy2R0pyh1BtEKzFI,1169
 fabricatio/journal.py,sha256=u6YTKrD9EoebbR8guvAe5zSqWV1nS-BGyKxZR_Sw2ik,684
-fabricatio/models/generic.py,sha256=Q92xkAL1S8zWMCGDUNQahT9atFamZKhHzKbI1OXjBmk,19264
+fabricatio/models/generic.py,sha256=pV84sWgZZItgotFTchfyixSLSnmv5GkumSv_0frPK38,22402
 fabricatio/models/action.py,sha256=VL_A8BWOuCYQCjwCAm0vExA88KSwm6VkluXHOtHijUw,4882
-fabricatio/models/task.py,sha256=-YB9SGqOYrPmk7ce5Go3OO5Cc7Q_KB7QdmeG49GeZpY,9090
+fabricatio/models/task.py,sha256=2jSttV7Yod2Waz16RByFZrRZT07YThs5Vw5pgbjbD1s,9063
 fabricatio/models/role.py,sha256=1vhpHRKJ6f7PDBLl6jYRZbxTJyrVhNN3FEEh_HAwU3o,990
 fabricatio/models/tool.py,sha256=PCsX5hdf6YnslkyoPA8FwybGvrO0fJFJpHsheBwg2bo,4886
 fabricatio/models/events.py,sha256=S0E7l5iQoJYdUn0pSSziHdXqPzIvvLy8ifVPQxXeAho,2570
@@ -24,7 +24,7 @@ fabricatio/actions/__init__.py,sha256=eLa_5ACZ-FqdrLtOfCHk5nQBxzhIs1kgMIXWmkm2P8
 fabricatio/actions/communication.py,sha256=4hNzyv5_j4HU291dDuBmBEb66VihQPI9SLRVZrxWu1Y,412
 fabricatio/actions/transmission.py,sha256=gLLKoi4keL64uaEkHDxwVrBGciheupQrO2fW3GkjfEw,1156
 fabricatio/_rust.pyi,sha256=EwMPoCZFOCYZ5h_9PZLvO3YrgBEG0Hwae0ux0hcBabM,1487
-fabricatio/parser.py,sha256=CAI7S3u2xI_-s1JWb4v2GkhiNeX1kr_C2olxy98KLR8,2400
-fabricatio/_rust.cpython-312-x86_64-linux-gnu.so,sha256=JUDe4N51fXgkY7c-vEXnFJaTBWzwwh93AwJPk_QWoGE,1190824
-fabricatio-0.2.0.dev12.data/scripts/tdown,sha256=7IEbm2H2hkcCNmm8b4BEfHtnauc5oO9fIiGn_UXYb3M,4549648
-fabricatio-0.2.0.dev12.dist-info/RECORD,,
+fabricatio/parser.py,sha256=Ef-uOmrx-1wQqNlcPyPf_LgKg4WLuCzjWgbHGtTTK5o,3190
+fabricatio/_rust.cpython-312-x86_64-linux-gnu.so,sha256=MXQf097U_QTbRfeJjDbsmUt68b-5bXBUmgIIRNuTQ7w,1189720
+fabricatio-0.2.0.dev13.data/scripts/tdown,sha256=Rzx0OwwHeu8DOmxOeyI_3m4EmyU70iWG4oR8NgLmoes,4549888
+fabricatio-0.2.0.dev13.dist-info/RECORD,,