From ea0d6c27e9a30a32762d3b38fbdac4aaea854d8b Mon Sep 17 00:00:00 2001 From: zachary62 Date: Thu, 20 Mar 2025 16:31:05 -0400 Subject: [PATCH] add streaming --- cookbook/pocketflow-llm-streaming/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cookbook/pocketflow-llm-streaming/main.py b/cookbook/pocketflow-llm-streaming/main.py index 756962c..3885b75 100644 --- a/cookbook/pocketflow-llm-streaming/main.py +++ b/cookbook/pocketflow-llm-streaming/main.py @@ -1,7 +1,7 @@ import time import threading from pocketflow import Node, Flow -from utils import fake_stream_llm +from utils import fake_stream_llm, stream_llm class StreamNode(Node): def __init__(self, max_retries=1, wait=0):