add streaming

2025-03-20 16:31:05 -04:00 · 2025-03-20 16:31:05 -04:00 · ea0d6c27e9
parent 795c132018
commit ea0d6c27e9
1 changed files with 1 additions and 1 deletions
--- 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):