ys-code-agent 2.0.0

package/dist/src/providers/anthropic.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../../src/src/providers/anthropic.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,cAAc,EAAE,WAAW,EAAE,gBAAgB,EAAE,mBAAmB,EAAY,MAAM,aAAa,CAAC;AAE3G,qBAAa,iBAAkB,SAAQ,YAAY;gBACrC,MAAM,EAAE,cAAc,EAAE,WAAW,EAAE,WAAW;IAI5D,OAAO,IAAI,MAAM;IAIjB,OAAO,IAAI,MAAM;IAIjB,SAAS,CAAC,YAAY,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAQhD,SAAS,CAAC,gBAAgB,CAAC,QAAQ,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,EAAE,KAAK,CAAC,EAAE,OAAO,EAAE,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IA2B1H,OAAO,CAAC,eAAe;IAqBjB,QAAQ,CACZ,QAAQ,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,EAClD,KAAK,CAAC,EAAE,OAAO,EAAE,GAChB,OAAO,CAAC,gBAAgB,CAAC;IA4DrB,cAAc,CACnB,QAAQ,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,EAClD,KAAK,CAAC,EAAE,OAAO,EAAE,GAChB,cAAc,CAAC,mBAAmB,CAAC;CAiGvC"}

package/dist/src/providers/anthropic.js

@@ -0,0 +1,202 @@
+import { BaseProvider } from './base.js';
+export class AnthropicProvider extends BaseProvider {
+    constructor(config, modelConfig) {
+        super(config, modelConfig);
+    }
+    getName() {
+        return 'Anthropic';
+    }
+    getType() {
+        return 'anthropic';
+    }
+    buildHeaders() {
+        return {
+            'Content-Type': 'application/json',
+            'x-api-key': this.getApiKey(),
+            'anthropic-version': '2023-06-01',
+        };
+    }
+    buildRequestBody(messages, tools) {
+        const systemMessages = messages.filter((m) => m.role === 'system');
+        const chatMessages = messages.filter((m) => m.role !== 'system');
+        const body = {
+            model: this.modelConfig.model,
+            messages: chatMessages.map((m) => ({
+                role: m.role === 'assistant' ? 'assistant' : 'user',
+                content: m.content,
+            })),
+            max_tokens: this.modelConfig.maxTokens,
+            temperature: this.modelConfig.temperature,
+            top_p: this.modelConfig.topP,
+            stream: false,
+        };
+        if (systemMessages.length > 0) {
+            body.system = systemMessages.map((m) => ({ type: 'text', text: m.content }));
+        }
+        if (tools && tools.length > 0) {
+            body.tools = tools;
+        }
+        return body;
+    }
+    convertMessages(messages) {
+        const systemMessages = [];
+        const chatMessages = [];
+        for (const msg of messages) {
+            if (msg.role === 'system') {
+                systemMessages.push({ type: 'text', text: msg.content });
+            }
+            else {
+                chatMessages.push({
+                    role: msg.role === 'assistant' ? 'assistant' : 'user',
+                    content: msg.content,
+                });
+            }
+        }
+        return { systemMessages, chatMessages };
+    }
+    async generate(messages, tools) {
+        const url = `${this.config.baseUrl}/messages`;
+        const { systemMessages, chatMessages } = this.convertMessages(messages);
+        const body = {
+            model: this.modelConfig.model,
+            messages: chatMessages,
+            max_tokens: this.modelConfig.maxTokens,
+            temperature: this.modelConfig.temperature,
+            top_p: this.modelConfig.topP,
+            stream: false,
+        };
+        if (systemMessages.length > 0) {
+            body.system = systemMessages;
+        }
+        if (tools && tools.length > 0) {
+            body.tools = tools;
+        }
+        const response = await this.fetchWithRetry(url, {
+            method: 'POST',
+            headers: this.buildHeaders(),
+            body: JSON.stringify(body),
+        });
+        const data = await response.json();
+        const content = data.content || [];
+        let textContent = '';
+        const toolCalls = [];
+        for (const block of content) {
+            if (block.type === 'text') {
+                textContent += block.text || '';
+            }
+            else if (block.type === 'tool_use') {
+                toolCalls.push({
+                    id: block.id,
+                    name: block.name,
+                    arguments: block.input || {},
+                    timestamp: Date.now(),
+                });
+            }
+        }
+        const usage = data.usage || {};
+        return {
+            content: textContent,
+            toolCalls,
+            usage: {
+                promptTokens: usage.input_tokens || 0,
+                completionTokens: usage.output_tokens || 0,
+                totalTokens: (usage.input_tokens || 0) + (usage.output_tokens || 0),
+            },
+            finishReason: data.stop_reason || 'end_turn',
+        };
+    }
+    async *generateStream(messages, tools) {
+        const url = `${this.config.baseUrl}/messages`;
+        const { systemMessages, chatMessages } = this.convertMessages(messages);
+        const body = {
+            model: this.modelConfig.model,
+            messages: chatMessages,
+            max_tokens: this.modelConfig.maxTokens,
+            temperature: this.modelConfig.temperature,
+            top_p: this.modelConfig.topP,
+            stream: true,
+        };
+        if (systemMessages.length > 0) {
+            body.system = systemMessages;
+        }
+        if (tools && tools.length > 0) {
+            body.tools = tools;
+        }
+        const response = await this.fetchWithRetry(url, {
+            method: 'POST',
+            headers: this.buildHeaders(),
+            body: JSON.stringify(body),
+        });
+        const reader = response.body?.getReader();
+        if (!reader)
+            throw new Error('No response body stream available');
+        const decoder = new TextDecoder();
+        let buffer = '';
+        try {
+            while (true) {
+                const { done, value } = await reader.read();
+                if (done)
+                    break;
+                buffer += decoder.decode(value, { stream: true });
+                const lines = buffer.split('\n');
+                buffer = lines.pop() || '';
+                for (const line of lines) {
+                    const trimmed = line.trim();
+                    if (!trimmed || !trimmed.startsWith('data: '))
+                        continue;
+                    const dataStr = trimmed.slice(6);
+                    if (dataStr === '[DONE]') {
+                        yield { type: 'done' };
+                        return;
+                    }
+                    try {
+                        const data = JSON.parse(dataStr);
+                        const eventType = data.type;
+                        if (eventType === 'content_block_delta') {
+                            const delta = data.delta || {};
+                            if (delta.type === 'text_delta' && delta.text) {
+                                yield { type: 'text', content: delta.text };
+                            }
+                        }
+                        else if (eventType === 'content_block_start') {
+                            const block = data.content_block || {};
+                            if (block.type === 'tool_use') {
+                                yield {
+                                    type: 'tool_call',
+                                    toolCall: {
+                                        id: block.id,
+                                        name: block.name,
+                                        arguments: block.input || {},
+                                        timestamp: Date.now(),
+                                    },
+                                };
+                            }
+                        }
+                        else if (eventType === 'message_done') {
+                            const usage = data.usage;
+                            if (usage) {
+                                yield {
+                                    type: 'done',
+                                    usage: {
+                                        promptTokens: usage.input_tokens || 0,
+                                        completionTokens: usage.output_tokens || 0,
+                                        totalTokens: (usage.input_tokens || 0) + (usage.output_tokens || 0),
+                                    },
+                                };
+                            }
+                            else {
+                                yield { type: 'done' };
+                            }
+                        }
+                    }
+                    catch {
+                    }
+                }
+            }
+        }
+        finally {
+            reader.releaseLock();
+        }
+    }
+}
+//# sourceMappingURL=anthropic.js.map

