|
1 # -*- coding: utf-8 -*- |
|
2 |
|
3 # Copyright (c) 2024 Detlev Offenbach <detlev@die-offenbachs.de> |
|
4 # |
|
5 |
|
6 """ |
|
7 Module implementing a dialog to show details of the running models. |
|
8 """ |
|
9 |
|
10 from PyQt6.QtWidgets import QDialog, QTreeWidgetItem |
|
11 |
|
12 from eric7 import Globals |
|
13 |
|
14 from .Ui_OllamaRunningModelsDialog import Ui_OllamaRunningModelsDialog |
|
15 |
|
16 |
|
17 class OllamaRunningModelsDialog(QDialog, Ui_OllamaRunningModelsDialog): |
|
18 """ |
|
19 Class implementing a dialog to show details of the running models. |
|
20 """ |
|
21 |
|
22 def __init__(self, models, parent=None): |
|
23 """ |
|
24 Constructor |
|
25 |
|
26 @param models list of available models with details |
|
27 @type list[dict[str: Any]] |
|
28 @param parent reference to the parent widget (defaults to None) |
|
29 @type QWidget (optional) |
|
30 """ |
|
31 super().__init__(parent) |
|
32 self.setupUi(self) |
|
33 |
|
34 for model in models: |
|
35 QTreeWidgetItem( |
|
36 self.modelsList, |
|
37 [ |
|
38 model["name"], |
|
39 model["id"], |
|
40 Globals.dataString(model["size"]), |
|
41 model["processor"], |
|
42 model["expires"].strftime("%Y-%m-%d %H:%M:%S"), |
|
43 ] |
|
44 ) |
|
45 |
|
46 for column in range(self.modelsList.columnCount()): |
|
47 self.modelsList.resizeColumnToContents(column) |