autocoder-rag-sdk 0.0.1

package/LICENSE

@@ -0,0 +1,22 @@
+MIT License
+
+Copyright (c) 2024 AutoCoder Team
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.
+

package/README.md

@@ -0,0 +1,105 @@
+# AutoCoder RAG SDK for Node.js
+
+一个便于在Node.js/TypeScript代码中调用`auto-coder.rag run`功能的SDK，用于文档问答和检索增强生成。
+
+## 特性
+
+- 🚀 **易于使用**: 提供简洁直观的API接口
+- 🔄 **异步优先**: 基于Promise和AsyncGenerator的现代异步设计
+- 📡 **流式处理**: 支持实时流式输出答案
+- 🛠 **完整配置**: 支持所有auto-coder.rag run命令行选项
+- 📦 **TypeScript优先**: 完整的类型定义和类型安全
+
+## 安装
+
+```bash
+cd rag-sdks/nodejs
+pnpm install
+pnpm run build
+```
+
+## 快速开始
+
+### 基础用法
+
+```typescript
+import { AutoCoderRAGClient } from 'autocoder-rag-sdk';
+
+const client = new AutoCoderRAGClient('/path/to/docs');
+
+const answer = await client.query('如何使用这个项目?');
+console.log(answer);
+```
+
+### 流式输出
+
+```typescript
+import { AutoCoderRAGClient } from 'autocoder-rag-sdk';
+
+const client = new AutoCoderRAGClient('/path/to/docs');
+
+for await (const chunk of client.queryStream('这个项目的主要功能是什么?')) {
+  process.stdout.write(chunk);
+}
+```
+
+### 获取上下文
+
+```typescript
+import { AutoCoderRAGClient } from 'autocoder-rag-sdk';
+
+const client = new AutoCoderRAGClient('/path/to/docs');
+
+const response = await client.queryWithContexts('如何安装?');
+
+console.log(`答案: ${response.answer}`);
+console.log(`上下文数量: ${response.contexts?.length || 0}`);
+```
+
+### 高级配置
+
+```typescript
+import { AutoCoderRAGClient, RAGConfig } from 'autocoder-rag-sdk';
+
+const config: RAGConfig = {
+  docDir: '/path/to/docs',
+  model: 'v3_chat',
+  agentic: true,  // 使用 AgenticRAG
+  productMode: 'pro',
+  enableHybridIndex: true,
+};
+
+const client = new AutoCoderRAGClient(config);
+
+const answer = await client.query('项目架构是什么?');
+console.log(answer);
+```
+
+## API 文档
+
+### AutoCoderRAGClient
+
+```typescript
+class AutoCoderRAGClient {
+  constructor(config: RAGConfig | string);
+  query(question: string, options?: RAGQueryOptions): Promise<string>;
+  queryStream(question: string, options?: RAGQueryOptions): AsyncGenerator<string>;
+  queryWithContexts(question: string, options?: RAGQueryOptions): Promise<RAGResponse>;
+  getVersion(): string;
+  checkAvailability(): boolean;
+}
+```
+
+## 示例
+
+运行示例：
+
+```bash
+pnpm run example:basic
+pnpm run example:stream
+```
+
+## 许可证
+
+MIT License
+

package/dist/client.d.ts

@@ -0,0 +1,17 @@
+/**
+ * AutoCoder RAG SDK 客户端
+ */
+import { RAGConfig, RAGQueryOptions, RAGResponse } from './types';
+export declare class AutoCoderRAGClient {
+    private config;
+    constructor(configOrDocDir: RAGConfig | string);
+    private getDefaultConfig;
+    private validateConfig;
+    query(question: string, options?: RAGQueryOptions): Promise<string>;
+    queryStream(question: string, options?: RAGQueryOptions): AsyncGenerator<string, void, unknown>;
+    queryWithContexts(question: string, options?: RAGQueryOptions): Promise<RAGResponse>;
+    private buildCommand;
+    getVersion(): string;
+    checkAvailability(): boolean;
+}
+//# sourceMappingURL=client.d.ts.map

