694 lines
30 KiB
JavaScript
694 lines
30 KiB
JavaScript
const fs = require('fs')
|
||
const path = require('path')
|
||
const http = require('http')
|
||
const https = require('https')
|
||
|
||
const ROOT = path.resolve(__dirname, '..')
|
||
const DOCS_DIR = path.join(ROOT, 'docs')
|
||
const ENV_FILE = path.join(__dirname, '.env.local')
|
||
const KNOWLEDGE_FILE = path.join(DOCS_DIR, 'wc-spc-knowledge-base.json')
|
||
|
||
function loadLocalEnv() {
|
||
if (!fs.existsSync(ENV_FILE)) return
|
||
fs.readFileSync(ENV_FILE, 'utf8').split(/\r?\n/).forEach(line => {
|
||
const text = line.trim()
|
||
if (!text || text[0] === '#') return
|
||
const index = text.indexOf('=')
|
||
if (index <= 0) return
|
||
const key = text.slice(0, index).trim()
|
||
const value = text.slice(index + 1).trim().replace(/^["']|["']$/g, '')
|
||
if (key && process.env[key] == null) process.env[key] = value
|
||
})
|
||
}
|
||
|
||
loadLocalEnv()
|
||
|
||
const PORT = Number(process.env.AI_MCP_PORT || 3100)
|
||
const DEEPSEEK_API_KEY = process.env.DEEPSEEK_API_KEY || ''
|
||
const DEEPSEEK_BASE_URL = (process.env.DEEPSEEK_BASE_URL || 'https://api.deepseek.com').replace(/\/$/, '')
|
||
const DEEPSEEK_MODEL = process.env.DEEPSEEK_MODEL || 'deepseek-v4-pro'
|
||
const MES_BACKEND_URL = process.env.MES_BACKEND_URL || 'http://127.0.0.1:10050/submit/MESCommonBase.ashx'
|
||
const SPC_BACKEND_URL = (process.env.SPC_BACKEND_URL || 'http://localhost:57966').replace(/\/$/, '')
|
||
const ALLOW_BACKEND_CALLS = process.env.AI_MCP_ALLOW_BACKEND_CALLS === 'true'
|
||
|
||
function readJson(file, fallback) {
|
||
try {
|
||
return JSON.parse(fs.readFileSync(file, 'utf8'))
|
||
} catch (error) {
|
||
return fallback
|
||
}
|
||
}
|
||
|
||
function emptyKnowledgeBase() {
|
||
return {
|
||
generatedAt: null,
|
||
modules: [],
|
||
apiMappings: [],
|
||
sqlTables: [],
|
||
qualityModules: [],
|
||
spcCharts: [],
|
||
queryHints: []
|
||
}
|
||
}
|
||
|
||
let knowledgeBase = readJson(KNOWLEDGE_FILE, emptyKnowledgeBase())
|
||
|
||
function lower(value) {
|
||
return String(value || '').toLowerCase()
|
||
}
|
||
|
||
function unique(values) {
|
||
return Array.from(new Set(values.filter(Boolean)))
|
||
}
|
||
|
||
function escapeHtml(value) {
|
||
return String(value == null ? '' : value)
|
||
.replace(/&/g, '&')
|
||
.replace(/</g, '<')
|
||
.replace(/>/g, '>')
|
||
.replace(/"/g, '"')
|
||
.replace(/'/g, ''')
|
||
}
|
||
|
||
function table(headers, rows) {
|
||
const dataRows = Array.isArray(rows) ? rows : []
|
||
let html = '<table class="ai-result-table"><thead><tr>'
|
||
headers.forEach(header => {
|
||
html += '<th>' + escapeHtml(header.label) + '</th>'
|
||
})
|
||
html += '</tr></thead><tbody>'
|
||
if (!dataRows.length) {
|
||
html += '<tr><td colspan="' + headers.length + '">无匹配数据</td></tr>'
|
||
} else {
|
||
dataRows.forEach(row => {
|
||
html += '<tr>'
|
||
headers.forEach(header => {
|
||
const value = row && row[header.key]
|
||
html += '<td>' + escapeHtml(Array.isArray(value) ? value.join(', ') : value) + '</td>'
|
||
})
|
||
html += '</tr>'
|
||
})
|
||
}
|
||
html += '</tbody></table>'
|
||
return html
|
||
}
|
||
|
||
function normalizeRows(rows, limit) {
|
||
if (!Array.isArray(rows)) return []
|
||
return rows.slice(0, limit || 30).map(row => {
|
||
if (!row || typeof row !== 'object') return { value: row }
|
||
return row
|
||
})
|
||
}
|
||
|
||
function htmlForRows(title, rows, limit) {
|
||
const normalized = normalizeRows(rows, limit || 30)
|
||
const keys = unique(normalized.flatMap(row => Object.keys(row))).slice(0, 12)
|
||
if (!keys.length) return '<h4>' + escapeHtml(title) + '</h4>' + table([{ key: 'message', label: '结果' }], [{ message: '无数据' }])
|
||
return '<h4>' + escapeHtml(title) + '</h4>' + table(keys.map(key => ({ key, label: key })), normalized)
|
||
}
|
||
|
||
function tokens(query) {
|
||
const source = lower(query)
|
||
const parts = source
|
||
.replace(/[^\w\u4e00-\u9fa5]+/g, ' ')
|
||
.split(/\s+/)
|
||
.map(item => item.trim())
|
||
.filter(item => item && item.length > 1)
|
||
const grams = []
|
||
parts.forEach(part => {
|
||
if (/[\u4e00-\u9fa5]/.test(part) && part.length > 2) {
|
||
for (let size = 2; size <= Math.min(4, part.length); size++) {
|
||
for (let index = 0; index <= part.length - size; index++) {
|
||
grams.push(part.substr(index, size))
|
||
}
|
||
}
|
||
}
|
||
})
|
||
return unique(parts.concat(grams)).filter(item => ['页面', '接口', '数据', '查询', '功能', '模块'].indexOf(item) === -1)
|
||
}
|
||
|
||
function scoreText(text, query) {
|
||
const haystack = lower(text)
|
||
const compactHaystack = haystack.replace(/\s+/g, '')
|
||
const q = lower(query)
|
||
const compactQuery = q.replace(/\s+/g, '')
|
||
if (!q) return 1
|
||
let score = compactQuery && compactHaystack.indexOf(compactQuery) > -1 ? 50 : 0
|
||
tokens(query).forEach(part => {
|
||
if (haystack.indexOf(part) > -1) score += part.length
|
||
})
|
||
return score
|
||
}
|
||
|
||
function pick(rows, query, fields, limit) {
|
||
return (rows || [])
|
||
.map(row => {
|
||
const text = fields.map(field => {
|
||
const value = row[field]
|
||
return Array.isArray(value) ? value.join(' ') : value
|
||
}).join(' ')
|
||
return { row, score: scoreText(text, query) }
|
||
})
|
||
.filter(item => item.score > 0)
|
||
.sort((a, b) => b.score - a.score)
|
||
.slice(0, limit || 20)
|
||
.map(item => item.row)
|
||
}
|
||
|
||
function routeCandidates(question, limit) {
|
||
const query = String(question || '').replace(/打开|跳转|进入|导航|前往|去到|查看|页面|模块/g, ' ')
|
||
return (knowledgeBase.modules || [])
|
||
.filter(row => row.routePath)
|
||
.map(row => {
|
||
const text = [row.module, row.title, row.view, row.routePath, row.domains, row.functionNames, row.keywords].join(' ')
|
||
let score = scoreText(text, query || question)
|
||
const compactQuestion = String(question || '').replace(/\s+/g, '').toLowerCase()
|
||
const compactPath = String(row.routePath || '').replace(/[\/_\s]+/g, '').toLowerCase()
|
||
const compactTitle = String(row.title || '').replace(/[\/_\s]+/g, '').toLowerCase()
|
||
if (compactQuestion.indexOf('spc分析') > -1 && compactPath.indexOf('searchdataspcanalysis') > -1) score += 100
|
||
if (compactQuestion.indexOf('spc') > -1 && compactTitle.indexOf('spc') > -1) score += 30
|
||
return Object.assign({ score }, row)
|
||
})
|
||
.filter(row => row.score > 0)
|
||
.sort((a, b) => b.score - a.score)
|
||
.slice(0, limit || 8)
|
||
}
|
||
|
||
function isNavigationQuestion(question) {
|
||
return /(打开|跳转|进入|导航|前往|去到|切换到)/.test(String(question || '')) || /查看.*页面/.test(String(question || ''))
|
||
}
|
||
|
||
function navigationResponse(question) {
|
||
if (!isNavigationQuestion(question)) return null
|
||
const candidates = routeCandidates(question, 8)
|
||
if (!candidates.length) {
|
||
return { html: table([{ key: '提示', label: '提示' }], [{ 提示: '未找到匹配页面，请换一个页面名或模块名。' }]), actions: [] }
|
||
}
|
||
const target = candidates[0]
|
||
return {
|
||
html: '<h4>页面导航</h4>' + table([
|
||
{ key: 'title', label: '页面' },
|
||
{ key: 'routePath', label: '路由' },
|
||
{ key: 'view', label: '源码文件' }
|
||
], [target]),
|
||
actions: [{ type: 'navigate', path: target.routePath, title: target.title, view: target.view }]
|
||
}
|
||
}
|
||
|
||
function isQualityQuestion(question) {
|
||
return /质量|SPC|巡检|条码|Andon|安灯|测量|工位|过程能力|趋势|直方|正态|排列|X-R|XR|X-S|XS/i.test(String(question || ''))
|
||
}
|
||
|
||
function listQualityModules(limit) {
|
||
return (knowledgeBase.qualityModules || []).slice(0, limit || 50).map(row => ({
|
||
title: row.Title,
|
||
routePath: row.RoutePath,
|
||
domains: row.Domains,
|
||
queryable: row.Queryable,
|
||
view: row.View
|
||
}))
|
||
}
|
||
|
||
function qualitySearchResponse(question) {
|
||
const rows = pick(knowledgeBase.qualityModules, question, ['Title', 'View', 'Domains', 'QueryNames'], 12)
|
||
return '<h4>质量管理匹配模块</h4>' + table([
|
||
{ key: 'Title', label: '页面' },
|
||
{ key: 'RoutePath', label: '路由' },
|
||
{ key: 'Domains', label: '业务域' },
|
||
{ key: 'Queryable', label: '可查询' }
|
||
], rows)
|
||
}
|
||
|
||
function spcChartType(question) {
|
||
const text = String(question || '')
|
||
const rules = [
|
||
{ type: 'xr', re: /X-R|XR|极差|均值.*极差/i },
|
||
{ type: 'xs', re: /X-S|XS|标准差|均值.*标准差/i },
|
||
{ type: 'pareto', re: /排列|帕累托|pareto/i },
|
||
{ type: 'processCapability', re: /过程能力|CPK|CP|能力/i },
|
||
{ type: 'normalDistribution', re: /正态|分布|normal/i },
|
||
{ type: 'histogram', re: /直方|频数|histogram/i },
|
||
{ type: 'sampleTrend', re: /样本趋势|样本/i },
|
||
{ type: 'basicTrend', re: /基本趋势|趋势|折线/i }
|
||
]
|
||
const match = rules.find(rule => rule.re.test(text))
|
||
return match ? match.type : ''
|
||
}
|
||
|
||
function parseDateRange(question) {
|
||
const text = String(question || '')
|
||
const explicit = text.match(/(\d{4}[-/]\d{1,2}[-/]\d{1,2}).{0,8}(\d{4}[-/]\d{1,2}[-/]\d{1,2})/)
|
||
if (explicit) return { startTime: explicit[1].replace(/\//g, '-'), endTime: explicit[2].replace(/\//g, '-') }
|
||
const today = new Date()
|
||
const end = today.toISOString().slice(0, 10)
|
||
const daysMatch = text.match(/最近\s*(\d+)\s*天/)
|
||
const days = daysMatch ? Math.min(Math.max(Number(daysMatch[1]), 1), 90) : 7
|
||
const startDate = new Date(today.getTime() - (days - 1) * 24 * 60 * 60 * 1000)
|
||
return { startTime: startDate.toISOString().slice(0, 10), endTime: end }
|
||
}
|
||
|
||
function parseQuotedOrNamed(question, names) {
|
||
const text = String(question || '')
|
||
const quoted = text.match(/[“"']([^“”"']+)[”"']/)
|
||
if (quoted) return quoted[1]
|
||
for (let i = 0; i < names.length; i++) {
|
||
const name = names[i]
|
||
const re = new RegExp(name + '\\s*[:：]?\\s*([^，,。\\s]+)')
|
||
const match = text.match(re)
|
||
if (match) return match[1]
|
||
}
|
||
return ''
|
||
}
|
||
|
||
function parseSpcParams(question) {
|
||
const dateRange = parseDateRange(question)
|
||
const sampleSize = (String(question).match(/样本容量\s*[:：]?\s*(\d+)/) || [])[1] || '5'
|
||
const sampleNumber = (String(question).match(/样本(?:个数|数量)\s*[:：]?\s*(\d+)/) || [])[1] || '20'
|
||
const usl = (String(question).match(/USL|上限|上规格限/i) || {}).index != null ? ((String(question).match(/(?:USL|上限|上规格限)\s*[:：]?\s*(-?\d+(?:\.\d+)?)/i) || [])[1] || '') : ''
|
||
const lsl = (String(question).match(/LSL|下限|下规格限/i) || {}).index != null ? ((String(question).match(/(?:LSL|下限|下规格限)\s*[:：]?\s*(-?\d+(?:\.\d+)?)/i) || [])[1] || '') : ''
|
||
return Object.assign({
|
||
opName: parseQuotedOrNamed(question, ['工位', '工位号', '工位名称']),
|
||
model: parseQuotedOrNamed(question, ['机型', '型号', '零件号']),
|
||
measureName: parseQuotedOrNamed(question, ['测量位置', '位置']),
|
||
measureContent: parseQuotedOrNamed(question, ['测量项目', '测量内容', '项目']),
|
||
sampleSize,
|
||
sampleNumber,
|
||
usl,
|
||
lsl,
|
||
pageSize: 20,
|
||
pageCurrent: 1
|
||
}, dateRange)
|
||
}
|
||
|
||
function missingSpcParams(params) {
|
||
const missing = []
|
||
if (!params.opName) missing.push('工位')
|
||
if (!params.measureName) missing.push('测量位置')
|
||
if (!params.measureContent) missing.push('测量项目')
|
||
return missing
|
||
}
|
||
|
||
function buildUrl(endpoint, type, params) {
|
||
const query = {
|
||
type,
|
||
OPNameCheck: params.opName ? 'true' : 'false',
|
||
OPName: params.opName || '',
|
||
shaftNameCheck: params.measureName ? 'true' : 'false',
|
||
shaftName: params.measureName || '',
|
||
EngineTypeIDCheck: params.model ? 'true' : 'false',
|
||
EngineTypeID: params.model || '',
|
||
itemNameCheck: params.measureContent ? 'true' : 'false',
|
||
itemName: params.measureContent || '',
|
||
startTime: params.startTime,
|
||
endTime: params.endTime,
|
||
Yangben: params.sampleSize || '5',
|
||
Num: params.sampleNumber || '20'
|
||
}
|
||
if (params.usl) query.Usl = params.usl
|
||
if (params.lsl) query.Lsl = params.lsl
|
||
if (type === 'GetDataTableByConditent') {
|
||
query.page = params.pageCurrent || 1
|
||
query.rows = params.pageSize || 20
|
||
}
|
||
const qs = Object.keys(query).map(key => encodeURIComponent(key) + '=' + encodeURIComponent(query[key])).join('&')
|
||
return SPC_BACKEND_URL + '/submit/' + endpoint + '?' + qs
|
||
}
|
||
|
||
function getJson(url) {
|
||
return new Promise((resolve, reject) => {
|
||
const target = new URL(url)
|
||
const client = target.protocol === 'https:' ? https : http
|
||
const req = client.request({
|
||
method: 'POST',
|
||
hostname: target.hostname,
|
||
port: target.port || (target.protocol === 'https:' ? 443 : 80),
|
||
path: target.pathname + target.search
|
||
}, res => {
|
||
let text = ''
|
||
res.setEncoding('utf8')
|
||
res.on('data', chunk => { text += chunk })
|
||
res.on('end', () => {
|
||
try {
|
||
resolve(JSON.parse(text))
|
||
} catch (error) {
|
||
resolve(text)
|
||
}
|
||
})
|
||
})
|
||
req.on('error', reject)
|
||
req.end()
|
||
})
|
||
}
|
||
|
||
function numericValuesFromRows(rows) {
|
||
return normalizeRows(rows, 1000).map(row => {
|
||
const keys = Object.keys(row)
|
||
const key = keys.find(item => /测量值|value|Value|数值/i.test(item)) || keys.find(item => !Number.isNaN(Number(row[item])))
|
||
return key ? Number(row[key]) : NaN
|
||
}).filter(value => !Number.isNaN(value))
|
||
}
|
||
|
||
function summaryForValues(values) {
|
||
if (!values.length) return []
|
||
const sum = values.reduce((acc, item) => acc + item, 0)
|
||
const avg = sum / values.length
|
||
const variance = values.reduce((acc, item) => acc + Math.pow(item - avg, 2), 0) / values.length
|
||
return [
|
||
{ name: '样本数', value: values.length },
|
||
{ name: '最小值', value: Math.min.apply(null, values).toFixed(4) },
|
||
{ name: '最大值', value: Math.max.apply(null, values).toFixed(4) },
|
||
{ name: '平均值', value: avg.toFixed(4) },
|
||
{ name: '标准差', value: Math.sqrt(variance).toFixed(4) }
|
||
]
|
||
}
|
||
|
||
function simpleLineChart(title, values) {
|
||
return {
|
||
id: 'chart_' + Date.now() + '_' + Math.floor(Math.random() * 10000),
|
||
title,
|
||
option: {
|
||
title: { text: title, left: 'center' },
|
||
tooltip: { trigger: 'axis' },
|
||
grid: { left: 45, right: 20, top: 55, bottom: 35 },
|
||
xAxis: { type: 'category', data: values.map((_, index) => String(index + 1)) },
|
||
yAxis: { type: 'value', scale: true },
|
||
series: [{ name: '测量值', type: 'line', smooth: true, data: values }]
|
||
}
|
||
}
|
||
}
|
||
|
||
function simpleBarChart(title, values) {
|
||
const bins = {}
|
||
values.forEach(value => {
|
||
const key = value.toFixed(2)
|
||
bins[key] = (bins[key] || 0) + 1
|
||
})
|
||
const keys = Object.keys(bins).sort((a, b) => Number(a) - Number(b))
|
||
return {
|
||
id: 'chart_' + Date.now() + '_' + Math.floor(Math.random() * 10000),
|
||
title,
|
||
option: {
|
||
title: { text: title, left: 'center' },
|
||
tooltip: { trigger: 'axis' },
|
||
grid: { left: 45, right: 20, top: 55, bottom: 45 },
|
||
xAxis: { type: 'category', data: keys },
|
||
yAxis: { type: 'value' },
|
||
series: [{ name: '频数', type: 'bar', data: keys.map(key => bins[key]) }]
|
||
}
|
||
}
|
||
}
|
||
|
||
async function querySpcChart(args) {
|
||
const chartType = args.chartType || spcChartType(args.query || '') || 'basicTrend'
|
||
const params = Object.assign(parseSpcParams(args.query || ''), args)
|
||
const missing = missingSpcParams(params)
|
||
if (missing.length) {
|
||
return {
|
||
html: '<h4>SPC 查询条件不足</h4>' + table([
|
||
{ key: 'field', label: '缺少条件' },
|
||
{ key: 'example', label: '示例' }
|
||
], missing.map(field => ({ field, example: field + '：请在问题中明确指定' }))),
|
||
charts: [],
|
||
data: { missing, params }
|
||
}
|
||
}
|
||
const chartMap = {
|
||
basicTrend: { endpoint: 'QualityData_TrendPictureBasic.ashx', type: 'GetBaseTrendData', title: '基本趋势图' },
|
||
sampleTrend: { endpoint: 'QualityData_TrendPicture.ashx', type: 'GetBaseTrendData', title: '样本趋势图' },
|
||
histogram: { endpoint: 'QualityData_Histogram.ashx', type: 'GetDataTableByConditent', title: '直方图' },
|
||
normalDistribution: { endpoint: 'QualityData_NormalDistribution.ashx', type: 'GetBaseTrendData', title: '正态分布图' },
|
||
processCapability: { endpoint: 'QualityData_NormalDistribution.ashx', type: 'GetBaseTrendData', title: '过程能力分析' },
|
||
pareto: { endpoint: 'QualityData_Pareto.ashx', type: 'GetChart', title: '排列图' },
|
||
xr: { endpoint: 'QualityData_XR.ashx', type: 'GetBaseTrendData', title: 'X-R 控制图' },
|
||
xs: { endpoint: 'QualityData_XS.ashx', type: 'GetBaseTrendData', title: 'X-S 控制图' }
|
||
}
|
||
const meta = chartMap[chartType] || chartMap.basicTrend
|
||
const url = buildUrl(meta.endpoint, meta.type, params)
|
||
let raw
|
||
try {
|
||
raw = await getJson(url)
|
||
} catch (error) {
|
||
return {
|
||
html: '<h4>SPC 接口调用失败</h4>' + table([
|
||
{ key: 'name', label: '项目' },
|
||
{ key: 'value', label: '内容' }
|
||
], [
|
||
{ name: '错误', value: error.message },
|
||
{ name: '接口', value: url }
|
||
]),
|
||
charts: [],
|
||
data: { error: error.message, params, url }
|
||
}
|
||
}
|
||
const rows = Array.isArray(raw) ? raw : (raw && Array.isArray(raw.rows) ? raw.rows : [])
|
||
const values = numericValuesFromRows(rows)
|
||
const charts = values.length ? [
|
||
/histogram|normalDistribution|processCapability|pareto/.test(chartType) ? simpleBarChart(meta.title, values) : simpleLineChart(meta.title, values)
|
||
] : []
|
||
const conditionRows = [
|
||
{ name: '图表类型', value: meta.title },
|
||
{ name: '开始时间', value: params.startTime },
|
||
{ name: '结束时间', value: params.endTime },
|
||
{ name: '工位', value: params.opName },
|
||
{ name: '测量位置', value: params.measureName },
|
||
{ name: '测量项目', value: params.measureContent },
|
||
{ name: '样本容量', value: params.sampleSize },
|
||
{ name: '样本个数', value: params.sampleNumber }
|
||
]
|
||
const summaryRows = summaryForValues(values)
|
||
const sourceRows = [{ endpoint: meta.endpoint, type: meta.type, url }]
|
||
let html = '<h4>SPC 查询条件</h4>' + table([{ key: 'name', label: '条件' }, { key: 'value', label: '值' }], conditionRows)
|
||
html += '<h4>SPC 统计摘要</h4>' + table([{ key: 'name', label: '指标' }, { key: 'value', label: '值' }], summaryRows)
|
||
html += htmlForRows('明细数据', rows, 20)
|
||
html += '<h4>数据来源</h4>' + table([{ key: 'endpoint', label: '接口' }, { key: 'type', label: '类型' }, { key: 'url', label: '调用地址' }], sourceRows)
|
||
if (!rows.length && typeof raw === 'string') {
|
||
html += '<h4>原始返回</h4><pre>' + escapeHtml(raw.slice(0, 1000)) + '</pre>'
|
||
}
|
||
return { html, charts, data: { params, raw, rows, values, url } }
|
||
}
|
||
|
||
function postJson(url, payload, headers) {
|
||
return new Promise((resolve, reject) => {
|
||
const target = new URL(url)
|
||
const client = target.protocol === 'https:' ? https : http
|
||
const body = typeof payload === 'string' ? payload : JSON.stringify(payload)
|
||
const req = client.request({
|
||
method: 'POST',
|
||
hostname: target.hostname,
|
||
port: target.port || (target.protocol === 'https:' ? 443 : 80),
|
||
path: target.pathname + target.search,
|
||
headers: Object.assign({ 'Content-Type': 'application/json', 'Content-Length': Buffer.byteLength(body) }, headers || {})
|
||
}, res => {
|
||
let text = ''
|
||
res.setEncoding('utf8')
|
||
res.on('data', chunk => { text += chunk })
|
||
res.on('end', () => {
|
||
try {
|
||
resolve(JSON.parse(text))
|
||
} catch (error) {
|
||
resolve({ raw: text, statusCode: res.statusCode })
|
||
}
|
||
})
|
||
})
|
||
req.on('error', reject)
|
||
req.write(body)
|
||
req.end()
|
||
})
|
||
}
|
||
|
||
async function askDeepSeek(question, localHtml) {
|
||
if (!DEEPSEEK_API_KEY) return localHtml
|
||
const context = {
|
||
matchedModules: routeCandidates(question, 8),
|
||
qualityModules: pick(knowledgeBase.qualityModules, question, ['Title', 'View', 'Domains', 'QueryNames'], 8),
|
||
spcCharts: pick(knowledgeBase.spcCharts, question, ['chartType', 'title', 'apiFile', 'endpoint', 'functions'], 8)
|
||
}
|
||
const system = [
|
||
'你是 WC-SPC MES-Manager_View 网页内置的 DeepSeekV4Pro AI 助手。',
|
||
'使用中文回答，优先输出 HTML 表格。',
|
||
'只能依据项目知识库和工具结果回答，不要编造数据。',
|
||
'不要输出可执行 JavaScript；图表由系统结构化渲染。'
|
||
].join('\n')
|
||
const response = await postJson(DEEPSEEK_BASE_URL + '/chat/completions', {
|
||
model: DEEPSEEK_MODEL,
|
||
messages: [
|
||
{ role: 'system', content: system },
|
||
{ role: 'user', content: '项目知识库匹配上下文：\n' + JSON.stringify(context, null, 2) },
|
||
{ role: 'user', content: question }
|
||
],
|
||
temperature: 0.2
|
||
}, { Authorization: 'Bearer ' + DEEPSEEK_API_KEY })
|
||
return response && response.choices && response.choices[0] && response.choices[0].message
|
||
? response.choices[0].message.content
|
||
: localHtml
|
||
}
|
||
|
||
function localAnswer(question) {
|
||
if (/项目|功能|模块|有哪些/.test(question)) {
|
||
return '<h4>项目功能概览</h4>' + table([
|
||
{ key: 'name', label: '项目' },
|
||
{ key: 'value', label: '数量' }
|
||
], [
|
||
{ name: '页面模块', value: knowledgeBase.modules.length },
|
||
{ name: 'API 映射', value: knowledgeBase.apiMappings.length },
|
||
{ name: '质量相关模块', value: knowledgeBase.qualityModules.length },
|
||
{ name: 'SPC 图表类型', value: knowledgeBase.spcCharts.length }
|
||
]) + '<h4>质量相关模块</h4>' + table([
|
||
{ key: 'title', label: '页面' },
|
||
{ key: 'routePath', label: '路由' },
|
||
{ key: 'domains', label: '业务域' }
|
||
], listQualityModules(12))
|
||
}
|
||
if (isQualityQuestion(question)) return qualitySearchResponse(question)
|
||
const rows = routeCandidates(question, 12)
|
||
return '<h4>项目知识库匹配结果</h4>' + table([
|
||
{ key: 'title', label: '页面' },
|
||
{ key: 'routePath', label: '路由' },
|
||
{ key: 'view', label: '源码文件' },
|
||
{ key: 'callCount', label: '调用数' }
|
||
], rows)
|
||
}
|
||
|
||
const tools = [
|
||
{ name: 'project_summary', description: '返回 WC-SPC 项目结构、知识库、质量模块和 SPC 图表概览。', inputSchema: { type: 'object', properties: {} } },
|
||
{ name: 'search_project_functions', description: '按关键字搜索项目页面、接口和功能。', inputSchema: { type: 'object', properties: { query: { type: 'string' }, limit: { type: 'number' }}, required: ['query'] } },
|
||
{ name: 'list_quality_modules', description: '列出质量管理相关页面。', inputSchema: { type: 'object', properties: { limit: { type: 'number' } } } },
|
||
{ name: 'search_quality_functions', description: '搜索质量管理相关页面和查询入口。', inputSchema: { type: 'object', properties: { query: { type: 'string' }, limit: { type: 'number' }}, required: ['query'] } },
|
||
{ name: 'list_spc_charts', description: '列出 SPC 图表类型与对应接口。', inputSchema: { type: 'object', properties: {} } },
|
||
{ name: 'spc_query_chart', description: '根据条件调用 SPC 接口并返回图表、摘要和明细。', inputSchema: { type: 'object', properties: { query: { type: 'string' }, chartType: { type: 'string' }, opName: { type: 'string' }, measureName: { type: 'string' }, measureContent: { type: 'string' }, startTime: { type: 'string' }, endTime: { type: 'string' } } } },
|
||
{ name: 'reload_project_index', description: '重新加载知识库。', inputSchema: { type: 'object', properties: {} } },
|
||
{ name: 'call_mes_backend', description: '调用 MESCommonBase.ashx，默认禁用。', inputSchema: { type: 'object', properties: { payload: { type: 'object' }}, required: ['payload'] } }
|
||
]
|
||
|
||
async function toolResult(name, args) {
|
||
const limit = Math.min(Number(args.limit || 30), 100)
|
||
if (name === 'project_summary') {
|
||
const rows = [
|
||
{ name: '知识库生成时间', value: knowledgeBase.generatedAt || '未生成' },
|
||
{ name: '页面模块', value: knowledgeBase.modules.length },
|
||
{ name: 'API 映射', value: knowledgeBase.apiMappings.length },
|
||
{ name: '质量相关模块', value: knowledgeBase.qualityModules.length },
|
||
{ name: 'SPC 图表类型', value: knowledgeBase.spcCharts.length },
|
||
{ name: 'DeepSeek 已配置', value: Boolean(DEEPSEEK_API_KEY) },
|
||
{ name: 'SPC 后端地址', value: SPC_BACKEND_URL }
|
||
]
|
||
return { text: JSON.stringify(rows, null, 2), html: '<h4>项目概览</h4>' + table([{ key: 'name', label: '项目' }, { key: 'value', label: '值' }], rows), data: rows }
|
||
}
|
||
if (name === 'search_project_functions') {
|
||
const rows = routeCandidates(args.query, limit)
|
||
return { text: JSON.stringify(rows, null, 2), html: table([{ key: 'title', label: '页面' }, { key: 'routePath', label: '路由' }, { key: 'view', label: '源码' }, { key: 'callCount', label: '调用数' }], rows), data: rows }
|
||
}
|
||
if (name === 'list_quality_modules') {
|
||
const rows = listQualityModules(limit)
|
||
return { text: JSON.stringify(rows, null, 2), html: table([{ key: 'title', label: '页面' }, { key: 'routePath', label: '路由' }, { key: 'domains', label: '业务域' }, { key: 'queryable', label: '可查询' }], rows), data: rows }
|
||
}
|
||
if (name === 'search_quality_functions') {
|
||
const rows = pick(knowledgeBase.qualityModules, args.query, ['Title', 'View', 'Domains', 'QueryNames'], limit)
|
||
return { text: JSON.stringify(rows, null, 2), html: table([{ key: 'Title', label: '页面' }, { key: 'RoutePath', label: '路由' }, { key: 'Domains', label: '业务域' }, { key: 'QueryNames', label: '查询入口' }], rows), data: rows }
|
||
}
|
||
if (name === 'list_spc_charts') {
|
||
const rows = knowledgeBase.spcCharts || []
|
||
return { text: JSON.stringify(rows, null, 2), html: table([{ key: 'title', label: '图表' }, { key: 'chartType', label: '类型' }, { key: 'apiFile', label: 'API 文件' }, { key: 'endpoint', label: '后端接口' }], rows), data: rows }
|
||
}
|
||
if (name === 'spc_query_chart') {
|
||
const result = await querySpcChart(args || {})
|
||
return { text: JSON.stringify(result.data, null, 2), html: result.html, data: result }
|
||
}
|
||
if (name === 'reload_project_index') {
|
||
knowledgeBase = readJson(KNOWLEDGE_FILE, emptyKnowledgeBase())
|
||
return { text: 'reloaded', html: table([{ key: 'name', label: '项目' }, { key: 'value', label: '值' }], [{ name: '状态', value: '知识库已重新加载' }]), data: { ok: true } }
|
||
}
|
||
if (name === 'call_mes_backend') {
|
||
if (!ALLOW_BACKEND_CALLS) {
|
||
return { text: 'disabled', html: table([{ key: 'name', label: '项目' }, { key: 'value', label: '值' }], [{ name: '状态', value: '已禁用，避免 AI 直接修改业务数据' }]), data: { disabled: true } }
|
||
}
|
||
const response = await postJson(MES_BACKEND_URL, args.payload || {})
|
||
return { text: JSON.stringify(response, null, 2), html: htmlForRows('MES 后端返回', Array.isArray(response) ? response : [response], 30), data: response }
|
||
}
|
||
throw new Error('Unknown tool: ' + name)
|
||
}
|
||
|
||
async function handleMcp(body) {
|
||
const id = body.id == null ? null : body.id
|
||
try {
|
||
if (body.method === 'initialize') {
|
||
return { jsonrpc: '2.0', id, result: { protocolVersion: '2025-06-18', serverInfo: { name: 'wc-spc-mcp', version: '1.0.0' }, capabilities: { tools: {} } } }
|
||
}
|
||
if (body.method === 'tools/list') return { jsonrpc: '2.0', id, result: { tools } }
|
||
if (body.method === 'tools/call') {
|
||
const result = await toolResult(body.params && body.params.name, (body.params && body.params.arguments) || {})
|
||
return { jsonrpc: '2.0', id, result: { content: [{ type: 'text', text: result.text }, { type: 'text', text: result.html }], structuredContent: result.data } }
|
||
}
|
||
return { jsonrpc: '2.0', id, error: { code: -32601, message: 'Method not found' } }
|
||
} catch (error) {
|
||
return { jsonrpc: '2.0', id, error: { code: -32000, message: error.message } }
|
||
}
|
||
}
|
||
|
||
function parseBody(req) {
|
||
return new Promise((resolve, reject) => {
|
||
const chunks = []
|
||
req.on('data', chunk => { chunks.push(Buffer.from(chunk)) })
|
||
req.on('end', () => {
|
||
try {
|
||
const text = Buffer.concat(chunks).toString('utf8')
|
||
resolve(text ? JSON.parse(text) : {})
|
||
} catch (error) {
|
||
reject(error)
|
||
}
|
||
})
|
||
req.on('error', reject)
|
||
})
|
||
}
|
||
|
||
function send(res, status, payload) {
|
||
const body = typeof payload === 'string' ? payload : JSON.stringify(payload)
|
||
res.writeHead(status, {
|
||
'Content-Type': typeof payload === 'string' ? 'text/html; charset=utf-8' : 'application/json; charset=utf-8',
|
||
'Access-Control-Allow-Origin': '*',
|
||
'Access-Control-Allow-Headers': 'Content-Type, Authorization',
|
||
'Access-Control-Allow-Methods': 'GET, POST, OPTIONS'
|
||
})
|
||
res.end(body)
|
||
}
|
||
|
||
const server = http.createServer(async (req, res) => {
|
||
if (req.method === 'OPTIONS') return send(res, 204, '')
|
||
try {
|
||
if (req.method === 'GET' && req.url === '/health') {
|
||
return send(res, 200, {
|
||
ok: true,
|
||
model: DEEPSEEK_MODEL,
|
||
apiKeyConfigured: Boolean(DEEPSEEK_API_KEY),
|
||
knowledgeBase: { loaded: Boolean(knowledgeBase.generatedAt), modules: knowledgeBase.modules.length, qualityModules: knowledgeBase.qualityModules.length, spcCharts: knowledgeBase.spcCharts.length },
|
||
endpoints: { mes: MES_BACKEND_URL, spc: SPC_BACKEND_URL }
|
||
})
|
||
}
|
||
if (req.method === 'GET' && req.url === '/api/tools') return send(res, 200, { tools })
|
||
if (req.method === 'POST' && req.url === '/mcp') return send(res, 200, await handleMcp(await parseBody(req)))
|
||
if (req.method === 'POST' && req.url === '/api/assistant/chat') {
|
||
const body = await parseBody(req)
|
||
const question = String(body.question || '').trim()
|
||
if (!question) return send(res, 400, { error: 'question is required' })
|
||
const navigation = navigationResponse(question)
|
||
if (navigation) return send(res, 200, navigation)
|
||
if (/图|趋势|直方|正态|过程能力|排列|X-R|XR|X-S|XS/i.test(question)) {
|
||
return send(res, 200, await querySpcChart({ query: question }))
|
||
}
|
||
const localHtml = localAnswer(question)
|
||
const html = await askDeepSeek(question, localHtml)
|
||
return send(res, 200, { html })
|
||
}
|
||
return send(res, 404, { error: 'Not found' })
|
||
} catch (error) {
|
||
return send(res, 500, { error: error.message })
|
||
}
|
||
})
|
||
|
||
server.listen(PORT, () => {
|
||
console.log('WC-SPC AI MCP server listening on http://127.0.0.1:' + PORT)
|
||
console.log('DeepSeek model: ' + DEEPSEEK_MODEL + ', API key configured: ' + Boolean(DEEPSEEK_API_KEY))
|
||
console.log('Knowledge base loaded: ' + Boolean(knowledgeBase.generatedAt))
|
||
})
|