duocode 0.1.0

package/dist/providers/openrouter.js

@@ -0,0 +1,178 @@
+import OpenAI from "openai";
+function toMessages(messages) {
+    return messages.map((message) => {
+        if (message.role === "tool") {
+            return {
+                role: "tool",
+                content: message.content,
+                tool_call_id: message.toolCallId,
+            };
+        }
+        return {
+            role: message.role,
+            content: message.content,
+            name: message.name,
+        };
+    });
+}
+function toTools(tools) {
+    if (!tools || tools.length === 0) {
+        return undefined;
+    }
+    return tools.map((tool) => ({
+        type: "function",
+        function: {
+            name: tool.name,
+            description: tool.description,
+            parameters: tool.inputSchema,
+        },
+    }));
+}
+function parseToolCalls(rawToolCalls) {
+    if (!rawToolCalls) {
+        return [];
+    }
+    const calls = [];
+    for (const call of rawToolCalls) {
+        const rawArguments = call?.function?.arguments;
+        let input = {};
+        if (typeof rawArguments === "string" && rawArguments.length > 0) {
+            try {
+                input = JSON.parse(rawArguments);
+            }
+            catch {
+                input = { __raw: rawArguments };
+            }
+        }
+        calls.push({
+            id: String(call.id ?? ""),
+            name: String(call?.function?.name ?? ""),
+            input,
+        });
+    }
+    return calls;
+}
+export class OpenRouterProvider {
+    name = "openrouter";
+    client;
+    constructor(params) {
+        const apiKey = params?.apiKey ?? process.env.OPENROUTER_API_KEY;
+        this.client = new OpenAI({
+            apiKey,
+            baseURL: params?.baseURL ?? "https://openrouter.ai/api/v1",
+            defaultHeaders: {
+                ...(params?.siteUrl ? { "HTTP-Referer": params.siteUrl } : {}),
+                ...(params?.appName ? { "X-Title": params.appName } : {}),
+            },
+        });
+    }
+    supportsModel(_model) {
+        return true;
+    }
+    async generate(request) {
+        const completion = await this.client.chat.completions.create({
+            model: request.model,
+            messages: toMessages(request.messages),
+            tools: toTools(request.tools),
+            tool_choice: request.toolChoice,
+            temperature: request.temperature,
+            max_tokens: request.maxOutputTokens,
+            metadata: request.metadata,
+        });
+        const choice = completion.choices?.[0];
+        const message = choice?.message;
+        const toolCalls = parseToolCalls(message?.tool_calls);
+        return {
+            provider: this.name,
+            model: completion.model,
+            text: typeof message?.content === "string" ? message.content : "",
+            toolCalls,
+            stopReason: choice?.finish_reason ?? null,
+            usage: {
+                inputTokens: completion.usage?.prompt_tokens ?? 0,
+                outputTokens: completion.usage?.completion_tokens ?? 0,
+            },
+            raw: completion,
+        };
+    }
+    async *stream(request) {
+        const stream = await this.client.chat.completions.create({
+            model: request.model,
+            messages: toMessages(request.messages),
+            tools: toTools(request.tools),
+            tool_choice: request.toolChoice,
+            temperature: request.temperature,
+            max_tokens: request.maxOutputTokens,
+            metadata: request.metadata,
+            stream: true,
+            stream_options: { include_usage: true },
+        });
+        const textParts = [];
+        const toolCallBuffers = new Map();
+        let finalUsage = { inputTokens: 0, outputTokens: 0 };
+        let model = request.model;
+        let stopReason = null;
+        for await (const chunk of stream) {
+            model = chunk.model ?? model;
+            if (chunk.usage) {
+                finalUsage = {
+                    inputTokens: chunk.usage.prompt_tokens ?? finalUsage.inputTokens,
+                    outputTokens: chunk.usage.completion_tokens ?? finalUsage.outputTokens,
+                };
+                yield { type: "usage", usage: finalUsage };
+            }
+            const delta = chunk.choices?.[0]?.delta;
+            const finishReason = chunk.choices?.[0]?.finish_reason;
+            if (finishReason) {
+                stopReason = finishReason;
+            }
+            if (typeof delta?.content === "string" && delta.content.length > 0) {
+                textParts.push(delta.content);
+                yield { type: "text-delta", delta: delta.content };
+            }
+            const toolCalls = delta?.tool_calls;
+            if (toolCalls && toolCalls.length > 0) {
+                for (const toolCall of toolCalls) {
+                    const key = String(toolCall.id ?? toolCall.index ?? "0");
+                    const existing = toolCallBuffers.get(key) ?? { name: "", args: "" };
+                    if (typeof toolCall?.function?.name === "string") {
+                        existing.name = toolCall.function.name;
+                    }
+                    if (typeof toolCall?.function?.arguments === "string") {
+                        existing.args += toolCall.function.arguments;
+                    }
+                    toolCallBuffers.set(key, existing);
+                }
+            }
+        }
+        const parsedCalls = [...toolCallBuffers.entries()].map(([id, raw]) => {
+            let input = {};
+            if (raw.args.length > 0) {
+                try {
+                    input = JSON.parse(raw.args);
+                }
+                catch {
+                    input = { __raw: raw.args };
+                }
+            }
+            return {
+                id,
+                name: raw.name,
+                input,
+            };
+        });
+        for (const call of parsedCalls) {
+            yield { type: "tool-call", call };
+        }
+        const response = {
+            provider: this.name,
+            model,
+            text: textParts.join(""),
+            toolCalls: parsedCalls,
+            stopReason,
+            usage: finalUsage,
+        };
+        yield { type: "done", response };
+    }
+}
+//# sourceMappingURL=openrouter.js.map

