coze-coding-dev-sdk 0.5.5 → 0.5.6

package/dist/cli/commands/knowledge.d.ts

@@ -0,0 +1,3 @@
+import { Command } from 'commander';
+export declare const knowledgeCommand: Command;
+//# sourceMappingURL=knowledge.d.ts.map

package/dist/cli/commands/knowledge.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"knowledge.d.ts","sourceRoot":"","sources":["../../../src/cli/commands/knowledge.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAMpC,eAAO,MAAM,gBAAgB,SACS,CAAC"}

package/dist/cli/commands/knowledge.js

@@ -0,0 +1,119 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.knowledgeCommand = void 0;
+const commander_1 = require("commander");
+const index_1 = require("../../index");
+const chalk_1 = __importDefault(require("chalk"));
+const ora_1 = __importDefault(require("ora"));
+const utils_1 = require("../utils");
+exports.knowledgeCommand = new commander_1.Command('knowledge')
+    .description('Knowledge Base tools');
+exports.knowledgeCommand.command('search')
+    .description('Search for knowledge chunks')
+    .requiredOption('-q, --query <text>', 'Search query')
+    .option('-d, --dataset <name>', 'Dataset names (tables) to search in (can be used multiple times)', (value, previous) => previous.concat([value]), [])
+    .option('-k, --top-k <number>', 'Number of results to return', '5')
+    .option('-m, --min-score <number>', 'Minimum similarity score', '0.0')
+    .option('-H, --header <header>', 'Custom HTTP header (format: "Key: Value" or "Key=Value", can be used multiple times)', (value, previous) => previous.concat([value]), [])
+    .option('--verbose', 'Enable verbose logging to print HTTP request details', false)
+    .action(async (options) => {
+    const spinner = (0, ora_1.default)('Searching...').start();
+    try {
+        const config = new index_1.Config();
+        const customHeaders = (0, utils_1.parseHeaders)(options.header);
+        const client = new index_1.KnowledgeClient(config, customHeaders, options.verbose);
+        const results = await client.search(options.query, options.dataset.length > 0 ? options.dataset : undefined, parseInt(options.topK), parseFloat(options.minScore));
+        if (results.code !== 0) {
+            spinner.fail(chalk_1.default.red(`Error ${results.code}: ${results.msg}`));
+            return;
+        }
+        spinner.succeed(chalk_1.default.green('Search completed!'));
+        console.log(chalk_1.default.bold('\nSearch Results:'));
+        console.log('────────────────────────────────────────────────────────────────────────────────');
+        if (results.chunks.length === 0) {
+            console.log(chalk_1.default.yellow('No matching chunks found.'));
+        }
+        results.chunks.forEach((chunk, index) => {
+            console.log(chalk_1.default.cyan(`[${index + 1}] Score: ${chunk.score.toFixed(4)}`));
+            if (chunk.doc_id)
+                console.log(chalk_1.default.dim(`Doc ID: ${chunk.doc_id}`));
+            if (chunk.chunk_id)
+                console.log(chalk_1.default.dim(`Chunk ID: ${chunk.chunk_id}`));
+            console.log(chalk_1.default.green('Content:'));
+            console.log(chunk.content);
+            console.log('────────────────────────────────────────────────────────────────────────────────');
+        });
+    }
+    catch (error) {
+        spinner.fail(chalk_1.default.red('Search failed'));
+        console.error(chalk_1.default.red(error.message));
+        process.exit(1);
+    }
+});
+exports.knowledgeCommand.command('add')
+    .description('Add documents to knowledge base')
+    .requiredOption('-d, --dataset <name>', 'Dataset name (table) to add to')
+    .option('-c, --content <text>', 'Raw text content to add (can be used multiple times)', (value, previous) => previous.concat([value]), [])
+    .option('-u, --url <url>', 'Web URL to add (can be used multiple times)', (value, previous) => previous.concat([value]), [])
+    .option('--chunk-separator <char>', 'Chunk separator', '\\n')
+    .option('--max-tokens <number>', 'Max tokens per chunk', '500')
+    .option('--remove-extra-spaces', 'Normalize extra spaces', false)
+    .option('--remove-urls-emails', 'Strip URLs and emails', false)
+    .option('-H, --header <header>', 'Custom HTTP header (format: "Key: Value" or "Key=Value", can be used multiple times)', (value, previous) => previous.concat([value]), [])
+    .option('--verbose', 'Enable verbose logging to print HTTP request details', false)
+    .action(async (options) => {
+    const spinner = (0, ora_1.default)('Adding documents...').start();
+    try {
+        const config = new index_1.Config();
+        const customHeaders = (0, utils_1.parseHeaders)(options.header);
+        const client = new index_1.KnowledgeClient(config, customHeaders, options.verbose);
+        const documents = [];
+        if (options.content) {
+            options.content.forEach((text) => {
+                documents.push({
+                    source: index_1.DataSourceType.TEXT,
+                    raw_data: text,
+                });
+            });
+        }
+        if (options.url) {
+            options.url.forEach((url) => {
+                documents.push({
+                    source: index_1.DataSourceType.URL,
+                    url: url,
+                });
+            });
+        }
+        if (documents.length === 0) {
+            spinner.fail(chalk_1.default.red('No content or URL provided to add.'));
+            return;
+        }
+        // Handle escaped newlines in chunk separator (e.g. from CLI input)
+        const separator = options.chunkSeparator.replace(/\\n/g, '\n');
+        const chunkConfig = {
+            separator: separator,
+            max_tokens: parseInt(options.maxTokens),
+            remove_extra_spaces: options.removeExtraSpaces,
+            remove_urls_emails: options.removeUrlsEmails,
+        };
+        const response = await client.addDocuments(documents, options.dataset, chunkConfig);
+        if (response.code !== 0) {
+            spinner.fail(chalk_1.default.red(`Error ${response.code}: ${response.msg}`));
+            return;
+        }
+        spinner.succeed(chalk_1.default.green('Documents added successfully!'));
+        if (response.doc_ids && response.doc_ids.length > 0) {
+            console.log(chalk_1.default.cyan(`\nAdded ${response.doc_ids.length} documents.`));
+            console.log(chalk_1.default.dim(`Doc IDs: ${response.doc_ids.join(', ')}`));
+        }
+    }
+    catch (error) {
+        spinner.fail(chalk_1.default.red('Add documents failed'));
+        console.error(chalk_1.default.red(error.message));
+        process.exit(1);
+    }
+});
+//# sourceMappingURL=knowledge.js.map

