agent-orcha 0.0.4 → 0.0.7

package/dist/lib/llm/providers/anthropic-chat-model.js

@@ -0,0 +1,262 @@
+import Anthropic from '@anthropic-ai/sdk';
+import { zodToJsonSchema } from 'zod-to-json-schema';
+import { contentToText, } from "../../types/llm-types.js";
+import { logger } from "../../logger.js";
+export class AnthropicChatModel {
+    client;
+    modelName;
+    temperature;
+    maxTokens;
+    thinkingBudget;
+    boundTools;
+    cachedProviderTools;
+    structuredSchema;
+    constructor(options) {
+        this.client = new Anthropic({
+            apiKey: options.apiKey ?? process.env.ANTHROPIC_API_KEY ?? 'not-set',
+            ...(options.baseURL ? { baseURL: options.baseURL } : {}),
+        });
+        this.modelName = options.modelName;
+        this.temperature = options.temperature;
+        this.maxTokens = options.maxTokens ?? 8192;
+        this.thinkingBudget = options.thinkingBudget;
+    }
+    extractSystemAndMessages(messages) {
+        let system;
+        const anthropicMessages = [];
+        for (const msg of messages) {
+            if (msg.role === 'system') {
+                const text = contentToText(msg.content);
+                system = system ? `${system}\n\n${text}` : text;
+                continue;
+            }
+            if (msg.role === 'human') {
+                if (Array.isArray(msg.content)) {
+                    const blocks = msg.content.map(p => {
+                        if (p.type === 'image') {
+                            return {
+                                type: 'image',
+                                source: { type: 'base64', media_type: p.mediaType, data: p.data },
+                            };
+                        }
+                        return { type: 'text', text: p.text };
+                    });
+                    anthropicMessages.push({ role: 'user', content: blocks });
+                }
+                else {
+                    anthropicMessages.push({ role: 'user', content: msg.content });
+                }
+            }
+            else if (msg.role === 'ai') {
+                const content = [];
+                const text = contentToText(msg.content);
+                if (text) {
+                    content.push({ type: 'text', text });
+                }
+                if (msg.tool_calls?.length) {
+                    for (const tc of msg.tool_calls) {
+                        content.push({
+                            type: 'tool_use',
+                            id: tc.id,
+                            name: tc.name,
+                            input: tc.args,
+                        });
+                    }
+                }
+                anthropicMessages.push({ role: 'assistant', content });
+            }
+            else if (msg.role === 'tool') {
+                if (Array.isArray(msg.content)) {
+                    const blocks = msg.content.map((p) => {
+                        if (p.type === 'image') {
+                            return {
+                                type: 'image',
+                                source: { type: 'base64', media_type: p.mediaType, data: p.data },
+                            };
+                        }
+                        return { type: 'text', text: p.text };
+                    });
+                    anthropicMessages.push({
+                        role: 'user',
+                        content: [
+                            {
+                                type: 'tool_result',
+                                tool_use_id: msg.tool_call_id,
+                                content: blocks,
+                            },
+                        ],
+                    });
+                }
+                else {
+                    anthropicMessages.push({
+                        role: 'user',
+                        content: [
+                            {
+                                type: 'tool_result',
+                                tool_use_id: msg.tool_call_id,
+                                content: msg.content,
+                            },
+                        ],
+                    });
+                }
+            }
+        }
+        return { system, messages: anthropicMessages };
+    }
+    toAnthropicTools() {
+        if (this.cachedProviderTools)
+            return this.cachedProviderTools;
+        if (!this.boundTools?.length)
+            return undefined;
+        this.cachedProviderTools = this.boundTools.map((t) => {
+            const { $schema, ...input_schema } = zodToJsonSchema(t.schema);
+            return {
+                name: t.name,
+                description: t.description,
+                input_schema: input_schema,
+            };
+        });
+        return this.cachedProviderTools;
+    }
+    parseResponse(response) {
+        let content = '';
+        let reasoning = '';
+        const toolCalls = [];
+        for (const block of response.content) {
+            if (block.type === 'thinking') {
+                reasoning += block.thinking;
+            }
+            else if (block.type === 'text') {
+                content += block.text;
+            }
+            else if (block.type === 'tool_use') {
+                toolCalls.push({
+                    id: block.id,
+                    name: block.name,
+                    args: block.input,
+                });
+            }
+        }
+        if (reasoning) {
+            logger.debug(`[Anthropic] Thinking content received (${reasoning.length} chars)`);
+        }
+        return {
+            content,
+            ...(reasoning ? { reasoning } : {}),
+            tool_calls: toolCalls.length > 0 ? toolCalls : undefined,
+            usage_metadata: {
+                input_tokens: response.usage.input_tokens,
+                output_tokens: response.usage.output_tokens,
+                total_tokens: response.usage.input_tokens + response.usage.output_tokens,
+            },
+        };
+    }
+    async invoke(messages) {
+        const { system, messages: anthropicMessages } = this.extractSystemAndMessages(messages);
+        const tools = this.toAnthropicTools();
+        const params = {
+            model: this.modelName,
+            max_tokens: this.maxTokens,
+            messages: anthropicMessages,
+            ...(system ? { system } : {}),
+            ...(!this.thinkingBudget && this.temperature !== undefined ? { temperature: this.temperature } : {}),
+            ...(tools ? { tools } : {}),
+            ...(this.thinkingBudget ? { thinking: { type: 'enabled', budget_tokens: this.thinkingBudget } } : {}),
+        };
+        const response = await this.client.messages.create(params);
+        return this.parseResponse(response);
+    }
+    async *stream(messages, options) {
+        const { system, messages: anthropicMessages } = this.extractSystemAndMessages(messages);
+        const tools = this.toAnthropicTools();
+        const stream = this.client.messages.stream({
+            model: this.modelName,
+            max_tokens: this.maxTokens,
+            messages: anthropicMessages,
+            ...(system ? { system } : {}),
+            ...(!this.thinkingBudget && this.temperature !== undefined ? { temperature: this.temperature } : {}),
+            ...(tools ? { tools } : {}),
+            ...(this.thinkingBudget ? { thinking: { type: 'enabled', budget_tokens: this.thinkingBudget } } : {}),
+        }, { signal: options?.signal });
+        const toolCalls = new Map();
+        let inputTokens = 0;
+        for await (const event of stream) {
+            if (event.type === 'message_start') {
+                const msg = event.message;
+                if (msg?.usage?.input_tokens) {
+                    inputTokens = msg.usage.input_tokens;
+                }
+            }
+            else if (event.type === 'content_block_start') {
+                if (event.content_block.type === 'tool_use') {
+                    toolCalls.set(event.index, {
+                        id: event.content_block.id,
+                        name: event.content_block.name,
+                        args: '',
+                    });
+                }
+            }
+            else if (event.type === 'content_block_delta') {
+                if (event.delta.type === 'text_delta') {
+                    yield { content: event.delta.text };
+                }
+                else if (event.delta.type === 'thinking_delta') {
+                    yield { content: '', reasoning: event.delta.thinking };
+                }
+                else if (event.delta.type === 'input_json_delta') {
+                    const existing = toolCalls.get(event.index);
+                    if (existing) {
+                        existing.args += event.delta.partial_json;
+                    }
+                }
+            }
+            else if (event.type === 'message_delta') {
+                // Final message with usage
+                const parsedToolCalls = toolCalls.size > 0
+                    ? Array.from(toolCalls.values()).map((tc) => ({
+                        id: tc.id,
+                        name: tc.name,
+                        args: JSON.parse(tc.args || '{}'),
+                    }))
+                    : undefined;
+                const outputTokens = event.usage?.output_tokens ?? 0;
+                yield {
+                    content: '',
+                    tool_calls: parsedToolCalls,
+                    usage_metadata: {
+                        input_tokens: inputTokens,
+                        output_tokens: outputTokens,
+                        total_tokens: inputTokens + outputTokens,
+                    },
+                };
+            }
+        }
+    }
+    bindTools(tools) {
+        const bound = new AnthropicChatModel({
+            apiKey: this.client.apiKey ?? undefined,
+            modelName: this.modelName,
+            temperature: this.temperature,
+            maxTokens: this.maxTokens,
+            thinkingBudget: this.thinkingBudget,
+        });
+        bound.client = this.client;
+        bound.boundTools = tools;
+        bound.structuredSchema = this.structuredSchema;
+        return bound;
+    }
+    withStructuredOutput(schema) {
+        const wrapped = new AnthropicChatModel({
+            apiKey: this.client.apiKey ?? undefined,
+            modelName: this.modelName,
+            temperature: this.temperature,
+            maxTokens: this.maxTokens,
+            thinkingBudget: this.thinkingBudget,
+        });
+        wrapped.client = this.client;
+        wrapped.structuredSchema = schema;
+        wrapped.boundTools = this.boundTools;
+        return wrapped;
+    }
+}
+//# sourceMappingURL=anthropic-chat-model.js.map