package/dist/providers/openrouter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openrouter.js","sourceRoot":"","sources":["../../src/providers/openrouter.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAU5B,SAAS,UAAU,CAAC,QAAuB;IACzC,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;QAC9B,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC5B,OAAO;gBACL,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,YAAY,EAAE,OAAO,CAAC,UAAU;aACjC,CAAC;QACJ,CAAC;QAED,OAAO;YACL,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,IAAI,EAAE,OAAO,CAAC,IAAI;SACnB,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,OAAO,CAAC,KAA+B;IAC9C,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACjC,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAC1B,IAAI,EAAE,UAAU;QAChB,QAAQ,EAAE;YACR,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,IAAI,CAAC,WAAW;SAC7B;KACF,CAAC,CAAC,CAAC;AACN,CAAC;AAED,SAAS,cAAc,CAAC,YAA+B;IACrD,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,KAAK,GAAe,EAAE,CAAC;IAC7B,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;QAChC,MAAM,YAAY,GAAG,IAAI,EAAE,QAAQ,EAAE,SAAS,CAAC;QAC/C,IAAI,KAAK,GAA4B,EAAE,CAAC;QAExC,IAAI,OAAO,YAAY,KAAK,QAAQ,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChE,IAAI,CAAC;gBACH,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAA4B,CAAC;YAC9D,CAAC;YAAC,MAAM,CAAC;gBACP,KAAK,GAAG,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC;YAClC,CAAC;QACH,CAAC;QAED,KAAK,CAAC,IAAI,CAAC;YACT,EAAE,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,CAAC;YACzB,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,QAAQ,EAAE,IAAI,IAAI,EAAE,CAAC;YACxC,KAAK;SACN,CAAC,CAAC;IACL,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,OAAO,kBAAkB;IACpB,IAAI,GAAG,YAAqB,CAAC;IAErB,MAAM,CAAS;IAEhC,YAAY,MAKX;QACC,MAAM,MAAM,GAAG,MAAM,EAAE,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;QAChE,IAAI,CAAC,MAAM,GAAG,IAAI,MAAM,CAAC;YACvB,MAAM;YACN,OAAO,EAAE,MAAM,EAAE,OAAO,IAAI,8BAA8B;YAC1D,cAAc,EAAE;gBACd,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC9D,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC1D;SACF,CAAC,CAAC;IACL,CAAC;IAED,aAAa,CAAC,MAAc;QAC1B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAAwB;QACrC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;YAC3D,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,QAAQ,EAAE,UAAU,CAAC,OAAO,CAAC,QAAQ,CAAQ;YAC7C,KAAK,EAAE,OAAO,CAAC,OAAO,CAAC,KAAK,CAAQ;YACpC,WAAW,EAAE,OAAO,CAAC,UAAU;YAC/B,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,UAAU,EAAE,OAAO,CAAC,eAAe;YACnC,QAAQ,EAAE,OAAO,CAAC,QAAQ;SACpB,CAAC,CAAC;QAEV,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,OAAO,GAAG,MAAM,EAAE,OAAO,CAAC;QAChC,MAAM,SAAS,GAAG,cAAc,CAAC,OAAO,EAAE,UAA+B,CAAC,CAAC;QAE3E,OAAO;YACL,QAAQ,EAAE,IAAI,CAAC,IAAI;YACnB,KAAK,EAAE,UAAU,CAAC,KAAK;YACvB,IAAI,EAAE,OAAO,OAAO,EAAE,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YACjE,SAAS;YACT,UAAU,EAAE,MAAM,EAAE,aAAa,IAAI,IAAI;YACzC,KAAK,EAAE;gBACL,WAAW,EAAE,UAAU,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;gBACjD,YAAY,EAAE,UAAU,CAAC,KAAK,EAAE,iBAAiB,IAAI,CAAC;aACvD;YACD,GAAG,EAAE,UAAU;SAChB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,OAAwB;QACpC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;YACvD,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,QAAQ,EAAE,UAAU,CAAC,OAAO,CAAC,QAAQ,CAAQ;YAC7C,KAAK,EAAE,OAAO,CAAC,OAAO,CAAC,KAAK,CAAQ;YACpC,WAAW,EAAE,OAAO,CAAC,UAAU;YAC/B,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,UAAU,EAAE,OAAO,CAAC,eAAe;YACnC,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,MAAM,EAAE,IAAI;YACZ,cAAc,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;SACjC,CAAC,CAAC;QAEV,MAAM,SAAS,GAAa,EAAE,CAAC;QAC/B,MAAM,eAAe,GAAG,IAAI,GAAG,EAA0C,CAAC;QAC1E,IAAI,UAAU,GAAG,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC;QACrD,IAAI,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC1B,IAAI,UAAU,GAAkB,IAAI,CAAC;QAErC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAa,EAAE,CAAC;YACxC,KAAK,GAAG,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC;YAE7B,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;gBAChB,UAAU,GAAG;oBACX,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,aAAa,IAAI,UAAU,CAAC,WAAW;oBAChE,YAAY,EAAE,KAAK,CAAC,KAAK,CAAC,iBAAiB,IAAI,UAAU,CAAC,YAAY;iBACvE,CAAC;gBACF,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;YAC7C,CAAC;YAED,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC;YACxC,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,CAAC;YACvD,IAAI,YAAY,EAAE,CAAC;gBACjB,UAAU,GAAG,YAAY,CAAC;YAC5B,CAAC;YAED,IAAI,OAAO,KAAK,EAAE,OAAO,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACnE,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;gBAC9B,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC;YACrD,CAAC;YAED,MAAM,SAAS,GAAG,KAAK,EAAE,UAA+B,CAAC;YACzD,IAAI,SAAS,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACtC,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;oBACjC,MAAM,GAAG,GAAG,MAAM,CAAC,QAAQ,CAAC,EAAE,IAAI,QAAQ,CAAC,KAAK,IAAI,GAAG,CAAC,CAAC;oBACzD,MAAM,QAAQ,GAAG,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;oBAEpE,IAAI,OAAO,QAAQ,EAAE,QAAQ,EAAE,IAAI,KAAK,QAAQ,EAAE,CAAC;wBACjD,QAAQ,CAAC,IAAI,GAAG,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC;oBACzC,CAAC;oBACD,IAAI,OAAO,QAAQ,EAAE,QAAQ,EAAE,SAAS,KAAK,QAAQ,EAAE,CAAC;wBACtD,QAAQ,CAAC,IAAI,IAAI,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC;oBAC/C,CAAC;oBAED,eAAe,CAAC,GAAG,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;gBACrC,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,WAAW,GAAe,CAAC,GAAG,eAAe,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE;YAC/E,IAAI,KAAK,GAA4B,EAAE,CAAC;YACxC,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxB,IAAI,CAAC;oBACH,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAA4B,CAAC;gBAC1D,CAAC;gBAAC,MAAM,CAAC;oBACP,KAAK,GAAG,EAAE,KAAK,EAAE,GAAG,CAAC,IAAI,EAAE,CAAC;gBAC9B,CAAC;YACH,CAAC;YAED,OAAO;gBACL,EAAE;gBACF,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,KAAK;aACN,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;YAC/B,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;QACpC,CAAC;QAED,MAAM,QAAQ,GAAqB;YACjC,QAAQ,EAAE,IAAI,CAAC,IAAI;YACnB,KAAK;YACL,IAAI,EAAE,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;YACxB,SAAS,EAAE,WAAW;YACtB,UAAU;YACV,KAAK,EAAE,UAAU;SAClB,CAAC;QAEF,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC;IACnC,CAAC;CACF"}

