enhance workflow

Author: leehuwuj

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

@@ -13,7 +13,7 @@ def create_artifact_generator(chat_history: List[ChatMessage]):
         name="ArtifactGenerator",
         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.",
+        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.",
         chat_history=chat_history,
         verbose=True,
     )

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

@@ -66,6 +66,7 @@ async def start(self, ctx: Context, ev: StartEvent) -> ResearchEvent:
         ctx.data["streaming"] = getattr(ev, "streaming", False)
         # start the workflow with researching about a topic
         ctx.data["task"] = ev.input
+        ctx.data["user_input"] = ev.input
         return ResearchEvent(input=f"Research for this task: {ev.input}")
 
     @step()
@@ -117,8 +118,9 @@ async def review(
             )
         )
         if post_is_good:
+            user_input = ctx.data["user_input"]
             return GenerateArtifactEvent(
-                input=f"You're blog post is ready for publication. Please respond with just the blog post. Blog post: ```{old_content}```",
+                input=f"Please generate an artifact for this content: ```{old_content}```. The user input is: ```{user_input}```",
             )
         else:
             return WriteEvent(