duocode 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/.env.example +36 -0
- package/LICENSE +21 -0
- package/README.md +52 -0
- package/dist/ast/context.d.ts +16 -0
- package/dist/ast/context.js +37 -0
- package/dist/ast/context.js.map +1 -0
- package/dist/ast/diff.d.ts +27 -0
- package/dist/ast/diff.js +44 -0
- package/dist/ast/diff.js.map +1 -0
- package/dist/ast/locks.d.ts +47 -0
- package/dist/ast/locks.js +88 -0
- package/dist/ast/locks.js.map +1 -0
- package/dist/ast/merge.d.ts +22 -0
- package/dist/ast/merge.js +120 -0
- package/dist/ast/merge.js.map +1 -0
- package/dist/ast/ownership.d.ts +31 -0
- package/dist/ast/ownership.js +111 -0
- package/dist/ast/ownership.js.map +1 -0
- package/dist/ast/parser.d.ts +44 -0
- package/dist/ast/parser.js +134 -0
- package/dist/ast/parser.js.map +1 -0
- package/dist/cli.d.ts +2 -0
- package/dist/cli.js +423 -0
- package/dist/cli.js.map +1 -0
- package/dist/commands/doctor.d.ts +5 -0
- package/dist/commands/doctor.js +63 -0
- package/dist/commands/doctor.js.map +1 -0
- package/dist/commands/duo.d.ts +9 -0
- package/dist/commands/duo.js +285 -0
- package/dist/commands/duo.js.map +1 -0
- package/dist/commands/github.d.ts +2 -0
- package/dist/commands/github.js +85 -0
- package/dist/commands/github.js.map +1 -0
- package/dist/commands/init.d.ts +2 -0
- package/dist/commands/init.js +33 -0
- package/dist/commands/init.js.map +1 -0
- package/dist/commands/negotiation.d.ts +2 -0
- package/dist/commands/negotiation.js +160 -0
- package/dist/commands/negotiation.js.map +1 -0
- package/dist/commands/repl_commands.d.ts +26 -0
- package/dist/commands/repl_commands.js +226 -0
- package/dist/commands/repl_commands.js.map +1 -0
- package/dist/commands/shell.d.ts +1 -0
- package/dist/commands/shell.js +110 -0
- package/dist/commands/shell.js.map +1 -0
- package/dist/commands/start.d.ts +2 -0
- package/dist/commands/start.js +231 -0
- package/dist/commands/start.js.map +1 -0
- package/dist/commands/task.d.ts +2 -0
- package/dist/commands/task.js +215 -0
- package/dist/commands/task.js.map +1 -0
- package/dist/config/loader.d.ts +193 -0
- package/dist/config/loader.js +106 -0
- package/dist/config/loader.js.map +1 -0
- package/dist/context/project_context.d.ts +79 -0
- package/dist/context/project_context.js +292 -0
- package/dist/context/project_context.js.map +1 -0
- package/dist/context/token_budget.d.ts +35 -0
- package/dist/context/token_budget.js +81 -0
- package/dist/context/token_budget.js.map +1 -0
- package/dist/db/queries.d.ts +121 -0
- package/dist/db/queries.js +109 -0
- package/dist/db/queries.js.map +1 -0
- package/dist/db/schema.d.ts +110 -0
- package/dist/db/schema.js +346 -0
- package/dist/db/schema.js.map +1 -0
- package/dist/duo/duo_orchestrator.d.ts +50 -0
- package/dist/duo/duo_orchestrator.js +510 -0
- package/dist/duo/duo_orchestrator.js.map +1 -0
- package/dist/duo/duo_session.d.ts +47 -0
- package/dist/duo/duo_session.js +127 -0
- package/dist/duo/duo_session.js.map +1 -0
- package/dist/duo/duo_types.d.ts +168 -0
- package/dist/duo/duo_types.js +53 -0
- package/dist/duo/duo_types.js.map +1 -0
- package/dist/duo/session_store.d.ts +71 -0
- package/dist/duo/session_store.js +177 -0
- package/dist/duo/session_store.js.map +1 -0
- package/dist/git/worktree.d.ts +21 -0
- package/dist/git/worktree.js +86 -0
- package/dist/git/worktree.js.map +1 -0
- package/dist/github/cache.d.ts +23 -0
- package/dist/github/cache.js +67 -0
- package/dist/github/cache.js.map +1 -0
- package/dist/github/issues.d.ts +17 -0
- package/dist/github/issues.js +93 -0
- package/dist/github/issues.js.map +1 -0
- package/dist/github/mcp_client.d.ts +57 -0
- package/dist/github/mcp_client.js +214 -0
- package/dist/github/mcp_client.js.map +1 -0
- package/dist/github/sync.d.ts +11 -0
- package/dist/github/sync.js +65 -0
- package/dist/github/sync.js.map +1 -0
- package/dist/github/webhook.d.ts +25 -0
- package/dist/github/webhook.js +197 -0
- package/dist/github/webhook.js.map +1 -0
- package/dist/negotiation/index.d.ts +1 -0
- package/dist/negotiation/index.js +2 -0
- package/dist/negotiation/index.js.map +1 -0
- package/dist/negotiation/protocol.d.ts +62 -0
- package/dist/negotiation/protocol.js +188 -0
- package/dist/negotiation/protocol.js.map +1 -0
- package/dist/orchestrator/complexity_scorer.d.ts +2 -0
- package/dist/orchestrator/complexity_scorer.js +79 -0
- package/dist/orchestrator/complexity_scorer.js.map +1 -0
- package/dist/orchestrator/dependency_graph.d.ts +7 -0
- package/dist/orchestrator/dependency_graph.js +73 -0
- package/dist/orchestrator/dependency_graph.js.map +1 -0
- package/dist/orchestrator/intent_parser.d.ts +11 -0
- package/dist/orchestrator/intent_parser.js +116 -0
- package/dist/orchestrator/intent_parser.js.map +1 -0
- package/dist/orchestrator/task_runner.d.ts +56 -0
- package/dist/orchestrator/task_runner.js +181 -0
- package/dist/orchestrator/task_runner.js.map +1 -0
- package/dist/orchestrator/types.d.ts +44 -0
- package/dist/orchestrator/types.js +21 -0
- package/dist/orchestrator/types.js.map +1 -0
- package/dist/providers/anthropic.d.ts +12 -0
- package/dist/providers/anthropic.js +258 -0
- package/dist/providers/anthropic.js.map +1 -0
- package/dist/providers/auction.d.ts +42 -0
- package/dist/providers/auction.js +190 -0
- package/dist/providers/auction.js.map +1 -0
- package/dist/providers/base.d.ts +103 -0
- package/dist/providers/base.js +2 -0
- package/dist/providers/base.js.map +1 -0
- package/dist/providers/cost_tracker.d.ts +45 -0
- package/dist/providers/cost_tracker.js +111 -0
- package/dist/providers/cost_tracker.js.map +1 -0
- package/dist/providers/duo_pair_router.d.ts +11 -0
- package/dist/providers/duo_pair_router.js +67 -0
- package/dist/providers/duo_pair_router.js.map +1 -0
- package/dist/providers/factory.d.ts +7 -0
- package/dist/providers/factory.js +130 -0
- package/dist/providers/factory.js.map +1 -0
- package/dist/providers/grading_rubric.d.ts +37 -0
- package/dist/providers/grading_rubric.js +238 -0
- package/dist/providers/grading_rubric.js.map +1 -0
- package/dist/providers/openai.d.ts +12 -0
- package/dist/providers/openai.js +229 -0
- package/dist/providers/openai.js.map +1 -0
- package/dist/providers/openrouter.d.ts +14 -0
- package/dist/providers/openrouter.js +178 -0
- package/dist/providers/openrouter.js.map +1 -0
- package/dist/providers/performance_tracker.d.ts +21 -0
- package/dist/providers/performance_tracker.js +63 -0
- package/dist/providers/performance_tracker.js.map +1 -0
- package/dist/providers/registry_loader.d.ts +6 -0
- package/dist/providers/registry_loader.js +54 -0
- package/dist/providers/registry_loader.js.map +1 -0
- package/dist/providers/retry.d.ts +66 -0
- package/dist/providers/retry.js +203 -0
- package/dist/providers/retry.js.map +1 -0
- package/dist/providers/role_scorer.d.ts +16 -0
- package/dist/providers/role_scorer.js +16 -0
- package/dist/providers/role_scorer.js.map +1 -0
- package/dist/providers/router.d.ts +84 -0
- package/dist/providers/router.js +542 -0
- package/dist/providers/router.js.map +1 -0
- package/dist/security/credentials.d.ts +6 -0
- package/dist/security/credentials.js +16 -0
- package/dist/security/credentials.js.map +1 -0
- package/dist/setup/browser.d.ts +1 -0
- package/dist/setup/browser.js +12 -0
- package/dist/setup/browser.js.map +1 -0
- package/dist/setup/global_config.d.ts +14 -0
- package/dist/setup/global_config.js +54 -0
- package/dist/setup/global_config.js.map +1 -0
- package/dist/setup/wizard.d.ts +2 -0
- package/dist/setup/wizard.js +206 -0
- package/dist/setup/wizard.js.map +1 -0
- package/dist/tools/agent_loop.d.ts +38 -0
- package/dist/tools/agent_loop.js +72 -0
- package/dist/tools/agent_loop.js.map +1 -0
- package/dist/tools/approval.d.ts +64 -0
- package/dist/tools/approval.js +172 -0
- package/dist/tools/approval.js.map +1 -0
- package/dist/tools/checkpoint.d.ts +65 -0
- package/dist/tools/checkpoint.js +342 -0
- package/dist/tools/checkpoint.js.map +1 -0
- package/dist/tools/definitions.d.ts +13 -0
- package/dist/tools/definitions.js +103 -0
- package/dist/tools/definitions.js.map +1 -0
- package/dist/tools/diff_display.d.ts +46 -0
- package/dist/tools/diff_display.js +298 -0
- package/dist/tools/diff_display.js.map +1 -0
- package/dist/tools/executor.d.ts +12 -0
- package/dist/tools/executor.js +340 -0
- package/dist/tools/executor.js.map +1 -0
- package/dist/tools/permissions.d.ts +17 -0
- package/dist/tools/permissions.js +139 -0
- package/dist/tools/permissions.js.map +1 -0
- package/dist/tools/tool_types.d.ts +48 -0
- package/dist/tools/tool_types.js +7 -0
- package/dist/tools/tool_types.js.map +1 -0
- package/dist/ui/banner.d.ts +4 -0
- package/dist/ui/banner.js +104 -0
- package/dist/ui/banner.js.map +1 -0
- package/dist/ui/callbacks.d.ts +30 -0
- package/dist/ui/callbacks.js +132 -0
- package/dist/ui/callbacks.js.map +1 -0
- package/dist/ui/colors.d.ts +14 -0
- package/dist/ui/colors.js +28 -0
- package/dist/ui/colors.js.map +1 -0
- package/dist/ui/dashboard.d.ts +51 -0
- package/dist/ui/dashboard.js +181 -0
- package/dist/ui/dashboard.js.map +1 -0
- package/dist/ui/leaderboard.d.ts +16 -0
- package/dist/ui/leaderboard.js +43 -0
- package/dist/ui/leaderboard.js.map +1 -0
- package/dist/ui/logger.d.ts +28 -0
- package/dist/ui/logger.js +117 -0
- package/dist/ui/logger.js.map +1 -0
- package/dist/ui/progress.d.ts +16 -0
- package/dist/ui/progress.js +62 -0
- package/dist/ui/progress.js.map +1 -0
- package/dist/ui/tokenizer.d.ts +5 -0
- package/dist/ui/tokenizer.js +54 -0
- package/dist/ui/tokenizer.js.map +1 -0
- package/package.json +63 -0
|
@@ -0,0 +1,45 @@
|
|
|
1
|
+
import type { ProviderName, TokenUsage } from "./base.js";
|
|
2
|
+
import type Database from "better-sqlite3";
|
|
3
|
+
export interface ModelPricing {
|
|
4
|
+
inputPer1kUsd: number;
|
|
5
|
+
outputPer1kUsd: number;
|
|
6
|
+
cacheReadInputPer1kUsd?: number;
|
|
7
|
+
cacheWriteInputPer1kUsd?: number;
|
|
8
|
+
}
|
|
9
|
+
export interface CostRecord {
|
|
10
|
+
requestId: string;
|
|
11
|
+
timestamp: number;
|
|
12
|
+
provider: ProviderName;
|
|
13
|
+
model: string;
|
|
14
|
+
usage: TokenUsage;
|
|
15
|
+
costUsd: number;
|
|
16
|
+
}
|
|
17
|
+
export interface CostTotals {
|
|
18
|
+
totalUsd: number;
|
|
19
|
+
byProvider: Record<string, number>;
|
|
20
|
+
byModel: Record<string, number>;
|
|
21
|
+
}
|
|
22
|
+
export declare const DEFAULT_MODEL_PRICING: Record<string, ModelPricing>;
|
|
23
|
+
export declare class CostTracker {
|
|
24
|
+
private readonly pricing;
|
|
25
|
+
private readonly records;
|
|
26
|
+
private db;
|
|
27
|
+
constructor(pricing?: Record<string, ModelPricing>);
|
|
28
|
+
/**
|
|
29
|
+
* Attach a database for persisting cost records to provider_usage table.
|
|
30
|
+
* Can be called after construction when DB becomes available.
|
|
31
|
+
*/
|
|
32
|
+
setDatabase(db: Database.Database): void;
|
|
33
|
+
setPricing(model: string, pricing: ModelPricing): void;
|
|
34
|
+
getPricing(model: string): ModelPricing | null;
|
|
35
|
+
estimate(model: string, usage: TokenUsage): number;
|
|
36
|
+
record(input: {
|
|
37
|
+
provider: ProviderName;
|
|
38
|
+
model: string;
|
|
39
|
+
usage: TokenUsage;
|
|
40
|
+
requestId?: string;
|
|
41
|
+
timestamp?: number;
|
|
42
|
+
}): CostRecord;
|
|
43
|
+
getTotals(): CostTotals;
|
|
44
|
+
getRecords(): CostRecord[];
|
|
45
|
+
}
|
|
@@ -0,0 +1,111 @@
|
|
|
1
|
+
export const DEFAULT_MODEL_PRICING = {
|
|
2
|
+
"openrouter/openai/gpt-5.3-codex": { inputPer1kUsd: 0.015, outputPer1kUsd: 0.06 },
|
|
3
|
+
"openrouter/openai/gpt-5.2": { inputPer1kUsd: 0.01, outputPer1kUsd: 0.04 },
|
|
4
|
+
"openrouter/anthropic/claude-sonnet-5": { inputPer1kUsd: 0.004, outputPer1kUsd: 0.02 },
|
|
5
|
+
"gpt-5.3-codex": { inputPer1kUsd: 0.015, outputPer1kUsd: 0.06 },
|
|
6
|
+
"gpt-5.2": { inputPer1kUsd: 0.01, outputPer1kUsd: 0.04 },
|
|
7
|
+
"gpt-4.1": { inputPer1kUsd: 0.01, outputPer1kUsd: 0.03 },
|
|
8
|
+
"o4-mini": { inputPer1kUsd: 0.003, outputPer1kUsd: 0.012 },
|
|
9
|
+
"claude-sonnet-4-5": { inputPer1kUsd: 0.003, outputPer1kUsd: 0.015 },
|
|
10
|
+
"claude-sonnet-5": { inputPer1kUsd: 0.004, outputPer1kUsd: 0.02 },
|
|
11
|
+
// OpenRouter models — internal names
|
|
12
|
+
"glm-5": { inputPer1kUsd: 0.0008, outputPer1kUsd: 0.00256 },
|
|
13
|
+
"qwen3-max-thinking": { inputPer1kUsd: 0.0012, outputPer1kUsd: 0.006 },
|
|
14
|
+
"qwen3-coder-next": { inputPer1kUsd: 0.00007, outputPer1kUsd: 0.0003 },
|
|
15
|
+
"kimi-k2.5": { inputPer1kUsd: 0.00045, outputPer1kUsd: 0.00225 },
|
|
16
|
+
// OpenRouter models — OpenRouter IDs (used after profile remapping)
|
|
17
|
+
"z-ai/glm-5": { inputPer1kUsd: 0.0008, outputPer1kUsd: 0.00256 },
|
|
18
|
+
"qwen/qwen3-max-thinking": { inputPer1kUsd: 0.0012, outputPer1kUsd: 0.006 },
|
|
19
|
+
"qwen/qwen3-coder-next": { inputPer1kUsd: 0.00007, outputPer1kUsd: 0.0003 },
|
|
20
|
+
"moonshotai/kimi-k2.5": { inputPer1kUsd: 0.00045, outputPer1kUsd: 0.00225 },
|
|
21
|
+
// OpenRouter existing models — OpenRouter IDs
|
|
22
|
+
"openai/gpt-4.1": { inputPer1kUsd: 0.01, outputPer1kUsd: 0.03 },
|
|
23
|
+
"openai/gpt-4.1-mini": { inputPer1kUsd: 0.004, outputPer1kUsd: 0.016 },
|
|
24
|
+
"openai/o4-mini": { inputPer1kUsd: 0.003, outputPer1kUsd: 0.012 },
|
|
25
|
+
"anthropic/claude-sonnet-4-5-20250514": { inputPer1kUsd: 0.003, outputPer1kUsd: 0.015 },
|
|
26
|
+
};
|
|
27
|
+
function estimateCost(usage, pricing) {
|
|
28
|
+
const input = (usage.inputTokens / 1000) * pricing.inputPer1kUsd;
|
|
29
|
+
const output = (usage.outputTokens / 1000) * pricing.outputPer1kUsd;
|
|
30
|
+
const cacheRead = ((usage.cacheReadInputTokens ?? 0) / 1000) * (pricing.cacheReadInputPer1kUsd ?? pricing.inputPer1kUsd);
|
|
31
|
+
const cacheWrite = ((usage.cacheWriteInputTokens ?? 0) / 1000) * (pricing.cacheWriteInputPer1kUsd ?? pricing.inputPer1kUsd);
|
|
32
|
+
return input + output + cacheRead + cacheWrite;
|
|
33
|
+
}
|
|
34
|
+
function makeRequestId() {
|
|
35
|
+
return `${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 8)}`;
|
|
36
|
+
}
|
|
37
|
+
export class CostTracker {
|
|
38
|
+
pricing = new Map();
|
|
39
|
+
records = [];
|
|
40
|
+
db = null;
|
|
41
|
+
constructor(pricing = DEFAULT_MODEL_PRICING) {
|
|
42
|
+
for (const [model, value] of Object.entries(pricing)) {
|
|
43
|
+
this.pricing.set(model, value);
|
|
44
|
+
}
|
|
45
|
+
}
|
|
46
|
+
/**
|
|
47
|
+
* Attach a database for persisting cost records to provider_usage table.
|
|
48
|
+
* Can be called after construction when DB becomes available.
|
|
49
|
+
*/
|
|
50
|
+
setDatabase(db) {
|
|
51
|
+
this.db = db;
|
|
52
|
+
}
|
|
53
|
+
setPricing(model, pricing) {
|
|
54
|
+
this.pricing.set(model, pricing);
|
|
55
|
+
}
|
|
56
|
+
getPricing(model) {
|
|
57
|
+
return this.pricing.get(model) ?? null;
|
|
58
|
+
}
|
|
59
|
+
estimate(model, usage) {
|
|
60
|
+
const pricing = this.pricing.get(model);
|
|
61
|
+
if (!pricing) {
|
|
62
|
+
return 0;
|
|
63
|
+
}
|
|
64
|
+
return estimateCost(usage, pricing);
|
|
65
|
+
}
|
|
66
|
+
record(input) {
|
|
67
|
+
const requestId = input.requestId ?? makeRequestId();
|
|
68
|
+
const timestamp = input.timestamp ?? Date.now();
|
|
69
|
+
const costUsd = this.estimate(input.model, input.usage);
|
|
70
|
+
const record = {
|
|
71
|
+
requestId,
|
|
72
|
+
timestamp,
|
|
73
|
+
provider: input.provider,
|
|
74
|
+
model: input.model,
|
|
75
|
+
usage: input.usage,
|
|
76
|
+
costUsd,
|
|
77
|
+
};
|
|
78
|
+
this.records.push(record);
|
|
79
|
+
// Persist to database if attached
|
|
80
|
+
if (this.db) {
|
|
81
|
+
try {
|
|
82
|
+
this.db.prepare(`
|
|
83
|
+
INSERT OR IGNORE INTO provider_usage (
|
|
84
|
+
request_id, task_id, provider, model,
|
|
85
|
+
input_tokens, output_tokens, cache_read_input_tokens,
|
|
86
|
+
cache_write_input_tokens, cost_usd
|
|
87
|
+
) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
88
|
+
`).run(requestId, null, input.provider, input.model, input.usage.inputTokens, input.usage.outputTokens, input.usage.cacheReadInputTokens ?? 0, input.usage.cacheWriteInputTokens ?? 0, costUsd);
|
|
89
|
+
}
|
|
90
|
+
catch {
|
|
91
|
+
// Don't let DB errors break the pipeline
|
|
92
|
+
}
|
|
93
|
+
}
|
|
94
|
+
return record;
|
|
95
|
+
}
|
|
96
|
+
getTotals() {
|
|
97
|
+
const byProvider = {};
|
|
98
|
+
const byModel = {};
|
|
99
|
+
let totalUsd = 0;
|
|
100
|
+
for (const record of this.records) {
|
|
101
|
+
totalUsd += record.costUsd;
|
|
102
|
+
byProvider[record.provider] = (byProvider[record.provider] ?? 0) + record.costUsd;
|
|
103
|
+
byModel[record.model] = (byModel[record.model] ?? 0) + record.costUsd;
|
|
104
|
+
}
|
|
105
|
+
return { totalUsd, byProvider, byModel };
|
|
106
|
+
}
|
|
107
|
+
getRecords() {
|
|
108
|
+
return [...this.records];
|
|
109
|
+
}
|
|
110
|
+
}
|
|
111
|
+
//# sourceMappingURL=cost_tracker.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cost_tracker.js","sourceRoot":"","sources":["../../src/providers/cost_tracker.ts"],"names":[],"mappings":"AAyBA,MAAM,CAAC,MAAM,qBAAqB,GAAiC;IACjE,iCAAiC,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE;IACjF,2BAA2B,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE;IAC1E,sCAAsC,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE;IACtF,eAAe,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE;IAC/D,SAAS,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE;IACxD,SAAS,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE;IACxD,SAAS,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,cAAc,EAAE,KAAK,EAAE;IAC1D,mBAAmB,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,cAAc,EAAE,KAAK,EAAE;IACpE,iBAAiB,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE;IACjE,qCAAqC;IACrC,OAAO,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,cAAc,EAAE,OAAO,EAAE;IAC3D,oBAAoB,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,cAAc,EAAE,KAAK,EAAE;IACtE,kBAAkB,EAAE,EAAE,aAAa,EAAE,OAAO,EAAE,cAAc,EAAE,MAAM,EAAE;IACtE,WAAW,EAAE,EAAE,aAAa,EAAE,OAAO,EAAE,cAAc,EAAE,OAAO,EAAE;IAChE,oEAAoE;IACpE,YAAY,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,cAAc,EAAE,OAAO,EAAE;IAChE,yBAAyB,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,cAAc,EAAE,KAAK,EAAE;IAC3E,uBAAuB,EAAE,EAAE,aAAa,EAAE,OAAO,EAAE,cAAc,EAAE,MAAM,EAAE;IAC3E,sBAAsB,EAAE,EAAE,aAAa,EAAE,OAAO,EAAE,cAAc,EAAE,OAAO,EAAE;IAC3E,8CAA8C;IAC9C,gBAAgB,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE;IAC/D,qBAAqB,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,cAAc,EAAE,KAAK,EAAE;IACtE,gBAAgB,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,cAAc,EAAE,KAAK,EAAE;IACjE,sCAAsC,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,cAAc,EAAE,KAAK,EAAE;CACxF,CAAC;AAEF,SAAS,YAAY,CAAC,KAAiB,EAAE,OAAqB;IAC5D,MAAM,KAAK,GAAG,CAAC,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,OAAO,CAAC,aAAa,CAAC;IACjE,MAAM,MAAM,GAAG,CAAC,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,OAAO,CAAC,cAAc,CAAC;IACpE,MAAM,SAAS,GAAG,CAAC,CAAC,KAAK,CAAC,oBAAoB,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,sBAAsB,IAAI,OAAO,CAAC,aAAa,CAAC,CAAC;IACzH,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC,qBAAqB,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,uBAAuB,IAAI,OAAO,CAAC,aAAa,CAAC,CAAC;IAC5H,OAAO,KAAK,GAAG,MAAM,GAAG,SAAS,GAAG,UAAU,CAAC;AACjD,CAAC;AAED,SAAS,aAAa;IACpB,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;AAChF,CAAC;AAED,MAAM,OAAO,WAAW;IACL,OAAO,GAAG,IAAI,GAAG,EAAwB,CAAC;IAC1C,OAAO,GAAiB,EAAE,CAAC;IACpC,EAAE,GAA6B,IAAI,CAAC;IAE5C,YAAY,UAAwC,qBAAqB;QACvE,KAAK,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YACrD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QACjC,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,WAAW,CAAC,EAAqB;QAC/B,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC;IACf,CAAC;IAED,UAAU,CAAC,KAAa,EAAE,OAAqB;QAC7C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IACnC,CAAC;IAED,UAAU,CAAC,KAAa;QACtB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC;IACzC,CAAC;IAED,QAAQ,CAAC,KAAa,EAAE,KAAiB;QACvC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACxC,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,CAAC,CAAC;QACX,CAAC;QAED,OAAO,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IACtC,CAAC;IAED,MAAM,CAAC,KAMN;QACC,MAAM,SAAS,GAAG,KAAK,CAAC,SAAS,IAAI,aAAa,EAAE,CAAC;QACrD,MAAM,SAAS,GAAG,KAAK,CAAC,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;QAChD,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;QAExD,MAAM,MAAM,GAAe;YACzB,SAAS;YACT,SAAS;YACT,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,OAAO;SACR,CAAC;QAEF,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAE1B,kCAAkC;QAClC,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;YACZ,IAAI,CAAC;gBACH,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC;;;;;;SAMf,CAAC,CAAC,GAAG,CACJ,SAAS,EACT,IAAI,EACJ,KAAK,CAAC,QAAQ,EACd,KAAK,CAAC,KAAK,EACX,KAAK,CAAC,KAAK,CAAC,WAAW,EACvB,KAAK,CAAC,KAAK,CAAC,YAAY,EACxB,KAAK,CAAC,KAAK,CAAC,oBAAoB,IAAI,CAAC,EACrC,KAAK,CAAC,KAAK,CAAC,qBAAqB,IAAI,CAAC,EACtC,OAAO,CACR,CAAC;YACJ,CAAC;YAAC,MAAM,CAAC;gBACP,yCAAyC;YAC3C,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,SAAS;QACP,MAAM,UAAU,GAA2B,EAAE,CAAC;QAC9C,MAAM,OAAO,GAA2B,EAAE,CAAC;QAE3C,IAAI,QAAQ,GAAG,CAAC,CAAC;QACjB,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAClC,QAAQ,IAAI,MAAM,CAAC,OAAO,CAAC;YAC3B,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC;YAClF,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC;QACxE,CAAC;QAED,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC;IAC3C,CAAC;IAED,UAAU;QACR,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC;IAC3B,CAAC;CACF"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import { CapabilityRouter } from "./router.js";
|
|
2
|
+
import type { DuoPairAssignment } from "../duo/duo_types.js";
|
|
3
|
+
export declare class DuoPairRouter {
|
|
4
|
+
private readonly capabilityRouter;
|
|
5
|
+
private readonly _costTracker?;
|
|
6
|
+
constructor(capabilityRouter: CapabilityRouter, _costTracker?: unknown);
|
|
7
|
+
assign(opts: {
|
|
8
|
+
preferredArchitect?: string;
|
|
9
|
+
preferredExecutor?: string;
|
|
10
|
+
}): DuoPairAssignment;
|
|
11
|
+
}
|
|
@@ -0,0 +1,67 @@
|
|
|
1
|
+
import { scoreAllModelsForRole, } from "./role_scorer.js";
|
|
2
|
+
export class DuoPairRouter {
|
|
3
|
+
capabilityRouter;
|
|
4
|
+
_costTracker;
|
|
5
|
+
constructor(capabilityRouter, _costTracker) {
|
|
6
|
+
this.capabilityRouter = capabilityRouter;
|
|
7
|
+
this._costTracker = _costTracker;
|
|
8
|
+
}
|
|
9
|
+
assign(opts) {
|
|
10
|
+
const profiles = this.capabilityRouter.listProfiles();
|
|
11
|
+
if (profiles.length === 0) {
|
|
12
|
+
throw new Error("No model profiles available for DuoPair routing.");
|
|
13
|
+
}
|
|
14
|
+
const architectScores = scoreAllModelsForRole({ profiles, role: "architect" });
|
|
15
|
+
const executorScores = scoreAllModelsForRole({ profiles, role: "executor" });
|
|
16
|
+
if (architectScores.length === 0 || executorScores.length === 0) {
|
|
17
|
+
throw new Error("No models scored for either architect or executor role.");
|
|
18
|
+
}
|
|
19
|
+
// Apply preferred model overrides
|
|
20
|
+
const filterPreferred = (scores, preferred) => {
|
|
21
|
+
if (!preferred)
|
|
22
|
+
return scores;
|
|
23
|
+
const filtered = scores.filter((s) => s.model === preferred);
|
|
24
|
+
return filtered.length > 0 ? filtered : scores;
|
|
25
|
+
};
|
|
26
|
+
const filteredArch = filterPreferred(architectScores, opts.preferredArchitect);
|
|
27
|
+
const filteredExec = filterPreferred(executorScores, opts.preferredExecutor);
|
|
28
|
+
// Build candidate pairs
|
|
29
|
+
const candidates = [];
|
|
30
|
+
for (const arch of filteredArch) {
|
|
31
|
+
for (const exec of filteredExec) {
|
|
32
|
+
const sameModel = arch.model === exec.model;
|
|
33
|
+
const overlapPenalty = sameModel ? 0.03 : 0;
|
|
34
|
+
const pairScore = arch.score + exec.score - overlapPenalty;
|
|
35
|
+
candidates.push({ architect: arch, executor: exec, pairScore, sameModel });
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
candidates.sort((a, b) => b.pairScore - a.pairScore);
|
|
39
|
+
const chosen = candidates[0];
|
|
40
|
+
const profileMap = new Map(profiles.map((p) => [p.model, p]));
|
|
41
|
+
const archProfile = profileMap.get(chosen.architect.model);
|
|
42
|
+
const execProfile = profileMap.get(chosen.executor.model);
|
|
43
|
+
const architectSlot = {
|
|
44
|
+
profile: archProfile,
|
|
45
|
+
role: "architect",
|
|
46
|
+
roleScore: chosen.architect.score,
|
|
47
|
+
reasons: chosen.architect.reasons,
|
|
48
|
+
};
|
|
49
|
+
const executorSlot = {
|
|
50
|
+
profile: execProfile,
|
|
51
|
+
role: "executor",
|
|
52
|
+
roleScore: chosen.executor.score,
|
|
53
|
+
reasons: chosen.executor.reasons,
|
|
54
|
+
};
|
|
55
|
+
const rationale = chosen.sameModel
|
|
56
|
+
? `Same model (${archProfile.model}) assigned to both roles.`
|
|
57
|
+
: `Split pair: ${archProfile.model} (architect, ${chosen.architect.score.toFixed(3)}) + ${execProfile.model} (executor, ${chosen.executor.score.toFixed(3)}).`;
|
|
58
|
+
return {
|
|
59
|
+
architect: architectSlot,
|
|
60
|
+
executor: executorSlot,
|
|
61
|
+
pairScore: chosen.pairScore,
|
|
62
|
+
sameModel: chosen.sameModel,
|
|
63
|
+
rationale,
|
|
64
|
+
};
|
|
65
|
+
}
|
|
66
|
+
}
|
|
67
|
+
//# sourceMappingURL=duo_pair_router.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"duo_pair_router.js","sourceRoot":"","sources":["../../src/providers/duo_pair_router.ts"],"names":[],"mappings":"AACA,OAAO,EACL,qBAAqB,GAEtB,MAAM,kBAAkB,CAAC;AAa1B,MAAM,OAAO,aAAa;IAEL;IACA;IAFnB,YACmB,gBAAkC,EAClC,YAAsB;QADtB,qBAAgB,GAAhB,gBAAgB,CAAkB;QAClC,iBAAY,GAAZ,YAAY,CAAU;IACtC,CAAC;IAEJ,MAAM,CAAC,IAGN;QACC,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC,YAAY,EAAE,CAAC;QACtD,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;QACtE,CAAC;QAED,MAAM,eAAe,GAAG,qBAAqB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC;QAC/E,MAAM,cAAc,GAAG,qBAAqB,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;QAE7E,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,IAAI,cAAc,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChE,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;QAC7E,CAAC;QAED,kCAAkC;QAClC,MAAM,eAAe,GAAG,CACtB,MAAmB,EACnB,SAAkB,EACL,EAAE;YACf,IAAI,CAAC,SAAS;gBAAE,OAAO,MAAM,CAAC;YAC9B,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC;YAC7D,OAAO,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC;QACjD,CAAC,CAAC;QAEF,MAAM,YAAY,GAAG,eAAe,CAAC,eAAe,EAAE,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAC/E,MAAM,YAAY,GAAG,eAAe,CAAC,cAAc,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAE7E,wBAAwB;QACxB,MAAM,UAAU,GAAoB,EAAE,CAAC;QAEvC,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;YAChC,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;gBAChC,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,KAAK,IAAI,CAAC,KAAK,CAAC;gBAC5C,MAAM,cAAc,GAAG,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC5C,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,GAAG,cAAc,CAAC;gBAE3D,UAAU,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;YAC7E,CAAC;QACH,CAAC;QAED,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;QACrD,MAAM,MAAM,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;QAE7B,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAE9D,MAAM,WAAW,GAAG,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAE,CAAC;QAC5D,MAAM,WAAW,GAAG,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAE,CAAC;QAE3D,MAAM,aAAa,GAAgB;YACjC,OAAO,EAAE,WAAW;YACpB,IAAI,EAAE,WAAW;YACjB,SAAS,EAAE,MAAM,CAAC,SAAS,CAAC,KAAK;YACjC,OAAO,EAAE,MAAM,CAAC,SAAS,CAAC,OAAO;SAClC,CAAC;QAEF,MAAM,YAAY,GAAgB;YAChC,OAAO,EAAE,WAAW;YACpB,IAAI,EAAE,UAAU;YAChB,SAAS,EAAE,MAAM,CAAC,QAAQ,CAAC,KAAK;YAChC,OAAO,EAAE,MAAM,CAAC,QAAQ,CAAC,OAAO;SACjC,CAAC;QAEF,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS;YAChC,CAAC,CAAC,eAAe,WAAW,CAAC,KAAK,2BAA2B;YAC7D,CAAC,CAAC,eAAe,WAAW,CAAC,KAAK,gBAAgB,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,WAAW,CAAC,KAAK,eAAe,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;QAEjK,OAAO;YACL,SAAS,EAAE,aAAa;YACxB,QAAQ,EAAE,YAAY;YACtB,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,SAAS;SACV,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1,130 @@
|
|
|
1
|
+
import { OpenRouterProvider } from "./openrouter.js";
|
|
2
|
+
import { CapabilityRouter, ProviderRouter } from "./router.js";
|
|
3
|
+
import { CostTracker, DEFAULT_MODEL_PRICING } from "./cost_tracker.js";
|
|
4
|
+
const OPENROUTER_PROFILES = [
|
|
5
|
+
// --- Frontier ---
|
|
6
|
+
{
|
|
7
|
+
model: "openai/gpt-5.3-codex",
|
|
8
|
+
provider: "openrouter",
|
|
9
|
+
capabilities: ["deep_reasoning", "code_generation", "tool_use", "streaming"],
|
|
10
|
+
supportsStreaming: true,
|
|
11
|
+
supportsTools: true,
|
|
12
|
+
contextWindow: 256000,
|
|
13
|
+
maxOutputTokens: 32000,
|
|
14
|
+
avgLatencyMs: 1800,
|
|
15
|
+
defaultScore: 0.96,
|
|
16
|
+
},
|
|
17
|
+
{
|
|
18
|
+
model: "anthropic/claude-sonnet-5",
|
|
19
|
+
provider: "openrouter",
|
|
20
|
+
capabilities: ["deep_reasoning", "architecture", "code_generation", "tool_use", "streaming"],
|
|
21
|
+
supportsStreaming: true,
|
|
22
|
+
supportsTools: true,
|
|
23
|
+
contextWindow: 200000,
|
|
24
|
+
maxOutputTokens: 24000,
|
|
25
|
+
avgLatencyMs: 1700,
|
|
26
|
+
defaultScore: 0.95,
|
|
27
|
+
},
|
|
28
|
+
{
|
|
29
|
+
model: "google/gemini-3.0-pro",
|
|
30
|
+
provider: "openrouter",
|
|
31
|
+
capabilities: ["deep_reasoning", "code_generation", "streaming", "tool_use"],
|
|
32
|
+
supportsStreaming: true,
|
|
33
|
+
supportsTools: true,
|
|
34
|
+
contextWindow: 2000000,
|
|
35
|
+
maxOutputTokens: 65000,
|
|
36
|
+
avgLatencyMs: 1100,
|
|
37
|
+
defaultScore: 0.95,
|
|
38
|
+
},
|
|
39
|
+
{
|
|
40
|
+
model: "anthropic/claude-opus-4-6",
|
|
41
|
+
provider: "openrouter",
|
|
42
|
+
capabilities: ["deep_reasoning", "architecture", "code_generation", "tool_use", "streaming"],
|
|
43
|
+
supportsStreaming: true,
|
|
44
|
+
supportsTools: true,
|
|
45
|
+
contextWindow: 200000,
|
|
46
|
+
maxOutputTokens: 32000,
|
|
47
|
+
avgLatencyMs: 2200,
|
|
48
|
+
defaultScore: 0.97,
|
|
49
|
+
},
|
|
50
|
+
// --- Strong ---
|
|
51
|
+
{
|
|
52
|
+
model: "anthropic/claude-sonnet-4-5-20250929",
|
|
53
|
+
provider: "openrouter",
|
|
54
|
+
capabilities: ["deep_reasoning", "code_generation", "tool_use", "streaming"],
|
|
55
|
+
supportsStreaming: true,
|
|
56
|
+
supportsTools: true,
|
|
57
|
+
contextWindow: 200000,
|
|
58
|
+
maxOutputTokens: 16000,
|
|
59
|
+
avgLatencyMs: 1500,
|
|
60
|
+
defaultScore: 0.93,
|
|
61
|
+
},
|
|
62
|
+
{
|
|
63
|
+
model: "openai/gpt-5.2",
|
|
64
|
+
provider: "openrouter",
|
|
65
|
+
capabilities: ["deep_reasoning", "code_generation", "streaming"],
|
|
66
|
+
supportsStreaming: true,
|
|
67
|
+
supportsTools: true,
|
|
68
|
+
contextWindow: 200000,
|
|
69
|
+
maxOutputTokens: 24000,
|
|
70
|
+
avgLatencyMs: 1400,
|
|
71
|
+
defaultScore: 0.90,
|
|
72
|
+
},
|
|
73
|
+
{
|
|
74
|
+
model: "openai/gpt-4o",
|
|
75
|
+
provider: "openrouter",
|
|
76
|
+
capabilities: ["code_generation", "streaming", "fast_response"],
|
|
77
|
+
supportsStreaming: true,
|
|
78
|
+
supportsTools: true,
|
|
79
|
+
contextWindow: 128000,
|
|
80
|
+
maxOutputTokens: 16000,
|
|
81
|
+
avgLatencyMs: 800,
|
|
82
|
+
defaultScore: 0.88,
|
|
83
|
+
},
|
|
84
|
+
// --- Fast / cheap ---
|
|
85
|
+
{
|
|
86
|
+
model: "openai/o4-mini",
|
|
87
|
+
provider: "openrouter",
|
|
88
|
+
capabilities: ["routing", "tool_use", "fast_response", "streaming"],
|
|
89
|
+
supportsStreaming: true,
|
|
90
|
+
supportsTools: true,
|
|
91
|
+
contextWindow: 128000,
|
|
92
|
+
maxOutputTokens: 16000,
|
|
93
|
+
avgLatencyMs: 650,
|
|
94
|
+
defaultScore: 0.82,
|
|
95
|
+
},
|
|
96
|
+
{
|
|
97
|
+
model: "anthropic/claude-haiku-3.5",
|
|
98
|
+
provider: "openrouter",
|
|
99
|
+
capabilities: ["code_generation", "fast_response", "streaming"],
|
|
100
|
+
supportsStreaming: true,
|
|
101
|
+
supportsTools: true,
|
|
102
|
+
contextWindow: 200000,
|
|
103
|
+
maxOutputTokens: 8000,
|
|
104
|
+
avgLatencyMs: 400,
|
|
105
|
+
defaultScore: 0.78,
|
|
106
|
+
},
|
|
107
|
+
];
|
|
108
|
+
const OPENROUTER_PRICING = {
|
|
109
|
+
"openai/gpt-5.3-codex": { inputPer1kUsd: 0.015, outputPer1kUsd: 0.06 },
|
|
110
|
+
"anthropic/claude-opus-4-6": { inputPer1kUsd: 0.015, outputPer1kUsd: 0.075 },
|
|
111
|
+
"anthropic/claude-sonnet-5": { inputPer1kUsd: 0.004, outputPer1kUsd: 0.02 },
|
|
112
|
+
"google/gemini-3.0-pro": { inputPer1kUsd: 0.0015, outputPer1kUsd: 0.01 },
|
|
113
|
+
"anthropic/claude-sonnet-4-5-20250929": { inputPer1kUsd: 0.003, outputPer1kUsd: 0.015 },
|
|
114
|
+
"openai/gpt-5.2": { inputPer1kUsd: 0.01, outputPer1kUsd: 0.04 },
|
|
115
|
+
"openai/gpt-4o": { inputPer1kUsd: 0.0025, outputPer1kUsd: 0.01 },
|
|
116
|
+
"openai/o4-mini": { inputPer1kUsd: 0.003, outputPer1kUsd: 0.012 },
|
|
117
|
+
"anthropic/claude-haiku-3.5": { inputPer1kUsd: 0.0008, outputPer1kUsd: 0.004 },
|
|
118
|
+
};
|
|
119
|
+
export function createProviderStack() {
|
|
120
|
+
const apiKey = process.env.OPENROUTER_API_KEY;
|
|
121
|
+
if (!apiKey) {
|
|
122
|
+
throw new Error("OPENROUTER_API_KEY not set. Run `duocode login` first.");
|
|
123
|
+
}
|
|
124
|
+
const provider = new OpenRouterProvider({ apiKey, appName: "DuoCode" });
|
|
125
|
+
const costTracker = new CostTracker({ ...DEFAULT_MODEL_PRICING, ...OPENROUTER_PRICING });
|
|
126
|
+
const capabilityRouter = new CapabilityRouter(OPENROUTER_PROFILES);
|
|
127
|
+
const router = new ProviderRouter([provider], capabilityRouter, costTracker);
|
|
128
|
+
return { router, costTracker, capabilityRouter };
|
|
129
|
+
}
|
|
130
|
+
//# sourceMappingURL=factory.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"factory.js","sourceRoot":"","sources":["../../src/providers/factory.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AACrD,OAAO,EAAE,gBAAgB,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC/D,OAAO,EAAE,WAAW,EAAE,qBAAqB,EAAE,MAAM,mBAAmB,CAAC;AAGvE,MAAM,mBAAmB,GAAmB;IAC1C,mBAAmB;IACnB;QACE,KAAK,EAAE,sBAAsB;QAC7B,QAAQ,EAAE,YAAY;QACtB,YAAY,EAAE,CAAC,gBAAgB,EAAE,iBAAiB,EAAE,UAAU,EAAE,WAAW,CAAC;QAC5E,iBAAiB,EAAE,IAAI;QACvB,aAAa,EAAE,IAAI;QACnB,aAAa,EAAE,MAAO;QACtB,eAAe,EAAE,KAAM;QACvB,YAAY,EAAE,IAAI;QAClB,YAAY,EAAE,IAAI;KACnB;IACD;QACE,KAAK,EAAE,2BAA2B;QAClC,QAAQ,EAAE,YAAY;QACtB,YAAY,EAAE,CAAC,gBAAgB,EAAE,cAAc,EAAE,iBAAiB,EAAE,UAAU,EAAE,WAAW,CAAC;QAC5F,iBAAiB,EAAE,IAAI;QACvB,aAAa,EAAE,IAAI;QACnB,aAAa,EAAE,MAAO;QACtB,eAAe,EAAE,KAAM;QACvB,YAAY,EAAE,IAAI;QAClB,YAAY,EAAE,IAAI;KACnB;IACD;QACE,KAAK,EAAE,uBAAuB;QAC9B,QAAQ,EAAE,YAAY;QACtB,YAAY,EAAE,CAAC,gBAAgB,EAAE,iBAAiB,EAAE,WAAW,EAAE,UAAU,CAAC;QAC5E,iBAAiB,EAAE,IAAI;QACvB,aAAa,EAAE,IAAI;QACnB,aAAa,EAAE,OAAS;QACxB,eAAe,EAAE,KAAM;QACvB,YAAY,EAAE,IAAI;QAClB,YAAY,EAAE,IAAI;KACnB;IACD;QACE,KAAK,EAAE,2BAA2B;QAClC,QAAQ,EAAE,YAAY;QACtB,YAAY,EAAE,CAAC,gBAAgB,EAAE,cAAc,EAAE,iBAAiB,EAAE,UAAU,EAAE,WAAW,CAAC;QAC5F,iBAAiB,EAAE,IAAI;QACvB,aAAa,EAAE,IAAI;QACnB,aAAa,EAAE,MAAO;QACtB,eAAe,EAAE,KAAM;QACvB,YAAY,EAAE,IAAI;QAClB,YAAY,EAAE,IAAI;KACnB;IACD,iBAAiB;IACjB;QACE,KAAK,EAAE,sCAAsC;QAC7C,QAAQ,EAAE,YAAY;QACtB,YAAY,EAAE,CAAC,gBAAgB,EAAE,iBAAiB,EAAE,UAAU,EAAE,WAAW,CAAC;QAC5E,iBAAiB,EAAE,IAAI;QACvB,aAAa,EAAE,IAAI;QACnB,aAAa,EAAE,MAAO;QACtB,eAAe,EAAE,KAAM;QACvB,YAAY,EAAE,IAAI;QAClB,YAAY,EAAE,IAAI;KACnB;IACD;QACE,KAAK,EAAE,gBAAgB;QACvB,QAAQ,EAAE,YAAY;QACtB,YAAY,EAAE,CAAC,gBAAgB,EAAE,iBAAiB,EAAE,WAAW,CAAC;QAChE,iBAAiB,EAAE,IAAI;QACvB,aAAa,EAAE,IAAI;QACnB,aAAa,EAAE,MAAO;QACtB,eAAe,EAAE,KAAM;QACvB,YAAY,EAAE,IAAI;QAClB,YAAY,EAAE,IAAI;KACnB;IACD;QACE,KAAK,EAAE,eAAe;QACtB,QAAQ,EAAE,YAAY;QACtB,YAAY,EAAE,CAAC,iBAAiB,EAAE,WAAW,EAAE,eAAe,CAAC;QAC/D,iBAAiB,EAAE,IAAI;QACvB,aAAa,EAAE,IAAI;QACnB,aAAa,EAAE,MAAO;QACtB,eAAe,EAAE,KAAM;QACvB,YAAY,EAAE,GAAG;QACjB,YAAY,EAAE,IAAI;KACnB;IACD,uBAAuB;IACvB;QACE,KAAK,EAAE,gBAAgB;QACvB,QAAQ,EAAE,YAAY;QACtB,YAAY,EAAE,CAAC,SAAS,EAAE,UAAU,EAAE,eAAe,EAAE,WAAW,CAAC;QACnE,iBAAiB,EAAE,IAAI;QACvB,aAAa,EAAE,IAAI;QACnB,aAAa,EAAE,MAAO;QACtB,eAAe,EAAE,KAAM;QACvB,YAAY,EAAE,GAAG;QACjB,YAAY,EAAE,IAAI;KACnB;IACD;QACE,KAAK,EAAE,4BAA4B;QACnC,QAAQ,EAAE,YAAY;QACtB,YAAY,EAAE,CAAC,iBAAiB,EAAE,eAAe,EAAE,WAAW,CAAC;QAC/D,iBAAiB,EAAE,IAAI;QACvB,aAAa,EAAE,IAAI;QACnB,aAAa,EAAE,MAAO;QACtB,eAAe,EAAE,IAAK;QACtB,YAAY,EAAE,GAAG;QACjB,YAAY,EAAE,IAAI;KACnB;CACF,CAAC;AAEF,MAAM,kBAAkB,GAAsE;IAC5F,sBAAsB,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE;IACtE,2BAA2B,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,cAAc,EAAE,KAAK,EAAE;IAC5E,2BAA2B,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE;IAC3E,uBAAuB,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,cAAc,EAAE,IAAI,EAAE;IACxE,sCAAsC,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,cAAc,EAAE,KAAK,EAAE;IACvF,gBAAgB,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE;IAC/D,eAAe,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,cAAc,EAAE,IAAI,EAAE;IAChE,gBAAgB,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,cAAc,EAAE,KAAK,EAAE;IACjE,4BAA4B,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,cAAc,EAAE,KAAK,EAAE;CAC/E,CAAC;AAEF,MAAM,UAAU,mBAAmB;IAKjC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;IAC9C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAC;IAC5E,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,kBAAkB,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;IACxE,MAAM,WAAW,GAAG,IAAI,WAAW,CAAC,EAAE,GAAG,qBAAqB,EAAE,GAAG,kBAAkB,EAAE,CAAC,CAAC;IACzF,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,CAAC,mBAAmB,CAAC,CAAC;IACnE,MAAM,MAAM,GAAG,IAAI,cAAc,CAAC,CAAC,QAAQ,CAAC,EAAE,gBAAgB,EAAE,WAAW,CAAC,CAAC;IAE7E,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,gBAAgB,EAAE,CAAC;AACnD,CAAC"}
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
import type { ProviderRouter } from "./router.js";
|
|
2
|
+
import { type DuoRole, type GradingResult } from "../duo/duo_types.js";
|
|
3
|
+
export declare class GradingRubric {
|
|
4
|
+
private readonly router;
|
|
5
|
+
constructor(router: ProviderRouter);
|
|
6
|
+
grade(opts: {
|
|
7
|
+
code: string;
|
|
8
|
+
taskDescription: string;
|
|
9
|
+
model: string;
|
|
10
|
+
role: DuoRole;
|
|
11
|
+
taskId?: string;
|
|
12
|
+
judgeModel?: string;
|
|
13
|
+
systemPrompt?: string;
|
|
14
|
+
}): Promise<GradingResult>;
|
|
15
|
+
gradeArchitectPlan(opts: {
|
|
16
|
+
plan: string;
|
|
17
|
+
taskDescription: string;
|
|
18
|
+
model: string;
|
|
19
|
+
taskId?: string;
|
|
20
|
+
judgeModel?: string;
|
|
21
|
+
}): Promise<GradingResult>;
|
|
22
|
+
compareOutputs(opts: {
|
|
23
|
+
codeA: string;
|
|
24
|
+
codeB: string;
|
|
25
|
+
taskDescription: string;
|
|
26
|
+
modelA: string;
|
|
27
|
+
modelB: string;
|
|
28
|
+
role: DuoRole;
|
|
29
|
+
taskId?: string;
|
|
30
|
+
}): Promise<{
|
|
31
|
+
gradeA: GradingResult;
|
|
32
|
+
gradeB: GradingResult;
|
|
33
|
+
winner: string;
|
|
34
|
+
isDraw: boolean;
|
|
35
|
+
scoreDelta: number;
|
|
36
|
+
}>;
|
|
37
|
+
}
|