124 lines
3.6 KiB
JavaScript
124 lines
3.6 KiB
JavaScript
/**
|
|
* Copyright (c) 2024-2025 Тарабанов Александр Викторович
|
|
* All rights reserved.
|
|
*
|
|
* This software is proprietary and confidential.
|
|
* Unauthorized copying, modification, or distribution is prohibited.
|
|
*
|
|
* For licensing inquiries: info@hb3-accelerator.com
|
|
* Website: https://hb3-accelerator.com
|
|
* GitHub: https://github.com/HB3-ACCELERATOR
|
|
*/
|
|
|
|
const express = require('express');
|
|
const router = express.Router();
|
|
const axios = require('axios');
|
|
const db = require('../db');
|
|
const aiAssistant = require('../services/ai-assistant');
|
|
const aiCache = require('../services/ai-cache');
|
|
const aiQueue = require('../services/ai-queue');
|
|
const logger = require('../utils/logger');
|
|
|
|
router.get('/', async (req, res) => {
|
|
const results = {};
|
|
// Backend
|
|
results.backend = { status: 'ok' };
|
|
|
|
// Vector Search
|
|
try {
|
|
const baseUrl = process.env.VECTOR_SEARCH_URL || 'http://vector-search:8001';
|
|
const healthUrl = baseUrl.replace(/\/$/, '') + '/health';
|
|
const vs = await axios.get(healthUrl, { timeout: 2000 });
|
|
results.vectorSearch = { status: 'ok', ...vs.data };
|
|
} catch (e) {
|
|
console.log('Vector Search error:', e.message, 'Status:', e.response?.status);
|
|
results.vectorSearch = { status: 'error', error: e.message };
|
|
}
|
|
|
|
// Ollama
|
|
try {
|
|
const ollama = await axios.get(process.env.OLLAMA_BASE_URL ? process.env.OLLAMA_BASE_URL + '/api/tags' : 'http://ollama:11434/api/tags', { timeout: 2000 });
|
|
results.ollama = { status: 'ok', models: ollama.data.models?.length || 0 };
|
|
} catch (e) {
|
|
results.ollama = { status: 'error', error: e.message };
|
|
}
|
|
|
|
// PostgreSQL
|
|
try {
|
|
await db.query('SELECT 1');
|
|
results.postgres = { status: 'ok' };
|
|
} catch (e) {
|
|
results.postgres = { status: 'error', error: e.message };
|
|
}
|
|
|
|
res.json({ status: 'ok', services: results, timestamp: new Date().toISOString() });
|
|
});
|
|
|
|
// GET /api/monitoring/ai-stats - статистика AI
|
|
router.get('/ai-stats', async (req, res) => {
|
|
try {
|
|
const aiHealth = await aiAssistant.checkHealth();
|
|
const cacheStats = aiCache.getStats();
|
|
const queueStats = aiQueue.getStats();
|
|
|
|
res.json({
|
|
status: 'ok',
|
|
timestamp: new Date().toISOString(),
|
|
ai: {
|
|
health: aiHealth,
|
|
model: process.env.OLLAMA_MODEL || 'qwen2.5:7b',
|
|
baseUrl: process.env.OLLAMA_BASE_URL || 'http://localhost:11434'
|
|
},
|
|
cache: {
|
|
...cacheStats,
|
|
hitRate: `${(cacheStats.hitRate * 100).toFixed(1)}%`
|
|
},
|
|
queue: {
|
|
...queueStats,
|
|
avgResponseTime: `${queueStats.avgResponseTime.toFixed(0)}ms`
|
|
}
|
|
});
|
|
} catch (error) {
|
|
logger.error('Error getting AI stats:', error);
|
|
res.status(500).json({
|
|
status: 'error',
|
|
error: error.message
|
|
});
|
|
}
|
|
});
|
|
|
|
// POST /api/monitoring/ai-cache/clear - очистка кэша
|
|
router.post('/ai-cache/clear', async (req, res) => {
|
|
try {
|
|
aiCache.clear();
|
|
res.json({
|
|
status: 'ok',
|
|
message: 'AI cache cleared successfully'
|
|
});
|
|
} catch (error) {
|
|
logger.error('Error clearing AI cache:', error);
|
|
res.status(500).json({
|
|
status: 'error',
|
|
error: error.message
|
|
});
|
|
}
|
|
});
|
|
|
|
// POST /api/monitoring/ai-queue/clear - очистка очереди
|
|
router.post('/ai-queue/clear', async (req, res) => {
|
|
try {
|
|
aiQueue.clear();
|
|
res.json({
|
|
status: 'ok',
|
|
message: 'AI queue cleared successfully'
|
|
});
|
|
} catch (error) {
|
|
logger.error('Error clearing AI queue:', error);
|
|
res.status(500).json({
|
|
status: 'error',
|
|
error: error.message
|
|
});
|
|
}
|
|
});
|
|
|
|
module.exports = router;
|