package/dist/providers/performance_tracker.d.ts

@@ -0,0 +1,21 @@
+import type Database from "better-sqlite3";
+import { type DuoRole, type EloRecord, type EloMatchResult } from "../duo/duo_types.js";
+export declare class PerformanceTracker {
+    private readonly db;
+    private readonly kFactor;
+    constructor(db: Database.Database, kFactor?: number);
+    getElo(model: string, role: DuoRole, bucket: string): number;
+    getAggregateElo(model: string, role: DuoRole): number;
+    recordMatch(result: EloMatchResult): {
+        winnerNewElo: number;
+        loserNewElo: number;
+    };
+    leaderboard(role: DuoRole): EloRecord[];
+    recordGradeResult(opts: {
+        model: string;
+        role: DuoRole;
+        capabilityBucket: string;
+        compositeScore: number;
+    }): number;
+    getMatchCount(model: string, role: DuoRole): number;
+}

package/dist/providers/performance_tracker.js

@@ -0,0 +1,63 @@
+import { DEFAULT_ELO, DEFAULT_K_FACTOR, } from "../duo/duo_types.js";
+import { getEloRating, upsertEloRating, listEloRatingsByRole, getAggregateElo, getAggregateMatchCount, } from "../db/queries.js";
+function expectedScore(ratingA, ratingB) {
+    return 1 / (1 + Math.pow(10, (ratingB - ratingA) / 400));
+}
+export class PerformanceTracker {
+    db;
+    kFactor;
+    constructor(db, kFactor = DEFAULT_K_FACTOR) {
+        this.db = db;
+        this.kFactor = kFactor;
+    }
+    getElo(model, role, bucket) {
+        const row = getEloRating(this.db, model, role, bucket);
+        return row?.elo ?? DEFAULT_ELO;
+    }
+    getAggregateElo(model, role) {
+        return getAggregateElo(this.db, model, role);
+    }
+    recordMatch(result) {
+        const { winnerModel, loserModel, role, capabilityBucket, isDraw, } = result;
+        const winnerRow = getEloRating(this.db, winnerModel, role, capabilityBucket);
+        const loserRow = getEloRating(this.db, loserModel, role, capabilityBucket);
+        const winnerElo = winnerRow?.elo ?? DEFAULT_ELO;
+        const loserElo = loserRow?.elo ?? DEFAULT_ELO;
+        const winnerCount = winnerRow?.match_count ?? 0;
+        const loserCount = loserRow?.match_count ?? 0;
+        const expectedWin = expectedScore(winnerElo, loserElo);
+        const expectedLose = expectedScore(loserElo, winnerElo);
+        const actualWin = isDraw ? 0.5 : 1;
+        const actualLose = isDraw ? 0.5 : 0;
+        const winnerNewElo = winnerElo + this.kFactor * (actualWin - expectedWin);
+        const loserNewElo = loserElo + this.kFactor * (actualLose - expectedLose);
+        upsertEloRating(this.db, winnerModel, role, capabilityBucket, winnerNewElo, winnerCount + 1);
+        upsertEloRating(this.db, loserModel, role, capabilityBucket, loserNewElo, loserCount + 1);
+        return { winnerNewElo, loserNewElo };
+    }
+    leaderboard(role) {
+        const rows = listEloRatingsByRole(this.db, role);
+        return rows.map((row) => ({
+            model: row.model,
+            role: row.role,
+            capabilityBucket: row.capability_bucket,
+            elo: row.elo,
+            matchCount: row.match_count,
+        }));
+    }
+    recordGradeResult(opts) {
+        const { model, role, capabilityBucket, compositeScore } = opts;
+        const row = getEloRating(this.db, model, role, capabilityBucket);
+        const currentElo = row?.elo ?? DEFAULT_ELO;
+        const currentCount = row?.match_count ?? 0;
+        const actual = (compositeScore - 1) / 4;
+        const expected = expectedScore(currentElo, DEFAULT_ELO);
+        const newElo = currentElo + this.kFactor * (actual - expected);
+        upsertEloRating(this.db, model, role, capabilityBucket, newElo, currentCount + 1);
+        return newElo;
+    }
+    getMatchCount(model, role) {
+        return getAggregateMatchCount(this.db, model, role);
+    }
+}
+//# sourceMappingURL=performance_tracker.js.map

