fabricatio 0.2.6.dev5__cp39-cp39-win_amd64.whl → 0.2.6.dev7__cp39-cp39-win_amd64.whl

fabricatio/models/kwargs_types.py

@@ -1,19 +1,29 @@
 """This module contains the types for the keyword arguments of the methods in the models module."""
 
+from importlib.util import find_spec
 from typing import Any, Required, TypedDict
 
 from litellm.caching.caching import CacheMode
 from litellm.types.caching import CachingSupportedCallTypes
 
+if find_spec("pymilvus"):
+    from pymilvus import CollectionSchema
+    from pymilvus.milvus_client import IndexParams
 
-class CollectionSimpleConfigKwargs(TypedDict, total=False):
-    """Configuration parameters for a vector collection.
+    class CollectionConfigKwargs(TypedDict, total=False):
+        """Configuration parameters for a vector collection.
 
-    These arguments are typically used when configuring connections to vector databases.
-    """
+        These arguments are typically used when configuring connections to vector databases.
+        """
 
-    dimension: int | None
-    timeout: float
+        dimension: int | None
+        primary_field_name: str
+        id_type: str
+        vector_field_name: str
+        metric_type: str
+        timeout: float | None
+        schema: CollectionSchema | None
+        index_params: IndexParams | None
 
 
 class FetchKwargs(TypedDict, total=False):
@@ -81,6 +91,7 @@ class ValidateKwargs[T](GenerateKwargs, total=False):
 
     default: T
     max_validations: int
+    co_extractor: GenerateKwargs
 
 
 # noinspection PyTypedDict

fabricatio/models/role.py

@@ -3,7 +3,6 @@
 from typing import Any, Self, Set
 
 from fabricatio.capabilities.correct import Correct
-from fabricatio.capabilities.covalidate import CoValidate
 from fabricatio.capabilities.task import HandleTask, ProposeTask
 from fabricatio.core import env
 from fabricatio.journal import logger
@@ -13,7 +12,7 @@ from fabricatio.models.tool import ToolBox
 from pydantic import Field
 
 
-class Role(ProposeTask, HandleTask, Correct, CoValidate):
+class Role(ProposeTask, HandleTask, Correct):
     """Class that represents a role with a registry of events and workflows.
 
     A Role serves as a container for workflows, managing their registration to events

fabricatio/models/usages.py

@@ -12,9 +12,9 @@ from fabricatio.models.generic import ScopedConfig, WithBriefing
 from fabricatio.models.kwargs_types import ChooseKwargs, EmbeddingKwargs, GenerateKwargs, LLMKwargs, ValidateKwargs
 from fabricatio.models.task import Task
 from fabricatio.models.tool import Tool, ToolBox
-from fabricatio.models.utils import Messages
+from fabricatio.models.utils import Messages, ok
 from fabricatio.parser import GenericCapture, JsonCapture
-from litellm import Router, stream_chunk_builder
+from litellm import Router, stream_chunk_builder  # pyright: ignore [reportPrivateImportUsage]
 from litellm.types.router import Deployment, LiteLLM_Params, ModelInfo
 from litellm.types.utils import (
     Choices,
@@ -70,14 +70,22 @@ class LLMUsage(ScopedConfig):
         """
         # Call the underlying asynchronous completion function with the provided and default parameters
         # noinspection PyTypeChecker,PydanticTypeChecker