package/dist/cli/commands/knowledge.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"knowledge.js","sourceRoot":"","sources":["../../../src/cli/commands/knowledge.ts"],"names":[],"mappings":";;;;;;AAAA,yCAAoC;AACpC,uCAAsG;AACtG,kDAA0B;AAC1B,8CAAsB;AACtB,oCAAwC;AAE3B,QAAA,gBAAgB,GAAG,IAAI,mBAAO,CAAC,WAAW,CAAC;KACrD,WAAW,CAAC,sBAAsB,CAAC,CAAC;AAEvC,wBAAgB,CAAC,OAAO,CAAC,QAAQ,CAAC;KAC/B,WAAW,CAAC,6BAA6B,CAAC;KAC1C,cAAc,CAAC,oBAAoB,EAAE,cAAc,CAAC;KACpD,MAAM,CAAC,sBAAsB,EAAE,kEAAkE,EAAE,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAc,CAAC;KACjK,MAAM,CAAC,sBAAsB,EAAE,6BAA6B,EAAE,GAAG,CAAC;KAClE,MAAM,CAAC,0BAA0B,EAAE,0BAA0B,EAAE,KAAK,CAAC;KACrE,MAAM,CAAC,uBAAuB,EAAE,sFAAsF,EAAE,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAc,CAAC;KACtL,MAAM,CAAC,WAAW,EAAE,sDAAsD,EAAE,KAAK,CAAC;KAClF,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;IACxB,MAAM,OAAO,GAAG,IAAA,aAAG,EAAC,cAAc,CAAC,CAAC,KAAK,EAAE,CAAC;IAE5C,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,cAAM,EAAE,CAAC;QAC5B,MAAM,aAAa,GAAG,IAAA,oBAAY,EAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACnD,MAAM,MAAM,GAAG,IAAI,uBAAe,CAAC,MAAM,EAAE,aAAa,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;QAE3E,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,MAAM,CACjC,OAAO,CAAC,KAAK,EACb,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,EACxD,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,EACtB,UAAU,CAAC,OAAO,CAAC,QAAQ,CAAC,CAC7B,CAAC;QAEF,IAAI,OAAO,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YACvB,OAAO,CAAC,IAAI,CAAC,eAAK,CAAC,GAAG,CAAC,SAAS,OAAO,CAAC,IAAI,KAAK,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;YACjE,OAAO;QACT,CAAC;QAED,OAAO,CAAC,OAAO,CAAC,eAAK,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC,CAAC;QAElD,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC,CAAC;QAC7C,OAAO,CAAC,GAAG,CAAC,kFAAkF,CAAC,CAAC;QAEhG,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChC,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,MAAM,CAAC,2BAA2B,CAAC,CAAC,CAAC;QACzD,CAAC;QAED,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;YACtC,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,IAAI,KAAK,GAAG,CAAC,YAAY,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAC3E,IAAI,KAAK,CAAC,MAAM;gBAAE,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,GAAG,CAAC,WAAW,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YACpE,IAAI,KAAK,CAAC,QAAQ;gBAAE,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,GAAG,CAAC,aAAa,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;YAC1E,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC;YACrC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC3B,OAAO,CAAC,GAAG,CAAC,kFAAkF,CAAC,CAAC;QAClG,CAAC,CAAC,CAAC;IAEL,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QACpB,OAAO,CAAC,IAAI,CAAC,eAAK,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC;QACzC,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;QACxC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC,CAAC,CAAC;AAEL,wBAAgB,CAAC,OAAO,CAAC,KAAK,CAAC;KAC5B,WAAW,CAAC,iCAAiC,CAAC;KAC9C,cAAc,CAAC,sBAAsB,EAAE,gCAAgC,CAAC;KACxE,MAAM,CAAC,sBAAsB,EAAE,sDAAsD,EAAE,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAc,CAAC;KACrJ,MAAM,CAAC,iBAAiB,EAAE,6CAA6C,EAAE,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAc,CAAC;KACvI,MAAM,CAAC,0BAA0B,EAAE,iBAAiB,EAAE,KAAK,CAAC;KAC5D,MAAM,CAAC,uBAAuB,EAAE,sBAAsB,EAAE,KAAK,CAAC;KAC9D,MAAM,CAAC,uBAAuB,EAAE,wBAAwB,EAAE,KAAK,CAAC;KAChE,MAAM,CAAC,sBAAsB,EAAE,uBAAuB,EAAE,KAAK,CAAC;KAC9D,MAAM,CAAC,uBAAuB,EAAE,sFAAsF,EAAE,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAc,CAAC;KACtL,MAAM,CAAC,WAAW,EAAE,sDAAsD,EAAE,KAAK,CAAC;KAClF,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;IACxB,MAAM,OAAO,GAAG,IAAA,aAAG,EAAC,qBAAqB,CAAC,CAAC,KAAK,EAAE,CAAC;IAEnD,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,cAAM,EAAE,CAAC;QAC5B,MAAM,aAAa,GAAG,IAAA,oBAAY,EAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACnD,MAAM,MAAM,GAAG,IAAI,uBAAe,CAAC,MAAM,EAAE,aAAa,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;QAE3E,MAAM,SAAS,GAAwB,EAAE,CAAC;QAE1C,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACpB,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAY,EAAE,EAAE;gBACvC,SAAS,CAAC,IAAI,CAAC;oBACb,MAAM,EAAE,sBAAc,CAAC,IAAI;oBAC3B,QAAQ,EAAE,IAAI;iBACf,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;YAChB,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,GAAW,EAAE,EAAE;gBAClC,SAAS,CAAC,IAAI,CAAC;oBACb,MAAM,EAAE,sBAAc,CAAC,GAAG;oBAC1B,GAAG,EAAE,GAAG;iBACT,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3B,OAAO,CAAC,IAAI,CAAC,eAAK,CAAC,GAAG,CAAC,oCAAoC,CAAC,CAAC,CAAC;YAC9D,OAAO;QACT,CAAC;QAED,mEAAmE;QACnE,MAAM,SAAS,GAAG,OAAO,CAAC,cAAc,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAE/D,MAAM,WAAW,GAAgB;YAC/B,SAAS,EAAE,SAAS;YACpB,UAAU,EAAE,QAAQ,CAAC,OAAO,CAAC,SAAS,CAAC;YACvC,mBAAmB,EAAE,OAAO,CAAC,iBAAiB;YAC9C,kBAAkB,EAAE,OAAO,CAAC,gBAAgB;SAC7C,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,YAAY,CACxC,SAAS,EACT,OAAO,CAAC,OAAO,EACf,WAAW,CACZ,CAAC;QAEF,IAAI,QAAQ,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YACxB,OAAO,CAAC,IAAI,CAAC,eAAK,CAAC,GAAG,CAAC,SAAS,QAAQ,CAAC,IAAI,KAAK,QAAQ,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;YACnE,OAAO;QACT,CAAC;QAED,OAAO,CAAC,OAAO,CAAC,eAAK,CAAC,KAAK,CAAC,+BAA+B,CAAC,CAAC,CAAC;QAE9D,IAAI,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpD,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,WAAW,QAAQ,CAAC,OAAO,CAAC,MAAM,aAAa,CAAC,CAAC,CAAC;YACzE,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,GAAG,CAAC,YAAY,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;QACpE,CAAC;IAEH,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QACpB,OAAO,CAAC,IAAI,CAAC,eAAK,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC,CAAC;QAChD,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;QACxC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC,CAAC,CAAC"}

