79 lines
3.0 KiB
Python
79 lines
3.0 KiB
Python
import sys
|
||
import requests
|
||
import json
|
||
from PySide6.QtWidgets import QApplication, QMainWindow, QListWidget, QVBoxLayout, QWidget, QLabel, QPushButton
|
||
from PySide6.QtCore import Qt
|
||
from loguru import logger
|
||
|
||
class OllamaModelViewer(QMainWindow):
|
||
def __init__(self):
|
||
super().__init__()
|
||
self.setWindowTitle("Ollama 模型查看器")
|
||
self.setGeometry(100, 100, 600, 400)
|
||
|
||
# 创建主窗口部件
|
||
self.central_widget = QWidget()
|
||
self.setCentralWidget(self.central_widget)
|
||
|
||
# 创建布局
|
||
self.layout = QVBoxLayout()
|
||
self.central_widget.setLayout(self.layout)
|
||
|
||
# 创建标题标签
|
||
self.title_label = QLabel("当前安装的Ollama模型:")
|
||
self.title_label.setStyleSheet("font-weight: bold; font-size: 14px;")
|
||
self.layout.addWidget(self.title_label)
|
||
|
||
# 创建列表部件
|
||
self.model_list = QListWidget()
|
||
self.model_list.setStyleSheet("font-family: monospace;")
|
||
self.layout.addWidget(self.model_list)
|
||
|
||
# 创建刷新按钮
|
||
self.refresh_button = QPushButton("刷新模型列表")
|
||
self.refresh_button.clicked.connect(self.fetch_models)
|
||
self.layout.addWidget(self.refresh_button)
|
||
|
||
# 初始加载模型
|
||
self.fetch_models()
|
||
|
||
def fetch_models(self):
|
||
"""从Ollama API获取模型列表"""
|
||
self.model_list.clear()
|
||
|
||
try:
|
||
logger.info("正在获取Ollama模型列表...")
|
||
response = requests.get("http://localhost:11434/api/tags", timeout=5)
|
||
|
||
if response.status_code == 200:
|
||
data = response.json()
|
||
models = data.get("models", [])
|
||
|
||
if models:
|
||
for model in models:
|
||
model_name = model.get("model", "")
|
||
if model_name:
|
||
self.model_list.addItem(model_name)
|
||
logger.info(f"找到模型: {model_name}")
|
||
else:
|
||
self.model_list.addItem("未找到任何模型")
|
||
logger.info("未找到任何模型")
|
||
else:
|
||
self.model_list.addItem(f"API请求失败，状态码: {response.status_code}")
|
||
logger.error(f"API请求失败，状态码: {response.status_code}")
|
||
|
||
except requests.exceptions.RequestException as e:
|
||
self.model_list.addItem("无法连接到Ollama API")
|
||
logger.error(f"无法连接到Ollama API: {str(e)}")
|
||
except json.JSONDecodeError as e:
|
||
self.model_list.addItem("API响应格式错误")
|
||
logger.error(f"API响应格式错误: {str(e)}")
|
||
except Exception as e:
|
||
self.model_list.addItem(f"发生错误: {str(e)}")
|
||
logger.error(f"发生未知错误: {str(e)}")
|
||
|
||
if __name__ == "__main__":
|
||
app = QApplication(sys.argv)
|
||
window = OllamaModelViewer()
|
||
window.show()
|
||
sys.exit(app.exec()) |