增加供应商智能添加功能
This commit is contained in:
Vendored
+2
@@ -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']
|
||||
|
||||
@@ -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
|
||||
})
|
||||
}
|
||||
|
||||
@@ -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">
|
||||
|
||||
Binary file not shown.
@@ -0,0 +1,243 @@
|
||||
// 内置演示 MCP 服务（stdio 传输）
|
||||
// 提供 get_time / calculate / weather_query / echo 等工具，用于验证「MCP 市场 → 会话工具调用」全链路。
|
||||
// 运行：go run ./cmd/demo-mcp 或直接运行已编译二进制 bin/demo-mcp
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"math"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/mark3labs/mcp-go/mcp"
|
||||
"github.com/mark3labs/mcp-go/server"
|
||||
)
|
||||
|
||||
func toolResultText(text string) *mcp.CallToolResult {
|
||||
return &mcp.CallToolResult{
|
||||
Content: []mcp.Content{
|
||||
mcp.TextContent{Type: "text", Text: text},
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func main() {
|
||||
srv := server.NewMCPServer("demo-mcp", "1.0.0")
|
||||
|
||||
// 1. 当前时间
|
||||
srv.AddTool(mcp.NewTool(
|
||||
"get_time",
|
||||
mcp.WithDescription("获取服务器当前日期和时间（含星期），不需要参数"),
|
||||
), func(_ context.Context, _ mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
now := time.Now()
|
||||
weekdays := []string{"日", "一", "二", "三", "四", "五", "六"}
|
||||
return toolResultText(fmt.Sprintf("当前时间：%s（星期%s）",
|
||||
now.Format("2006-01-02 15:04:05"), weekdays[int(now.Weekday())])), nil
|
||||
})
|
||||
|
||||
// 2. 计算器
|
||||
srv.AddTool(mcp.NewTool(
|
||||
"calculate",
|
||||
mcp.WithDescription("计算一个数学表达式，支持 + - * / % 与括号。例如：2+3*4"),
|
||||
mcp.WithString("expression", mcp.Required(), mcp.Description("数学表达式字符串")),
|
||||
), func(_ context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
expr := req.GetArguments()["expression"]
|
||||
exprStr, _ := expr.(string)
|
||||
exprStr = strings.TrimSpace(exprStr)
|
||||
if exprStr == "" {
|
||||
return toolResultText("错误：expression 参数不能为空"), nil
|
||||
}
|
||||
val, err := safeEval(exprStr)
|
||||
if err != nil {
|
||||
return toolResultText("错误：" + err.Error()), nil
|
||||
}
|
||||
return toolResultText(fmt.Sprintf("%s = %v", exprStr, val)), nil
|
||||
})
|
||||
|
||||
// 3. 模拟天气
|
||||
srv.AddTool(mcp.NewTool(
|
||||
"weather_query",
|
||||
mcp.WithDescription("查询指定城市的模拟天气信息（演示数据，非真实天气）"),
|
||||
mcp.WithString("city", mcp.Required(), mcp.Description("城市名称，如 北京、上海、连云港")),
|
||||
), func(_ context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
city, _ := req.GetArguments()["city"].(string)
|
||||
city = strings.TrimSpace(city)
|
||||
if city == "" {
|
||||
city = "未知城市"
|
||||
}
|
||||
conditions := []string{"晴", "多云", "小雨", "阴", "晴转多云"}
|
||||
cond := conditions[int(time.Now().Unix())%len(conditions)]
|
||||
temp := 18 + int(time.Now().Unix())%15
|
||||
return toolResultText(fmt.Sprintf("【%s】今天 %s，气温 %d℃，空气质量：良（模拟数据）", city, cond, temp)), nil
|
||||
})
|
||||
|
||||
// 4. 回声
|
||||
srv.AddTool(mcp.NewTool(
|
||||
"echo",
|
||||
mcp.WithDescription("原样返回你输入的文字，用于测试"),
|
||||
mcp.WithString("text", mcp.Required(), mcp.Description("要回显的文字")),
|
||||
), func(_ context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
text, _ := req.GetArguments()["text"].(string)
|
||||
return toolResultText("echo: " + text), nil
|
||||
})
|
||||
|
||||
// 5. 获取用户信息（演示带对象参数的 JSON 参数）
|
||||
srv.AddTool(mcp.NewTool(
|
||||
"get_user_profile",
|
||||
mcp.WithDescription("获取用户画像信息（演示），可指定姓名与年龄"),
|
||||
mcp.WithString("name", mcp.Required(), mcp.Description("姓名")),
|
||||
mcp.WithNumber("age", mcp.Description("年龄")),
|
||||
), func(_ context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
args := req.GetArguments()
|
||||
name, _ := args["name"].(string)
|
||||
age, _ := args["age"].(float64)
|
||||
out, _ := json.Marshal(map[string]interface{}{
|
||||
"name": name,
|
||||
"age": int(age),
|
||||
"role": "演示用户",
|
||||
"note": "这是一条来自内置演示 MCP 服务的测试数据",
|
||||
})
|
||||
return toolResultText(string(out)), nil
|
||||
})
|
||||
|
||||
if err := server.ServeStdio(srv); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
|
||||
// safeEval 极简安全表达式求值（仅支持数字与 + - * / % 和括号）
|
||||
func safeEval(expr string) (float64, error) {
|
||||
parser := &exprParser{s: expr}
|
||||
parser.skipSpace()
|
||||
v, err := parser.parseExpr()
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
parser.skipSpace()
|
||||
if parser.pos < len(parser.s) {
|
||||
return 0, fmt.Errorf("无法解析的表达式（非法字符）")
|
||||
}
|
||||
if math.IsInf(v, 0) || math.IsNaN(v) {
|
||||
return 0, fmt.Errorf("计算结果无效")
|
||||
}
|
||||
return v, nil
|
||||
}
|
||||
|
||||
type exprParser struct {
|
||||
s string
|
||||
pos int
|
||||
}
|
||||
|
||||
func (p *exprParser) skipSpace() {
|
||||
for p.pos < len(p.s) && (p.s[p.pos] == ' ' || p.s[p.pos] == '\t') {
|
||||
p.pos++
|
||||
}
|
||||
}
|
||||
|
||||
func (p *exprParser) peek() byte {
|
||||
if p.pos < len(p.s) {
|
||||
return p.s[p.pos]
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
||||
func (p *exprParser) parseExpr() (float64, error) {
|
||||
v, err := p.parseTerm()
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
for {
|
||||
p.skipSpace()
|
||||
c := p.peek()
|
||||
if c == '+' || c == '-' {
|
||||
p.pos++
|
||||
rhs, err := p.parseTerm()
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
if c == '+' {
|
||||
v += rhs
|
||||
} else {
|
||||
v -= rhs
|
||||
}
|
||||
} else {
|
||||
return v, nil
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (p *exprParser) parseTerm() (float64, error) {
|
||||
v, err := p.parseFactor()
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
for {
|
||||
p.skipSpace()
|
||||
c := p.peek()
|
||||
if c == '*' || c == '/' || c == '%' {
|
||||
p.pos++
|
||||
rhs, err := p.parseFactor()
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
switch c {
|
||||
case '*':
|
||||
v *= rhs
|
||||
case '/':
|
||||
if rhs == 0 {
|
||||
return 0, fmt.Errorf("除数不能为 0")
|
||||
}
|
||||
v /= rhs
|
||||
case '%':
|
||||
if rhs == 0 {
|
||||
return 0, fmt.Errorf("取模除数不能为 0")
|
||||
}
|
||||
v = math.Mod(v, rhs)
|
||||
}
|
||||
} else {
|
||||
return v, nil
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (p *exprParser) parseFactor() (float64, error) {
|
||||
p.skipSpace()
|
||||
c := p.peek()
|
||||
if c == '(' {
|
||||
p.pos++
|
||||
v, err := p.parseExpr()
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
p.skipSpace()
|
||||
if p.peek() != ')' {
|
||||
return 0, fmt.Errorf("缺少右括号")
|
||||
}
|
||||
p.pos++
|
||||
return v, nil
|
||||
}
|
||||
return p.parseNumber()
|
||||
}
|
||||
|
||||
func (p *exprParser) parseNumber() (float64, error) {
|
||||
start := p.pos
|
||||
for p.pos < len(p.s) {
|
||||
c := p.s[p.pos]
|
||||
if (c >= '0' && c <= '9') || c == '.' {
|
||||
p.pos++
|
||||
} else {
|
||||
break
|
||||
}
|
||||
}
|
||||
if start == p.pos {
|
||||
return 0, fmt.Errorf("表达式格式错误（无法识别的字符）")
|
||||
}
|
||||
raw := p.s[start:p.pos]
|
||||
var v float64
|
||||
if _, err := fmt.Sscanf(raw, "%g", &v); err != nil {
|
||||
return 0, fmt.Errorf("数字格式错误: %s", raw)
|
||||
}
|
||||
return v, nil
|
||||
}
|
||||
+928
-462
File diff suppressed because it is too large
Load Diff
@@ -56,6 +56,7 @@ type aiProviderPayload struct {
|
||||
ApiBase string `json:"api_base"`
|
||||
ApiKey string `json:"api_key"`
|
||||
Models []string `json:"models"`
|
||||
DefaultModel string `json:"default_model"` // 默认使用的模型
|
||||
IsDefault int8 `json:"is_default"`
|
||||
Status int8 `json:"status"`
|
||||
Remark string `json:"remark"`
|
||||
@@ -108,11 +109,8 @@ func (c *BackendAiProviderController) List() {
|
||||
return
|
||||
}
|
||||
|
||||
// 脱敏api_key + 解析模型列表
|
||||
// 解析模型列表（不脱敏api_key，前端需要完整api_key进行测试）
|
||||
for i := range list {
|
||||
if list[i].ApiKey != "" && len(list[i].ApiKey) > 8 {
|
||||
list[i].ApiKey = list[i].ApiKey[:4] + "****" + list[i].ApiKey[len(list[i].ApiKey)-4:]
|
||||
}
|
||||
list[i].ModelsList = parseProviderModels(list[i].Models)
|
||||
}
|
||||
|
||||
@@ -171,6 +169,7 @@ func (c *BackendAiProviderController) Create() {
|
||||
ApiBase: strings.TrimSpace(p.ApiBase),
|
||||
ApiKey: strings.TrimSpace(p.ApiKey),
|
||||
Models: string(modelsJSON),
|
||||
DefaultModel: strings.TrimSpace(p.DefaultModel),
|
||||
IsDefault: p.IsDefault,
|
||||
Status: p.Status,
|
||||
Remark: strings.TrimSpace(p.Remark),
|
||||
@@ -240,6 +239,7 @@ func (c *BackendAiProviderController) Update() {
|
||||
modelsJSON, _ := json.Marshal(p.Models)
|
||||
provider.Models = string(modelsJSON)
|
||||
}
|
||||
provider.DefaultModel = strings.TrimSpace(p.DefaultModel)
|
||||
provider.IsDefault = p.IsDefault
|
||||
provider.Status = p.Status
|
||||
provider.Remark = strings.TrimSpace(p.Remark)
|
||||
|
||||
@@ -4,6 +4,8 @@ import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"math"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
@@ -53,6 +55,75 @@ type smartGeneratePayload struct {
|
||||
Type string `json:"type"` // supplier / customer
|
||||
}
|
||||
|
||||
// extractJSONFromText 从文本中提取JSON对象（查找 { 和 } 之间的内容）
|
||||
func extractJSONFromText(text string) string {
|
||||
text = strings.TrimSpace(text)
|
||||
|
||||
// 找到第一个 { 和最后一个 }
|
||||
start := strings.Index(text, "{")
|
||||
end := strings.LastIndex(text, "}")
|
||||
|
||||
if start == -1 || end == -1 || end <= start {
|
||||
return ""
|
||||
}
|
||||
|
||||
// 提取可能的JSON字符串
|
||||
jsonStr := text[start : end+1]
|
||||
|
||||
// 验证是否是有效的JSON（允许额外内容）
|
||||
var test map[string]interface{}
|
||||
if err := json.Unmarshal([]byte(jsonStr), &test); err == nil {
|
||||
return jsonStr
|
||||
}
|
||||
|
||||
// 尝试更宽松的提取：从第一个 { 开始找完整的JSON结构
|
||||
return findCompleteJSON(text, start)
|
||||
}
|
||||
|
||||
// findCompleteJSON 从指定位置开始查找完整的JSON对象
|
||||
func findCompleteJSON(text string, startPos int) string {
|
||||
if startPos == -1 || startPos >= len(text) {
|
||||
return ""
|
||||
}
|
||||
|
||||
stack := 0
|
||||
start := -1
|
||||
|
||||
for i := startPos; i < len(text); i++ {
|
||||
ch := text[i]
|
||||
|
||||
switch ch {
|
||||
case '{':
|
||||
if stack == 0 {
|
||||
start = i
|
||||
}
|
||||
stack++
|
||||
case '}':
|
||||
stack--
|
||||
if stack == 0 && start != -1 {
|
||||
// 找到完整的JSON对象
|
||||
jsonStr := text[start : i+1]
|
||||
// 验证是否是有效JSON
|
||||
var test map[string]interface{}
|
||||
if err := json.Unmarshal([]byte(jsonStr), &test); err == nil {
|
||||
return jsonStr
|
||||
}
|
||||
}
|
||||
case '"':
|
||||
// 跳过字符串中的括号
|
||||
i++
|
||||
for i < len(text) && text[i] != '"' {
|
||||
if text[i] == '\\' {
|
||||
i++ // 跳过转义字符
|
||||
}
|
||||
i++
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
// GenerateCompany POST /backend/ai/smart-generate/company
|
||||
// 根据公司名称智能生成详细信息
|
||||
func (c *BackendAiSmartGenerateController) GenerateCompany() {
|
||||
@@ -97,26 +168,53 @@ func (c *BackendAiSmartGenerateController) GenerateCompany() {
|
||||
}
|
||||
provider := providers[0]
|
||||
|
||||
// 确定使用的模型（第一个）
|
||||
// 确定使用的模型
|
||||
providerModels := parseProviderModels(provider.Models)
|
||||
if len(providerModels) == 0 {
|
||||
c.sgJsonErr(400, 400, "默认接入配置没有可用模型")
|
||||
return
|
||||
}
|
||||
|
||||
// 优先使用默认模型，如果没有设置则使用第一个模型
|
||||
useModel := providerModels[0]
|
||||
if provider.DefaultModel != "" {
|
||||
// 检查default_model是否在models列表中
|
||||
for _, m := range providerModels {
|
||||
if m == provider.DefaultModel {
|
||||
useModel = m
|
||||
break
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 调试：打印使用的模型（使用fmt替代beego.Info）
|
||||
fmt.Printf("SmartGenerate: Using model=%s, default_model=%s, tenant=%d\n", useModel, provider.DefaultModel, claims.TenantId)
|
||||
|
||||
// 构造prompt
|
||||
typeLabel := "供应商"
|
||||
nameField := "supplier_name"
|
||||
typeField := "supplier_type"
|
||||
typeAllowed := "原材料供应商/设备供应商/服务供应商/其他"
|
||||
if p.Type == "customer" {
|
||||
typeLabel = "客户"
|
||||
nameField = "customer_name"
|
||||
typeField = "customer_type"
|
||||
typeAllowed = "企业/政府机构/国企/教育机构/个人"
|
||||
}
|
||||
|
||||
prompt := fmt.Sprintf(`你是一个企业信息助手。请根据%s名称"%s"，生成该公司的详细信息，以JSON格式返回，包含以下字段：
|
||||
prompt := fmt.Sprintf(`你是一个企业信息助手。请根据%s名称"%s"，先用可用的MCP工具（如天眼查 search_companies / get_company_basic_profile 等）查询该企业的真实信息，然后输出以下两部分内容：
|
||||
|
||||
【第一部分：企业画像（默认展示内容）】
|
||||
用清晰的结构化markdown输出该企业画像，包含：基础工商信息（法定代表人、注册资本、成立日期、登记状态、企业类型、登记机关、注册地址、所属行业）、规模与人员（人员规模、参保人数）、企业标签（如高新技术企业/专精特新等）、经营范围（节选）、经营概况（对外投资/招投标/商标/专利数量）、风险提示（司法案件/合作风险等）。有真实数据的填真实数据，查询不到的字段注明"暂无"，不要编造数据。
|
||||
|
||||
【第二部分：结构化JSON（供系统自动填充）】
|
||||
在回复最后输出一个【合法的JSON对象】，必须以 { 开头、以 } 结尾，键和字符串值都要用英文双引号，键值之间用英文冒号，键值对之间用英文逗号。例如：
|
||||
{"customer_name":"连云港如年实业有限公司","contact_person":"张昊","registered_capital":"6000万元"}
|
||||
包含以下字段（每个字段一个键值，查不到就留空字符串）：
|
||||
- %s: 公司全称
|
||||
- %s: %s类型（必须从以下取值中选一个：%s）
|
||||
- contact_person: 联系人
|
||||
- contact_phone: 联系电话
|
||||
- contact_phone: 联系电话（务必从查询结果中提取真实电话，缺失时与注册电话互相补齐，不要编造号码）
|
||||
- contact_email: 邮箱
|
||||
- address: 公司地址
|
||||
- industry: 所属行业
|
||||
@@ -132,53 +230,337 @@ func (c *BackendAiSmartGenerateController) GenerateCompany() {
|
||||
- bank_name: 开户银行
|
||||
- bank_account: 银行账号
|
||||
- registered_address: 注册地址
|
||||
- registered_phone: 注册电话
|
||||
- registered_phone: 注册电话（与联系电话互相补齐，缺失时用联系电话，不要编造号码）
|
||||
- remark: 备注
|
||||
|
||||
请只返回JSON对象，不要返回其他任何文字、解释或markdown代码块标记。如果某些信息不确定，请留空字符串。`, typeLabel, p.CompanyName, nameField)
|
||||
注意：JSON对象必须放在回复最后且用 { } 包裹，不要用markdown代码块包裹，不要输出"customer_name: xxx"这类非JSON格式。`, typeLabel, p.CompanyName, nameField, typeField, typeLabel, typeAllowed)
|
||||
|
||||
messages := []openaiMessage{
|
||||
{Role: "user", Content: prompt},
|
||||
}
|
||||
|
||||
// 调用AI（非流式，超时120s）
|
||||
reply, err := callAI(provider, useModel, "", messages)
|
||||
// 调用AI（非流式，含已启用 MCP 工具：AI 会先调用天眼查等工具获取真实数据再生成）
|
||||
reply, err := runToolLoop(claims, provider, useModel, "", messages)
|
||||
if err != nil {
|
||||
c.sgJsonErr(500, 500, "AI生成失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
// 解析AI返回的JSON
|
||||
reply = strings.TrimSpace(reply)
|
||||
// 去掉可能的markdown代码块标记
|
||||
reply = strings.TrimPrefix(reply, "```json")
|
||||
reply = strings.TrimPrefix(reply, "```")
|
||||
reply = strings.TrimSuffix(reply, "```")
|
||||
// 解析AI返回的JSON（前面的markdown企业画像作为raw返回给前端展示）
|
||||
reply = strings.TrimSpace(reply)
|
||||
|
||||
var result map[string]interface{}
|
||||
if err := json.Unmarshal([]byte(reply), &result); err != nil {
|
||||
// 解析失败，返回原始文本让用户自己处理
|
||||
c.sgOk(map[string]interface{}{
|
||||
"raw": reply,
|
||||
"parsed": false,
|
||||
"name": p.CompanyName,
|
||||
"type": p.Type,
|
||||
"generate_time": time.Now().Format("2006-01-02 15:04:05"),
|
||||
})
|
||||
return
|
||||
var jsonStr string
|
||||
|
||||
// 方法1: 尝试直接解析整个响应
|
||||
if err := json.Unmarshal([]byte(reply), &result); err == nil {
|
||||
jsonStr = reply
|
||||
} else {
|
||||
// 方法2: 从回复中提取最后一个完整JSON对象（结构化字段在回复末尾）
|
||||
jsonStr = extractLastJSON(reply)
|
||||
if jsonStr == "" {
|
||||
// 方法3: 兜底——模型可能输出 "字段名: 值" 的非JSON格式文本，逐行摘取到对应字段
|
||||
result = parseSmartGenKV(reply, nameField)
|
||||
if len(result) == 0 {
|
||||
// 全部失败，返回原始文本让用户自己处理
|
||||
c.sgOk(map[string]interface{}{
|
||||
"raw": reply,
|
||||
"parsed": false,
|
||||
"error": "AI返回格式异常，无法解析JSON",
|
||||
"name": p.CompanyName,
|
||||
"type": p.Type,
|
||||
"generate_time": time.Now().Format("2006-01-02 15:04:05"),
|
||||
})
|
||||
return
|
||||
}
|
||||
// KV兜底成功，继续走后面的结果组装
|
||||
c.sgOk(smartGenResult(reply, result, p.CompanyName, p.Type, nameField))
|
||||
return
|
||||
}
|
||||
|
||||
if err := json.Unmarshal([]byte(jsonStr), &result); err != nil {
|
||||
c.sgOk(map[string]interface{}{
|
||||
"raw": reply,
|
||||
"parsed": false,
|
||||
"error": fmt.Sprintf("解析JSON失败: %v", err),
|
||||
"name": p.CompanyName,
|
||||
"type": p.Type,
|
||||
"generate_time": time.Now().Format("2006-01-02 15:04:05"),
|
||||
})
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
c.sgOk(smartGenResult(reply, result, p.CompanyName, p.Type, nameField))
|
||||
}
|
||||
|
||||
// smartGenResult 组装智能生成成功响应（含完整AI回复 raw），并对表单受限字段做规整
|
||||
func smartGenResult(reply string, result map[string]interface{}, companyName, genType, nameField string) map[string]interface{} {
|
||||
// 确保公司名称字段存在
|
||||
if _, ok := result[nameField]; !ok || result[nameField] == "" {
|
||||
result[nameField] = p.CompanyName
|
||||
result[nameField] = companyName
|
||||
}
|
||||
|
||||
c.sgOk(map[string]interface{}{
|
||||
// 规整受限字段，使其符合新增客户/供应商表单的下拉选项
|
||||
normalizeStrField(result, "enterprise_type", normalizeEnterpriseType)
|
||||
normalizeStrField(result, "industry", normalizeIndustry)
|
||||
normalizeStrField(result, "taxpayer_qualification", normalizeTaxpayer)
|
||||
// 注册资本/实缴资本规整为纯数值（单位万元），与表单"万元"后缀一致
|
||||
normalizeStrField(result, "registered_capital", normalizeCapital)
|
||||
normalizeStrField(result, "paid_capital", normalizeCapital)
|
||||
// 客户/供应商类型规整为表单下拉选项代码
|
||||
if genType == "supplier" {
|
||||
normalizeStrField(result, "supplier_type", normalizeSupplierType)
|
||||
} else {
|
||||
normalizeStrField(result, "customer_type", normalizeCustomerType)
|
||||
}
|
||||
// 联系电话与注册电话互相补齐，尽量保证有电话可填（不编造）
|
||||
crossFillPhone(result, "contact_phone", "registered_phone")
|
||||
return map[string]interface{}{
|
||||
"parsed": true,
|
||||
"data": result,
|
||||
"name": p.CompanyName,
|
||||
"type": p.Type,
|
||||
"raw": reply, // AI完整回复（含企业画像markdown，供前端"AI响应数据"展示）
|
||||
"name": companyName,
|
||||
"type": genType,
|
||||
"generate_time": time.Now().Format("2006-01-02 15:04:05"),
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
// normalizeStrField 对 result 中的字符串字段应用规整函数
|
||||
func normalizeStrField(result map[string]interface{}, key string, fn func(string) string) {
|
||||
if v, ok := result[key]; ok {
|
||||
if s, ok2 := v.(string); ok2 {
|
||||
result[key] = fn(s)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// crossFillPhone 两个字段有任一非空时互相补齐（如 联系电话 缺则用 注册电话，反之亦然）
|
||||
func crossFillPhone(result map[string]interface{}, a, b string) {
|
||||
av, aOK := result[a].(string)
|
||||
bv, bOK := result[b].(string)
|
||||
if aOK && strings.TrimSpace(av) == "" && bOK && strings.TrimSpace(bv) != "" {
|
||||
result[a] = bv
|
||||
}
|
||||
if bOK && strings.TrimSpace(bv) == "" && aOK && strings.TrimSpace(av) != "" {
|
||||
result[b] = av
|
||||
}
|
||||
}
|
||||
|
||||
// normalizeEnterpriseType 将AI返回的企业类型规整为新增表单下拉选项之一
|
||||
func normalizeEnterpriseType(raw string) string {
|
||||
s := strings.TrimSpace(raw)
|
||||
// 去掉括号及其内容，如 "有限责任公司(自然人投资或控股)" -> "有限责任公司"
|
||||
if i := strings.IndexAny(s, "（("); i > 0 {
|
||||
s = strings.TrimSpace(s[:i])
|
||||
}
|
||||
switch {
|
||||
case strings.Contains(s, "有限责任"):
|
||||
return "有限责任公司"
|
||||
case strings.Contains(s, "股份"):
|
||||
return "股份有限公司"
|
||||
case strings.Contains(s, "合伙"):
|
||||
return "合伙企业"
|
||||
case strings.Contains(s, "个人独资"):
|
||||
return "个人独资企业"
|
||||
case strings.Contains(s, "国有"):
|
||||
return "国有企业"
|
||||
case strings.Contains(s, "集体"):
|
||||
return "集体企业"
|
||||
case strings.Contains(s, "外商"):
|
||||
return "外商投资企业"
|
||||
case s == "其他":
|
||||
return "其他"
|
||||
case s == "":
|
||||
return ""
|
||||
default:
|
||||
return "其他"
|
||||
}
|
||||
}
|
||||
|
||||
// normalizeIndustry 规整行业：去掉"大行业>细分行业"路径，取最后一段
|
||||
func normalizeIndustry(raw string) string {
|
||||
s := strings.TrimSpace(raw)
|
||||
if i := strings.LastIndex(s, ">"); i >= 0 && i < len(s)-1 {
|
||||
return strings.TrimSpace(s[i+1:])
|
||||
}
|
||||
return s
|
||||
}
|
||||
|
||||
// normalizeTaxpayer 将AI返回的纳税人资质规整为下拉选项之一
|
||||
func normalizeTaxpayer(raw string) string {
|
||||
s := strings.TrimSpace(raw)
|
||||
switch {
|
||||
case strings.Contains(s, "一般"):
|
||||
return "一般纳税人"
|
||||
case strings.Contains(s, "小规模"):
|
||||
return "小规模纳税人"
|
||||
case s == "其他":
|
||||
return "其他"
|
||||
case s == "":
|
||||
return ""
|
||||
default:
|
||||
return "其他"
|
||||
}
|
||||
}
|
||||
|
||||
// normalizeCustomerType 将AI返回的客户类型规整为下拉选项代码(1企业/2政府机构/3国企/4教育机构/5个人)
|
||||
func normalizeCustomerType(raw string) string {
|
||||
s := strings.TrimSpace(raw)
|
||||
switch {
|
||||
case s == "1" || s == "2" || s == "3" || s == "4" || s == "5":
|
||||
return s
|
||||
case strings.Contains(s, "政府"):
|
||||
return "2"
|
||||
case strings.Contains(s, "教育") || strings.Contains(s, "学校"):
|
||||
return "4"
|
||||
case strings.Contains(s, "个人") || strings.Contains(s, "个体"):
|
||||
return "5"
|
||||
case strings.Contains(s, "国企") || strings.Contains(s, "国有"):
|
||||
return "3"
|
||||
case s == "":
|
||||
return ""
|
||||
default:
|
||||
return "1" // 企业
|
||||
}
|
||||
}
|
||||
|
||||
// normalizeSupplierType 将AI返回的供应商类型规整为下拉选项代码(1原材料/2设备/3服务/4其他)
|
||||
func normalizeSupplierType(raw string) string {
|
||||
s := strings.TrimSpace(raw)
|
||||
switch {
|
||||
case s == "1" || s == "2" || s == "3" || s == "4":
|
||||
return s
|
||||
case strings.Contains(s, "原材料"):
|
||||
return "1"
|
||||
case strings.Contains(s, "设备"):
|
||||
return "2"
|
||||
case strings.Contains(s, "服务"):
|
||||
return "3"
|
||||
case s == "":
|
||||
return ""
|
||||
default:
|
||||
return "4" // 其他
|
||||
}
|
||||
}
|
||||
|
||||
// normalizeCapital 将AI返回的注册资本/实缴资本规整为纯数值（单位万元），与表单"万元"后缀一致。
|
||||
// 例: "6000万人民币"->"6000", "6000.00万"->"6000", "1亿"->"10000", "5000万元"->"5000"
|
||||
func normalizeCapital(raw string) string {
|
||||
s := strings.TrimSpace(raw)
|
||||
if s == "" {
|
||||
return ""
|
||||
}
|
||||
// 去掉千分位逗号
|
||||
s = strings.ReplaceAll(s, ",", "")
|
||||
s = strings.ReplaceAll(s, "，", "")
|
||||
// 提取数字部分（含小数与可能的负号）
|
||||
i := 0
|
||||
if len(s) > 0 && s[0] == '-' {
|
||||
i = 1
|
||||
}
|
||||
for i < len(s) && ((s[i] >= '0' && s[i] <= '9') || s[i] == '.') {
|
||||
i++
|
||||
}
|
||||
numStr := s[:i]
|
||||
if numStr == "" || numStr == "." || numStr == "-" {
|
||||
return ""
|
||||
}
|
||||
num, err := strconv.ParseFloat(numStr, 64)
|
||||
if err != nil {
|
||||
return ""
|
||||
}
|
||||
// 判断数字后的单位，统一换算为万元
|
||||
rest := strings.TrimSpace(s[i:])
|
||||
if strings.HasPrefix(rest, "亿") {
|
||||
num *= 10000
|
||||
} else if strings.HasPrefix(rest, "千") {
|
||||
num *= 0.1
|
||||
}
|
||||
// 格式化：整数去小数位，非整数保留有效小数
|
||||
if num == math.Trunc(num) {
|
||||
return strconv.FormatFloat(num, 'f', 0, 64)
|
||||
}
|
||||
return strconv.FormatFloat(num, 'f', -1, 64)
|
||||
}
|
||||
|
||||
// smartGenAllowedFields 智能生成JSON允许的字段名
|
||||
var smartGenAllowedFields = map[string]bool{
|
||||
"customer_name": true, "supplier_name": true, "contact_person": true,
|
||||
"contact_phone": true, "contact_email": true, "address": true,
|
||||
"industry": true, "registered_capital": true, "paid_capital": true,
|
||||
"establish_date": true, "administrative_division": true, "enterprise_type": true,
|
||||
"taxpayer_qualification": true, "business_scope": true, "invoice_title": true,
|
||||
"tax_number": true, "bank_name": true, "bank_account": true,
|
||||
"registered_address": true, "registered_phone": true, "remark": true,
|
||||
"customer_type": true, "supplier_type": true,
|
||||
}
|
||||
|
||||
// parseSmartGenKV 兜底解析：当模型输出 "字段名: 值" 的非JSON文本时，逐行摘取到对应字段
|
||||
func parseSmartGenKV(text, nameField string) map[string]interface{} {
|
||||
result := make(map[string]interface{})
|
||||
lines := strings.Split(text, "\n")
|
||||
for _, line := range lines {
|
||||
line = strings.TrimSpace(line)
|
||||
// 去掉可能的列表/引用前缀
|
||||
line = strings.TrimLeft(line, "-*|#> `")
|
||||
line = strings.TrimSpace(line)
|
||||
idx := strings.Index(line, ":")
|
||||
if idx <= 0 {
|
||||
continue
|
||||
}
|
||||
key := strings.Trim(strings.TrimSpace(line[:idx]), "\"'`")
|
||||
if !smartGenAllowedFields[key] && key != nameField {
|
||||
continue
|
||||
}
|
||||
val := strings.TrimSpace(line[idx+1:])
|
||||
val = strings.Trim(val, "\"'`")
|
||||
val = strings.TrimSpace(val)
|
||||
if val == "" || val == "null" || val == "NULL" || val == "undefined" || val == "暂无" {
|
||||
val = ""
|
||||
}
|
||||
result[key] = val
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
// extractLastJSON 从文本中提取最后一个完整的JSON对象（忽略markdown等前置内容）
|
||||
func extractLastJSON(text string) string {
|
||||
var lastValid string
|
||||
depth := 0
|
||||
start := -1
|
||||
inStr := false
|
||||
escaped := false
|
||||
for i := 0; i < len(text); i++ {
|
||||
ch := text[i]
|
||||
if inStr {
|
||||
if escaped {
|
||||
escaped = false
|
||||
} else if ch == '\\' {
|
||||
escaped = true
|
||||
} else if ch == '"' {
|
||||
inStr = false
|
||||
}
|
||||
continue
|
||||
}
|
||||
switch ch {
|
||||
case '"':
|
||||
inStr = true
|
||||
case '{':
|
||||
if depth == 0 {
|
||||
start = i
|
||||
}
|
||||
depth++
|
||||
case '}':
|
||||
depth--
|
||||
if depth == 0 && start != -1 {
|
||||
cand := text[start : i+1]
|
||||
var test map[string]interface{}
|
||||
if json.Unmarshal([]byte(cand), &test) == nil {
|
||||
lastValid = cand
|
||||
}
|
||||
start = -1
|
||||
}
|
||||
}
|
||||
}
|
||||
return lastValid
|
||||
}
|
||||
|
||||
@@ -0,0 +1,506 @@
|
||||
package controllers
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"server/models"
|
||||
"server/pkg/jwtutil"
|
||||
"server/services"
|
||||
|
||||
beego "github.com/beego/beego/v2/server/web"
|
||||
)
|
||||
|
||||
// BackendMcpServerController MCP服务器配置控制器
|
||||
type BackendMcpServerController struct {
|
||||
beego.Controller
|
||||
}
|
||||
|
||||
func (c *BackendMcpServerController) mcpSrvClaims() (*jwtutil.Claims, error) {
|
||||
auth := c.Ctx.Request.Header.Get("Authorization")
|
||||
if auth == "" {
|
||||
return nil, fmt.Errorf("未登录")
|
||||
}
|
||||
parts := strings.SplitN(auth, " ", 2)
|
||||
if len(parts) != 2 || parts[0] != "Bearer" {
|
||||
return nil, fmt.Errorf("认证信息格式错误")
|
||||
}
|
||||
claims, err := jwtutil.ParseToken(parts[1])
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("无效的token")
|
||||
}
|
||||
if claims.UserType != "backend" {
|
||||
return nil, fmt.Errorf("无权访问")
|
||||
}
|
||||
return claims, nil
|
||||
}
|
||||
|
||||
func (c *BackendMcpServerController) mcpSrvErr(httpStatus, bizCode int, msg string) {
|
||||
c.Ctx.Output.SetStatus(httpStatus)
|
||||
c.Data["json"] = map[string]interface{}{"code": bizCode, "msg": msg}
|
||||
_ = c.ServeJSON()
|
||||
}
|
||||
|
||||
func (c *BackendMcpServerController) mcpSrvOk(data interface{}) {
|
||||
c.Data["json"] = map[string]interface{}{"code": 200, "msg": "success", "data": data}
|
||||
_ = c.ServeJSON()
|
||||
}
|
||||
|
||||
type mcpServerPayload struct {
|
||||
Name string `json:"name"`
|
||||
Transport string `json:"transport"`
|
||||
Command string `json:"command"`
|
||||
Args string `json:"args"`
|
||||
Env string `json:"env"`
|
||||
URL string `json:"url"`
|
||||
Headers string `json:"headers"`
|
||||
Description string `json:"description"`
|
||||
Provider string `json:"provider"`
|
||||
FromMarket string `json:"from_market"`
|
||||
Enabled int8 `json:"enabled"`
|
||||
Remark string `json:"remark"`
|
||||
}
|
||||
|
||||
func (p *mcpServerPayload) validate() error {
|
||||
if strings.TrimSpace(p.Name) == "" {
|
||||
return fmt.Errorf("服务名称不能为空")
|
||||
}
|
||||
p.Transport = strings.ToLower(strings.TrimSpace(p.Transport))
|
||||
if p.Transport == "" {
|
||||
p.Transport = "http"
|
||||
}
|
||||
switch p.Transport {
|
||||
case "stdio":
|
||||
if strings.TrimSpace(p.Command) == "" {
|
||||
return fmt.Errorf("stdio 传输必须填写启动命令")
|
||||
}
|
||||
case "http", "sse":
|
||||
if strings.TrimSpace(p.URL) == "" {
|
||||
return fmt.Errorf("请填写服务地址 URL")
|
||||
}
|
||||
default:
|
||||
return fmt.Errorf("不支持的传输类型: %s（可选 stdio/http/sse）")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (p *mcpServerPayload) toModel(claims *jwtutil.Claims) models.BackendMcpServer {
|
||||
return models.BackendMcpServer{
|
||||
TenantID: fmt.Sprintf("%d", claims.TenantId),
|
||||
UserID: uint64(claims.UserID),
|
||||
Name: strings.TrimSpace(p.Name),
|
||||
Transport: p.Transport,
|
||||
Command: strings.TrimSpace(p.Command),
|
||||
Args: strings.TrimSpace(p.Args),
|
||||
Env: strings.TrimSpace(p.Env),
|
||||
URL: strings.TrimSpace(p.URL),
|
||||
Headers: strings.TrimSpace(p.Headers),
|
||||
Description: strings.TrimSpace(p.Description),
|
||||
Provider: strings.TrimSpace(p.Provider),
|
||||
FromMarket: strings.TrimSpace(p.FromMarket),
|
||||
Enabled: p.Enabled,
|
||||
Remark: strings.TrimSpace(p.Remark),
|
||||
CreateTime: time.Now(),
|
||||
UpdateTime: time.Now(),
|
||||
}
|
||||
}
|
||||
|
||||
// List GET /backend/mcp/server/list
|
||||
func (c *BackendMcpServerController) List() {
|
||||
claims, err := c.mcpSrvClaims()
|
||||
if err != nil {
|
||||
c.mcpSrvErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
var list []models.BackendMcpServer
|
||||
_, err = models.Orm.QueryTable(new(models.BackendMcpServer)).
|
||||
Filter("tenant_id", fmt.Sprintf("%d", claims.TenantId)).
|
||||
Filter("user_id", uint64(claims.UserID)).
|
||||
Filter("delete_time__isnull", true).
|
||||
OrderBy("-id").
|
||||
All(&list)
|
||||
if err != nil {
|
||||
c.mcpSrvErr(500, 500, "查询失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
c.mcpSrvOk(map[string]interface{}{"list": list})
|
||||
}
|
||||
|
||||
// Create POST /backend/mcp/server
|
||||
func (c *BackendMcpServerController) Create() {
|
||||
claims, err := c.mcpSrvClaims()
|
||||
if err != nil {
|
||||
c.mcpSrvErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
body, err := io.ReadAll(c.Ctx.Request.Body)
|
||||
if err != nil {
|
||||
c.mcpSrvErr(400, 400, "读取请求体失败")
|
||||
return
|
||||
}
|
||||
var p mcpServerPayload
|
||||
if err := json.Unmarshal(body, &p); err != nil {
|
||||
c.mcpSrvErr(400, 400, "参数格式错误")
|
||||
return
|
||||
}
|
||||
if err := p.validate(); err != nil {
|
||||
c.mcpSrvErr(400, 400, err.Error())
|
||||
return
|
||||
}
|
||||
m := p.toModel(claims)
|
||||
id, err := models.Orm.Insert(&m)
|
||||
if err != nil {
|
||||
c.mcpSrvErr(500, 500, "创建失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
c.mcpSrvOk(map[string]interface{}{"id": id})
|
||||
}
|
||||
|
||||
// Update PUT /backend/mcp/server/:id
|
||||
func (c *BackendMcpServerController) Update() {
|
||||
claims, err := c.mcpSrvClaims()
|
||||
if err != nil {
|
||||
c.mcpSrvErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
idStr := c.Ctx.Input.Param(":id")
|
||||
id, err := strconv.ParseUint(idStr, 10, 64)
|
||||
if err != nil {
|
||||
c.mcpSrvErr(400, 400, "ID格式错误")
|
||||
return
|
||||
}
|
||||
srv := models.BackendMcpServer{ID: id}
|
||||
if err := models.Orm.Read(&srv); err != nil {
|
||||
c.mcpSrvErr(404, 404, "服务不存在")
|
||||
return
|
||||
}
|
||||
if srv.TenantID != fmt.Sprintf("%d", claims.TenantId) || srv.UserID != uint64(claims.UserID) {
|
||||
c.mcpSrvErr(403, 403, "无权操作")
|
||||
return
|
||||
}
|
||||
body, err := io.ReadAll(c.Ctx.Request.Body)
|
||||
if err != nil {
|
||||
c.mcpSrvErr(400, 400, "读取请求体失败")
|
||||
return
|
||||
}
|
||||
var p mcpServerPayload
|
||||
if err := json.Unmarshal(body, &p); err != nil {
|
||||
c.mcpSrvErr(400, 400, "参数格式错误")
|
||||
return
|
||||
}
|
||||
if err := p.validate(); err != nil {
|
||||
c.mcpSrvErr(400, 400, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
srv.Name = strings.TrimSpace(p.Name)
|
||||
srv.Transport = p.Transport
|
||||
srv.Command = strings.TrimSpace(p.Command)
|
||||
srv.Args = strings.TrimSpace(p.Args)
|
||||
srv.Env = strings.TrimSpace(p.Env)
|
||||
srv.URL = strings.TrimSpace(p.URL)
|
||||
srv.Headers = strings.TrimSpace(p.Headers)
|
||||
srv.Description = strings.TrimSpace(p.Description)
|
||||
srv.Provider = strings.TrimSpace(p.Provider)
|
||||
srv.FromMarket = strings.TrimSpace(p.FromMarket)
|
||||
srv.Enabled = p.Enabled
|
||||
srv.Remark = strings.TrimSpace(p.Remark)
|
||||
srv.UpdateTime = time.Now()
|
||||
|
||||
_, err = models.Orm.Update(&srv)
|
||||
if err != nil {
|
||||
c.mcpSrvErr(500, 500, "更新失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
// 配置变更后断开旧连接，下次使用时按新配置重连
|
||||
services.McpClientManager.Disconnect(srv.ID)
|
||||
c.mcpSrvOk(nil)
|
||||
}
|
||||
|
||||
// Delete DELETE /backend/mcp/server/:id
|
||||
func (c *BackendMcpServerController) Delete() {
|
||||
claims, err := c.mcpSrvClaims()
|
||||
if err != nil {
|
||||
c.mcpSrvErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
idStr := c.Ctx.Input.Param(":id")
|
||||
id, err := strconv.ParseUint(idStr, 10, 64)
|
||||
if err != nil {
|
||||
c.mcpSrvErr(400, 400, "ID格式错误")
|
||||
return
|
||||
}
|
||||
srv := models.BackendMcpServer{ID: id}
|
||||
if err := models.Orm.Read(&srv); err != nil {
|
||||
c.mcpSrvErr(404, 404, "服务不存在")
|
||||
return
|
||||
}
|
||||
if srv.TenantID != fmt.Sprintf("%d", claims.TenantId) || srv.UserID != uint64(claims.UserID) {
|
||||
c.mcpSrvErr(403, 403, "无权操作")
|
||||
return
|
||||
}
|
||||
now := time.Now()
|
||||
srv.DeleteTime = &now
|
||||
if _, err := models.Orm.Update(&srv, "delete_time"); err != nil {
|
||||
c.mcpSrvErr(500, 500, "删除失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
services.McpClientManager.Disconnect(srv.ID)
|
||||
c.mcpSrvOk(nil)
|
||||
}
|
||||
|
||||
// Toggle PUT /backend/mcp/server/:id/toggle
|
||||
func (c *BackendMcpServerController) Toggle() {
|
||||
claims, err := c.mcpSrvClaims()
|
||||
if err != nil {
|
||||
c.mcpSrvErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
idStr := c.Ctx.Input.Param(":id")
|
||||
id, err := strconv.ParseUint(idStr, 10, 64)
|
||||
if err != nil {
|
||||
c.mcpSrvErr(400, 400, "ID格式错误")
|
||||
return
|
||||
}
|
||||
srv := models.BackendMcpServer{ID: id}
|
||||
if err := models.Orm.Read(&srv); err != nil {
|
||||
c.mcpSrvErr(404, 404, "服务不存在")
|
||||
return
|
||||
}
|
||||
if srv.TenantID != fmt.Sprintf("%d", claims.TenantId) || srv.UserID != uint64(claims.UserID) {
|
||||
c.mcpSrvErr(403, 403, "无权操作")
|
||||
return
|
||||
}
|
||||
body, err := io.ReadAll(c.Ctx.Request.Body)
|
||||
if err != nil {
|
||||
c.mcpSrvErr(400, 400, "读取请求体失败")
|
||||
return
|
||||
}
|
||||
var req struct {
|
||||
Enabled *int8 `json:"enabled"`
|
||||
}
|
||||
if len(body) > 0 {
|
||||
_ = json.Unmarshal(body, &req)
|
||||
}
|
||||
if req.Enabled != nil {
|
||||
srv.Enabled = *req.Enabled
|
||||
} else if srv.Enabled == 1 {
|
||||
srv.Enabled = 0
|
||||
} else {
|
||||
srv.Enabled = 1
|
||||
}
|
||||
srv.UpdateTime = time.Now()
|
||||
if _, err := models.Orm.Update(&srv, "enabled", "update_time"); err != nil {
|
||||
c.mcpSrvErr(500, 500, "更新失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
c.mcpSrvOk(map[string]interface{}{"enabled": srv.Enabled})
|
||||
}
|
||||
|
||||
// Test POST /backend/mcp/server/test
|
||||
// 使用服务器配置新建连接并列出工具（不进缓存），同时回写连接状态
|
||||
func (c *BackendMcpServerController) Test() {
|
||||
claims, err := c.mcpSrvClaims()
|
||||
if err != nil {
|
||||
c.mcpSrvErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
body, err := io.ReadAll(c.Ctx.Request.Body)
|
||||
if err != nil {
|
||||
c.mcpSrvErr(400, 400, "读取请求体失败")
|
||||
return
|
||||
}
|
||||
var req struct {
|
||||
ID uint64 `json:"id"`
|
||||
}
|
||||
_ = json.Unmarshal(body, &req)
|
||||
|
||||
var srv models.BackendMcpServer
|
||||
if req.ID > 0 {
|
||||
srv = models.BackendMcpServer{ID: req.ID}
|
||||
if err := models.Orm.Read(&srv); err != nil {
|
||||
c.mcpSrvErr(404, 404, "服务不存在")
|
||||
return
|
||||
}
|
||||
if srv.TenantID != fmt.Sprintf("%d", claims.TenantId) || srv.UserID != uint64(claims.UserID) {
|
||||
c.mcpSrvErr(403, 403, "无权操作")
|
||||
return
|
||||
}
|
||||
} else {
|
||||
// 未保存的配置直接测试
|
||||
var p mcpServerPayload
|
||||
if err := json.Unmarshal(body, &p); err != nil {
|
||||
c.mcpSrvErr(400, 400, "参数格式错误")
|
||||
return
|
||||
}
|
||||
if err := p.validate(); err != nil {
|
||||
c.mcpSrvErr(400, 400, err.Error())
|
||||
return
|
||||
}
|
||||
srv = p.toModel(claims)
|
||||
}
|
||||
|
||||
tools, err := services.McpClientManager.TestConnection(&srv)
|
||||
if err != nil {
|
||||
// 回写失败状态
|
||||
if req.ID > 0 {
|
||||
srv.Status = 2
|
||||
srv.LastError = err.Error()
|
||||
srv.ToolCount = 0
|
||||
srv.UpdateTime = time.Now()
|
||||
_, _ = models.Orm.Update(&srv, "status", "last_error", "tool_count", "update_time")
|
||||
}
|
||||
c.mcpSrvErr(400, 400, "连接失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
// 回写成功状态
|
||||
if req.ID > 0 {
|
||||
srv.Status = 1
|
||||
srv.LastError = ""
|
||||
srv.ToolCount = len(tools)
|
||||
srv.UpdateTime = time.Now()
|
||||
_, _ = models.Orm.Update(&srv, "status", "last_error", "tool_count", "update_time")
|
||||
}
|
||||
|
||||
c.mcpSrvOk(map[string]interface{}{
|
||||
"success": true,
|
||||
"tools": tools,
|
||||
"count": len(tools),
|
||||
})
|
||||
}
|
||||
|
||||
// Tools POST /backend/mcp/server/:id/tools
|
||||
// 使用缓存连接列出工具（供会话注入与前端查看）
|
||||
func (c *BackendMcpServerController) Tools() {
|
||||
claims, err := c.mcpSrvClaims()
|
||||
if err != nil {
|
||||
c.mcpSrvErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
idStr := c.Ctx.Input.Param(":id")
|
||||
id, err := strconv.ParseUint(idStr, 10, 64)
|
||||
if err != nil {
|
||||
c.mcpSrvErr(400, 400, "ID格式错误")
|
||||
return
|
||||
}
|
||||
srv := models.BackendMcpServer{ID: id}
|
||||
if err := models.Orm.Read(&srv); err != nil {
|
||||
c.mcpSrvErr(404, 404, "服务不存在")
|
||||
return
|
||||
}
|
||||
if srv.TenantID != fmt.Sprintf("%d", claims.TenantId) || srv.UserID != uint64(claims.UserID) {
|
||||
c.mcpSrvErr(403, 403, "无权操作")
|
||||
return
|
||||
}
|
||||
|
||||
tools, err := services.McpClientManager.EnsureConnected(&srv)
|
||||
if err != nil {
|
||||
srv.Status = 2
|
||||
srv.LastError = err.Error()
|
||||
srv.UpdateTime = time.Now()
|
||||
_, _ = models.Orm.Update(&srv, "status", "last_error", "update_time")
|
||||
c.mcpSrvErr(400, 400, "连接失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
if srv.Status != 1 || srv.ToolCount != len(tools) {
|
||||
srv.Status = 1
|
||||
srv.LastError = ""
|
||||
srv.ToolCount = len(tools)
|
||||
srv.UpdateTime = time.Now()
|
||||
_, _ = models.Orm.Update(&srv, "status", "last_error", "tool_count", "update_time")
|
||||
}
|
||||
|
||||
c.mcpSrvOk(map[string]interface{}{"tools": tools, "count": len(tools)})
|
||||
}
|
||||
|
||||
// Market GET /backend/mcp/server/market
|
||||
func (c *BackendMcpServerController) Market() {
|
||||
if _, err := c.mcpSrvClaims(); err != nil {
|
||||
c.mcpSrvErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
c.mcpSrvOk(map[string]interface{}{"list": services.GetMcpMarket()})
|
||||
}
|
||||
|
||||
// AddFromMarket POST /backend/mcp/server/from-market
|
||||
// 从市场一键添加服务
|
||||
func (c *BackendMcpServerController) AddFromMarket() {
|
||||
claims, err := c.mcpSrvClaims()
|
||||
if err != nil {
|
||||
c.mcpSrvErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
body, err := io.ReadAll(c.Ctx.Request.Body)
|
||||
if err != nil {
|
||||
c.mcpSrvErr(400, 400, "读取请求体失败")
|
||||
return
|
||||
}
|
||||
var req struct {
|
||||
Key string `json:"key"`
|
||||
Name string `json:"name"`
|
||||
Enabled *int8 `json:"enabled"`
|
||||
}
|
||||
if err := json.Unmarshal(body, &req); err != nil {
|
||||
c.mcpSrvErr(400, 400, "参数格式错误")
|
||||
return
|
||||
}
|
||||
|
||||
item, ok := services.FindMarketItem(req.Key)
|
||||
if !ok {
|
||||
c.mcpSrvErr(404, 404, "市场不存在该服务: "+req.Key)
|
||||
return
|
||||
}
|
||||
|
||||
name := strings.TrimSpace(req.Name)
|
||||
if name == "" {
|
||||
name = item.Name
|
||||
}
|
||||
|
||||
// 防止重复添加同一市场服务
|
||||
count, _ := models.Orm.QueryTable(new(models.BackendMcpServer)).
|
||||
Filter("tenant_id", fmt.Sprintf("%d", claims.TenantId)).
|
||||
Filter("user_id", uint64(claims.UserID)).
|
||||
Filter("from_market", item.Key).
|
||||
Filter("delete_time__isnull", true).
|
||||
Count()
|
||||
if count > 0 {
|
||||
c.mcpSrvErr(400, 400, "该市场服务已添加，请勿重复添加")
|
||||
return
|
||||
}
|
||||
|
||||
enabled := int8(0)
|
||||
if req.Enabled != nil {
|
||||
enabled = *req.Enabled
|
||||
}
|
||||
|
||||
argsJSON, _ := json.Marshal(item.Args)
|
||||
envJSON, _ := json.Marshal(item.Env)
|
||||
srv := models.BackendMcpServer{
|
||||
TenantID: fmt.Sprintf("%d", claims.TenantId),
|
||||
UserID: uint64(claims.UserID),
|
||||
Name: name,
|
||||
Transport: item.Transport,
|
||||
Command: item.Command,
|
||||
Args: string(argsJSON),
|
||||
Env: string(envJSON),
|
||||
URL: item.URL,
|
||||
Description: item.Description,
|
||||
Provider: item.Provider,
|
||||
FromMarket: item.Key,
|
||||
Enabled: enabled,
|
||||
CreateTime: time.Now(),
|
||||
UpdateTime: time.Now(),
|
||||
}
|
||||
id, err := models.Orm.Insert(&srv)
|
||||
if err != nil {
|
||||
c.mcpSrvErr(500, 500, "添加失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
c.mcpSrvOk(map[string]interface{}{"id": id})
|
||||
}
|
||||
@@ -0,0 +1,265 @@
|
||||
package controllers
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"server/models"
|
||||
"server/pkg/jwtutil"
|
||||
|
||||
beego "github.com/beego/beego/v2/server/web"
|
||||
)
|
||||
|
||||
// BackendMcpToolController MCP工具控制器
|
||||
type BackendMcpToolController struct {
|
||||
beego.Controller
|
||||
}
|
||||
|
||||
func (c *BackendMcpToolController) mcpClaims() (*jwtutil.Claims, error) {
|
||||
auth := c.Ctx.Request.Header.Get("Authorization")
|
||||
if auth == "" {
|
||||
return nil, fmt.Errorf("未登录")
|
||||
}
|
||||
parts := strings.SplitN(auth, " ", 2)
|
||||
if len(parts) != 2 || parts[0] != "Bearer" {
|
||||
return nil, fmt.Errorf("认证信息格式错误")
|
||||
}
|
||||
claims, err := jwtutil.ParseToken(parts[1])
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("无效的token")
|
||||
}
|
||||
if claims.UserType != "backend" {
|
||||
return nil, fmt.Errorf("无权访问")
|
||||
}
|
||||
return claims, nil
|
||||
}
|
||||
|
||||
func (c *BackendMcpToolController) mcpJsonErr(httpStatus, bizCode int, msg string) {
|
||||
c.Ctx.Output.SetStatus(httpStatus)
|
||||
c.Data["json"] = map[string]interface{}{"code": bizCode, "msg": msg}
|
||||
_ = c.ServeJSON()
|
||||
}
|
||||
|
||||
func (c *BackendMcpToolController) mcpOk(data interface{}) {
|
||||
c.Data["json"] = map[string]interface{}{"code": 200, "msg": "success", "data": data}
|
||||
_ = c.ServeJSON()
|
||||
}
|
||||
|
||||
type mcpToolPayload struct {
|
||||
Name string `json:"name"`
|
||||
Type string `json:"type"`
|
||||
URL string `json:"url"`
|
||||
Method string `json:"method"`
|
||||
Headers string `json:"headers"`
|
||||
Params string `json:"params"`
|
||||
Body string `json:"body"`
|
||||
Enabled int8 `json:"enabled"`
|
||||
Remark string `json:"remark"`
|
||||
}
|
||||
|
||||
// List GET /backend/mcp/tool/list
|
||||
func (c *BackendMcpToolController) List() {
|
||||
claims, err := c.mcpClaims()
|
||||
if err != nil {
|
||||
c.mcpJsonErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
var list []models.BackendMcpTool
|
||||
_, err = models.Orm.QueryTable(new(models.BackendMcpTool)).
|
||||
Filter("tenant_id", fmt.Sprintf("%d", claims.TenantId)).
|
||||
Filter("user_id", uint64(claims.UserID)).
|
||||
Filter("delete_time__isnull", true).
|
||||
OrderBy("-id").
|
||||
All(&list)
|
||||
if err != nil {
|
||||
c.mcpJsonErr(500, 500, "查询失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
c.mcpOk(map[string]interface{}{"list": list})
|
||||
}
|
||||
|
||||
// Create POST /backend/mcp/tool
|
||||
func (c *BackendMcpToolController) Create() {
|
||||
claims, err := c.mcpClaims()
|
||||
if err != nil {
|
||||
c.mcpJsonErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
body, err := io.ReadAll(c.Ctx.Request.Body)
|
||||
if err != nil {
|
||||
c.mcpJsonErr(400, 400, "读取请求体失败")
|
||||
return
|
||||
}
|
||||
var p mcpToolPayload
|
||||
if err := json.Unmarshal(body, &p); err != nil {
|
||||
c.mcpJsonErr(400, 400, "参数格式错误")
|
||||
return
|
||||
}
|
||||
|
||||
if strings.TrimSpace(p.Name) == "" {
|
||||
c.mcpJsonErr(400, 400, "工具名称不能为空")
|
||||
return
|
||||
}
|
||||
if strings.TrimSpace(p.URL) == "" {
|
||||
c.mcpJsonErr(400, 400, "API地址不能为空")
|
||||
return
|
||||
}
|
||||
|
||||
tool := models.BackendMcpTool{
|
||||
TenantID: fmt.Sprintf("%d", claims.TenantId),
|
||||
UserID: uint64(claims.UserID),
|
||||
Name: strings.TrimSpace(p.Name),
|
||||
Type: strings.TrimSpace(p.Type),
|
||||
URL: strings.TrimSpace(p.URL),
|
||||
Method: strings.TrimSpace(p.Method),
|
||||
Headers: strings.TrimSpace(p.Headers),
|
||||
Params: strings.TrimSpace(p.Params),
|
||||
Body: strings.TrimSpace(p.Body),
|
||||
Enabled: p.Enabled,
|
||||
Remark: strings.TrimSpace(p.Remark),
|
||||
CreateTime: time.Now(),
|
||||
UpdateTime: time.Now(),
|
||||
}
|
||||
|
||||
id, err := models.Orm.Insert(&tool)
|
||||
if err != nil {
|
||||
c.mcpJsonErr(500, 500, "创建失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
c.mcpOk(map[string]interface{}{"id": id})
|
||||
}
|
||||
|
||||
// Update PUT /backend/mcp/tool/:id
|
||||
func (c *BackendMcpToolController) Update() {
|
||||
claims, err := c.mcpClaims()
|
||||
if err != nil {
|
||||
c.mcpJsonErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
idStr := c.Ctx.Input.Param(":id")
|
||||
id, err := strconv.ParseUint(idStr, 10, 64)
|
||||
if err != nil {
|
||||
c.mcpJsonErr(400, 400, "ID格式错误")
|
||||
return
|
||||
}
|
||||
|
||||
tool := models.BackendMcpTool{ID: id}
|
||||
if err := models.Orm.Read(&tool); err != nil {
|
||||
c.mcpJsonErr(404, 404, "工具不存在")
|
||||
return
|
||||
}
|
||||
if tool.TenantID != fmt.Sprintf("%d", claims.TenantId) {
|
||||
c.mcpJsonErr(403, 403, "无权操作")
|
||||
return
|
||||
}
|
||||
|
||||
body, err := io.ReadAll(c.Ctx.Request.Body)
|
||||
if err != nil {
|
||||
c.mcpJsonErr(400, 400, "读取请求体失败")
|
||||
return
|
||||
}
|
||||
var p mcpToolPayload
|
||||
if err := json.Unmarshal(body, &p); err != nil {
|
||||
c.mcpJsonErr(400, 400, "参数格式错误")
|
||||
return
|
||||
}
|
||||
|
||||
tool.Name = strings.TrimSpace(p.Name)
|
||||
tool.Type = strings.TrimSpace(p.Type)
|
||||
tool.URL = strings.TrimSpace(p.URL)
|
||||
tool.Method = strings.TrimSpace(p.Method)
|
||||
tool.Headers = strings.TrimSpace(p.Headers)
|
||||
tool.Params = strings.TrimSpace(p.Params)
|
||||
tool.Body = strings.TrimSpace(p.Body)
|
||||
tool.Enabled = p.Enabled
|
||||
tool.Remark = strings.TrimSpace(p.Remark)
|
||||
tool.UpdateTime = time.Now()
|
||||
|
||||
_, err = models.Orm.Update(&tool)
|
||||
if err != nil {
|
||||
c.mcpJsonErr(500, 500, "更新失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
c.mcpOk(nil)
|
||||
}
|
||||
|
||||
// Delete DELETE /backend/mcp/tool/:id
|
||||
func (c *BackendMcpToolController) Delete() {
|
||||
claims, err := c.mcpClaims()
|
||||
if err != nil {
|
||||
c.mcpJsonErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
idStr := c.Ctx.Input.Param(":id")
|
||||
id, err := strconv.ParseUint(idStr, 10, 64)
|
||||
if err != nil {
|
||||
c.mcpJsonErr(400, 400, "ID格式错误")
|
||||
return
|
||||
}
|
||||
|
||||
tool := models.BackendMcpTool{ID: id}
|
||||
if err := models.Orm.Read(&tool); err != nil {
|
||||
c.mcpJsonErr(404, 404, "工具不存在")
|
||||
return
|
||||
}
|
||||
if tool.TenantID != fmt.Sprintf("%d", claims.TenantId) {
|
||||
c.mcpJsonErr(403, 403, "无权操作")
|
||||
return
|
||||
}
|
||||
|
||||
now := time.Now()
|
||||
tool.DeleteTime = &now
|
||||
_, err = models.Orm.Update(&tool, "delete_time")
|
||||
if err != nil {
|
||||
c.mcpJsonErr(500, 500, "删除失败: "+err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
c.mcpOk(nil)
|
||||
}
|
||||
|
||||
// Test POST /backend/mcp/tool/test
|
||||
func (c *BackendMcpToolController) Test() {
|
||||
// 验证用户登录状态
|
||||
if _, err := c.mcpClaims(); err != nil {
|
||||
c.mcpJsonErr(401, 401, err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
body, err := io.ReadAll(c.Ctx.Request.Body)
|
||||
if err != nil {
|
||||
c.mcpJsonErr(400, 400, "读取请求体失败")
|
||||
return
|
||||
}
|
||||
|
||||
var p mcpToolPayload
|
||||
if err := json.Unmarshal(body, &p); err != nil {
|
||||
c.mcpJsonErr(400, 400, "参数格式错误")
|
||||
return
|
||||
}
|
||||
|
||||
if strings.TrimSpace(p.URL) == "" {
|
||||
c.mcpJsonErr(400, 400, "API地址不能为空")
|
||||
return
|
||||
}
|
||||
|
||||
// TODO: 实现MCP工具测试逻辑
|
||||
// 这里只是一个示例实现，实际测试逻辑需要根据工具类型和配置来实现
|
||||
c.mcpOk(map[string]interface{}{
|
||||
"success": true,
|
||||
"message": "测试连接成功",
|
||||
"method": p.Method,
|
||||
"url": p.URL,
|
||||
})
|
||||
}
|
||||
@@ -1,38 +1,42 @@
|
||||
module server
|
||||
|
||||
go 1.17
|
||||
|
||||
require (
|
||||
github.com/beego/beego/v2 v2.1.0
|
||||
github.com/golang-jwt/jwt/v5 v5.2.1
|
||||
github.com/qiniu/go-sdk/v7 v7.18.2
|
||||
golang.org/x/crypto v0.1.0
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/go-sql-driver/mysql v1.7.0
|
||||
github.com/google/uuid v1.6.0
|
||||
golang.org/x/net v0.7.0
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/beorn7/perks v1.0.1 // indirect
|
||||
github.com/cespare/xxhash/v2 v2.2.0 // indirect
|
||||
github.com/golang/protobuf v1.5.3 // indirect
|
||||
github.com/hashicorp/golang-lru v0.5.4 // indirect
|
||||
github.com/matttproud/golang_protobuf_extensions v1.0.4 // indirect
|
||||
github.com/mitchellh/mapstructure v1.5.0 // indirect
|
||||
github.com/pkg/errors v0.9.1 // indirect
|
||||
github.com/prometheus/client_golang v1.15.1 // indirect
|
||||
github.com/prometheus/client_model v0.3.0 // indirect
|
||||
github.com/prometheus/common v0.42.0 // indirect
|
||||
github.com/prometheus/procfs v0.9.0 // indirect
|
||||
github.com/shiena/ansicolor v0.0.0-20200904210342-c7312218db18 // indirect
|
||||
golang.org/x/sync v0.1.0 // indirect
|
||||
golang.org/x/sys v0.6.0 // indirect
|
||||
golang.org/x/text v0.7.0 // indirect
|
||||
google.golang.org/protobuf v1.30.0 // indirect
|
||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||
)
|
||||
|
||||
exclude github.com/mattn/go-sqlite3 v1.14.31
|
||||
module server
|
||||
|
||||
go 1.23.0
|
||||
|
||||
require (
|
||||
github.com/beego/beego/v2 v2.1.0
|
||||
github.com/golang-jwt/jwt/v5 v5.2.1
|
||||
github.com/qiniu/go-sdk/v7 v7.18.2
|
||||
golang.org/x/crypto v0.1.0
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/go-sql-driver/mysql v1.7.0
|
||||
github.com/google/uuid v1.6.0
|
||||
github.com/mark3labs/mcp-go v0.33.0
|
||||
golang.org/x/net v0.7.0
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/beorn7/perks v1.0.1 // indirect
|
||||
github.com/cespare/xxhash/v2 v2.2.0 // indirect
|
||||
github.com/golang/protobuf v1.5.3 // indirect
|
||||
github.com/google/go-cmp v0.7.0 // indirect
|
||||
github.com/hashicorp/golang-lru v0.5.4 // indirect
|
||||
github.com/matttproud/golang_protobuf_extensions v1.0.4 // indirect
|
||||
github.com/mitchellh/mapstructure v1.5.0 // indirect
|
||||
github.com/pkg/errors v0.9.1 // indirect
|
||||
github.com/prometheus/client_golang v1.15.1 // indirect
|
||||
github.com/prometheus/client_model v0.3.0 // indirect
|
||||
github.com/prometheus/common v0.42.0 // indirect
|
||||
github.com/prometheus/procfs v0.9.0 // indirect
|
||||
github.com/shiena/ansicolor v0.0.0-20200904210342-c7312218db18 // indirect
|
||||
github.com/spf13/cast v1.7.1 // indirect
|
||||
github.com/yosida95/uritemplate/v3 v3.0.2 // indirect
|
||||
golang.org/x/sync v0.1.0 // indirect
|
||||
golang.org/x/sys v0.6.0 // indirect
|
||||
golang.org/x/text v0.14.0 // indirect
|
||||
google.golang.org/protobuf v1.30.0 // indirect
|
||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||
)
|
||||
|
||||
exclude github.com/mattn/go-sqlite3 v1.14.31
|
||||
|
||||
BIN
Binary file not shown.
@@ -13,6 +13,7 @@ type BackendAiProvider struct {
|
||||
ApiKey string `orm:"column(api_key);size(255)" json:"api_key"`
|
||||
Models string `orm:"column(models);type(text)" json:"-"` // JSON数组存储，如 ["gpt-4o","gpt-3.5-turbo"]
|
||||
ModelsList []string `orm:"-" json:"models"` // 返回给前端的模型列表
|
||||
DefaultModel string `orm:"column(default_model);size(255)" json:"default_model"` // 默认使用的模型（从models列表中选择）
|
||||
IsDefault int8 `orm:"column(is_default);default(0)" json:"is_default"` // 0-否 1-是（租户默认模型，用于智能添加等企业级功能）
|
||||
Status int8 `orm:"column(status);default(1)" json:"status"` // 0-禁用 1-启用
|
||||
Remark string `orm:"column(remark);size(255)" json:"remark"`
|
||||
|
||||
@@ -0,0 +1,46 @@
|
||||
package models
|
||||
|
||||
import "time"
|
||||
|
||||
// BackendMcpServer MCP服务器配置表（真正的 MCP 协议接入配置）
|
||||
// transport 支持三种：
|
||||
// - stdio：本地进程方式（command + args + env）
|
||||
// - http ：Streamable HTTP 传输（url + headers）
|
||||
// - sse ：SSE 传输（url + headers）
|
||||
type BackendMcpServer struct {
|
||||
ID uint64 `orm:"column(id);pk;auto" json:"id"`
|
||||
TenantID string `orm:"column(tenant_id);size(64)" json:"tenant_id"`
|
||||
UserID uint64 `orm:"column(user_id);default(0)" json:"user_id"`
|
||||
Name string `orm:"column(name);size(100)" json:"name"`
|
||||
Transport string `orm:"column(transport);size(20)" json:"transport"` // stdio/http/sse
|
||||
Command string `orm:"column(command);size(255)" json:"command"` // stdio 命令
|
||||
Args string `orm:"column(args);type(text)" json:"args"` // stdio 参数 JSON数组
|
||||
Env string `orm:"column(env);type(text)" json:"env"` // 环境变量 JSON对象
|
||||
URL string `orm:"column(url);size(512)" json:"url"` // http/sse 地址
|
||||
Headers string `orm:"column(headers);type(text)" json:"headers"` // 请求头 JSON对象
|
||||
Description string `orm:"column(description);size(255)" json:"description"`
|
||||
Provider string `orm:"column(provider);size(50)" json:"provider"` // 提供方名称
|
||||
FromMarket string `orm:"column(from_market);size(50)" json:"from_market"` // 来源市场key
|
||||
Enabled int8 `orm:"column(enabled);default(0)" json:"enabled"` // 0-禁用 1-启用（仅启用项在会话中生效）
|
||||
Status int8 `orm:"column(status);default(0)" json:"status"` // 0-未测试 1-连接正常 2-连接失败
|
||||
LastError string `orm:"column(last_error);size(500)" json:"last_error"`
|
||||
ToolCount int `orm:"column(tool_count);default(0)" json:"tool_count"`
|
||||
Remark string `orm:"column(remark);size(255)" json:"remark"`
|
||||
CreateTime time.Time `orm:"column(create_time);auto_now_add;type(datetime)" json:"create_time"`
|
||||
UpdateTime time.Time `orm:"column(update_time);auto_now;type(datetime)" json:"update_time"`
|
||||
DeleteTime *time.Time `orm:"column(delete_time);type(datetime);null" json:"delete_time"`
|
||||
|
||||
// 非数据库字段：连接后发现的工具列表（用于前端展示）
|
||||
Tools []McpToolInfo `orm:"-" json:"tools,omitempty"`
|
||||
}
|
||||
|
||||
// McpToolInfo MCP 工具信息（用于前端展示与 LLM 工具注入）
|
||||
type McpToolInfo struct {
|
||||
Name string `json:"name"`
|
||||
Description string `json:"description"`
|
||||
InputSchema interface{} `json:"input_schema"`
|
||||
}
|
||||
|
||||
func (m *BackendMcpServer) TableName() string {
|
||||
return "yz_backend_mcp_server"
|
||||
}
|
||||
@@ -0,0 +1,26 @@
|
||||
package models
|
||||
|
||||
import "time"
|
||||
|
||||
// BackendMcpTool MCP工具配置表
|
||||
type BackendMcpTool struct {
|
||||
ID uint64 `orm:"column(id);pk;auto" json:"id"`
|
||||
TenantID string `orm:"column(tenant_id);size(64)" json:"tenant_id"`
|
||||
UserID uint64 `orm:"column(user_id);default(0)" json:"user_id"`
|
||||
Name string `orm:"column(name);size(100)" json:"name"`
|
||||
Type string `orm:"column(type);size(20)" json:"type"` // rest/http/other
|
||||
URL string `orm:"column(url);size(255)" json:"url"`
|
||||
Method string `orm:"column(method);size(10)" json:"method"` // GET/POST/PUT/DELETE
|
||||
Headers string `orm:"column(headers);type(text)" json:"headers"` // JSON
|
||||
Params string `orm:"column(params);type(text)" json:"params"` // JSON
|
||||
Body string `orm:"column(body);type(text)" json:"body"` // JSON
|
||||
Enabled int8 `orm:"column(enabled);default(1)" json:"enabled"` // 0-禁用 1-启用
|
||||
Remark string `orm:"column(remark);size(255)" json:"remark"`
|
||||
CreateTime time.Time `orm:"column(create_time);auto_now_add;type(datetime)" json:"create_time"`
|
||||
UpdateTime time.Time `orm:"column(update_time);auto_now;type(datetime)" json:"update_time"`
|
||||
DeleteTime *time.Time `orm:"column(delete_time);type(datetime);null" json:"delete_time"`
|
||||
}
|
||||
|
||||
func (m *BackendMcpTool) TableName() string {
|
||||
return "yz_backend_mcp_tool"
|
||||
}
|
||||
@@ -81,6 +81,8 @@ func Init(_ string) {
|
||||
new(BackendAiChatSession),
|
||||
new(BackendAiChatMessage),
|
||||
new(BackendAiChatPreset),
|
||||
new(BackendMcpTool),
|
||||
new(BackendMcpServer),
|
||||
|
||||
new(CmsArticleCategory),
|
||||
new(CmsArticle),
|
||||
|
||||
@@ -364,4 +364,20 @@ func registerOrganizationRoutes(module string) {
|
||||
beego.Router("/backend/ai/preset", &controllers.BackendAiChatPresetController{}, "post:Create")
|
||||
beego.Router("/backend/ai/preset/:id", &controllers.BackendAiChatPresetController{}, "put:Update;delete:Delete")
|
||||
beego.Router("/backend/ai/preset/:id/default", &controllers.BackendAiChatPresetController{}, "put:SetDefault")
|
||||
|
||||
// MCP工具管理
|
||||
beego.Router("/backend/mcp/tool/list", &controllers.BackendMcpToolController{}, "get:List")
|
||||
beego.Router("/backend/mcp/tool", &controllers.BackendMcpToolController{}, "post:Create")
|
||||
beego.Router("/backend/mcp/tool/:id", &controllers.BackendMcpToolController{}, "put:Update;delete:Delete")
|
||||
beego.Router("/backend/mcp/tool/test", &controllers.BackendMcpToolController{}, "post:Test")
|
||||
|
||||
// MCP服务器配置管理（真正接入 MCP 协议的服务）
|
||||
beego.Router("/backend/mcp/server/list", &controllers.BackendMcpServerController{}, "get:List")
|
||||
beego.Router("/backend/mcp/server", &controllers.BackendMcpServerController{}, "post:Create")
|
||||
beego.Router("/backend/mcp/server/market", &controllers.BackendMcpServerController{}, "get:Market")
|
||||
beego.Router("/backend/mcp/server/from-market", &controllers.BackendMcpServerController{}, "post:AddFromMarket")
|
||||
beego.Router("/backend/mcp/server/test", &controllers.BackendMcpServerController{}, "post:Test")
|
||||
beego.Router("/backend/mcp/server/:id", &controllers.BackendMcpServerController{}, "put:Update;delete:Delete")
|
||||
beego.Router("/backend/mcp/server/:id/toggle", &controllers.BackendMcpServerController{}, "put:Toggle")
|
||||
beego.Router("/backend/mcp/server/:id/tools", &controllers.BackendMcpServerController{}, "post:Tools")
|
||||
}
|
||||
|
||||
@@ -0,0 +1,325 @@
|
||||
package services
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"path/filepath"
|
||||
"runtime"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"server/models"
|
||||
|
||||
mcpclient "github.com/mark3labs/mcp-go/client"
|
||||
"github.com/mark3labs/mcp-go/client/transport"
|
||||
"github.com/mark3labs/mcp-go/mcp"
|
||||
)
|
||||
|
||||
// McpSession 一次已建立的 MCP 连接会话
|
||||
type McpSession struct {
|
||||
Server models.BackendMcpServer
|
||||
Client *mcpclient.Client
|
||||
Tools []models.McpToolInfo
|
||||
Finger string // 配置指纹，配置变更时自动重连
|
||||
}
|
||||
|
||||
// McpManager MCP 客户端管理器（全局单例，带连接缓存）
|
||||
type McpManager struct {
|
||||
mu sync.Mutex
|
||||
sessions map[uint64]*McpSession
|
||||
}
|
||||
|
||||
// NewMcpManager 创建管理器
|
||||
func NewMcpManager() *McpManager {
|
||||
return &McpManager{sessions: make(map[uint64]*McpSession)}
|
||||
}
|
||||
|
||||
// McpClientManager 全局 MCP 客户端管理器
|
||||
var McpClientManager = NewMcpManager()
|
||||
|
||||
const (
|
||||
connectTimeout = 25 * time.Second
|
||||
callTimeout = 90 * time.Second
|
||||
)
|
||||
|
||||
// serverFinger 计算服务器配置指纹（用于配置变更自动重连）
|
||||
func serverFinger(s *models.BackendMcpServer) string {
|
||||
return fmt.Sprintf("%s|%s|%s|%s|%s|%s|%s",
|
||||
s.Transport, s.Command, s.Args, s.Env, s.URL, s.Headers, s.Name)
|
||||
}
|
||||
|
||||
// parseStringArray 解析 JSON 数组字符串为 []string
|
||||
func parseStringArray(s string) []string {
|
||||
s = strings.TrimSpace(s)
|
||||
if s == "" {
|
||||
return nil
|
||||
}
|
||||
var arr []string
|
||||
if err := json.Unmarshal([]byte(s), &arr); err == nil {
|
||||
return arr
|
||||
}
|
||||
// 兼容逗号分隔
|
||||
parts := strings.Split(s, ",")
|
||||
out := make([]string, 0, len(parts))
|
||||
for _, p := range parts {
|
||||
if v := strings.TrimSpace(p); v != "" {
|
||||
out = append(out, v)
|
||||
}
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
// parseStringMap 解析 JSON 对象字符串为 map
|
||||
func parseStringMap(s string) map[string]string {
|
||||
s = strings.TrimSpace(s)
|
||||
if s == "" {
|
||||
return nil
|
||||
}
|
||||
var m map[string]string
|
||||
if err := json.Unmarshal([]byte(s), &m); err == nil {
|
||||
return m
|
||||
}
|
||||
var raw map[string]interface{}
|
||||
if err := json.Unmarshal([]byte(s), &raw); err == nil {
|
||||
out := make(map[string]string, len(raw))
|
||||
for k, v := range raw {
|
||||
out[k] = fmt.Sprintf("%v", v)
|
||||
}
|
||||
return out
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// envToSlice 将环境变量 map 转为 "K=V" 切片
|
||||
func envToSlice(env map[string]string) []string {
|
||||
if len(env) == 0 {
|
||||
return nil
|
||||
}
|
||||
out := make([]string, 0, len(env))
|
||||
for k, v := range env {
|
||||
out = append(out, k+"="+v)
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
// resolveStdioCommand Windows 下 .cmd/.bat/npx 需要 cmd /c 包装
|
||||
func resolveStdioCommand(cmd string, args []string) (string, []string) {
|
||||
if runtime.GOOS != "windows" {
|
||||
return cmd, args
|
||||
}
|
||||
lower := strings.ToLower(strings.TrimSpace(cmd))
|
||||
base := filepath.Base(lower)
|
||||
if base == "npx" || base == "npm" || base == "npx.cmd" || base == "npm.cmd" ||
|
||||
base == "uvx" || base == "uvx.exe" ||
|
||||
strings.HasSuffix(lower, ".cmd") || strings.HasSuffix(lower, ".bat") {
|
||||
all := append([]string{cmd}, args...)
|
||||
return "cmd", append([]string{"/c"}, all...)
|
||||
}
|
||||
return cmd, args
|
||||
}
|
||||
|
||||
// buildClient 按传输类型创建 MCP 客户端（不连接、不初始化）
|
||||
func buildClient(s *models.BackendMcpServer) (*mcpclient.Client, error) {
|
||||
switch s.Transport {
|
||||
case "stdio":
|
||||
cmd, args := resolveStdioCommand(strings.TrimSpace(s.Command), parseStringArray(s.Args))
|
||||
if cmd == "" {
|
||||
return nil, fmt.Errorf("stdio 传输必须配置 command")
|
||||
}
|
||||
return mcpclient.NewStdioMCPClient(cmd, envToSlice(parseStringMap(s.Env)), args...)
|
||||
case "sse":
|
||||
if strings.TrimSpace(s.URL) == "" {
|
||||
return nil, fmt.Errorf("sse 传输必须配置 url")
|
||||
}
|
||||
return mcpclient.NewSSEMCPClient(strings.TrimSpace(s.URL), transport.WithHeaders(parseStringMap(s.Headers)))
|
||||
case "http":
|
||||
if strings.TrimSpace(s.URL) == "" {
|
||||
return nil, fmt.Errorf("http 传输必须配置 url")
|
||||
}
|
||||
return mcpclient.NewStreamableHttpClient(strings.TrimSpace(s.URL), transport.WithHTTPHeaders(parseStringMap(s.Headers)))
|
||||
default:
|
||||
return nil, fmt.Errorf("不支持的传输类型: %s", s.Transport)
|
||||
}
|
||||
}
|
||||
|
||||
// connectAndList 建立连接并列出工具
|
||||
func connectAndList(ctx context.Context, s *models.BackendMcpServer) (*mcpclient.Client, []models.McpToolInfo, error) {
|
||||
cl, err := buildClient(s)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
|
||||
if s.Transport != "stdio" {
|
||||
// stdio 的传输在构造函数中已启动，其余需手动 Start
|
||||
if err := cl.Start(ctx); err != nil {
|
||||
_ = cl.Close()
|
||||
return nil, nil, fmt.Errorf("启动连接失败: %w", err)
|
||||
}
|
||||
}
|
||||
|
||||
initReq := mcp.InitializeRequest{}
|
||||
initReq.Params.ProtocolVersion = mcp.LATEST_PROTOCOL_VERSION
|
||||
initReq.Params.ClientInfo = mcp.Implementation{
|
||||
Name: "xiaozhi-ai-backend",
|
||||
Version: "1.0.0",
|
||||
}
|
||||
if _, err := cl.Initialize(ctx, initReq); err != nil {
|
||||
_ = cl.Close()
|
||||
return nil, nil, fmt.Errorf("MCP 握手失败: %w", err)
|
||||
}
|
||||
|
||||
toolsResult, err := cl.ListTools(ctx, mcp.ListToolsRequest{})
|
||||
if err != nil {
|
||||
_ = cl.Close()
|
||||
return nil, nil, fmt.Errorf("获取工具列表失败: %w", err)
|
||||
}
|
||||
|
||||
tools := make([]models.McpToolInfo, 0, len(toolsResult.Tools))
|
||||
for _, t := range toolsResult.Tools {
|
||||
info := models.McpToolInfo{
|
||||
Name: t.Name,
|
||||
Description: t.Description,
|
||||
}
|
||||
// 优先使用 RawInputSchema（完整 JSON Schema）
|
||||
if len(t.RawInputSchema) > 0 {
|
||||
var schema interface{}
|
||||
if err := json.Unmarshal(t.RawInputSchema, &schema); err == nil {
|
||||
info.InputSchema = schema
|
||||
}
|
||||
} else if t.InputSchema.Type != "" || len(t.InputSchema.Properties) > 0 {
|
||||
info.InputSchema = t.InputSchema
|
||||
}
|
||||
tools = append(tools, info)
|
||||
}
|
||||
return cl, tools, nil
|
||||
}
|
||||
|
||||
// TestConnection 测试连接：新建连接 → 列出工具 → 关闭（不进入缓存）
|
||||
func (m *McpManager) TestConnection(s *models.BackendMcpServer) ([]models.McpToolInfo, error) {
|
||||
ctx, cancel := context.WithTimeout(context.Background(), connectTimeout)
|
||||
defer cancel()
|
||||
|
||||
cl, tools, err := connectAndList(ctx, s)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
_ = cl.Close()
|
||||
return tools, nil
|
||||
}
|
||||
|
||||
// EnsureConnected 获取（或建立）连接，返回该服务器已发现的工具
|
||||
func (m *McpManager) EnsureConnected(s *models.BackendMcpServer) ([]models.McpToolInfo, error) {
|
||||
finger := serverFinger(s)
|
||||
|
||||
m.mu.Lock()
|
||||
if sess, ok := m.sessions[s.ID]; ok && sess.Finger == finger {
|
||||
tools := sess.Tools
|
||||
m.mu.Unlock()
|
||||
return tools, nil
|
||||
}
|
||||
m.mu.Unlock()
|
||||
|
||||
// 新建连接（放在锁外，避免长时间占用锁）
|
||||
ctx, cancel := context.WithTimeout(context.Background(), connectTimeout)
|
||||
defer cancel()
|
||||
cl, tools, err := connectAndList(ctx, s)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
// 关闭旧连接
|
||||
if old, ok := m.sessions[s.ID]; ok && old.Client != nil {
|
||||
_ = old.Client.Close()
|
||||
}
|
||||
m.sessions[s.ID] = &McpSession{
|
||||
Server: *s,
|
||||
Client: cl,
|
||||
Tools: tools,
|
||||
Finger: finger,
|
||||
}
|
||||
return tools, nil
|
||||
}
|
||||
|
||||
// ListTools 返回缓存中的工具列表（未连接返回 nil）
|
||||
func (m *McpManager) ListTools(serverID uint64) []models.McpToolInfo {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
if sess, ok := m.sessions[serverID]; ok {
|
||||
return sess.Tools
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// CallTool 调用 MCP 工具，返回文本结果
|
||||
func (m *McpManager) CallTool(serverID uint64, name string, args map[string]interface{}) (string, bool, error) {
|
||||
m.mu.Lock()
|
||||
sess, ok := m.sessions[serverID]
|
||||
m.mu.Unlock()
|
||||
if !ok || sess.Client == nil {
|
||||
return "", false, fmt.Errorf("MCP 服务未连接")
|
||||
}
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), callTimeout)
|
||||
defer cancel()
|
||||
|
||||
req := mcp.CallToolRequest{Params: mcp.CallToolParams{
|
||||
Name: name,
|
||||
Arguments: args,
|
||||
}}
|
||||
result, err := sess.Client.CallTool(ctx, req)
|
||||
if err != nil {
|
||||
return "", false, err
|
||||
}
|
||||
return mcpResultToText(result), result.IsError, nil
|
||||
}
|
||||
|
||||
// mcpResultToText 将 MCP CallToolResult 内容转为文本
|
||||
func mcpResultToText(result *mcp.CallToolResult) string {
|
||||
if result == nil {
|
||||
return ""
|
||||
}
|
||||
var sb strings.Builder
|
||||
for _, c := range result.Content {
|
||||
switch v := c.(type) {
|
||||
case mcp.TextContent:
|
||||
sb.WriteString(v.Text)
|
||||
case mcp.ImageContent:
|
||||
sb.WriteString(fmt.Sprintf("[图片: %s, %d 字节]", v.MIMEType, len(v.Data)))
|
||||
case mcp.AudioContent:
|
||||
sb.WriteString(fmt.Sprintf("[音频: %s, %d 字节]", v.MIMEType, len(v.Data)))
|
||||
default:
|
||||
if b, err := json.Marshal(c); err == nil {
|
||||
sb.Write(b)
|
||||
}
|
||||
}
|
||||
}
|
||||
return sb.String()
|
||||
}
|
||||
|
||||
// Disconnect 断开并移除指定服务器的连接
|
||||
func (m *McpManager) Disconnect(serverID uint64) {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
if sess, ok := m.sessions[serverID]; ok {
|
||||
if sess.Client != nil {
|
||||
_ = sess.Client.Close()
|
||||
}
|
||||
delete(m.sessions, serverID)
|
||||
}
|
||||
}
|
||||
|
||||
// CloseAll 关闭所有连接（服务退出时调用）
|
||||
func (m *McpManager) CloseAll() {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
for id, sess := range m.sessions {
|
||||
if sess.Client != nil {
|
||||
_ = sess.Client.Close()
|
||||
}
|
||||
delete(m.sessions, id)
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,120 @@
|
||||
package services
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
"runtime"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// McpMarketItem MCP 市场条目
|
||||
type McpMarketItem struct {
|
||||
Key string `json:"key"`
|
||||
Name string `json:"name"`
|
||||
Description string `json:"description"`
|
||||
Provider string `json:"provider"`
|
||||
Transport string `json:"transport"` // stdio/http/sse
|
||||
Command string `json:"command"`
|
||||
Args []string `json:"args"`
|
||||
Env []string `json:"env"`
|
||||
URL string `json:"url"`
|
||||
Tags []string `json:"tags"`
|
||||
BuiltIn bool `json:"built_in"` // 是否为项目内置（演示）服务
|
||||
}
|
||||
|
||||
// demoMcpPath 解析内置演示 MCP 服务可执行文件路径
|
||||
// 优先级：1) 服务器工作目录下 bin/demo-mcp 2) go run ./cmd/demo-mcp 兜底
|
||||
func demoMcpPath() (string, []string) {
|
||||
ext := ""
|
||||
if runtime.GOOS == "windows" {
|
||||
ext = ".exe"
|
||||
}
|
||||
candidates := []string{
|
||||
filepath.Join("bin", "demo-mcp"+ext),
|
||||
filepath.Join("..", "bin", "demo-mcp"+ext),
|
||||
}
|
||||
if wd, err := os.Getwd(); err == nil {
|
||||
for _, c := range candidates {
|
||||
p := filepath.Join(wd, c)
|
||||
if info, err := os.Stat(p); err == nil && !info.IsDir() {
|
||||
return p, nil
|
||||
}
|
||||
}
|
||||
}
|
||||
// 兜底：go run ./cmd/demo-mcp（需要 Go 工具链）
|
||||
return "go", []string{"run", "./cmd/demo-mcp"}
|
||||
}
|
||||
|
||||
// GetMcpMarket 返回 MCP 市场服务列表
|
||||
func GetMcpMarket() []McpMarketItem {
|
||||
cmd, args := demoMcpPath()
|
||||
list := []McpMarketItem{
|
||||
{
|
||||
Key: "demo",
|
||||
Name: "内置演示 MCP",
|
||||
Description: "项目自带演示服务：提供当前时间、计算器、模拟天气、回声等工具，用于验证 MCP 全链路。",
|
||||
Provider: "本项目",
|
||||
Transport: "stdio",
|
||||
Command: cmd,
|
||||
Args: args,
|
||||
Tags: []string{"演示", "零配置"},
|
||||
BuiltIn: true,
|
||||
},
|
||||
{
|
||||
Key: "tianyancha",
|
||||
Name: "天眼查 MCP",
|
||||
Description: "企业工商信息查询、股权穿透、司法风险等数据服务。",
|
||||
Provider: "天眼查",
|
||||
Transport: "http",
|
||||
URL: "https://mcp.tianyancha.com/mcp",
|
||||
Tags: []string{"企业信息"},
|
||||
},
|
||||
{
|
||||
Key: "qcc",
|
||||
Name: "企查查 MCP",
|
||||
Description: "企业信用信息、工商资料、经营风险等数据查询服务。",
|
||||
Provider: "企查查",
|
||||
Transport: "http",
|
||||
URL: "https://mcp.qcc.com/mcp",
|
||||
Tags: []string{"企业信息"},
|
||||
},
|
||||
{
|
||||
Key: "qianzhan",
|
||||
Name: "前瞻 MCP",
|
||||
Description: "前瞻产业研究院行业数据、研究报告等。",
|
||||
Provider: "前瞻",
|
||||
Transport: "http",
|
||||
URL: "https://mcp.qianzhan.com/mcp",
|
||||
Tags: []string{"行业数据"},
|
||||
},
|
||||
{
|
||||
Key: "itjuzi",
|
||||
Name: "IT桔子 MCP",
|
||||
Description: "创业公司数据、投融资事件、行业洞察等。",
|
||||
Provider: "IT桔子",
|
||||
Transport: "http",
|
||||
URL: "https://mcp.itjuzi.com/mcp",
|
||||
Tags: []string{"创投数据"},
|
||||
},
|
||||
{
|
||||
Key: "boss",
|
||||
Name: "BOSS直聘 MCP",
|
||||
Description: "BOSS直聘企业招聘、职位等数据服务。",
|
||||
Provider: "BOSS直聘",
|
||||
Transport: "http",
|
||||
URL: "https://mcp.zhipin.com/mcp",
|
||||
Tags: []string{"招聘"},
|
||||
},
|
||||
}
|
||||
return list
|
||||
}
|
||||
|
||||
// FindMarketItem 按 key 查找市场条目
|
||||
func FindMarketItem(key string) (McpMarketItem, bool) {
|
||||
for _, item := range GetMcpMarket() {
|
||||
if strings.EqualFold(item.Key, key) {
|
||||
return item, true
|
||||
}
|
||||
}
|
||||
return McpMarketItem{}, false
|
||||
}
|
||||
Reference in New Issue
Block a user