fix minor glitch related to llamaindex and finish reason

yjg30737 · yjg30737 · commit d39de5786c2a · 2023-07-20T14:39:23.000+09:00
diff --git a/pyqt_openai/chat_widget/chatBrowser.py b/pyqt_openai/chat_widget/chatBrowser.py
@@ -73,7 +73,6 @@ def showText(self, text, stream_f, user_f):
             self.widget().setCurrentIndex(1)
         return self.__setLabel(text, stream_f, user_f)
 
-
     def __setLabel(self, text, stream_f, user_f):
         chatUnit = QLabel()
         if user_f:
diff --git a/pyqt_openai/openAiThread.py b/pyqt_openai/openAiThread.py
@@ -15,6 +15,7 @@ class OpenAIThread(QThread):
     First: response
     Second: user or AI
     Third: streaming or not streaming
+    Forth: Finish reason
     """
     replyGenerated = Signal(str, bool, bool, str)
     streamFinished = Signal(str)
@@ -64,7 +65,6 @@ def run(self):
         try:
             self.__llama_idx_instance.set_openai_arg(**self.__openai_arg)
             resp = self.__llama_idx_instance.get_response(self.__query_text)
-            print(resp)
             f = isinstance(resp, StreamingResponse)
             if f:
                 for response_text in resp.response_gen:
@@ -74,6 +74,6 @@ def run(self):
                 self.replyGenerated.emit(resp.response, False, f, '')
         except openai.error.InvalidRequestError as e:
             self.replyGenerated.emit('<p style="color:red">Your request was rejected as a result of our safety system.<br/>'
-                                     'Your prompt may contain text that is not allowed by our safety system.</p>', False)
+                                     'Your prompt may contain text that is not allowed by our safety system.</p>', False, False, 'Error')
         except openai.error.RateLimitError as e:
-            self.replyGenerated.emit(f'<p style="color:red">{e}<br/>Check the usage: https://platform.openai.com/account/usage<br/>Update to paid account: https://platform.openai.com/account/billing/overview', False)
+            self.replyGenerated.emit(f'<p style="color:red">{e}<br/>Check the usage: https://platform.openai.com/account/usage<br/>Update to paid account: https://platform.openai.com/account/billing/overview', False, False, 'Error')
