pocketflow/cookbook/pocketflow-tool-search/flow.py

19 lines
419 B
Python

from pocketflow import Flow
from nodes import SearchNode, AnalyzeResultsNode
def create_flow() -> Flow:
"""Create and configure the search flow
Returns:
Flow: Configured flow ready to run
"""
# Create nodes
search = SearchNode()
analyze = AnalyzeResultsNode()
# Connect nodes
search >> analyze
# Create flow starting with search
return Flow(start=search)