package/dist/src/providers/anthropic.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../../src/src/providers/anthropic.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAGzC,MAAM,OAAO,iBAAkB,SAAQ,YAAY;IACjD,YAAY,MAAsB,EAAE,WAAwB;QAC1D,KAAK,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;IAC7B,CAAC;IAED,OAAO;QACL,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,OAAO;QACL,OAAO,WAAW,CAAC;IACrB,CAAC;IAES,YAAY;QACpB,OAAO;YACL,cAAc,EAAE,kBAAkB;YAClC,WAAW,EAAE,IAAI,CAAC,SAAS,EAAE;YAC7B,mBAAmB,EAAE,YAAY;SAClC,CAAC;IACJ,CAAC;IAES,gBAAgB,CAAC,QAAkD,EAAE,KAAiB;QAC9F,MAAM,cAAc,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QACnE,MAAM,YAAY,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QAEjE,MAAM,IAAI,GAA4B;YACpC,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,KAAK;YAC7B,QAAQ,EAAE,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACjC,IAAI,EAAE,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM;gBACnD,OAAO,EAAE,CAAC,CAAC,OAAO;aACnB,CAAC,CAAC;YACH,UAAU,EAAE,IAAI,CAAC,WAAW,CAAC,SAAS;YACtC,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,WAAW;YACzC,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI;YAC5B,MAAM,EAAE,KAAK;SACd,CAAC;QAEF,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAC/E,CAAC;QAED,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACrB,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,eAAe,CAAC,QAAkD;QAIxE,MAAM,cAAc,GAA0C,EAAE,CAAC;QACjE,MAAM,YAAY,GAA6C,EAAE,CAAC;QAElE,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC3B,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC1B,cAAc,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;YAC3D,CAAC;iBAAM,CAAC;gBACN,YAAY,CAAC,IAAI,CAAC;oBAChB,IAAI,EAAE,GAAG,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM;oBACrD,OAAO,EAAE,GAAG,CAAC,OAAO;iBACrB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,QAAkD,EAClD,KAAiB;QAEjB,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,WAAW,CAAC;QAC9C,MAAM,EAAE,cAAc,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QAExE,MAAM,IAAI,GAA4B;YACpC,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,KAAK;YAC7B,QAAQ,EAAE,YAAY;YACtB,UAAU,EAAE,IAAI,CAAC,WAAW,CAAC,SAAS;YACtC,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,WAAW;YACzC,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI;YAC5B,MAAM,EAAE,KAAK;SACd,CAAC;QAEF,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC;QAC/B,CAAC;QAED,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACrB,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE;YAC9C,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,IAAI,CAAC,YAAY,EAAE;YAC5B,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAA6B,CAAC;QAC9D,MAAM,OAAO,GAAG,IAAI,CAAC,OAAyC,IAAI,EAAE,CAAC;QAErE,IAAI,WAAW,GAAG,EAAE,CAAC;QACrB,MAAM,SAAS,GAAe,EAAE,CAAC;QAEjC,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC1B,WAAW,IAAI,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC;YAClC,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBACrC,SAAS,CAAC,IAAI,CAAC;oBACb,EAAE,EAAE,KAAK,CAAC,EAAY;oBACtB,IAAI,EAAE,KAAK,CAAC,IAAc;oBAC1B,SAAS,EAAE,KAAK,CAAC,KAAgC,IAAI,EAAE;oBACvD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;iBACtB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,KAAgC,IAAI,EAAE,CAAC;QAE1D,OAAO;YACL,OAAO,EAAE,WAAW;YACpB,SAAS;YACT,KAAK,EAAE;gBACL,YAAY,EAAG,KAAK,CAAC,YAAuB,IAAI,CAAC;gBACjD,gBAAgB,EAAG,KAAK,CAAC,aAAwB,IAAI,CAAC;gBACtD,WAAW,EAAE,CAAE,KAAK,CAAC,YAAuB,IAAI,CAAC,CAAC,GAAG,CAAE,KAAK,CAAC,aAAwB,IAAI,CAAC,CAAC;aAC5F;YACD,YAAY,EAAG,IAAI,CAAC,WAAsB,IAAI,UAAU;SACzD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,CAAC,cAAc,CACnB,QAAkD,EAClD,KAAiB;QAEjB,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,WAAW,CAAC;QAC9C,MAAM,EAAE,cAAc,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QAExE,MAAM,IAAI,GAA4B;YACpC,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,KAAK;YAC7B,QAAQ,EAAE,YAAY;YACtB,UAAU,EAAE,IAAI,CAAC,WAAW,CAAC,SAAS;YACtC,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,WAAW;YACzC,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI;YAC5B,MAAM,EAAE,IAAI;SACb,CAAC;QAEF,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC;QAC/B,CAAC;QAED,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACrB,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE;YAC9C,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,IAAI,CAAC,YAAY,EAAE;YAC5B,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC3B,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;QAC1C,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QAElE,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;QAClC,IAAI,MAAM,GAAG,EAAE,CAAC;QAEhB,IAAI,CAAC;YACH,OAAO,IAAI,EAAE,CAAC;gBACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC5C,IAAI,IAAI;oBAAE,MAAM;gBAEhB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBAClD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;gBAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACzB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;oBAC5B,IAAI,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC;wBAAE,SAAS;oBAExD,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;oBACjC,IAAI,OAAO,KAAK,QAAQ,EAAE,CAAC;wBACzB,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;wBACvB,OAAO;oBACT,CAAC;oBAED,IAAI,CAAC;wBACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAA4B,CAAC;wBAC5D,MAAM,SAAS,GAAG,IAAI,CAAC,IAAc,CAAC;wBAEtC,IAAI,SAAS,KAAK,qBAAqB,EAAE,CAAC;4BACxC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAgC,IAAI,EAAE,CAAC;4BAC1D,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;gCAC9C,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,IAAc,EAAE,CAAC;4BACxD,CAAC;wBACH,CAAC;6BAAM,IAAI,SAAS,KAAK,qBAAqB,EAAE,CAAC;4BAC/C,MAAM,KAAK,GAAG,IAAI,CAAC,aAAwC,IAAI,EAAE,CAAC;4BAClE,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gCAC9B,MAAM;oCACJ,IAAI,EAAE,WAAW;oCACjB,QAAQ,EAAE;wCACR,EAAE,EAAE,KAAK,CAAC,EAAY;wCACtB,IAAI,EAAE,KAAK,CAAC,IAAc;wCAC1B,SAAS,EAAG,KAAK,CAAC,KAAiC,IAAI,EAAE;wCACzD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;qCACtB;iCACF,CAAC;4BACJ,CAAC;wBACH,CAAC;6BAAM,IAAI,SAAS,KAAK,cAAc,EAAE,CAAC;4BACxC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAgC,CAAC;4BACpD,IAAI,KAAK,EAAE,CAAC;gCACV,MAAM;oCACJ,IAAI,EAAE,MAAM;oCACZ,KAAK,EAAE;wCACL,YAAY,EAAG,KAAK,CAAC,YAAuB,IAAI,CAAC;wCACjD,gBAAgB,EAAG,KAAK,CAAC,aAAwB,IAAI,CAAC;wCACtD,WAAW,EAAE,CAAE,KAAK,CAAC,YAAuB,IAAI,CAAC,CAAC,GAAG,CAAE,KAAK,CAAC,aAAwB,IAAI,CAAC,CAAC;qCAC5F;iCACF,CAAC;4BACJ,CAAC;iCAAM,CAAC;gCACN,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;4BACzB,CAAC;wBACH,CAAC;oBACH,CAAC;oBAAC,MAAM,CAAC;oBACT,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,CAAC;IACH,CAAC;CACF"}