package/dist/cli/index.js

@@ -7,6 +7,7 @@ const search_1 = require("./commands/search");
 const voice_1 = require("./commands/voice");
 const chat_1 = require("./commands/chat");
 const db_1 = require("./commands/db");
+const knowledge_1 = require("./commands/knowledge");
 const program = new commander_1.Command();
 program
     .name('coze-coding-ai')
@@ -19,5 +20,6 @@ program.addCommand(voice_1.ttsCommand);
 program.addCommand(voice_1.asrCommand);
 program.addCommand(chat_1.chatCommand);
 program.addCommand(db_1.dbCommand);
+program.addCommand(knowledge_1.knowledgeCommand);
 program.parse(process.argv);
 //# sourceMappingURL=index.js.map

package/dist/cli/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/cli/index.ts"],"names":[],"mappings":";;AAAA,yCAAoC;AACpC,4CAAgD;AAChD,4CAAgD;AAChD,8CAAkD;AAClD,4CAA0D;AAC1D,0CAA8C;AAC9C,sCAA0C;AAE1C,MAAM,OAAO,GAAG,IAAI,mBAAO,EAAE,CAAC;AAE9B,OAAO;KACJ,IAAI,CAAC,gBAAgB,CAAC;KACtB,WAAW,CAAC,oGAAoG,CAAC;KACjH,OAAO,CAAC,OAAO,CAAC,CAAC;AACpB,OAAO,CAAC,UAAU,CAAC,oBAAY,CAAC,CAAC;AACjC,OAAO,CAAC,UAAU,CAAC,oBAAY,CAAC,CAAC;AACjC,OAAO,CAAC,UAAU,CAAC,sBAAa,CAAC,CAAC;AAClC,OAAO,CAAC,UAAU,CAAC,kBAAU,CAAC,CAAC;AAC/B,OAAO,CAAC,UAAU,CAAC,kBAAU,CAAC,CAAC;AAC/B,OAAO,CAAC,UAAU,CAAC,kBAAW,CAAC,CAAC;AAChC,OAAO,CAAC,UAAU,CAAC,cAAS,CAAC,CAAC;AAE9B,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/cli/index.ts"],"names":[],"mappings":";;AAAA,yCAAoC;AACpC,4CAAgD;AAChD,4CAAgD;AAChD,8CAAkD;AAClD,4CAA0D;AAC1D,0CAA8C;AAC9C,sCAA0C;AAC1C,oDAAwD;AAExD,MAAM,OAAO,GAAG,IAAI,mBAAO,EAAE,CAAC;AAE9B,OAAO;KACJ,IAAI,CAAC,gBAAgB,CAAC;KACtB,WAAW,CAAC,oGAAoG,CAAC;KACjH,OAAO,CAAC,OAAO,CAAC,CAAC;AACpB,OAAO,CAAC,UAAU,CAAC,oBAAY,CAAC,CAAC;AACjC,OAAO,CAAC,UAAU,CAAC,oBAAY,CAAC,CAAC;AACjC,OAAO,CAAC,UAAU,CAAC,sBAAa,CAAC,CAAC;AAClC,OAAO,CAAC,UAAU,CAAC,kBAAU,CAAC,CAAC;AAC/B,OAAO,CAAC,UAAU,CAAC,kBAAU,CAAC,CAAC;AAC/B,OAAO,CAAC,UAAU,CAAC,kBAAW,CAAC,CAAC;AAChC,OAAO,CAAC,UAAU,CAAC,cAAS,CAAC,CAAC;AAC9B,OAAO,CAAC,UAAU,CAAC,4BAAgB,CAAC,CAAC;AAErC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC"}

