@anonx3247/universal-agent-harness 0.1.0

package/dist/lib/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/lib/utils.ts"],"names":[],"mappings":"AAAA,wBAAgB,WAAW,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,GAAG,IAAI,GAAG,SAAS,CAAC,EAAE,GAAG,CAAC,EAAE,CAEjE;AAED,wBAAgB,MAAM,IAAI,MAAM,CAE/B;AAED,wBAAgB,SAAS,CAAC,CAAC,EACzB,GAAG,EAAE,OAAO,EACZ,EAAE,EAAE,CAAC,CAAC,EAAE,OAAO,KAAK,CAAC,IAAI,CAAC,GACzB,GAAG,IAAI,CAAC,EAAE,CAEZ;AAED,wBAAgB,QAAQ,CAAC,GAAG,EAAE,OAAO,GAAG,GAAG,IAAI,MAAM,CAEpD"}

package/dist/lib/utils.js

@@ -0,0 +1,13 @@
+export function removeNulls(arr) {
+    return arr.filter((v) => v !== null && v !== undefined);
+}
+export function newID6() {
+    return Math.random().toString(36).substring(2, 8);
+}
+export function isArrayOf(arr, tg) {
+    return Array.isArray(arr) && arr.every(tg);
+}
+export function isString(str) {
+    return typeof str === "string";
+}
+//# sourceMappingURL=utils.js.map

package/dist/lib/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/lib/utils.ts"],"names":[],"mappings":"AAAA,MAAM,UAAU,WAAW,CAAI,GAA6B;IAC1D,OAAO,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAU,EAAE,CAAC,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,SAAS,CAAC,CAAC;AAClE,CAAC;AAED,MAAM,UAAU,MAAM;IACpB,OAAO,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;AACpD,CAAC;AAED,MAAM,UAAU,SAAS,CACvB,GAAY,EACZ,EAA0B;IAE1B,OAAO,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;AAC7C,CAAC;AAED,MAAM,UAAU,QAAQ,CAAC,GAAY;IACnC,OAAO,OAAO,GAAG,KAAK,QAAQ,CAAC;AACjC,CAAC"}

package/dist/models/anthropic.d.ts

@@ -0,0 +1,20 @@
+import { MessageParam } from "@anthropic-ai/sdk/resources/messages";
+import { LLM, ModelConfig, Message, Tool, TokenUsage } from "./index";
+import { Result } from "@app/lib/error";
+export type AnthropicModel = "claude-opus-4-5" | "claude-sonnet-4-5" | "claude-haiku-4-5";
+export declare function isAnthropicModel(model: string): model is AnthropicModel;
+export declare class AnthropicLLM extends LLM {
+    private client;
+    private model;
+    constructor(config: ModelConfig, model?: AnthropicModel);
+    messages(messages: Message[]): MessageParam[];
+    run(messages: Message[], prompt: string, tools: Tool[]): Promise<Result<{
+        message: Message;
+        tokenUsage?: TokenUsage;
+    }>>;
+    tokens(messages: Message[], prompt: string, tools: Tool[]): Promise<Result<number>>;
+    private tokenUsage;
+    protected costPerTokenUsage(tokenUsage: TokenUsage): number;
+    maxTokens(): number;
+}
+//# sourceMappingURL=anthropic.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../src/models/anthropic.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,sCAAsC,CAAC;AACpE,OAAO,EAAE,GAAG,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AAEtE,OAAO,EAAE,MAAM,EAAW,MAAM,gBAAgB,CAAC;AAqCjD,MAAM,MAAM,cAAc,GACtB,iBAAiB,GACjB,mBAAmB,GACnB,kBAAkB,CAAC;AACvB,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,MAAM,GAAG,KAAK,IAAI,cAAc,CAIvE;AAED,qBAAa,YAAa,SAAQ,GAAG;IACnC,OAAO,CAAC,MAAM,CAAY;IAC1B,OAAO,CAAC,KAAK,CAAiB;gBAG5B,MAAM,EAAE,WAAW,EACnB,KAAK,GAAE,cAAoC;IAS7C,QAAQ,CAAC,QAAQ,EAAE,OAAO,EAAE;IA6GtB,GAAG,CACP,QAAQ,EAAE,OAAO,EAAE,EACnB,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,IAAI,EAAE,GACZ,OAAO,CAAC,MAAM,CAAC;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,UAAU,CAAC,EAAE,UAAU,CAAA;KAAE,CAAC,CAAC;IAgH3D,MAAM,CACV,QAAQ,EAAE,OAAO,EAAE,EACnB,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,IAAI,EAAE,GACZ,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAmC1B,OAAO,CAAC,UAAU;IAgBlB,SAAS,CAAC,iBAAiB,CAAC,UAAU,EAAE,UAAU,GAAG,MAAM;IAW3D,SAAS,IAAI,MAAM;CAUpB"}

package/dist/models/anthropic.js

