Initial commit

skills/pocketflow/assets/examples/02_workflow.py

@@ -0,0 +1,120 @@
+"""
+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"])