from pocketflow import Flow

from nodes import Decide, Search, Answer

def createFlow(logger):
    decide = Decide()
    decide.setLogger(logger)
    search = Search()
    search.setLogger(logger)
    answer = Answer()
    answer.setLogger(logger)

    decide - "search" >> search
    decide - "answer" >> answer
    search - "decide" >> decide

    return Flow(start=decide)