@@ -0,0 +1,312 @@
+import { LLM } from "./index";
+import Anthropic from "@anthropic-ai/sdk";
+import { err, ok } from "@app/lib/error";
+import { assertNever } from "@app/lib/assert";
+import { removeNulls } from "@app/lib/utils";
+const DEFAULT_TIMEOUT = 600000 * 2; // 20 minutes (double the default)
+const DEFAULT_MAX_TOKENS = 8192;
+const DEFAULT_THINKING_TOKENS = 8192;
+function normalizeTokenPrices(costPerMillionInputTokens, costPerMillionOutputTokens) {
+    return {
+        baseInput: costPerMillionInputTokens / 1_000_000,
+        cache5m: (costPerMillionInputTokens * 1.25) / 1_000_000,
+        cache1h: (costPerMillionInputTokens * 2) / 1_000_000,
+        cacheHits: (costPerMillionInputTokens * 0.1) / 1_000_000,
+        output: costPerMillionOutputTokens / 1_000_000,
+    };
+}
+// https://docs.claude.com/en/docs/about-claude/pricing#model-pricing
+const TOKEN_PRICING = {
+    "claude-opus-4-5": normalizeTokenPrices(5, 25),
+    "claude-sonnet-4-5": normalizeTokenPrices(3, 15),
+    "claude-haiku-4-5": normalizeTokenPrices(1, 5),
+};
+export function isAnthropicModel(model) {
+    return ["claude-opus-4-5", "claude-sonnet-4-5", "claude-haiku-4-5"].includes(model);
+}
+export class AnthropicLLM extends LLM {
+    client;
+    model;
+    constructor(config, model = "claude-sonnet-4-5") {
+        super(config);
+        this.client = new Anthropic({
+            timeout: DEFAULT_TIMEOUT,
+        });
+        this.model = model;
+    }
+    messages(messages) {
+        const anthropicMessages = messages.map((msg) => ({
+            role: msg.role === "agent" ? "assistant" : "user",
+            content: removeNulls(msg.content.map((content) => {
+                switch (content.type) {
+                    case "text":
+                        return {
+                            type: "text",
+                            text: content.text,
+                        };
+                    case "tool_use":
+                        return {
+                            type: "tool_use",
+                            id: content.id,
+                            name: content.name,
+                            input: content.input,
+                        };
+                    case "tool_result":
+                        return {
+                            type: "tool_result",
+                            tool_use_id: content.toolUseId,
+                            content: content.content.map((content) => {
+                                switch (content.type) {
+                                    case "text":
+                                        return {
+                                            type: "text",
+                                            text: content.text,
+                                        };
+                                    case "image": {
+                                        return {
+                                            type: "image",
+                                            source: {
+                                                data: content.data,
+                                                media_type: content.mimeType,
+                                                type: "base64",
+                                            },
+                                        };
+                                    }
+                                    case "audio":
+                                        return {
+                                            type: "text",
+                                            text: "(unsupported audio content)",
+                                        };
+                                    case "resource":
+                                        return {
+                                            type: "text",
+                                            text: JSON.stringify(content, null, 2),
+                                        };
+                                    case "resource_link":
+                                        return {
+                                            type: "text",
+                                            text: JSON.stringify(content, null, 2),
+                                        };
+                                    default:
+                                        assertNever(content);
+                                }
+                            }),
+                            is_error: content.isError,
+                        };
+                    case "thinking": {
+                        if (content.provider?.anthropic) {
+                            switch (content.provider.anthropic.type) {
+                                case "thinking": {
+                                    return {
+                                        type: "thinking",
+                                        thinking: content.thinking,
+                                        signature: content.provider.anthropic.signature,
+                                    };
+                                }
+                                case "redacted_thinking": {
+                                    return {
+                                        type: "redacted_thinking",
+                                        data: content.provider.anthropic.data,
+                                    };
+                                }
+                                default:
+                                    return null;
+                            }
+                        }
+                        return null;
+                    }
+                    default:
+                        assertNever(content);
+                }
+            })),
+        }));
+        for (let i = anthropicMessages.length - 1; i >= 0; i--) {
+            if (anthropicMessages[i].role === "user") {
+                let found = false;
+                for (let j = anthropicMessages[i].content.length - 1; j >= 0; j--) {
+                    const c = anthropicMessages[i].content[j];
+                    if (typeof c !== "string" && c.type === "text") {
+                        c.cache_control = { type: "ephemeral" };
+                        found = true;
+                        break;
+                    }
+                }
+                if (found) {
+                    break;
+                }
+            }
+        }
+        return anthropicMessages;
+    }
+    async run(messages, prompt, tools) {
+        try {
+            const message = await this.client.beta.messages.create({
+                model: this.model,
+                max_tokens: this.config.maxTokens ??
+                    (() => {
+                        // thinking: true = enabled, false/undefined = disabled
+                        if (this.config.thinking === false) {
+                            return DEFAULT_MAX_TOKENS;
+                        }
+                        // Default or explicitly enabled: use thinking tokens
+                        return DEFAULT_THINKING_TOKENS + DEFAULT_MAX_TOKENS;
+                    })(),
+                messages: this.messages(messages),
+                system: [
+                    {
+                        type: "text",
+                        text: prompt,
+                        cache_control: {
+                            type: "ephemeral",
+                        },
+                    },
+                ],
+                thinking: (() => {
+                    // thinking: true = enabled, false/undefined = disabled
+                    if (this.config.thinking === false) {
+                        return { type: "disabled" };
+                    }
+                    // Default or explicitly enabled
+                    return {
+                        type: "enabled",
+                        budget_tokens: DEFAULT_THINKING_TOKENS,
+                    };
+                })(),
+                tools: tools.map((tool) => ({
+                    name: tool.name,
+                    description: tool.description,
+                    input_schema: tool.inputSchema,
+                })),
+                tool_choice: { type: "auto" },
+                betas: ["interleaved-thinking-2025-05-14"],
+            });
+            const tokenUsage = this.tokenUsage(message.usage);
+            // console.log(message.usage);
+            return ok({
+                message: {
+                    role: message.role === "assistant" ? "agent" : "user",
+                    content: removeNulls(message.content.map((c) => {
+                        switch (c.type) {
+                            case "text":
+                                return {
+                                    type: "text",
+                                    text: c.text,
+                                    provider: null,
+                                };
+                            case "tool_use":
+                                return {
+                                    type: "tool_use",
+                                    id: c.id,
+                                    name: c.name,
+                                    input: c.input,
+                                    provider: null,
+                                };
+                            case "thinking": {
+                                return {
+                                    type: "thinking",
+                                    thinking: c.thinking,
+                                    provider: {
+                                        anthropic: {
+                                            type: c.type,
+                                            signature: c.signature,
+                                        },
+                                    },
+                                };
+                            }
+                            case "redacted_thinking": {
+                                return {
+                                    type: "thinking",
+                                    thinking: "<redacted>",
+                                    provider: {
+                                        anthropic: {
+                                            type: c.type,
+                                            data: c.data,
+                                        },
+                                    },
+                                };
+                            }
+                            case "server_tool_use":
+                            case "mcp_tool_use":
+                            case "mcp_tool_result":
+                            case "code_execution_tool_result":
+                            case "container_upload":
+                            case "web_search_tool_result": {
+                                return null;
+                            }
+                            default:
+                                assertNever(c);
+                        }
+                    })),
+                },
+                tokenUsage, // also inlcude cached or input tokens ?
+            });
+        }
+        catch (error) {
+            return err("model_error", "Failed to run model", error);
+        }
+    }
+    async tokens(messages, prompt, tools) {
+        try {
+            const response = await this.client.messages.countTokens({
+                model: this.model,
+                messages: this.messages(messages),
+                system: prompt,
+                thinking: (() => {
+                    // thinking: true = enabled, false/undefined = disabled
+                    if (this.config.thinking === false) {
+                        return { type: "disabled" };
+                    }
+                    // Default or explicitly enabled
+                    return {
+                        type: "enabled",
+                        budget_tokens: DEFAULT_THINKING_TOKENS,
+                    };
+                })(),
+                tools: tools.map((tool) => ({
+                    name: tool.name,
+                    description: tool.description,
+                    input_schema: tool.inputSchema,
+                })),
+                tool_choice: { type: "auto" },
+            });
+            return ok(response.input_tokens);
+        }
+        catch (error) {
+            return err("model_error", "Failed to count tokens", error);
+        }
+    }
+    tokenUsage(usage) {
+        const input = usage.input_tokens +
+            (usage.cache_read_input_tokens ?? 0) +
+            (usage.cache_creation?.ephemeral_1h_input_tokens ?? 0) +
+            (usage.cache_creation?.ephemeral_5m_input_tokens ?? 0);
+        return {
+            total: usage.output_tokens + input,
+            input,
+            output: usage.output_tokens,
+            cached: usage.cache_read_input_tokens ?? 0,
+            thinking: 0, // Anthropic doesn't give thinking token usage
+        };
+    }
+    costPerTokenUsage(tokenUsage) {
+        const pricing = TOKEN_PRICING[this.model];
+        // For Anthropic, we use a conservative estimate:
+        // baseInput for non-cached tokens, cacheHits for cached tokens
+        const nonCachedInput = tokenUsage.input - tokenUsage.cached;
+        let c = nonCachedInput * pricing.baseInput;
+        c += tokenUsage.cached * pricing.cacheHits;
+        c += tokenUsage.output * pricing.output;
+        return c;
+    }
+    maxTokens() {
+        switch (this.model) {
+            case "claude-opus-4-5":
+            case "claude-sonnet-4-5":
+            case "claude-haiku-4-5":
+                return 200000 - 64000;
+            default:
+                assertNever(this.model);
+        }
+    }
+}
+//# sourceMappingURL=anthropic.js.map

