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

fabricatio/utils.py

@@ -1,13 +1,39 @@
 """A collection of utility functions for the fabricatio package."""
 
-from typing import Any, Dict, List, Mapping, Optional, TypedDict, Unpack, overload
-
-import aiohttp
-import requests
+from typing import Any, Dict, Iterable, List, Mapping, Optional, Tuple, Type, overload
 
 from fabricatio.decorators import precheck_package
-from fabricatio.journal import logger
-from fabricatio.models.kwargs_types import RerankOptions
+
+
+def is_subclass_of_base(cls: Type, base_module: str, base_name: str) -> bool:
+    """Determines if the given class is a subclass of an unimported base class.
+
+    Args:
+        cls: The class to check
+        base_module: The module name of the base class
+        base_name: The class name of the base class
+
+    Returns:
+        bool: True if cls is a subclass of the specified base class, False otherwise
+    """
+    return any(ancestor.__module__ == base_module and ancestor.__name__ == base_name for ancestor in cls.__mro__)
+
+
+def is_subclass_of_any_base(cls: Type, bases: List[Tuple[str, str]]) -> bool:
+    """Determines if the given class is a subclass of the candidate base classes.
+
+    Args:
+        cls: The class to check
+        bases: A list of tuples where each tuple contains (module_name, class_name)
+
+    Returns:
+        bool: True if cls is a subclass of the specified base classes, False otherwise
+    """
+    for ancestor in cls.__mro__:
+        for base_module, base_name in bases:
+            if ancestor.__module__ == base_module and ancestor.__name__ == base_name:
+                return True
+    return False
 
 
 @precheck_package(
@@ -85,6 +111,37 @@ def ok[T](val: Optional[T], msg: str = "Value is None") -> T:
     return val
 
 
+def first_available[T](iterable: Iterable[T], msg: str = "No available item found in the iterable.") -> T:
+    """Return the first available item in the iterable that's not None.
+
+    This function searches through the provided iterable and returns the first
+    item that is not None. If all items are None or the iterable is empty,
+    it raises a ValueError.
+
+    Args:
+        iterable: The iterable collection to search through.
+        msg: The message to include in the ValueError if no non-None item is found.
+
+    Returns:
+        T: The first non-None item found in the iterable.
+        If no non-None item is found, it raises a ValueError.
+
+    Raises:
+        ValueError: If no non-None item is found in the iterable.
+
+    Examples:
+        >>> first_available([None, None, "value", "another"])
+        'value'
+        >>> first_available([1, 2, 3])
+        1
+        >>> assert (first_available([None, None]))
+        ValueError: No available item found in the iterable.
+    """
+    if (first := next((item for item in iterable if item is not None), None)) is not None:
+        return first
+    raise ValueError(msg)
+
+
 def wrapp_in_block(string: str, title: str, style: str = "-") -> str:
     """Wraps a string in a block with a title.
 
@@ -97,159 +154,3 @@ def wrapp_in_block(string: str, title: str, style: str = "-") -> str:
         str: The wrapped string.
     """
     return f"--- Start of {title} ---\n{string}\n--- End of {title} ---".replace("-", style)
-
-
-class RerankResult(TypedDict):
-    """The rerank result."""
-
-    index: int
-    score: float
-
-
-class RerankerAPI:
-    """A class to interact with the /rerank API for text reranking."""
-
-    def __init__(self, base_url: str) -> None:
-        """Initialize the RerankerAPI instance.
-
-        Args:
-            base_url (str): The base URL of the TEI-deployed reranker model API.
-                Example: "http://localhost:8000".
-        """
-        self.base_url = base_url.rstrip("/")  # Ensure no trailing slashes
-
-    @staticmethod
-    def _map_error_code(status_code: int, error_data: Dict[str, str]) -> Exception:
-        """Map HTTP status codes and error data to specific exceptions.
-
-        Args:
-            status_code (int): The HTTP status code returned by the API.
-            error_data (Dict[str, str]): The error details returned by the API.
-
-        Returns:
-            Exception: A specific exception based on the error code and message.
-        """
-        error_message = error_data.get("error", "Unknown error")
-
-        if status_code == 400:
-            return ValueError(f"Bad request: {error_message}")
-        if status_code == 413:
-            return ValueError(f"Batch size error: {error_message}")
-        if status_code == 422:
-            return RuntimeError(f"Tokenization error: {error_message}")
-        if status_code == 424:
-            return RuntimeError(f"Rerank error: {error_message}")
-        if status_code == 429:
-            return RuntimeError(f"Model overloaded: {error_message}")
-        return RuntimeError(f"Unexpected error ({status_code}): {error_message}")
-
-    def rerank(self, query: str, texts: List[str], **kwargs: Unpack[RerankOptions]) -> List[RerankResult]:
-        """Call the /rerank API to rerank a list of texts based on a query (synchronous).
-
-        Args:
-            query (str): The query string used for matching with the texts.
-            texts (List[str]): A list of texts to be reranked.
-            **kwargs (Unpack[RerankOptions]): Optional keyword arguments:
-                - raw_scores (bool, optional): Whether to return raw scores. Defaults to False.
-                - truncate (bool, optional): Whether to truncate the texts. Defaults to False.
-                - truncation_direction (Literal["left", "right"], optional): Direction of truncation. Defaults to "right".
-
-        Returns:
-            List[RerankResult]: A list of dictionaries containing the reranked results.
-                Each dictionary includes:
-                - "index" (int): The original index of the text.
-                - "score" (float): The relevance score.
-
-        Raises:
-            ValueError: If input parameters are invalid or the API returns a client-side error.
-            RuntimeError: If the API call fails or returns a server-side error.
-        """
-        # Validate inputs
-        if not isinstance(query, str) or not query.strip():
-            raise ValueError("Query must be a non-empty string.")
-        if not isinstance(texts, list) or not all(isinstance(text, str) for text in texts):
-            raise ValueError("Texts must be a list of strings.")
-
-        # Construct the request payload
-        payload = {
-            "query": query,
-            "texts": texts,
-            **kwargs,
-        }
-
-        try:
-            # Send POST request to the API
-            response = requests.post(f"{self.base_url}/rerank", json=payload)
-
-            # Handle non-200 status codes
-            if not response.ok:
-                error_data = None
-                if "application/json" in response.headers.get("Content-Type", ""):
-                    error_data = response.json()
-                else:
-                    error_data = {"error": response.text, "error_type": "unexpected_mimetype"}
-                raise self._map_error_code(response.status_code, error_data)
-
-            # Parse the JSON response
-            data: List[RerankResult] = response.json()
-            logger.debug(f"Rerank for `{query}` get {[s['score'] for s in data]}")
-            return data
-
-        except requests.exceptions.RequestException as e:
-            raise RuntimeError(f"Failed to connect to the API: {e}") from e
-
-    async def arerank(self, query: str, texts: List[str], **kwargs: Unpack[RerankOptions]) -> List[RerankResult]:
-        """Call the /rerank API to rerank a list of texts based on a query (asynchronous).
-
-        Args:
-            query (str): The query string used for matching with the texts.
-            texts (List[str]): A list of texts to be reranked.
-            **kwargs (Unpack[RerankOptions]): Optional keyword arguments:
-                - raw_scores (bool, optional): Whether to return raw scores. Defaults to False.
-                - truncate (bool, optional): Whether to truncate the texts. Defaults to False.
-                - truncation_direction (Literal["left", "right"], optional): Direction of truncation. Defaults to "right".
-
-        Returns:
-            List[RerankResult]: A list of dictionaries containing the reranked results.
-                Each dictionary includes:
-                - "index" (int): The original index of the text.
-                - "score" (float): The relevance score.
-
-        Raises:
-            ValueError: If input parameters are invalid or the API returns a client-side error.
-            RuntimeError: If the API call fails or returns a server-side error.
-        """
-        # Validate inputs
-        if not isinstance(query, str) or not query.strip():
-            raise ValueError("Query must be a non-empty string.")
-        if not isinstance(texts, list) or not all(isinstance(text, str) for text in texts):
-            raise ValueError("Texts must be a list of strings.")
-
-        # Construct the request payload
-        payload = {
-            "query": query,
-            "texts": texts,
-            **kwargs,
-        }
-
-        try:
-            # Send POST request to the API using aiohttp
-            async with (
-                aiohttp.ClientSession() as session,
-                session.post(f"{self.base_url}/rerank", json=payload) as response,
-            ):
-                # Handle non-200 status codes
-                if not response.ok:
-                    if "application/json" in response.headers.get("Content-Type", ""):
-                        error_data = await response.json()
-                    else:
-                        error_data = {"error": await response.text(), "error_type": "unexpected_mimetype"}
-                    raise self._map_error_code(response.status, error_data)
-
-                # Parse the JSON response
-                data: List[RerankResult] = await response.json()
-                logger.debug(f"Rerank for `{query}` get {[s['score'] for s in data]}")
-                return data
-
-        except aiohttp.ClientError as e:
-            raise RuntimeError(f"Failed to connect to the API: {e}") from e

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fabricatio
-Version: 0.2.13.dev3
+Version: 0.3.14
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: 3.12
@@ -8,7 +8,6 @@ Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Framework :: AsyncIO
 Classifier: Framework :: Pydantic :: 2
 Classifier: Typing :: Typed
-Requires-Dist: appdirs>=1.4.4
 Requires-Dist: asyncio>=3.4.3
 Requires-Dist: asyncstdlib>=3.13.0
 Requires-Dist: json-repair>=0.39.1
@@ -16,21 +15,16 @@ Requires-Dist: litellm>=1.60.0
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: more-itertools>=10.6.0
 Requires-Dist: pydantic>=2.10.6
-Requires-Dist: pydantic-settings>=2.7.1
 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
@@ -38,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
@@ -88,7 +82,7 @@ make bdist
 
 ```python
 import asyncio
-from fabricatio import Action, Role, Task, logger, WorkFlow
+from fabricatio import Action, Role, Task, logger, WorkFlow, Event
 from typing import Any
 
 
@@ -103,14 +97,14 @@ class Hello(Action):
 
 
 async def main() -> None:
-    role = Role(
+    Role(
         name="talker",
         description="talker role",
-        registry={Task.pending_label: WorkFlow(name="talk", steps=(Hello,))}
+        registry={Event.quick_instantiate("talk"): WorkFlow(name="talk", steps=(Hello,))}
     )
 
-    task = Task(name="say hello", goals="say hello", description="say hello to the world")
-    result = await task.delegate()
+    task = Task(name="say hello", goals=["say hello"], description="say hello to the world")
+    result = await task.delegate("talk")
     logger.success(f"Result: {result}")
 
 
@@ -165,11 +159,11 @@ max_tokens = 8192
     ```bash
     make test
     ```
+
 ## TODO
 
 - Add an element based format strategy
 
-
 ## Contributing
 
 Contributions are welcome! Follow these steps:
@@ -191,4 +185,5 @@ Special thanks to the contributors and maintainers of:
 - [PyO3](https://github.com/PyO3/pyo3)
 - [Maturin](https://github.com/PyO3/maturin)
 - [Handlebars.rs](https://github.com/sunng87/handlebars-rust)
+- [LiteLLM](https://github.com/BerriAI/litellm)
 

fabricatio-0.3.14.dist-info/RECORD

@@ -0,0 +1,64 @@
+fabricatio-0.3.14.data/scripts/tdown.exe,sha256=tcwy8uOlTTSW7vbDwiRkEWIbWNGDCtt7x30sEec54vs,3450880
+fabricatio-0.3.14.data/scripts/ttm.exe,sha256=hR8wJEEp6UfrGVmrIOwrBhozKEpA-PE-ZcmfDAk8qjA,2559488
+fabricatio-0.3.14.dist-info/METADATA,sha256=mYmZNSNR5-e2R-LgjvRgAsQvnPtpFzJFuFLZiEEkP14,5160
+fabricatio-0.3.14.dist-info/WHEEL,sha256=YpU2aDuTyBIvwRZn8idqScP-vkQ8DUGkCILtYmMfnFY,96
+fabricatio-0.3.14.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=gRAkE2eSzSNQh0eCk9P1FGF8xLPvHEVfxOLNDxrCI3U,14470
+fabricatio/actions/article_rag.py,sha256=ohS1CRtYuv2rJNgoIsBl2yv-PuuoypA3y223rUUyDBg,17989
+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/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
+fabricatio/capabilities/check.py,sha256=6IC6F0IhYVpSf9pJ8r9lq40l_FF3qf-iJcTRWwpnkdg,8591
+fabricatio/capabilities/correct.py,sha256=-JR8ZUAtagmNXepVyY679MBUyFCZwtKPjv8dANJMZiE,10403
+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=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/decorators.py,sha256=FmUDSr6RFtRnIXZ2OWYmbxCgTM1lsHKuyw4jTFgJbDo,9094
+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/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=wg7EaxDW3ScOoYHPc-e9HXzllNgaJemNFmrAuF_mgzI,12009
+fabricatio/models/extra/article_base.py,sha256=b70frbOUmPywCcWGxQhOzvKcqhqx5bRB5aTGfROHlFE,18009
+fabricatio/models/extra/article_essence.py,sha256=z3Qz6xVsB9k-K-c4Y2CoKzxZrXaUd4oyt2Mb6hGDYdg,2793
+fabricatio/models/extra/article_main.py,sha256=kBLqKMeBPy2fxioOZa9oqFjGPPjHvXHWDF_Fpx16aLU,11307
+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/generic.py,sha256=NQW2hfZ-_OTuok_gvIYepq49VThrl9guFPfN-68LbPw,27870
+fabricatio/models/kwargs_types.py,sha256=Ik8-Oi_NmwfkvC9B8K4NsoZc_vSWV85xKCSthA1Xv_k,3403
+fabricatio/models/role.py,sha256=PrVwmGUi3xh2uyxEDw0fygXDhllFOB65dKOuQnepoUc,4253
+fabricatio/models/task.py,sha256=XZ1l1P-iS02ZF9P8cXv8gEfJKBa17PFPNJ1SbhyhT4Q,11033
+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.cp312-win_amd64.pyd,sha256=Xzt_eDhOHiObHe5KVXonpySrstNCFLeSHbcQRZopujU,7819776
+fabricatio/rust.pyi,sha256=P_Tb54nW3Pepw--fgs3Tj8GhC1ufUCJXhcnFNElVtn8,26168
+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/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-0.3.14.dist-info/RECORD,,

{fabricatio-0.2.13.dev3.dist-info → fabricatio-0.3.14.dist-info}/WHEEL

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