package/dist/index.d.ts

@@ -4,6 +4,7 @@ export * from './search';
 export * from './voice';
 export * from './video';
 export * from './llm';
+export * from './knowledge';
 export * from './database';
 export * from './s3';
 export { VERSION } from './version';

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,QAAQ,CAAC;AACvB,cAAc,SAAS,CAAC;AACxB,cAAc,UAAU,CAAC;AACzB,cAAc,SAAS,CAAC;AACxB,cAAc,SAAS,CAAC;AACxB,cAAc,OAAO,CAAC;AACtB,cAAc,YAAY,CAAC;AAC3B,cAAc,MAAM,CAAC;AACrB,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,QAAQ,CAAC;AACvB,cAAc,SAAS,CAAC;AACxB,cAAc,UAAU,CAAC;AACzB,cAAc,SAAS,CAAC;AACxB,cAAc,SAAS,CAAC;AACxB,cAAc,OAAO,CAAC;AACtB,cAAc,aAAa,CAAC;AAC5B,cAAc,YAAY,CAAC;AAC3B,cAAc,MAAM,CAAC;AACrB,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC"}

package/dist/index.js

@@ -21,6 +21,7 @@ __exportStar(require("./search"), exports);
 __exportStar(require("./voice"), exports);
 __exportStar(require("./video"), exports);
 __exportStar(require("./llm"), exports);
