Skip to content

Commit cbbc870

Browse files
authored
fix lm studio backend
1 parent bd6ed9b commit cbbc870

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

src/chat_lm_studio.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@ def ask_local_chatgpt(self, query, selected_database):
124124
self.signals.response_signal.emit("\n")
125125

126126
citations = self.handle_response_and_cleanup(full_response, metadata_list)
127-
self.signals.citation_signal.emit(citations)
127+
self.signals.citations_signal.emit(citations)
128128
self.signals.finished_signal.emit()
129129

130130
class LMStudioChatThread(QThread):
@@ -175,7 +175,7 @@ def is_lm_studio_available():
175175
| - response_signal - chunk.choices[0].delta.content
176176
| - error_signal |
177177
| - finished_signal |
178-
| - citation_signal |
178+
| - citations_signal |
179179
| | |
180180
| GUI Updates: Cleanup Operations:
181181
| - update_response_lm_studio() - handle_response_and_cleanup()
@@ -184,6 +184,6 @@ def is_lm_studio_available():
184184
| - display_citations_in_widget() - gc.collect()
185185
| |
186186
| Emit Final Signals:
187-
| - citation_signal
187+
| - citations_signal
188188
| - finished_signal
189189
"""

src/gui_tabs_database_query.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -482,7 +482,8 @@ def toggle_recording(self):
482482
self.is_recording = not self.is_recording
483483

484484
def update_response_lm_studio(self, response_chunk):
485-
self.response_widget.setPlainText(self.response_widget.toPlainText() + response_chunk)
485+
self.raw_response += response_chunk
486+
self._render_html()
486487
self.response_widget.verticalScrollBar().setValue(
487488
self.response_widget.verticalScrollBar().maximum()
488489
)

0 commit comments

Comments
 (0)