Sun, 25 Aug 2024 19:44:24 +0200
Implemented some more menu actions.
- Start and Stop a local 'ollama' server in the background.
- List available models.
- List running models.
- Open the 'ollama' model library in a browser.
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
1 | # -*- coding: utf-8 -*- |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
2 | # Copyright (c) 2024 Detlev Offenbach <detlev@die-offenbachs.de> |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
3 | # |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
4 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
5 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
6 | Module implementing the main ollama interface widget. |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
7 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
8 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
9 | import json |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
10 | import os |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
11 | |
8
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
12 | from PyQt6.QtCore import QProcess, QProcessEnvironment, Qt, QTimer, pyqtSlot, QUrl |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
13 | from PyQt6.QtGui import QDesktopServices |
6
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
14 | from PyQt6.QtWidgets import ( |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
15 | QDialog, |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
16 | QInputDialog, |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
17 | QLineEdit, |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
18 | QMenu, |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
19 | QVBoxLayout, |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
20 | QWidget, |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
21 | ) |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
22 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
23 | from eric7 import Globals |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
24 | from eric7.EricGui import EricPixmapCache |
6
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
25 | from eric7.EricWidgets import EricFileDialog, EricMessageBox |
5
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
26 | from eric7.EricWidgets.EricApplication import ericApp |
6
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
27 | from eric7.EricWidgets.EricListSelectionDialog import EricListSelectionDialog |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
28 | |
5
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
29 | from .OllamaChatWidget import OllamaChatWidget |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
30 | from .OllamaClient import OllamaClient |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
31 | from .OllamaHistoryWidget import OllamaHistoryWidget |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
32 | from .Ui_OllamaWidget import Ui_OllamaWidget |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
33 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
34 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
35 | class OllamaWidget(QWidget, Ui_OllamaWidget): |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
36 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
37 | Class implementing the main ollama interface widget. |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
38 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
39 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
40 | OllamaHistoryFile = "ollama_history.json" |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
41 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
42 | def __init__(self, plugin, fromEric=True, parent=None): |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
43 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
44 | Constructor |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
45 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
46 | @param plugin reference to the plug-in object |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
47 | @type PluginOllamaInterface |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
48 | @param fromEric flag indicating the eric-ide mode (defaults to True) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
49 | (True = eric-ide mode, False = application mode) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
50 | @type bool (optional) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
51 | @param parent reference to the parent widget (defaults to None) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
52 | @type QWidget (optional) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
53 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
54 | super().__init__(parent) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
55 | self.setupUi(self) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
56 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
57 | self.__plugin = plugin |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
58 | self.__client = OllamaClient(plugin, self) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
59 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
60 | if fromEric: |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
61 | self.layout().setContentsMargins(0, 3, 0, 0) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
62 | else: |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
63 | self.layout().setContentsMargins(0, 0, 0, 0) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
64 | |
5
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
65 | iconSuffix = "-dark" if ericApp().usesDarkPalette() else "-light" |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
66 | |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
67 | self.ollamaMenuButton.setIcon(EricPixmapCache.getIcon("superMenu")) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
68 | self.reloadModelsButton.setIcon(EricPixmapCache.getIcon("reload")) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
69 | self.newChatButton.setIcon(EricPixmapCache.getIcon("plus")) |
5
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
70 | self.sendButton.setIcon( |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
71 | EricPixmapCache.getIcon( |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
72 | os.path.join("OllamaInterface", "icons", "send{0}".format(iconSuffix)) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
73 | ) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
74 | ) |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
75 | |
6
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
76 | self.ollamaMenuButton.setAutoRaise(True) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
77 | self.ollamaMenuButton.setShowMenuInside(True) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
78 | |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
79 | self.__chatHistoryLayout = QVBoxLayout() |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
80 | self.historyScrollWidget.setLayout(self.__chatHistoryLayout) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
81 | self.__chatHistoryLayout.addStretch(1) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
82 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
83 | self.mainSplitter.setSizes([200, 2000]) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
84 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
85 | self.newChatButton.setEnabled(False) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
86 | self.__handleServerStateChanged(False) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
87 | |
7
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
88 | self.__localServerDialog = None |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
89 | self.__localServerProcess = None |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
90 | |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
91 | self.__connectClient() |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
92 | |
6
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
93 | self.__initOllamaMenu() |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
94 | |
5
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
95 | self.sendButton.clicked.connect(self.__sendMessage) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
96 | self.messageEdit.returnPressed.connect(self.__sendMessage) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
97 | |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
98 | self.__loadHistory() |
5
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
99 | self.__updateMessageEditState() |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
100 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
101 | def __connectClient(self): |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
102 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
103 | Private method to connect the client signals. |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
104 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
105 | self.__client.serverStateChanged.connect(self.__handleServerStateChanged) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
106 | self.__client.serverVersion.connect(self.__setHeaderLabel) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
107 | self.__client.modelsList.connect(self.__populateModelSelector) |
6
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
108 | self.__client.modelsList.connect(self.__checkHistoryModels) |
5
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
109 | self.__client.replyReceived.connect(self.__handleServerMessage) |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
110 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
111 | @pyqtSlot(bool) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
112 | def __handleServerStateChanged(self, ok): |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
113 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
114 | Private slot handling a change in the 'ollama' server responsiveness. |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
115 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
116 | @param ok flag indicating a responsive 'ollama' server |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
117 | @type bool |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
118 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
119 | if ok: |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
120 | self.__finishSetup() |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
121 | else: |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
122 | self.ollamaVersionLabel.setText( |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
123 | self.tr("<b>Error: The configured server is not responding.</b>") |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
124 | ) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
125 | self.setEnabled(ok) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
126 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
127 | @pyqtSlot() |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
128 | def __finishSetup(self): |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
129 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
130 | Private slot to finish the UI setup. |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
131 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
132 | self.__client.version() |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
133 | self.__client.list() |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
134 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
135 | @pyqtSlot() |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
136 | def on_reloadModelsButton_clicked(self): |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
137 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
138 | Private slot to reload the list of available models. |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
139 | """ |
8
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
140 | self.__finishSetup() |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
141 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
142 | @pyqtSlot(str) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
143 | def on_modelComboBox_currentTextChanged(self, model): |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
144 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
145 | Private slot handling the selection of a model. |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
146 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
147 | @param model name of the selected model |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
148 | @type str |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
149 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
150 | self.newChatButton.setEnabled(bool(model)) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
151 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
152 | ############################################################################ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
153 | ## Methods handling signals from the 'ollama' client. |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
154 | ############################################################################ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
155 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
156 | @pyqtSlot(str) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
157 | def __setHeaderLabel(self, version): |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
158 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
159 | Private slot to receive the 'ollama' server version and set the header. |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
160 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
161 | @param version 'ollama' server version' |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
162 | @type str |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
163 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
164 | self.ollamaVersionLabel.setText( |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
165 | self.tr("<b>ollama Server Version {0}</b>").format(version) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
166 | ) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
167 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
168 | @pyqtSlot(list) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
169 | def __populateModelSelector(self, modelNames): |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
170 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
171 | Private slot to receive the list of available model names and populate |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
172 | the model selector with them. |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
173 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
174 | @param modelNames list of model names |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
175 | @type list[str] |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
176 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
177 | self.modelComboBox.clear() |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
178 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
179 | self.modelComboBox.addItem("") |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
180 | self.modelComboBox.addItems(sorted(modelNames)) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
181 | |
6
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
182 | @pyqtSlot(list) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
183 | def __checkHistoryModels(self, modelNames): |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
184 | """ |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
185 | Private slot to set the chat history entry states according to available |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
186 | models. |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
187 | |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
188 | @param modelNames list of model names |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
189 | @type list[str] |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
190 | """ |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
191 | for index in range(self.__chatHistoryLayout.count() - 1): |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
192 | self.__chatHistoryLayout.itemAt(index).widget().checkModelAvailable( |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
193 | modelNames |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
194 | ) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
195 | |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
196 | ############################################################################ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
197 | ## Methods handling signals from the chat history widgets. |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
198 | ############################################################################ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
199 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
200 | def __createHistoryWidget(self, title, model, jsonStr=None): |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
201 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
202 | Private method to create a chat history widget and insert it into the |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
203 | respective layout. |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
204 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
205 | @param title title of the chat |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
206 | @type str |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
207 | @param model name of the model |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
208 | @type str |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
209 | @param jsonStr string containing JSON serialize chat history data (defaults |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
210 | to None) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
211 | @type str (optional) |
5
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
212 | @return reference to the created history widget |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
213 | @rtype OllamaHistoryWidget |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
214 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
215 | history = OllamaHistoryWidget(title=title, model=model, jsonStr=jsonStr) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
216 | self.__chatHistoryLayout.insertWidget( |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
217 | self.__chatHistoryLayout.count() - 1, history |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
218 | ) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
219 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
220 | history.deleteChatHistory.connect(self.__deleteHistory) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
221 | history.dataChanged.connect(self.__saveHistory) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
222 | history.newChatWithHistory.connect(self.__newChatWithHistory) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
223 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
224 | self.__saveHistory() |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
225 | |
5
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
226 | QTimer.singleShot(0, self.__scrollHistoryToBottom) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
227 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
228 | return history |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
229 | |
5
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
230 | @pyqtSlot() |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
231 | def __scrollHistoryToBottom(self): |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
232 | """ |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
233 | Private slot to scroll the history widget to the bottom. |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
234 | """ |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
235 | scrollbar = self.historyScrollArea.verticalScrollBar() |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
236 | scrollbar.setMaximum(self.historyScrollWidget.height()) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
237 | scrollbar.setValue(scrollbar.maximum()) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
238 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
239 | def __findHistoryWidget(self, hid): |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
240 | """ |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
241 | Private method to find the widget of a given chat history ID. |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
242 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
243 | @param hid ID of the chat history |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
244 | @type str |
5
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
245 | @return reference to the chat history widget |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
246 | @rtype OllamaHistoryWidget |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
247 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
248 | for index in range(self.__chatHistoryLayout.count() - 1): |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
249 | widget = self.__chatHistoryLayout.itemAt(index).widget() |
5
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
250 | if widget.getId() == hid: |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
251 | return widget |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
252 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
253 | return None |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
254 | |
6
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
255 | def __getHistoryIds(self): |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
256 | """ |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
257 | Private method to get a list of all history IDs. |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
258 | |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
259 | @return list of history IDs |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
260 | @rtype list[str] |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
261 | """ |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
262 | hids = [] |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
263 | for index in range(self.__chatHistoryLayout.count() - 1): |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
264 | widget = self.__chatHistoryLayout.itemAt(index).widget() |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
265 | hids.append(widget.getId()) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
266 | |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
267 | return hids |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
268 | |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
269 | def __historyFilePath(self): |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
270 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
271 | Private method to get the path name of the chat history file. |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
272 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
273 | @return file path of the chat history file |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
274 | @rtype str |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
275 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
276 | return os.path.join(Globals.getConfigDir(), OllamaWidget.OllamaHistoryFile) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
277 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
278 | @pyqtSlot() |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
279 | def __saveHistory(self): |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
280 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
281 | Private method to save the current chat history to the history file. |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
282 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
283 | # step 1: collect all history entries |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
284 | entries = {} |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
285 | for index in range(self.__chatHistoryLayout.count() - 1): |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
286 | widget = self.__chatHistoryLayout.itemAt(index).widget() |
5
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
287 | hid = widget.getId() |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
288 | entries[hid] = widget.saveToJson() |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
289 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
290 | # step 2: save the collected chat histories |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
291 | filePath = self.__historyFilePath() |
6
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
292 | self.__saveChatHistoryFile(filePath, entries) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
293 | |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
294 | def __saveChatHistoryFile(self, filePath, entries): |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
295 | """ |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
296 | Private method to save the chat history entries to a file. |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
297 | |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
298 | @param filePath file name to save to |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
299 | @type str |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
300 | @param entries dictionary containing the chat history entries as a |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
301 | JSON serialized string indexed by their ID |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
302 | @type dict[str, str] |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
303 | """ |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
304 | try: |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
305 | with open(filePath, "w") as f: |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
306 | json.dump(entries, f) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
307 | except OSError as err: |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
308 | EricMessageBox.critical( |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
309 | self, |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
310 | self.tr("Save Chat History"), |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
311 | self.tr( |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
312 | "<p>The chat history could not be saved to <b>{0}</b>.</p>" |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
313 | "<p>Reason: {1}</p>" |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
314 | ).format(filePath, str(err)), |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
315 | ) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
316 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
317 | def __loadHistory(self): |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
318 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
319 | Private method to load a previously saved history file. |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
320 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
321 | # step 1: load the history file, if it exists |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
322 | filePath = self.__historyFilePath() |
6
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
323 | self.__loadChatHistoriesFile(filePath) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
324 | |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
325 | def __loadChatHistoriesFile(self, filePath, reportDuplicates=False): |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
326 | """ |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
327 | Private method to load chat history entries from a given file. |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
328 | |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
329 | @param filePath path of the chat history file |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
330 | @type str |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
331 | @param reportDuplicates flag indicating to report skipped chat history entries |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
332 | (defaults to False) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
333 | @type bool (optional) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
334 | @return flag indicating success |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
335 | @rtype str |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
336 | """ |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
337 | if not os.path.exists(filePath): |
6
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
338 | return False |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
339 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
340 | try: |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
341 | with open(filePath, "r") as f: |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
342 | entries = json.load(f) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
343 | except OSError as err: |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
344 | EricMessageBox.critical( |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
345 | self, |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
346 | self.tr("Load Chat History"), |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
347 | self.tr( |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
348 | "<p>The chat history could not be loaded from <b>{0}</b>.</p>" |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
349 | "<p>Reason: {1}</p>" |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
350 | ).format(filePath, str(err)), |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
351 | ) |
6
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
352 | return False |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
353 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
354 | # step 2: create history widgets |
6
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
355 | existingIDs = self.__getHistoryIds() |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
356 | skipped = [] |
5
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
357 | for hid in entries: |
6
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
358 | if hid in existingIDs: |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
359 | data = json.loads(entries[hid]) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
360 | skipped.append(data["title"]) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
361 | else: |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
362 | self.__createHistoryWidget("", "", jsonStr=entries[hid]) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
363 | |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
364 | if skipped and reportDuplicates: |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
365 | EricMessageBox.warning( |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
366 | self, |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
367 | self.tr("Load Chat History"), |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
368 | self.tr( |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
369 | "<p>These chats were not loaded because they already existed.</p>" |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
370 | "{0}" |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
371 | ).format("<ul><li>{0}</li></ul>".format("</li><li>".join(skipped))), |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
372 | ) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
373 | |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
374 | return True |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
375 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
376 | def clearHistory(self): |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
377 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
378 | Public method to clear the history entries and close all chats. |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
379 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
380 | while self.__chatHistoryLayout.count() > 1: |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
381 | # do not delete the spacer at the end of the list |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
382 | item = self.__chatHistoryLayout.takeAt(0) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
383 | if item is not None: |
6
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
384 | hid = item.widget().getId() |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
385 | self.__removeChatWidget(hid) |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
386 | item.widget().deleteLater() |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
387 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
388 | self.__saveHistory() |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
389 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
390 | @pyqtSlot(str) |
5
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
391 | def __deleteHistory(self, hid): |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
392 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
393 | Private slot to delete the history with the given ID. |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
394 | |
5
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
395 | @param hid ID of the history to be deleted |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
396 | @type str |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
397 | """ |
5
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
398 | widget = self.__findHistoryWidget(hid) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
399 | if widget is not None: |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
400 | widgetIndex = self.__chatHistoryLayout.indexOf(widget) |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
401 | item = self.__chatHistoryLayout.takeAt(widgetIndex) |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
402 | if item is not None: |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
403 | item.widget().deleteLater() |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
404 | |
5
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
405 | self.__saveHistory() |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
406 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
407 | self.__removeChatWidget(hid) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
408 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
409 | ####################################################################### |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
410 | ## Chat related methods below |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
411 | ####################################################################### |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
412 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
413 | def __findChatWidget(self, hid): |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
414 | """ |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
415 | Private method to find a chat widget given a chat history ID. |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
416 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
417 | @param hid chat history ID |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
418 | @type str |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
419 | @return reference to the chat widget related to the given ID |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
420 | @rtype OllamaChatWidget |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
421 | """ |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
422 | for index in range(self.chatStackWidget.count()): |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
423 | widget = self.chatStackWidget.widget(index) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
424 | if widget.getHistoryId() == hid: |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
425 | return widget |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
426 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
427 | return None |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
428 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
429 | @pyqtSlot() |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
430 | def on_newChatButton_clicked(self): |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
431 | """ |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
432 | Private slot to start a new chat with the 'ollama' server. |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
433 | """ |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
434 | model = self.modelComboBox.currentText() |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
435 | if not model: |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
436 | EricMessageBox.critical( |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
437 | self, |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
438 | self.tr("New Chat"), |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
439 | self.tr("""A model has to be selected first. Aborting..."""), |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
440 | ) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
441 | return |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
442 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
443 | title, ok = QInputDialog.getText( |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
444 | self, |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
445 | self.tr("New Chat"), |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
446 | self.tr("Enter a title for the new chat:"), |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
447 | QLineEdit.EchoMode.Normal, |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
448 | ) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
449 | if ok and title: |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
450 | historyWidget = self.__createHistoryWidget(title, model) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
451 | hid = historyWidget.getId() |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
452 | chatWidget = OllamaChatWidget(hid=hid, title=title, model=model) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
453 | index = self.chatStackWidget.addWidget(chatWidget) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
454 | self.chatStackWidget.setCurrentIndex(index) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
455 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
456 | self.__updateMessageEditState() |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
457 | self.messageEdit.setFocus(Qt.FocusReason.OtherFocusReason) |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
458 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
459 | @pyqtSlot(str) |
5
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
460 | def __newChatWithHistory(self, hid): |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
461 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
462 | Private slot to start a new chat using a previously saved history. |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
463 | |
5
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
464 | @param hid ID of the history to be used |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
465 | @type str |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
466 | """ |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
467 | chatWidget = self.__findChatWidget(hid) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
468 | if chatWidget is None: |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
469 | historyWidget = self.__findHistoryWidget(hid) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
470 | if historyWidget is None: |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
471 | # Oops, treat it as a new chat. |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
472 | self.on_newChatButton_clicked() |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
473 | return |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
474 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
475 | chatWidget = OllamaChatWidget( |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
476 | hid=hid, title=historyWidget.getTitle(), model=historyWidget.getModel() |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
477 | ) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
478 | index = self.chatStackWidget.addWidget(chatWidget) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
479 | self.chatStackWidget.setCurrentIndex(index) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
480 | for message in historyWidget.getMessages(): |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
481 | chatWidget.addMessage(role=message["role"], message=message["content"]) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
482 | else: |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
483 | # simply switch to the already existing chatWidget |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
484 | self.chatStackWidget.setCurrentWidget(chatWidget) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
485 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
486 | self.__updateMessageEditState() |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
487 | self.messageEdit.setFocus(Qt.FocusReason.OtherFocusReason) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
488 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
489 | def __removeChatWidget(self, hid): |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
490 | """ |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
491 | Private method to remove a chat widget given its chat history ID. |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
492 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
493 | @param hid chat history ID |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
494 | @type str |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
495 | """ |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
496 | widget = self.__findChatWidget(hid) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
497 | if widget is not None: |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
498 | self.chatStackWidget.removeWidget(widget) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
499 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
500 | @pyqtSlot() |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
501 | def __updateMessageEditState(self): |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
502 | """ |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
503 | Private slot to set the enabled state of the message line edit and the send |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
504 | button. |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
505 | """ |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
506 | chatActive = bool(self.chatStackWidget.count()) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
507 | hasText = bool(self.messageEdit.text()) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
508 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
509 | self.messageEdit.setEnabled(chatActive) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
510 | self.sendButton.setEnabled(chatActive and hasText) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
511 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
512 | @pyqtSlot(str) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
513 | def on_messageEdit_textChanged(self, msg): |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
514 | """ |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
515 | Private slot to handle a change of the entered message. |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
516 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
517 | @param msg text of the message line edit |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
518 | @type str |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
519 | """ |
5
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
520 | self.sendButton.setEnabled(bool(msg)) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
521 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
522 | @pyqtSlot() |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
523 | def __sendMessage(self): |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
524 | """ |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
525 | Private method to send the given message of the current chat to the |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
526 | 'ollama' server. |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
527 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
528 | This sends the message with context (i.e. the history of the current chat). |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
529 | """ |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
530 | msg = self.messageEdit.text() |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
531 | if not msg: |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
532 | # empty message => ignore |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
533 | return |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
534 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
535 | if not bool(self.chatStackWidget.count()): |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
536 | # no current stack => ignore |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
537 | return |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
538 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
539 | # 1. determine hid of the current chat via chat stack widget |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
540 | chatWidget = self.chatStackWidget.currentWidget() |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
541 | hid = chatWidget.getHistoryId() |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
542 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
543 | # 2. get chat history widget via hid from chat history widget |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
544 | historyWidget = self.__findHistoryWidget(hid) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
545 | if historyWidget is not None: |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
546 | # 3. append the message to the history |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
547 | historyWidget.addToMessages("user", msg) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
548 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
549 | # 4. get the complete messages list from the history |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
550 | messages = historyWidget.getMessages() |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
551 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
552 | # 5. add the message to the current chat and an empty one |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
553 | # for the response |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
554 | chatWidget.addMessage("user", msg) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
555 | chatWidget.addMessage("assistant", "") |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
556 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
557 | # 6. send the request via the client (non-streaming (?)) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
558 | model = historyWidget.getModel() |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
559 | self.__client.chat( |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
560 | model=model, |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
561 | messages=messages, |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
562 | streaming=self.__plugin.getPreferences("StreamingChatResponse"), |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
563 | ) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
564 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
565 | # 7. clear the message editor and give input focus back |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
566 | self.messageEdit.clear() |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
567 | self.messageEdit.setFocus(Qt.FocusReason.OtherFocusReason) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
568 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
569 | @pyqtSlot(str, str, bool) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
570 | def __handleServerMessage(self, content, role, done): |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
571 | """ |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
572 | Private slot handling an 'ollama' server chat response. |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
573 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
574 | @param content message sent by the server |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
575 | @type str |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
576 | @param role role name |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
577 | @type str |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
578 | @param done flag indicating the last chat response |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
579 | @type bool |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
580 | """ |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
581 | if not bool(self.chatStackWidget.count()): |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
582 | # no current stack => ignore |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
583 | return |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
584 | |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
585 | chatWidget = self.chatStackWidget.currentWidget() |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
586 | chatWidget.appendMessage(content) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
587 | if done: |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
588 | hid = chatWidget.getHistoryId() |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
589 | historyWidget = self.__findHistoryWidget(hid) |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
590 | if historyWidget is not None: |
6e8af43d537d
Implemented the 'chat' functionality.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
4
diff
changeset
|
591 | historyWidget.addToMessages(role, chatWidget.getRecentMessage()) |
4
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
592 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
593 | ####################################################################### |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
594 | ## Menu related methods below |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
595 | ####################################################################### |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
596 | |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
597 | def __initOllamaMenu(self): |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
598 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
599 | Private method to create the super menu and attach it to the super |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
600 | menu button. |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
601 | """ |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
602 | # TODO: implement the menu and menu methods |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
603 | # * Pull Model |
7dd1b9cd3150
Implemented most of the Chat History widgets.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
diff
changeset
|
604 | # * Remove Model |
6
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
605 | ################################################################### |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
606 | ## Menu with Chat History related actions |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
607 | ################################################################### |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
608 | |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
609 | self.__chatHistoryMenu = QMenu(self.tr("Chat History")) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
610 | self.__chatHistoryMenu.addAction(self.tr("Load"), self.__loadHistory) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
611 | self.__chatHistoryMenu.addSeparator() |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
612 | self.__clearHistoriesAct = self.__chatHistoryMenu.addAction( |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
613 | self.tr("Clear All"), self.__menuClearAllHistories |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
614 | ) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
615 | self.__chatHistoryMenu.addSeparator() |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
616 | self.__chatHistoryMenu.addAction(self.tr("Import"), self.__menuImportHistories) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
617 | self.__chatHistoryMenu.addAction(self.tr("Export"), self.__menuExportHistories) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
618 | |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
619 | ################################################################### |
8
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
620 | ## Menu with Model related actions |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
621 | ################################################################### |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
622 | |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
623 | self.__modelMenu = QMenu(self.tr("Model Management")) |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
624 | self.__modelMenu.addAction(self.tr("List Models"), self.__showModels) |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
625 | self.__modelMenu.addAction( |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
626 | self.tr("List Running Models"), self.__showRunningModels |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
627 | ) |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
628 | self.__modelMenu.addSeparator() |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
629 | self.__modelMenu.addAction( |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
630 | self.tr("Show Model Library"), self.__showModelLibrary |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
631 | ) |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
632 | self.__modelMenu.addSeparator() |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
633 | self.__modelMenu.addAction(self.tr("Download Model"), self.__pullModel) |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
634 | self.__modelMenu.addAction(self.tr("Remove Model"), self.__removeModel) |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
635 | |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
636 | ################################################################### |
7
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
637 | ## Menu with Local Server related actions |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
638 | ################################################################### |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
639 | |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
640 | self.__localServerMenu = QMenu(self.tr("Local Server")) |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
641 | self.__localServerStartMonitorAct = self.__localServerMenu.addAction( |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
642 | self.tr("Start with Monitoring"), self.__startLocalServerMonitoring |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
643 | ) |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
644 | self.__localServerMenu.addSeparator() |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
645 | self.__startLocalServerAct = self.__localServerMenu.addAction( |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
646 | self.tr("Start"), self.__startLocalServer |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
647 | ) |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
648 | self.__stopLocalServerAct = self.__localServerMenu.addAction( |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
649 | self.tr("Stop"), self.__stopLocalServer |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
650 | ) |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
651 | |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
652 | ################################################################### |
6
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
653 | ## Main menu |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
654 | ################################################################### |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
655 | |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
656 | self.__ollamaMenu = QMenu() |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
657 | self.__ollamaMenu.addMenu(self.__chatHistoryMenu) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
658 | self.__ollamaMenu.addSeparator() |
8
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
659 | self.__ollamaMenu.addMenu(self.__modelMenu) |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
660 | self.__ollamaMenu.addSeparator() |
7
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
661 | self.__ollamaMenu.addMenu(self.__localServerMenu) |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
662 | self.__ollamaMenu.addSeparator() |
6
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
663 | self.__ollamaMenu.addAction(self.tr("Configure..."), self.__ollamaConfigure) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
664 | |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
665 | self.__ollamaMenu.aboutToShow.connect(self.__aboutToShowOllamaMenu) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
666 | |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
667 | self.ollamaMenuButton.setMenu(self.__ollamaMenu) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
668 | |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
669 | @pyqtSlot() |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
670 | def __aboutToShowOllamaMenu(self): |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
671 | """ |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
672 | Private slot to set the action enabled status. |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
673 | """ |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
674 | self.__clearHistoriesAct.setEnabled(self.__chatHistoryLayout.count() > 1) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
675 | |
7
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
676 | self.__localServerStartMonitorAct.setEnabled( |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
677 | self.__localServerProcess is None and self.__localServerDialog is None |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
678 | ) |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
679 | self.__startLocalServerAct.setEnabled( |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
680 | self.__localServerProcess is None and self.__localServerDialog is None |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
681 | ) |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
682 | self.__stopLocalServerAct.setEnabled( |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
683 | self.__localServerProcess is not None and self.__localServerDialog is None |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
684 | ) |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
685 | |
6
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
686 | @pyqtSlot() |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
687 | def __ollamaConfigure(self): |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
688 | """ |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
689 | Private slot to show the ollama configuration page. |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
690 | """ |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
691 | ericApp().getObject("UserInterface").showPreferences("ollamaPage") |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
692 | |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
693 | @pyqtSlot() |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
694 | def __menuClearAllHistories(self): |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
695 | """ |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
696 | Private slot to clear all chat history entries. |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
697 | """ |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
698 | yes = EricMessageBox.yesNo( |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
699 | self, |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
700 | self.tr("Clear All Chat Histories"), |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
701 | self.tr( |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
702 | "<p>Do you really want to delete all chat histories? This is" |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
703 | " <b>irreversible</b>.</p>" |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
704 | ), |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
705 | ) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
706 | if yes: |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
707 | self.clearHistory() |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
708 | |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
709 | @pyqtSlot() |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
710 | def __menuImportHistories(self): |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
711 | """ |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
712 | Private slot to import chat history entries from a file. |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
713 | """ |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
714 | historyFile = EricFileDialog.getOpenFileName( |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
715 | self, |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
716 | self.tr("Import Chat History"), |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
717 | "", |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
718 | self.tr("Chat History Files (*.json);;All Files (*)"), |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
719 | self.tr("Chat History Files (*.json)"), |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
720 | ) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
721 | if historyFile: |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
722 | self.__loadChatHistoriesFile(historyFile, reportDuplicates=True) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
723 | |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
724 | @pyqtSlot() |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
725 | def __menuExportHistories(self): |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
726 | """ |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
727 | Private slot to export chat history entries to a file. |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
728 | """ |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
729 | entries = [] |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
730 | for index in range(self.__chatHistoryLayout.count() - 1): |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
731 | item = self.__chatHistoryLayout.itemAt(index) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
732 | widget = item.widget() |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
733 | hid = widget.getId() |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
734 | title = widget.getTitle() |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
735 | entries.append((title, hid)) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
736 | |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
737 | dlg = EricListSelectionDialog( |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
738 | entries, |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
739 | title=self.tr("Export Chat History"), |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
740 | message=self.tr("Select the chats to be exported:"), |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
741 | checkBoxSelection=True, |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
742 | showSelectAll=True, |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
743 | ) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
744 | if dlg.exec() == QDialog.DialogCode.Accepted: |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
745 | selectedChats = dlg.getSelection() |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
746 | |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
747 | fileName = EricFileDialog.getSaveFileName( |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
748 | self, |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
749 | self.tr("Export Chat History"), |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
750 | "", |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
751 | self.tr("Chat History Files (*.json)"), |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
752 | None, |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
753 | EricFileDialog.DontConfirmOverwrite, |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
754 | ) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
755 | if fileName: |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
756 | if not fileName.endswith(".json"): |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
757 | fileName += ".json" |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
758 | |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
759 | entries = {} |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
760 | for _, hid in selectedChats: |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
761 | historyWidget = self.__findHistoryWidget(hid) |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
762 | if historyWidget is not None: |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
763 | entries[hid] = historyWidget.saveToJson() |
d8064fb63eac
Completed the chat history functions and menu entries and refactored the history file loading and saving functions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
5
diff
changeset
|
764 | self.__saveChatHistoryFile(fileName, entries) |
7
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
765 | |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
766 | def prepareServerRuntimeEnvironment(self): |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
767 | """ |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
768 | Public method to prepare a QProcessEnvironment object. |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
769 | |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
770 | @return prepared environment object to be used with QProcess |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
771 | @rtype QProcessEnvironment |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
772 | """ |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
773 | env = QProcessEnvironment.systemEnvironment() |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
774 | env.insert( |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
775 | "OLLAMA_HOST", |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
776 | "127.0.0.1:{0}".format(self.__plugin.getPreferences("OllamaLocalPort")), |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
777 | ) |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
778 | |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
779 | return env |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
780 | |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
781 | @pyqtSlot() |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
782 | def __startLocalServerMonitoring(self): |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
783 | """ |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
784 | Private slot to open a dialog for running a local 'ollama' server instance |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
785 | and monitor its output. |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
786 | """ |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
787 | from .RunOllamaServerDialog import RunOllamaServerDialog |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
788 | |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
789 | self.__localServerDialog = RunOllamaServerDialog( |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
790 | self.__client, self.__plugin, self |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
791 | ) |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
792 | self.__localServerDialog.serverStarted.connect(self.__serverStarted) |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
793 | self.__localServerDialog.serverStopped.connect(self.__serverStopped) |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
794 | self.__localServerDialog.finished.connect(self.__serverDialogClosed) |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
795 | self.__localServerDialog.show() |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
796 | self.__localServerDialog.startServer() |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
797 | |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
798 | @pyqtSlot() |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
799 | def __serverStarted(self): |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
800 | """ |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
801 | Private slot to handle the start of a local server. |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
802 | """ |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
803 | self.__client.setMode(True) |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
804 | self.on_reloadModelsButton_clicked() |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
805 | |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
806 | @pyqtSlot() |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
807 | def __serverStopped(self): |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
808 | """ |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
809 | Private slot to handle the stopping of a local server. |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
810 | """ |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
811 | self.__client.setMode(False) |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
812 | self.on_reloadModelsButton_clicked() |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
813 | |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
814 | @pyqtSlot() |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
815 | def __serverDialogClosed(self): |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
816 | """ |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
817 | Private slot handling the closing of the local server dialog. |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
818 | """ |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
819 | self.__localServerDialog.deleteLater() |
eb1dec15b2f0
Implemented a menu, dialog and actions to start a local 'ollama' server with the capability to monitor its log output.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
6
diff
changeset
|
820 | self.__localServerDialog = None |
8
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
821 | |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
822 | @pyqtSlot() |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
823 | def __startLocalServer(self): |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
824 | """ |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
825 | Private slot to start a local 'ollama' server instance in the background. |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
826 | """ |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
827 | env = self.prepareServerRuntimeEnvironment() |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
828 | self.__localServerProcess = QProcess() |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
829 | self.__localServerProcess.setProcessEnvironment(env) |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
830 | self.__localServerProcess.finished.connect(self.__localServerProcessFinished) |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
831 | |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
832 | command = "ollama" |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
833 | args = ["serve"] |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
834 | |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
835 | self.__localServerProcess.start(command, args) |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
836 | ok = self.__localServerProcess.waitForStarted(10000) |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
837 | if not ok: |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
838 | EricMessageBox.critical( |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
839 | None, |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
840 | self.tr("Run Local 'ollama' Server"), |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
841 | self.tr("""The loacl 'ollama' server process could not be started."""), |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
842 | ) |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
843 | self.__localServerProcess = None |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
844 | else: |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
845 | self.__serverStarted() |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
846 | |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
847 | @pyqtSlot() |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
848 | def __stopLocalServer(self): |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
849 | """ |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
850 | Private slot to stop a running local 'ollama' server instance. |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
851 | """ |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
852 | if self.__localServerProcess is not None: |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
853 | self.__localServerProcess.terminate() |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
854 | |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
855 | @pyqtSlot() |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
856 | def __localServerProcessFinished(self): |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
857 | """ |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
858 | Private slot handling the finishing of the local 'ollama' server process. |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
859 | """ |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
860 | if ( |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
861 | self.__localServerProcess is not None |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
862 | and self.__localServerProcess.state() != QProcess.ProcessState.NotRunning |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
863 | ): |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
864 | self.__localServerProcess.terminate() |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
865 | QTimer.singleShot(2000, self.__localServerProcess.kill) |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
866 | self.__localServerProcess.waitForFinished(3000) |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
867 | |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
868 | self.__localServerProcess = None |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
869 | |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
870 | self.__serverStopped() |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
871 | |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
872 | @pyqtSlot() |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
873 | def __showModels(self): |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
874 | """ |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
875 | Private slot to ask the 'ollama' server for a list of available models with |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
876 | some details. |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
877 | """ |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
878 | from .OllamaDetailedModelsDialog import OllamaDetailedModelsDialog |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
879 | |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
880 | models = self.__client.listDetails() |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
881 | if models: |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
882 | dlg = OllamaDetailedModelsDialog(models, self) |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
883 | dlg.exec() |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
884 | else: |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
885 | EricMessageBox.information( |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
886 | self, |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
887 | self.tr("List Models"), |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
888 | self.tr("There are no models available."), |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
889 | ) |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
890 | |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
891 | @pyqtSlot() |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
892 | def __showRunningModels(self): |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
893 | """ |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
894 | Private slot to show a dialog with data of the running models. |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
895 | """ |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
896 | from .OllamaRunningModelsDialog import OllamaRunningModelsDialog |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
897 | |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
898 | models = self.__client.listRunning() |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
899 | if models: |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
900 | dlg = OllamaRunningModelsDialog(models, self) |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
901 | dlg.exec() |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
902 | else: |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
903 | EricMessageBox.information( |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
904 | self, |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
905 | self.tr("List Running Models"), |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
906 | self.tr("There are no models running."), |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
907 | ) |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
908 | |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
909 | @pyqtSlot() |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
910 | def __showModelLibrary(self): |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
911 | """ |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
912 | Private slot to open the 'ollama' model librayr web site. |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
913 | """ |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
914 | urlStr = self.__plugin.getPreferences("OllamaModelLibraryUrl") |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
915 | url = QUrl.fromUserInput(urlStr) |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
916 | QDesktopServices.openUrl(url) |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
917 | |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
918 | @pyqtSlot() |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
919 | def __pullModel(self): |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
920 | """ |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
921 | Private slot to download a model from the 'ollama' model library. |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
922 | """ |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
923 | # TODO: not implemented yet |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
924 | pass |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
925 | |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
926 | @pyqtSlot() |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
927 | def __removeModel(self): |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
928 | """ |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
929 | Private slot to remove a model from the 'ollama' server. |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
930 | """ |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
931 | # TODO: not implemented yet |
3118d16e526e
Implemented some more menu actions.
Detlev Offenbach <detlev@die-offenbachs.de>
parents:
7
diff
changeset
|
932 | pass |