package/dist/src/providers/base.d.ts

@@ -0,0 +1,29 @@
+import { ProviderConfig, ModelConfig, ProviderResponse, ProviderStreamChunk, ToolCall } from '../types.js';
+export declare abstract class BaseProvider {
+    protected config: ProviderConfig;
+    protected modelConfig: ModelConfig;
+    constructor(config: ProviderConfig, modelConfig: ModelConfig);
+    abstract getName(): string;
+    abstract getType(): string;
+    abstract generate(messages: Array<{
+        role: string;
+        content: string;
+    }>, tools?: unknown[]): Promise<ProviderResponse>;
+    abstract generateStream(messages: Array<{
+        role: string;
+        content: string;
+    }>, tools?: unknown[]): AsyncGenerator<ProviderStreamChunk>;
+    protected getApiKey(): string;
+    protected buildHeaders(): Record<string, string>;
+    protected buildRequestBody(messages: Array<{
+        role: string;
+        content: string;
+    }>, tools?: unknown[]): Record<string, unknown>;
+    protected buildStreamBody(messages: Array<{
+        role: string;
+        content: string;
+    }>, tools?: unknown[]): Record<string, unknown>;
+    protected fetchWithRetry(url: string, options: RequestInit, retries?: number): Promise<Response>;
+    protected parseToolCalls(data: Record<string, unknown>): ToolCall[];
+}
+//# sourceMappingURL=base.d.ts.map