package/dist/client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AAAA;;GAEG;AAIH,OAAO,EACL,SAAS,EACT,eAAe,EACf,WAAW,EAIZ,MAAM,SAAS,CAAC;AAOjB,qBAAa,kBAAkB;IAC7B,OAAO,CAAC,MAAM,CAAoB;gBAEtB,cAAc,EAAE,SAAS,GAAG,MAAM;IAa9C,OAAO,CAAC,gBAAgB;IA0BxB,OAAO,CAAC,cAAc;IAYhB,KAAK,CACT,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE,eAAoB,GAC5B,OAAO,CAAC,MAAM,CAAC;IA2DX,WAAW,CAChB,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE,eAAoB,GAC5B,cAAc,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC;IAiDlC,iBAAiB,CACrB,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE,eAAoB,GAC5B,OAAO,CAAC,WAAW,CAAC;IA6BvB,OAAO,CAAC,YAAY;IAiFpB,UAAU,IAAI,MAAM;IAYpB,iBAAiB,IAAI,OAAO;CAS7B"}

package/dist/client.js

@@ -0,0 +1,247 @@
+"use strict";
+/**
+ * AutoCoder RAG SDK 客户端
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AutoCoderRAGClient = void 0;
+const child_process_1 = require("child_process");
+const fs_1 = require("fs");
+const types_1 = require("./types");
+class AutoCoderRAGClient {
+    constructor(configOrDocDir) {
+        if (typeof configOrDocDir === 'string') {
+            this.config = this.getDefaultConfig(configOrDocDir);
+        }
+        else {
+            this.config = {
+                ...this.getDefaultConfig(configOrDocDir.docDir),
+                ...configOrDocDir,
+            };
+        }
+        this.validateConfig();
+    }
+    getDefaultConfig(docDir) {
+        return {
+            docDir,
+            model: 'v3_chat',
+            timeout: 300000, // 默认5分钟（毫秒）
+            ragContextWindowLimit: 56000,
+            fullTextRatio: 0.7,
+            segmentRatio: 0.2,
+            ragDocFilterRelevance: 5,
+            agentic: false,
+            productMode: 'lite',
+            enableHybridIndex: false,
+            disableAutoWindow: false,
+            disableSegmentReorder: false,
+            recallModel: '',
+            chunkModel: '',
+            qaModel: '',
+            embModel: '',
+            agenticModel: '',
+            contextPruneModel: '',
+            tokenizerPath: undefined,
+            requiredExts: '',
+            rayAddress: 'auto',
+        };
+    }
+    validateConfig() {
+        if (!(0, fs_1.existsSync)(this.config.docDir)) {
+            throw new types_1.ValidationError(`文档目录不存在: ${this.config.docDir}`);
+        }
+        if (!['lite', 'pro'].includes(this.config.productMode)) {
+            throw new types_1.ValidationError(`不支持的产品模式: ${this.config.productMode}`);
+        }
+    }
+    async query(question, options = {}) {
+        const opts = { outputFormat: 'text', ...options };
+        if (!['text', 'json', 'stream-json'].includes(opts.outputFormat)) {
+            throw new types_1.ValidationError(`不支持的输出格式: ${opts.outputFormat}`);
+        }
+        // 获取超时时间
+        const timeout = opts.timeout !== undefined ? opts.timeout : this.config.timeout;
+        const cmd = this.buildCommand(opts);
+        return new Promise((resolve, reject) => {
+            const process = (0, child_process_1.spawn)(cmd[0], cmd.slice(1));
+            // 设置超时
+            const timeoutId = setTimeout(() => {
+                process.kill();
+                reject(new types_1.ExecutionError('查询执行超时'));
+            }, timeout);
+            let stdout = '';
+            let stderr = '';
+            // 发送问题
+            if (process.stdin) {
+                process.stdin.write(question);
+                process.stdin.end();
+            }
+            // 收集输出
+            process.stdout?.on('data', data => {
+                stdout += data.toString();
+            });
+            process.stderr?.on('data', data => {
+                stderr += data.toString();
+            });
+            process.on('close', code => {
+                clearTimeout(timeoutId); // 清除超时定时器
+                if (code !== 0) {
+                    reject(new types_1.ExecutionError(stderr || `命令执行失败，退出码: ${code}`, code || undefined));
+                }
+                else {
+                    resolve(stdout.trim());
+                }
+            });
+            process.on('error', err => {
+                reject(new types_1.RAGError(`进程错误: ${err.message}`));
+            });
+        });
+    }
+    async *queryStream(question, options = {}) {
+        const opts = { outputFormat: 'text', ...options };
+        const cmd = this.buildCommand(opts);
+        const process = (0, child_process_1.spawn)(cmd[0], cmd.slice(1));
+        // 发送问题
+        if (process.stdin) {
+            process.stdin.write(question);
+            process.stdin.end();
+        }
+        // 流式输出
+        if (!process.stdout) {
+            throw new types_1.RAGError('无法获取进程输出');
+        }
+        let buffer = '';
+        for await (const chunk of process.stdout) {
+            buffer += chunk.toString();
+            const lines = buffer.split('\n');
+            // 保留最后一行（可能不完整）
+            buffer = lines.pop() || '';
+            for (const line of lines) {
+                yield line;
+            }
+        }
+        // 输出最后剩余的内容
+        if (buffer) {
+            yield buffer;
+        }
+        // 检查退出码
+        const exitCode = await new Promise(resolve => {
+            process.on('close', code => resolve(code));
+        });
+        if (exitCode !== 0) {
+            throw new types_1.ExecutionError(`命令执行失败，退出码: ${exitCode}`, exitCode || undefined);
+        }
+    }
+    async queryWithContexts(question, options = {}) {
+        const opts = { ...options, outputFormat: 'json' };
+        try {
+            const result = await this.query(question, opts);
+            const data = JSON.parse(result);
+            return {
+                success: true,
+                answer: data.answer || '',
+                contexts: data.contexts || [],
+                metadata: data.metadata || {},
+            };
+        }
+        catch (error) {
+            if (error instanceof types_1.RAGError) {
+                return {
+                    success: false,
+                    answer: '',
+                    error: error.message,
+                };
+            }
+            return {
+                success: false,
+                answer: '',
+                error: `查询失败: ${error}`,
+            };
+        }
+    }
+    buildCommand(options) {
+        const cmd = ['auto-coder.rag', 'run', '--doc_dir', this.config.docDir];
+        // 模型参数
+        const model = options.model || this.config.model;
+        if (model) {
+            cmd.push('--model', model);
+        }
+        // 输出格式
+        cmd.push('--output_format', options.outputFormat || 'text');
+        // RAG 模式
+        const agentic = options.agentic !== undefined ? options.agentic : this.config.agentic;
+        if (agentic) {
+            cmd.push('--agentic');
+        }
+        // 产品模式
+        const productMode = options.productMode || this.config.productMode;
+        if (productMode === 'pro') {
+            cmd.push('--pro');
+        }
+        else if (productMode === 'lite') {
+            cmd.push('--lite');
+        }
+        // RAG 参数
+        cmd.push('--rag_context_window_limit', String(this.config.ragContextWindowLimit), '--full_text_ratio', String(this.config.fullTextRatio), '--segment_ratio', String(this.config.segmentRatio), '--rag_doc_filter_relevance', String(this.config.ragDocFilterRelevance));
+        // 可选模型
+        if (this.config.recallModel) {
+            cmd.push('--recall_model', this.config.recallModel);
+        }
+        if (this.config.chunkModel) {
+            cmd.push('--chunk_model', this.config.chunkModel);
+        }
+        if (this.config.qaModel) {
+            cmd.push('--qa_model', this.config.qaModel);
+        }
+        if (this.config.embModel) {
+            cmd.push('--emb_model', this.config.embModel);
+        }
+        if (this.config.agenticModel) {
+            cmd.push('--agentic_model', this.config.agenticModel);
+        }
+        if (this.config.contextPruneModel) {
+            cmd.push('--context_prune_model', this.config.contextPruneModel);
+        }
+        // 索引选项
+        if (this.config.enableHybridIndex) {
+            cmd.push('--enable_hybrid_index');
+        }
+        if (this.config.disableAutoWindow) {
+            cmd.push('--disable_auto_window');
+        }
+        if (this.config.disableSegmentReorder) {
+            cmd.push('--disable_segment_reorder');
+        }
+        // 其他参数
+        if (this.config.requiredExts) {
+            cmd.push('--required_exts', this.config.requiredExts);
+        }
+        if (this.config.tokenizerPath) {
+            cmd.push('--tokenizer_path', this.config.tokenizerPath);
+        }
+        return cmd;
+    }
+    getVersion() {
+        try {
+            const result = (0, child_process_1.execSync)('auto-coder.rag --version', {
+                encoding: 'utf-8',
+                timeout: 60000, // 60秒超时
+            });
+            return result.trim();
+        }
+        catch {
+            return 'unknown';
+        }
+    }
+    checkAvailability() {
+        try {
+            (0, child_process_1.execSync)('which auto-coder.rag', { timeout: 60000 }); // 60秒超时
+            (0, child_process_1.execSync)('auto-coder.rag --help', { timeout: 60000 }); // 60秒超时
+            return true;
+        }
+        catch {
+            return false;
+        }
+    }
+}
+exports.AutoCoderRAGClient = AutoCoderRAGClient;
+//# sourceMappingURL=client.js.map

package/dist/client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.js","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAEH,iDAAgD;AAChD,2BAAgC;AAChC,mCAOiB;AAOjB,MAAa,kBAAkB;IAG7B,YAAY,cAAkC;QAC5C,IAAI,OAAO,cAAc,KAAK,QAAQ,EAAE,CAAC;YACvC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,gBAAgB,CAAC,cAAc,CAAC,CAAC;QACtD,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,MAAM,GAAG;gBACZ,GAAG,IAAI,CAAC,gBAAgB,CAAC,cAAc,CAAC,MAAM,CAAC;gBAC/C,GAAG,cAAc;aACG,CAAC;QACzB,CAAC;QAED,IAAI,CAAC,cAAc,EAAE,CAAC;IACxB,CAAC;IAEO,gBAAgB,CAAC,MAAc;QACrC,OAAO;YACL,MAAM;YACN,KAAK,EAAE,SAAS;YAChB,OAAO,EAAE,MAAM,EAAG,YAAY;YAC9B,qBAAqB,EAAE,KAAK;YAC5B,aAAa,EAAE,GAAG;YAClB,YAAY,EAAE,GAAG;YACjB,qBAAqB,EAAE,CAAC;YACxB,OAAO,EAAE,KAAK;YACd,WAAW,EAAE,MAAM;YACnB,iBAAiB,EAAE,KAAK;YACxB,iBAAiB,EAAE,KAAK;YACxB,qBAAqB,EAAE,KAAK;YAC5B,WAAW,EAAE,EAAE;YACf,UAAU,EAAE,EAAE;YACd,OAAO,EAAE,EAAE;YACX,QAAQ,EAAE,EAAE;YACZ,YAAY,EAAE,EAAE;YAChB,iBAAiB,EAAE,EAAE;YACrB,aAAa,EAAE,SAAS;YACxB,YAAY,EAAE,EAAE;YAChB,UAAU,EAAE,MAAM;SACnB,CAAC;IACJ,CAAC;IAEO,cAAc;QACpB,IAAI,CAAC,IAAA,eAAU,EAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YACpC,MAAM,IAAI,uBAAe,CAAC,YAAY,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QAC9D,CAAC;QAED,IAAI,CAAC,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC;YACvD,MAAM,IAAI,uBAAe,CACvB,aAAa,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CACvC,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK,CACT,QAAgB,EAChB,UAA2B,EAAE;QAE7B,MAAM,IAAI,GAAG,EAAE,YAAY,EAAE,MAAe,EAAE,GAAG,OAAO,EAAE,CAAC;QAE3D,IAAI,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;YACjE,MAAM,IAAI,uBAAe,CAAC,aAAa,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;QAC9D,CAAC;QAED,SAAS;QACT,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;QAEhF,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QAEpC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,OAAO,GAAG,IAAA,qBAAK,EAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YAE5C,OAAO;YACP,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;gBAChC,OAAO,CAAC,IAAI,EAAE,CAAC;gBACf,MAAM,CAAC,IAAI,sBAAc,CAAC,QAAQ,CAAC,CAAC,CAAC;YACvC,CAAC,EAAE,OAAO,CAAC,CAAC;YACZ,IAAI,MAAM,GAAG,EAAE,CAAC;YAChB,IAAI,MAAM,GAAG,EAAE,CAAC;YAEhB,OAAO;YACP,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;gBAClB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;gBAC9B,OAAO,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;YACtB,CAAC;YAED,OAAO;YACP,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,IAAI,CAAC,EAAE;gBAChC,MAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC5B,CAAC,CAAC,CAAC;YAEH,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,IAAI,CAAC,EAAE;gBAChC,MAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC5B,CAAC,CAAC,CAAC;YAEH,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,EAAE;gBACzB,YAAY,CAAC,SAAS,CAAC,CAAC,CAAE,UAAU;gBAEpC,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;oBACf,MAAM,CACJ,IAAI,sBAAc,CAChB,MAAM,IAAI,eAAe,IAAI,EAAE,EAC/B,IAAI,IAAI,SAAS,CAClB,CACF,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;gBACzB,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,CAAC,EAAE;gBACxB,MAAM,CAAC,IAAI,gBAAQ,CAAC,SAAS,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YAC/C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,CAAC,WAAW,CAChB,QAAgB,EAChB,UAA2B,EAAE;QAE7B,MAAM,IAAI,GAAG,EAAE,YAAY,EAAE,MAAe,EAAE,GAAG,OAAO,EAAE,CAAC;QAC3D,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QAEpC,MAAM,OAAO,GAAG,IAAA,qBAAK,EAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QAE5C,OAAO;QACP,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAC9B,OAAO,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;QACtB,CAAC;QAED,OAAO;QACP,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YACpB,MAAM,IAAI,gBAAQ,CAAC,UAAU,CAAC,CAAC;QACjC,CAAC;QAED,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACzC,MAAM,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;YAC3B,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAEjC,gBAAgB;YAChB,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;YAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,IAAI,CAAC;YACb,CAAC;QACH,CAAC;QAED,YAAY;QACZ,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,MAAM,CAAC;QACf,CAAC;QAED,QAAQ;QACR,MAAM,QAAQ,GAAG,MAAM,IAAI,OAAO,CAAgB,OAAO,CAAC,EAAE;YAC1D,OAAO,CAAC,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,IAAI,QAAQ,KAAK,CAAC,EAAE,CAAC;YACnB,MAAM,IAAI,sBAAc,CACtB,eAAe,QAAQ,EAAE,EACzB,QAAQ,IAAI,SAAS,CACtB,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,iBAAiB,CACrB,QAAgB,EAChB,UAA2B,EAAE;QAE7B,MAAM,IAAI,GAAG,EAAE,GAAG,OAAO,EAAE,YAAY,EAAE,MAAe,EAAE,CAAC;QAE3D,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;YAChD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAEhC,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,EAAE;gBACzB,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,EAAE;gBAC7B,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,EAAE;aAC9B,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,gBAAQ,EAAE,CAAC;gBAC9B,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,MAAM,EAAE,EAAE;oBACV,KAAK,EAAE,KAAK,CAAC,OAAO;iBACrB,CAAC;YACJ,CAAC;YACD,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,EAAE;gBACV,KAAK,EAAE,SAAS,KAAK,EAAE;aACxB,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,YAAY,CAAC,OAAwB;QAC3C,MAAM,GAAG,GAAG,CAAC,gBAAgB,EAAE,KAAK,EAAE,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAEvE,OAAO;QACP,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;QACjD,IAAI,KAAK,EAAE,CAAC;YACV,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAC7B,CAAC;QAED,OAAO;QACP,GAAG,CAAC,IAAI,CAAC,iBAAiB,EAAE,OAAO,CAAC,YAAY,IAAI,MAAM,CAAC,CAAC;QAE5D,SAAS;QACT,MAAM,OAAO,GACX,OAAO,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;QACxE,IAAI,OAAO,EAAE,CAAC;YACZ,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACxB,CAAC;QAED,OAAO;QACP,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC;QACnE,IAAI,WAAW,KAAK,KAAK,EAAE,CAAC;YAC1B,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACpB,CAAC;aAAM,IAAI,WAAW,KAAK,MAAM,EAAE,CAAC;YAClC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrB,CAAC;QAED,SAAS;QACT,GAAG,CAAC,IAAI,CACN,4BAA4B,EAC5B,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,qBAAqB,CAAC,EACzC,mBAAmB,EACnB,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,EACjC,iBAAiB,EACjB,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,EAChC,4BAA4B,EAC5B,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,qBAAqB,CAAC,CAC1C,CAAC;QAEF,OAAO;QACP,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;YAC5B,GAAG,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QACtD,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;YAC3B,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QACpD,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACxB,GAAG,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;YACzB,GAAG,CAAC,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAChD,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;YAC7B,GAAG,CAAC,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QACxD,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC;YAClC,GAAG,CAAC,IAAI,CAAC,uBAAuB,EAAE,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC;QACnE,CAAC;QAED,OAAO;QACP,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC;YAClC,GAAG,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QACpC,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC;YAClC,GAAG,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QACpC,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,CAAC,qBAAqB,EAAE,CAAC;YACtC,GAAG,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;QACxC,CAAC;QAED,OAAO;QACP,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;YAC7B,GAAG,CAAC,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QACxD,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;YAC9B,GAAG,CAAC,IAAI,CAAC,kBAAkB,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QAC1D,CAAC;QAED,OAAO,GAAG,CAAC;IACb,CAAC;IAED,UAAU;QACR,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAA,wBAAQ,EAAC,0BAA0B,EAAE;gBAClD,QAAQ,EAAE,OAAO;gBACjB,OAAO,EAAE,KAAK,EAAG,QAAQ;aAC1B,CAAC,CAAC;YACH,OAAO,MAAM,CAAC,IAAI,EAAE,CAAC;QACvB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,SAAS,CAAC;QACnB,CAAC;IACH,CAAC;IAED,iBAAiB;QACf,IAAI,CAAC;YACH,IAAA,wBAAQ,EAAC,sBAAsB,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAE,QAAQ;YAC/D,IAAA,wBAAQ,EAAC,uBAAuB,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAE,QAAQ;YAChE,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;CACF;AA9SD,gDA8SC"}

package/dist/index.d.ts

@@ -0,0 +1,10 @@
+/**
+ * AutoCoder RAG SDK for Node.js
+ *
+ * 一个便于在Node.js代码中调用auto-coder.rag run功能的SDK。
+ */
+export { AutoCoderRAGClient } from './client';
+export type { RAGConfig, RAGQueryOptions, RAGResponse, } from './types';
+export { RAGError, ValidationError, ExecutionError } from './types';
+export declare const VERSION = "1.0.0";
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,kBAAkB,EAAE,MAAM,UAAU,CAAC;AAC9C,YAAY,EACV,SAAS,EACT,eAAe,EACf,WAAW,GACZ,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,QAAQ,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,SAAS,CAAC;AAEpE,eAAO,MAAM,OAAO,UAAU,CAAC"}