+__exportStar(require("./knowledge"), exports);
 __exportStar(require("./database"), exports);
 __exportStar(require("./s3"), exports);
 var version_1 = require("./version");

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AAAA,yCAAuB;AACvB,0CAAwB;AACxB,2CAAyB;AACzB,0CAAwB;AACxB,0CAAwB;AACxB,wCAAsB;AACtB,6CAA2B;AAC3B,uCAAqB;AACrB,qCAAoC;AAA3B,kGAAA,OAAO,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AAAA,yCAAuB;AACvB,0CAAwB;AACxB,2CAAyB;AACzB,0CAAwB;AACxB,0CAAwB;AACxB,wCAAsB;AACtB,8CAA4B;AAC5B,6CAA2B;AAC3B,uCAAqB;AACrB,qCAAoC;AAA3B,kGAAA,OAAO,OAAA"}

package/dist/knowledge/client.d.ts

@@ -0,0 +1,10 @@
+import { BaseClient } from '../core/client';
+import { Config } from '../core/config';
+import { ChunkConfig, KnowledgeDocument, KnowledgeInsertResponse, KnowledgeSearchResponse } from './models';
+export declare class KnowledgeClient extends BaseClient {
+    private baseUrl;
+    constructor(config?: Config, customHeaders?: Record<string, string>, verbose?: boolean);
+    search(query: string, tableNames?: string[], topK?: number, minScore?: number, extraHeaders?: Record<string, string>): Promise<KnowledgeSearchResponse>;
+    addDocuments(documents: KnowledgeDocument[], tableName: string, chunkConfig?: ChunkConfig, extraHeaders?: Record<string, string>): Promise<KnowledgeInsertResponse>;
+}
+//# sourceMappingURL=client.d.ts.map