package/dist/src/providers/base.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../src/src/providers/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,WAAW,EAAE,gBAAgB,EAAE,mBAAmB,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAE3G,8BAAsB,YAAY;IAChC,SAAS,CAAC,MAAM,EAAE,cAAc,CAAC;IACjC,SAAS,CAAC,WAAW,EAAE,WAAW,CAAC;gBAEvB,MAAM,EAAE,cAAc,EAAE,WAAW,EAAE,WAAW;IAK5D,QAAQ,CAAC,OAAO,IAAI,MAAM;IAC1B,QAAQ,CAAC,OAAO,IAAI,MAAM;IAC1B,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,EAAE,KAAK,CAAC,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,gBAAgB,CAAC;IACnH,QAAQ,CAAC,cAAc,CACrB,QAAQ,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,EAClD,KAAK,CAAC,EAAE,OAAO,EAAE,GAChB,cAAc,CAAC,mBAAmB,CAAC;IAEtC,SAAS,CAAC,SAAS,IAAI,MAAM;IAI7B,SAAS,CAAC,YAAY,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAOhD,SAAS,CAAC,gBAAgB,CAAC,QAAQ,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,EAAE,KAAK,CAAC,EAAE,OAAO,EAAE,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IA2B1H,SAAS,CAAC,eAAe,CAAC,QAAQ,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,EAAE,KAAK,CAAC,EAAE,OAAO,EAAE,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;cAMzG,cAAc,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,OAAO,SAAyB,GAAG,OAAO,CAAC,QAAQ,CAAC;IAoDtH,SAAS,CAAC,cAAc,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,QAAQ,EAAE;CA0BpE"}

package/dist/src/providers/base.js