package/dist/models/anthropic.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../src/models/anthropic.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,GAAG,EAA0C,MAAM,SAAS,CAAC;AACtE,OAAO,SAAS,MAAM,mBAAmB,CAAC;AAC1C,OAAO,EAAU,GAAG,EAAE,EAAE,EAAE,MAAM,gBAAgB,CAAC;AACjD,OAAO,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAG7C,MAAM,eAAe,GAAG,MAAM,GAAG,CAAC,CAAC,CAAC,kCAAkC;AACtE,MAAM,kBAAkB,GAAG,IAAI,CAAC;AAChC,MAAM,uBAAuB,GAAG,IAAI,CAAC;AAUrC,SAAS,oBAAoB,CAC3B,yBAAiC,EACjC,0BAAkC;IAElC,OAAO;QACL,SAAS,EAAE,yBAAyB,GAAG,SAAS;QAChD,OAAO,EAAE,CAAC,yBAAyB,GAAG,IAAI,CAAC,GAAG,SAAS;QACvD,OAAO,EAAE,CAAC,yBAAyB,GAAG,CAAC,CAAC,GAAG,SAAS;QACpD,SAAS,EAAE,CAAC,yBAAyB,GAAG,GAAG,CAAC,GAAG,SAAS;QACxD,MAAM,EAAE,0BAA0B,GAAG,SAAS;KAC/C,CAAC;AACJ,CAAC;AAED,qEAAqE;AACrE,MAAM,aAAa,GAAiD;IAClE,iBAAiB,EAAE,oBAAoB,CAAC,CAAC,EAAE,EAAE,CAAC;IAC9C,mBAAmB,EAAE,oBAAoB,CAAC,CAAC,EAAE,EAAE,CAAC;IAChD,kBAAkB,EAAE,oBAAoB,CAAC,CAAC,EAAE,CAAC,CAAC;CAC/C,CAAC;AAMF,MAAM,UAAU,gBAAgB,CAAC,KAAa;IAC5C,OAAO,CAAC,iBAAiB,EAAE,mBAAmB,EAAE,kBAAkB,CAAC,CAAC,QAAQ,CAC1E,KAAK,CACN,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,YAAa,SAAQ,GAAG;IAC3B,MAAM,CAAY;IAClB,KAAK,CAAiB;IAE9B,YACE,MAAmB,EACnB,QAAwB,mBAAmB;QAE3C,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC;YAC1B,OAAO,EAAE,eAAe;SACzB,CAAC,CAAC;QACH,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,QAAQ,CAAC,QAAmB;QAC1B,MAAM,iBAAiB,GAAmB,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YAC/D,IAAI,EAAE,GAAG,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM;YACjD,OAAO,EAAE,WAAW,CAClB,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;gBAC1B,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;oBACrB,KAAK,MAAM;wBACT,OAAO;4BACL,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,OAAO,CAAC,IAAI;yBACnB,CAAC;oBACJ,KAAK,UAAU;wBACb,OAAO;4BACL,IAAI,EAAE,UAAU;4BAChB,EAAE,EAAE,OAAO,CAAC,EAAE;4BACd,IAAI,EAAE,OAAO,CAAC,IAAI;4BAClB,KAAK,EAAE,OAAO,CAAC,KAAK;yBACrB,CAAC;oBACJ,KAAK,aAAa;wBAChB,OAAO;4BACL,IAAI,EAAE,aAAa;4BACnB,WAAW,EAAE,OAAO,CAAC,SAAS;4BAC9B,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;gCACvC,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;oCACrB,KAAK,MAAM;wCACT,OAAO;4CACL,IAAI,EAAE,MAAM;4CACZ,IAAI,EAAE,OAAO,CAAC,IAAI;yCACnB,CAAC;oCACJ,KAAK,OAAO,CAAC,CAAC,CAAC;wCACb,OAAO;4CACL,IAAI,EAAE,OAAO;4CACb,MAAM,EAAE;gDACN,IAAI,EAAE,OAAO,CAAC,IAAI;gDAClB,UAAU,EAAE,OAAO,CAAC,QAAe;gDACnC,IAAI,EAAE,QAAQ;6CACf;yCACF,CAAC;oCACJ,CAAC;oCACD,KAAK,OAAO;wCACV,OAAO;4CACL,IAAI,EAAE,MAAM;4CACZ,IAAI,EAAE,6BAA6B;yCACpC,CAAC;oCACJ,KAAK,UAAU;wCACb,OAAO;4CACL,IAAI,EAAE,MAAM;4CACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;yCACvC,CAAC;oCACJ,KAAK,eAAe;wCAClB,OAAO;4CACL,IAAI,EAAE,MAAM;4CACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;yCACvC,CAAC;oCACJ;wCACE,WAAW,CAAC,OAAO,CAAC,CAAC;gCACzB,CAAC;4BACH,CAAC,CAAC;4BACF,QAAQ,EAAE,OAAO,CAAC,OAAO;yBAC1B,CAAC;oBACJ,KAAK,UAAU,CAAC,CAAC,CAAC;wBAChB,IAAI,OAAO,CAAC,QAAQ,EAAE,SAAS,EAAE,CAAC;4BAChC,QAAQ,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;gCACxC,KAAK,UAAU,CAAC,CAAC,CAAC;oCAChB,OAAO;wCACL,IAAI,EAAE,UAAU;wCAChB,QAAQ,EAAE,OAAO,CAAC,QAAQ;wCAC1B,SAAS,EAAE,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,SAAS;qCAChD,CAAC;gCACJ,CAAC;gCACD,KAAK,mBAAmB,CAAC,CAAC,CAAC;oCACzB,OAAO;wCACL,IAAI,EAAE,mBAAmB;wCACzB,IAAI,EAAE,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI;qCACtC,CAAC;gCACJ,CAAC;gCACD;oCACE,OAAO,IAAI,CAAC;4BAChB,CAAC;wBACH,CAAC;wBACD,OAAO,IAAI,CAAC;oBACd,CAAC;oBACD;wBACE,WAAW,CAAC,OAAO,CAAC,CAAC;gBACzB,CAAC;YACH,CAAC,CAAC,CACH;SACF,CAAC,CAAC,CAAC;QAEJ,KAAK,IAAI,CAAC,GAAG,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YACvD,IAAI,iBAAiB,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACzC,IAAI,KAAK,GAAG,KAAK,CAAC;gBAClB,KAAK,IAAI,CAAC,GAAG,iBAAiB,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;oBAClE,MAAM,CAAC,GAAG,iBAAiB,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;oBAC1C,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;wBAC/C,CAAC,CAAC,aAAa,GAAG,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;wBACxC,KAAK,GAAG,IAAI,CAAC;wBACb,MAAM;oBACR,CAAC;gBACH,CAAC;gBACD,IAAI,KAAK,EAAE,CAAC;oBACV,MAAM;gBACR,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,GAAG,CACP,QAAmB,EACnB,MAAc,EACd,KAAa;QAEb,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;gBACrD,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,UAAU,EACR,IAAI,CAAC,MAAM,CAAC,SAAS;oBACrB,CAAC,GAAG,EAAE;wBACJ,uDAAuD;wBACvD,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,KAAK,KAAK,EAAE,CAAC;4BACnC,OAAO,kBAAkB,CAAC;wBAC5B,CAAC;wBACD,qDAAqD;wBACrD,OAAO,uBAAuB,GAAG,kBAAkB,CAAC;oBACtD,CAAC,CAAC,EAAE;gBACN,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;gBACjC,MAAM,EAAE;oBACN;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,MAAM;wBACZ,aAAa,EAAE;4BACb,IAAI,EAAE,WAAW;yBAClB;qBACF;iBACF;gBACD,QAAQ,EAAE,CAAC,GAAG,EAAE;oBACd,uDAAuD;oBACvD,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,KAAK,KAAK,EAAE,CAAC;wBACnC,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;oBAC9B,CAAC;oBACD,gCAAgC;oBAChC,OAAO;wBACL,IAAI,EAAE,SAAS;wBACf,aAAa,EAAE,uBAAuB;qBACvC,CAAC;gBACJ,CAAC,CAAC,EAAE;gBACJ,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;oBAC1B,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,WAAW,EAAE,IAAI,CAAC,WAAW;oBAC7B,YAAY,EAAE,IAAI,CAAC,WAAkB;iBACtC,CAAC,CAAC;gBACH,WAAW,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;gBAC7B,KAAK,EAAE,CAAC,iCAAiC,CAAC;aAC3C,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAClD,8BAA8B;YAE9B,OAAO,EAAE,CAAC;gBACR,OAAO,EAAE;oBACP,IAAI,EAAE,OAAO,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM;oBACrD,OAAO,EAAE,WAAW,CAClB,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;wBACxB,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;4BACf,KAAK,MAAM;gCACT,OAAO;oCACL,IAAI,EAAE,MAAM;oCACZ,IAAI,EAAE,CAAC,CAAC,IAAI;oCACZ,QAAQ,EAAE,IAAI;iCACf,CAAC;4BACJ,KAAK,UAAU;gCACb,OAAO;oCACL,IAAI,EAAE,UAAU;oCAChB,EAAE,EAAE,CAAC,CAAC,EAAE;oCACR,IAAI,EAAE,CAAC,CAAC,IAAI;oCACZ,KAAK,EAAE,CAAC,CAAC,KAAK;oCACd,QAAQ,EAAE,IAAI;iCACf,CAAC;4BACJ,KAAK,UAAU,CAAC,CAAC,CAAC;gCAChB,OAAO;oCACL,IAAI,EAAE,UAAU;oCAChB,QAAQ,EAAE,CAAC,CAAC,QAAQ;oCACpB,QAAQ,EAAE;wCACR,SAAS,EAAE;4CACT,IAAI,EAAE,CAAC,CAAC,IAAI;4CACZ,SAAS,EAAE,CAAC,CAAC,SAAS;yCACvB;qCACF;iCACF,CAAC;4BACJ,CAAC;4BACD,KAAK,mBAAmB,CAAC,CAAC,CAAC;gCACzB,OAAO;oCACL,IAAI,EAAE,UAAU;oCAChB,QAAQ,EAAE,YAAY;oCACtB,QAAQ,EAAE;wCACR,SAAS,EAAE;4CACT,IAAI,EAAE,CAAC,CAAC,IAAI;4CACZ,IAAI,EAAE,CAAC,CAAC,IAAI;yCACb;qCACF;iCACF,CAAC;4BACJ,CAAC;4BACD,KAAK,iBAAiB,CAAC;4BACvB,KAAK,cAAc,CAAC;4BACpB,KAAK,iBAAiB,CAAC;4BACvB,KAAK,4BAA4B,CAAC;4BAClC,KAAK,kBAAkB,CAAC;4BACxB,KAAK,wBAAwB,CAAC,CAAC,CAAC;gCAC9B,OAAO,IAAI,CAAC;4BACd,CAAC;4BACD;gCACE,WAAW,CAAC,CAAC,CAAC,CAAC;wBACnB,CAAC;oBACH,CAAC,CAAC,CACH;iBACF;gBACD,UAAU,EAAE,wCAAwC;aACrD,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,GAAG,CAAC,aAAa,EAAE,qBAAqB,EAAE,KAAK,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CACV,QAAmB,EACnB,MAAc,EACd,KAAa;QAEb,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC;gBACtD,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;gBACjC,MAAM,EAAE,MAAM;gBACd,QAAQ,EAAE,CAAC,GAAG,EAAE;oBACd,uDAAuD;oBACvD,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,KAAK,KAAK,EAAE,CAAC;wBACnC,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;oBAC9B,CAAC;oBACD,gCAAgC;oBAChC,OAAO;wBACL,IAAI,EAAE,SAAS;wBACf,aAAa,EAAE,uBAAuB;qBACvC,CAAC;gBACJ,CAAC,CAAC,EAAE;gBACJ,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;oBAC1B,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,WAAW,EAAE,IAAI,CAAC,WAAW;oBAC7B,YAAY,EAAE,IAAI,CAAC,WAAkB;iBACtC,CAAC,CAAC;gBACH,WAAW,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;aAC9B,CAAC,CAAC;YAEH,OAAO,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;QACnC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,GAAG,CACR,aAAa,EACb,wBAAwB,EACxB,KAAK,CACN,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,UAAU,CAAC,KAAgB;QACjC,MAAM,KAAK,GACT,KAAK,CAAC,YAAY;YAClB,CAAC,KAAK,CAAC,uBAAuB,IAAI,CAAC,CAAC;YACpC,CAAC,KAAK,CAAC,cAAc,EAAE,yBAAyB,IAAI,CAAC,CAAC;YACtD,CAAC,KAAK,CAAC,cAAc,EAAE,yBAAyB,IAAI,CAAC,CAAC,CAAC;QAEzD,OAAO;YACL,KAAK,EAAE,KAAK,CAAC,aAAa,GAAG,KAAK;YAClC,KAAK;YACL,MAAM,EAAE,KAAK,CAAC,aAAa;YAC3B,MAAM,EAAE,KAAK,CAAC,uBAAuB,IAAI,CAAC;YAC1C,QAAQ,EAAE,CAAC,EAAE,8CAA8C;SAC5D,CAAC;IACJ,CAAC;IAES,iBAAiB,CAAC,UAAsB;QAChD,MAAM,OAAO,GAAG,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1C,iDAAiD;QACjD,+DAA+D;QAC/D,MAAM,cAAc,GAAG,UAAU,CAAC,KAAK,GAAG,UAAU,CAAC,MAAM,CAAC;QAC5D,IAAI,CAAC,GAAG,cAAc,GAAG,OAAO,CAAC,SAAS,CAAC;QAC3C,CAAC,IAAI,UAAU,CAAC,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC;QAC3C,CAAC,IAAI,UAAU,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QACxC,OAAO,CAAC,CAAC;IACX,CAAC;IAED,SAAS;QACP,QAAQ,IAAI,CAAC,KAAK,EAAE,CAAC;YACnB,KAAK,iBAAiB,CAAC;YACvB,KAAK,mBAAmB,CAAC;YACzB,KAAK,kBAAkB;gBACrB,OAAO,MAAM,GAAG,KAAK,CAAC;YACxB;gBACE,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC;CACF"}