-
         return await self._deploy(
             Deployment(
-                model_name=(m_name := kwargs.get("model") or self.llm_model or configs.llm.model),
+                model_name=(
+                    m_name := ok(
+                        kwargs.get("model") or self.llm_model or configs.llm.model, "model name is not set at any place"
+                    )
+                ),  # pyright: ignore [reportCallIssue]
                 litellm_params=(
                     p := LiteLLM_Params(
-                        api_key=(self.llm_api_key or configs.llm.api_key).get_secret_value(),
-                        api_base=(self.llm_api_endpoint or configs.llm.api_endpoint).unicode_string(),
+                        api_key=ok(
+                            self.llm_api_key or configs.llm.api_key, "llm api key is not set at any place"
+                        ).get_secret_value(),
+                        api_base=ok(
+                            self.llm_api_endpoint or configs.llm.api_endpoint,
+                            "llm api endpoint is not set at any place",
+                        ).unicode_string(),
                         model=m_name,
                         tpm=self.llm_tpm or configs.llm.tpm,
                         rpm=self.llm_rpm or configs.llm.rpm,
@@ -88,14 +96,14 @@ class LLMUsage(ScopedConfig):
                 model_info=ModelInfo(id=hash(m_name + p.model_dump_json(exclude_none=True))),
             )
         ).acompletion(
-            messages=messages,
+            messages=messages,  # pyright: ignore [reportArgumentType]
             n=n or self.llm_generation_count or configs.llm.generation_count,
             model=m_name,
             temperature=kwargs.get("temperature") or self.llm_temperature or configs.llm.temperature,
             stop=kwargs.get("stop") or self.llm_stop_sign or configs.llm.stop_sign,
             top_p=kwargs.get("top_p") or self.llm_top_p or configs.llm.top_p,
             max_tokens=kwargs.get("max_tokens") or self.llm_max_tokens or configs.llm.max_tokens,
-            stream=kwargs.get("stream") or self.llm_stream or configs.llm.stream,
+            stream=ok(kwargs.get("stream") or self.llm_stream or configs.llm.stream, "stream is not set at any place"),
             cache={
                 "no-cache": kwargs.get("no_cache"),
                 "no-store": kwargs.get("no_store"),
@@ -196,15 +204,15 @@ class LLMUsage(ScopedConfig):
                         for q, sm in zip(q_seq, sm_seq, strict=True)
                     ]
                 )
-                return [r[0].message.content for r in res]
+                return [r[0].message.content for r in res]  # pyright: ignore [reportReturnType, reportAttributeAccessIssue]
             case (list(q_seq), str(sm)):
                 res = await gather(*[self.ainvoke(n=1, question=q, system_message=sm, **kwargs) for q in q_seq])
-                return [r[0].message.content for r in res]
+                return [r[0].message.content for r in res]  # pyright: ignore [reportReturnType, reportAttributeAccessIssue]
             case (str(q), list(sm_seq)):
                 res = await gather(*[self.ainvoke(n=1, question=q, system_message=sm, **kwargs) for sm in sm_seq])
-                return [r[0].message.content for r in res]
+                return [r[0].message.content for r in res]  # pyright: ignore [reportReturnType, reportAttributeAccessIssue]
             case (str(q), str(sm)):
-                return ((await self.ainvoke(n=1, question=q, system_message=sm, **kwargs))[0]).message.content
+                return ((await self.ainvoke(n=1, question=q, system_message=sm, **kwargs))[0]).message.content  # pyright: ignore [reportReturnType, reportAttributeAccessIssue]
             case _:
                 raise RuntimeError("Should not reach here.")
 
@@ -215,6 +223,7 @@ class LLMUsage(ScopedConfig):
         validator: Callable[[str], T | None],
         default: T = ...,
         max_validations: PositiveInt = 2,
+        co_extractor: Optional[GenerateKwargs] = None,
         **kwargs: Unpack[GenerateKwargs],
     ) -> T: ...
     @overload
@@ -224,6 +233,7 @@ class LLMUsage(ScopedConfig):
         validator: Callable[[str], T | None],
         default: T = ...,
         max_validations: PositiveInt = 2,
+        co_extractor: Optional[GenerateKwargs] = None,
         **kwargs: Unpack[GenerateKwargs],
     ) -> List[T]: ...
     @overload
@@ -233,6 +243,7 @@ class LLMUsage(ScopedConfig):
         validator: Callable[[str], T | None],
         default: None = None,
         max_validations: PositiveInt = 2,
+        co_extractor: Optional[GenerateKwargs] = None,
         **kwargs: Unpack[GenerateKwargs],
     ) -> Optional[T]: ...
 