@@ -0,0 +1,112 @@
+export class BaseProvider {
+    config;
+    modelConfig;
+    constructor(config, modelConfig) {
+        this.config = config;
+        this.modelConfig = modelConfig;
+    }
+    getApiKey() {
+        return this.config.apiKey || process.env[`${this.config.type.toUpperCase()}_API_KEY`] || '';
+    }
+    buildHeaders() {
+        return {
+            'Content-Type': 'application/json',
+            'Authorization': `Bearer ${this.getApiKey()}`,
+        };
+    }
+    buildRequestBody(messages, tools) {
+        const body = {
+            model: this.modelConfig.model,
+            messages: messages.map((m) => ({
+                role: m.role,
+                content: m.content,
+            })),
+            temperature: this.modelConfig.temperature,
+            max_tokens: this.modelConfig.maxTokens,
+            top_p: this.modelConfig.topP,
+            frequency_penalty: this.modelConfig.frequencyPenalty,
+            presence_penalty: this.modelConfig.presencePenalty,
+            stream: false,
+        };
+        if (this.modelConfig.stop && this.modelConfig.stop.length > 0) {
+            body.stop = this.modelConfig.stop;
+        }
+        if (tools && tools.length > 0) {
+            body.tools = tools;
+            body.tool_choice = 'auto';
+        }
+        return body;
+    }
+    buildStreamBody(messages, tools) {
+        const body = this.buildRequestBody(messages, tools);
+        body.stream = true;
+        return body;
+    }
+    async fetchWithRetry(url, options, retries = this.config.maxRetries) {
+        let lastError = null;
+        for (let attempt = 0; attempt <= retries; attempt++) {
+            try {
+                const controller = new AbortController();
+                const timeoutId = setTimeout(() => controller.abort(), this.config.timeout);
+                const response = await fetch(url, {
+                    ...options,
+                    signal: controller.signal,
+                });
+                clearTimeout(timeoutId);
+                if (response.status === 429) {
+                    const retryAfter = parseInt(response.headers.get('retry-after') || '5', 10);
+                    await new Promise((resolve) => setTimeout(resolve, retryAfter * 1000));
+                    continue;
+                }
+                if (response.status === 401) {
+                    throw new Error('Authentication failed. Please check your API key.');
+                }
+                if (response.status === 403) {
+                    throw new Error('Access forbidden. Your API key may not have access to this model.');
+                }
+                if (!response.ok) {
+                    const errorText = await response.text();
+                    throw new Error(`HTTP ${response.status}: ${errorText}`);
+                }
+                return response;
+            }
+            catch (error) {
+                lastError = error instanceof Error ? error : new Error(String(error));
+                if (error instanceof Error && error.name === 'AbortError') {
+                    throw new Error('Request timed out');
+                }
+                if (attempt < retries) {
+                    const delay = Math.min(1000 * Math.pow(2, attempt), 10000);
+                    await new Promise((resolve) => setTimeout(resolve, delay));
+                }
+            }
+        }
+        throw lastError || new Error('Request failed after all retries');
+    }
+    parseToolCalls(data) {
+        const toolCalls = [];
+        const choices = data.choices;
+        if (!choices || choices.length === 0)
+            return toolCalls;
+        const message = choices[0].message;
+        if (!message)
+            return toolCalls;
+        const calls = message.tool_calls;
+        if (!calls)
+            return toolCalls;
+        for (const call of calls) {
+            try {
+                toolCalls.push({
+                    id: call.id,
+                    name: call.function ? call.function.name : '',
+                    arguments: call.function ? JSON.parse(call.function.arguments) : {},
+                    timestamp: Date.now(),
+                });
+            }
+            catch {
+            }
+        }
+        return toolCalls;
+    }
+}
+//# sourceMappingURL=base.js.map