package/dist/models/deepseek/index.d.ts

@@ -0,0 +1,20 @@
+import { ChatCompletionMessageParam } from "openai/resources/chat";
+import { LLM, ModelConfig, Message, Tool, TokenUsage } from "../index";
+import { Result } from "@app/lib/error";
+export type DeepseekModel = "deepseek-chat" | "deepseek-reasoner";
+export declare function isDeepseekModel(model: string): model is DeepseekModel;
+export declare class DeepseekLLM extends LLM {
+    private client;
+    private model;
+    constructor(config: ModelConfig, model?: DeepseekModel);
+    messages(prompt: string, messages: Message[]): ChatCompletionMessageParam[];
+    run(messages: Message[], prompt: string, tools: Tool[]): Promise<Result<{
+        message: Message;
+        tokenUsage?: TokenUsage;
+    }>>;
+    private tokenUsage;
+    protected costPerTokenUsage(tokenUsage: TokenUsage): number;
+    tokens(messages: Message[], prompt: string, tools: Tool[]): Promise<Result<number>>;
+    maxTokens(): number;
+}
+//# sourceMappingURL=index.d.ts.map

package/dist/models/deepseek/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/models/deepseek/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,0BAA0B,EAC3B,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EACL,GAAG,EACH,WAAW,EACX,OAAO,EACP,IAAI,EACJ,UAAU,EACX,MAAM,UAAU,CAAC;AAIlB,OAAO,EAAE,MAAM,EAAW,MAAM,gBAAgB,CAAC;AAKjD,MAAM,MAAM,aAAa,GAAG,eAAe,GAAG,mBAAmB,CAAC;AAClE,wBAAgB,eAAe,CAAC,KAAK,EAAE,MAAM,GAAG,KAAK,IAAI,aAAa,CAErE;AA0BD,qBAAa,WAAY,SAAQ,GAAG;IAClC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,KAAK,CAAgB;gBAEjB,MAAM,EAAE,WAAW,EAAE,KAAK,GAAE,aAA+B;IASvE,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE;IAiEtC,GAAG,CACP,QAAQ,EAAE,OAAO,EAAE,EACnB,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,IAAI,EAAE,GACZ,OAAO,CAAC,MAAM,CAAC;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,UAAU,CAAC,EAAE,UAAU,CAAA;KAAE,CAAC,CAAC;IA4FjE,OAAO,CAAC,UAAU;IAUlB,SAAS,CAAC,iBAAiB,CAAC,UAAU,EAAE,UAAU,GAAG,MAAM;IAUrD,MAAM,CACV,QAAQ,EAAE,OAAO,EAAE,EACnB,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,IAAI,EAAE,GACZ,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAgB1B,SAAS,IAAI,MAAM;CAUpB"}