package/dist/index.js

@@ -0,0 +1,16 @@
+"use strict";
+/**
+ * AutoCoder RAG SDK for Node.js
+ *
+ * 一个便于在Node.js代码中调用auto-coder.rag run功能的SDK。
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.VERSION = exports.ExecutionError = exports.ValidationError = exports.RAGError = exports.AutoCoderRAGClient = void 0;
+var client_1 = require("./client");
+Object.defineProperty(exports, "AutoCoderRAGClient", { enumerable: true, get: function () { return client_1.AutoCoderRAGClient; } });
+var types_1 = require("./types");
+Object.defineProperty(exports, "RAGError", { enumerable: true, get: function () { return types_1.RAGError; } });
+Object.defineProperty(exports, "ValidationError", { enumerable: true, get: function () { return types_1.ValidationError; } });
+Object.defineProperty(exports, "ExecutionError", { enumerable: true, get: function () { return types_1.ExecutionError; } });
+exports.VERSION = '1.0.0';
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAAA;;;;GAIG;;;AAEH,mCAA8C;AAArC,4GAAA,kBAAkB,OAAA;AAM3B,iCAAoE;AAA3D,iGAAA,QAAQ,OAAA;AAAE,wGAAA,eAAe,OAAA;AAAE,uGAAA,cAAc,OAAA;AAErC,QAAA,OAAO,GAAG,OAAO,CAAC"}

package/dist/types.d.ts

@@ -0,0 +1,51 @@
+/**
+ * AutoCoder RAG SDK 类型定义
+ */
+export declare class RAGError extends Error {
+    constructor(message: string);
+}
+export declare class ValidationError extends RAGError {
+    constructor(message: string);
+}
+export declare class ExecutionError extends RAGError {
+    readonly exitCode?: number;
+    constructor(message: string, exitCode?: number);
+}
+export interface RAGConfig {
+    docDir: string;
+    model?: string;
+    timeout?: number;
+    ragContextWindowLimit?: number;
+    fullTextRatio?: number;
+    segmentRatio?: number;
+    ragDocFilterRelevance?: number;
+    agentic?: boolean;
+    productMode?: 'lite' | 'pro';
+    enableHybridIndex?: boolean;
+    disableAutoWindow?: boolean;
+    disableSegmentReorder?: boolean;
+    recallModel?: string;
+    chunkModel?: string;
+    qaModel?: string;
+    embModel?: string;
+    agenticModel?: string;
+    contextPruneModel?: string;
+    tokenizerPath?: string;
+    requiredExts?: string;
+    rayAddress?: string;
+}
+export interface RAGQueryOptions {
+    outputFormat?: 'text' | 'json' | 'stream-json';
+    agentic?: boolean;
+    productMode?: 'lite' | 'pro';
+    model?: string;
+    timeout?: number;
+}
+export interface RAGResponse {
+    success: boolean;
+    answer: string;
+    contexts?: string[];
+    error?: string;
+    metadata?: Record<string, any>;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,qBAAa,QAAS,SAAQ,KAAK;gBACrB,OAAO,EAAE,MAAM;CAI5B;AAED,qBAAa,eAAgB,SAAQ,QAAQ;gBAC/B,OAAO,EAAE,MAAM;CAI5B;AAED,qBAAa,cAAe,SAAQ,QAAQ;IAC1C,SAAgB,QAAQ,CAAC,EAAE,MAAM,CAAC;gBAEtB,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM;CAK/C;AAED,MAAM,WAAW,SAAS;IAExB,MAAM,EAAE,MAAM,CAAC;IAGf,KAAK,CAAC,EAAE,MAAM,CAAC;IAGf,OAAO,CAAC,EAAE,MAAM,CAAC;IAGjB,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAG/B,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,WAAW,CAAC,EAAE,MAAM,GAAG,KAAK,CAAC;IAG7B,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAGhC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAG3B,aAAa,CAAC,EAAE,MAAM,CAAC;IAGvB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,eAAe;IAE9B,YAAY,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,aAAa,CAAC;IAG/C,OAAO,CAAC,EAAE,OAAO,CAAC;IAGlB,WAAW,CAAC,EAAE,MAAM,GAAG,KAAK,CAAC;IAG7B,KAAK,CAAC,EAAE,MAAM,CAAC;IAGf,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;IACpB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAChC"}

package/dist/types.js

@@ -0,0 +1,29 @@
+"use strict";
+/**
+ * AutoCoder RAG SDK 类型定义
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ExecutionError = exports.ValidationError = exports.RAGError = void 0;
+class RAGError extends Error {
+    constructor(message) {
+        super(message);
+        this.name = 'RAGError';
+    }
+}
+exports.RAGError = RAGError;
+class ValidationError extends RAGError {
+    constructor(message) {
+        super(message);
+        this.name = 'ValidationError';
+    }
+}
+exports.ValidationError = ValidationError;
+class ExecutionError extends RAGError {
+    constructor(message, exitCode) {
+        super(message);
+        this.name = 'ExecutionError';
+        this.exitCode = exitCode;
+    }
+}
+exports.ExecutionError = ExecutionError;
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAEH,MAAa,QAAS,SAAQ,KAAK;IACjC,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,UAAU,CAAC;IACzB,CAAC;CACF;AALD,4BAKC;AAED,MAAa,eAAgB,SAAQ,QAAQ;IAC3C,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,iBAAiB,CAAC;IAChC,CAAC;CACF;AALD,0CAKC;AAED,MAAa,cAAe,SAAQ,QAAQ;IAG1C,YAAY,OAAe,EAAE,QAAiB;QAC5C,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,gBAAgB,CAAC;QAC7B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;CACF;AARD,wCAQC"}

package/package.json

@@ -0,0 +1,49 @@
+{
+  "name": "autocoder-rag-sdk",
+  "version": "0.0.1",
+  "description": "Node.js SDK for AutoCoder RAG - 便于在Node.js代码中调用auto-coder.rag run功能",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "files": [
+    "dist/**/*",
+    "README.md",
+    "LICENSE"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsx --watch examples/basic-usage.ts",
+    "example:basic": "tsx examples/basic-usage.ts",
+    "example:stream": "tsx examples/stream-usage.ts",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": [
+    "autocoder",
+    "rag",
+    "sdk",
+    "document-qa",
+    "ai",
+    "nodejs",
+    "typescript"
+  ],
+  "author": "AutoCoder Team <support@autocoder.com>",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/allwefantasy/auto-coder.git",
+    "directory": "rag-sdks/nodejs"
+  },
+  "bugs": {
+    "url": "https://github.com/allwefantasy/auto-coder/issues"
+  },
+  "homepage": "https://github.com/allwefantasy/auto-coder/tree/master/rag-sdks/nodejs",
+  "engines": {
+    "node": ">=14.0.0"
+  },
+  "devDependencies": {
+    "@types/node": "^20.0.0",
+    "tsx": "^4.0.0",
+    "typescript": "^5.0.0"
+  },
+  "packageManager": "pnpm@10.12.1"
+}
+