package/dist/src/providers/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../src/src/providers/base.ts"],"names":[],"mappings":"AAEA,MAAM,OAAgB,YAAY;IACtB,MAAM,CAAiB;IACvB,WAAW,CAAc;IAEnC,YAAY,MAAsB,EAAE,WAAwB;QAC1D,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;IACjC,CAAC;IAUS,SAAS;QACjB,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,UAAU,CAAC,IAAI,EAAE,CAAC;IAC9F,CAAC;IAES,YAAY;QACpB,OAAO;YACL,cAAc,EAAE,kBAAkB;YAClC,eAAe,EAAE,UAAU,IAAI,CAAC,SAAS,EAAE,EAAE;SAC9C,CAAC;IACJ,CAAC;IAES,gBAAgB,CAAC,QAAkD,EAAE,KAAiB;QAC9F,MAAM,IAAI,GAA4B;YACpC,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,KAAK;YAC7B,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC7B,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,OAAO,EAAE,CAAC,CAAC,OAAO;aACnB,CAAC,CAAC;YACH,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,WAAW;YACzC,UAAU,EAAE,IAAI,CAAC,WAAW,CAAC,SAAS;YACtC,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI;YAC5B,iBAAiB,EAAE,IAAI,CAAC,WAAW,CAAC,gBAAgB;YACpD,gBAAgB,EAAE,IAAI,CAAC,WAAW,CAAC,eAAe;YAClD,MAAM,EAAE,KAAK;SACd,CAAC;QAEF,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9D,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;QACpC,CAAC;QAED,IAAI,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9B,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;YACnB,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC;QAC5B,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAES,eAAe,CAAC,QAAkD,EAAE,KAAiB;QAC7F,MAAM,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QACpD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,OAAO,IAAI,CAAC;IACd,CAAC;IAES,KAAK,CAAC,cAAc,CAAC,GAAW,EAAE,OAAoB,EAAE,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU;QAChG,IAAI,SAAS,GAAiB,IAAI,CAAC;QAEnC,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC;YACpD,IAAI,CAAC;gBACH,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;gBACzC,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;gBAE5E,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;oBAChC,GAAG,OAAO;oBACV,MAAM,EAAE,UAAU,CAAC,MAAM;iBAC1B,CAAC,CAAC;gBAEH,YAAY,CAAC,SAAS,CAAC,CAAC;gBAExB,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;oBAC5B,MAAM,UAAU,GAAG,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,IAAI,GAAG,EAAE,EAAE,CAAC,CAAC;oBAC5E,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,UAAU,GAAG,IAAI,CAAC,CAAC,CAAC;oBACvE,SAAS;gBACX,CAAC;gBAED,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;oBAC5B,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;gBACvE,CAAC;gBAED,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;oBAC5B,MAAM,IAAI,KAAK,CAAC,mEAAmE,CAAC,CAAC;gBACvF,CAAC;gBAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;oBACxC,MAAM,IAAI,KAAK,CAAC,QAAQ,QAAQ,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC,CAAC;gBAC3D,CAAC;gBAED,OAAO,QAAQ,CAAC;YAClB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,SAAS,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;gBAEtE,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;oBAC1D,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;gBACvC,CAAC;gBAED,IAAI,OAAO,GAAG,OAAO,EAAE,CAAC;oBACtB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,KAAK,CAAC,CAAC;oBAC3D,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;gBAC7D,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,SAAS,IAAI,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;IACnE,CAAC;IAES,cAAc,CAAC,IAA6B;QACpD,MAAM,SAAS,GAAe,EAAE,CAAC;QACjC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAqD,CAAC;QAE3E,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,SAAS,CAAC;QAEvD,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,OAA8C,CAAC;QAC1E,IAAI,CAAC,OAAO;YAAE,OAAO,SAAS,CAAC;QAE/B,MAAM,KAAK,GAAG,OAAO,CAAC,UAAwD,CAAC;QAC/E,IAAI,CAAC,KAAK;YAAE,OAAO,SAAS,CAAC;QAE7B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC;gBACH,SAAS,CAAC,IAAI,CAAC;oBACb,EAAE,EAAE,IAAI,CAAC,EAAY;oBACrB,IAAI,EAAE,IAAI,CAAC,QAAmB,CAAC,CAAC,CAAE,IAAI,CAAC,QAAoC,CAAC,IAAc,CAAC,CAAC,CAAC,EAAE;oBAC/F,SAAS,EAAE,IAAI,CAAC,QAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAE,IAAI,CAAC,QAAoC,CAAC,SAAmB,CAAC,CAAC,CAAC,CAAC,EAAE;oBACrH,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;iBACtB,CAAC,CAAC;YACL,CAAC;YAAC,MAAM,CAAC;YACT,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;CACF"}

package/dist/src/providers/custom.d.ts

@@ -0,0 +1,8 @@
+import { OpenAIProvider } from './openai.js';
+import { ProviderConfig, ModelConfig } from '../types.js';
+export declare class CustomProvider extends OpenAIProvider {
+    constructor(config: ProviderConfig, modelConfig: ModelConfig);
+    getName(): string;
+    getType(): string;
+}
+//# sourceMappingURL=custom.d.ts.map

package/dist/src/providers/custom.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"custom.d.ts","sourceRoot":"","sources":["../../../src/src/providers/custom.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE1D,qBAAa,cAAe,SAAQ,cAAc;gBACpC,MAAM,EAAE,cAAc,EAAE,WAAW,EAAE,WAAW;IAI5D,OAAO,IAAI,MAAM;IAIjB,OAAO,IAAI,MAAM;CAGlB"}

package/dist/src/providers/custom.js

@@ -0,0 +1,13 @@
+import { OpenAIProvider } from './openai.js';
+export class CustomProvider extends OpenAIProvider {
+    constructor(config, modelConfig) {
+        super(config, modelConfig);
+    }
+    getName() {
+        return this.config.name || 'Custom';
+    }
+    getType() {
+        return 'custom';
+    }
+}
+//# sourceMappingURL=custom.js.map

package/dist/src/providers/custom.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"custom.js","sourceRoot":"","sources":["../../../src/src/providers/custom.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAG7C,MAAM,OAAO,cAAe,SAAQ,cAAc;IAChD,YAAY,MAAsB,EAAE,WAAwB;QAC1D,KAAK,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;IAC7B,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,QAAQ,CAAC;IACtC,CAAC;IAED,OAAO;QACL,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF"}

package/dist/src/providers/deepseek.d.ts