package/dist/models/deepseek/index.js

@@ -0,0 +1,210 @@
+import { LLM, } from "../index";
+import { PreTrainedTokenizer } from "@huggingface/transformers";
+import OpenAI from "openai";
+import { err, ok } from "@app/lib/error";
+import { assertNever } from "@app/lib/assert";
+import { removeNulls } from "@app/lib/utils";
+export function isDeepseekModel(model) {
+    return ["deepseek-chat", "deepseek-reasoner"].includes(model);
+}
+function normalizeTokenPrices(costPerMillionInputTokens, costPerMillionOutputTokens, costPerMillionCacheTokens) {
+    return {
+        input: costPerMillionInputTokens / 1_000_000,
+        output: costPerMillionOutputTokens / 1_000_000,
+        cacheHits: costPerMillionCacheTokens / 1_000_000,
+    };
+}
+// https://api-docs.deepseek.com/quick_start/pricing
+const TOKEN_PRICING = {
+    "deepseek-chat": normalizeTokenPrices(0.28, 0.42, 0.028),
+    "deepseek-reasoner": normalizeTokenPrices(0.28, 0.42, 0.028),
+};
+export class DeepseekLLM extends LLM {
+    client;
+    model;
+    constructor(config, model = "deepseek-chat") {
+        super(config);
+        this.client = new OpenAI({
+            apiKey: process.env.DEEPSEEK_API_KEY,
+            baseURL: "https://api.deepseek.com/v1",
+        });
+        this.model = model;
+    }
+    messages(prompt, messages) {
+        const inputItems = [
+            { role: "system", content: prompt },
+            ...removeNulls(messages
+                .map((msg) => {
+                switch (msg.role) {
+                    case "user":
+                        return msg.content.map((c) => {
+                            switch (c.type) {
+                                case "text":
+                                    return { role: "user", content: c.text };
+                                case "tool_result":
+                                    return {
+                                        role: "tool",
+                                        name: c.toolUseName,
+                                        tool_call_id: c.toolUseId,
+                                        id: c.toolUseId,
+                                        content: JSON.stringify(c.content),
+                                    };
+                                default:
+                                    return undefined;
+                            }
+                        });
+                    case "agent":
+                        const message = {
+                            role: "assistant",
+                            content: null,
+                        };
+                        msg.content.forEach((c) => {
+                            switch (c.type) {
+                                case "text":
+                                    message.content = c.text;
+                                    break;
+                                case "thinking":
+                                    message.reasoning_content = c.thinking;
+                                    break;
+                                case "tool_use":
+                                    message.tool_calls = message.tool_calls ?? [];
+                                    message.tool_calls.push({
+                                        type: "function",
+                                        id: c.id,
+                                        function: {
+                                            name: c.name,
+                                            arguments: JSON.stringify(c.input),
+                                        },
+                                    });
+                                    break;
+                            }
+                        });
+                        if (message.tool_calls && message.tool_calls.length > 0 && !message.reasoning_content) {
+                            message.reasoning_content = "";
+                        }
+                        return [message];
+                }
+            })
+                .flat()),
+        ];
+        return inputItems;
+    }
+    async run(messages, prompt, tools) {
+        try {
+            const input = this.messages(prompt, messages);
+            const response = await this.client.chat.completions.create({
+                model: this.model,
+                messages: input,
+                tool_choice: "auto",
+                tools: tools.map((tool) => ({
+                    type: "function",
+                    function: {
+                        name: tool.name,
+                        description: tool.description,
+                        parameters: tool.inputSchema,
+                    },
+                    strict: false,
+                })),
+            }, {});
+            const message = response.choices[0].message;
+            const textContent = message.content;
+            const thinkingContent = "reasoning_content" in message
+                ? message.reasoning_content
+                : undefined;
+            const toolCalls = message.tool_calls;
+            const output = [];
+            if (textContent) {
+                output.push({
+                    type: "text",
+                    text: textContent,
+                    provider: null,
+                });
+            }
+            if (thinkingContent) {
+                output.push({
+                    type: "thinking",
+                    thinking: thinkingContent,
+                    provider: null,
+                });
+            }
+            if (toolCalls) {
+                output.push(...toolCalls
+                    .filter((t) => t.type === "function")
+                    .map((toolCall) => {
+                    return {
+                        type: "tool_use",
+                        id: toolCall.id,
+                        name: toolCall.function.name,
+                        input: JSON.parse(toolCall.function.arguments),
+                        provider: {
+                            moonshotai: {
+                                id: toolCall.id,
+                            },
+                        },
+                    };
+                }));
+            }
+            if (!textContent && !toolCalls) {
+                output.push({
+                    type: "text",
+                    text: "",
+                    provider: null,
+                });
+            }
+            const tokenUsage = response.usage
+                ? this.tokenUsage(response.usage)
+                : undefined;
+            return ok({
+                message: {
+                    role: "agent",
+                    content: output,
+                },
+                tokenUsage,
+            });
+        }
+        catch (error) {
+            return err("model_error", "Failed to run model", error);
+        }
+    }
+    tokenUsage(usage) {
+        return {
+            total: usage.total_tokens,
+            input: usage.prompt_tokens,
+            output: usage.completion_tokens,
+            cached: usage.prompt_tokens_details?.cached_tokens ?? 0,
+            thinking: usage.completion_tokens_details?.reasoning_tokens ?? 0,
+        };
+    }
+    costPerTokenUsage(tokenUsage) {
+        const pricing = TOKEN_PRICING[this.model];
+        const nonCachedInput = tokenUsage.input - tokenUsage.cached;
+        const c = nonCachedInput * pricing.input +
+            tokenUsage.output * pricing.output +
+            tokenUsage.cached * pricing.cacheHits;
+        return c;
+    }
+    async tokens(messages, prompt, tools) {
+        const str = [messages, prompt, tools]
+            .map((x) => JSON.stringify(x))
+            .reduce((acc, cur) => acc + cur);
+        try {
+            const tokenizer = await PreTrainedTokenizer.from_pretrained(__dirname);
+            const encoded = tokenizer.encode(str);
+            return ok(encoded.length);
+        }
+        catch (e) {
+            return err("model_error", "Could not tokenize", e);
+        }
+    }
+    maxTokens() {
+        switch (this.model) {
+            case "deepseek-chat":
+                return 128000;
+            case "deepseek-reasoner":
+                return 128000;
+            default:
+                assertNever(this.model);
+        }
+    }
+}
+//# sourceMappingURL=index.js.map