@@ -243,6 +254,7 @@ class LLMUsage(ScopedConfig):
         validator: Callable[[str], T | None],
         default: None = None,
         max_validations: PositiveInt = 2,
+        co_extractor: Optional[GenerateKwargs] = None,
         **kwargs: Unpack[GenerateKwargs],
     ) -> List[Optional[T]]: ...
 
@@ -252,6 +264,7 @@ class LLMUsage(ScopedConfig):
         validator: Callable[[str], T | None],
         default: Optional[T] = None,
         max_validations: PositiveInt = 2,
+        co_extractor: Optional[GenerateKwargs] = None,
         **kwargs: Unpack[GenerateKwargs],
     ) -> Optional[T] | List[Optional[T]] | List[T] | T:
         """Asynchronously asks a question and validates the response using a given validator.
@@ -261,6 +274,7 @@ class LLMUsage(ScopedConfig):
             validator (Callable[[str], T | None]): A function to validate the response.
             default (T | None): Default value to return if validation fails. Defaults to None.
             max_validations (PositiveInt): Maximum number of validation attempts. Defaults to 2.
+            co_extractor (Optional[GenerateKwargs]): Keyword arguments for the co-extractor, if provided will enable co-extraction.
             **kwargs (Unpack[LLMKwargs]): Additional keyword arguments for the LLM usage.
 
         Returns:
@@ -274,6 +288,23 @@ class LLMUsage(ScopedConfig):
                     if (response := await self.aask(question=q, **kwargs)) and (validated := validator(response)):
                         logger.debug(f"Successfully validated the response at {lap}th attempt.")
                         return validated
+
+                    if co_extractor and (
+                        (
+                            co_response := await self.aask(
+                                question=(
+                                    TEMPLATE_MANAGER.render_template(
+                                        configs.templates.co_validation_template,
+                                        {"original_q": q, "original_a": response},
+                                    )
+                                ),
+                                **co_extractor,
+                            )
+                        )
+                        and (validated := validator(co_response))
+                    ):
+                        logger.debug(f"Successfully validated the co-response at {lap}th attempt.")
+                        return validated
                 except Exception as e:  # noqa: BLE001
                     logger.error(f"Error during validation: \n{e}")
                     break
@@ -291,7 +322,7 @@ class LLMUsage(ScopedConfig):
 
     async def aliststr(
         self, requirement: str, k: NonNegativeInt = 0, **kwargs: Unpack[ValidateKwargs[List[str]]]
-    ) -> List[str]:
+    ) -> Optional[List[str]]:
         """Asynchronously generates a list of strings based on a given requirement.
 
         Args:
@@ -311,7 +342,7 @@ class LLMUsage(ScopedConfig):
             **kwargs,
         )
 
-    async def apathstr(self, requirement: str, **kwargs: Unpack[ChooseKwargs[List[str]]]) -> List[str]:
+    async def apathstr(self, requirement: str, **kwargs: Unpack[ChooseKwargs[List[str]]]) -> Optional[List[str]]:
         """Asynchronously generates a list of strings based on a given requirement.
 
         Args:
@@ -339,7 +370,7 @@ class LLMUsage(ScopedConfig):
         Returns:
             str: The validated response as a single string.
         """