package/dist/lib/llm/providers/anthropic-chat-model.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic-chat-model.js","sourceRoot":"","sources":["../../../../lib/llm/providers/anthropic-chat-model.ts"],"names":[],"mappings":"AAAA,OAAO,SAAS,MAAM,mBAAmB,CAAC;AAC1C,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EACL,aAAa,GAMd,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AAWzC,MAAM,OAAO,kBAAkB;IACrB,MAAM,CAAY;IAClB,SAAS,CAAS;IAClB,WAAW,CAAU;IACrB,SAAS,CAAS;IAClB,cAAc,CAAU;IACxB,UAAU,CAAoB;IAC9B,mBAAmB,CAAoB;IACvC,gBAAgB,CAA2B;IAEnD,YAAY,OAAkC;QAC5C,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC;YAC1B,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,SAAS;YACpE,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACzD,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;QACvC,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,IAAI,CAAC;QAC3C,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC;IAC/C,CAAC;IAEO,wBAAwB,CAC9B,QAAuB;QAEvB,IAAI,MAA0B,CAAC;QAC/B,MAAM,iBAAiB,GAA6B,EAAE,CAAC;QAEvD,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC1B,MAAM,IAAI,GAAG,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACxC,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,OAAO,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;gBAChD,SAAS;YACX,CAAC;YAED,IAAI,GAAG,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBACzB,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC/B,MAAM,MAAM,GAAkC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;wBAChE,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;4BACvB,OAAO;gCACL,IAAI,EAAE,OAAgB;gCACtB,MAAM,EAAE,EAAE,IAAI,EAAE,QAAiB,EAAE,UAAU,EAAE,CAAC,CAAC,SAAyB,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE;6BAC3F,CAAC;wBACJ,CAAC;wBACD,OAAO,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;oBACjD,CAAC,CAAC,CAAC;oBACH,iBAAiB,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;gBAC5D,CAAC;qBAAM,CAAC;oBACN,iBAAiB,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;gBACjE,CAAC;YACH,CAAC;iBAAM,IAAI,GAAG,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;gBAC7B,MAAM,OAAO,GAAkC,EAAE,CAAC;gBAClD,MAAM,IAAI,GAAG,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACxC,IAAI,IAAI,EAAE,CAAC;oBACT,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBACvC,CAAC;gBACD,IAAI,GAAG,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC;oBAC3B,KAAK,MAAM,EAAE,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;wBAChC,OAAO,CAAC,IAAI,CAAC;4BACX,IAAI,EAAE,UAAU;4BAChB,EAAE,EAAE,EAAE,CAAC,EAAE;4BACT,IAAI,EAAE,EAAE,CAAC,IAAI;4BACb,KAAK,EAAE,EAAE,CAAC,IAAI;yBACf,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;gBACD,iBAAiB,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC;YACzD,CAAC;iBAAM,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC/B,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC/B,MAAM,MAAM,GAA8C,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;wBAC9E,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;4BACvB,OAAO;gCACL,IAAI,EAAE,OAAgB;gCACtB,MAAM,EAAE,EAAE,IAAI,EAAE,QAAiB,EAAE,UAAU,EAAE,CAAC,CAAC,SAAyB,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE;6BAC3F,CAAC;wBACJ,CAAC;wBACD,OAAO,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;oBACjD,CAAC,CAAC,CAAC;oBACH,iBAAiB,CAAC,IAAI,CAAC;wBACrB,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE;4BACP;gCACE,IAAI,EAAE,aAAa;gCACnB,WAAW,EAAE,GAAG,CAAC,YAAa;gCAC9B,OAAO,EAAE,MAAM;6BAChB;yBACF;qBACF,CAAC,CAAC;gBACL,CAAC;qBAAM,CAAC;oBACN,iBAAiB,CAAC,IAAI,CAAC;wBACrB,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE;4BACP;gCACE,IAAI,EAAE,aAAa;gCACnB,WAAW,EAAE,GAAG,CAAC,YAAa;gCAC9B,OAAO,EAAE,GAAG,CAAC,OAAO;6BACrB;yBACF;qBACF,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,iBAAiB,EAAE,CAAC;IACjD,CAAC;IAEO,gBAAgB;QACtB,IAAI,IAAI,CAAC,mBAAmB;YAAE,OAAO,IAAI,CAAC,mBAAmB,CAAC;QAC9D,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,MAAM;YAAE,OAAO,SAAS,CAAC;QAC/C,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACnD,MAAM,EAAE,OAAO,EAAE,GAAG,YAAY,EAAE,GAAG,eAAe,CAAC,CAAC,CAAC,MAAM,CAA4B,CAAC;YAC1F,OAAO;gBACL,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;gBAC1B,YAAY,EAAE,YAA0C;aACzD,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,OAAO,IAAI,CAAC,mBAAmB,CAAC;IAClC,CAAC;IAEO,aAAa,CAAC,QAA2B;QAC/C,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,IAAI,SAAS,GAAG,EAAE,CAAC;QACnB,MAAM,SAAS,GAAe,EAAE,CAAC;QAEjC,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACrC,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC9B,SAAS,IAAK,KAAa,CAAC,QAAQ,CAAC;YACvC,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACjC,OAAO,IAAI,KAAK,CAAC,IAAI,CAAC;YACxB,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBACrC,SAAS,CAAC,IAAI,CAAC;oBACb,EAAE,EAAE,KAAK,CAAC,EAAE;oBACZ,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,IAAI,EAAE,KAAK,CAAC,KAAgC;iBAC7C,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,IAAI,SAAS,EAAE,CAAC;YACd,MAAM,CAAC,KAAK,CAAC,0CAA0C,SAAS,CAAC,MAAM,SAAS,CAAC,CAAC;QACpF,CAAC;QAED,OAAO;YACL,OAAO;YACP,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACnC,UAAU,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;YACxD,cAAc,EAAE;gBACd,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;gBACzC,aAAa,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa;gBAC3C,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,aAAa;aACzE;SACF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,QAAuB;QAClC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,iBAAiB,EAAE,GAAG,IAAI,CAAC,wBAAwB,CAAC,QAAQ,CAAC,CAAC;QAExF,MAAM,KAAK,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACtC,MAAM,MAAM,GAA8C;YACxD,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,QAAQ,EAAE,iBAAiB;YAC3B,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7B,GAAG,CAAC,CAAC,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACpG,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3B,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,SAAkB,EAAE,aAAa,EAAE,IAAI,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC/G,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC3D,OAAO,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CACX,QAAuB,EACvB,OAAkC;QAElC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,iBAAiB,EAAE,GAAG,IAAI,CAAC,wBAAwB,CAAC,QAAQ,CAAC,CAAC;QAExF,MAAM,KAAK,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;YACzC,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,QAAQ,EAAE,iBAAiB;YAC3B,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7B,GAAG,CAAC,CAAC,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACpG,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC3B,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,SAAkB,EAAE,aAAa,EAAE,IAAI,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC/G,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAEhC,MAAM,SAAS,GAA4D,IAAI,GAAG,EAAE,CAAC;QACrF,IAAI,WAAW,GAAG,CAAC,CAAC;QAEpB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACjC,IAAI,KAAK,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;gBACnC,MAAM,GAAG,GAAI,KAAa,CAAC,OAAO,CAAC;gBACnC,IAAI,GAAG,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC;oBAC7B,WAAW,GAAG,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC;gBACvC,CAAC;YACH,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,qBAAqB,EAAE,CAAC;gBAChD,IAAI,KAAK,CAAC,aAAa,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBAC5C,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE;wBACzB,EAAE,EAAE,KAAK,CAAC,aAAa,CAAC,EAAE;wBAC1B,IAAI,EAAE,KAAK,CAAC,aAAa,CAAC,IAAI;wBAC9B,IAAI,EAAE,EAAE;qBACT,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,qBAAqB,EAAE,CAAC;gBAChD,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;oBACtC,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;gBACtC,CAAC;qBAAM,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;oBACjD,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,SAAS,EAAG,KAAK,CAAC,KAAa,CAAC,QAAQ,EAAE,CAAC;gBAClE,CAAC;qBAAM,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;oBACnD,MAAM,QAAQ,GAAG,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;oBAC5C,IAAI,QAAQ,EAAE,CAAC;wBACb,QAAQ,CAAC,IAAI,IAAI,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC;oBAC5C,CAAC;gBACH,CAAC;YACH,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;gBAC1C,2BAA2B;gBAC3B,MAAM,eAAe,GAA2B,SAAS,CAAC,IAAI,GAAG,CAAC;oBAChE,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;wBAC1C,EAAE,EAAE,EAAE,CAAC,EAAE;wBACT,IAAI,EAAE,EAAE,CAAC,IAAI;wBACb,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,IAAI,IAAI,CAAC;qBAClC,CAAC,CAAC;oBACL,CAAC,CAAC,SAAS,CAAC;gBAEd,MAAM,YAAY,GAAG,KAAK,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC,CAAC;gBACrD,MAAM;oBACJ,OAAO,EAAE,EAAE;oBACX,UAAU,EAAE,eAAe;oBAC3B,cAAc,EAAE;wBACd,YAAY,EAAE,WAAW;wBACzB,aAAa,EAAE,YAAY;wBAC3B,YAAY,EAAE,WAAW,GAAG,YAAY;qBACzC;iBACF,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,SAAS,CAAC,KAAuB;QAC/B,MAAM,KAAK,GAAG,IAAI,kBAAkB,CAAC;YACnC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,SAAS;YACvC,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,cAAc,EAAE,IAAI,CAAC,cAAc;SACpC,CAAC,CAAC;QACH,KAAK,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC3B,KAAK,CAAC,UAAU,GAAG,KAAK,CAAC;QACzB,KAAK,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC;QAC/C,OAAO,KAAK,CAAC;IACf,CAAC;IAED,oBAAoB,CAAC,MAA+B;QAClD,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC;YACrC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,SAAS;YACvC,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,cAAc,EAAE,IAAI,CAAC,cAAc;SACpC,CAAC,CAAC;QACH,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC7B,OAAO,CAAC,gBAAgB,GAAG,MAAM,CAAC;QAClC,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC;QACrC,OAAO,OAAO,CAAC;IACjB,CAAC;CACF"}

