fabricatio 0.2.6.dev6__tar.gz → 0.2.6.dev7__tar.gz

{fabricatio-0.2.6.dev6 → fabricatio-0.2.6.dev7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fabricatio
-Version: 0.2.6.dev6
+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.dev7/examples/extract_and_inject/.gitignore

@@ -0,0 +1,2 @@
+bpdf_out
+output*

fabricatio-0.2.6.dev7/examples/extract_and_inject/extract_and_inject.py

@@ -0,0 +1,53 @@
+"""Example of proposing a task to a role."""
+
+import asyncio
+from pathlib import Path
+from typing import List
+
+from fabricatio import Action, Event, Role, Task, WorkFlow, logger
+from fabricatio.actions.article import ExtractArticleEssence
+from fabricatio.actions.rag import InjectToDB
+from fabricatio.fs.curd import dump_text, gather_files
+from fabricatio.models.extra import ArticleEssence
+from pydantic import HttpUrl
+
+
+class SaveToFS(Action):
+    """Save to file system."""
+
+    async def _execute(self, to_inject: List[ArticleEssence], output_dir: Path, **_) -> None:
+        output_dir.mkdir(parents=True, exist_ok=True)
+        for i, e in enumerate(to_inject):
+            dump_text(output_dir / f"{i}.json", e.display())
+
+
+async def main() -> None:
+    """Main function."""
+    role = Role(
+        name="Researcher",
+        description="Extract article essence",
+        llm_api_endpoint=HttpUrl("https://dashscope.aliyuncs.com/compatible-mode/v1"),
+        llm_model="deepseek-r1-distill-llama-70b",
+        registry={
+            Event.quick_instantiate("article"): WorkFlow(
+                name="extract",
+                steps=(ExtractArticleEssence(output_key="to_inject"), InjectToDB(output_key="task_output")),
+                extra_init_context={"collection_name": "article_essence", "output_dir": Path("output")},
+            )
+        },
+    )
+
+    task: Task[str] = await role.propose_task(
+        "Extract the essence of the article from the files in './bpdf_out'",
+    )
+
+    col_name = await task.override_dependencies(gather_files("bpdf_out", "md")).delegate("article")
+
+    if col_name is None:
+        logger.error("No essence found")
+        return
+    logger.success(f"Injected to collection: {col_name}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

{fabricatio-0.2.6.dev6 → fabricatio-0.2.6.dev7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "fabricatio"
-version = "0.2.6-dev6"
+version = "0.2.6-dev7"
 description = "A LLM multi-agent framework."
 readme = "README.md"
 license = { file = "LICENSE" }

{fabricatio-0.2.6.dev6 → fabricatio-0.2.6.dev7}/python/fabricatio/capabilities/rag.py

@@ -15,7 +15,7 @@ from fabricatio.config import configs
 from fabricatio.journal import logger
 from fabricatio.models.kwargs_types import (
     ChooseKwargs,
-    CollectionSimpleConfigKwargs,
+    CollectionConfigKwargs,
     EmbeddingKwargs,
     FetchKwargs,
     LLMKwargs,
@@ -102,14 +102,14 @@ class RAG(EmbeddingUsage):
         ]
 
     def view(
-        self, collection_name: Optional[str], create: bool = False, **kwargs: Unpack[CollectionSimpleConfigKwargs]
+        self, collection_name: Optional[str], create: bool = False, **kwargs: Unpack[CollectionConfigKwargs]
     ) -> Self:
         """View the specified collection.
 
         Args:
             collection_name (str): The name of the collection.
             create (bool): Whether to create the collection if it does not exist.
-            **kwargs (Unpack[CollectionSimpleConfigKwargs]): Additional keyword arguments for collection configuration.
+            **kwargs (Unpack[CollectionConfigKwargs]): Additional keyword arguments for collection configuration.
         """
         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