30 lines
730 B
Python
30 lines
730 B
Python
from pocketflow import Flow
|
|
|
|
from nodes import (
|
|
DecideAction,
|
|
CheckWeather,
|
|
BookHotel,
|
|
FollowUp,
|
|
ResultNotification,
|
|
)
|
|
|
|
|
|
def create_flow():
|
|
"""
|
|
Create and connect the nodes to form a complete agent flow.
|
|
"""
|
|
decide_action = DecideAction()
|
|
check_weather = CheckWeather()
|
|
book_hotel = BookHotel()
|
|
follow_up = FollowUp()
|
|
result_notification = ResultNotification()
|
|
|
|
decide_action - "check-weather" >> check_weather
|
|
check_weather >> decide_action
|
|
decide_action - "book-hotel" >> book_hotel
|
|
book_hotel >> decide_action
|
|
decide_action - "follow-up" >> follow_up
|
|
decide_action - "result-notification" >> result_notification
|
|
|
|
return Flow(start=decide_action)
|