diff --git a/cookbook/pocketflow-llm-streaming/main.py b/cookbook/pocketflow-llm-streaming/main.py index 532d6b2..d5728d7 100644 --- a/cookbook/pocketflow-llm-streaming/main.py +++ b/cookbook/pocketflow-llm-streaming/main.py @@ -18,7 +18,7 @@ class StreamNode(Node): # Get prompt from shared store prompt = shared["prompt"] # Get chunks from LLM function - chunks = fake_stream_llm(prompt) + chunks = stream_llm(prompt) return chunks, interrupt_event, listener_thread def exec(self, prep_res):