@@ -0,0 +1,8 @@
+import { OpenAIProvider } from './openai.js';
+import { ProviderConfig, ModelConfig } from '../types.js';
+export declare class DeepSeekProvider extends OpenAIProvider {
+    constructor(config: ProviderConfig, modelConfig: ModelConfig);
+    getName(): string;
+    getType(): string;
+}
+//# sourceMappingURL=deepseek.d.ts.map

package/dist/src/providers/deepseek.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"deepseek.d.ts","sourceRoot":"","sources":["../../../src/src/providers/deepseek.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE1D,qBAAa,gBAAiB,SAAQ,cAAc;gBACtC,MAAM,EAAE,cAAc,EAAE,WAAW,EAAE,WAAW;IAI5D,OAAO,IAAI,MAAM;IAIjB,OAAO,IAAI,MAAM;CAGlB"}

package/dist/src/providers/deepseek.js

@@ -0,0 +1,13 @@
+import { OpenAIProvider } from './openai.js';
+export class DeepSeekProvider extends OpenAIProvider {
+    constructor(config, modelConfig) {
+        super(config, modelConfig);
+    }
+    getName() {
+        return 'DeepSeek';
+    }
+    getType() {
+        return 'deepseek';
+    }
+}
+//# sourceMappingURL=deepseek.js.map

package/dist/src/providers/deepseek.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"deepseek.js","sourceRoot":"","sources":["../../../src/src/providers/deepseek.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAG7C,MAAM,OAAO,gBAAiB,SAAQ,cAAc;IAClD,YAAY,MAAsB,EAAE,WAAwB;QAC1D,KAAK,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;IAC7B,CAAC;IAED,OAAO;QACL,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,OAAO;QACL,OAAO,UAAU,CAAC;IACpB,CAAC;CACF"}

package/dist/src/providers/gemini.d.ts

@@ -0,0 +1,18 @@
+import { BaseProvider } from './base.js';
+import { ProviderConfig, ModelConfig, ProviderResponse, ProviderStreamChunk } from '../types.js';
+export declare class GeminiProvider extends BaseProvider {
+    constructor(config: ProviderConfig, modelConfig: ModelConfig);
+    getName(): string;
+    getType(): string;
+    protected buildHeaders(): Record<string, string>;
+    private convertMessages;
+    generate(messages: Array<{
+        role: string;
+        content: string;
+    }>, tools?: unknown[]): Promise<ProviderResponse>;
+    generateStream(messages: Array<{
+        role: string;
+        content: string;
+    }>, tools?: unknown[]): AsyncGenerator<ProviderStreamChunk>;
+}
+//# sourceMappingURL=gemini.d.ts.map

package/dist/src/providers/gemini.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../../src/src/providers/gemini.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,cAAc,EAAE,WAAW,EAAE,gBAAgB,EAAE,mBAAmB,EAAY,MAAM,aAAa,CAAC;AAE3G,qBAAa,cAAe,SAAQ,YAAY;gBAClC,MAAM,EAAE,cAAc,EAAE,WAAW,EAAE,WAAW;IAI5D,OAAO,IAAI,MAAM;IAIjB,OAAO,IAAI,MAAM;IAIjB,SAAS,CAAC,YAAY,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAMhD,OAAO,CAAC,eAAe;IAsBjB,QAAQ,CACZ,QAAQ,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,EAClD,KAAK,CAAC,EAAE,OAAO,EAAE,GAChB,OAAO,CAAC,gBAAgB,CAAC;IAsErB,cAAc,CACnB,QAAQ,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,EAClD,KAAK,CAAC,EAAE,OAAO,EAAE,GAChB,cAAc,CAAC,mBAAmB,CAAC;CA+FvC"}

package/dist/src/providers/gemini.js