package/dist/models/deepseek/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/models/deepseek/index.ts"],"names":[],"mappings":"AAIA,OAAO,EACL,GAAG,GAKJ,MAAM,UAAU,CAAC;AAClB,OAAO,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAEhE,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAU,GAAG,EAAE,EAAE,EAAE,MAAM,gBAAgB,CAAC;AACjD,OAAO,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAI7C,MAAM,UAAU,eAAe,CAAC,KAAa;IAC3C,OAAO,CAAC,eAAe,EAAE,mBAAmB,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;AAChE,CAAC;AAQD,SAAS,oBAAoB,CAC3B,yBAAiC,EACjC,0BAAkC,EAClC,yBAAiC;IAEjC,OAAO;QACL,KAAK,EAAE,yBAAyB,GAAG,SAAS;QAC5C,MAAM,EAAE,0BAA0B,GAAG,SAAS;QAC9C,SAAS,EAAE,yBAAyB,GAAG,SAAS;KACjD,CAAC;AACJ,CAAC;AAED,oDAAoD;AACpD,MAAM,aAAa,GAA+C;IAChE,eAAe,EAAE,oBAAoB,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC;IACxD,mBAAmB,EAAE,oBAAoB,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC;CAC7D,CAAC;AAEF,MAAM,OAAO,WAAY,SAAQ,GAAG;IAC1B,MAAM,CAAS;IACf,KAAK,CAAgB;IAE7B,YAAY,MAAmB,EAAE,QAAuB,eAAe;QACrE,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC;YACvB,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,gBAAgB;YACpC,OAAO,EAAE,6BAA6B;SACvC,CAAC,CAAC;QACH,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,QAAQ,CAAC,MAAc,EAAE,QAAmB;QAC1C,MAAM,UAAU,GAAiC;YAC/C,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE;YACnC,GAAG,WAAW,CACZ,QAAQ;iBACL,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;gBACX,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;oBACjB,KAAK,MAAM;wBACT,OAAO,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;4BAC3B,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;gCACf,KAAK,MAAM;oCACT,OAAO,EAAE,IAAI,EAAE,MAAe,EAAE,OAAO,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;gCACpD,KAAK,aAAa;oCAChB,OAAO;wCACL,IAAI,EAAE,MAAe;wCACrB,IAAI,EAAE,CAAC,CAAC,WAAW;wCACnB,YAAY,EAAE,CAAC,CAAC,SAAS;wCACzB,EAAE,EAAE,CAAC,CAAC,SAAS;wCACf,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC;qCACnC,CAAC;gCACJ;oCACE,OAAO,SAAS,CAAC;4BACrB,CAAC;wBACH,CAAC,CAAC,CAAC;oBACL,KAAK,OAAO;wBACV,MAAM,OAAO,GAET;4BACF,IAAI,EAAE,WAAW;4BACjB,OAAO,EAAE,IAAI;yBACd,CAAC;wBACF,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;4BACxB,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;gCACf,KAAK,MAAM;oCACT,OAAO,CAAC,OAAO,GAAG,CAAC,CAAC,IAAI,CAAC;oCACzB,MAAM;gCACR,KAAK,UAAU;oCACb,OAAO,CAAC,iBAAiB,GAAG,CAAC,CAAC,QAAQ,CAAC;oCACvC,MAAM;gCACR,KAAK,UAAU;oCACb,OAAO,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,EAAE,CAAC;oCAC9C,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC;wCACtB,IAAI,EAAE,UAAmB;wCACzB,EAAE,EAAE,CAAC,CAAC,EAAE;wCACR,QAAQ,EAAE;4CACR,IAAI,EAAE,CAAC,CAAC,IAAI;4CACZ,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC;yCACnC;qCACF,CAAC,CAAC;oCACH,MAAM;4BACV,CAAC;wBACH,CAAC,CAAC,CAAC;wBACH,IAAI,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAAE,CAAC;4BACtF,OAAO,CAAC,iBAAiB,GAAG,EAAE,CAAC;wBACjC,CAAC;wBACD,OAAO,CAAC,OAAO,CAAC,CAAC;gBACrB,CAAC;YACH,CAAC,CAAC;iBACD,IAAI,EAAE,CACV;SACF,CAAC;QAEF,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,GAAG,CACP,QAAmB,EACnB,MAAc,EACd,KAAa;QAEb,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YAE9C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CACxD;gBACE,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,QAAQ,EAAE,KAAK;gBACf,WAAW,EAAE,MAAM;gBACnB,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;oBAC1B,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE;wBACR,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,WAAW,EAAE,IAAI,CAAC,WAAW;wBAC7B,UAAU,EAAE,IAAI,CAAC,WAAkB;qBACpC;oBACD,MAAM,EAAE,KAAK;iBACd,CAAC,CAAC;aACJ,EACD,EAAE,CACH,CAAC;YAEF,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YAC5C,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC;YACpC,MAAM,eAAe,GACnB,mBAAmB,IAAI,OAAO;gBAC5B,CAAC,CAAE,OAAO,CAAC,iBAA4B;gBACvC,CAAC,CAAC,SAAS,CAAC;YAChB,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC;YAErC,MAAM,MAAM,GAAG,EAAE,CAAC;YAElB,IAAI,WAAW,EAAE,CAAC;gBAChB,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,WAAW;oBACjB,QAAQ,EAAE,IAAI;iBACf,CAAC,CAAC;YACL,CAAC;YAED,IAAI,eAAe,EAAE,CAAC;gBACpB,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,UAAmB;oBACzB,QAAQ,EAAE,eAAe;oBACzB,QAAQ,EAAE,IAAI;iBACf,CAAC,CAAC;YACL,CAAC;YAED,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,CAAC,IAAI,CACT,GAAG,SAAS;qBACT,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC;qBACpC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;oBAChB,OAAO;wBACL,IAAI,EAAE,UAAmB;wBACzB,EAAE,EAAE,QAAQ,CAAC,EAAE;wBACf,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI;wBAC5B,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC;wBAC9C,QAAQ,EAAE;4BACR,UAAU,EAAE;gCACV,EAAE,EAAE,QAAQ,CAAC,EAAE;6BAChB;yBACF;qBACF,CAAC;gBACJ,CAAC,CAAC,CACL,CAAC;YACJ,CAAC;YAED,IAAI,CAAC,WAAW,IAAI,CAAC,SAAS,EAAE,CAAC;gBAC/B,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,MAAe;oBACrB,IAAI,EAAE,EAAE;oBACR,QAAQ,EAAE,IAAI;iBACf,CAAC,CAAC;YACL,CAAC;YAED,MAAM,UAAU,GAAG,QAAQ,CAAC,KAAK;gBAC/B,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC;gBACjC,CAAC,CAAC,SAAS,CAAC;YAEd,OAAO,EAAE,CAAC;gBACR,OAAO,EAAE;oBACP,IAAI,EAAE,OAAO;oBACb,OAAO,EAAE,MAAM;iBAChB;gBACD,UAAU;aACX,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,GAAG,CAAC,aAAa,EAAE,qBAAqB,EAAE,KAAK,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAEO,UAAU,CAAC,KAAsB;QACvC,OAAO;YACL,KAAK,EAAE,KAAK,CAAC,YAAY;YACzB,KAAK,EAAE,KAAK,CAAC,aAAa;YAC1B,MAAM,EAAE,KAAK,CAAC,iBAAiB;YAC/B,MAAM,EAAE,KAAK,CAAC,qBAAqB,EAAE,aAAa,IAAI,CAAC;YACvD,QAAQ,EAAE,KAAK,CAAC,yBAAyB,EAAE,gBAAgB,IAAI,CAAC;SACjE,CAAC;IACJ,CAAC;IAES,iBAAiB,CAAC,UAAsB;QAChD,MAAM,OAAO,GAAG,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1C,MAAM,cAAc,GAAG,UAAU,CAAC,KAAK,GAAG,UAAU,CAAC,MAAM,CAAC;QAC5D,MAAM,CAAC,GACL,cAAc,GAAG,OAAO,CAAC,KAAK;YAC9B,UAAU,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM;YAClC,UAAU,CAAC,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC;QACxC,OAAO,CAAC,CAAC;IACX,CAAC;IAED,KAAK,CAAC,MAAM,CACV,QAAmB,EACnB,MAAc,EACd,KAAa;QAEb,MAAM,GAAG,GAAG,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC;aAClC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;aAC7B,MAAM,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC;QAEnC,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,MAAM,mBAAmB,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;YAEvE,MAAM,OAAO,GAAG,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAEtC,OAAO,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC5B,CAAC;QAAC,OAAO,CAAM,EAAE,CAAC;YAChB,OAAO,GAAG,CAAC,aAAa,EAAE,oBAAoB,EAAE,CAAC,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAED,SAAS;QACP,QAAQ,IAAI,CAAC,KAAK,EAAE,CAAC;YACnB,KAAK,eAAe;gBAClB,OAAO,MAAM,CAAC;YAChB,KAAK,mBAAmB;gBACtB,OAAO,MAAM,CAAC;YAChB;gBACE,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC;CACF"}

