fabricatio 0.2.5.dev5__tar.gz → 0.2.6.dev0__tar.gz

{fabricatio-0.2.5.dev5 → fabricatio-0.2.6.dev0}/PKG-INFO

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

{fabricatio-0.2.5.dev5 → fabricatio-0.2.6.dev0}/examples/extract_and_review/extract_and_review.py

@@ -27,8 +27,10 @@ async def main() -> None:
 
     unchecked_ess = await task.override_dependencies(gather_files("bpdf_out", "md")).delegate("article")
 
+    if unchecked_ess is None:
+        logger.error("No essence found")
+        return
     ess = list(filter(lambda x: x is not None, unchecked_ess))
-
     logger.success(f"Essence Count:{len(ess)}, invalid count: {len(unchecked_ess) - len(ess)}")
 
     Path("output").mkdir(exist_ok=True)

{fabricatio-0.2.5.dev5 → fabricatio-0.2.6.dev0}/examples/minor/hello_fabricatio.py

@@ -13,7 +13,7 @@ class Hello(Action):
 
     output_key: str = "task_output"
 
-    async def _execute(self, task_input: Task[str], **_) -> Any:
+    async def _execute(self, **_) -> Any:
         ret = "Hello fabricatio!"
         logger.info("executing talk action")
         return ret

{fabricatio-0.2.5.dev5 → fabricatio-0.2.6.dev0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "fabricatio"
-version = "0.2.5-dev5"
+version = "0.2.6-dev0"
 description = "A LLM multi-agent framework."
 readme = "README.md"
 license = { file = "LICENSE" }
@@ -72,6 +72,7 @@ dev = [
     "pytest-mock>=3.14.0",
     "pytest-rerunfailures>=15.0",
     "pytest-xdist>=3.6.1",
+    "viztracer>=1.0.2",
 ]
 
 [project.optional-dependencies]
@@ -139,6 +140,10 @@ testpaths = [
 asyncio_mode = "auto"
 asyncio_default_fixture_loop_scope = "function"
 
+[[tool.uv.index]]
+url = "https://mirrors.bfsu.edu.cn/pypi/web/simple"
+default = true
+
 [[tool.uv.index]]
 url = "https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple"
 
@@ -148,7 +153,3 @@ url = "https://mirrors.huaweicloud.com/repository/pypi/simple"
 [[tool.uv.index]]
 url = "https://mirrors.aliyun.com/pypi/simple/"
 
-[[tool.uv.index]]
-url = "https://mirrors.bfsu.edu.cn/pypi/web/simple"
-default = true
-

{fabricatio-0.2.5.dev5 → fabricatio-0.2.6.dev0}/python/fabricatio/actions/article.py

@@ -2,7 +2,7 @@
 
 from os import PathLike
 from pathlib import Path
-from typing import Callable, List
+from typing import Callable, List, Optional
 
 from fabricatio.fs import safe_text_read
 from fabricatio.journal import logger
@@ -27,7 +27,7 @@ class ExtractArticleEssence(Action):
         task_input: Task,
         reader: Callable[[P], str] = lambda p: Path(p).read_text(encoding="utf-8"),
         **_,
-    ) -> List[ArticleEssence]:
+    ) -> Optional[List[ArticleEssence]]:
         if not task_input.dependencies:
             logger.info(err := "Task not approved, since no dependencies are provided.")
             raise RuntimeError(err)
@@ -51,7 +51,7 @@ class GenerateArticleProposal(Action):
         self,
         task_input: Task,
         **_,
-    ) -> ArticleProposal:
+    ) -> Optional[ArticleProposal]:
         input_path = await self.awhich_pathstr(
             f"{task_input.briefing}\nExtract the path of file, which contains the article briefing that I need to read."
         )
@@ -73,7 +73,7 @@ class GenerateOutline(Action):
         self,
         article_proposal: ArticleProposal,
         **_,