package/dist/knowledge/client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../src/knowledge/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5C,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AACxC,OAAO,EACL,WAAW,EAEX,iBAAiB,EACjB,uBAAuB,EACvB,uBAAuB,EACxB,MAAM,UAAU,CAAC;AAElB,qBAAa,eAAgB,SAAQ,UAAU;IAC7C,OAAO,CAAC,OAAO,CAAS;gBAGtB,MAAM,CAAC,EAAE,MAAM,EACf,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EACtC,OAAO,GAAE,OAAe;IAapB,MAAM,CACV,KAAK,EAAE,MAAM,EACb,UAAU,CAAC,EAAE,MAAM,EAAE,EACrB,IAAI,GAAE,MAAU,EAChB,QAAQ,GAAE,MAAY,EACtB,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GACpC,OAAO,CAAC,uBAAuB,CAAC;IA0C7B,YAAY,CAChB,SAAS,EAAE,iBAAiB,EAAE,EAC9B,SAAS,EAAE,MAAM,EACjB,WAAW,CAAC,EAAE,WAAW,EACzB,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GACpC,OAAO,CAAC,uBAAuB,CAAC;CAuCpC"}

package/dist/knowledge/client.js

@@ -0,0 +1,73 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.KnowledgeClient = void 0;
+const client_1 = require("../core/client");
+class KnowledgeClient extends client_1.BaseClient {
+    constructor(config, customHeaders, verbose = false) {
+        super(config, customHeaders, verbose);
+        // Use COZE_INTEGRATION_BASE_URL as endpoint, defaults to config.baseUrl
+        this.baseUrl = this.config.baseUrl;
+        // Add default PPE headers for testing
+        Object.assign(this.axiosInstance.defaults.headers.common, {
+            'x-tt-env': 'ppe_coze_knowledge',
+            'x-use-ppe': '1',
+        });
+    }
+    async search(query, tableNames, topK = 5, minScore = 0.0, extraHeaders) {
+        const payload = {
+            query,
+            top_k: topK,
+            min_score: minScore,
+        };
+        if (tableNames) {
+            payload.dataset = tableNames;
+        }
+        const url = `${this.baseUrl}/v1/knowledge_base/recall`;
+        const headers = { ...extraHeaders };
+        if (this.config.apiKey) {
+            headers['x-coze-token'] = `Bearer ${this.config.apiKey}`;
+        }
+        const response = await this.request('POST', url, payload, { headers });
+        // Response structure: { "data": [ ... ], "code": 0, "msg": "" }
+        const dataList = response.data || [];
+        const chunks = dataList.map((item) => ({
+            content: item.slice || '',
+            score: item.score || 0.0,
+            chunk_id: item.chunk_id,
+            doc_id: item.doc_id,
+        }));
+        return {
+            chunks,
+            code: response.code || 0,
+            msg: response.msg || '',
+        };
+    }
+    async addDocuments(documents, tableName, chunkConfig, extraHeaders) {
+        const payload = {
+            dataset: tableName,
+            data: documents, // TypeScript automatically serializes objects, so no need for explicit to_api_format equivalent if keys match
+        };
+        if (chunkConfig) {
+            // Remove undefined fields
+            const cleanChunkConfig = Object.fromEntries(Object.entries(chunkConfig).filter(([_, v]) => v !== undefined));
+            payload.chunk_config = cleanChunkConfig;
+        }
+        const url = `${this.baseUrl}/v1/knowledge_base/batch_import`;
+        const headers = { ...extraHeaders };
+        if (this.config.apiKey) {
+            headers['x-coze-token'] = `Bearer ${this.config.apiKey}`;
+        }
+        const response = await this.request('POST', url, payload, { headers });
+        let msg = response.msg || '';
+        if (response.code === 0) {
+            msg = "成功！文档正在异步导入中，请稍候。您可以查看数据库中的 'knowledge' schema 来验证导入状态。";
+        }
+        return {
+            doc_ids: response.doc_ids,
+            code: response.code || 0,
+            msg: msg,
+        };
+    }
+}
+exports.KnowledgeClient = KnowledgeClient;
+//# sourceMappingURL=client.js.map

