logan
cd319ce7bd
- 侧栏 Ask|Agent 分段；Ask 仍走 /api/ai/chat，Agent 走 /api/ai/agent - RunAgent 多轮 tool call → 进程内 aitools（与 MCP 同源只读/白名单/审计） - 回复下步骤时间线；openai-chat tools（DeepSeek）；agent-accept 冒烟
55 lines
1.4 KiB
Go
55 lines
1.4 KiB
Go
// opsxdb-mcp：stdio MCP Server，暴露只读数据库工具（P0）。 // Logan.Li
|
||
//
|
||
// 配置：
|
||
// ~/.opsxdb/mcp.json — allowed_connection_ids；mode（P0 执行层强制只读）
|
||
// ~/.opsxdb/connections.json — 桌面双写的连接列表（密码 ox1. 混淆）
|
||
// ~/.opsxdb/logs/mcp-audit.jsonl
|
||
//
|
||
// Cursor 示例见 docs/mcp-cursor.example.json
|
||
package main
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"fmt"
|
||
"os"
|
||
|
||
"github.com/mark3labs/mcp-go/mcp"
|
||
"github.com/mark3labs/mcp-go/server"
|
||
|
||
"opsxdb/server/internal/aitools"
|
||
)
|
||
|
||
func main() {
|
||
rt, err := aitools.DefaultRuntime()
|
||
if err != nil {
|
||
fmt.Fprintf(os.Stderr, "opsxdb-mcp: load policy: %v\n", err)
|
||
os.Exit(1)
|
||
}
|
||
|
||
s := server.NewMCPServer(
|
||
"opsxdb",
|
||
"0.1.0",
|
||
server.WithToolCapabilities(true),
|
||
)
|
||
|
||
for _, spec := range aitools.ToolSpecs() {
|
||
raw, _ := json.Marshal(spec.InputSchema())
|
||
t := mcp.NewToolWithRawSchema(spec.Name, spec.Description, raw)
|
||
toolName := spec.Name
|
||
s.AddTool(t, func(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||
args := req.GetArguments()
|
||
out, err := rt.Call(ctx, toolName, args)
|
||
if err != nil {
|
||
return mcp.NewToolResultError(err.Error()), nil
|
||
}
|
||
return mcp.NewToolResultText(aitools.ResultJSON(out)), nil
|
||
})
|
||
}
|
||
|
||
if err := server.ServeStdio(s); err != nil {
|
||
fmt.Fprintf(os.Stderr, "opsxdb-mcp: %v\n", err)
|
||
os.Exit(1)
|
||
}
|
||
}
|