package/dist/lib/llm/providers/gemini-chat-model.d.ts

@@ -0,0 +1,30 @@
+import { type ChatModel, type ChatModelResponse, type BaseMessage, type StructuredTool } from '../../types/llm-types.ts';
+interface GeminiChatModelOptions {
+    apiKey?: string;
+    modelName: string;
+    temperature?: number;
+    maxTokens?: number;
+}
+export declare class GeminiChatModel implements ChatModel {
+    private genAI;
+    private modelName;
+    private temperature?;
+    private maxTokens?;
+    private boundTools?;
+    private cachedProviderTools?;
+    private structuredSchema?;
+    constructor(options: GeminiChatModelOptions);
+    private extractSystemAndContents;
+    private toGeminiTools;
+    private convertToGeminiSchema;
+    private convertPropertyToGemini;
+    private mapSchemaType;
+    invoke(messages: BaseMessage[]): Promise<ChatModelResponse>;
+    stream(messages: BaseMessage[], options?: {
+        signal?: AbortSignal;
+    }): AsyncIterable<ChatModelResponse>;
+    bindTools(tools: StructuredTool[]): ChatModel;
+    withStructuredOutput(schema: Record<string, unknown>): ChatModel;
+}
+export {};
+//# sourceMappingURL=gemini-chat-model.d.ts.map

