Merge pull request #52 from Ming-jiayou/Fixed-pocketflow-llm-streaming

Fixed pocketflow-llm-streaming
2025-05-09 11:45:41 -04:00 · 2025-05-09 11:45:41 -04:00 · 3c970ba2d1
parent 2e968df11a b91494fd38
commit 3c970ba2d1
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):