-    ) -> ArticleOutline:
+    ) -> Optional[ArticleOutline]:
         return await self.propose(
             ArticleOutline,
             article_proposal.display(),

{fabricatio-0.2.5.dev5 → fabricatio-0.2.6.dev0}/python/fabricatio/actions/output.py

@@ -1,7 +1,5 @@
 """Dump the finalized output to a file."""
 
-from typing import Unpack
-
 from fabricatio.models.action import Action
 from fabricatio.models.generic import FinalizedDumpAble
 from fabricatio.models.task import Task
@@ -12,7 +10,7 @@ class DumpFinalizedOutput(Action):
 
     output_key: str = "dump_path"
 
-    async def _execute(self, task_input: Task, to_dump: FinalizedDumpAble, **cxt: Unpack) -> str:
+    async def _execute(self, task_input: Task, to_dump: FinalizedDumpAble, **_) -> str:
         dump_path = await self.awhich_pathstr(
             f"{task_input.briefing}\n\nExtract a single path of the file, to which I will dump the data."
         )

{fabricatio-0.2.5.dev5 → fabricatio-0.2.6.dev0}/python/fabricatio/actions/rag.py

@@ -1,6 +1,6 @@
 """Inject data into the database."""
 
-from typing import List, Optional, Unpack
+from typing import List, Optional
 
 from fabricatio.capabilities.rag import RAG
 from fabricatio.models.action import Action
@@ -13,7 +13,7 @@ class InjectToDB(Action, RAG):
     output_key: str = "collection_name"
 
     async def _execute[T: PrepareVectorization](
-        self, to_inject: T | List[T], collection_name: Optional[str] = "my_collection", **cxt: Unpack
+        self, to_inject: T | List[T], collection_name: Optional[str] = "my_collection", **_
     ) -> Optional[str]:
         if not isinstance(to_inject, list):
             to_inject = [to_inject]

{fabricatio-0.2.5.dev5 → fabricatio-0.2.6.dev0}/python/fabricatio/capabilities/rag.py

@@ -8,7 +8,7 @@ from functools import lru_cache
 from operator import itemgetter
 from os import PathLike
 from pathlib import Path
-from typing import Any, Callable, Dict, List, Optional, Self, Union, Unpack, overload
+from typing import Any, Callable, Dict, List, Optional, Self, Union, Unpack, cast, overload
 
 from fabricatio._rust_instances import template_manager
 from fabricatio.config import configs
@@ -111,9 +111,9 @@ class RAG(EmbeddingUsage):
             create (bool): Whether to create the collection if it does not exist.
             **kwargs (Unpack[CollectionSimpleConfigKwargs]): Additional keyword arguments for collection configuration.
         """
-        if create and collection_name and not self._client.has_collection(collection_name):
+        if create and collection_name and self.client.has_collection(collection_name):
             kwargs["dimension"] = kwargs.get("dimension") or self.milvus_dimensions or configs.rag.milvus_dimensions
-            self._client.create_collection(collection_name, auto_id=True, **kwargs)
+            self.client.create_collection(collection_name, auto_id=True, **kwargs)
             logger.info(f"Creating collection {collection_name}")
 
         self.target_collection = collection_name
@@ -152,15 +152,17 @@ class RAG(EmbeddingUsage):
             Self: The current instance, allowing for method chaining.
         """
         if isinstance(data, MilvusData):
-            data = data.prepare_insertion()
-        if isinstance(data, list):
-            data = [d.prepare_insertion() if isinstance(d, MilvusData) else d for d in data]
+            prepared_data = data.prepare_insertion()
+        elif isinstance(data, list):
+            prepared_data = [d.prepare_insertion() if isinstance(d, MilvusData) else d for d in data]
+        else:
+            raise TypeError(f"Expected MilvusData or list of MilvusData, got {type(data)}")
         c_name = collection_name or self.safe_target_collection
-        self._client.insert(c_name, data)
+        self.client.insert(c_name, prepared_data)
 
         if flush:
             logger.debug(f"Flushing collection {c_name}")
-            self._client.flush(c_name)
+            self.client.flush(c_name)
         return self
 
     async def consume_file(
@@ -196,14 +198,14 @@ class RAG(EmbeddingUsage):
         self.add_document(await self.pack(text), collection_name or self.safe_target_collection, flush=True)
         return self
 
-    async def afetch_document(
+    async def afetch_document[V: (int, str, float, bytes)](
         self,
         vecs: List[List[float]],
         desired_fields: List[str] | str,
         collection_name: Optional[str] = None,
         similarity_threshold: float = 0.37,
         result_per_query: int = 10,
-    ) -> List[Dict[str, Any]] | List[Any]:
+    ) -> List[Dict[str, Any]] | List[V]:
         """Fetch data from the collection.
 
         Args:
@@ -217,7 +219,7 @@ class RAG(EmbeddingUsage):
             List[Dict[str, Any]] | List[Any]: The retrieved data.
         """
         # Step 1: Search for vectors
-        search_results = self._client.search(
+        search_results = self.client.search(
             collection_name or self.safe_target_collection,
             vecs,
             search_params={"radius": similarity_threshold},
@@ -237,7 +239,7 @@ class RAG(EmbeddingUsage):
 
         if isinstance(desired_fields, list):
             return resp
-        return [r.get(desired_fields) for r in resp]
+        return [r.get(desired_fields) for r in resp]  # extract the single field as list
 
     async def aretrieve(
         self,
@@ -257,12 +259,13 @@ class RAG(EmbeddingUsage):
         """
         if isinstance(query, str):
             query = [query]
-        return (
+        return cast(
+            List[str],
             await self.afetch_document(
                 vecs=(await self.vectorize(query)),
                 desired_fields="text",
                 **kwargs,
-            )
+            ),
         )[:final_limit]
 
     async def aask_retrieved(

{fabricatio-0.2.5.dev5 → fabricatio-0.2.6.dev0}/python/fabricatio/capabilities/task.py

@@ -7,13 +7,13 @@ import orjson
 from fabricatio._rust_instances import template_manager
 from fabricatio.capabilities.propose import Propose
 from fabricatio.config import configs
+from fabricatio.journal import logger
 from fabricatio.models.generic import WithBriefing
 from fabricatio.models.kwargs_types import ChooseKwargs, ValidateKwargs
 from fabricatio.models.task import Task
 from fabricatio.models.tool import Tool, ToolExecutor
 from fabricatio.models.usages import ToolBoxUsage
 from fabricatio.parser import JsonCapture, PythonCapture
-from loguru import logger
 
 
 class ProposeTask(WithBriefing, Propose):

{fabricatio-0.2.5.dev5 → fabricatio-0.2.6.dev0}/python/fabricatio/config.py

@@ -68,7 +68,7 @@ class LLMConfig(BaseModel):
     temperature: NonNegativeFloat = Field(default=1.0)
     """The temperature of the LLM model. Controls randomness in generation. Set to 1.0 as per request."""
 
-    stop_sign: str | List[str] = Field(default_factory=lambda :["\n\n\n", "User:"])
+    stop_sign: str | List[str] = Field(default_factory=lambda: ["\n\n\n", "User:"])
     """The stop sign of the LLM model. No default stop sign specified."""
 
     top_p: NonNegativeFloat = Field(default=0.35)
@@ -83,6 +83,12 @@ class LLMConfig(BaseModel):
     max_tokens: PositiveInt = Field(default=8192)
     """The maximum number of tokens to generate. Set to 8192 as per request."""
 
+    rpm: Optional[PositiveInt] = Field(default=100)
+    """The rate limit of the LLM model in requests per minute. None means not checked."""
+
+    tpm: Optional[PositiveInt] = Field(default=1000000)
+    """The rate limit of the LLM model in tokens per minute. None means not checked."""
+
 
 class EmbeddingConfig(BaseModel):
     """Embedding configuration class."""
@@ -285,6 +291,19 @@ class CacheConfig(BaseModel):
     """Whether to enable cache."""
 
 
+class RoutingConfig(BaseModel):
+    """Routing configuration class."""
+
+    model_config = ConfigDict(use_attribute_docstrings=True)
+
+    allowed_fails: Optional[int] = 1
+    """The number of allowed fails before the routing is considered failed."""
+    retry_after: int = 15
+    """The time in seconds to wait before retrying the routing after a fail."""
+    cooldown_time: Optional[int] = 120
+    """The time in seconds to wait before retrying the routing after a cooldown."""
+
+
 class Settings(BaseSettings):
     """Application settings class.
 
@@ -310,6 +329,9 @@ class Settings(BaseSettings):
     llm: LLMConfig = Field(default_factory=LLMConfig)
     """LLM Configuration"""
 
+    routing: RoutingConfig = Field(default_factory=RoutingConfig)
+    """Routing Configuration"""
+
     embedding: EmbeddingConfig = Field(default_factory=EmbeddingConfig)
     """Embedding Configuration"""
 

{fabricatio-0.2.5.dev5 → fabricatio-0.2.6.dev0}/python/fabricatio/models/action.py

@@ -3,7 +3,7 @@
 import traceback
 from abc import abstractmethod
 from asyncio import Queue, create_task
-from typing import Any, Dict, Self, Tuple, Type, Union, Unpack, final
+from typing import Any, Dict, Self, Tuple, Type, Union, final
 
 from fabricatio.capabilities.review import Review
 from fabricatio.capabilities.task import HandleTask, ProposeTask
@@ -37,7 +37,7 @@ class Action(HandleTask, ProposeTask, Review):
         self.description = self.description or self.__class__.__doc__ or ""
 
     @abstractmethod
-    async def _execute(self, **cxt: Unpack) -> Any:
+    async def _execute(self, **cxt) -> Any:
         """Execute the action with the provided arguments.
 
         Args:

{fabricatio-0.2.5.dev5 → fabricatio-0.2.6.dev0}/python/fabricatio/models/generic.py

@@ -2,7 +2,7 @@
 
 from abc import abstractmethod
 from pathlib import Path
-from typing import Any, Callable, Dict, Iterable, List, Optional, Self, Union, final
+from typing import Any, Callable, Dict, Iterable, List, Optional, Self, Union, final, overload
 
 import orjson
 from fabricatio._rust import blake3_hash
@@ -100,7 +100,15 @@ class CreateJsonObjPrompt(WithFormatedJsonSchema):
     """Class that provides a prompt for creating a JSON object."""
 
     @classmethod
-    def create_json_prompt(cls, requirement: str) -> str:
+    @overload
+    def create_json_prompt(cls, requirement: List[str]) -> List[str]: ...
+
+    @classmethod
+    @overload
+    def create_json_prompt(cls, requirement: str) -> str: ...
+
+    @classmethod
+    def create_json_prompt(cls, requirement: str | List[str]) -> str | List[str]:
         """Create the prompt for creating a JSON object with given requirement.
 
         Args:
@@ -109,10 +117,18 @@ class CreateJsonObjPrompt(WithFormatedJsonSchema):
         Returns:
             str: The prompt for creating a JSON object with given requirement.
         """
-        return template_manager.render_template(
-            configs.templates.create_json_obj_template,
-            {"requirement": requirement, "json_schema": cls.formated_json_schema()},
-        )
+        if isinstance(requirement, str):
+            return template_manager.render_template(
+                configs.templates.create_json_obj_template,
+                {"requirement": requirement, "json_schema": cls.formated_json_schema()},
+            )
+        return [
+            template_manager.render_template(
+                configs.templates.create_json_obj_template,
+                {"requirement": r, "json_schema": cls.formated_json_schema()},
+            )
+            for r in requirement
+        ]
 
 
 class InstantiateFromString(Base):
@@ -307,6 +323,12 @@ class ScopedConfig(Base):
     llm_max_tokens: Optional[PositiveInt] = None
     """The maximum number of tokens to generate."""
 
+    llm_tpm: Optional[PositiveInt] = None
+    """The tokens per minute of the LLM model."""
+
+    llm_rpm: Optional[PositiveInt] = None
+    """The requests per minute of the LLM model."""
+
     embedding_api_endpoint: Optional[HttpUrl] = None
     """The OpenAI API endpoint."""
 

{fabricatio-0.2.5.dev5 → fabricatio-0.2.6.dev0}/python/fabricatio/models/kwargs_types.py

@@ -12,7 +12,7 @@ class CollectionSimpleConfigKwargs(TypedDict, total=False):
     These arguments are typically used when configuring connections to vector databases.
     """
 
-    dimension: int
+    dimension: int|None
     timeout: float
 
 
@@ -23,7 +23,7 @@ class FetchKwargs(TypedDict, total=False):
     and result limiting parameters.
     """
 
-    collection_name: str
+    collection_name: str|None
     similarity_threshold: float
     result_per_query: int
 

{fabricatio-0.2.5.dev5 → fabricatio-0.2.6.dev0}/python/fabricatio/models/usages.py

@@ -14,7 +14,8 @@ from fabricatio.models.task import Task
 from fabricatio.models.tool import Tool, ToolBox
 from fabricatio.models.utils import Messages
 from fabricatio.parser import JsonCapture
-from litellm import stream_chunk_builder
+from litellm import Router, stream_chunk_builder
+from litellm.types.router import Deployment, LiteLLM_Params, ModelInfo
 from litellm.types.utils import (
     Choices,
     EmbeddingResponse,
@@ -22,28 +23,40 @@ from litellm.types.utils import (
     StreamingChoices,
     TextChoices,
 )
-from litellm.utils import CustomStreamWrapper
+from litellm.utils import CustomStreamWrapper  # pyright: ignore [reportPrivateImportUsage]
 from more_itertools import duplicates_everseen
 from pydantic import Field, NonNegativeInt, PositiveInt
 
 if configs.cache.enabled and configs.cache.type:
     litellm.enable_cache(type=configs.cache.type, **configs.cache.params)
     logger.success(f"{configs.cache.type.name} Cache enabled")
+ROUTER = Router(
+    routing_strategy="usage-based-routing-v2",
+    allowed_fails=configs.routing.allowed_fails,
+    retry_after=configs.routing.retry_after,
+    cooldown_time=configs.routing.cooldown_time,
+)
 
 
 class LLMUsage(ScopedConfig):
     """Class that manages LLM (Large Language Model) usage parameters and methods."""
 
+    def _deploy(self, deployment: Deployment) -> Router:
+        """Add a deployment to the router."""
+        self._added_deployment = ROUTER.upsert_deployment(deployment)
+        return ROUTER
+
     @classmethod
     def _scoped_model(cls) -> Type["LLMUsage"]:
         return LLMUsage
 
+    # noinspection PyTypeChecker,PydanticTypeChecker
     async def aquery(
         self,
         messages: List[Dict[str, str]],
         n: PositiveInt | None = None,
         **kwargs: Unpack[LLMKwargs],
-    ) -> ModelResponse:
+    ) -> ModelResponse | CustomStreamWrapper:
         """Asynchronously queries the language model to generate a response based on the provided messages and parameters.
 
         Args:
@@ -55,19 +68,33 @@ class LLMUsage(ScopedConfig):
             ModelResponse | CustomStreamWrapper: An object containing the generated response and other metadata from the model.
         """
         # Call the underlying asynchronous completion function with the provided and default parameters
-        return await litellm.acompletion(
+        # noinspection PyTypeChecker,PydanticTypeChecker
+
+        return await self._deploy(
+            Deployment(
+                model_name=(m_name := kwargs.get("model") or self.llm_model or configs.llm.model),
+                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(),
+                        model=m_name,
+                        tpm=self.llm_tpm or configs.llm.tpm,
+                        rpm=self.llm_rpm or configs.llm.rpm,
+                        max_retries=kwargs.get("max_retries") or self.llm_max_retries or configs.llm.max_retries,
+                        timeout=kwargs.get("timeout") or self.llm_timeout or configs.llm.timeout,
+                    )
+                ),
+                model_info=ModelInfo(id=hash(m_name + p.model_dump_json(exclude_none=True))),
+            )
+        ).acompletion(
             messages=messages,
             n=n or self.llm_generation_count or configs.llm.generation_count,
-            model=kwargs.get("model") or self.llm_model or configs.llm.model,
+            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,
-            timeout=kwargs.get("timeout") or self.llm_timeout or configs.llm.timeout,
-            max_retries=kwargs.get("max_retries") or self.llm_max_retries or configs.llm.max_retries,
-            api_key=(self.llm_api_key or configs.llm.api_key).get_secret_value(),
-            base_url=(self.llm_api_endpoint or configs.llm.api_endpoint).unicode_string(),
             cache={
                 "no-cache": kwargs.get("no_cache"),
                 "no-store": kwargs.get("no_store"),
@@ -192,31 +219,31 @@ class LLMUsage(ScopedConfig):
     @overload
     async def aask_validate[T](
         self,
-        question: str,
+        question: List[str],
         validator: Callable[[str], T | None],
-        default: None = None,
+        default: T,
         max_validations: PositiveInt = 2,
         **kwargs: Unpack[GenerateKwargs],
-    ) -> Optional[T]: ...
-
+    ) -> List[T]: ...
     @overload
     async def aask_validate[T](
         self,
-        question: List[str],
+        question: str,
         validator: Callable[[str], T | None],
         default: None = None,
         max_validations: PositiveInt = 2,
         **kwargs: Unpack[GenerateKwargs],
-    ) -> List[Optional[T]]: ...
+    ) -> Optional[T]: ...
+
     @overload
     async def aask_validate[T](
         self,
         question: List[str],
         validator: Callable[[str], T | None],
-        default: T,
+        default: None = None,
         max_validations: PositiveInt = 2,
         **kwargs: Unpack[GenerateKwargs],
-    ) -> List[T]: ...
+    ) -> List[Optional[T]]: ...
 
     async def aask_validate[T](
         self,

{fabricatio-0.2.5.dev5 → fabricatio-0.2.6.dev0}/uv.lock

@@ -1,5 +1,4 @@
 version = 1
-revision = 1
 requires-python = "==3.12.*"
 
 [[package]]
@@ -216,7 +215,7 @@ wheels = [
 
 [[package]]
 name = "fabricatio"
-version = "0.2.5.dev5"
+version = "0.2.6.dev0"
 source = { editable = "." }
 dependencies = [
     { name = "appdirs" },
@@ -258,6 +257,7 @@ dev = [
     { name = "pytest-xdist" },
     { name = "ruff" },
     { name = "uv" },
+    { name = "viztracer" },
 ]
 
 [package.metadata]
@@ -280,7 +280,6 @@ requires-dist = [
     { name = "regex", specifier = ">=2024.11.6" },
     { name = "rich", specifier = ">=13.9.4" },
 ]
-provides-extras = ["rag", "full"]
 
 [package.metadata.requires-dev]
 dev = [
@@ -296,6 +295,7 @@ dev = [
     { name = "pytest-xdist", specifier = ">=3.6.1" },
     { name = "ruff", specifier = ">=0.9.4" },
     { name = "uv", specifier = ">=0.5.29" },
+    { name = "viztracer", specifier = ">=1.0.2" },
 ]
 
 [[package]]
@@ -750,6 +750,15 @@ wheels = [
     { url = "https://mirrors.bfsu.edu.cn/pypi/web/packages/16/2e/86f24451c2d530c88daf997cb8d6ac622c1d40d19f5a031ed68a4b73a374/numpy-1.26.4-cp312-cp312-win_amd64.whl", hash = "sha256:08beddf13648eb95f8d867350f6a018a4be2e5ad54c8d8caed89ebca558b2818", size = 15517754 },
 ]
 
+[[package]]
+name = "objprint"
+version = "0.3.0"
+source = { registry = "https://mirrors.bfsu.edu.cn/pypi/web/simple" }
+sdist = { url = "https://mirrors.bfsu.edu.cn/pypi/web/packages/81/b8/c10e96120f1585824a1992655334b49da3924edfb364e84a26cc0ecdb89b/objprint-0.3.0.tar.gz", hash = "sha256:b5d83f9d62db5b95353bb42959106e1cd43010dcaa3eed1ad8d7d0b2df9b2d5a", size = 47481 }
+wheels = [
+    { url = "https://mirrors.bfsu.edu.cn/pypi/web/packages/ec/af/572825252f16f36eeecbc8e3b721913d2640d69b984fdb8907aa8b4b0975/objprint-0.3.0-py3-none-any.whl", hash = "sha256:489083bfc8baf0526f8fd6af74673799511532636f0ce4141133255ded773405", size = 41619 },
+]
+
 [[package]]
 name = "onnxruntime"
 version = "1.20.1"
@@ -1462,6 +1471,24 @@ wheels = [
     { url = "https://mirrors.bfsu.edu.cn/pypi/web/packages/94/5e/0132a00365066b058d096c008f8ef7dc36bb00a5676efae97dc919669c78/uv-0.5.29-py3-none-win_arm64.whl", hash = "sha256:e8a5e18487c33a0c29867da123ef0f035ee1ba880640fcbf8743ca80d7158ed0", size = 15693104 },
 ]
 
+[[package]]
+name = "viztracer"
+version = "1.0.2"
+source = { registry = "https://mirrors.bfsu.edu.cn/pypi/web/simple" }
+dependencies = [
+    { name = "objprint" },
+]
+sdist = { url = "https://mirrors.bfsu.edu.cn/pypi/web/packages/ab/20/c699e4c355fc63fa9a3516c000e6e0ea1fc12a8ac1fef3ccceeb4b4acc04/viztracer-1.0.2.tar.gz", hash = "sha256:83f3fbc9530808ac048942acaa684ffca3a850de0b098783f67de059bf460764", size = 14270242 }
+wheels = [
+    { url = "https://mirrors.bfsu.edu.cn/pypi/web/packages/76/fc/5d32541bc6b758fccf59642103e29b9f039dd34c53b1c213eecba86bfbe1/viztracer-1.0.2-cp312-cp312-macosx_11_0_arm64.whl", hash = "sha256:37f9bcc2bd42cde5c4b4fb1ee4c65b795a1c4ed2636876abb49cc6e67721d72e", size = 14437011 },
+    { url = "https://mirrors.bfsu.edu.cn/pypi/web/packages/68/ed/63d051ffb12db1cf659e5ca5a0373b9d471c7d5525f763e0f59e9689db6f/viztracer-1.0.2-cp312-cp312-macosx_11_0_x86_64.whl", hash = "sha256:01feb66558942c6ac21d07400c95a6254709d5e050bb2cd9ace6bee989da413f", size = 14436844 },
+    { url = "https://mirrors.bfsu.edu.cn/pypi/web/packages/c4/e3/7de2b047e27f4d98e04504654b63c24b878ede178e6631d5c81248336668/viztracer-1.0.2-cp312-cp312-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:5d7964ae404427c6feea6460a1bccd88fd9898b36bf256fd0a3b20c3ec556e92", size = 14556048 },
+    { url = "https://mirrors.bfsu.edu.cn/pypi/web/packages/7e/cb/37863d6efdc65e89ab89e92a559c6c2736a7a5fe60d094175d17d407c4bf/viztracer-1.0.2-cp312-cp312-manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:d2680b2bda7b8c42a1a32b391034d58e7ace81a4c601c727a333598bf70df794", size = 14552601 },
+    { url = "https://mirrors.bfsu.edu.cn/pypi/web/packages/f3/e3/25f5ade889c92440cf279bd66e32965523513642a0e5f8ed58eb087e17ce/viztracer-1.0.2-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:8d5c412c2f3bf6d05a0bc5f50e4bce3642a863a001f0d3b290c6bd7085f045ca", size = 14564496 },
+    { url = "https://mirrors.bfsu.edu.cn/pypi/web/packages/8d/b1/60ef00a773c631a068f3ef53a7d598408cce4bd60f19e8ef355a9012e77d/viztracer-1.0.2-cp312-cp312-win32.whl", hash = "sha256:f0ba07dff91cdb7a489376f35206594c90b211367abfcb109c7c59f5b6691c18", size = 14600798 },
+    { url = "https://mirrors.bfsu.edu.cn/pypi/web/packages/5d/45/a53c0fe058780c8262c09a3c8279974e31cc54bc9aaa7d026c4a2f0289d2/viztracer-1.0.2-cp312-cp312-win_amd64.whl", hash = "sha256:0c114f56839e78113f3721e013a6742aca74acfef987eae2ba7c670533950c82", size = 14603623 },
+]
+
 [[package]]
 name = "wcwidth"
 version = "0.2.13"