package/dist/lib/llm/providers/gemini-chat-model.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini-chat-model.d.ts","sourceRoot":"","sources":["../../../../lib/llm/providers/gemini-chat-model.ts"],"names":[],"mappings":"AAEA,OAAO,EAEL,KAAK,SAAS,EACd,KAAK,iBAAiB,EACtB,KAAK,WAAW,EAChB,KAAK,cAAc,EAEpB,MAAM,0BAA0B,CAAC;AAGlC,UAAU,sBAAsB;IAC9B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,qBAAa,eAAgB,YAAW,SAAS;IAC/C,OAAO,CAAC,KAAK,CAAqB;IAClC,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,WAAW,CAAC,CAAS;IAC7B,OAAO,CAAC,SAAS,CAAC,CAAS;IAC3B,OAAO,CAAC,UAAU,CAAC,CAAmB;IACtC,OAAO,CAAC,mBAAmB,CAAC,CAAoD;IAChF,OAAO,CAAC,gBAAgB,CAAC,CAA0B;gBAEvC,OAAO,EAAE,sBAAsB;IAO3C,OAAO,CAAC,wBAAwB;IAgFhC,OAAO,CAAC,aAAa;IAerB,OAAO,CAAC,qBAAqB;IAyB7B,OAAO,CAAC,uBAAuB;IAgC/B,OAAO,CAAC,aAAa;IAYf,MAAM,CAAC,QAAQ,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,iBAAiB,CAAC;IA2D1D,MAAM,CACX,QAAQ,EAAE,WAAW,EAAE,EACvB,OAAO,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,WAAW,CAAA;KAAE,GACjC,aAAa,CAAC,iBAAiB,CAAC;IAyDnC,SAAS,CAAC,KAAK,EAAE,cAAc,EAAE,GAAG,SAAS;IAa7C,oBAAoB,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS;CAYjE"}

package/dist/lib/llm/providers/gemini-chat-model.js

