xx-ai-cli 0.1.0

package/dist/core/stream.js

@@ -0,0 +1,102 @@
+import { createParser } from 'eventsource-parser';
+export async function chat(modelConfig, messages, options = {}) {
+    const { stream = true, onChunk, tools } = options;
+    const { baseURL, apiKey, model, maxTokens, temperature } = modelConfig;
+    const body = {
+        model,
+        messages,
+        max_tokens: maxTokens,
+        temperature,
+        stream,
+    };
+    // 注入工具定义
+    if (tools && tools.length > 0) {
+        body.tools = tools;
+        body.tool_choice = 'auto';
+        // 工具调用时不能流式（部分模型限制），但我们仍尝试
+    }
+    const response = await fetch(`${baseURL}/chat/completions`, {
+        method: 'POST',
+        headers: {
+            'Content-Type': 'application/json',
+            Authorization: `Bearer ${apiKey}`,
+        },
+        body: JSON.stringify(body),
+    });
+    if (!response.ok) {
+        const err = await response.text();
+        throw new Error(`API 请求失败 [${response.status}]: ${err}`);
+    }
+    // ─── 非流式 ────────────────────────────────────────────────
+    if (!stream) {
+        const data = (await response.json());
+        const choice = data.choices[0];
+        const toolCalls = choice?.message?.tool_calls;
+        const content = choice?.message?.content ?? '';
+        return {
+            content,
+            toolCalls: toolCalls && toolCalls.length > 0 ? toolCalls : undefined,
+            finishReason: choice?.finish_reason,
+        };
+    }
+    // ─── 流式处理 ─────────────────────────────────────────────
+    let fullContent = '';
+    const partialToolCalls = {};
+    const parser = createParser({
+        onEvent(event) {
+            if (event.data === '[DONE]')
+                return;
+            try {
+                const parsed = JSON.parse(event.data);
+                const delta = parsed.choices[0]?.delta;
+                if (!delta)
+                    return;
+                // 普通文本
+                if (delta.content) {
+                    fullContent += delta.content;
+                    onChunk?.(delta.content);
+                }
+                // 工具调用 delta 拼接
+                if (delta.tool_calls) {
+                    for (const tc of delta.tool_calls) {
+                        if (!partialToolCalls[tc.index]) {
+                            partialToolCalls[tc.index] = { id: tc.id ?? '', name: '', arguments: '' };
+                        }
+                        if (tc.function?.name)
+                            partialToolCalls[tc.index].name += tc.function.name;
+                        if (tc.function?.arguments)
+                            partialToolCalls[tc.index].arguments += tc.function.arguments;
+                        if (tc.id)
+                            partialToolCalls[tc.index].id = tc.id;
+                    }
+                }
+            }
+            catch {
+                // 忽略解析错误
+            }
+        },
+    });
+    const reader = response.body?.getReader();
+    if (!reader)
+        throw new Error('无法读取响应流');
+    const decoder = new TextDecoder();
+    while (true) {
+        const { done, value } = await reader.read();
+        if (done)
+            break;
+        parser.feed(decoder.decode(value, { stream: true }));
+    }
+    // 组装 toolCalls
+    const toolCallsResult = Object.values(partialToolCalls).length > 0
+        ? Object.values(partialToolCalls).map((tc, i) => ({
+            id: tc.id || `call_${i}`,
+            type: 'function',
+            function: { name: tc.name, arguments: tc.arguments },
+        }))
+        : undefined;
+    return {
+        content: fullContent,
+        toolCalls: toolCallsResult,
+    };
+}
+//# sourceMappingURL=stream.js.map

