121 lines
3.4 KiB
Python
121 lines
3.4 KiB
Python
"""
|
|
PocketFlow Cookbook Example: Article Writing Workflow
|
|
|
|
Difficulty: ☆☆☆ Dummy Level
|
|
Source: https://github.com/The-Pocket/PocketFlow/tree/main/cookbook/pocketflow-workflow
|
|
|
|
Description:
|
|
A writing workflow that outlines, writes content, and applies styling.
|
|
Demonstrates:
|
|
- Sequential multi-step workflow
|
|
- Progressive content generation
|
|
- Task decomposition pattern
|
|
"""
|
|
|
|
from pocketflow import Node, Flow
|
|
# from utils import call_llm # You need to implement this
|
|
|
|
|
|
class GenerateOutlineNode(Node):
|
|
"""Generate article outline from topic"""
|
|
|
|
def prep(self, shared):
|
|
return shared["topic"]
|
|
|
|
def exec(self, topic):
|
|
"""Create outline with LLM"""
|
|
prompt = f"Create a detailed outline for an article about: {topic}"
|
|
# outline = call_llm(prompt)
|
|
outline = f"Outline for {topic}:\n1. Introduction\n2. Main Points\n3. Conclusion"
|
|
print(f"\n📋 Outline Generated ({len(outline)} chars)")
|
|
return outline
|
|
|
|
def post(self, shared, prep_res, exec_res):
|
|
shared["outline"] = exec_res
|
|
return "default"
|
|
|
|
|
|
class WriteDraftNode(Node):
|
|
"""Write article draft from outline"""
|
|
|
|
def prep(self, shared):
|
|
return shared["outline"]
|
|
|
|
def exec(self, outline):
|
|
"""Generate content based on outline"""
|
|
prompt = f"Write content based on this outline:\n{outline}"
|
|
# draft = call_llm(prompt)
|
|
draft = f"Draft article based on outline:\n\n{outline}\n\n[Article content here...]"
|
|
print(f"\n✍️ Draft Written ({len(draft)} chars)")
|
|
return draft
|
|
|
|
def post(self, shared, prep_res, exec_res):
|
|
shared["draft"] = exec_res
|
|
return "default"
|
|
|
|
|
|
class RefineArticleNode(Node):
|
|
"""Polish and refine the draft"""
|
|
|
|
def prep(self, shared):
|
|
return shared["draft"]
|
|
|
|
def exec(self, draft):
|
|
"""Improve draft quality"""
|
|
prompt = f"Review and improve this draft:\n{draft}"
|
|
# final = call_llm(prompt)
|
|
final = f"Refined version:\n\n{draft}\n\n[Enhanced with better flow and clarity]"
|
|
print(f"\n✨ Article Refined ({len(final)} chars)")
|
|
return final
|
|
|
|
def post(self, shared, prep_res, exec_res):
|
|
shared["final_article"] = exec_res
|
|
print("\n✅ Article Complete!")
|
|
return "default"
|
|
|
|
|
|
# Build the workflow
|
|
def create_article_flow():
|
|
"""Create sequential article writing workflow"""
|
|
outline = GenerateOutlineNode()
|
|
draft = WriteDraftNode()
|
|
refine = RefineArticleNode()
|
|
|
|
# Sequential pipeline
|
|
outline >> draft >> refine
|
|
|
|
flow = Flow(start=outline)
|
|
return flow
|
|
|
|
|
|
# Example usage
|
|
def run_flow(topic="AI Safety"):
|
|
"""Run the article writing workflow"""
|
|
shared = {"topic": topic}
|
|
|
|
print(f"\n=== Starting Article Workflow: {topic} ===\n")
|
|
|
|
flow = create_article_flow()
|
|
flow.run(shared)
|
|
|
|
# Output summary
|
|
print("\n=== Workflow Statistics ===")
|
|
print(f"Topic: {shared['topic']}")
|
|
print(f"Outline: {len(shared['outline'])} characters")
|
|
print(f"Draft: {len(shared['draft'])} characters")
|
|
print(f"Final: {len(shared['final_article'])} characters")
|
|
|
|
return shared
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import sys
|
|
|
|
# Get topic from command line or use default
|
|
topic = " ".join(sys.argv[1:]) if len(sys.argv) > 1 else "AI Safety"
|
|
result = run_flow(topic)
|
|
|
|
# Print final article
|
|
print("\n=== Final Article ===")
|
|
print(result["final_article"])
|