-        return (
+        return ok(
             await self.apathstr(
                 requirement,
                 k=1,
@@ -347,7 +378,7 @@ class LLMUsage(ScopedConfig):
             )
         ).pop()
 
-    async def ageneric_string(self, requirement: str, **kwargs: Unpack[ValidateKwargs[str]]) -> str:
+    async def ageneric_string(self, requirement: str, **kwargs: Unpack[ValidateKwargs[str]]) -> Optional[str]:
         """Asynchronously generates a generic string based on a given requirement.
 
         Args:
@@ -357,7 +388,7 @@ class LLMUsage(ScopedConfig):
         Returns:
             str: The generated string.
         """
-        return await self.aask_validate(
+        return await self.aask_validate(  # pyright: ignore [reportReturnType]
             TEMPLATE_MANAGER.render_template(
                 configs.templates.generic_string_template,
                 {"requirement": requirement, "language": GenericCapture.capture_type},
@@ -372,7 +403,7 @@ class LLMUsage(ScopedConfig):
         choices: List[T],
         k: NonNegativeInt = 0,
         **kwargs: Unpack[ValidateKwargs[List[T]]],
-    ) -> List[T]:
+    ) -> Optional[List[T]]:
         """Asynchronously executes a multi-choice decision-making process, generating a prompt based on the instruction and options, and validates the returned selection results.
 
         Args:
@@ -437,13 +468,13 @@ class LLMUsage(ScopedConfig):
         Raises:
             ValueError: If validation fails after maximum attempts or if no valid selection is made.
         """
-        return (
+        return ok(
             await self.achoose(
                 instruction=instruction,
                 choices=choices,
                 k=1,
                 **kwargs,
-            )
+            ),
         )[0]
 
     async def ajudge(
@@ -500,7 +531,7 @@ class EmbeddingUsage(LLMUsage):
         """
         # check seq length
         max_len = self.embedding_max_sequence_length or configs.embedding.max_sequence_length
-        if any(len(t) > max_len for t in input_text):
+        if max_len and any(len(t) > max_len for t in input_text):
             logger.error(err := f"Input text exceeds maximum sequence length {max_len}.")
             raise ValueError(err)
 
@@ -514,10 +545,10 @@ class EmbeddingUsage(LLMUsage):
             or configs.embedding.timeout
             or self.llm_timeout
             or configs.llm.timeout,
-            api_key=(
+            api_key=ok(
                 self.embedding_api_key or configs.embedding.api_key or self.llm_api_key or configs.llm.api_key
             ).get_secret_value(),
-            api_base=(
+            api_base=ok(
                 self.embedding_api_endpoint
                 or configs.embedding.api_endpoint
                 or self.llm_api_endpoint
@@ -566,7 +597,7 @@ class ToolBoxUsage(LLMUsage):
         self,
         task: Task,
         **kwargs: Unpack[ChooseKwargs[List[ToolBox]]],
-    ) -> List[ToolBox]:
+    ) -> Optional[List[ToolBox]]:
         """Asynchronously executes a multi-choice decision-making process to choose toolboxes.
 
         Args:
@@ -591,7 +622,7 @@ class ToolBoxUsage(LLMUsage):
         task: Task,
         toolbox: ToolBox,
         **kwargs: Unpack[ChooseKwargs[List[Tool]]],
-    ) -> List[Tool]:
+    ) -> Optional[List[Tool]]:
         """Asynchronously executes a multi-choice decision-making process to choose tools.
 
         Args:
@@ -631,11 +662,11 @@ class ToolBoxUsage(LLMUsage):
         tool_choose_kwargs = tool_choose_kwargs or {}
 
         # Choose the toolboxes
-        chosen_toolboxes = await self.choose_toolboxes(task, **box_choose_kwargs)
+        chosen_toolboxes = ok(await self.choose_toolboxes(task, **box_choose_kwargs))
         # Choose the tools
         chosen_tools = []
         for toolbox in chosen_toolboxes:
-            chosen_tools.extend(await self.choose_tools(task, toolbox, **tool_choose_kwargs))
+            chosen_tools.extend(ok(await self.choose_tools(task, toolbox, **tool_choose_kwargs)))
         return chosen_tools
 
     async def gather_tools(self, task: Task, **kwargs: Unpack[ChooseKwargs]) -> List[Tool]:

fabricatio/models/utils.py

@@ -165,3 +165,24 @@ async def ask_edit(
         if edited:
             res.append(edited)
     return res
+
+
+def override_kwargs[T](kwargs: Dict[str, T], **overrides) -> Dict[str, T]:
+    """Override the values in kwargs with the provided overrides."""
+    kwargs.update({k: v for k, v in overrides.items() if v is not None})
+    return kwargs
+
+
+def ok[T](val: Optional[T], msg:str="Value is None") -> T:
+    """Check if a value is None and raise a ValueError with the provided message if it is.
+
+    Args:
+        val: The value to check.
+        msg: The message to include in the ValueError if val is None.
+
+    Returns:
+        T: The value if it is not None.
+    """
+    if val is None:
+        raise ValueError(msg)
+    return val

fabricatio/parser.py

@@ -52,6 +52,7 @@ class Capture(BaseModel):
             case _:
                 return text
 
+
     def capture(self, text: str) -> Tuple[str, ...] | str | None:
         """Capture the first occurrence of the pattern in the given text.
 

{fabricatio-0.2.6.dev5.dist-info → fabricatio-0.2.6.dev7.dist-info}/METADATA

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

{fabricatio-0.2.6.dev5.dist-info → fabricatio-0.2.6.dev7.dist-info}/RECORD

@@ -1,34 +1,33 @@
-fabricatio-0.2.6.dev5.dist-info/METADATA,sha256=W68vsoplegz0qZ6jTc47JXuhgTVJBXGOKJtxdMh7p_Y,14085
-fabricatio-0.2.6.dev5.dist-info/WHEEL,sha256=SmPT9fUKPAPiE6hwAZ9_NHUVRjWSQ_RENTrzrvPx4p0,94
-fabricatio-0.2.6.dev5.dist-info/licenses/LICENSE,sha256=do7J7EiCGbq0QPbMAL_FqLYufXpHnCnXBOuqVPwSV8Y,1088
-fabricatio/actions/article.py,sha256=d3i4i7h88yG8gvEW2SLDKM-FSsUv7qqiLgWpe6AJfSk,4463
+fabricatio-0.2.6.dev7.dist-info/METADATA,sha256=KJKAdxeQyQdtZjg8fyqf5BCnAZA4EJkAAMaNsDgXYCQ,14085
+fabricatio-0.2.6.dev7.dist-info/WHEEL,sha256=mDFV3bKFgwlxLHvOsPqpR9up9dUKYzsUQNKBdkW5c08,94
+fabricatio-0.2.6.dev7.dist-info/licenses/LICENSE,sha256=do7J7EiCGbq0QPbMAL_FqLYufXpHnCnXBOuqVPwSV8Y,1088
+fabricatio/actions/article.py,sha256=LfIWnbFYB9e3Bq2YDPk1geWDbJTq7zCitLtpFhAhYHM,4563
 fabricatio/actions/output.py,sha256=KSSLvEvXsA10ACN2mbqGo98QwKLVUAoMUJNKYk6HhGc,645
 fabricatio/actions/rag.py,sha256=GpT7YlqOYznZyaT-6Y84_33HtZGT-5s71ZK8iroQA9g,813
 fabricatio/capabilities/correct.py,sha256=0BYhjo9WrLwKsXQR8bTPvdQITbrMs7RX1xpzhuQt_yY,5222
-fabricatio/capabilities/covalidate.py,sha256=zl0b0Z8ZC3XkpzISIZJY4CZZAdVsx4qd1rdTLrFHFz8,6621
 fabricatio/capabilities/propose.py,sha256=y3kge5g6bb8HYuV8e9h4MdqOMTlsfAIZpqE_cagWPTY,1593
-fabricatio/capabilities/rag.py,sha256=OebdGps8LGniN_HkRAOuwZd1ZQsyQe3WrduNAmBSxLM,15773
+fabricatio/capabilities/rag.py,sha256=R1yUD675CDEmGakXb2nzEzZe0vjN7edMS7VHtPOAriU,15771
 fabricatio/capabilities/rating.py,sha256=R9otyZVE2E3kKxrOCTZMeesBCPbC-fSb7bXgZPMQzfU,14406
 fabricatio/capabilities/review.py,sha256=XYzpSnFCT9HS2XytQT8HDgV4SjXehexoJgucZFMx6P8,11102
 fabricatio/capabilities/task.py,sha256=MBiDyC3oHwTbTiLiGyqUEVfVGSN42lU03ndeapTpyjQ,4609
-fabricatio/config.py,sha256=orbMYzGZqdQy5cRlw0M6kOYmrRDq7uo8yEEZIjBfR18,16594
+fabricatio/config.py,sha256=f3B_Mwhc4mGEdECG4EqcxGww0Eu7KhCAwPXXJlHf1a8,16635
 fabricatio/core.py,sha256=VQ_JKgUGIy2gZ8xsTBZCdr_IP7wC5aPg0_bsOmjQ588,6458
 fabricatio/decorators.py,sha256=uzsP4tFKQNjDHBkofsjjoJA0IUAaYOtt6YVedoyOqlo,6551
 fabricatio/fs/curd.py,sha256=N6l2MncjrFfnXBRtteRouXp5Rjy8EAKC_i29_G-zz98,4618
 fabricatio/fs/readers.py,sha256=EZKN_AZdrp8DggJECP53QHw3uHeSDf-AwCAA_V7fNKU,1202
 fabricatio/fs/__init__.py,sha256=PCf0s_9KDjVfNw7AfPoJzGt3jMq4gJOfbcT4pb0D0ZY,588
 fabricatio/journal.py,sha256=stnEP88aUBA_GmU9gfTF2EZI8FS2OyMLGaMSTgK4QgA,476
-fabricatio/models/action.py,sha256=-WutQKRKBSLU6TNscT-puS6TnQdLxSmatof4JxltXWU,8281
+fabricatio/models/action.py,sha256=dSmwIrW68JhCrkhWENRgTLIQ-0grVA4408QAUy23HZo,8210
 fabricatio/models/events.py,sha256=QvlnS8FEELg6KNabcJMeh2GV_y0ZBzKOPphcteKYWYU,4183
-fabricatio/models/extra.py,sha256=s2y_zFH9kdCFHjrJadrHTODh_wFNcr6d_RDCLixgeyw,7390
+fabricatio/models/extra.py,sha256=oPCrh80u-O5XoFMVvZ6D6SVpSSW0zkxw4zfaTeK_wLU,26263
 fabricatio/models/generic.py,sha256=IdPJMf3qxZFq8yqd6OuAYKfCM0wBlJkozgxvxQZVEEc,14025
-fabricatio/models/kwargs_types.py,sha256=Dfmd18SABDeV9JsI1JfPNpoB8FtB6qVYgJshZBsN1P0,4593
-fabricatio/models/role.py,sha256=GVe8Rzjxzn9Fiava82XWJJQKjbxl6Fxi3U1pebiWeYU,2853
+fabricatio/models/kwargs_types.py,sha256=H6DI3Jdben-FER_kx7owiRzmbSFKuu0sFjCADA1LJB0,5008
+fabricatio/models/role.py,sha256=mmQbJ6GKr2Gx3wtjEz8d-vYoXs09ffcEkT_eCXaDd3E,2782
 fabricatio/models/task.py,sha256=8NaR7ojQWyM740EDTqt9stwHKdrD6axCRpLKo0QzS-I,10492
 fabricatio/models/tool.py,sha256=4b-v4WIC_LuLOKzzXL9bvKXr8vmGZ8O2uAFv5-1KRA0,7052
-fabricatio/models/usages.py,sha256=Rp4AGYFgpm5L2McNSaztWMZplmE_PnMmRc9NdcD5Jmw,28702
-fabricatio/models/utils.py,sha256=1bCqeB6za7ecCAM3cU1raNWuN56732m45rXtlIlc3I4,5017
-fabricatio/parser.py,sha256=b1Em7zoEepQIrxgM51Damnbsx_AhPab-BefIWHGo1Ss,6249
+fabricatio/models/usages.py,sha256=-689ssQ5F1SmxDToDHbv0EH8YaPTjhkn14l_M6Aer-M,30859
+fabricatio/models/utils.py,sha256=3HW0tM6WwOK8g14tnIzVWTXzIRLHjMKPjjSl9pMRWkw,5668
+fabricatio/parser.py,sha256=9Jzw-yV6uKbFvf6sPna-XHdziVGVBZWvPctgX_6ODL8,6251
 fabricatio/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 fabricatio/toolboxes/arithmetic.py,sha256=WLqhY-Pikv11Y_0SGajwZx3WhsLNpHKf9drzAqOf_nY,1369
 fabricatio/toolboxes/fs.py,sha256=l4L1CVxJmjw9Ld2XUpIlWfV0_Fu_2Og6d3E13I-S4aE,736
@@ -38,6 +37,6 @@ fabricatio/workflows/rag.py,sha256=-YYp2tlE9Vtfgpg6ROpu6QVO8j8yVSPa6yDzlN3qVxs,5
 fabricatio/_rust.pyi,sha256=eawBfpyGrB-JtOh4I6RSbjFSq83SSl-0syBeZ-g8270,3491
 fabricatio/_rust_instances.py,sha256=2GwF8aVfYNemRI2feBzH1CZfBGno-XJJE5imJokGEYw,314
 fabricatio/__init__.py,sha256=SzBYsRhZeL77jLtfJEjmoHOSwHwUGyvMATX6xfndLDM,1135
-fabricatio/_rust.cp39-win_amd64.pyd,sha256=rDkMzkreCAsU9bmxconUNqFNOaSLgSr7m--MqnSbXUU,1826304
-fabricatio-0.2.6.dev5.data/scripts/tdown.exe,sha256=ZrN_zXI0oDCkHx_r7-QrdqncIH0mFqq19mN9sE81kxs,3397632
-fabricatio-0.2.6.dev5.dist-info/RECORD,,
+fabricatio/_rust.cp39-win_amd64.pyd,sha256=GvYOGn9Xya6YMX-nhmqv-w908ndgc2HSinAYMkhypKo,1826304
+fabricatio-0.2.6.dev7.data/scripts/tdown.exe,sha256=5mZx7mp19U-nnWHwoZTyRmJun2iR77nar1wab1j_Jj8,3397632
+fabricatio-0.2.6.dev7.dist-info/RECORD,,

{fabricatio-0.2.6.dev5.dist-info → fabricatio-0.2.6.dev7.dist-info}/WHEEL

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

fabricatio/capabilities/covalidate.py

@@ -1,160 +0,0 @@
-"""Co-validation capability for LLMs."""
-
-from asyncio import gather
-from typing import Callable, List, Optional, Union, Unpack, overload
-
-from fabricatio import TEMPLATE_MANAGER
-from fabricatio.config import configs
-from fabricatio.journal import logger
-from fabricatio.models.kwargs_types import GenerateKwargs
-from fabricatio.models.usages import LLMUsage
-
-
-class CoValidate(LLMUsage):
-    """Class that represents a co-validation capability using multiple LLMs.
-
-    This class provides methods to validate responses by attempting multiple approaches:
-    1. Using the primary LLM to generate a response
-    2. Using a secondary (co-) model to refine responses that fail validation
-    3. Trying multiple times if needed
-    """
-
-    @overload
-    async def aask_covalidate[T](
-        self,
-        question: str,
-        validator: Callable[[str], T | None],
-        co_model: Optional[str] = None,
-        co_temperature: Optional[float] = None,
-        co_top_p: Optional[float] = None,
-        co_max_tokens: Optional[int] = None,
-        max_validations: int = 2,
-        default: None = None,
-        **kwargs: Unpack[GenerateKwargs],
-    ) -> T | None: ...
-
-    @overload
-    async def aask_covalidate[T](
-        self,
-        question: str,
-        validator: Callable[[str], T | None],
-        co_model: Optional[str] = None,
-        co_temperature: Optional[float] = None,
-        co_top_p: Optional[float] = None,
-        co_max_tokens: Optional[int] = None,
-        max_validations: int = 2,
-        default: T = ...,
-        **kwargs: Unpack[GenerateKwargs],
-    ) -> T: ...
-
-    @overload
-    async def aask_covalidate[T](
-        self,
-        question: List[str],
-        validator: Callable[[str], T | None],
-        co_model: Optional[str] = None,
-        co_temperature: Optional[float] = None,
-        co_top_p: Optional[float] = None,
-        co_max_tokens: Optional[int] = None,
-        max_validations: int = 2,
-        default: None = None,
-        **kwargs: Unpack[GenerateKwargs],
-    ) -> List[T | None]: ...
-
-    @overload
-    async def aask_covalidate[T](
-        self,
-        question: List[str],
-        validator: Callable[[str], T | None],
-        co_model: Optional[str] = None,
-        co_temperature: Optional[float] = None,
-        co_top_p: Optional[float] = None,
-        co_max_tokens: Optional[int] = None,
-        max_validations: int = 2,
-        default: T = ...,
-        **kwargs: Unpack[GenerateKwargs],
-    ) -> List[T]: ...
-
-    async def aask_covalidate[T](
-        self,
-        question: Union[str, List[str]],
-        validator: Callable[[str], T | None],
-        co_model: Optional[str] = None,
-        co_temperature: Optional[float] = None,
-        co_top_p: Optional[float] = None,
-        co_max_tokens: Optional[int] = None,
-        max_validations: int = 2,
-        default: Optional[T] = None,
-        **kwargs: Unpack[GenerateKwargs],
-    ) -> Union[T | None, List[T | None]]:
-        """Ask the LLM with co-validation to obtain a validated response.
-
-        This method attempts to generate a response that passes validation using two approaches:
-        1. First, it asks the primary LLM using the original question
-        2. If validation fails, it uses a secondary (co-) model with a template to improve the response
-        3. The process repeats up to max_validations times
-
-        Args:
-            question: String question or list of questions to ask
-            validator: Function that validates responses, returns result or None if invalid
-            co_model: Optional model name for the co-validator
-            co_temperature: Optional temperature setting for the co-validator
-            co_top_p: Optional top_p setting for the co-validator
-            co_max_tokens: Optional maximum tokens for the co-validator response
-            max_validations: Maximum number of validation attempts
-            default: Default value to return if validation fails
-            **kwargs: Additional keyword arguments passed to aask method
-
-        Returns:
-            The validated result (T) or default if validation fails.
-            If input is a list of questions, returns a list of results.
-        """
-
-        async def validate_single_question(q: str) -> Optional[T]:
-            """Process a single question with validation attempts."""
-            validation_kwargs = kwargs.copy()
-
-            for lap in range(max_validations):
-                try:
-                    # First attempt: direct question to primary model
-                    response = await self.aask(question=q, **validation_kwargs)
-                    if response and (validated := validator(response)):
-                        logger.debug(f"Successfully validated the primary response at {lap}th attempt.")
-                        return validated
-
-                    # Second attempt: use co-model with validation template
-                    co_prompt = TEMPLATE_MANAGER.render_template(
-                        configs.templates.co_validation_template,
-                        {"original_q": q, "original_a": response},
-                    )
-                    co_response = await self.aask(
-                        question=co_prompt,
-                        model=co_model,
-                        temperature=co_temperature,
-                        top_p=co_top_p,
-                        max_tokens=co_max_tokens,
-                    )
-
-                    if co_response and (validated := validator(co_response)):
-                        logger.debug(f"Successfully validated the co-response at {lap}th attempt.")
-                        return validated
-
-                except Exception as e:  # noqa: BLE001
-                    logger.error(f"Error during validation: \n{e}")
-                    break
-
-                # Disable caching for subsequent attempts
-                if not validation_kwargs.get("no_cache"):
-                    validation_kwargs["no_cache"] = True
-                    logger.debug("Disabled cache for the next attempt")
-
-            if default is None:
-                logger.error(f"Failed to validate the response after {max_validations} attempts.")
-            return default
-
-        # Handle single question or list of questions
-        if isinstance(question, str):
-            return await validate_single_question(question)
-
-        # Process multiple questions in parallel
-        return await gather(*[validate_single_question(q) for q in question])