package/dist/knowledge/client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/knowledge/client.ts"],"names":[],"mappings":";;;AAAA,2CAA4C;AAU5C,MAAa,eAAgB,SAAQ,mBAAU;IAG7C,YACE,MAAe,EACf,aAAsC,EACtC,UAAmB,KAAK;QAExB,KAAK,CAAC,MAAM,EAAE,aAAa,EAAE,OAAO,CAAC,CAAC;QACtC,wEAAwE;QACxE,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;QAEnC,sCAAsC;QACtC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE;YACxD,UAAU,EAAE,oBAAoB;YAChC,WAAW,EAAE,GAAG;SACjB,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,MAAM,CACV,KAAa,EACb,UAAqB,EACrB,OAAe,CAAC,EAChB,WAAmB,GAAG,EACtB,YAAqC;QAErC,MAAM,OAAO,GAAQ;YACnB,KAAK;YACL,KAAK,EAAE,IAAI;YACX,SAAS,EAAE,QAAQ;SACpB,CAAC;QAEF,IAAI,UAAU,EAAE,CAAC;YACf,OAAO,CAAC,OAAO,GAAG,UAAU,CAAC;QAC/B,CAAC;QAED,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,2BAA2B,CAAC;QAEvD,MAAM,OAAO,GAA2B,EAAE,GAAG,YAAY,EAAE,CAAC;QAC5D,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACvB,OAAO,CAAC,cAAc,CAAC,GAAG,UAAU,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QAC3D,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CACjC,MAAM,EACN,GAAG,EACH,OAAO,EACP,EAAE,OAAO,EAAE,CACZ,CAAC;QAEF,gEAAgE;QAChE,MAAM,QAAQ,GAAG,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAC;QAErC,MAAM,MAAM,GAAqB,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,CAAC;YAC5D,OAAO,EAAE,IAAI,CAAC,KAAK,IAAI,EAAE;YACzB,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,GAAG;YACxB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC,CAAC,CAAC;QAEJ,OAAO;YACL,MAAM;YACN,IAAI,EAAE,QAAQ,CAAC,IAAI,IAAI,CAAC;YACxB,GAAG,EAAE,QAAQ,CAAC,GAAG,IAAI,EAAE;SACxB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,SAA8B,EAC9B,SAAiB,EACjB,WAAyB,EACzB,YAAqC;QAErC,MAAM,OAAO,GAAQ;YACnB,OAAO,EAAE,SAAS;YAClB,IAAI,EAAE,SAAS,EAAE,8GAA8G;SAChI,CAAC;QAEF,IAAI,WAAW,EAAE,CAAC;YAChB,0BAA0B;YAC1B,MAAM,gBAAgB,GAAG,MAAM,CAAC,WAAW,CACzC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAChE,CAAC;YACF,OAAO,CAAC,YAAY,GAAG,gBAAgB,CAAC;QAC1C,CAAC;QAED,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,iCAAiC,CAAC;QAE7D,MAAM,OAAO,GAA2B,EAAE,GAAG,YAAY,EAAE,CAAC;QAC5D,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACvB,OAAO,CAAC,cAAc,CAAC,GAAG,UAAU,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QAC3D,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CACjC,MAAM,EACN,GAAG,EACH,OAAO,EACP,EAAE,OAAO,EAAE,CACZ,CAAC;QAEF,IAAI,GAAG,GAAG,QAAQ,CAAC,GAAG,IAAI,EAAE,CAAC;QAC7B,IAAI,QAAQ,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YACxB,GAAG,GAAG,yDAAyD,CAAC;QAClE,CAAC;QAED,OAAO;YACL,OAAO,EAAE,QAAQ,CAAC,OAAO;YACzB,IAAI,EAAE,QAAQ,CAAC,IAAI,IAAI,CAAC;YACxB,GAAG,EAAE,GAAG;SACT,CAAC;IACJ,CAAC;CACF;AA/GD,0CA+GC"}

