Spaces:

AIfenaike
/

multi_func_chatbot

Build error

Create streaming.py

1929340 verified over 2 years ago

354 Bytes

from langchain_core.callbacks.base import BaseCallbackHandler

class StreamHandler(BaseCallbackHandler):
    
    def __init__(self, container, initial_text=""):
        self.container = container
        self.text = initial_text

    def on_llm_new_token(self, token: str, **kwargs):
        self.text += token
        self.container.markdown(self.text)

	from langchain_core.callbacks.base import BaseCallbackHandler

	class StreamHandler(BaseCallbackHandler):

	def __init__(self, container, initial_text=""):
	self.container = container
	self.text = initial_text

	def on_llm_new_token(self, token: str, **kwargs):
	self.text += token
	self.container.markdown(self.text)