@@ -0,0 +1,300 @@
+import { GoogleGenerativeAI, SchemaType } from '@google/generative-ai';
+import { zodToJsonSchema } from 'zod-to-json-schema';
+import { contentToText, } from "../../types/llm-types.js";
+import { logger } from "../../logger.js";
+export class GeminiChatModel {
+    genAI;
+    modelName;
+    temperature;
+    maxTokens;
+    boundTools;
+    cachedProviderTools;
+    structuredSchema;
+    constructor(options) {
+        this.genAI = new GoogleGenerativeAI(options.apiKey ?? process.env.GOOGLE_API_KEY ?? '');
+        this.modelName = options.modelName;
+        this.temperature = options.temperature;
+        this.maxTokens = options.maxTokens;
+    }
+    extractSystemAndContents(messages) {
+        let systemInstruction;
+        const contents = [];
+        for (const msg of messages) {
+            if (msg.role === 'system') {
+                const text = contentToText(msg.content);
+                systemInstruction = systemInstruction
+                    ? `${systemInstruction}\n\n${text}`
+                    : text;
+                continue;
+            }
+            if (msg.role === 'human') {
+                if (Array.isArray(msg.content)) {
+                    const parts = msg.content.map(p => {
+                        if (p.type === 'image') {
+                            return { inlineData: { mimeType: p.mediaType, data: p.data } };
+                        }
+                        return { text: p.text };
+                    });
+                    contents.push({ role: 'user', parts });
+                }
+                else {
+                    contents.push({ role: 'user', parts: [{ text: msg.content }] });
+                }
+            }
+            else if (msg.role === 'ai') {
+                const parts = [];
+                const text = contentToText(msg.content);
+                if (text) {
+                    parts.push({ text });
+                }
+                if (msg.tool_calls?.length) {
+                    for (const tc of msg.tool_calls) {
+                        parts.push({
+                            functionCall: { name: tc.name, args: tc.args },
+                        });
+                    }
+                }
+                contents.push({ role: 'model', parts });
+            }
+            else if (msg.role === 'tool') {
+                if (Array.isArray(msg.content)) {
+                    const parts = [];
+                    // Add function response with text content
+                    const textContent = contentToText(msg.content);
+                    parts.push({
+                        functionResponse: {
+                            name: msg.name,
+                            response: { result: textContent },
+                        },
+                    });
+                    // Add inline image data for image parts
+                    for (const p of msg.content) {
+                        if (p.type === 'image') {
+                            parts.push({
+                                inlineData: { mimeType: p.mediaType, data: p.data },
+                            });
+                        }
+                    }
+                    contents.push({ role: 'function', parts });
+                }
+                else {
+                    contents.push({
+                        role: 'function',
+                        parts: [
+                            {
+                                functionResponse: {
+                                    name: msg.name,
+                                    response: { result: msg.content },
+                                },
+                            },
+                        ],
+                    });
+                }
+            }
+        }
+        return { systemInstruction, contents };
+    }
+    toGeminiTools() {
+        if (this.cachedProviderTools)
+            return this.cachedProviderTools;
+        if (!this.boundTools?.length)
+            return undefined;
+        const declarations = this.boundTools.map((t) => {
+            const { $schema, ...jsonSchema } = zodToJsonSchema(t.schema);
+            return {
+                name: t.name,
+                description: t.description,
+                parameters: this.convertToGeminiSchema(jsonSchema),
+            };
+        });
+        this.cachedProviderTools = [{ functionDeclarations: declarations }];
+        return this.cachedProviderTools;
+    }
+    convertToGeminiSchema(jsonSchema) {
+        const schemaType = jsonSchema.type;
+        const properties = jsonSchema.properties;
+        const required = jsonSchema.required;
+        const result = {
+            type: this.mapSchemaType(schemaType),
+            properties: {},
+        };
+        if (properties) {
+            const geminiProps = {};
+            for (const [key, prop] of Object.entries(properties)) {
+                geminiProps[key] = this.convertPropertyToGemini(prop);
+            }
+            result.properties = geminiProps;
+        }
+        if (required) {
+            result.required = required;
+        }
+        return result;
+    }
+    convertPropertyToGemini(prop) {
+        const type = prop.type;
+        const result = {
+            type: this.mapSchemaType(type),
+        };
+        if (prop.description) {
+            result.description = prop.description;
+        }
+        if (type === 'array') {
+            const items = prop.items;
+            result.items = items
+                ? this.convertPropertyToGemini(items)
+                : { type: SchemaType.STRING };
+        }
+        if (type === 'object' && prop.properties) {
+            const nested = prop.properties;
+            const geminiProps = {};
+            for (const [key, nestedProp] of Object.entries(nested)) {
+                geminiProps[key] = this.convertPropertyToGemini(nestedProp);
+            }
+            result.properties = geminiProps;
+            if (prop.required) {
+                result.required = prop.required;
+            }
+        }
+        return result;
+    }
+    mapSchemaType(type) {
+        switch (type) {
+            case 'string': return SchemaType.STRING;
+            case 'number': return SchemaType.NUMBER;
+            case 'integer': return SchemaType.INTEGER;
+            case 'boolean': return SchemaType.BOOLEAN;
+            case 'array': return SchemaType.ARRAY;
+            case 'object': return SchemaType.OBJECT;
+            default: return SchemaType.STRING;
+        }
+    }
+    async invoke(messages) {
+        const { systemInstruction, contents } = this.extractSystemAndContents(messages);
+        const tools = this.toGeminiTools();
+        const model = this.genAI.getGenerativeModel({
+            model: this.modelName,
+            ...(systemInstruction ? { systemInstruction } : {}),
+            generationConfig: {
+                ...(this.temperature !== undefined ? { temperature: this.temperature } : {}),
+                ...(this.maxTokens ? { maxOutputTokens: this.maxTokens } : {}),
+            },
+            ...(tools ? { tools } : {}),
+        });
+        const result = await model.generateContent({ contents });
+        const response = result.response;
+        const candidate = response.candidates?.[0];
+        let content = '';
+        let reasoning = '';
+        const toolCalls = [];
+        if (candidate?.content?.parts) {
+            for (const part of candidate.content.parts) {
+                if (part.thought && part.text) {
+                    reasoning += part.text;
+                }
+                else if (part.text) {
+                    content += part.text;
+                }
+                if (part.functionCall) {
+                    toolCalls.push({
+                        id: `call_${crypto.randomUUID()}`,
+                        name: part.functionCall.name,
+                        args: (part.functionCall.args ?? {}),
+                    });
+                }
+            }
+        }
+        if (reasoning) {
+            logger.debug(`[Gemini] Thinking content received (${reasoning.length} chars)`);
+        }
+        const usage = response.usageMetadata;
+        return {
+            content,
+            ...(reasoning ? { reasoning } : {}),
+            tool_calls: toolCalls.length > 0 ? toolCalls : undefined,
+            usage_metadata: usage
+                ? {
+                    input_tokens: usage.promptTokenCount ?? 0,
+                    output_tokens: usage.candidatesTokenCount ?? 0,
+                    total_tokens: usage.totalTokenCount ?? 0,
+                }
+                : undefined,
+        };
+    }
+    async *stream(messages, options) {
+        const { systemInstruction, contents } = this.extractSystemAndContents(messages);
+        const signal = options?.signal;
+        const tools = this.toGeminiTools();
+        const model = this.genAI.getGenerativeModel({
+            model: this.modelName,
+            ...(systemInstruction ? { systemInstruction } : {}),
+            generationConfig: {
+                ...(this.temperature !== undefined ? { temperature: this.temperature } : {}),
+                ...(this.maxTokens ? { maxOutputTokens: this.maxTokens } : {}),
+            },
+            ...(tools ? { tools } : {}),
+        });
+        const result = await model.generateContentStream({ contents });
+        for await (const chunk of result.stream) {
+            if (signal?.aborted)
+                break;
+            const candidate = chunk.candidates?.[0];
+            let content = '';
+            let reasoning = '';
+            const toolCalls = [];
+            if (candidate?.content?.parts) {
+                for (const part of candidate.content.parts) {
+                    if (part.thought && part.text) {
+                        reasoning += part.text;
+                    }
+                    else if (part.text) {
+                        content += part.text;
+                    }
+                    if (part.functionCall) {
+                        toolCalls.push({
+                            id: `call_${crypto.randomUUID()}`,
+                            name: part.functionCall.name,
+                            args: (part.functionCall.args ?? {}),
+                        });
+                    }
+                }
+            }
+            yield {
+                content,
+                ...(reasoning ? { reasoning } : {}),
+                tool_calls: toolCalls.length > 0 ? toolCalls : undefined,
+                usage_metadata: chunk.usageMetadata
+                    ? {
+                        input_tokens: chunk.usageMetadata.promptTokenCount ?? 0,
+                        output_tokens: chunk.usageMetadata.candidatesTokenCount ?? 0,
+                        total_tokens: chunk.usageMetadata.totalTokenCount ?? 0,
+                    }
+                    : undefined,
+            };
+        }
+    }
+    bindTools(tools) {
+        const bound = new GeminiChatModel({
+            apiKey: '',
+            modelName: this.modelName,
+            temperature: this.temperature,
+            maxTokens: this.maxTokens,
+        });
+        bound.genAI = this.genAI;
+        bound.boundTools = tools;
+        bound.structuredSchema = this.structuredSchema;
+        return bound;
+    }
+    withStructuredOutput(schema) {
+        const wrapped = new GeminiChatModel({
+            apiKey: '',
+            modelName: this.modelName,
+            temperature: this.temperature,
+            maxTokens: this.maxTokens,
+        });
+        wrapped.genAI = this.genAI;
+        wrapped.structuredSchema = schema;
+        wrapped.boundTools = this.boundTools;
+        return wrapped;
+    }
+}
+//# sourceMappingURL=gemini-chat-model.js.map

