@cnbcool/mcp-server 0.6.1 → 0.6.3

package/README.md

@@ -43,6 +43,8 @@ CNB(https://cnb.cool) toolkits for LLMs supporting the MCP protocol
 | cnb_stopBuild | Stop a build |
 | cnb_list_workspaces | Get a list of the current user's cloud-native development environments on the CNB platform |
 | cnb_delete_workspace | Delete a specified cloud-native development environment on the CNB platform |
+| cnb_getKnowledgeBaseInfo | Get knowledge base information |
+| cnb_queryKnowledgeBase | Query knowledge base |
 
 ## How to use
 

package/README_zh.md

@@ -43,6 +43,8 @@ CNB(https://cnb.cool) 支持 MCP 协议的 MCP Server
 | cnb_stopBuild | 停止一个构建 |
 | cnb_list_workspaces | 获取当前用户在CNB平台的云原生开发环境列表 |
 | cnb_delete_workspace | 在CNB平台删除指定的云原生开发环境 |
+| cnb_getKnowledgeBaseInfo | 获取知识库信息 |
+| cnb_queryKnowledgeBase | 查询知识库 |
 
 ## 使用方法
 

package/dist/api/interfaces/dto.queryknowledgebasereq.js

@@ -0,0 +1,15 @@
+"use strict";
+// @ts-nocheck
+/* tslint:disable */
+/* eslint-disable */
+/*
+ * -------------------------------------------------------------------------
+ * ## THIS FILE WAS GENERATED VIA CNB-API-GENERATE                        ##
+ * ##                                                                     ##
+ * ## AUTHOR: bapelin                                                     ##
+ * ## SOURCE: https://cnb.woa.com/cnb/frontend-science/cnb-api-generate   ##
+ * -------------------------------------------------------------------------
+ * @Version 1.5.2
+ * @Source dto.QueryKnowledgeBaseReq
+ */
+Object.defineProperty(exports, "__esModule", { value: true });

package/dist/api/interfaces/dto.queryknowledgebaseres.js

@@ -0,0 +1,15 @@
+"use strict";
+// @ts-nocheck
+/* tslint:disable */
+/* eslint-disable */
+/*
+ * -------------------------------------------------------------------------
+ * ## THIS FILE WAS GENERATED VIA CNB-API-GENERATE                        ##
+ * ##                                                                     ##
+ * ## AUTHOR: bapelin                                                     ##
+ * ## SOURCE: https://cnb.woa.com/cnb/frontend-science/cnb-api-generate   ##
+ * -------------------------------------------------------------------------
+ * @Version 1.5.2
+ * @Source dto.QueryKnowledgeBaseRes
+ */
+Object.defineProperty(exports, "__esModule", { value: true });

package/dist/api/knowledge-base/get-knowledge-base-info.js

@@ -0,0 +1,42 @@
+"use strict";
+// @ts-nocheck
+/* tslint:disable */
+/* eslint-disable */
+/*
+ * -------------------------------------------------------------------------
+ * ## THIS FILE WAS GENERATED VIA CNB-API-GENERATE                        ##
+ * ##                                                                     ##
+ * ## AUTHOR: bapelin                                                     ##
+ * ## SOURCE: https://cnb.woa.com/cnb/frontend-science/cnb-api-generate   ##
+ * -------------------------------------------------------------------------
+ * @Version 1.5.2
+ * @Source /{repo}/-/knowledge/base
+ */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getKnowledgeBaseInfo = getKnowledgeBaseInfo;
+const request_1 = __importDefault(require("../../helpers/request"));
+/**
+* @description 访问令牌调用此接口需包含以下权限。Required permissions for access token.
+* repo-code:r
+* @tags KnowledgeBase
+* @name getKnowledgeBaseInfo
+* @summary 获取知识库信息
+* @request get:/{repo}/-/knowledge/base
+
+----------------------------------
+* @param {string} arg0
+* @param {RequestConfig} arg1 - Other reuqest params
+*/
+async function getKnowledgeBaseInfo(repo, { req, options, ...axiosConfig } = {}) {
+    return await request_1.default.request({
+        ...axiosConfig,
+        _next_req: req,
+        options: options,
+        url: `/${repo}/-/knowledge/base`,
+        _apiTag: '/{repo}/-/knowledge/base',
+        method: 'get'
+    });
+}

package/dist/api/knowledge-base/query-knowledge-base.js

@@ -0,0 +1,44 @@
+"use strict";
+// @ts-nocheck
+/* tslint:disable */
+/* eslint-disable */
+/*
+ * -------------------------------------------------------------------------
+ * ## THIS FILE WAS GENERATED VIA CNB-API-GENERATE                        ##
+ * ##                                                                     ##
+ * ## AUTHOR: bapelin                                                     ##
+ * ## SOURCE: https://cnb.woa.com/cnb/frontend-science/cnb-api-generate   ##
+ * -------------------------------------------------------------------------
+ * @Version 1.5.2
+ * @Source /{repo}/-/knowledge/base/query
+ */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.queryKnowledgeBase = queryKnowledgeBase;
+const request_1 = __importDefault(require("../../helpers/request"));
+/**
+* @description 访问令牌调用此接口需包含以下权限。Required permissions for access token.
+* repo-code:r
+* @tags KnowledgeBase
+* @name queryKnowledgeBase
+* @summary 查询知识库，使用文档：https://docs.cnb.cool/zh/ai/knowledge-base.html
+* @request post:/{repo}/-/knowledge/base/query
+
+----------------------------------
+* @param {string} arg0
+* @param {DtoQueryKnowledgeBaseReq} arg1
+* @param {RequestConfig} arg2 - Other reuqest params
+*/
+async function queryKnowledgeBase(repo, query, { req, options, ...axiosConfig } = {}) {
+    return await request_1.default.request({
+        ...axiosConfig,
+        _next_req: req,
+        options: options,
+        url: `/${repo}/-/knowledge/base/query`,
+        _apiTag: '/{repo}/-/knowledge/base/query',
+        method: 'post',
+        data: query
+    });
+}

package/dist/constants/toolDescriptions.js

@@ -45,5 +45,8 @@ exports.toolDescriptions = {
     [toolNames_1.ToolNames.STOP_BUILD]: '停止一个构建',
     // 云原生开发工具
     [toolNames_1.ToolNames.LIST_WORKSPACES]: '获取当前用户在CNB平台的云原生开发环境列表',
-    [toolNames_1.ToolNames.DELETE_WORKSPACE]: '在CNB平台删除指定的云原生开发环境'
+    [toolNames_1.ToolNames.DELETE_WORKSPACE]: '在CNB平台删除指定的云原生开发环境',
+    // 知识库工具
+    [toolNames_1.ToolNames.GET_KNOWLEDGE_BASE_INFO]: '获取知识库信息',
+    [toolNames_1.ToolNames.QUERY_KNOWLEDGE_BASE]: '查询知识库'
 };

package/dist/constants/toolNames.js

@@ -46,4 +46,7 @@ var ToolNames;
     // 云原生开发工具
     ToolNames["LIST_WORKSPACES"] = "cnb_list_workspaces";
     ToolNames["DELETE_WORKSPACE"] = "cnb_delete_workspace";
+    // 知识库工具
+    ToolNames["GET_KNOWLEDGE_BASE_INFO"] = "cnb_getKnowledgeBaseInfo";
+    ToolNames["QUERY_KNOWLEDGE_BASE"] = "cnb_queryKnowledgeBase";
 })(ToolNames || (exports.ToolNames = ToolNames = {}));