package/dist/models/gemini.d.ts

@@ -0,0 +1,20 @@
+import { Content } from "@google/genai";
+import { LLM, ModelConfig, Message, Tool, TokenUsage } from "./index";
+import { Result } from "@app/lib/error";
+export type GeminiModel = "gemini-3-pro-preview" | "gemini-2.5-pro" | "gemini-2.5-flash" | "gemini-2.5-flash-lite";
+export declare function isGeminiModel(model: string): model is GeminiModel;
+export declare class GeminiLLM extends LLM {
+    private client;
+    private model;
+    constructor(config: ModelConfig, model?: GeminiModel);
+    contents(messages: Message[]): Content[];
+    run(messages: Message[], prompt: string, tools: Tool[]): Promise<Result<{
+        message: Message;
+        tokenUsage?: TokenUsage;
+    }>>;
+    private tokenUsage;
+    protected costPerTokenUsage(tokenUsage: TokenUsage): number;
+    tokens(messages: Message[], prompt: string, _tools: Tool[]): Promise<Result<number>>;
+    maxTokens(): number;
+}
+//# sourceMappingURL=gemini.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../src/models/gemini.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,OAAO,EAKR,MAAM,eAAe,CAAC;AACvB,OAAO,EACL,GAAG,EACH,WAAW,EACX,OAAO,EACP,IAAI,EAGJ,UAAU,EACX,MAAM,SAAS,CAAC;AACjB,OAAO,EAAE,MAAM,EAAW,MAAM,gBAAgB,CAAC;AAIjD,MAAM,MAAM,WAAW,GACnB,sBAAsB,GACtB,gBAAgB,GAChB,kBAAkB,GAClB,uBAAuB,CAAC;AAC5B,wBAAgB,aAAa,CAAC,KAAK,EAAE,MAAM,GAAG,KAAK,IAAI,WAAW,CAOjE;AAyBD,qBAAa,SAAU,SAAQ,GAAG;IAChC,OAAO,CAAC,MAAM,CAAc;IAC5B,OAAO,CAAC,KAAK,CAAc;gBAGzB,MAAM,EAAE,WAAW,EACnB,KAAK,GAAE,WAAqC;IAO9C,QAAQ,CAAC,QAAQ,EAAE,OAAO,EAAE;IAuDtB,GAAG,CACP,QAAQ,EAAE,OAAO,EAAE,EACnB,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,IAAI,EAAE,GACZ,OAAO,CAAC,MAAM,CAAC;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,UAAU,CAAC,EAAE,UAAU,CAAA;KAAE,CAAC,CAAC;IAqHjE,OAAO,CAAC,UAAU;IAUlB,SAAS,CAAC,iBAAiB,CAAC,UAAU,EAAE,UAAU,GAAG,MAAM;IAQrD,MAAM,CACV,QAAQ,EAAE,OAAO,EAAE,EACnB,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,IAAI,EAAE,GACb,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IA8B1B,SAAS,IAAI,MAAM;CAYpB"}