rename to publisher

Author: leehuwuj

templates/types/multiagent/fastapi/app/examples/choreography.py

@@ -2,14 +2,14 @@
 
 from app.agents.multi import AgentCallingAgent
 from app.agents.single import FunctionCallingAgent
-from app.examples.artifact_generator import create_artifact_generator
+from app.examples.publisher import create_publisher
 from app.examples.researcher import create_researcher
 from llama_index.core.chat_engine.types import ChatMessage
 
 
 def create_choreography(chat_history: Optional[List[ChatMessage]] = None):
     researcher = create_researcher(chat_history)
-    artifact_generator = create_artifact_generator(chat_history)
+    publisher = create_publisher(chat_history)
     reviewer = FunctionCallingAgent(
         name="reviewer",
         role="expert in reviewing blog posts",
@@ -19,12 +19,12 @@ def create_choreography(chat_history: Optional[List[ChatMessage]] = None):
     )
     return AgentCallingAgent(
         name="writer",
-        agents=[researcher, reviewer, artifact_generator],
+        agents=[researcher, reviewer, publisher],
         role="expert in writing blog posts",
         system_prompt="""You are an expert in writing blog posts. You are given a task to write a blog post. Before starting to write the post, consult the researcher agent to get the information you need. Don't make up any information yourself.
         After creating a draft for the post, send it to the reviewer agent to receive some feedback and make sure to incorporate the feedback from the reviewer.
         You can consult the reviewer and researcher maximal two times. Your output should just contain the blog post. 
-        Finally, always request the artifact generator to create an artifact (pdf, html) that user can download and use for publishing the blog post.""",
+        Finally, always request the publisher to create an artifact (pdf, html) and publish the blog post.""",
         # TODO: add chat_history support to AgentCallingAgent
         # chat_history=chat_history,
         verbose=True,

templates/types/multiagent/fastapi/app/examples/orchestrator.py

@@ -2,7 +2,7 @@
 
 from app.agents.multi import AgentOrchestrator
 from app.agents.single import FunctionCallingAgent
-from app.examples.artifact_generator import create_artifact_generator
+from app.examples.publisher import create_publisher
 from app.examples.researcher import create_researcher
 from llama_index.core.chat_engine.types import ChatMessage
 
@@ -22,8 +22,8 @@ def create_orchestrator(chat_history: Optional[List[ChatMessage]] = None):
         Especially check for logical inconsistencies and proofread the post for grammar and spelling errors.""",
         chat_history=chat_history,
     )
-    artifact_generator = create_artifact_generator(chat_history)
+    publisher = create_publisher(chat_history)
     return AgentOrchestrator(
-        agents=[writer, reviewer, researcher, artifact_generator],
+        agents=[writer, reviewer, researcher, publisher],
         refine_plan=False,
     )

templates/types/multiagent/fastapi/app/examples/artifact_generator.py renamed to templates/types/multiagent/fastapi/app/examples/publisher.py

@@ -6,11 +6,11 @@
 from llama_index.core.tools import FunctionTool
 
 
-def create_artifact_generator(chat_history: List[ChatMessage]):
+def create_publisher(chat_history: List[ChatMessage]):
     artifact_tool = FunctionTool.from_defaults(ArtifactGenerator.generate_artifact)
 
     return FunctionCallingAgent(
-        name="ArtifactGenerator",
+        name="publisher",
         tools=[artifact_tool],
         role="expert in generating artifacts (pdf, html)",
         system_prompt="You are generator that help generate artifacts (pdf, html) from a given content. Please always respond the content again along with the generated artifact.",

templates/types/multiagent/fastapi/app/examples/workflow.py

@@ -1,7 +1,7 @@
 from typing import AsyncGenerator, List, Optional
 
 from app.agents.single import AgentRunEvent, AgentRunResult, FunctionCallingAgent
-from app.examples.artifact_generator import create_artifact_generator
+from app.examples.publisher import create_publisher
 from app.examples.researcher import create_researcher
 from llama_index.core.chat_engine.types import ChatMessage
 from llama_index.core.workflow import (
@@ -18,7 +18,7 @@ def create_workflow(chat_history: Optional[List[ChatMessage]] = None):
     researcher = create_researcher(
         chat_history=chat_history,
     )
-    artifact_generator = create_artifact_generator(
+    publisher = create_publisher(
         chat_history=chat_history,
     )
     writer = FunctionCallingAgent(
@@ -38,7 +38,7 @@ def create_workflow(chat_history: Optional[List[ChatMessage]] = None):
         researcher=researcher,
         writer=writer,
         reviewer=reviewer,
-        artifact_generator=artifact_generator,
+        publisher=publisher,
     )
     return workflow
 
@@ -55,7 +55,7 @@ class ReviewEvent(Event):
     input: str
 
 
-class GenerateArtifactEvent(Event):
+class PublishEvent(Event):
     input: str
 
 
@@ -106,7 +106,7 @@ async def write(
     @step()
     async def review(
         self, ctx: Context, ev: ReviewEvent, reviewer: FunctionCallingAgent
-    ) -> WriteEvent | GenerateArtifactEvent:
+    ) -> WriteEvent | PublishEvent:
         result: AgentRunResult = await self.run_agent(ctx, reviewer, ev.input)
         review = result.response.message.content
         old_content = ctx.data["result"].response.message.content
@@ -119,8 +119,8 @@ async def review(
         )
         if post_is_good:
             user_input = ctx.data["user_input"]
-            return GenerateArtifactEvent(
-                input=f"Please generate an artifact for this content: ```{old_content}```. The user input is: ```{user_input}```",
+            return PublishEvent(
+                input=f"Please publish this content: ```{old_content}```. The user request was: ```{user_input}```",
             )
         else:
             return WriteEvent(
@@ -137,13 +137,13 @@ async def review(
             )
 
     @step()
-    async def generate_artifact(
+    async def publish(
         self,
         ctx: Context,
-        ev: GenerateArtifactEvent,
-        artifact_generator: FunctionCallingAgent,
+        ev: PublishEvent,
+        publisher: FunctionCallingAgent,
     ) -> StopEvent:
-        result: AgentRunResult = await self.run_agent(ctx, artifact_generator, ev.input)
+        result: AgentRunResult = await self.run_agent(ctx, publisher, ev.input)
         return StopEvent(result=result)
 
     async def run_agent(

templates/types/streaming/nextjs/app/components/ui/chat/chat-message/chat-agent-events.tsx

@@ -17,6 +17,7 @@ const AgentIcons: Record<string, LucideIcon> = {
   researcher: icons.ScanSearch,
   writer: icons.PenLine,
   reviewer: icons.MessageCircle,
+  publisher: icons.BookCheck,
 };
 
 type MergedEvent = {