增加供应商智能添加功能

This commit is contained in:
2026-09-04 00:09:32 +08:00
parent d8936d9730
commit 20be7f4817
29 changed files with 4317 additions and 1476 deletions
+2
View File
@@ -25,6 +25,8 @@ declare module 'vue' {
ElCheckbox: typeof import('element-plus/es')['ElCheckbox']
ElCheckboxGroup: typeof import('element-plus/es')['ElCheckboxGroup']
ElCol: typeof import('element-plus/es')['ElCol']
ElCollapse: typeof import('element-plus/es')['ElCollapse']
ElCollapseItem: typeof import('element-plus/es')['ElCollapseItem']
ElCollapseTransition: typeof import('element-plus/es')['ElCollapseTransition']
ElColorPicker: typeof import('element-plus/es')['ElColorPicker']
ElContainer: typeof import('element-plus/es')['ElContainer']
+109
View File
@@ -140,3 +140,112 @@ export function smartGenerateCompany(data) {
data
})
}
// MCP工具管理
export function getMcpToolList() {
return request({
url: '/backend/mcp/tool/list',
method: 'get'
})
}
export function createMcpTool(data) {
return request({
url: '/backend/mcp/tool',
method: 'post',
data
})
}
export function updateMcpTool(id, data) {
return request({
url: `/backend/mcp/tool/${id}`,
method: 'put',
data
})
}
export function deleteMcpTool(id) {
return request({
url: `/backend/mcp/tool/${id}`,
method: 'delete'
})
}
export function testMcpTool(data) {
return request({
url: '/backend/mcp/tool/test',
method: 'post',
data
})
}
// MCP服务器配置管理
export function getMcpServerList() {
return request({
url: '/backend/mcp/server/list',
method: 'get'
})
}
export function createMcpServer(data) {
return request({
url: '/backend/mcp/server',
method: 'post',
data
})
}
export function updateMcpServer(id, data) {
return request({
url: `/backend/mcp/server/${id}`,
method: 'put',
data
})
}
export function deleteMcpServer(id) {
return request({
url: `/backend/mcp/server/${id}`,
method: 'delete'
})
}
export function toggleMcpServer(id, enabled) {
return request({
url: `/backend/mcp/server/${id}/toggle`,
method: 'put',
data: { enabled }
})
}
export function testMcpServer(data) {
return request({
url: '/backend/mcp/server/test',
method: 'post',
data
})
}
export function getMcpServerTools(id) {
return request({
url: `/backend/mcp/server/${id}/tools`,
method: 'post'
})
}
// MCP市场
export function getMcpMarket() {
return request({
url: '/backend/mcp/server/market',
method: 'get'
})
}
export function addMcpFromMarket(data) {
return request({
url: '/backend/mcp/server/from-market',
method: 'post',
data
})
}
+98
View File
@@ -0,0 +1,98 @@
# MCP 工具中心(AI 模块)
本模块为「AI 智能体」接入 **MCP(Model Context Protocol)** 工具调用能力:
- **MCP 市场**:内置一批可一键添加的服务,以及项目自带的「内置演示 MCP」(零配置,直接可用)。
- **选择即用**:在「我的服务」里通过「启用」开关控制哪些 MCP 在会话中被调用;**只有启用状态的服务** 才会注入到 AI 会话。
- **会话自动调用**:聊天时,AI 会自动决定是否调用已启用的 MCP 工具,前端实时展示工具调用卡片(参数 / 结果 / 状态)。
---
## 一、如何使用
1. 打开「AI 广场」→ 左侧边栏点击 **CPU(MCP 工具中心)** 按钮。
2. 「MCP 市场」页 → 点击「内置演示 MCP」的 **添加**(或添加天眼查/企查查等,需自行申请对应的 URL/Key)。
3. 切到「我的服务」页 → 把要用的服务 **启用** 开关打开;可点击「测试」验证连接并预览工具。
4. 回到聊天窗 → 提问。例如:
- `现在几点钟了?`(触发 `get_time`)
- `帮我计算 (3+5)*2`(触发 `calculate`)
- `连云港今天天气怎么样?`(触发 `weather_query`)
5. 回答过程中,AI 气泡内会出现工具调用卡片:展开可看参数与返回结果。
> 头部「MCP」绿色指示器表示当前会话已接入 N 个 MCP 工具。
## 二、手动新增 MCP 服务
「我的服务」→「新增配置」,支持三种传输:
| 传输 | 说明 | 必填字段 |
|---|---|---|
| **stdio** | 本地进程(MCP Server 以子进程方式启动) | 启动命令 command;可选参数 args(JSON数组)、环境变量 env(JSON对象) |
| **http** | Streamable HTTP 传输 | 服务地址 url;可选请求头 headers(JSON对象) |
| **sse** | 传统 SSE 传输 | 服务地址 url;可选请求头 headers(JSON对象) |
示例(stdio + npx 远程包,Windows 会自动用 `cmd /c` 包装):
```
命令: npx -y @modelcontextprotocol/server-everything
```
示例(http):
```
地址: https://mcp.tianyancha.com/mcp
请求头: {"Authorization":"Bearer 你的Key"}
```
## 三、内置演示 MCP(cmd/demo-mcp)
项目自带一个 Go 编写的 stdio MCP Server,提供 5 个工具,用于零配置验证全链路:
`get_time`(当前时间)、`calculate`(表达式计算)、`weather_query`(模拟天气)、`echo`、`get_user_profile`。
重新编译(如需):
```bash
cd go
go build -o bin/demo-mcp.exe ./cmd/demo-mcp
```
## 四、实现说明(文件清单)
### 后端(Go / Beego,`go/`)
| 文件 | 作用 |
|---|---|
| `models/backend_mcp_server.go` | MCP 服务器配置模型(传输、命令、URL、启用、状态、工具数) |
| `controllers/backend_mcp_server.go` | MCP 服务器接口:市场 / 增删改查 / 测试 / 工具发现 / 启停 |
| `services/mcp_client.go` | MCP 客户端管理器:stdio/http/sse 连接、工具列表、工具调用、连接缓存 |
| `services/mcp_market.go` | MCP 市场目录(内置演示 + 天眼查/企查查/前瞻/IT桔子/BOSS直聘) |
| `controllers/backend_ai_chat.go` | 聊天工具调用循环:注入已启用 MCP 工具 → LLM 返回 tool_calls → 执行 → 回填 → 流式输出 |
| `cmd/demo-mcp/main.go` | 内置演示 MCP Server(stdio) |
| `sql/create_backend_mcp_server.sql` | 建表 SQL(`yz_backend_mcp_server`) |
**SSE 事件扩展**(`POST /backend/ai/chat/send-stream`):
`session` / `tool_list`(本次启用的工具)/ `tool_start`(开始调用某工具)/ `tool_result`(工具返回)/ `content`(文本增量)/ `done` / `error`
### 前端(Vue3 + Element Plus,`backend/src/`)
| 文件 | 作用 |
|---|---|
| `views/apps/ai/components/mcpConfig.vue` | MCP 工具中心(市场 + 我的服务 + 新增/编辑 + 工具预览) |
| `views/apps/ai/components/chatWindow.vue` | 聊天窗:SSE 工具事件解析 + 工具调用卡片渲染 + MCP 状态指示 |
| `views/apps/ai/components/chatSidebar.vue` | CPU 按钮入口(标题更新为「MCP 工具中心」) |
| `api/ai.js` | MCP 服务器 / 市场 API 封装 |
### 模型要求
- 需支持 **Function Calling / Tool Calling**(OpenAI 兼容与 Anthropic 均已适配)。
- 已内置两种接入:OpenAI 兼容(含豆包/DeepSeek/OpenRouter 等)与 Anthropic。
## 五、依赖说明
- 新增 Go 依赖:`github.com/mark3labs/mcp-go v0.33.0`(MCP 官方协议的 Go 客户端/服务端库)。
- `go.mod` 的 `go` 指令由 `1.17` 调整为 `1.23`(mcp-go v0.33 的最低要求;本机 Go 1.25 完全兼容)。
- 数据库:需先执行 `sql/create_backend_mcp_server.sql`(本环境已执行建表)。
## 六、构建与运行
```bash
# 后端(go/ 目录)
go build -o main.exe .
./main.exe # 监听 9000
# 前端(backend/ 目录,Vue SPA)
npm run dev # 开发,vite 端口 4001,/api 代理到 9000
npm run build # 生产构建
```
@@ -0,0 +1,18 @@
<template>
<div class="ai-playground-page">
<el-empty description="AI 实验室(建设中)" />
</div>
</template>
<script setup>
// 占位页面:AI 实验室
</script>
<style scoped>
.ai-playground-page {
display: flex;
align-items: center;
justify-content: center;
height: 100%;
}
</style>
@@ -4,6 +4,7 @@
<el-button type="primary" class="new-chat-btn" :icon="ChatDotRound" @click="handleNewChat">
新建对话
</el-button>
<el-button :icon="Cpu" class="mcp-btn" @click="handleMcpConfig" title="MCP 工具中心" />
<el-button :icon="Setting" class="settings-btn" @click="settingsVisible = true" title="AI接入设置" />
</div>
@@ -52,15 +53,17 @@
</div>
<ProviderDialog v-model="settingsVisible" @change="handleProviderChange" />
<McpConfig v-model="mcpConfigVisible" />
</div>
</template>
<script setup>
import { ref, onMounted, nextTick } from 'vue'
import { ref, onMounted, nextTick, watch } from 'vue'
import { ElMessage, ElMessageBox } from 'element-plus'
import { ChatDotRound, Setting, Delete, Edit } from '@element-plus/icons-vue'
import { ChatDotRound, Setting, Delete, Edit, Cpu } from '@element-plus/icons-vue'
import { getChatSessionList, createChatSession, deleteChatSession, updateChatSession } from '@/api/ai'
import ProviderDialog from './providerDialog.vue'
import McpConfig from './mcpConfig.vue'
const emit = defineEmits(['select', 'new', 'rename'])
@@ -68,6 +71,7 @@ const loading = ref(false)
const list = ref([])
const currentId = ref(null)
const settingsVisible = ref(false)
const mcpConfigVisible = ref(false)
// 重命名
const editingId = ref(null)
@@ -176,6 +180,10 @@ function handleProviderChange() {
// 配置变更后不需要特殊处理
}
function handleMcpConfig() {
mcpConfigVisible.value = true
}
defineExpose({
fetchList,
setCurrent: (id) => { currentId.value = id }
@@ -205,6 +213,11 @@ defineExpose({
width: 36px;
padding: 0;
}
.mcp-btn {
width: 36px;
padding: 0;
}
}
.session-list {
@@ -42,6 +42,13 @@
:value="m"
/>
</el-select>
<el-tooltip :content="mcpTip" placement="bottom">
<span class="mcp-indicator" :class="{ 'has-mcp': activeMcpTools.length > 0 }">
<el-icon><Cpu /></el-icon>
<span class="mcp-ind-text">MCP</span>
<span v-if="activeMcpTools.length > 0" class="mcp-ind-count">{{ activeMcpTools.length }}</span>
</span>
</el-tooltip>
</div>
</div>
@@ -81,13 +88,46 @@
</div>
<!-- 正常显示 -->
<template v-else>
<div class="message-bubble" :class="{ typing: msg.role === 'assistant' && !msg.content && sending }">
<div
class="message-bubble"
:class="{
typing: msg.role === 'assistant' && !msg.content && sending,
'has-tools': msg.role === 'assistant' && msg.toolCalls && msg.toolCalls.length
}"
>
<template v-if="msg.role === 'assistant' && !msg.content && sending">
<span class="typing-dot"></span>
<span class="typing-dot"></span>
<span class="typing-dot"></span>
</template>
<pre v-else class="message-text">{{ msg.content }}</pre>
<template v-else>
<div v-if="msg.role === 'assistant' && msg.toolCalls && msg.toolCalls.length" class="tool-call-list">
<div v-for="tc in msg.toolCalls" :key="tc.id" class="tool-call-card">
<div class="tool-call-head" @click="tc.expanded = !tc.expanded">
<el-icon class="tool-call-icon"><Cpu /></el-icon>
<span class="tool-call-name">
<span v-if="tc.serverName" class="tool-server">{{ tc.serverName }}</span>
{{ tc.tool }}
</span>
<el-icon v-if="tc.status === 'running'" class="is-loading tool-status-icon"><Loading /></el-icon>
<el-tag v-else-if="tc.status === 'ok'" type="success" size="small" effect="plain">完成</el-tag>
<el-tag v-else-if="tc.status === 'error'" type="danger" size="small" effect="plain">失败</el-tag>
<span class="tool-call-toggle">{{ tc.expanded ? '收起' : '详情' }}</span>
</div>
<div v-if="tc.expanded" class="tool-call-detail">
<div v-if="tc.args && Object.keys(tc.args).length" class="tool-call-block">
<div class="tool-call-label">参数</div>
<pre class="tool-call-json">{{ JSON.stringify(tc.args, null, 2) }}</pre>
</div>
<div v-if="tc.status !== 'running'" class="tool-call-block">
<div class="tool-call-label">结果</div>
<pre class="tool-call-json" :class="{ 'is-error': tc.status === 'error' }">{{ tc.status === 'error' ? tc.error : tc.result }}</pre>
</div>
</div>
</div>
</div>
<pre v-if="msg.content" class="message-text">{{ msg.content }}</pre>
</template>
</div>
<div class="message-actions">
<el-tooltip content="复制" placement="top">
@@ -140,8 +180,8 @@
<script setup>
import { ref, nextTick, watch, computed, onMounted } from 'vue'
import { ElMessage } from 'element-plus'
import { User, Cpu, Promotion, Edit, CopyDocument, RefreshRight, Delete, UserFilled } from '@element-plus/icons-vue'
import { getChatMessageList, sendChatMessage, getAiProviderList, deleteChatMessage, getAiPresetList } from '@/api/ai'
import { User, Cpu, Promotion, Edit, CopyDocument, RefreshRight, Delete, UserFilled, Loading } from '@element-plus/icons-vue'
import { getChatMessageList, sendChatMessage, getAiProviderList, deleteChatMessage, getAiPresetList, getMcpServerList } from '@/api/ai'
import PresetDialog from './presetDialog.vue'
const props = defineProps({
@@ -171,6 +211,16 @@ const selectedModel = ref('')
const presetVisible = ref(false)
const defaultPreset = ref(null)
// MCP 接入状态
const activeMcpTools = ref([])
const mcpTip = computed(() => {
if (activeMcpTools.value.length === 0) {
return '当前未启用 MCP 工具,可在侧边栏 CPU 按钮中配置'
}
const servers = [...new Set(activeMcpTools.value.map((t) => t.server_name).filter(Boolean))]
return `已接入 ${activeMcpTools.value.length} 个 MCP 工具(${servers.join('、')}),会话中 AI 会自动调用`
})
const currentModels = computed(() => {
const p = providerList.value.find(item => item.id === selectedProviderId.value)
return p && p.models ? p.models : []
@@ -179,8 +229,25 @@ const currentModels = computed(() => {
onMounted(() => {
fetchProviders()
fetchDefaultPreset()
fetchMcpStatus()
})
async function fetchMcpStatus() {
try {
const res = await getMcpServerList()
if (res.code === 200 && res.data && res.data.list) {
const enabled = res.data.list.filter((s) => s.enabled === 1)
const tools = []
enabled.forEach((s) => {
tools.push({ key: '', server_id: s.id, server_name: s.name, tool_name: '' })
})
activeMcpTools.value = tools
}
} catch (e) {
// 静默失败
}
}
async function fetchDefaultPreset() {
try {
const res = await getAiPresetList()
@@ -386,7 +453,7 @@ async function doSend(content) {
// 添加用户消息
messages.value.push({ role: 'user', content })
// 添加空的AI消息,用于流式更新
const aiMsg = { id: null, role: 'assistant', content: '' }
const aiMsg = { id: null, role: 'assistant', content: '', toolCalls: [] }
messages.value.push(aiMsg)
sending.value = true
scrollToBottom()
@@ -434,6 +501,35 @@ async function doSend(content) {
} else if (eventType === 'content') {
aiMsg.content += parsed.content
scrollToBottom()
} else if (eventType === 'tool_list') {
// 当前会话启用的 MCP 工具
if (parsed.tools) {
activeMcpTools.value = parsed.tools.filter((t) => t.tool_name)
}
} else if (eventType === 'tool_start') {
// AI 开始调用某个 MCP 工具
const tc = {
id: parsed.id,
key: parsed.key,
tool: parsed.tool,
serverName: parsed.server_name,
args: parsed.args || {},
status: 'running',
expanded: false,
result: '',
error: ''
}
aiMsg.toolCalls.push(tc)
scrollToBottom()
} else if (eventType === 'tool_result') {
// 工具调用返回结果
const tc = aiMsg.toolCalls.find((t) => t.id === parsed.id)
if (tc) {
tc.status = parsed.ok ? 'ok' : 'error'
tc.result = parsed.result || ''
tc.error = parsed.error || ''
}
scrollToBottom()
} else if (eventType === 'done') {
aiMsg.id = parsed.message_id
} else if (eventType === 'error') {
@@ -743,4 +839,130 @@ defineExpose({
margin-top: 8px;
}
}
// MCP 状态指示
.mcp-indicator {
display: inline-flex;
align-items: center;
gap: 4px;
padding: 4px 10px;
border-radius: 16px;
border: 1px solid var(--el-border-color);
color: var(--el-text-color-secondary);
font-size: 12px;
cursor: default;
height: 32px;
&.has-mcp {
background: var(--el-color-success-light-9);
border-color: var(--el-color-success);
color: var(--el-color-success);
}
.mcp-ind-count {
background: var(--el-color-success);
color: #fff;
border-radius: 8px;
font-size: 11px;
line-height: 16px;
padding: 0 5px;
min-width: 16px;
text-align: center;
}
}
// 工具调用卡片
.tool-call-list {
display: flex;
flex-direction: column;
gap: 6px;
margin-bottom: 8px;
}
.tool-call-card {
border: 1px solid var(--el-border-color-lighter);
border-radius: 8px;
background: var(--el-bg-color);
overflow: hidden;
.tool-call-head {
display: flex;
align-items: center;
gap: 6px;
padding: 7px 10px;
cursor: pointer;
font-size: 13px;
.tool-call-icon {
color: var(--el-color-primary);
flex-shrink: 0;
}
.tool-call-name {
flex: 1;
font-family: 'JetBrains Mono', Consolas, monospace;
font-weight: 500;
overflow: hidden;
text-overflow: ellipsis;
white-space: nowrap;
.tool-server {
color: var(--el-text-color-secondary);
font-family: inherit;
margin-right: 4px;
}
}
.tool-status-icon {
color: var(--el-color-primary);
}
.tool-call-toggle {
font-size: 12px;
color: var(--el-text-color-secondary);
flex-shrink: 0;
}
}
.tool-call-detail {
border-top: 1px dashed var(--el-border-color-lighter);
padding: 8px 10px;
.tool-call-block {
margin-bottom: 6px;
&:last-child {
margin-bottom: 0;
}
}
.tool-call-label {
font-size: 12px;
color: var(--el-text-color-secondary);
margin-bottom: 4px;
}
.tool-call-json {
margin: 0;
background: #f5f7fa;
border-radius: 6px;
padding: 8px 10px;
font-size: 12px;
line-height: 1.5;
white-space: pre-wrap;
word-break: break-all;
max-height: 200px;
overflow: auto;
&.is-error {
color: var(--el-color-danger);
}
}
}
}
.message-bubble.has-tools {
display: block;
max-width: 100%;
}
</style>
@@ -0,0 +1,561 @@
<template>
<el-dialog
:model-value="modelValue"
title="MCP 工具中心"
width="920px"
top="6vh"
append-to-body
class="mcp-center-dialog"
@update:model-value="emitUpdate"
@closed="handleClosed"
>
<el-tabs v-model="activeTab">
<!-- ============ MCP 市场 ============ -->
<el-tab-pane label="MCP 市场" name="market">
<el-alert
title="从市场选择 MCP 服务并「添加」,然后在「我的服务」中开启「启用」开关;开启后,会话中 AI 会自动调用这些 MCP 工具。"
type="info"
:closable="false"
show-icon
class="mb12"
/>
<el-row :gutter="12" v-loading="marketLoading">
<el-col :span="8" v-for="item in marketList" :key="item.key" class="mb12">
<el-card shadow="hover" class="market-card">
<div class="market-head">
<div class="market-name">
<el-tag v-if="item.built_in" type="success" size="small" effect="dark" class="mr6">内置</el-tag>
<span class="name-text">{{ item.name }}</span>
</div>
<el-button
v-if="!isAdded(item.key)"
type="primary"
size="small"
:loading="addingKey === item.key"
@click="handleAddMarket(item)"
>添加</el-button>
<el-tag v-else type="info" size="small">已添加</el-tag>
</div>
<div class="market-provider">{{ item.provider }}</div>
<div class="market-desc">{{ item.description }}</div>
<div class="market-tags">
<el-tag v-for="t in item.tags" :key="t" size="small" type="warning" effect="plain" class="mr6">{{ t }}</el-tag>
<el-tag size="small" effect="plain">{{ item.transport.toUpperCase() }}</el-tag>
</div>
</el-card>
</el-col>
<el-col v-if="!marketLoading && marketList.length === 0" :span="24">
<el-empty description="暂无市场服务" />
</el-col>
</el-row>
</el-tab-pane>
<!-- ============ 我的服务 ============ -->
<el-tab-pane label="我的服务" name="mine">
<div class="mine-toolbar">
<el-button type="primary" :icon="Plus" @click="openEdit(null)">新增配置</el-button>
<el-button :icon="Refresh" @click="loadServers">刷新</el-button>
</div>
<el-table :data="serverList" v-loading="serversLoading" size="default" border>
<el-table-column label="名称" min-width="150">
<template #default="{ row }">
<div class="srv-name">
<el-tag v-if="row.from_market" type="warning" size="small" class="mr6" effect="plain">市场</el-tag>
{{ row.name }}
</div>
</template>
</el-table-column>
<el-table-column label="传输" width="80">
<template #default="{ row }">
<el-tag size="small" effect="plain">{{ row.transport ? row.transport.toUpperCase() : 'HTTP' }}</el-tag>
</template>
</el-table-column>
<el-table-column label="工具数" width="80" align="center">
<template #default="{ row }">{{ row.tool_count || 0 }}</template>
</el-table-column>
<el-table-column label="连接状态" width="100" align="center">
<template #default="{ row }">
<el-tag v-if="row.status === 1" type="success" size="small">正常</el-tag>
<el-tag v-else-if="row.status === 2" type="danger" size="small">失败</el-tag>
<el-tag v-else type="info" size="small">未测试</el-tag>
</template>
</el-table-column>
<el-table-column label="启用" width="80" align="center">
<template #default="{ row }">
<el-switch
:model-value="row.enabled === 1"
:loading="togglingId === row.id"
@change="(v) => handleToggle(row, v)"
/>
</template>
</el-table-column>
<el-table-column label="操作" min-width="230" align="center">
<template #default="{ row }">
<el-button link type="primary" size="small" @click="handleTest(row)">测试</el-button>
<el-button link type="success" size="small" @click="handleViewTools(row)">工具</el-button>
<el-button link type="primary" size="small" @click="openEdit(row)">编辑</el-button>
<el-button link type="danger" size="small" @click="handleDelete(row)">删除</el-button>
</template>
</el-table-column>
<template #empty>
<el-empty description="尚未配置 MCP 服务,请先在「MCP 市场」中添加或手动新增" :image-size="80" />
</template>
</el-table>
<el-alert
v-if="serverList.length"
title="仅「启用」状态的服务会在 AI 会话中被自动调用。"
type="warning"
:closable="false"
class="mt12"
/>
</el-tab-pane>
</el-tabs>
<!-- ============ 新增/编辑服务 ============ -->
<el-dialog
v-model="editVisible"
:title="form.id ? '编辑 MCP 服务' : '新增 MCP 服务'"
width="620px"
append-to-body
>
<el-form :model="form" label-width="96px" :rules="formRules" ref="formRef">
<el-form-item label="服务名称" prop="name">
<el-input v-model="form.name" placeholder="如:内置演示MCP、天眼查" />
</el-form-item>
<el-form-item label="传输方式">
<el-radio-group v-model="form.transport">
<el-radio value="stdio">本地进程 (stdio)</el-radio>
<el-radio value="http">HTTP</el-radio>
<el-radio value="sse">SSE</el-radio>
</el-radio-group>
</el-form-item>
<template v-if="form.transport === 'stdio'">
<el-form-item label="启动命令" prop="command">
<el-input v-model="form.command" placeholder="如:E:\...\bin\demo-mcp.exe 或 npx -y @xxx/server" />
</el-form-item>
<el-form-item label="启动参数">
<el-input
v-model="form.args"
type="textarea"
:rows="2"
placeholder='JSON 数组或逗号分隔,如 ["--port","8080"]'
/>
</el-form-item>
<el-form-item label="环境变量">
<el-input
v-model="form.env"
type="textarea"
:rows="3"
placeholder='JSON 对象,如 {"API_KEY":"xxx"}'
/>
</el-form-item>
</template>
<template v-else>
<el-form-item label="服务地址" prop="url">
<el-input v-model="form.url" placeholder="如:https://mcp.example.com/mcp" />
</el-form-item>
<el-form-item label="请求头">
<el-input
v-model="form.headers"
type="textarea"
:rows="3"
placeholder='JSON 对象,如 {"Authorization":"Bearer xxx"}'
/>
</el-form-item>
</template>
<el-form-item label="描述">
<el-input v-model="form.description" type="textarea" :rows="2" placeholder="简要说明该服务提供的能力" />
</el-form-item>
<el-form-item label="立即启用">
<el-switch v-model="form.enabled" />
<span class="form-tip">开启后 AI 会话即可自动调用</span>
</el-form-item>
</el-form>
<template #footer>
<el-button @click="editVisible = false">取消</el-button>
<el-button type="primary" :loading="saving" @click="handleSave">保存</el-button>
</template>
</el-dialog>
<!-- ============ 工具预览 ============ -->
<el-dialog v-model="toolsVisible" title="工具列表" width="720px" append-to-body>
<template v-if="toolsData.tools && toolsData.tools.length">
<el-collapse>
<el-collapse-item v-for="t in toolsData.tools" :key="t.name" :name="t.name">
<template #title>
<span class="tool-name">{{ t.name }}</span>
<el-tag size="small" effect="plain" class="ml8">server: {{ toolsData.server_name }}</el-tag>
</template>
<div class="tool-desc">{{ t.description || '(无描述)' }}</div>
<pre class="tool-schema">{{ formatSchema(t.input_schema) }}</pre>
</el-collapse-item>
</el-collapse>
</template>
<el-empty v-else-if="toolsData.tools" description="该服务未暴露任何工具" />
<el-empty v-else description="请先测试连接以发现工具" />
</el-dialog>
<template #footer>
<span class="dialog-tip">MCP 市场 · 选择即用,会话自动调用开启的服务</span>
<el-button @click="emitUpdate(false)">关闭</el-button>
</template>
</el-dialog>
</template>
<script setup>
import { ref, reactive, computed, onMounted, watch } from 'vue'
import { ElMessage, ElMessageBox } from 'element-plus'
import { Plus, Refresh } from '@element-plus/icons-vue'
import {
getMcpMarket,
addMcpFromMarket,
getMcpServerList,
createMcpServer,
updateMcpServer,
deleteMcpServer,
toggleMcpServer,
testMcpServer,
getMcpServerTools
} from '@/api/ai'
const props = defineProps({
modelValue: { type: Boolean, default: false }
})
const emit = defineEmits(['update:modelValue'])
const activeTab = ref('market')
const marketList = ref([])
const serverList = ref([])
const marketLoading = ref(false)
const serversLoading = ref(false)
const addingKey = ref('')
const togglingId = ref(0)
const saving = ref(false)
const editVisible = ref(false)
const formRef = ref()
const form = reactive({
id: 0,
name: '',
transport: 'stdio',
command: '',
args: '',
env: '',
url: '',
headers: '',
description: '',
enabled: false
})
const formRules = {
name: [{ required: true, message: '请输入服务名称', trigger: 'blur' }],
command: [{ required: true, message: '请输入启动命令', trigger: 'blur' }],
url: [{ required: true, message: '请输入服务地址', trigger: 'blur' }]
}
const toolsVisible = ref(false)
const toolsData = reactive({ tools: null, server_name: '' })
const marketKeys = computed(() => new Set(serverList.value.filter((s) => s.from_market).map((s) => s.from_market)))
function emitUpdate(v) {
emit('update:modelValue', v)
}
function handleClosed() {
activeTab.value = 'market'
}
function isAdded(key) {
return marketKeys.value.has(key)
}
async function loadMarket() {
marketLoading.value = true
try {
const res = await getMcpMarket()
marketList.value = res.data?.list || []
} catch (e) {
ElMessage.error(e.message || '加载市场失败')
} finally {
marketLoading.value = false
}
}
async function loadServers() {
serversLoading.value = true
try {
const res = await getMcpServerList()
serverList.value = res.data?.list || []
} catch (e) {
ElMessage.error(e.message || '加载服务列表失败')
} finally {
serversLoading.value = false
}
}
async function handleAddMarket(item) {
addingKey.value = item.key
try {
await addMcpFromMarket({ key: item.key })
ElMessage.success(`已添加「${item.name}」,请到「我的服务」中开启启用开关`)
await loadServers()
activeTab.value = 'mine'
} catch (e) {
ElMessage.error(e.message || '添加失败')
} finally {
addingKey.value = ''
}
}
function openEdit(row) {
if (row) {
Object.assign(form, {
id: row.id,
name: row.name || '',
transport: row.transport || 'http',
command: row.command || '',
args: row.args || '',
env: row.env || '',
url: row.url || '',
headers: row.headers || '',
description: row.description || '',
enabled: row.enabled === 1
})
} else {
Object.assign(form, {
id: 0,
name: '',
transport: 'stdio',
command: '',
args: '',
env: '',
url: '',
headers: '',
description: '',
enabled: true
})
}
editVisible.value = true
}
async function handleSave() {
await formRef.value.validate()
const payload = {
name: form.name,
transport: form.transport,
command: form.command,
args: form.args,
env: form.env,
url: form.url,
headers: form.headers,
description: form.description,
enabled: form.enabled ? 1 : 0
}
// 校验 JSON 字段
for (const key of ['args', 'env', 'headers']) {
if (payload[key] && !isValidJson(payload[key])) {
ElMessage.error(`「${key}」必须是合法 JSON`)
return
}
}
saving.value = true
try {
if (form.id) {
await updateMcpServer(form.id, payload)
} else {
await createMcpServer(payload)
}
ElMessage.success('保存成功')
editVisible.value = false
await loadServers()
} catch (e) {
ElMessage.error(e.message || '保存失败')
} finally {
saving.value = false
}
}
async function handleToggle(row, val) {
togglingId.value = row.id
try {
await toggleMcpServer(row.id, val ? 1 : 0)
row.enabled = val ? 1 : 0
ElMessage.success(val ? '已启用,AI 会话将自动调用该服务' : '已停用')
} catch (e) {
ElMessage.error(e.message || '操作失败')
} finally {
togglingId.value = 0
}
}
async function handleTest(row) {
const loading = ElMessage({ message: '正在连接测试...', type: 'info', duration: 0 })
try {
const res = await testMcpServer({ id: row.id })
loading.close()
ElMessage.success(`连接成功,发现 ${res.data?.count || 0} 个工具`)
row.status = 1
row.tool_count = res.data?.count || 0
toolsData.tools = res.data?.tools || []
toolsData.server_name = row.name
toolsVisible.value = true
} catch (e) {
loading.close()
ElMessage.error(e.message || '连接失败')
row.status = 2
}
}
async function handleViewTools(row) {
try {
const res = await getMcpServerTools(row.id)
toolsData.tools = res.data?.tools || []
toolsData.server_name = row.name
toolsVisible.value = true
} catch (e) {
ElMessage.error(e.message || '获取工具失败')
}
}
async function handleDelete(row) {
try {
await ElMessageBox.confirm(`确定删除 MCP 服务「${row.name}」?`, '删除确认', { type: 'warning' })
await deleteMcpServer(row.id)
ElMessage.success('已删除')
await loadServers()
} catch (e) {
// 取消或失败
}
}
function isValidJson(s) {
try {
JSON.parse(s)
return true
} catch {
return false
}
}
function formatSchema(schema) {
if (!schema) return '{}'
try {
return JSON.stringify(schema, null, 2)
} catch {
return String(schema)
}
}
watch(
() => props.modelValue,
(v) => {
if (v) {
loadMarket()
loadServers()
}
}
)
onMounted(() => {
if (props.modelValue) {
loadMarket()
loadServers()
}
})
</script>
<style scoped>
.mcp-center-dialog :deep(.el-dialog__body) {
padding-top: 8px;
}
.mb12 {
margin-bottom: 12px;
}
.mt12 {
margin-top: 12px;
}
.mr6 {
margin-right: 6px;
}
.ml8 {
margin-left: 8px;
}
.market-card {
height: 100%;
}
.market-head {
display: flex;
align-items: center;
justify-content: space-between;
margin-bottom: 4px;
}
.market-name {
display: flex;
align-items: center;
min-width: 0;
}
.name-text {
font-weight: 600;
overflow: hidden;
text-overflow: ellipsis;
white-space: nowrap;
}
.market-provider {
font-size: 12px;
color: #909399;
margin-bottom: 6px;
}
.market-desc {
font-size: 13px;
color: #606266;
line-height: 1.5;
min-height: 40px;
margin-bottom: 8px;
display: -webkit-box;
-webkit-line-clamp: 2;
-webkit-box-orient: vertical;
overflow: hidden;
}
.market-tags {
min-height: 22px;
}
.mine-toolbar {
margin-bottom: 12px;
}
.srv-name {
display: flex;
align-items: center;
}
.form-tip {
margin-left: 10px;
font-size: 12px;
color: #909399;
}
.tool-name {
font-weight: 600;
font-family: 'JetBrains Mono', Consolas, monospace;
}
.tool-desc {
font-size: 13px;
color: #606266;
margin-bottom: 8px;
}
.tool-schema {
background: #f5f7fa;
border-radius: 6px;
padding: 10px;
max-height: 220px;
overflow: auto;
font-size: 12px;
line-height: 1.5;
margin: 0;
}
.dialog-tip {
color: #909399;
font-size: 12px;
margin-right: 12px;
}
</style>
@@ -126,9 +126,23 @@
<el-form-item label="状态" prop="status">
<el-switch v-model="form.status" :active-value="1" :inactive-value="0" active-text="启用" inactive-text="禁用" />
</el-form-item>
<el-form-item label="默认模型" prop="is_default">
<el-form-item label="默认模型" prop="default_model">
<el-select v-model="form.default_model" placeholder="请选择默认模型" clearable style="width: 100%">
<el-option
v-for="(model, idx) in form.models"
:key="idx"
:label="model"
:value="model"
>
<span class="model-option">{{ model }}</span>
<el-tag v-if="idx === 0" size="small" type="info">首个</el-tag>
</el-option>
</el-select>
<div class="form-tip">选择后,智能添加供应商/客户等企业级功能将调用此模型</div>
</el-form-item>
<el-form-item label="设为默认接入" prop="is_default">
<el-switch v-model="form.is_default" :active-value="1" :inactive-value="0" active-text="是" inactive-text="否" />
<div class="form-tip">设为默认后,智能添加供应商/客户等企业级功能将调用此接入</div>
<div class="form-tip">设为默认后,此接入将作为租户的默认AI接入</div>
</el-form-item>
<el-form-item label="备注" prop="remark">
<el-input v-model="form.remark" type="textarea" :rows="2" placeholder="选填" />
@@ -176,6 +190,7 @@ const defaultForm = () => ({
api_base: '',
api_key: '',
models: [''],
default_model: '',
is_default: 0,
status: 1,
remark: ''
@@ -286,6 +301,7 @@ function openEdit(row) {
api_base: row.api_base,
api_key: row.api_key,
models: row.models && row.models.length > 0 ? [...row.models] : [''],
default_model: row.default_model || '',
is_default: row.is_default || 0,
status: row.status,
remark: row.remark
@@ -311,7 +327,11 @@ async function handleSave() {
submitting.value = true
try {
const payload = { ...form, models: validModels }
const payload = {
...form,
models: validModels,
default_model: form.default_model
}
if (editData.value) {
await updateAiProvider(editData.value.id, payload)
ElMessage.success('更新成功')
@@ -349,23 +369,12 @@ async function handleTestProvider(row) {
return
}
// 需要先编辑才能看到真实的api_key进行测试
// 或者直接从当前编辑的form中获取(如果正在编辑这个provider)
let apiKey = row.api_key
// 如果是脱敏的key,需要打开编辑对话框
if (apiKey && apiKey.includes('****')) {
ElMessage.warning('需要打开编辑表单才能看到完整API Key进行测试')
openEdit(row)
return
}
testingProviderId.value = row.id
try {
const res = await testAiProvider({
provider_type: row.provider_type,
api_base: row.api_base,
api_key: apiKey,
api_key: row.api_key,
models: row.models
})
@@ -0,0 +1,18 @@
<template>
<div class="ai-settings-page">
<el-empty description="AI 设置(建设中)" />
</div>
</template>
<script setup>
// 占位页面:AI 设置
</script>
<style scoped>
.ai-settings-page {
display: flex;
align-items: center;
justify-content: center;
height: 100%;
}
</style>
@@ -50,6 +50,47 @@
<el-input v-model="form.name" />
</el-form-item>
</el-col>
<el-col :span="12" v-if="type === 'customer'">
<el-form-item label="客户类型">
<el-select v-model="form.customer_type" placeholder="请选择" style="width: 100%">
<el-option label="企业" value="1" />
<el-option label="政府机构" value="2" />
<el-option label="国企" value="3" />
<el-option label="教育机构" value="4" />
<el-option label="个人" value="5" />
</el-select>
</el-form-item>
</el-col>
<el-col :span="12" v-if="type === 'supplier'">
<el-form-item label="供应商类型">
<el-select v-model="form.supplier_type" placeholder="请选择" style="width: 100%">
<el-option label="原材料供应商" value="1" />
<el-option label="设备供应商" value="2" />
<el-option label="服务供应商" value="3" />
<el-option label="其他" value="4" />
</el-select>
</el-form-item>
</el-col>
<el-col :span="12" v-if="type === 'customer'">
<el-form-item label="客户等级">
<el-select v-model="form.customer_level" placeholder="请选择" style="width: 100%">
<el-option label="核心客户" value="1" />
<el-option label="重要客户" value="2" />
<el-option label="普通客户" value="3" />
<el-option label="潜在客户" value="4" />
</el-select>
</el-form-item>
</el-col>
<el-col :span="12" v-if="type === 'supplier'">
<el-form-item label="供应商等级">
<el-select v-model="form.supplier_level" placeholder="请选择" style="width: 100%">
<el-option label="核心供应商" value="1" />
<el-option label="重要供应商" value="2" />
<el-option label="普通供应商" value="3" />
<el-option label="潜在供应商" value="4" />
</el-select>
</el-form-item>
</el-col>
<el-col :span="12">
<el-form-item label="所属行业">
<el-input v-model="form.industry" />
@@ -57,7 +98,16 @@
</el-col>
<el-col :span="12">
<el-form-item label="企业类型">
<el-input v-model="form.enterprise_type" placeholder="如:有限责任公司" />
<el-select v-model="form.enterprise_type" placeholder="请选择" style="width: 100%">
<el-option label="有限责任公司" value="有限责任公司" />
<el-option label="股份有限公司" value="股份有限公司" />
<el-option label="合伙企业" value="合伙企业" />
<el-option label="个人独资企业" value="个人独资企业" />
<el-option label="国有企业" value="国有企业" />
<el-option label="集体企业" value="集体企业" />
<el-option label="外商投资企业" value="外商投资企业" />
<el-option label="其他" value="其他" />
</el-select>
</el-form-item>
</el-col>
<el-col :span="12">
@@ -65,17 +115,32 @@
<el-select v-model="form.taxpayer_qualification" placeholder="请选择" style="width: 100%">
<el-option label="一般纳税人" value="一般纳税人" />
<el-option label="小规模纳税人" value="小规模纳税人" />
<el-option label="其他" value="其他" />
</el-select>
</el-form-item>
</el-col>
<el-col :span="12">
<el-form-item label="状态">
<el-select v-model="form.status" placeholder="请选择" style="width: 100%">
<el-option label="正常" value="1" />
<el-option label="禁用" value="0" />
<el-option label="冻结" value="2" />
<el-option label="已注销" value="3" />
</el-select>
</el-form-item>
</el-col>
<el-col :span="12">
<el-form-item label="注册资本">
<el-input v-model="form.registered_capital" placeholder="如:100万元" />
<el-input v-model="form.registered_capital" placeholder="万元">
<template #append>万元</template>
</el-input>
</el-form-item>
</el-col>
<el-col :span="12">
<el-form-item label="实缴资本">
<el-input v-model="form.paid_capital" placeholder="如:50万元" />
<el-input v-model="form.paid_capital" placeholder="万元">
<template #append>万元</template>
</el-input>
</el-form-item>
</el-col>
<el-col :span="12">
@@ -157,12 +222,21 @@
<el-form-item label="备注">
<el-input v-model="form.remark" type="textarea" :rows="2" />
</el-form-item>
<!-- 显示原始AI响应数据 -->
<el-divider content-position="left">AI响应数据</el-divider>
<el-form-item label="详细数据">
<div class="ai-data-box">
<pre>{{ formattedAiData }}</pre>
</div>
<div class="form-tip">此数据为AI返回的原始响应,仅作查看参考,已自动填充可编辑字段到上方表单</div>
</el-form-item>
</el-form>
</div>
<template #footer>
<el-button v-if="step === 2" @click="step = 1">返回修改名称</el-button>
<el-button @click="emit('update:visible', false)">取消</el-button>
<!-- <el-button @click="emit('update:visible', false)">取消</el-button> -->
<el-button
v-if="step === 2"
type="primary"
@@ -174,7 +248,7 @@
</template>
<script setup>
import { ref, reactive } from 'vue'
import { ref, reactive, watch } from 'vue'
import { ElMessage } from 'element-plus'
import { Cpu } from '@element-plus/icons-vue'
import { smartGenerateCompany } from '@/api/ai'
@@ -194,8 +268,41 @@ const generating = ref(false)
const saving = ref(false)
const formRef = ref()
// 从文本中提取JSON对象
function extractJSON(text) {
text = text.trim()
// 方法1: 直接解析整个文本
try {
JSON.parse(text)
return text
} catch (e) {
// 继续尝试其他方法
}
// 方法2: 查找第一个 { 和最后一个 }
const start = text.indexOf('{')
const end = text.lastIndexOf('}')
if (start !== -1 && end !== -1 && end > start) {
const jsonStr = text.substring(start, end + 1)
try {
JSON.parse(jsonStr)
return jsonStr
} catch (e) {
// 继续尝试其他方法
}
}
return ''
}
const defaultForm = () => ({
name: '',
customer_type: '',
supplier_type: '',
customer_level: '3',
supplier_level: '3',
status: '1',
industry: '',
enterprise_type: '',
taxpayer_qualification: '',
@@ -219,6 +326,15 @@ const defaultForm = () => ({
const form = reactive(defaultForm())
// 原始AI响应数据(JSON格式字符串)
const rawAiData = ref('')
// formattedAiData 直接显示 rawAiData 的内容(后端已返回格式化好的JSON或原始文本)
const formattedAiData = ref('')
watch(rawAiData, (newVal) => {
formattedAiData.value = newVal
}, { immediate: true })
async function handleGenerate() {
if (!companyName.value.trim()) {
ElMessage.warning('请输入公司名称')
@@ -236,6 +352,14 @@ async function handleGenerate() {
const data = res.data.data
const nameField = props.type === 'supplier' ? 'supplier_name' : 'customer_name'
form.name = data[nameField] || data.name || companyName.value.trim()
if (props.type === 'customer') {
form.customer_type = data.customer_type || '1'
form.customer_level = data.customer_level || '3'
} else {
form.supplier_type = data.supplier_type || ''
form.supplier_level = data.supplier_level || '3'
}
form.status = data.status || '1'
form.industry = data.industry || ''
form.enterprise_type = data.enterprise_type || ''
form.taxpayer_qualification = data.taxpayer_qualification || ''
@@ -255,10 +379,28 @@ async function handleGenerate() {
form.registered_address = data.registered_address || ''
form.registered_phone = data.registered_phone || ''
form.remark = data.remark || ''
// 保存原始AI响应数据(使用后端返回的原始数据)
if (res.data.raw) {
rawAiData.value = res.data.raw
} else {
rawAiData.value = JSON.stringify(data, null, 2)
}
} else {
// 解析失败,只填名称
// 解析失败,显示AI原始返回内容
let rawContent = res.data.raw || '无原始数据'
let errorMsg = res.data.error || '未知错误'
// 保存原始AI响应数据
rawAiData.value = rawContent
// 限制显示长度
if (rawContent.length > 1500) {
rawContent = rawContent.substring(0, 1500) + '...'
}
ElMessage.error({
message: `AI生成失败:${errorMsg}\n\n原始响应:\n${rawContent}`,
showClose: true,
duration: 0 // 不自动关闭
})
form.name = companyName.value.trim()
ElMessage.warning('AI返回格式异常,请手动填写其他信息')
}
step.value = 2
}
@@ -280,6 +422,11 @@ async function handleSave() {
const payload = {
supplier_name: form.name,
customer_name: form.name,
customer_type: form.customer_type,
supplier_type: form.supplier_type,
customer_level: form.customer_level,
supplier_level: form.supplier_level,
status: form.status,
industry: form.industry,
enterprise_type: form.enterprise_type,
taxpayer_qualification: form.taxpayer_qualification,
@@ -299,7 +446,6 @@ async function handleSave() {
registered_address: form.registered_address,
registered_phone: form.registered_phone,
remark: form.remark,
status: '1',
is_draft: 0
}
@@ -323,6 +469,8 @@ function handleClosed() {
step.value = 1
companyName.value = ''
Object.assign(form, defaultForm())
rawAiData.value = ''
formattedAiData.value = ''
}
</script>
@@ -347,5 +495,16 @@ function handleClosed() {
overflow-y: auto;
padding-right: 8px;
}
.ai-data-box {
background: var(--el-fill-color-lighter);
border: 1px solid var(--el-border-color);
border-radius: 4px;
padding: 12px;
font-family: 'Consolas', 'Monaco', monospace;
font-size: 12px;
white-space: pre-wrap;
word-break: break-all;
}
}
</style>
@@ -63,7 +63,6 @@
<div class="table-container" v-loading="loading">
<el-table :data="tableData" stripe border>
<el-table-column type="index" label="#" width="60" align="center" />
<el-table-column prop="customer_name" label="客户名称" min-width="180" show-overflow-tooltip>
<template #default="{ row }">
<div class="name-cell-wrapper">
@@ -62,7 +62,6 @@
<div class="table-container" v-loading="loading">
<el-table :data="tableData" stripe border>
<el-table-column type="index" label="#" width="60" align="center" />
<el-table-column prop="supplier_name" label="供应商名称" min-width="180" show-overflow-tooltip>
<template #default="{ row }">
<div class="name-cell-wrapper">