package/dist/core/stream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream.js","sourceRoot":"","sources":["../../src/core/stream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAiBlD,MAAM,CAAC,KAAK,UAAU,IAAI,CACxB,WAAwB,EACxB,QAAmB,EACnB,UAAuB,EAAE;IAEzB,MAAM,EAAE,MAAM,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC;IAClD,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,WAAW,EAAE,GAAG,WAAW,CAAC;IAEvE,MAAM,IAAI,GAA4B;QACpC,KAAK;QACL,QAAQ;QACR,UAAU,EAAE,SAAS;QACrB,WAAW;QACX,MAAM;KACP,CAAC;IAEF,SAAS;IACT,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC9B,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC;QAC1B,2BAA2B;IAC7B,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,mBAAmB,EAAE;QAC1D,MAAM,EAAE,MAAM;QACd,OAAO,EAAE;YACP,cAAc,EAAE,kBAAkB;YAClC,aAAa,EAAE,UAAU,MAAM,EAAE;SAClC;QACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;KAC3B,CAAC,CAAC;IAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CAAC,aAAa,QAAQ,CAAC,MAAM,MAAM,GAAG,EAAE,CAAC,CAAC;IAC3D,CAAC;IAED,2DAA2D;IAC3D,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAQlC,CAAC;QAEF,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,SAAS,GAAG,MAAM,EAAE,OAAO,EAAE,UAAU,CAAC;QAC9C,MAAM,OAAO,GAAG,MAAM,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;QAE/C,OAAO;YACL,OAAO;YACP,SAAS,EAAE,SAAS,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;YACpE,YAAY,EAAE,MAAM,EAAE,aAAa;SACpC,CAAC;IACJ,CAAC;IAED,yDAAyD;IACzD,IAAI,WAAW,GAAG,EAAE,CAAC;IACrB,MAAM,gBAAgB,GAAoE,EAAE,CAAC;IAE7F,MAAM,MAAM,GAAG,YAAY,CAAC;QAC1B,OAAO,CAAC,KAAK;YACX,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ;gBAAE,OAAO;YACpC,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAYnC,CAAC;gBAEF,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC;gBACvC,IAAI,CAAC,KAAK;oBAAE,OAAO;gBAEnB,OAAO;gBACP,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;oBAClB,WAAW,IAAI,KAAK,CAAC,OAAO,CAAC;oBAC7B,OAAO,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBAC3B,CAAC;gBAED,gBAAgB;gBAChB,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;oBACrB,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;wBAClC,IAAI,CAAC,gBAAgB,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;4BAChC,gBAAgB,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;wBAC5E,CAAC;wBACD,IAAI,EAAE,CAAC,QAAQ,EAAE,IAAI;4BAAE,gBAAgB,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC;wBAC3E,IAAI,EAAE,CAAC,QAAQ,EAAE,SAAS;4BAAE,gBAAgB,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,SAAS,IAAI,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;wBAC1F,IAAI,EAAE,CAAC,EAAE;4BAAE,gBAAgB,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC;oBACnD,CAAC;gBACH,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,SAAS;YACX,CAAC;QACH,CAAC;KACF,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;IAC1C,IAAI,CAAC,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC;IAExC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;IAClC,OAAO,IAAI,EAAE,CAAC;QACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QAC5C,IAAI,IAAI;YAAE,MAAM;QAChB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IACvD,CAAC;IAED,eAAe;IACf,MAAM,eAAe,GAAG,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,MAAM,GAAG,CAAC;QAChE,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YAChD,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,QAAQ,CAAC,EAAE;YACxB,IAAI,EAAE,UAAmB;YACzB,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC,SAAS,EAAE;SACrD,CAAC,CAAC;QACH,CAAC,CAAC,SAAS,CAAC;IAEd,OAAO;QACL,OAAO,EAAE,WAAW;QACpB,SAAS,EAAE,eAAe;KAC3B,CAAC;AACJ,CAAC"}

package/dist/core/tools.d.ts