package/dist/providers/performance_tracker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"performance_tracker.js","sourceRoot":"","sources":["../../src/providers/performance_tracker.ts"],"names":[],"mappings":"AACA,OAAO,EAIL,WAAW,EACX,gBAAgB,GACjB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,YAAY,EACZ,eAAe,EACf,oBAAoB,EACpB,eAAe,EACf,sBAAsB,GACvB,MAAM,kBAAkB,CAAC;AAE1B,SAAS,aAAa,CAAC,OAAe,EAAE,OAAe;IACrD,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,OAAO,GAAG,OAAO,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC;AAC3D,CAAC;AAED,MAAM,OAAO,kBAAkB;IAIV;IAHF,OAAO,CAAS;IAEjC,YACmB,EAAqB,EACtC,UAAkB,gBAAgB;QADjB,OAAE,GAAF,EAAE,CAAmB;QAGtC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,MAAM,CAAC,KAAa,EAAE,IAAa,EAAE,MAAc;QACjD,MAAM,GAAG,GAAG,YAAY,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;QACvD,OAAO,GAAG,EAAE,GAAG,IAAI,WAAW,CAAC;IACjC,CAAC;IAED,eAAe,CAAC,KAAa,EAAE,IAAa;QAC1C,OAAO,eAAe,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;IAC/C,CAAC;IAED,WAAW,CAAC,MAAsB;QAChC,MAAM,EACJ,WAAW,EACX,UAAU,EACV,IAAI,EACJ,gBAAgB,EAChB,MAAM,GACP,GAAG,MAAM,CAAC;QAEX,MAAM,SAAS,GAAG,YAAY,CAAC,IAAI,CAAC,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,gBAAgB,CAAC,CAAC;QAC7E,MAAM,QAAQ,GAAG,YAAY,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,gBAAgB,CAAC,CAAC;QAE3E,MAAM,SAAS,GAAG,SAAS,EAAE,GAAG,IAAI,WAAW,CAAC;QAChD,MAAM,QAAQ,GAAG,QAAQ,EAAE,GAAG,IAAI,WAAW,CAAC;QAC9C,MAAM,WAAW,GAAG,SAAS,EAAE,WAAW,IAAI,CAAC,CAAC;QAChD,MAAM,UAAU,GAAG,QAAQ,EAAE,WAAW,IAAI,CAAC,CAAC;QAE9C,MAAM,WAAW,GAAG,aAAa,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QACvD,MAAM,YAAY,GAAG,aAAa,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;QAExD,MAAM,SAAS,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,UAAU,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAEpC,MAAM,YAAY,GAAG,SAAS,GAAG,IAAI,CAAC,OAAO,GAAG,CAAC,SAAS,GAAG,WAAW,CAAC,CAAC;QAC1E,MAAM,WAAW,GAAG,QAAQ,GAAG,IAAI,CAAC,OAAO,GAAG,CAAC,UAAU,GAAG,YAAY,CAAC,CAAC;QAE1E,eAAe,CACb,IAAI,CAAC,EAAE,EACP,WAAW,EACX,IAAI,EACJ,gBAAgB,EAChB,YAAY,EACZ,WAAW,GAAG,CAAC,CAChB,CAAC;QAEF,eAAe,CACb,IAAI,CAAC,EAAE,EACP,UAAU,EACV,IAAI,EACJ,gBAAgB,EAChB,WAAW,EACX,UAAU,GAAG,CAAC,CACf,CAAC;QAEF,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC;IACvC,CAAC;IAED,WAAW,CAAC,IAAa;QACvB,MAAM,IAAI,GAAG,oBAAoB,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;QACjD,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACxB,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,IAAI,EAAE,GAAG,CAAC,IAAe;YACzB,gBAAgB,EAAE,GAAG,CAAC,iBAAiB;YACvC,GAAG,EAAE,GAAG,CAAC,GAAG;YACZ,UAAU,EAAE,GAAG,CAAC,WAAW;SAC5B,CAAC,CAAC,CAAC;IACN,CAAC;IAED,iBAAiB,CAAC,IAKjB;QACC,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,gBAAgB,EAAE,cAAc,EAAE,GAAG,IAAI,CAAC;QAE/D,MAAM,GAAG,GAAG,YAAY,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,gBAAgB,CAAC,CAAC;QACjE,MAAM,UAAU,GAAG,GAAG,EAAE,GAAG,IAAI,WAAW,CAAC;QAC3C,MAAM,YAAY,GAAG,GAAG,EAAE,WAAW,IAAI,CAAC,CAAC;QAE3C,MAAM,MAAM,GAAG,CAAC,cAAc,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;QACxC,MAAM,QAAQ,GAAG,aAAa,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QACxD,MAAM,MAAM,GAAG,UAAU,GAAG,IAAI,CAAC,OAAO,GAAG,CAAC,MAAM,GAAG,QAAQ,CAAC,CAAC;QAE/D,eAAe,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,gBAAgB,EAAE,MAAM,EAAE,YAAY,GAAG,CAAC,CAAC,CAAC;QAElF,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,aAAa,CAAC,KAAa,EAAE,IAAa;QACxC,OAAO,sBAAsB,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;IACtD,CAAC;CAEF"}

package/dist/providers/registry_loader.d.ts

@@ -0,0 +1,6 @@
+import type { ModelProfile } from "./base.js";
+import type { ModelPricing } from "./cost_tracker.js";
+export declare function loadModelProfilesFromRegistry(registryPath: string): {
+    profiles: ModelProfile[];
+    pricing: Record<string, ModelPricing>;
+};

package/dist/providers/registry_loader.js

