fix: minor response bug

datvodinh · May 12, 2024 · 22cb425 · 22cb425
1 parent f3a4bd8
commit 22cb425
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rag_chatbot/ui/ui.py b/rag_chatbot/ui/ui.py
@@ -65,12 +65,12 @@ def stream_response(
         for text in response.response_gen:
             answer.append(text)
             yield (
-                {'text': DefaultElement.DEFAULT_MESSAGE},
+                DefaultElement.DEFAULT_MESSAGE,
                 history + [[message, "".join(answer)]],
                 DefaultElement.ANSWERING_STATUS
             )
         yield (
-            {'text': DefaultElement.DEFAULT_MESSAGE},
+            DefaultElement.DEFAULT_MESSAGE,
             history + [[message, "".join(answer)]],
             DefaultElement.COMPLETED_STATUS
         )