thzxx
9735a1ed38
补日志的模块： - vector-store.ts: 0→7（init/create/delete/add/search 全覆盖） - rag.ts: 3→7（initVectorStore/embedBatch/removeDoc/retrieveContext） - chat-area.ts: 2→5（3个导出函数加日志） - history-modal.ts: 4→5（打开历史记录） - kb-modal.ts: 7→8（打开知识库管理） - header.ts: 5→7（运行中模型状态）
95 lines
3.3 KiB
TypeScript
95 lines
3.3 KiB
TypeScript
/**
|
|
* Header - 顶部导航组件
|
|
*/
|
|
|
|
import { state, KEYS } from '../state/state.js';
|
|
import { formatSize } from '../utils/utils.js';
|
|
import { OllamaAPI } from '../api/ollama.js';
|
|
import { logConnection, logError, logDebug } from '../services/log-service.js';
|
|
|
|
let connStatusEl: HTMLElement;
|
|
|
|
export function initHeader(): void {
|
|
connStatusEl = document.querySelector('#connStatus')!;
|
|
}
|
|
|
|
export async function checkConnection(): Promise<{ ok: boolean; version?: { version: string }; error?: Error }> {
|
|
const api = state.get<OllamaAPI>(KEYS.API);
|
|
if (!api) return { ok: false, error: new Error('API 未初始化') };
|
|
|
|
connStatusEl.className = 'conn-status pending';
|
|
connStatusEl.querySelector('.status-label')!.textContent = '连接中...';
|
|
|
|
try {
|
|
const version = await api.getVersion();
|
|
connStatusEl.className = 'conn-status connected';
|
|
connStatusEl.title = `已连接 (v${version.version})`;
|
|
connStatusEl.querySelector('.status-label')!.textContent = '已连接';
|
|
logConnection('connected', `Ollama v${version.version}`);
|
|
return { ok: true, version };
|
|
} catch (err) {
|
|
connStatusEl.className = 'conn-status disconnected';
|
|
connStatusEl.title = '未连接';
|
|
connStatusEl.querySelector('.status-label')!.textContent = '未连接';
|
|
logConnection('disconnected', (err as Error).message);
|
|
return { ok: false, error: err as Error };
|
|
}
|
|
}
|
|
|
|
export async function updateConnectionInfo(): Promise<void> {
|
|
const api = state.get<OllamaAPI>(KEYS.API);
|
|
if (!api) return;
|
|
|
|
const badge = document.querySelector('#connBadge')!;
|
|
const verEl = document.querySelector('#serverVersion')!;
|
|
const corsHint = document.querySelector('#corsHint') as HTMLElement;
|
|
|
|
badge.textContent = '检测中...';
|
|
badge.className = 'status-badge warning';
|
|
verEl.textContent = '';
|
|
|
|
try {
|
|
const version = await api.getVersion();
|
|
badge.textContent = '✓ 已连接';
|
|
badge.className = 'status-badge success';
|
|
verEl.textContent = `v${version.version}`;
|
|
corsHint.style.display = 'none';
|
|
logConnection('info', `Ollama v${version.version}`);
|
|
} catch (err) {
|
|
badge.textContent = '✗ 未连接';
|
|
badge.className = 'status-badge error';
|
|
verEl.textContent = '';
|
|
logError('连接检测失败', (err as Error).message);
|
|
|
|
if ((err as Error).message.includes('Failed to fetch') || (err as Error).message.includes('NetworkError') || (err as Error).message.includes('CORS')) {
|
|
corsHint.style.display = '';
|
|
} else {
|
|
corsHint.style.display = 'none';
|
|
}
|
|
}
|
|
}
|
|
|
|
export async function updateRunningModels(): Promise<void> {
|
|
const api = state.get<OllamaAPI>(KEYS.API);
|
|
if (!api) return;
|
|
|
|
const container = document.querySelector('#runningModels')!;
|
|
try {
|
|
const data = await api.psModels();
|
|
if (!data.models || data.models.length === 0) {
|
|
container.innerHTML = '<p class="text-muted">无运行中的模型</p>';
|
|
logDebug('运行中模型: 0');
|
|
return;
|
|
}
|
|
container.innerHTML = data.models.map(m => `
|
|
<div class="running-model">
|
|
<span class="model-name">${m.name}</span>
|
|
<span class="model-vram">${formatSize(m.size_vram || 0)} VRAM</span>
|
|
</div>
|
|
`).join('');
|
|
logDebug(`运行中模型: ${data.models.length}`, data.models.map(m => m.name).join(', '));
|
|
} catch {
|
|
container.innerHTML = '<p class="text-muted">无法获取运行信息</p>';
|
|
}
|
|
}
|