15 lines
515 B
Python
15 lines
515 B
Python
from pocketflow import Flow
|
|
from nodes import GetTopicNode, GenerateJokeNode, GetFeedbackNode
|
|
|
|
def create_joke_flow() -> Flow:
|
|
"""Creates and returns the joke generation flow."""
|
|
get_topic_node = GetTopicNode()
|
|
generate_joke_node = GenerateJokeNode()
|
|
get_feedback_node = GetFeedbackNode()
|
|
|
|
get_topic_node >> generate_joke_node
|
|
generate_joke_node >> get_feedback_node
|
|
get_feedback_node - "Disapprove" >> generate_joke_node
|
|
|
|
joke_flow = Flow(start=get_topic_node)
|
|
return joke_flow |