jproman's picture
Refactor to use langchain
522e6c3
raw
history blame contribute delete
520 Bytes
from pocketflow import Flow
from nodes import Decide, Search, Wikipedia, Answer
def createFlow(logger):
decide = Decide()
decide.setLogger(logger)
search = Search()
search.setLogger(logger)
wikipedia = Wikipedia()
wikipedia.setLogger(logger)
answer = Answer()
answer.setLogger(logger)
decide - "search" >> search
decide - "wikipedia" >> wikipedia
decide - "answer" >> answer
search - "decide" >> decide
wikipedia - "decide" >> decide
return Flow(start=decide)