Fixed pocketflow-llm-streaming

2025-05-09 15:51:54 +08:00 · 2025-05-09 15:51:54 +08:00 · b91494fd38
parent 2e968df11a
commit b91494fd38
1 changed files with 1 additions and 1 deletions
--- 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):