eagle-rag 1.0.1

package/README.md

@@ -0,0 +1,116 @@
+# Coze RAG MCP Server
+
+通过 npx 运行的 MCP Server，对接 Coze 工作流，实现 RAG 问答功能。
+
+## 功能特性
+
+- **身份识别**: 自动获取机器唯一 ID，作为用户标识
+- **RAG 问答**: 将用户问题传递给 Coze 工作流，获取知识库相关答案
+- **MCP 协议**: 完全兼容 Model Context Protocol
+
+## 快速开始
+
+### 1. 安装依赖
+
+```bash
+npm install
+```
+
+### 2. 配置环境变量（可选）
+
+可以通过环境变量覆盖默认配置：
+
+```bash
+# 设置 Coze API Token
+export COZE_API_TOKEN="your_api_token"
+
+# 设置 Coze 工作流 ID
+export COZE_WORKFLOW_ID="your_workflow_id"
+```
+
+### 3. 运行 MCP Server
+
+```bash
+# 使用 npx 运行
+npx coze-rag-mcp
+
+# 或者使用 npm 脚本
+npm start
+```
+
+## MCP 工具
+
+### ask
+
+向 Coze RAG 工作流提问，获取知识库相关答案。
+
+**参数:**
+- `question` (string): 用户问题
+
+**返回值:**
+- 知识库检索结果
+
+**示例:**
+```json
+{
+  "question": "如何创建无边框窗口？"
+}
+```
+
+### get_machine_id
+
+获取当前机器的唯一标识 ID。
+
+**参数:**
+- 无
+
+**返回值:**
+- 机器唯一 ID
+
+## 配置说明
+
+| 环境变量 | 默认值 | 说明 |
+|---------|--------|------|
+| `COZE_API_TOKEN` | `sat_pqqZVXMBiDTas0khz2x3vBqrCcicmfOt9XlNi4KrbtUsRtbroa8i1z5wfGrcXWdn` | Coze API Token |
+| `COZE_WORKFLOW_ID` | `7610685653369798719` | Coze 工作流 ID |
+
+## 在 Cursor 中使用
+
+在 Cursor 的 `mcp.json` 配置文件中添加：
+
+```json
+{
+  "mcpServers": {
+    "coze-rag": {
+      "command": "npx",
+      "args": ["coze-rag-mcp"],
+      "env": {
+        "COZE_API_TOKEN": "your_api_token"
+      }
+    }
+  }
+}
+```
+
+## API 请求格式
+
+```javascript
+POST https://api.coze.cn/v1/workflow/run
+Authorization: Bearer <token>
+Content-Type: application/json
+
+{
+  "workflow_id": "7610685653369798719",
+  "parameters": {
+    "user_id": "<machine_id>",
+    "question": "<user_question>"
+  }
+}
+```
+
+## 依赖
+
+- `@modelcontextprotocol/sdk` - MCP Server SDK
+- `axios` - HTTP 请求库
+- `node-machine-id` - 机器唯一 ID 获取
+- `zod` - 参数验证

package/package.json

@@ -0,0 +1,25 @@
+{
+  "name": "eagle-rag",
+  "version": "1.0.1",
+  "description": "MCP Server对接Coze工作流，实现RAG问答",
+  "type": "module",
+  "main": "src/index.js",
+  "bin": {
+    "eagle-rag": "./src/index.js"
+  },
+  "scripts": {
+    "start": "node src/index.js"
+  },
+  "keywords": ["mcp", "coze", "rag", "ai", "cursor"],
+  "author": "",
+  "license": "MIT",
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.0.0",
+    "axios": "^1.6.0",
+    "node-machine-id": "^1.1.12",
+    "zod": "^3.22.0"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  }
+}

package/src/index.js

@@ -0,0 +1,139 @@
+#!/usr/bin/env node
+
+/**
+ * Coze RAG MCP Server
+ * 通过 npx 运行的 MCP Server，对接 Coze 工作流
+ */
+
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { z } from "zod";
+import axios from "axios";
+import nodeMachineId from "node-machine-id";
+
+const { machineIdSync } = nodeMachineId;
+
+// ==================== 配置 ====================
+// Coze API 配置
+const COZE_API_URL = "https://api.coze.cn/v1/workflow/run";
+// Coze API Token (写死)
+const COZE_API_TOKEN = "sat_pqqZVXMBiDTas0khz2x3vBqrCcicmfOt9XlNi4KrbtUsRtbroa8i1z5wfGrcXWdn";
+// Coze 工作流 ID
+const WORKFLOW_ID = "7610685653369798719";
+
+// ==================== 获取机器唯一 ID ====================
+let cachedMachineId = null;
+
+function getMachineId() {
+  if (!cachedMachineId) {
+    try {
+      // 获取机器唯一 ID
+      cachedMachineId = machineIdSync({ original: true });
+      console.error(`[Coze RAG MCP] Machine ID: ${cachedMachineId}`);
+    } catch (error) {
+      console.error(`[Coze RAG MCP] Failed to get machine ID: ${error.message}`);
+      // 使用默认 ID
+      cachedMachineId = "unknown";
+    }
+  }
+  return cachedMachineId;
+}
+
+// ==================== Coze API 调用 ====================
+async function callCozeWorkflow(userId, question) {
+  try {
+    const response = await axios.post(
+      COZE_API_URL,
+      {
+        workflow_id: WORKFLOW_ID,
+        parameters: {
+          user_id: userId,
+          question: question
+        }
+      },
+      {
+        headers: {
+          "Authorization": `Bearer ${COZE_API_TOKEN}`,
+          "Content-Type": "application/json"
+        },
+        timeout: 60000 // 60秒超时
+      }
+    );
+
+    return response.data;
+  } catch (error) {
+    console.error(`[Coze RAG MCP] API Error: ${error.message}`);
+    if (error.response) {
+      return {
+        code: error.response.status,
+        msg: error.message,
+        data: JSON.stringify({ error: error.response.data })
+      };
+    }
+    return {
+      code: -1,
+      msg: error.message,
+      data: JSON.stringify({ error: "Network error" })
+    };
+  }
+}
+
+// ==================== MCP Server 工具 ====================
+const server = new McpServer({
+  name: "coze-rag",
+  version: "1.0.0",
+  capabilities: {
+    resources: {},
+    tools: {},
+  },
+});
+
+// 注册问答工具
+server.tool(
+  "ask",
+  "向 Coze RAG 工作流提问，获取知识库相关答案",
+  {
+    question: z.string().describe("用户问题")
+  },
+  async ({ question }) => {
+    const userId = getMachineId();
+    console.error(`[Coze RAG MCP] Processing question: "${question}" from user: ${userId}`);
+
+    const result = await callCozeWorkflow(userId, question);
+
+    // 解析返回结果
+    let responseText = "";
+    if (result.code === 0) {
+      try {
+        // data 字段是 JSON 字符串，需要解析
+        const data = JSON.parse(result.data);
+        responseText = data.result || JSON.stringify(data);
+      } catch {
+        responseText = result.data;
+      }
+    } else {
+      responseText = `错误: ${result.msg}`;
+    }
+
+    return {
+      content: [
+        {
+          type: "text",
+          text: responseText
+        }
+      ]
+    };
+  }
+);
+
+// ==================== 启动 Server ====================
+async function main() {
+  const transport = new StdioServerTransport();
+  await server.connect(transport);
+  console.error("[Coze RAG MCP] Server started successfully");
+}
+
+main().catch((error) => {
+  console.error("[Coze RAG MCP] Server error:", error);
+  process.exit(1);
+});