package/dist/helpers/{getToken.js → getConfig.js}

@@ -1,6 +1,17 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.default = getToken;
+exports.getBaseUrl = getBaseUrl;
+exports.getToken = getToken;
+/**
+ * baseUrl 降级获取
+ *
+ * 环境变量 API_BASE_URL（mcp 客户端配置）--不存在--> CNB 流水线提供的 CNB_API_ENDPOINT ---不存在--> 'https://api.cnb.cool'
+ */
+function getBaseUrl() {
+    var _a, _b;
+    const baseUrl = (_b = (_a = process.env.API_BASE_URL) !== null && _a !== void 0 ? _a : process.env.CNB_API_ENDPOINT) !== null && _b !== void 0 ? _b : 'https://api.cnb.cool';
+    return baseUrl;
+}
 /**
  * token 降级获取
  *

package/dist/helpers/request/index.js

@@ -1,12 +1,9 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.createHeaders = createHeaders;
-const getToken_1 = __importDefault(require("../getToken"));
+const getConfig_1 = require("../getConfig");
 const cnb_1 = require("./cnb");
-const baseURL = process.env.API_BASE_URL || 'https://api.cnb.cool';
+const baseURL = (0, getConfig_1.getBaseUrl)();
 const timeout = 10000; // 客户端10秒超时
 const CnbFetcher = new cnb_1.CnbRequest({
     baseURL,
@@ -15,7 +12,7 @@ const CnbFetcher = new cnb_1.CnbRequest({
 exports.default = CnbFetcher;
 function createHeaders(token) {
     return {
-        Authorization: `Bearer ${(0, getToken_1.default)(token)}`,
+        Authorization: `Bearer ${(0, getConfig_1.getToken)(token)}`,
         Accept: 'application/vnd.cnb.api+json'
     };
 }

package/dist/tools/index.js

@@ -4,7 +4,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.registerTools = registerTools;
-const getToken_js_1 = __importDefault(require("../helpers/getToken.js"));
+const getConfig_js_1 = require("../helpers/getConfig.js");
 const client_js_1 = __importDefault(require("../api/client.js"));
 const groupTools_js_1 = __importDefault(require("./groupTools.js"));
 const repoTools_js_1 = __importDefault(require("./repoTools.js"));
@@ -12,10 +12,11 @@ const issueTools_js_1 = __importDefault(require("./issueTools.js"));
 const workspaceTools_js_1 = __importDefault(require("./workspaceTools.js"));
 const pullTools_js_1 = __importDefault(require("./pullTools.js"));
 const buildTools_js_1 = __importDefault(require("./buildTools.js"));
+const knowledgeBaseTools_js_1 = __importDefault(require("./knowledgeBaseTools.js"));
 function registerTools(server, token) {
     const client = new client_js_1.default({
-        baseUrl: process.env.API_BASE_URL || 'https://api.cnb.cool',
-        token: (0, getToken_js_1.default)(token)
+        baseUrl: (0, getConfig_js_1.getBaseUrl)(),
+        token: (0, getConfig_js_1.getToken)(token)
     });
     (0, groupTools_js_1.default)(server, client);
     (0, repoTools_js_1.default)(server, client);
@@ -26,4 +27,5 @@ function registerTools(server, token) {
     // 使用 cnb-request
     // ================
     (0, buildTools_js_1.default)(server, token);
+    (0, knowledgeBaseTools_js_1.default)(server, token);
 }

package/dist/tools/knowledgeBaseTools.js

@@ -0,0 +1,64 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.default = registerKnowledgeBaseTools;
+const zod_1 = require("zod");
+const toolNames_js_1 = require("../constants/toolNames.js");
+const toolDescriptions_js_1 = require("../constants/toolDescriptions.js");
+const formatToolResult_js_1 = require("../helpers/formatToolResult.js");
+const index_js_1 = require("../helpers/request/index.js");
+const get_knowledge_base_info_js_1 = require("../api/knowledge-base/get-knowledge-base-info.js");
+const query_knowledge_base_js_1 = require("../api/knowledge-base/query-knowledge-base.js");
+function registerKnowledgeBaseTools(server, token) {
+    const headers = (0, index_js_1.createHeaders)(token);
+    server.tool(toolNames_js_1.ToolNames.GET_KNOWLEDGE_BASE_INFO, toolDescriptions_js_1.toolDescriptions[toolNames_js_1.ToolNames.GET_KNOWLEDGE_BASE_INFO], {
+        repo: zod_1.z.string().describe('仓库路径')
+    }, async ({ repo }) => {
+        try {
+            const { error, result } = await (0, get_knowledge_base_info_js_1.getKnowledgeBaseInfo)(repo, { headers });
+            if (error)
+                throw new Error(`ERROR ${error.code}: ${error.message}`);
+            return (0, formatToolResult_js_1.formatTextToolResult)(JSON.stringify(result, null, 2), toolNames_js_1.ToolNames.GET_KNOWLEDGE_BASE_INFO);
+        }
+        catch (error) {
+            return (0, formatToolResult_js_1.formatToolError)(error, toolNames_js_1.ToolNames.GET_KNOWLEDGE_BASE_INFO);
+        }
+    });
+    server.tool(toolNames_js_1.ToolNames.QUERY_KNOWLEDGE_BASE, toolDescriptions_js_1.toolDescriptions[toolNames_js_1.ToolNames.QUERY_KNOWLEDGE_BASE], {
+        repo: zod_1.z.string().describe('仓库路径'),
+        query: zod_1.z.string().describe('要查询的关键词或问题'),
+        score_threshold: zod_1.z.number().default(0).describe('匹配相关性分数阈值'),
+        top_k: zod_1.z.number().default(5).describe('返回结果的最大数量'),
+        metadata_filtering_conditions: zod_1.z
+            .object({
+            conditions: zod_1.z.array(zod_1.z.object({
+                comparison_operator: zod_1.z
+                    .enum([
+                    'is',
+                    'is not',
+                    'contains',
+                    'not contains',
+                    'starts with',
+                    'ends with',
+                    'is empty',
+                    'is not empty'
+                ])
+                    .describe('运算符'),
+                name: zod_1.z.enum(['position', 'path', 'type']).describe('字段名称'),
+                value: zod_1.z.string().describe('比较值。运算符 "is empty" 和 "is not empty" 时忽略此字段')
+            })),
+            logical_operator: zod_1.z.enum(['and', 'or']).default('and').describe('逻辑连接条件')
+        })
+            .optional()
+            .describe('元数据过滤条件')
+    }, async ({ repo, query, score_threshold, top_k, metadata_filtering_conditions }) => {
+        try {
+            const { error, result } = await (0, query_knowledge_base_js_1.queryKnowledgeBase)(repo, { query, score_threshold, top_k, metadata_filtering_conditions }, { headers });
+            if (error)
+                throw new Error(`ERROR ${error.code}: ${error.message}`);
+            return (0, formatToolResult_js_1.formatTextToolResult)(JSON.stringify(result, null, 2), toolNames_js_1.ToolNames.QUERY_KNOWLEDGE_BASE);
+        }
+        catch (error) {
+            return (0, formatToolResult_js_1.formatToolError)(error, toolNames_js_1.ToolNames.QUERY_KNOWLEDGE_BASE);
+        }
+    });
+}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@cnbcool/mcp-server",
   "description": "CNB MCP Server. A comprehensive MCP server that provides seamless integration to the CNB's API(https://cnb.cool), offering a wide range of tools for repository management, pipelines operations and collaboration features",
-  "version": "0.6.1",
+  "version": "0.6.3",
   "main": "./dist/stdio.js",
   "bin": {
     "cnb-mcp-stdio": "dist/stdio.js",