package/dist/knowledge/index.d.ts

@@ -0,0 +1,3 @@
+export * from './models';
+export * from './client';
+//# sourceMappingURL=index.d.ts.map

package/dist/knowledge/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/knowledge/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAC;AACzB,cAAc,UAAU,CAAC"}

package/dist/knowledge/index.js

@@ -0,0 +1,19 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./models"), exports);
+__exportStar(require("./client"), exports);
+//# sourceMappingURL=index.js.map

package/dist/knowledge/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/knowledge/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,2CAAyB;AACzB,2CAAyB"}

package/dist/knowledge/models.d.ts

@@ -0,0 +1,34 @@
+export declare enum DataSourceType {
+    TEXT = 0,
+    URL = 1,
+    URI = 2
+}
+export interface ChunkConfig {
+    separator: string;
+    max_tokens: number;
+    remove_extra_spaces?: boolean;
+    remove_urls_emails?: boolean;
+}
+export interface KnowledgeDocument {
+    source: DataSourceType;
+    raw_data?: string;
+    uri?: string;
+    url?: string;
+}
+export interface KnowledgeChunk {
+    content: string;
+    score: number;
+    chunk_id?: string;
+    doc_id?: string;
+}
+export interface KnowledgeSearchResponse {
+    chunks: KnowledgeChunk[];
+    code: number;
+    msg: string;
+}
+export interface KnowledgeInsertResponse {
+    doc_ids?: string[];
+    code: number;
+    msg: string;
+}
+//# sourceMappingURL=models.d.ts.map

package/dist/knowledge/models.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../../src/knowledge/models.ts"],"names":[],"mappings":"AAAA,oBAAY,cAAc;IACxB,IAAI,IAAI;IACR,GAAG,IAAI;IACP,GAAG,IAAI;CACR;AAED,MAAM,WAAW,WAAW;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,kBAAkB,CAAC,EAAE,OAAO,CAAC;CAC9B;AAED,MAAM,WAAW,iBAAiB;IAChC,MAAM,EAAE,cAAc,CAAC;IACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,GAAG,CAAC,EAAE,MAAM,CAAC;CACd;AAED,MAAM,WAAW,cAAc;IAC7B,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,uBAAuB;IACtC,MAAM,EAAE,cAAc,EAAE,CAAC;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,EAAE,MAAM,CAAC;CACb;AAED,MAAM,WAAW,uBAAuB;IACtC,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,EAAE,MAAM,CAAC;CACb"}

package/dist/knowledge/models.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DataSourceType = void 0;
+var DataSourceType;
+(function (DataSourceType) {
+    DataSourceType[DataSourceType["TEXT"] = 0] = "TEXT";
+    DataSourceType[DataSourceType["URL"] = 1] = "URL";
+    DataSourceType[DataSourceType["URI"] = 2] = "URI";
+})(DataSourceType || (exports.DataSourceType = DataSourceType = {}));
+//# sourceMappingURL=models.js.map

package/dist/knowledge/models.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"models.js","sourceRoot":"","sources":["../../src/knowledge/models.ts"],"names":[],"mappings":";;;AAAA,IAAY,cAIX;AAJD,WAAY,cAAc;IACxB,mDAAQ,CAAA;IACR,iDAAO,CAAA;IACP,iDAAO,CAAA;AACT,CAAC,EAJW,cAAc,8BAAd,cAAc,QAIzB"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "coze-coding-dev-sdk",
-  "version": "0.5.5",
+  "version": "0.5.6",
   "description": "Coze Coding Dev SDK - 优雅的多功能 AI SDK，支持图片生成、视频生成、语音合成、语音识别、大语言模型和联网搜索",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",