mirror of
https://github.com/aljazceru/Tutorial-Codebase-Knowledge.git
synced 2025-12-18 06:54:24 +01:00
33 lines
1.0 KiB
Python
33 lines
1.0 KiB
Python
from pocketflow import Flow
|
|
# Import all node classes from nodes.py
|
|
from nodes import (
|
|
FetchRepo,
|
|
IdentifyAbstractions,
|
|
AnalyzeRelationships,
|
|
OrderChapters,
|
|
WriteChapters,
|
|
CombineTutorial
|
|
)
|
|
|
|
def create_tutorial_flow():
|
|
"""Creates and returns the codebase tutorial generation flow."""
|
|
|
|
# Instantiate nodes
|
|
fetch_repo = FetchRepo()
|
|
identify_abstractions = IdentifyAbstractions(max_retries=3, wait=10)
|
|
analyze_relationships = AnalyzeRelationships(max_retries=3, wait=10)
|
|
order_chapters = OrderChapters(max_retries=3, wait=10)
|
|
write_chapters = WriteChapters(max_retries=3, wait=10) # This is a BatchNode
|
|
combine_tutorial = CombineTutorial()
|
|
|
|
# Connect nodes in sequence based on the design
|
|
fetch_repo >> identify_abstractions
|
|
identify_abstractions >> analyze_relationships
|
|
analyze_relationships >> order_chapters
|
|
order_chapters >> write_chapters
|
|
write_chapters >> combine_tutorial
|
|
|
|
# Create the flow starting with FetchRepo
|
|
tutorial_flow = Flow(start=fetch_repo)
|
|
|
|
return tutorial_flow |