@@ -0,0 +1,54 @@
+import fs from "node:fs";
+import YAML from "yaml";
+function normalizeCapabilities(caps, tags) {
+    const capabilities = new Set();
+    for (const item of caps ?? []) {
+        if (typeof item === "string") {
+            capabilities.add(item);
+        }
+        else if (item && typeof item.capability === "string") {
+            capabilities.add(item.capability);
+        }
+    }
+    if ((tags ?? []).some((tag) => tag.includes("coding"))) {
+        capabilities.add("code_generation");
+    }
+    return [...capabilities];
+}
+export function loadModelProfilesFromRegistry(registryPath) {
+    const raw = fs.readFileSync(registryPath, "utf8");
+    const doc = (YAML.parse(raw) ?? {});
+    const profiles = [];
+    const pricing = {};
+    for (const provider of doc.providers ?? []) {
+        for (const model of provider.models ?? []) {
+            if (model.enabled === false) {
+                continue;
+            }
+            const capabilities = normalizeCapabilities(model.capabilities, model.tags);
+            profiles.push({
+                model: model.modelId,
+                provider: provider.name,
+                capabilities,
+                supportsStreaming: true,
+                supportsTools: true,
+                contextWindow: model.contextWindow,
+                maxOutputTokens: model.maxOutputTokens,
+                avgLatencyMs: model.performance?.avgTimeToFirstTokenMs,
+                defaultScore: 0.8,
+            });
+            const inPrice = model.pricing?.inputPer1kTokens;
+            const outPrice = model.pricing?.outputPer1kTokens;
+            if (typeof inPrice === "number" && typeof outPrice === "number") {
+                pricing[model.modelId] = {
+                    inputPer1kUsd: inPrice,
+                    outputPer1kUsd: outPrice,
+                    cacheReadInputPer1kUsd: model.pricing?.cacheReadInputPer1kTokens,
+                    cacheWriteInputPer1kUsd: model.pricing?.cacheWriteInputPer1kTokens,
+                };
+            }
+        }
+    }
+    return { profiles, pricing };
+}
+//# sourceMappingURL=registry_loader.js.map

package/dist/providers/registry_loader.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry_loader.js","sourceRoot":"","sources":["../../src/providers/registry_loader.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,MAAM,CAAC;AA6BxB,SAAS,qBAAqB,CAC5B,IAAwD,EACxD,IAA0B;IAE1B,MAAM,YAAY,GAAG,IAAI,GAAG,EAAU,CAAC;IAEvC,KAAK,MAAM,IAAI,IAAI,IAAI,IAAI,EAAE,EAAE,CAAC;QAC9B,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC7B,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACzB,CAAC;aAAM,IAAI,IAAI,IAAI,OAAO,IAAI,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;YACvD,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACpC,CAAC;IACH,CAAC;IAED,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC;QACvD,YAAY,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;IACtC,CAAC;IAED,OAAO,CAAC,GAAG,YAAY,CAAC,CAAC;AAC3B,CAAC;AAED,MAAM,UAAU,6BAA6B,CAAC,YAAoB;IAIhE,MAAM,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;IAClD,MAAM,GAAG,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,CAAqB,CAAC;IAExD,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,MAAM,OAAO,GAAiC,EAAE,CAAC;IAEjD,KAAK,MAAM,QAAQ,IAAI,GAAG,CAAC,SAAS,IAAI,EAAE,EAAE,CAAC;QAC3C,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,MAAM,IAAI,EAAE,EAAE,CAAC;YAC1C,IAAI,KAAK,CAAC,OAAO,KAAK,KAAK,EAAE,CAAC;gBAC5B,SAAS;YACX,CAAC;YAED,MAAM,YAAY,GAAG,qBAAqB,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;YAE3E,QAAQ,CAAC,IAAI,CAAC;gBACZ,KAAK,EAAE,KAAK,CAAC,OAAO;gBACpB,QAAQ,EAAE,QAAQ,CAAC,IAAI;gBACvB,YAAY;gBACZ,iBAAiB,EAAE,IAAI;gBACvB,aAAa,EAAE,IAAI;gBACnB,aAAa,EAAE,KAAK,CAAC,aAAa;gBAClC,eAAe,EAAE,KAAK,CAAC,eAAe;gBACtC,YAAY,EAAE,KAAK,CAAC,WAAW,EAAE,qBAAqB;gBACtD,YAAY,EAAE,GAAG;aAClB,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,EAAE,gBAAgB,CAAC;YAChD,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,EAAE,iBAAiB,CAAC;YAElD,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBAChE,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG;oBACvB,aAAa,EAAE,OAAO;oBACtB,cAAc,EAAE,QAAQ;oBACxB,sBAAsB,EAAE,KAAK,CAAC,OAAO,EAAE,yBAAyB;oBAChE,uBAAuB,EAAE,KAAK,CAAC,OAAO,EAAE,0BAA0B;iBACnE,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;AAC/B,CAAC"}

package/dist/providers/retry.d.ts

@@ -0,0 +1,66 @@
+/**
+ * Retry with exponential backoff, resilient provider wrapper,
+ * and failover chain for LLM calls.
+ */
+/// <reference types="node" resolution-mode="require"/>
+import type { GenerateRequest, ModelProvider, ProviderName, ProviderResponse, ProviderStreamEvent, StreamRequest } from "./base.js";
+export interface RetryConfig {
+    maxAttempts?: number;
+    baseDelayMs?: number;
+    backoffMultiplier?: number;
+    maxDelayMs?: number;
+    timeoutMs?: number;
+    retryableStatuses?: number[];
+    retryablePatterns?: RegExp[];
+}
+export interface ClassifiedError {
+    retryable: boolean;
+    rateLimited: boolean;
+    statusCode: number | null;
+    retryAfterMs: number | null;
+    original: Error;
+}
+export declare function classifyError(error: unknown, config?: RetryConfig): ClassifiedError;
+export interface RetryAttempt {
+    attempt: number;
+    success: boolean;
+    durationMs: number;
+    error?: ClassifiedError;
+}
+export type RetryListener = (attempt: RetryAttempt) => void;
+export declare class ResilientProvider implements ModelProvider {
+    readonly name: ProviderName;
+    private readonly inner;
+    private readonly config;
+    private readonly listener?;
+    constructor(inner: ModelProvider, config?: RetryConfig, listener?: RetryListener);
+    supportsModel(model: string): boolean;
+    generate(request: GenerateRequest): Promise<ProviderResponse>;
+    stream(request: StreamRequest): AsyncIterable<ProviderStreamEvent>;
+}
+export interface FailoverEntry {
+    provider: ModelProvider;
+    config?: RetryConfig;
+}
+export type FailoverListener = (from: ProviderName, to: ProviderName) => void;
+export declare class FailoverChain implements ModelProvider {
+    readonly name: ProviderName;
+    private readonly entries;
+    private readonly onFailover?;
+    constructor(entries: FailoverEntry[], onFailover?: FailoverListener);
+    supportsModel(model: string): boolean;
+    generate(request: GenerateRequest): Promise<ProviderResponse>;
+    stream(request: StreamRequest): AsyncIterable<ProviderStreamEvent>;
+}
+export interface RetryOptions {
+    maxRetries?: number;
+    initialDelayMs?: number;
+    maxDelayMs?: number;
+    signal?: AbortSignal;
+}
+export declare function isTransientError(error: unknown): boolean;
+export declare function withRetry<T>(fn: (signal?: AbortSignal) => Promise<T>, opts?: RetryOptions): Promise<T>;
+export declare function timeoutSignal(ms: number): {
+    signal: AbortSignal;
+    cleanup: () => void;
+};

