fabricatio 0.2.11.dev2__tar.gz → 0.2.11.dev3__tar.gz

{fabricatio-0.2.11.dev2 → fabricatio-0.2.11.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fabricatio
-Version: 0.2.11.dev2
+Version: 0.2.11.dev3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: 3.12
@@ -20,18 +20,20 @@ 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] ; extra == 'full'
+Requires-Dist: fabricatio[calc,ftd,plot,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
 License-File: LICENSE
 Summary: A LLM multi-agent framework.
 Keywords: ai,agents,multi-agent,llm,pyo3

fabricatio-0.2.11.dev3/examples/write_article/write_article.py

@@ -0,0 +1,140 @@
+"""Example of using the library."""
+
+import asyncio
+from pathlib import Path
+
+import typer
+from fabricatio import Event, Role, WorkFlow, logger
+from fabricatio.actions.article import ExtractOutlineFromRaw, GenerateArticleProposal, GenerateInitialOutline
+from fabricatio.actions.article_rag import WriteArticleContentRAG
+from fabricatio.actions.output import (
+    DumpFinalizedOutput,
+    PersistentAll,
+)
+from fabricatio.models.task import Task
+from fabricatio.utils import ok
+from typer import Typer
+
+Role(
+    name="Undergraduate Researcher",
+    description="Write an outline for an article in typst format.",
+    llm_model="openai/qwen-max",
+    llm_temperature=0.63,
+    llm_stream=True,
+    llm_top_p=0.85,
+    llm_max_tokens=8191,
+    llm_rpm=600,
+    llm_tpm=900000,
+    registry={
+        Event.quick_instantiate(ns := "article"): WorkFlow(
+            name="Generate Article",
+            description="Generate an article. dump the outline to the given path. in typst format.",
+            steps=(
+                GenerateArticleProposal,
+                GenerateInitialOutline(output_key="article_outline"),
+                PersistentAll,
+                WriteArticleContentRAG(
+                    output_key="to_dump",
+                    llm_top_p=0.9,
+                    ref_limit=35,
+                    llm_model="openai/qwq-plus",
+                    target_collection="article_chunks",
+                    extractor_model="openai/qwen-plus",
+                    query_model="openai/qwen-max",
+                ),
+                DumpFinalizedOutput(output_key="task_output"),
+                PersistentAll,
+            ),
+        ),
+        Event.quick_instantiate(ns2 := ""): WorkFlow(
+            name="Generate Article",
+            description="Generate an article with given raw article outline. dump the outline to the given path. in typst format.",
+            steps=(
+                ExtractOutlineFromRaw,
+                PersistentAll,
+                WriteArticleContentRAG(
+                    output_key="to_dump",
+                    llm_top_p=0.9,
+                    ref_limit=35,
+                    llm_model="openai/qwq-plus",
+                    target_collection="article_chunks",
+                    extractor_model="openai/qwen-plus",
+                    query_model="openai/qwen-max",
+                ),
+                DumpFinalizedOutput(output_key="task_output"),
+                PersistentAll,
+            ),
+        ),
+    },
+)
+
+
+app = Typer()
+
+
+@app.command()
+def completion(
+    article_outline_raw_path: Path = typer.Option(  # noqa: B008
+        Path("article_outline_raw.txt"), "-a", "--article-outline-raw", help="Path to the article outline raw file."
+    ),
+    dump_path: Path = typer.Option(Path("out.typ"), "-d", "--dump-path", help="Path to dump the final output."),  # noqa: B008
+    persist_dir: Path = typer.Option(  # noqa: B008
+        Path("persistent"), "-p", "--persist-dir", help="Directory to persist the output."
+    ),
+    collection_name: str = typer.Option("article_chunks", "-c", "--collection-name", help="Name of the collection."),
+    supervisor: bool = typer.Option(False, "-s", "--supervisor", help="Whether to use the supervisor mode."),
+) -> None:
+    """Write an article based on a raw article outline."""
+    path = ok(
+        asyncio.run(
+            Task(name="write an article")
+            .update_init_context(
+                article_outline_raw_path=article_outline_raw_path,
+                dump_path=dump_path,
+                persist_dir=persist_dir,
+                collection_name=collection_name,
+                supervisor=supervisor,
+            )
+            .delegate(ns)
+        ),
+        "Failed to generate an article ",
+    )
+    logger.success(f"The outline is saved in:\n{path}")
+
+
+@app.command()
+def write(
+    article_briefing: Path = typer.Option(  # noqa: B008
+        Path("article_briefing.txt"), "-a", "--article-briefing", help="Path to the article briefing file."
+    ),
+    dump_path: Path = typer.Option(Path("out.typ"), "-d", "--dump-path", help="Path to dump the final output."),  # noqa: B008
+    persist_dir: Path = typer.Option(  # noqa: B008
+        Path("persistent"), "-p", "--persist-dir", help="Directory to persist the output."
+    ),
+    collection_name: str = typer.Option("article_chunks", "-c", "--collection-name", help="Name of the collection."),
+    supervisor: bool = typer.Option(False, "-s", "--supervisor", help="Whether to use the supervisor mode."),
+) -> None:
+    """Write an article based on a briefing.
+
+    This function generates an article outline and content based on the provided briefing.
+    The outline and content are then dumped to the specified path and persisted in the given directory.
+    """
+    path = ok(
+        asyncio.run(
+            Task(name="write an article")
+            .update_init_context(
+                article_briefing=article_briefing.read_text(),
+                dump_path=dump_path,
+                persist_dir=persist_dir,
+                collection_name=collection_name,
+                supervisor=supervisor,
+            )
+            .delegate(ns)
+        ),
+        "Failed to generate an article ",
+    )
+    logger.success(f"The outline is saved in:\n{path}")
+
+
+if __name__ == "__main__":
+    app()

{fabricatio-0.2.11.dev2 → fabricatio-0.2.11.dev3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "fabricatio"
-version = "0.2.11-dev2"
+version = "0.2.11-dev3"
 description = "A LLM multi-agent framework."
 readme = "README.md"
 license = { file = "LICENSE" }
@@ -74,7 +74,7 @@ dev = [
 
 [project.optional-dependencies]
 full = [
-    "fabricatio[calc,ftd,plot,qa,rag]",
+    "fabricatio[calc,ftd,plot,qa,rag,cli]",
 ]
 rag = [
     "pymilvus>=2.5.4",
@@ -92,6 +92,9 @@ qa = [
 ftd = [
     "magika>=0.6.1",
 ]
+cli = [
+    "typer-slim[standard]>=0.15.2",
+]
 
 [tool.ruff]
 include = ["pyproject.toml", "python/fabricatio/*.py", "python/fabricatio/*.pyi", "examples/*.py"]

{fabricatio-0.2.11.dev2 → fabricatio-0.2.11.dev3}/python/fabricatio/actions/article.py

@@ -149,10 +149,10 @@ class GenerateInitialOutline(Action, Extract):
     async def _execute(
         self,
         article_proposal: ArticleProposal,
+        supervisor: Optional[bool] = None,
         **_,
     ) -> Optional[ArticleOutline]:
-        raw_outline = await self.aask(
-            f"{(article_proposal.as_prompt())}\n"
+        req = (
             f"Design each chapter of a proper and academic and ready for release manner.\n"
             f"You Must make sure every chapter have sections, and every section have subsections.\n"
             f"Make the chapter and sections and subsections bing divided into a specific enough article component.\n"
@@ -160,14 +160,16 @@ class GenerateInitialOutline(Action, Extract):
             f"Note that you SHALL use `{article_proposal.language}` as written language",
         )
 
-        if self.supervisor:
+        raw_outline = await self.aask(f"{(article_proposal.as_prompt())}\n{req}")
+
+        if supervisor or (supervisor is None and self.supervisor):
             from questionary import confirm, text
 
             r_print(raw_outline)
             while not await confirm("Accept this version and continue?", default=True).ask_async():
                 imp = await text("Enter the improvement:").ask_async()
                 raw_outline = await self.aask(
-                    f"{article_proposal.as_prompt()}\n{wrapp_in_block(raw_outline, 'Previous ArticleOutline')}\n{imp}"
+                    f"{article_proposal.as_prompt()}\n{wrapp_in_block(raw_outline, 'Previous ArticleOutline')}\n{req}\n{wrapp_in_block(imp, title='Improvement')}"
                 )
                 r_print(raw_outline)
 
@@ -177,6 +179,20 @@ class GenerateInitialOutline(Action, Extract):
         ).update_ref(article_proposal)
 
 
+class ExtractOutlineFromRaw(Action, Extract):
+    """Extract the outline from the raw outline."""
+
+    output_key: str = "article_outline_from_raw"
+
+    async def _execute(self, article_outline_raw_path: str | Path, **cxt) -> ArticleOutline:
+        logger.info(f"Extracting outline from raw: {Path(article_outline_raw_path).as_posix()}")
+
+        return ok(
+            await self.extract(ArticleOutline, safe_text_read(article_outline_raw_path)),
+            "Could not extract the outline from raw.",
+        )
+
+
 class FixIntrospectedErrors(Action, Censor):
     """Fix introspected errors in the article outline."""
 

{fabricatio-0.2.11.dev2 → fabricatio-0.2.11.dev3}/python/fabricatio/actions/article_rag.py

@@ -56,13 +56,13 @@ class WriteArticleContentRAG(Action, RAG, Extract):
     async def _execute(
         self,
         article_outline: ArticleOutline,
-        writing_ruleset: RuleSet,
         collection_name: str = "article_chunks",
+        supervisor: Optional[bool] = None,
         **cxt,
     ) -> Article:
         article = Article.from_outline(article_outline).update_ref(article_outline)
 
-        if self.supervisor:
+        if supervisor or (supervisor is None and self.supervisor):
             await gather(
                 *[
                     self._supervisor_inner(article, article_outline, chap, sec, subsec)
@@ -98,9 +98,10 @@ class WriteArticleContentRAG(Action, RAG, Extract):
 
         raw = await self.write_raw(article, article_outline, chap, sec, subsec, cm)
         r_print(raw)
+
         while not await confirm("Accept this version and continue?").ask_async():
             if await confirm("Search for more refs?").ask_async():
-                new_refs = await self.search_database(article, article_outline, chap, sec, subsec)
+                new_refs = await self.search_database(article, article_outline, chap, sec, subsec, supervisor=True)
                 cm.add_chunks(await ask_retain([r.chunk for r in new_refs], new_refs))
 
             instruction = await text("Enter the instructions to improve").ask_async()
@@ -185,6 +186,7 @@ class WriteArticleContentRAG(Action, RAG, Extract):
         sec: ArticleSection,
         subsec: ArticleSubsection,
         extra_instruction: str = "",
+        supervisor: bool = False,
     ) -> List[ArticleChunk]:
         """Search database for related references."""
         ref_q = ok(
@@ -199,7 +201,7 @@ class WriteArticleContentRAG(Action, RAG, Extract):
             "Failed to refine query.",
         )
 
-        if self.supervisor:
+        if supervisor:
             ref_q = await ask_retain(ref_q)
 
         return await self.aretrieve(

{fabricatio-0.2.11.dev2 → fabricatio-0.2.11.dev3}/python/fabricatio/models/action.py

@@ -18,6 +18,7 @@ from fabricatio.journal import logger
 from fabricatio.models.generic import WithBriefing
 from fabricatio.models.task import Task
 from fabricatio.models.usages import LLMUsage, ToolBoxUsage
+from fabricatio.utils import override_kwargs
 from pydantic import Field, PrivateAttr
 
 OUTPUT_KEY = "task_output"
@@ -55,7 +56,7 @@ class Action(WithBriefing, LLMUsage):
         self.description = self.description or self.__class__.__doc__ or ""
 
     @abstractmethod
-    async def _execute(self, *_:Any, **cxt) -> Any:
+    async def _execute(self, *_: Any, **cxt) -> Any:
         """Implement the core logic of the action.
 
         Args:
@@ -95,11 +96,12 @@ class Action(WithBriefing, LLMUsage):
             return f"## Your personality: \n{self.personality}\n# The action you are going to perform: \n{super().briefing}"
         return f"# The action you are going to perform: \n{super().briefing}"
 
-    def to_task_output(self)->Self:
+    def to_task_output(self) -> Self:
         """Set the output key to OUTPUT_KEY and return the action instance."""
-        self.output_key=OUTPUT_KEY
+        self.output_key = OUTPUT_KEY
         return self
 
+
 class WorkFlow(WithBriefing, ToolBoxUsage):
     """Manages sequences of actions to fulfill tasks.
 
@@ -177,7 +179,7 @@ class WorkFlow(WithBriefing, ToolBoxUsage):
         current_action = None
         try:
             # Process each action in sequence
-            for i,step in enumerate(self._instances):
+            for i, step in enumerate(self._instances):
                 current_action = step.name
                 logger.info(f"Executing step [{i}] >> {current_action}")
 
@@ -227,8 +229,13 @@ class WorkFlow(WithBriefing, ToolBoxUsage):
             - Any extra_init_context values
         """
         logger.debug(f"Initializing context for workflow: {self.name}")
-        initial_context = {self.task_input_key: task, **dict(self.extra_init_context)}
-        await self._context.put(initial_context)
+        ctx = override_kwargs(self.extra_init_context, **task.extra_init_context)
+        if self.task_input_key in ctx:
+            raise ValueError(
+                f"Task input key: `{self.task_input_key}`, which is reserved, is already set in the init context"
+            )
+
+        await self._context.put({self.task_input_key: task, **ctx})
 
     def steps_fallback_to_self(self) -> Self:
         """Configure all steps to use this workflow's configuration as fallback.
@@ -245,7 +252,7 @@ class WorkFlow(WithBriefing, ToolBoxUsage):
         Returns:
             Self: The workflow instance for method chaining.
         """
-        self.provide_tools_to(i for i in self._instances if isinstance(i,ToolBoxUsage))
+        self.provide_tools_to(i for i in self._instances if isinstance(i, ToolBoxUsage))
         return self
 
     def update_init_context(self, /, **kwargs) -> Self:

{fabricatio-0.2.11.dev2 → fabricatio-0.2.11.dev3}/python/fabricatio/models/task.py

@@ -4,7 +4,7 @@ It includes methods to manage the task's lifecycle, such as starting, finishing,
 """
 
 from asyncio import Queue
-from typing import Any, List, Optional, Self
+from typing import Any, Dict, List, Optional, Self
 
 from fabricatio.config import configs
 from fabricatio.constants import TaskStatus
@@ -50,6 +50,18 @@ class Task[T](WithBriefing, ProposedAble, WithDependency):
 
     _namespace: Event = PrivateAttr(default_factory=Event)
     """The namespace of the task as an event, which is generated from the namespace list."""
+    _extra_init_context: Dict = PrivateAttr(default_factory=dict)
+    """Extra initialization context for the task, which is designed to override the one of the Workflow."""
+
+    @property
+    def extra_init_context(self) -> Dict:
+        """Extra initialization context for the task, which is designed to override the one of the Workflow."""
+        return self._extra_init_context
+
+    def update_init_context(self, /, **kwargs) -> Self:
+        """Update the extra initialization context for the task."""
+        self.extra_init_context.update(kwargs)
+        return self
 
     def model_post_init(self, __context: Any) -> None:
         """Initialize the task with a namespace event."""

{fabricatio-0.2.11.dev2 → fabricatio-0.2.11.dev3}/python/fabricatio/models/usages.py

@@ -31,7 +31,7 @@ from pydantic import BaseModel, ConfigDict, 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")
+    logger.debug(f"{configs.cache.type.name} Cache enabled")
 
 ROUTER = Router(
     routing_strategy="usage-based-routing-v2",

{fabricatio-0.2.11.dev2 → fabricatio-0.2.11.dev3}/python/fabricatio/utils.py

@@ -54,14 +54,14 @@ async def ask_retain[V](candidates: List[str], value_mapping: Optional[List[V]]
 def override_kwargs(kwargs: Mapping[str, Any], **overrides) -> Dict[str, Any]:
     """Override the values in kwargs with the provided overrides."""
     new_kwargs = dict(kwargs.items())
-    new_kwargs.update({k: v for k, v in overrides.items() if v is not None})
+    new_kwargs.update(overrides)
     return new_kwargs
 
 
-def fallback_kwargs(kwargs: Mapping[str, Any], **overrides) -> Dict[str, Any]:
-    """Fallback the values in kwargs with the provided overrides."""
+def fallback_kwargs(kwargs: Mapping[str, Any], **fallbacks) -> Dict[str, Any]:
+    """Fallback the values in kwargs with the provided fallbacks."""
     new_kwargs = dict(kwargs.items())
-    new_kwargs.update({k: v for k, v in overrides.items() if k not in new_kwargs and v is not None})
+    new_kwargs.update({k: v for k, v in fallbacks.items() if k not in new_kwargs})
     return new_kwargs
 
 

{fabricatio-0.2.11.dev2 → fabricatio-0.2.11.dev3}/uv.lock

@@ -248,7 +248,7 @@ wheels = [
 
 [[package]]
 name = "fabricatio"
-version = "0.2.11.dev2"
+version = "0.2.11.dev3"
 source = { editable = "." }
 dependencies = [
     { name = "appdirs" },
@@ -269,6 +269,9 @@ dependencies = [
 calc = [
     { name = "sympy" },
 ]
+cli = [
+    { name = "typer-slim", extra = ["standard"] },
+]
 ftd = [
     { name = "magika" },
 ]
@@ -278,6 +281,7 @@ full = [
     { name = "pymilvus" },
     { name = "questionary" },
     { name = "sympy" },
+    { name = "typer-slim", extra = ["standard"] },
 ]
 plot = [
     { name = "matplotlib" },
@@ -311,7 +315,7 @@ requires-dist = [
     { name = "appdirs", specifier = ">=1.4.4" },
     { name = "asyncio", specifier = ">=3.4.3" },
     { name = "asyncstdlib", specifier = ">=3.13.0" },
-    { name = "fabricatio", extras = ["calc", "ftd", "plot", "qa", "rag"], marker = "extra == 'full'", editable = "." },
+    { name = "fabricatio", extras = ["calc", "ftd", "plot", "qa", "rag", "cli"], marker = "extra == 'full'", editable = "." },
     { name = "json-repair", specifier = ">=0.39.1" },
     { name = "litellm", specifier = ">=1.60.0" },
     { name = "loguru", specifier = ">=0.7.3" },
@@ -325,9 +329,10 @@ requires-dist = [
     { name = "questionary", marker = "extra == 'qa'", specifier = ">=2.1.0" },
     { name = "rich", specifier = ">=13.9.4" },
     { name = "sympy", marker = "extra == 'calc'", specifier = ">=1.13.3" },
+    { name = "typer-slim", extras = ["standard"], marker = "extra == 'cli'", specifier = ">=0.15.2" },
     { name = "ujson", specifier = ">=5.10.0" },
 ]
-provides-extras = ["full", "rag", "calc", "plot", "qa", "ftd"]
+provides-extras = ["full", "rag", "calc", "plot", "qa", "ftd", "cli"]
 
 [package.metadata.requires-dev]
 dev = [
@@ -1430,6 +1435,15 @@ wheels = [
     { url = "https://mirrors.bfsu.edu.cn/pypi/web/packages/54/21/f43f0a1fa8b06b32812e0975981f4677d28e0f3271601dc88ac5a5b83220/setuptools-78.1.0-py3-none-any.whl", hash = "sha256:3e386e96793c8702ae83d17b853fb93d3e09ef82ec62722e61da5cd22376dcd8", size = 1256108 },
 ]
 
+[[package]]
+name = "shellingham"
+version = "1.5.4"
+source = { registry = "https://mirrors.bfsu.edu.cn/pypi/web/simple" }
+sdist = { url = "https://mirrors.bfsu.edu.cn/pypi/web/packages/58/15/8b3609fd3830ef7b27b655beb4b4e9c62313a4e8da8c676e142cc210d58e/shellingham-1.5.4.tar.gz", hash = "sha256:8dbca0739d487e5bd35ab3ca4b36e11c4078f3a234bfce294b0a0291363404de" }
+wheels = [
+    { url = "https://mirrors.bfsu.edu.cn/pypi/web/packages/e0/f9/0595336914c5619e5f28a1fb793285925a8cd4b432c9da0a987836c7f822/shellingham-1.5.4-py2.py3-none-any.whl", hash = "sha256:7ecfff8f2fd72616f7481040475a65b2bf8af90a56c89140852d1120324e8686" },
+]
+
 [[package]]
 name = "six"
 version = "1.17.0"
@@ -1515,6 +1529,25 @@ wheels = [
     { url = "https://mirrors.bfsu.edu.cn/pypi/web/packages/d0/30/dc54f88dd4a2b5dc8a0279bdd7270e735851848b762aeb1c1184ed1f6b14/tqdm-4.67.1-py3-none-any.whl", hash = "sha256:26445eca388f82e72884e0d580d5464cd801a3ea01e63e5601bdff9ba6a48de2", size = 78540 },
 ]
 
+[[package]]
+name = "typer-slim"
+version = "0.15.2"
+source = { registry = "https://mirrors.bfsu.edu.cn/pypi/web/simple" }
+dependencies = [
+    { name = "click" },
+    { name = "typing-extensions" },
+]
+sdist = { url = "https://mirrors.bfsu.edu.cn/pypi/web/packages/29/67/88189eb827c491646511dc6c806e2e6e543241cae5438383aa042b1dfa40/typer_slim-0.15.2.tar.gz", hash = "sha256:4a666bb7839a88f51dd25d078d36dbc1d0f37c8c2696e184fbc1f3eaa314a91b", size = 100755 }
+wheels = [
+    { url = "https://mirrors.bfsu.edu.cn/pypi/web/packages/9e/84/9b68e98bf7417d25e38b27a0296bfcbc6719b15d7000f4c09d9716fa9d11/typer_slim-0.15.2-py3-none-any.whl", hash = "sha256:4273014a3378b24367bffed45c2ce8dd3d85bd201a6f02e51ba6b19f336009be", size = 45117 },
+]
+
+[package.optional-dependencies]
+standard = [
+    { name = "rich" },
+    { name = "shellingham" },
+]
+
 [[package]]
 name = "typing-extensions"
 version = "4.13.2"

fabricatio-0.2.11.dev2/examples/write_article/write_article.py

@@ -1,90 +0,0 @@
-"""Example of using the library."""
-
-import asyncio
-from pathlib import Path
-
-from fabricatio import Event, Role, WorkFlow, logger
-from fabricatio.actions.article import (
-    GenerateArticleProposal,
-    GenerateInitialOutline,
-)
-from fabricatio.actions.article_rag import WriteArticleContentRAG
-from fabricatio.actions.output import (
-    DumpFinalizedOutput,
-    PersistentAll,
-)
-from fabricatio.models.action import Action
-from fabricatio.models.extra.article_main import Article
-from fabricatio.models.extra.article_outline import ArticleOutline
-from fabricatio.models.extra.article_proposal import ArticleProposal
-from fabricatio.models.task import Task
-from fabricatio.utils import ok
-
-
-class Connect(Action):
-    """Connect the article with the article_outline and article_proposal."""
-
-    output_key: str = "article"
-    """Connect the article with the article_outline and article_proposal."""
-
-    async def _execute(
-        self,
-        article_briefing: str,
-        article_proposal: ArticleProposal,
-        article_outline: ArticleOutline,
-        article: Article,
-        **cxt,
-    ) -> Article:
-        """Connect the article with the article_outline and article_proposal."""
-        return article.update_ref(article_outline.update_ref(article_proposal.update_ref(article_briefing)))
-
-
-async def main(article: bool) -> None:
-    """Main function."""
-    Role(
-        name="Undergraduate Researcher",
-        description="Write an outline for an article in typst format.",
-        llm_model="openai/qwen-max",
-        llm_temperature=0.63,
-        llm_stream=True,
-        llm_top_p=0.85,
-        llm_max_tokens=8191,
-        llm_rpm=600,
-        llm_tpm=900000,
-        registry={
-            Event.quick_instantiate(ns := "article"): WorkFlow(
-                name="Generate Article Outline",
-                description="Generate an outline for an article. dump the outline to the given path. in typst format.",
-                steps=(
-                    GenerateArticleProposal,
-                    GenerateInitialOutline(output_key="article_outline", supervisor=False),
-                    PersistentAll,
-                    WriteArticleContentRAG(
-                        output_key="to_dump",
-                        llm_top_p=0.9,
-                        ref_limit=60,
-                        llm_model="openai/qwq-plus",
-                        target_collection="article_chunks",
-                        extractor_model="openai/qwen-plus",
-                        query_model="openai/qwen-max",
-                    ),
-                    DumpFinalizedOutput(output_key="task_output"),
-                    PersistentAll,
-                ),
-            ).update_init_context(
-                article_briefing=Path("./article_briefing.txt").read_text(),
-                dump_path="out.typ",
-                persist_dir="persistent",
-                collection_name="article_chunks",
-            ),
-        },
-    )
-
-    if article:
-        proposed_task = Task(name="write an article")
-        path = ok(await proposed_task.delegate(ns), "Failed to generate an article ")
-        logger.success(f"The outline is saved in:\n{path}")
-
-
-if __name__ == "__main__":
-    asyncio.run(main(True))