implement showing result feature

yjg30737 · yjg30737 · commit aae599068607 · 2023-11-11T14:40:14.000+09:00
diff --git a/pyqt_openai/chat_widget/aiChatUnit.py b/pyqt_openai/chat_widget/aiChatUnit.py
@@ -161,9 +161,10 @@ def disableGUIDuringGenerateResponse(self):
         self.__copyBtn.setEnabled(False)
         self.__infoBtn.setEnabled(False)
 
-    def showConvResultInfo(self, finish_reason):
+    def showConvResultInfo(self, info_dict):
         self.__copyBtn.setEnabled(True)
         self.__infoBtn.setEnabled(True)
+        self.__result_info = info_dict
 
     def setText(self, text: str):
         self.__lbl = QLabel(text)
diff --git a/pyqt_openai/chat_widget/convUnitResultDialog.py b/pyqt_openai/chat_widget/convUnitResultDialog.py
@@ -15,11 +15,9 @@ def __initUi(self):
         self.setWindowTitle('Conversation Result')
         self.setWindowFlags(Qt.Window | Qt.WindowCloseButtonHint)
 
-        model_nameLbl = QLabel('model_nameLbl')
-        finish_reasonLbl = QLabel('finish_reasonLbl')
-        prompt_tokensLbl = QLabel('prompt_tokensLbl')
-        completion_tokensLbl = QLabel('completion_tokensLbl')
-        total_tokensLbl = QLabel('total_tokensLbl')
+        lbls = []
+        for k, v in self.__result_info.items():
+            lbls.append(QLabel(f'{k}: {v}'))
 
         sep = QFrame()
         sep.setFrameShape(QFrame.HLine)
@@ -29,11 +27,8 @@ def __initUi(self):
         okBtn.clicked.connect(self.accept)
 
         lay = QFormLayout()
-        lay.addWidget(model_nameLbl)
-        lay.addWidget(finish_reasonLbl)
-        lay.addWidget(prompt_tokensLbl)
-        lay.addWidget(completion_tokensLbl)
-        lay.addWidget(total_tokensLbl)
+        for lbl in lbls:
+            lay.addWidget(lbl)
         lay.addWidget(sep)
         lay.addWidget(okBtn)
 
diff --git a/pyqt_openai/sqlite.py b/pyqt_openai/sqlite.py
@@ -545,12 +545,15 @@ def selectAllContentOfConv(self):
 
     def insertConvUnit(self, id, user_f, conv, info):
         try:
-            # TODO 2023-11-11
-            # Insert a row into the table
             finish_reason = info['finish_reason']
+            model_name = info['model_name']
+            prompt_tokens = info['prompt_tokens']
+            completion_tokens = info['completion_tokens']
+            total_tokens = info['total_tokens']
             self.__c.execute(
-                f'INSERT INTO {self.__conv_unit_tb_nm}{id} (id_fk, is_user, conv, finish_reason) VALUES (?, ?, ?, ?)',
-                (id, user_f, conv, finish_reason))
+                f'INSERT INTO {self.__conv_unit_tb_nm}{id} (id_fk, is_user, conv, finish_reason, model_name, '
+                f'prompt_tokens, completion_tokens, total_tokens) VALUES (?, ?, ?, ?, ?, ?, ?, ?)',
+                (id, user_f, conv, finish_reason, model_name, prompt_tokens, completion_tokens, total_tokens))
             new_id = self.__c.lastrowid
             # Commit the transaction
             self.__conn.commit()