package/dist/lib/llm/providers/gemini-chat-model.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini-chat-model.js","sourceRoot":"","sources":["../../../../lib/llm/providers/gemini-chat-model.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAqF,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAC1J,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EACL,aAAa,GAMd,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AASzC,MAAM,OAAO,eAAe;IAClB,KAAK,CAAqB;IAC1B,SAAS,CAAS;IAClB,WAAW,CAAU;IACrB,SAAS,CAAU;IACnB,UAAU,CAAoB;IAC9B,mBAAmB,CAAqD;IACxE,gBAAgB,CAA2B;IAEnD,YAAY,OAA+B;QACzC,IAAI,CAAC,KAAK,GAAG,IAAI,kBAAkB,CAAC,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC;QACxF,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;QACvC,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;IACrC,CAAC;IAEO,wBAAwB,CAC9B,QAAuB;QAEvB,IAAI,iBAAqC,CAAC;QAC1C,MAAM,QAAQ,GAAc,EAAE,CAAC;QAE/B,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC1B,MAAM,IAAI,GAAG,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACxC,iBAAiB,GAAG,iBAAiB;oBACnC,CAAC,CAAC,GAAG,iBAAiB,OAAO,IAAI,EAAE;oBACnC,CAAC,CAAC,IAAI,CAAC;gBACT,SAAS;YACX,CAAC;YAED,IAAI,GAAG,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBACzB,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC/B,MAAM,KAAK,GAAW,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;wBACxC,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;4BACvB,OAAO,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;wBACjE,CAAC;wBACD,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;oBAC1B,CAAC,CAAC,CAAC;oBACH,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;gBACzC,CAAC;qBAAM,CAAC;oBACN,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;gBAClE,CAAC;YACH,CAAC;iBAAM,IAAI,GAAG,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;gBAC7B,MAAM,KAAK,GAAW,EAAE,CAAC;gBACzB,MAAM,IAAI,GAAG,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACxC,IAAI,IAAI,EAAE,CAAC;oBACT,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;gBACvB,CAAC;gBACD,IAAI,GAAG,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC;oBAC3B,KAAK,MAAM,EAAE,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;wBAChC,KAAK,CAAC,IAAI,CAAC;4BACT,YAAY,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE;yBAC/C,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;gBACD,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;YAC1C,CAAC;iBAAM,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC/B,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC/B,MAAM,KAAK,GAAW,EAAE,CAAC;oBACzB,0CAA0C;oBAC1C,MAAM,WAAW,GAAG,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;oBAC/C,KAAK,CAAC,IAAI,CAAC;wBACT,gBAAgB,EAAE;4BAChB,IAAI,EAAE,GAAG,CAAC,IAAK;4BACf,QAAQ,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE;yBAClC;qBACF,CAAC,CAAC;oBACH,wCAAwC;oBACxC,KAAK,MAAM,CAAC,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;wBAC5B,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;4BACvB,KAAK,CAAC,IAAI,CAAC;gCACT,UAAU,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE;6BACpD,CAAC,CAAC;wBACL,CAAC;oBACH,CAAC;oBACD,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC;gBAC7C,CAAC;qBAAM,CAAC;oBACN,QAAQ,CAAC,IAAI,CAAC;wBACZ,IAAI,EAAE,UAAU;wBAChB,KAAK,EAAE;4BACL;gCACE,gBAAgB,EAAE;oCAChB,IAAI,EAAE,GAAG,CAAC,IAAK;oCACf,QAAQ,EAAE,EAAE,MAAM,EAAE,GAAG,CAAC,OAAO,EAAE;iCAClC;6BACF;yBACF;qBACF,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,EAAE,iBAAiB,EAAE,QAAQ,EAAE,CAAC;IACzC,CAAC;IAEO,aAAa;QACnB,IAAI,IAAI,CAAC,mBAAmB;YAAE,OAAO,IAAI,CAAC,mBAAmB,CAAC;QAC9D,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,MAAM;YAAE,OAAO,SAAS,CAAC;QAC/C,MAAM,YAAY,GAA0B,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACpE,MAAM,EAAE,OAAO,EAAE,GAAG,UAAU,EAAE,GAAG,eAAe,CAAC,CAAC,CAAC,MAAM,CAA4B,CAAC;YACxF,OAAO;gBACL,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;gBAC1B,UAAU,EAAE,IAAI,CAAC,qBAAqB,CAAC,UAAU,CAAC;aACnD,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,mBAAmB,GAAG,CAAC,EAAE,oBAAoB,EAAE,YAAY,EAAE,CAAC,CAAC;QACpE,OAAO,IAAI,CAAC,mBAAmB,CAAC;IAClC,CAAC;IAEO,qBAAqB,CAAC,UAAmC;QAC/D,MAAM,UAAU,GAAG,UAAU,CAAC,IAAc,CAAC;QAC7C,MAAM,UAAU,GAAG,UAAU,CAAC,UAAiE,CAAC;QAChG,MAAM,QAAQ,GAAG,UAAU,CAAC,QAAgC,CAAC;QAE7D,MAAM,MAAM,GAA8B;YACxC,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC;YACpC,UAAU,EAAE,EAAE;SACf,CAAC;QAEF,IAAI,UAAU,EAAE,CAAC;YACf,MAAM,WAAW,GAAwB,EAAE,CAAC;YAC5C,KAAK,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;gBACrD,WAAW,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,CAAC;YACxD,CAAC;YACD,MAAM,CAAC,UAAU,GAAG,WAAW,CAAC;QAClC,CAAC;QAED,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,CAAC,QAAQ,GAAG,QAAQ,CAAC;QAC7B,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,uBAAuB,CAAC,IAA6B;QAC3D,MAAM,IAAI,GAAG,IAAI,CAAC,IAAc,CAAC;QACjC,MAAM,MAAM,GAA4B;YACtC,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;SAC/B,CAAC;QAEF,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,MAAM,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;QACxC,CAAC;QAED,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;YACrB,MAAM,KAAK,GAAG,IAAI,CAAC,KAA4C,CAAC;YAChE,MAAM,CAAC,KAAK,GAAG,KAAK;gBAClB,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,KAAK,CAAC;gBACrC,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,CAAC,MAAM,EAAE,CAAC;QAClC,CAAC;QAED,IAAI,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACzC,MAAM,MAAM,GAAG,IAAI,CAAC,UAAqD,CAAC;YAC1E,MAAM,WAAW,GAA4B,EAAE,CAAC;YAChD,KAAK,MAAM,CAAC,GAAG,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBACvD,WAAW,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,uBAAuB,CAAC,UAAU,CAAC,CAAC;YAC9D,CAAC;YACD,MAAM,CAAC,UAAU,GAAG,WAAW,CAAC;YAChC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClB,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;YAClC,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,aAAa,CAAC,IAAY;QAChC,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,QAAQ,CAAC,CAAC,OAAO,UAAU,CAAC,MAAM,CAAC;YACxC,KAAK,QAAQ,CAAC,CAAC,OAAO,UAAU,CAAC,MAAM,CAAC;YACxC,KAAK,SAAS,CAAC,CAAC,OAAO,UAAU,CAAC,OAAO,CAAC;YAC1C,KAAK,SAAS,CAAC,CAAC,OAAO,UAAU,CAAC,OAAO,CAAC;YAC1C,KAAK,OAAO,CAAC,CAAC,OAAO,UAAU,CAAC,KAAK,CAAC;YACtC,KAAK,QAAQ,CAAC,CAAC,OAAO,UAAU,CAAC,MAAM,CAAC;YACxC,OAAO,CAAC,CAAC,OAAO,UAAU,CAAC,MAAM,CAAC;QACpC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,QAAuB;QAClC,MAAM,EAAE,iBAAiB,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,wBAAwB,CAAC,QAAQ,CAAC,CAAC;QAEhF,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;QACnC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACnD,gBAAgB,EAAE;gBAChB,GAAG,CAAC,IAAI,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC5E,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC/D;YACD,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC5B,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,eAAe,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;QACzD,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QACjC,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;QAE3C,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,IAAI,SAAS,GAAG,EAAE,CAAC;QACnB,MAAM,SAAS,GAAe,EAAE,CAAC;QAEjC,IAAI,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YAC9B,KAAK,MAAM,IAAI,IAAI,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;gBAC3C,IAAK,IAAY,CAAC,OAAO,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;oBACvC,SAAS,IAAI,IAAI,CAAC,IAAI,CAAC;gBACzB,CAAC;qBAAM,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;oBACrB,OAAO,IAAI,IAAI,CAAC,IAAI,CAAC;gBACvB,CAAC;gBACD,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;oBACtB,SAAS,CAAC,IAAI,CAAC;wBACb,EAAE,EAAE,QAAQ,MAAM,CAAC,UAAU,EAAE,EAAE;wBACjC,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI;wBAC5B,IAAI,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,IAAI,EAAE,CAA4B;qBAChE,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,SAAS,EAAE,CAAC;YACd,MAAM,CAAC,KAAK,CAAC,uCAAuC,SAAS,CAAC,MAAM,SAAS,CAAC,CAAC;QACjF,CAAC;QAED,MAAM,KAAK,GAAG,QAAQ,CAAC,aAAa,CAAC;QAErC,OAAO;YACL,OAAO;YACP,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACnC,UAAU,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;YACxD,cAAc,EAAE,KAAK;gBACnB,CAAC,CAAC;oBACE,YAAY,EAAE,KAAK,CAAC,gBAAgB,IAAI,CAAC;oBACzC,aAAa,EAAE,KAAK,CAAC,oBAAoB,IAAI,CAAC;oBAC9C,YAAY,EAAE,KAAK,CAAC,eAAe,IAAI,CAAC;iBACzC;gBACH,CAAC,CAAC,SAAS;SACd,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CACX,QAAuB,EACvB,OAAkC;QAElC,MAAM,EAAE,iBAAiB,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,wBAAwB,CAAC,QAAQ,CAAC,CAAC;QAChF,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,CAAC;QAE/B,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;QACnC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACnD,gBAAgB,EAAE;gBAChB,GAAG,CAAC,IAAI,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC5E,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC/D;YACD,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC5B,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,qBAAqB,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;QAE/D,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YACxC,IAAI,MAAM,EAAE,OAAO;gBAAE,MAAM;YAE3B,MAAM,SAAS,GAAG,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC;YACxC,IAAI,OAAO,GAAG,EAAE,CAAC;YACjB,IAAI,SAAS,GAAG,EAAE,CAAC;YACnB,MAAM,SAAS,GAAe,EAAE,CAAC;YAEjC,IAAI,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;gBAC9B,KAAK,MAAM,IAAI,IAAI,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;oBAC3C,IAAK,IAAY,CAAC,OAAO,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;wBACvC,SAAS,IAAI,IAAI,CAAC,IAAI,CAAC;oBACzB,CAAC;yBAAM,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;wBACrB,OAAO,IAAI,IAAI,CAAC,IAAI,CAAC;oBACvB,CAAC;oBACD,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;wBACtB,SAAS,CAAC,IAAI,CAAC;4BACb,EAAE,EAAE,QAAQ,MAAM,CAAC,UAAU,EAAE,EAAE;4BACjC,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI;4BAC5B,IAAI,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,IAAI,EAAE,CAA4B;yBAChE,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;YACH,CAAC;YAED,MAAM;gBACJ,OAAO;gBACP,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACnC,UAAU,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;gBACxD,cAAc,EAAE,KAAK,CAAC,aAAa;oBACjC,CAAC,CAAC;wBACE,YAAY,EAAE,KAAK,CAAC,aAAa,CAAC,gBAAgB,IAAI,CAAC;wBACvD,aAAa,EAAE,KAAK,CAAC,aAAa,CAAC,oBAAoB,IAAI,CAAC;wBAC5D,YAAY,EAAE,KAAK,CAAC,aAAa,CAAC,eAAe,IAAI,CAAC;qBACvD;oBACH,CAAC,CAAC,SAAS;aACd,CAAC;QACJ,CAAC;IACH,CAAC;IAED,SAAS,CAAC,KAAuB;QAC/B,MAAM,KAAK,GAAG,IAAI,eAAe,CAAC;YAChC,MAAM,EAAE,EAAE;YACV,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,CAAC,CAAC;QACH,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACzB,KAAK,CAAC,UAAU,GAAG,KAAK,CAAC;QACzB,KAAK,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC;QAC/C,OAAO,KAAK,CAAC;IACf,CAAC;IAED,oBAAoB,CAAC,MAA+B;QAClD,MAAM,OAAO,GAAG,IAAI,eAAe,CAAC;YAClC,MAAM,EAAE,EAAE;YACV,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,CAAC,CAAC;QACH,OAAO,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QAC3B,OAAO,CAAC,gBAAgB,GAAG,MAAM,CAAC;QAClC,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC;QACrC,OAAO,OAAO,CAAC;IACjB,CAAC;CACF"}

package/dist/lib/llm/providers/gemini-embeddings.d.ts

@@ -0,0 +1,14 @@
+import type { Embeddings } from '../../types/llm-types.ts';
+interface GeminiEmbeddingsOptions {
+    apiKey?: string;
+    modelName: string;
+}
+export declare class GeminiEmbeddingsProvider implements Embeddings {
+    private genAI;
+    private modelName;
+    constructor(options: GeminiEmbeddingsOptions);
+    embedQuery(text: string): Promise<number[]>;
+    embedDocuments(texts: string[]): Promise<number[][]>;
+}
+export {};
+//# sourceMappingURL=gemini-embeddings.d.ts.map

package/dist/lib/llm/providers/gemini-embeddings.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini-embeddings.d.ts","sourceRoot":"","sources":["../../../../lib/llm/providers/gemini-embeddings.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AAE3D,UAAU,uBAAuB;IAC/B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,qBAAa,wBAAyB,YAAW,UAAU;IACzD,OAAO,CAAC,KAAK,CAAqB;IAClC,OAAO,CAAC,SAAS,CAAS;gBAEd,OAAO,EAAE,uBAAuB;IAKtC,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAM3C,cAAc,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;CAO3D"}