pocketflow/cookbook/pocketflow-fastapi-websocket/flow.py

6 lines
168 B
Python

from pocketflow import Flow
from nodes import StreamingChatNode
def create_streaming_chat_flow():
chat_node = StreamingChatNode()
return Flow(start=chat_node)