package/dist/providers/retry.js

@@ -0,0 +1,203 @@
+/**
+ * Retry with exponential backoff, resilient provider wrapper,
+ * and failover chain for LLM calls.
+ */
+const DEFAULT_CONFIG = {
+    maxAttempts: 3,
+    baseDelayMs: 500,
+    backoffMultiplier: 2,
+    maxDelayMs: 30000,
+    timeoutMs: 60000,
+    retryableStatuses: [429, 500, 502, 503, 529],
+    retryablePatterns: [/rate.?limit/i, /timeout/i, /ECONNRESET/, /ETIMEDOUT/, /ECONNREFUSED/, /fetch failed/i, /overloaded/i],
+};
+function mergeConfig(partial) {
+    return { ...DEFAULT_CONFIG, ...partial };
+}
+export function classifyError(error, config = DEFAULT_CONFIG) {
+    const err = error instanceof Error ? error : new Error(String(error));
+    const merged = mergeConfig(config);
+    const msg = err.message;
+    // Extract status code from error object
+    const statusError = error;
+    const statusCode = statusError.status ?? statusError.statusCode ?? null;
+    // Check retryable by status
+    const statusRetryable = statusCode !== null && merged.retryableStatuses.includes(statusCode);
+    // Check retryable by message pattern
+    const patternRetryable = merged.retryablePatterns.some((p) => p.test(msg));
+    const retryable = statusRetryable || patternRetryable;
+    // Rate limited = 429 or rate limit pattern
+    const rateLimited = statusCode === 429 || /rate.?limit/i.test(msg);
+    // Extract Retry-After header
+    let retryAfterMs = null;
+    const retryError = error;
+    const headerValue = retryError.headers?.get?.("retry-after") ?? retryError.headers?.["retry-after"];
+    if (headerValue) {
+        const seconds = Number(headerValue);
+        if (!Number.isNaN(seconds) && seconds > 0 && seconds < 300) {
+            retryAfterMs = seconds * 1000;
+        }
+    }
+    return { retryable, rateLimited, statusCode, retryAfterMs, original: err };
+}
+// ── Helpers ──────────────────────────────────────────────────────
+function sleep(ms) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+}
+function computeDelay(attempt, config, classified) {
+    if (classified.retryAfterMs !== null) {
+        return Math.min(classified.retryAfterMs, config.maxDelayMs);
+    }
+    const delay = config.baseDelayMs * config.backoffMultiplier ** attempt;
+    const jitter = delay * (0.5 + Math.random() * 0.5);
+    return Math.min(jitter, config.maxDelayMs);
+}
+// ── ResilientProvider ────────────────────────────────────────────
+export class ResilientProvider {
+    name;
+    inner;
+    config;
+    listener;
+    constructor(inner, config, listener) {
+        this.inner = inner;
+        this.name = inner.name;
+        this.config = mergeConfig(config);
+        this.listener = listener;
+    }
+    supportsModel(model) {
+        return this.inner.supportsModel(model);
+    }
+    async generate(request) {
+        let lastError;
+        for (let attempt = 0; attempt < this.config.maxAttempts; attempt++) {
+            const start = Date.now();
+            try {
+                const result = await this.inner.generate(request);
+                return result;
+            }
+            catch (error) {
+                const classified = classifyError(error, this.config);
+                lastError = classified.original;
+                const durationMs = Date.now() - start;
+                this.listener?.({ attempt, success: false, durationMs, error: classified });
+                if (!classified.retryable || attempt >= this.config.maxAttempts - 1) {
+                    throw lastError;
+                }
+                const delay = computeDelay(attempt, this.config, classified);
+                await sleep(delay);
+            }
+        }
+        throw lastError ?? new Error("ResilientProvider exhausted all attempts");
+    }
+    async *stream(request) {
+        let lastError;
+        for (let attempt = 0; attempt < this.config.maxAttempts; attempt++) {
+            const start = Date.now();
+            try {
+                yield* this.inner.stream(request);
+                return;
+            }
+            catch (error) {
+                const classified = classifyError(error, this.config);
+                lastError = classified.original;
+                const durationMs = Date.now() - start;
+                this.listener?.({ attempt, success: false, durationMs, error: classified });
+                if (!classified.retryable || attempt >= this.config.maxAttempts - 1) {
+                    throw lastError;
+                }
+                const delay = computeDelay(attempt, this.config, classified);
+                await sleep(delay);
+            }
+        }
+        throw lastError ?? new Error("ResilientProvider exhausted all attempts");
+    }
+}
+export class FailoverChain {
+    name;
+    entries;
+    onFailover;
+    constructor(entries, onFailover) {
+        if (entries.length === 0) {
+            throw new Error("FailoverChain requires at least one provider");
+        }
+        this.name = entries[0].provider.name;
+        this.onFailover = onFailover;
+        this.entries = entries.map((e) => ({
+            resilient: new ResilientProvider(e.provider, e.config),
+            provider: e.provider,
+        }));
+    }
+    supportsModel(model) {
+        return this.entries.some((e) => e.provider.supportsModel(model));
+    }
+    async generate(request) {
+        let lastError;
+        for (let i = 0; i < this.entries.length; i++) {
+            try {
+                return await this.entries[i].resilient.generate(request);
+            }
+            catch (error) {
+                lastError = error instanceof Error ? error : new Error(String(error));
+                if (i + 1 < this.entries.length) {
+                    this.onFailover?.(this.entries[i].provider.name, this.entries[i + 1].provider.name);
+                }
+            }
+        }
+        throw lastError ?? new Error("FailoverChain: all providers failed");
+    }
+    async *stream(request) {
+        let lastError;
+        for (let i = 0; i < this.entries.length; i++) {
+            try {
+                yield* this.entries[i].resilient.stream(request);
+                return;
+            }
+            catch (error) {
+                lastError = error instanceof Error ? error : new Error(String(error));
+                if (i + 1 < this.entries.length) {
+                    this.onFailover?.(this.entries[i].provider.name, this.entries[i + 1].provider.name);
+                }
+            }
+        }
+        throw lastError ?? new Error("FailoverChain: all providers failed");
+    }
+}
+export function isTransientError(error) {
+    const classified = classifyError(error);
+    return classified.retryable;
+}
+export async function withRetry(fn, opts) {
+    const maxRetries = opts?.maxRetries ?? 2;
+    const initialDelay = opts?.initialDelayMs ?? 1000;
+    const maxDelay = opts?.maxDelayMs ?? 10000;
+    const signal = opts?.signal;
+    let lastError;
+    for (let attempt = 0; attempt <= maxRetries; attempt++) {
+        if (signal?.aborted) {
+            throw signal.reason ?? new Error("Aborted");
+        }
+        try {
+            return await fn(signal);
+        }
+        catch (error) {
+            lastError = error instanceof Error ? error : new Error(String(error));
+            if (attempt >= maxRetries || !isTransientError(error)) {
+                throw lastError;
+            }
+            const classified = classifyError(error);
+            const backoff = classified.retryAfterMs ?? Math.min(initialDelay * 2 ** attempt, maxDelay);
+            const jitter = backoff * (0.5 + Math.random() * 0.5);
+            await sleep(jitter);
+        }
+    }
+    throw lastError ?? new Error("withRetry exhausted all attempts");
+}
+export function timeoutSignal(ms) {
+    const controller = new AbortController();
+    const timer = setTimeout(() => controller.abort(new Error(`Request timed out after ${ms}ms`)), ms);
+    return {
+        signal: controller.signal,
+        cleanup: () => clearTimeout(timer),
+    };
+}
+//# sourceMappingURL=retry.js.map

