add streaming
This commit is contained in:
@@ -1,7 +1,7 @@
|
|||||||
import time
|
import time
|
||||||
import threading
|
import threading
|
||||||
from pocketflow import Node, Flow
|
from pocketflow import Node, Flow
|
||||||
from utils import fake_stream_llm
|
from utils import fake_stream_llm, stream_llm
|
||||||
|
|
||||||
class StreamNode(Node):
|
class StreamNode(Node):
|
||||||
def __init__(self, max_retries=1, wait=0):
|
def __init__(self, max_retries=1, wait=0):
|
||||||
|
|||||||
Reference in New Issue
Block a user