ciorant commited on
Commit
985eea4
·
verified ·
1 Parent(s): cf96ea1

Update src/chatbot.py

Browse files
Files changed (1) hide show
  1. src/chatbot.py +7 -4
src/chatbot.py CHANGED
@@ -168,14 +168,17 @@ class BioethicsChatbot:
168
  """
169
 
170
  self.stream_handler.current_text = ""
 
 
 
171
 
172
- _ = self.llm.invoke(prompt) # streaming happens here
173
- print() # newline after streaming
174
-
175
  answer = self.stream_handler.get_text()
176
  self.history.append((question, answer))
177
-
178
  return answer
179
 
180
 
181
 
 
 
168
  """
169
 
170
  self.stream_handler.current_text = ""
171
+
172
+ if stream_placeholder:
173
+ self.stream_handler.placeholder = stream_placeholder
174
 
175
+ # streaming happens here
176
+ _ = self.llm.invoke(prompt)
177
+
178
  answer = self.stream_handler.get_text()
179
  self.history.append((question, answer))
 
180
  return answer
181
 
182
 
183
 
184
+