package/dist/providers/retry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"retry.js","sourceRoot":"","sources":["../../src/providers/retry.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAuBH,MAAM,cAAc,GAA0B;IAC5C,WAAW,EAAE,CAAC;IACd,WAAW,EAAE,GAAG;IAChB,iBAAiB,EAAE,CAAC;IACpB,UAAU,EAAE,KAAM;IAClB,SAAS,EAAE,KAAM;IACjB,iBAAiB,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC;IAC5C,iBAAiB,EAAE,CAAC,cAAc,EAAE,UAAU,EAAE,YAAY,EAAE,WAAW,EAAE,cAAc,EAAE,eAAe,EAAE,aAAa,CAAC;CAC3H,CAAC;AAEF,SAAS,WAAW,CAAC,OAAqB;IACxC,OAAO,EAAE,GAAG,cAAc,EAAE,GAAG,OAAO,EAAE,CAAC;AAC3C,CAAC;AAYD,MAAM,UAAU,aAAa,CAAC,KAAc,EAAE,SAAsB,cAAc;IAChF,MAAM,GAAG,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IACtE,MAAM,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;IACnC,MAAM,GAAG,GAAG,GAAG,CAAC,OAAO,CAAC;IAExB,wCAAwC;IACxC,MAAM,WAAW,GAAG,KAAiD,CAAC;IACtE,MAAM,UAAU,GAAG,WAAW,CAAC,MAAM,IAAI,WAAW,CAAC,UAAU,IAAI,IAAI,CAAC;IAExE,4BAA4B;IAC5B,MAAM,eAAe,GAAG,UAAU,KAAK,IAAI,IAAI,MAAM,CAAC,iBAAiB,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;IAE7F,qCAAqC;IACrC,MAAM,gBAAgB,GAAG,MAAM,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IAE3E,MAAM,SAAS,GAAG,eAAe,IAAI,gBAAgB,CAAC;IAEtD,2CAA2C;IAC3C,MAAM,WAAW,GAAG,UAAU,KAAK,GAAG,IAAI,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAEnE,6BAA6B;IAC7B,IAAI,YAAY,GAAkB,IAAI,CAAC;IACvC,MAAM,UAAU,GAAG,KAAqF,CAAC;IACzG,MAAM,WAAW,GAAG,UAAU,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,IAAI,UAAU,CAAC,OAAO,EAAE,CAAC,aAAa,CAAC,CAAC;IACpG,IAAI,WAAW,EAAE,CAAC;QAChB,MAAM,OAAO,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC;QACpC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,OAAO,GAAG,CAAC,IAAI,OAAO,GAAG,GAAG,EAAE,CAAC;YAC3D,YAAY,GAAG,OAAO,GAAG,IAAI,CAAC;QAChC,CAAC;IACH,CAAC;IAED,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,UAAU,EAAE,YAAY,EAAE,QAAQ,EAAE,GAAG,EAAE,CAAC;AAC7E,CAAC;AAaD,oEAAoE;AAEpE,SAAS,KAAK,CAAC,EAAU;IACvB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3D,CAAC;AAED,SAAS,YAAY,CAAC,OAAe,EAAE,MAA6B,EAAE,UAA2B;IAC/F,IAAI,UAAU,CAAC,YAAY,KAAK,IAAI,EAAE,CAAC;QACrC,OAAO,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,YAAY,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC;IAC9D,CAAC;IACD,MAAM,KAAK,GAAG,MAAM,CAAC,WAAW,GAAG,MAAM,CAAC,iBAAiB,IAAI,OAAO,CAAC;IACvE,MAAM,MAAM,GAAG,KAAK,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,GAAG,CAAC,CAAC;IACnD,OAAO,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC;AAC7C,CAAC;AAED,oEAAoE;AAEpE,MAAM,OAAO,iBAAiB;IACnB,IAAI,CAAe;IACX,KAAK,CAAgB;IACrB,MAAM,CAAwB;IAC9B,QAAQ,CAAiB;IAE1C,YAAY,KAAoB,EAAE,MAAoB,EAAE,QAAwB;QAC9E,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;QAClC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,aAAa,CAAC,KAAa;QACzB,OAAO,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAAwB;QACrC,IAAI,SAA4B,CAAC;QAEjC,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,OAAO,EAAE,EAAE,CAAC;YACnE,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACzB,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;gBAClD,OAAO,MAAM,CAAC;YAChB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,UAAU,GAAG,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;gBACrD,SAAS,GAAG,UAAU,CAAC,QAAQ,CAAC;gBAChC,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;gBAEtC,IAAI,CAAC,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;gBAE5E,IAAI,CAAC,UAAU,CAAC,SAAS,IAAI,OAAO,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,GAAG,CAAC,EAAE,CAAC;oBACpE,MAAM,SAAS,CAAC;gBAClB,CAAC;gBAED,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;gBAC7D,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;QACH,CAAC;QAED,MAAM,SAAS,IAAI,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;IAC3E,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,OAAsB;QAClC,IAAI,SAA4B,CAAC;QAEjC,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,OAAO,EAAE,EAAE,CAAC;YACnE,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACzB,IAAI,CAAC;gBACH,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;gBAClC,OAAO;YACT,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,UAAU,GAAG,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;gBACrD,SAAS,GAAG,UAAU,CAAC,QAAQ,CAAC;gBAChC,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;gBAEtC,IAAI,CAAC,QAAQ,EAAE,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;gBAE5E,IAAI,CAAC,UAAU,CAAC,SAAS,IAAI,OAAO,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,GAAG,CAAC,EAAE,CAAC;oBACpE,MAAM,SAAS,CAAC;gBAClB,CAAC;gBAED,MAAM,KAAK,GAAG,YAAY,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;gBAC7D,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;QACH,CAAC;QAED,MAAM,SAAS,IAAI,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;IAC3E,CAAC;CACF;AAWD,MAAM,OAAO,aAAa;IACf,IAAI,CAAe;IACX,OAAO,CAAmE;IAC1E,UAAU,CAAoB;IAE/C,YAAY,OAAwB,EAAE,UAA6B;QACjE,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;QAClE,CAAC;QAED,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC;QACrC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACjC,SAAS,EAAE,IAAI,iBAAiB,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC;YACtD,QAAQ,EAAE,CAAC,CAAC,QAAQ;SACrB,CAAC,CAAC,CAAC;IACN,CAAC;IAED,aAAa,CAAC,KAAa;QACzB,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;IACnE,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAAwB;QACrC,IAAI,SAA4B,CAAC;QAEjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC7C,IAAI,CAAC;gBACH,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YAC3D,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;gBACtE,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;oBAChC,IAAI,CAAC,UAAU,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;gBACtF,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,SAAS,IAAI,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACtE,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,OAAsB;QAClC,IAAI,SAA4B,CAAC;QAEjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC7C,IAAI,CAAC;gBACH,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;gBACjD,OAAO;YACT,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;gBACtE,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;oBAChC,IAAI,CAAC,UAAU,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;gBACtF,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,SAAS,IAAI,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACtE,CAAC;CACF;AAWD,MAAM,UAAU,gBAAgB,CAAC,KAAc;IAC7C,MAAM,UAAU,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;IACxC,OAAO,UAAU,CAAC,SAAS,CAAC;AAC9B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,SAAS,CAC7B,EAAwC,EACxC,IAAmB;IAEnB,MAAM,UAAU,GAAG,IAAI,EAAE,UAAU,IAAI,CAAC,CAAC;IACzC,MAAM,YAAY,GAAG,IAAI,EAAE,cAAc,IAAI,IAAI,CAAC;IAClD,MAAM,QAAQ,GAAG,IAAI,EAAE,UAAU,IAAI,KAAM,CAAC;IAC5C,MAAM,MAAM,GAAG,IAAI,EAAE,MAAM,CAAC;IAE5B,IAAI,SAA4B,CAAC;IAEjC,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,UAAU,EAAE,OAAO,EAAE,EAAE,CAAC;QACvD,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;YACpB,MAAM,MAAM,CAAC,MAAM,IAAI,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC;QAC9C,CAAC;QAED,IAAI,CAAC;YACH,OAAO,MAAM,EAAE,CAAC,MAAM,CAAC,CAAC;QAC1B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,SAAS,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YAEtE,IAAI,OAAO,IAAI,UAAU,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC;gBACtD,MAAM,SAAS,CAAC;YAClB,CAAC;YAED,MAAM,UAAU,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;YACxC,MAAM,OAAO,GAAG,UAAU,CAAC,YAAY,IAAI,IAAI,CAAC,GAAG,CAAC,YAAY,GAAG,CAAC,IAAI,OAAO,EAAE,QAAQ,CAAC,CAAC;YAC3F,MAAM,MAAM,GAAG,OAAO,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,GAAG,CAAC,CAAC;YAErD,MAAM,KAAK,CAAC,MAAM,CAAC,CAAC;QACtB,CAAC;IACH,CAAC;IAED,MAAM,SAAS,IAAI,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;AACnE,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,EAAU;IACtC,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,2BAA2B,EAAE,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IACnG,OAAO;QACL,MAAM,EAAE,UAAU,CAAC,MAAM;QACzB,OAAO,EAAE,GAAG,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC;KACnC,CAAC;AACJ,CAAC"}

package/dist/providers/role_scorer.d.ts

@@ -0,0 +1,16 @@
+import type { ModelProfile } from "./base.js";
+import type { DuoRole } from "../duo/duo_types.js";
+export interface RoleScore {
+    model: string;
+    role: DuoRole;
+    score: number;
+    reasons: string[];
+}
+export declare function scoreModelForRole(opts: {
+    profile: ModelProfile;
+    role: DuoRole;
+}): RoleScore;
+export declare function scoreAllModelsForRole(opts: {
+    profiles: ModelProfile[];
+    role: DuoRole;
+}): RoleScore[];