diff --git a/routes/llmTest.js b/routes/llmTest.js new file mode 100644 index 0000000..70186b6 --- /dev/null +++ b/routes/llmTest.js @@ -0,0 +1,219 @@ +import express from 'express'; +import { asyncHandler } from '../middleware/errorHandler.js'; +import { requireAuth, requireAdmin } from '../middleware/auth.js'; + +const router = express.Router(); + +/** + * GET /api/llm-test/models + * 列出可用的 LLM 模型(從 API 動態獲取) + */ +router.get('/models', requireAuth, asyncHandler(async (req, res) => { + const { api_url } = req.query; + const targetUrl = api_url || 'https://ollama_pjapi.theaken.com'; + + try { + const axios = (await import('axios')).default; + + const response = await axios.get(`${targetUrl}/v1/models`, { + timeout: 10000 + }); + + if (response.data && response.data.data) { + const models = response.data.data.map(model => ({ + id: model.id, + name: model.info?.name || model.id, + description: model.info?.description || '', + best_for: model.info?.best_for || '', + owned_by: model.owned_by || 'unknown' + })); + + res.json({ + success: true, + data: models + }); + } else { + throw new Error('Invalid response format'); + } + } catch (error) { + res.status(500).json({ + success: false, + error: '無法取得模型列表', + message: error.message + }); + } +})); + +/** + * POST /api/llm-test/quick + * 快速測試 LLM 連線(僅管理員) + */ +router.post('/quick', requireAdmin, asyncHandler(async (req, res) => { + const { api_url, api_key, model_name } = req.body; + + if (!api_url || !model_name) { + return res.status(400).json({ + success: false, + error: '請提供 API 端點和模型名稱' + }); + } + + try { + const axios = (await import('axios')).default; + const startTime = Date.now(); + + const response = await axios.post( + `${api_url}/v1/chat/completions`, + { + model: model_name, + messages: [ + { role: 'user', content: 'Hello, please respond with just "OK"' } + ], + max_tokens: 10 + }, + { + timeout: 15000, + headers: { + 'Content-Type': 'application/json', + ...(api_key && { 'Authorization': `Bearer ${api_key}` }) + } + } + ); + + const responseTime = Date.now() - startTime; + + if (response.data && response.data.choices) { + res.json({ + success: true, + message: 'LLM API 連線測試成功', + responseTime: responseTime, + response: response.data.choices[0]?.message?.content || '' + }); + } else { + throw new Error('Invalid API response format'); + } + } catch (error) { + res.status(500).json({ + success: false, + error: 'LLM API 連線測試失敗', + message: error.message + }); + } +})); + +/** + * POST /api/llm-test/chat + * 直接與 LLM 對話(測試用,僅管理員) + */ +router.post('/chat', requireAdmin, asyncHandler(async (req, res) => { + const { api_url, api_key, model_name, messages, temperature, max_tokens, stream } = req.body; + + if (!api_url || !model_name || !messages) { + return res.status(400).json({ + success: false, + error: '請提供必要參數' + }); + } + + try { + const axios = (await import('axios')).default; + const startTime = Date.now(); + + // 非串流模式 + if (!stream) { + const response = await axios.post( + `${api_url}/v1/chat/completions`, + { + model: model_name, + messages: messages, + temperature: temperature || 0.7, + max_tokens: max_tokens || 2000, + stream: false + }, + { + timeout: 120000, + headers: { + 'Content-Type': 'application/json', + ...(api_key && { 'Authorization': `Bearer ${api_key}` }) + } + } + ); + + const responseTime = Date.now() - startTime; + + res.json({ + success: true, + data: { + content: response.data.choices[0]?.message?.content || '', + usage: response.data.usage, + model: response.data.model, + responseTime: responseTime + } + }); + } else { + // 串流模式 - 使用 Server-Sent Events + res.setHeader('Content-Type', 'text/event-stream'); + res.setHeader('Cache-Control', 'no-cache'); + res.setHeader('Connection', 'keep-alive'); + + const response = await axios.post( + `${api_url}/v1/chat/completions`, + { + model: model_name, + messages: messages, + temperature: temperature || 0.7, + max_tokens: max_tokens || 2000, + stream: true + }, + { + timeout: 120000, + responseType: 'stream', + headers: { + 'Content-Type': 'application/json', + ...(api_key && { 'Authorization': `Bearer ${api_key}` }) + } + } + ); + + response.data.on('data', (chunk) => { + const lines = chunk.toString().split('\n'); + for (const line of lines) { + if (line.startsWith('data: ')) { + const dataStr = line.slice(6).trim(); + if (dataStr && dataStr !== '[DONE]') { + try { + const data = JSON.parse(dataStr); + const content = data.choices?.[0]?.delta?.content; + if (content) { + res.write(`data: ${JSON.stringify({ content })}\n\n`); + } + } catch (e) { + // Skip invalid JSON + } + } + } + } + }); + + response.data.on('end', () => { + res.write('data: [DONE]\n\n'); + res.end(); + }); + + response.data.on('error', (error) => { + res.write(`data: ${JSON.stringify({ error: error.message })}\n\n`); + res.end(); + }); + } + } catch (error) { + if (!res.headersSent) { + res.status(500).json({ + success: false, + error: 'LLM 對話失敗', + message: error.message + }); + } + } +})); + +export default router; diff --git a/scripts/fix-indexes.js b/scripts/fix-indexes.js new file mode 100644 index 0000000..5d4728b --- /dev/null +++ b/scripts/fix-indexes.js @@ -0,0 +1,204 @@ +#!/usr/bin/env node +/** + * Fix Database Indexes + * Ensures all indexes exist on renamed 5Why_ tables + * + * Run: node scripts/fix-indexes.js + */ + +import mysql from 'mysql2/promise'; +import dotenv from 'dotenv'; + +dotenv.config(); + +const dbConfig = { + host: process.env.DB_HOST || 'mysql.theaken.com', + port: parseInt(process.env.DB_PORT) || 33306, + user: process.env.DB_USER || 'A102', + password: process.env.DB_PASSWORD || 'Bb123456', + database: process.env.DB_NAME || 'db_A102' +}; + +// Index definitions for each table +const indexDefinitions = { + '5Why_users': [ + { name: 'idx_employee_id', columns: 'employee_id' }, + { name: 'idx_email', columns: 'email' }, + { name: 'idx_role', columns: 'role' } + ], + '5Why_analyses': [ + { name: 'idx_user_id', columns: 'user_id' }, + { name: 'idx_status', columns: 'status' }, + { name: 'idx_created_at', columns: 'created_at' } + ], + '5Why_analysis_perspectives': [ + { name: 'idx_analysis_id', columns: 'analysis_id' } + ], + '5Why_analysis_whys': [ + { name: 'idx_perspective_id', columns: 'perspective_id' }, + { name: 'idx_level', columns: 'level' } + ], + '5Why_llm_configs': [ + { name: 'idx_provider', columns: 'provider' }, + { name: 'idx_is_active', columns: 'is_active' } + ], + '5Why_system_settings': [ + { name: 'idx_setting_key', columns: 'setting_key' }, + { name: 'idx_is_public', columns: 'is_public' } + ], + '5Why_audit_logs': [ + { name: 'idx_user_id', columns: 'user_id' }, + { name: 'idx_action', columns: 'action' }, + { name: 'idx_created_at', columns: 'created_at' }, + { name: 'idx_entity', columns: 'entity_type, entity_id' } + ], + '5Why_sessions': [ + { name: 'idx_expires', columns: 'expires' } + ] +}; + +async function fixIndexes() { + let connection; + + try { + console.log('\n╔════════════════════════════════════════════════════════╗'); + console.log('║ 5 Why Analyzer - Fix Database Indexes ║'); + console.log('╚════════════════════════════════════════════════════════╝\n'); + + console.log('🔄 Connecting to database...'); + connection = await mysql.createConnection(dbConfig); + console.log('✅ Connected successfully\n'); + + // Process each table + for (const [tableName, indexes] of Object.entries(indexDefinitions)) { + console.log(`📋 Processing table: ${tableName}`); + + // Check if table exists + const [tableExists] = await connection.execute( + `SELECT COUNT(*) as count FROM information_schema.tables + WHERE table_schema = ? AND table_name = ?`, + [dbConfig.database, tableName] + ); + + if (tableExists[0].count === 0) { + console.log(` ⚠ Table ${tableName} not found, skipping...`); + continue; + } + + // Get existing indexes + const [existingIndexes] = await connection.execute( + `SELECT DISTINCT INDEX_NAME FROM information_schema.statistics + WHERE table_schema = ? AND table_name = ?`, + [dbConfig.database, tableName] + ); + const existingIndexNames = existingIndexes.map(idx => idx.INDEX_NAME); + + // Create missing indexes + for (const index of indexes) { + if (existingIndexNames.includes(index.name)) { + console.log(` ✓ Index ${index.name} already exists`); + } else { + try { + await connection.execute( + `CREATE INDEX ${index.name} ON \`${tableName}\` (${index.columns})` + ); + console.log(` ✓ Created index: ${index.name} on (${index.columns})`); + } catch (err) { + if (err.code === 'ER_DUP_KEYNAME') { + console.log(` ⚠ Index ${index.name} already exists (duplicate)`); + } else { + console.error(` ✗ Error creating index ${index.name}: ${err.message}`); + } + } + } + } + } + + // Verify unique constraints on 5Why_users + console.log('\n📋 Verifying unique constraints on 5Why_users...'); + + // Check employee_id unique constraint + const [empUnique] = await connection.execute( + `SELECT COUNT(*) as count FROM information_schema.statistics + WHERE table_schema = ? AND table_name = '5Why_users' + AND column_name = 'employee_id' AND non_unique = 0`, + [dbConfig.database] + ); + + if (empUnique[0].count > 0) { + console.log(' ✓ employee_id has unique constraint'); + } else { + console.log(' ⚠ employee_id missing unique constraint, adding...'); + try { + await connection.execute( + `ALTER TABLE 5Why_users ADD UNIQUE INDEX idx_employee_id_unique (employee_id)` + ); + console.log(' ✓ Added unique constraint on employee_id'); + } catch (err) { + if (err.code === 'ER_DUP_KEYNAME' || err.message.includes('Duplicate')) { + console.log(' ✓ Unique constraint already exists'); + } else { + console.error(` ✗ Error: ${err.message}`); + } + } + } + + // Check email unique constraint + const [emailUnique] = await connection.execute( + `SELECT COUNT(*) as count FROM information_schema.statistics + WHERE table_schema = ? AND table_name = '5Why_users' + AND column_name = 'email' AND non_unique = 0`, + [dbConfig.database] + ); + + if (emailUnique[0].count > 0) { + console.log(' ✓ email has unique constraint'); + } else { + console.log(' ⚠ email missing unique constraint, adding...'); + try { + await connection.execute( + `ALTER TABLE 5Why_users ADD UNIQUE INDEX idx_email_unique (email)` + ); + console.log(' ✓ Added unique constraint on email'); + } catch (err) { + if (err.code === 'ER_DUP_KEYNAME' || err.message.includes('Duplicate')) { + console.log(' ✓ Unique constraint already exists'); + } else { + console.error(` ✗ Error: ${err.message}`); + } + } + } + + console.log('\n╔════════════════════════════════════════════════════════╗'); + console.log('║ Index Fix Complete! ║'); + console.log('╚════════════════════════════════════════════════════════╝\n'); + + // Show current indexes for 5Why_users + console.log('📊 Current indexes on 5Why_users:'); + const [userIndexes] = await connection.execute( + `SELECT INDEX_NAME, COLUMN_NAME, NON_UNIQUE + FROM information_schema.statistics + WHERE table_schema = ? AND table_name = '5Why_users' + ORDER BY INDEX_NAME, SEQ_IN_INDEX`, + [dbConfig.database] + ); + + userIndexes.forEach(idx => { + const unique = idx.NON_UNIQUE === 0 ? ' (UNIQUE)' : ''; + console.log(` - ${idx.INDEX_NAME}: ${idx.COLUMN_NAME}${unique}`); + }); + + } catch (error) { + console.error('\n❌ Fix indexes failed:'); + console.error(' Error:', error.message); + process.exit(1); + } finally { + if (connection) { + await connection.end(); + console.log('\n🔌 Database connection closed\n'); + } + } +} + +// Execute +fixIndexes(); diff --git a/vite.config.js b/vite.config.js index 3922762..a8d2087 100644 --- a/vite.config.js +++ b/vite.config.js @@ -4,6 +4,7 @@ import react from '@vitejs/plugin-react' export default defineConfig({ plugins: [react()], server: { + host: 'localhost', port: 5173, open: true }