Spaces:

neerajkalyank
/

HRMS_Chatbot

Sleeping

HRMS_Chatbot / streaming_callback.py

Update streaming_callback.py

a159664 verified 10 months ago

318 Bytes

from langchain_core.callbacks import BaseCallbackHandler

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

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

	from langchain_core.callbacks import BaseCallbackHandler

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

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