@@ -0,0 +1,343 @@
+export declare const TOOL_DEFINITIONS: ({
+    type: string;
+    function: {
+        name: string;
+        description: string;
+        parameters: {
+            type: string;
+            properties: {
+                query: {
+                    type: string;
+                    description: string;
+                };
+                max_results: {
+                    type: string;
+                    description: string;
+                    default: number;
+                };
+                path?: undefined;
+                max_lines?: undefined;
+                content?: undefined;
+                search?: undefined;
+                replace?: undefined;
+                show_hidden?: undefined;
+                depth?: undefined;
+                pattern?: undefined;
+                file_glob?: undefined;
+                command?: undefined;
+                description?: undefined;
+                package?: undefined;
+                global?: undefined;
+                dev?: undefined;
+            };
+            required: string[];
+        };
+    };
+} | {
+    type: string;
+    function: {
+        name: string;
+        description: string;
+        parameters: {
+            type: string;
+            properties: {
+                path: {
+                    type: string;
+                    description: string;
+                    default?: undefined;
+                };
+                max_lines: {
+                    type: string;
+                    description: string;
+                };
+                query?: undefined;
+                max_results?: undefined;
+                content?: undefined;
+                search?: undefined;
+                replace?: undefined;
+                show_hidden?: undefined;
+                depth?: undefined;
+                pattern?: undefined;
+                file_glob?: undefined;
+                command?: undefined;
+                description?: undefined;
+                package?: undefined;
+                global?: undefined;
+                dev?: undefined;
+            };
+            required: string[];
+        };
+    };
+} | {
+    type: string;
+    function: {
+        name: string;
+        description: string;
+        parameters: {
+            type: string;
+            properties: {
+                path: {
+                    type: string;
+                    description: string;
+                    default?: undefined;
+                };
+                content: {
+                    type: string;
+                    description: string;
+                };
+                query?: undefined;
+                max_results?: undefined;
+                max_lines?: undefined;
+                search?: undefined;
+                replace?: undefined;
+                show_hidden?: undefined;
+                depth?: undefined;
+                pattern?: undefined;
+                file_glob?: undefined;
+                command?: undefined;
+                description?: undefined;
+                package?: undefined;
+                global?: undefined;
+                dev?: undefined;
+            };
+            required: string[];
+        };
+    };
+} | {
+    type: string;
+    function: {
+        name: string;
+        description: string;
+        parameters: {
+            type: string;
+            properties: {
+                path: {
+                    type: string;
+                    description: string;
+                    default?: undefined;
+                };
+                search: {
+                    type: string;
+                    description: string;
+                };
+                replace: {
+                    type: string;
+                    description: string;
+                };
+                query?: undefined;
+                max_results?: undefined;
+                max_lines?: undefined;
+                content?: undefined;
+                show_hidden?: undefined;
+                depth?: undefined;
+                pattern?: undefined;
+                file_glob?: undefined;
+                command?: undefined;
+                description?: undefined;
+                package?: undefined;
+                global?: undefined;
+                dev?: undefined;
+            };
+            required: string[];
+        };
+    };
+} | {
+    type: string;
+    function: {
+        name: string;
+        description: string;
+        parameters: {
+            type: string;
+            properties: {
+                path: {
+                    type: string;
+                    description: string;
+                    default: string;
+                };
+                show_hidden: {
+                    type: string;
+                    description: string;
+                    default: boolean;
+                };
+                query?: undefined;
+                max_results?: undefined;
+                max_lines?: undefined;
+                content?: undefined;
+                search?: undefined;
+                replace?: undefined;
+                depth?: undefined;
+                pattern?: undefined;
+                file_glob?: undefined;
+                command?: undefined;
+                description?: undefined;
+                package?: undefined;
+                global?: undefined;
+                dev?: undefined;
+            };
+            required?: undefined;
+        };
+    };
+} | {
+    type: string;
+    function: {
+        name: string;
+        description: string;
+        parameters: {
+            type: string;
+            properties: {
+                path: {
+                    type: string;
+                    description: string;
+                    default: string;
+                };
+                depth: {
+                    type: string;
+                    description: string;
+                    default: number;
+                };
+                query?: undefined;
+                max_results?: undefined;
+                max_lines?: undefined;
+                content?: undefined;
+                search?: undefined;
+                replace?: undefined;
+                show_hidden?: undefined;
+                pattern?: undefined;
+                file_glob?: undefined;
+                command?: undefined;
+                description?: undefined;
+                package?: undefined;
+                global?: undefined;
+                dev?: undefined;
+            };
+            required?: undefined;
+        };
+    };
+} | {
+    type: string;
+    function: {
+        name: string;
+        description: string;
+        parameters: {
+            type: string;
+            properties: {
+                pattern: {
+                    type: string;
+                    description: string;
+                };
+                path: {
+                    type: string;
+                    description: string;
+                    default: string;
+                };
+                file_glob: {
+                    type: string;
+                    description: string;
+                    default: string;
+                };
+                query?: undefined;
+                max_results?: undefined;
+                max_lines?: undefined;
+                content?: undefined;
+                search?: undefined;
+                replace?: undefined;
+                show_hidden?: undefined;
+                depth?: undefined;
+                command?: undefined;
+                description?: undefined;
+                package?: undefined;
+                global?: undefined;
+                dev?: undefined;
+            };
+            required: string[];
+        };
+    };
+} | {
+    type: string;
+    function: {
+        name: string;
+        description: string;
+        parameters: {
+            type: string;
+            properties: {
+                command: {
+                    type: string;
+                    description: string;
+                };
+                description: {
+                    type: string;
+                    description: string;
+                };
+                query?: undefined;
+                max_results?: undefined;
+                path?: undefined;
+                max_lines?: undefined;
+                content?: undefined;
+                search?: undefined;
+                replace?: undefined;
+                show_hidden?: undefined;
+                depth?: undefined;
+                pattern?: undefined;
+                file_glob?: undefined;
+                package?: undefined;
+                global?: undefined;
+                dev?: undefined;
+            };
+            required: string[];
+        };
+    };
+} | {
+    type: string;
+    function: {
+        name: string;
+        description: string;
+        parameters: {
+            type: string;
+            properties: {
+                package: {
+                    type: string;
+                    description: string;
+                };
+                global: {
+                    type: string;
+                    description: string;
+                    default: boolean;
+                };
+                dev: {
+                    type: string;
+                    description: string;
+                    default: boolean;
+                };
+                query?: undefined;
+                max_results?: undefined;
+                path?: undefined;
+                max_lines?: undefined;
+                content?: undefined;
+                search?: undefined;
+                replace?: undefined;
+                show_hidden?: undefined;
+                depth?: undefined;
+                pattern?: undefined;
+                file_glob?: undefined;
+                command?: undefined;
+                description?: undefined;
+            };
+            required: string[];
+        };
+    };
+})[];
+export interface ToolCall {
+    id: string;
+    type: 'function';
+    function: {
+        name: string;
+        arguments: string;
+    };
+}
+export interface ToolResult {
+    tool_call_id: string;
+    role: 'tool';
+    content: string;
+}
+export declare function executeTool(toolCall: ToolCall): Promise<ToolResult>;
+//# sourceMappingURL=tools.d.ts.map

package/dist/core/tools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.d.ts","sourceRoot":"","sources":["../../src/core/tools.ts"],"names":[],"mappings":"AASA,eAAO,MAAM,gBAAgB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;IAyI5B,CAAC;AAgCF,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,UAAU,CAAC;IACjB,QAAQ,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC;CAC/C;AAED,MAAM,WAAW,UAAU;IACzB,YAAY,EAAE,MAAM,CAAC;IACrB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,wBAAsB,WAAW,CAAC,QAAQ,EAAE,QAAQ,GAAG,OAAO,CAAC,UAAU,CAAC,CA8IzE"}