@@ -0,0 +1,183 @@
+import { BaseProvider } from './base.js';
+export class GeminiProvider extends BaseProvider {
+    constructor(config, modelConfig) {
+        super(config, modelConfig);
+    }
+    getName() {
+        return 'Gemini';
+    }
+    getType() {
+        return 'gemini';
+    }
+    buildHeaders() {
+        return {
+            'Content-Type': 'application/json',
+        };
+    }
+    convertMessages(messages) {
+        const contents = [];
+        let systemInstruction;
+        for (const msg of messages) {
+            if (msg.role === 'system') {
+                systemInstruction = msg.content;
+            }
+            else {
+                const role = msg.role === 'assistant' ? 'model' : 'user';
+                contents.push({
+                    role,
+                    parts: [{ text: msg.content }],
+                });
+            }
+        }
+        return { systemInstruction, contents };
+    }
+    async generate(messages, tools) {
+        const apiKey = this.getApiKey();
+        const url = `${this.config.baseUrl}/models/${this.modelConfig.model}:generateContent?key=${apiKey}`;
+        const { systemInstruction, contents } = this.convertMessages(messages);
+        const body = {
+            contents,
+            generationConfig: {
+                temperature: this.modelConfig.temperature,
+                maxOutputTokens: this.modelConfig.maxTokens,
+                topP: this.modelConfig.topP,
+                stopSequences: this.modelConfig.stop,
+            },
+        };
+        if (systemInstruction) {
+            body.systemInstruction = { parts: [{ text: systemInstruction }] };
+        }
+        const response = await this.fetchWithRetry(url, {
+            method: 'POST',
+            headers: this.buildHeaders(),
+            body: JSON.stringify(body),
+        });
+        const data = await response.json();
+        if (data.error) {
+            const err = data.error;
+            throw new Error(`Gemini API error: ${err.message || JSON.stringify(err)}`);
+        }
+        const candidates = data.candidates || [];
+        const candidate = candidates[0] || {};
+        const content = candidate.content || {};
+        const parts = content.parts || [];
+        let textContent = '';
+        const toolCalls = [];
+        for (const part of parts) {
+            if (part.text) {
+                textContent += part.text;
+            }
+            else if (part.functionCall) {
+                const fc = part.functionCall;
+                const args = fc.args ? (typeof fc.args === 'string' ? JSON.parse(fc.args) : fc.args) : {};
+                toolCalls.push({
+                    id: `call_${Date.now()}`,
+                    name: fc.name,
+                    arguments: args,
+                    timestamp: Date.now(),
+                });
+            }
+        }
+        const usageMetadata = data.usageMetadata || {};
+        return {
+            content: textContent,
+            toolCalls,
+            usage: {
+                promptTokens: usageMetadata.promptTokenCount || 0,
+                completionTokens: usageMetadata.candidatesTokenCount || 0,
+                totalTokens: usageMetadata.totalTokenCount || 0,
+            },
+            finishReason: candidate.finishReason || 'STOP',
+        };
+    }
+    async *generateStream(messages, tools) {
+        const apiKey = this.getApiKey();
+        const url = `${this.config.baseUrl}/models/${this.modelConfig.model}:streamGenerateContent?key=${apiKey}&alt=sse`;
+        const { systemInstruction, contents } = this.convertMessages(messages);
+        const body = {
+            contents,
+            generationConfig: {
+                temperature: this.modelConfig.temperature,
+                maxOutputTokens: this.modelConfig.maxTokens,
+                topP: this.modelConfig.topP,
+                stopSequences: this.modelConfig.stop,
+            },
+        };
+        if (systemInstruction) {
+            body.systemInstruction = { parts: [{ text: systemInstruction }] };
+        }
+        const response = await this.fetchWithRetry(url, {
+            method: 'POST',
+            headers: this.buildHeaders(),
+            body: JSON.stringify(body),
+        });
+        const reader = response.body?.getReader();
+        if (!reader)
+            throw new Error('No response body stream available');
+        const decoder = new TextDecoder();
+        let buffer = '';
+        try {
+            while (true) {
+                const { done, value } = await reader.read();
+                if (done)
+                    break;
+                buffer += decoder.decode(value, { stream: true });
+                const lines = buffer.split('\n');
+                buffer = lines.pop() || '';
+                for (const line of lines) {
+                    const trimmed = line.trim();
+                    if (!trimmed || !trimmed.startsWith('data: '))
+                        continue;
+                    const dataStr = trimmed.slice(6);
+                    try {
+                        const data = JSON.parse(dataStr);
+                        const candidates = data.candidates || [];
+                        const candidate = candidates[0] || {};
+                        const content = candidate.content || {};
+                        const parts = content.parts || [];
+                        for (const part of parts) {
+                            if (part.text) {
+                                yield { type: 'text', content: part.text };
+                            }
+                            else if (part.functionCall) {
+                                const fc = part.functionCall;
+                                const args = fc.args ? (typeof fc.args === 'string' ? JSON.parse(fc.args) : fc.args) : {};
+                                yield {
+                                    type: 'tool_call',
+                                    toolCall: {
+                                        id: `call_${Date.now()}`,
+                                        name: fc.name,
+                                        arguments: args,
+                                        timestamp: Date.now(),
+                                    },
+                                };
+                            }
+                        }
+                        if (candidate.finishReason) {
+                            const usageMetadata = data.usageMetadata;
+                            if (usageMetadata) {
+                                yield {
+                                    type: 'done',
+                                    usage: {
+                                        promptTokens: usageMetadata.promptTokenCount || 0,
+                                        completionTokens: usageMetadata.candidatesTokenCount || 0,
+                                        totalTokens: usageMetadata.totalTokenCount || 0,
+                                    },
+                                };
+                            }
+                            else {
+                                yield { type: 'done' };
+                            }
+                        }
+                    }
+                    catch {
+                    }
+                }
+            }
+        }
+        finally {
+            reader.releaseLock();
+        }
+    }
+}
+//# sourceMappingURL=gemini.js.map