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,258 @@
|
|
|
1
|
+
import Anthropic from "@anthropic-ai/sdk";
|
|
2
|
+
function toAnthropicInput(messages) {
|
|
3
|
+
const systemParts = [];
|
|
4
|
+
const converted = [];
|
|
5
|
+
for (const message of messages) {
|
|
6
|
+
if (message.role === "system") {
|
|
7
|
+
systemParts.push(message.content);
|
|
8
|
+
continue;
|
|
9
|
+
}
|
|
10
|
+
if (message.role === "tool") {
|
|
11
|
+
const toolResult = {
|
|
12
|
+
type: "tool_result",
|
|
13
|
+
tool_use_id: message.toolCallId ?? "tool_call",
|
|
14
|
+
content: message.content,
|
|
15
|
+
};
|
|
16
|
+
converted.push({
|
|
17
|
+
role: "user",
|
|
18
|
+
content: [toolResult],
|
|
19
|
+
});
|
|
20
|
+
continue;
|
|
21
|
+
}
|
|
22
|
+
// Assistant message carrying tool calls
|
|
23
|
+
if (message.role === "assistant" &&
|
|
24
|
+
message.toolCalls &&
|
|
25
|
+
message.toolCalls.length > 0) {
|
|
26
|
+
const blocks = [];
|
|
27
|
+
if (message.content) {
|
|
28
|
+
blocks.push({ type: "text", text: message.content });
|
|
29
|
+
}
|
|
30
|
+
for (const tc of message.toolCalls) {
|
|
31
|
+
blocks.push({
|
|
32
|
+
type: "tool_use",
|
|
33
|
+
id: tc.id,
|
|
34
|
+
name: tc.name,
|
|
35
|
+
input: tc.input,
|
|
36
|
+
});
|
|
37
|
+
}
|
|
38
|
+
converted.push({ role: "assistant", content: blocks });
|
|
39
|
+
continue;
|
|
40
|
+
}
|
|
41
|
+
converted.push({
|
|
42
|
+
role: message.role === "assistant" ? "assistant" : "user",
|
|
43
|
+
content: message.content,
|
|
44
|
+
});
|
|
45
|
+
}
|
|
46
|
+
return {
|
|
47
|
+
system: systemParts.length > 0 ? systemParts.join("\n\n") : undefined,
|
|
48
|
+
messages: converted,
|
|
49
|
+
};
|
|
50
|
+
}
|
|
51
|
+
function toAnthropicTools(tools) {
|
|
52
|
+
if (!tools || tools.length === 0) {
|
|
53
|
+
return undefined;
|
|
54
|
+
}
|
|
55
|
+
return tools.map((tool) => ({
|
|
56
|
+
name: tool.name,
|
|
57
|
+
description: tool.description,
|
|
58
|
+
input_schema: tool.inputSchema,
|
|
59
|
+
}));
|
|
60
|
+
}
|
|
61
|
+
function toAnthropicToolChoice(choice) {
|
|
62
|
+
if (choice === "required")
|
|
63
|
+
return { type: "any" };
|
|
64
|
+
if (choice === "none")
|
|
65
|
+
return { type: "auto" };
|
|
66
|
+
return undefined;
|
|
67
|
+
}
|
|
68
|
+
function toAnthropicMetadata(metadata) {
|
|
69
|
+
if (!metadata)
|
|
70
|
+
return undefined;
|
|
71
|
+
const userId = metadata.user_id;
|
|
72
|
+
if (typeof userId === "string")
|
|
73
|
+
return { user_id: userId };
|
|
74
|
+
return undefined;
|
|
75
|
+
}
|
|
76
|
+
export class AnthropicProvider {
|
|
77
|
+
name = "anthropic";
|
|
78
|
+
client;
|
|
79
|
+
constructor(params) {
|
|
80
|
+
this.client = new Anthropic({
|
|
81
|
+
apiKey: params?.apiKey ?? process.env.ANTHROPIC_API_KEY,
|
|
82
|
+
baseURL: params?.baseURL,
|
|
83
|
+
});
|
|
84
|
+
}
|
|
85
|
+
supportsModel(model) {
|
|
86
|
+
// Anthropic only supports Claude models
|
|
87
|
+
if (model.startsWith("claude") || model.includes("/claude"))
|
|
88
|
+
return true;
|
|
89
|
+
// Reject anything that looks like an OpenAI model
|
|
90
|
+
if (model.startsWith("gpt") || model.startsWith("o4") || model.startsWith("openai/"))
|
|
91
|
+
return false;
|
|
92
|
+
// Accept unknown models cautiously (user may use custom fine-tunes)
|
|
93
|
+
return false;
|
|
94
|
+
}
|
|
95
|
+
async generate(request) {
|
|
96
|
+
const input = toAnthropicInput(request.messages);
|
|
97
|
+
const params = {
|
|
98
|
+
model: request.model,
|
|
99
|
+
system: input.system,
|
|
100
|
+
messages: input.messages,
|
|
101
|
+
tools: toAnthropicTools(request.tools),
|
|
102
|
+
tool_choice: toAnthropicToolChoice(request.toolChoice),
|
|
103
|
+
temperature: request.temperature,
|
|
104
|
+
max_tokens: request.maxOutputTokens ?? 2048,
|
|
105
|
+
metadata: toAnthropicMetadata(request.metadata),
|
|
106
|
+
};
|
|
107
|
+
const response = await this.client.messages.create(params);
|
|
108
|
+
const textParts = [];
|
|
109
|
+
const toolCalls = [];
|
|
110
|
+
for (const block of response.content) {
|
|
111
|
+
if (block.type === "text") {
|
|
112
|
+
textParts.push(block.text);
|
|
113
|
+
}
|
|
114
|
+
if (block.type === "tool_use") {
|
|
115
|
+
toolCalls.push({
|
|
116
|
+
id: block.id,
|
|
117
|
+
name: block.name,
|
|
118
|
+
input: (block.input ?? {}),
|
|
119
|
+
});
|
|
120
|
+
}
|
|
121
|
+
}
|
|
122
|
+
return {
|
|
123
|
+
provider: this.name,
|
|
124
|
+
model: response.model,
|
|
125
|
+
text: textParts.join(""),
|
|
126
|
+
toolCalls,
|
|
127
|
+
stopReason: response.stop_reason ?? null,
|
|
128
|
+
usage: {
|
|
129
|
+
inputTokens: response.usage.input_tokens ?? 0,
|
|
130
|
+
outputTokens: response.usage.output_tokens ?? 0,
|
|
131
|
+
cacheReadInputTokens: response.usage.cache_read_input_tokens ?? 0,
|
|
132
|
+
cacheWriteInputTokens: response.usage.cache_creation_input_tokens ?? 0,
|
|
133
|
+
},
|
|
134
|
+
raw: response,
|
|
135
|
+
};
|
|
136
|
+
}
|
|
137
|
+
async *stream(request) {
|
|
138
|
+
const input = toAnthropicInput(request.messages);
|
|
139
|
+
const params = {
|
|
140
|
+
model: request.model,
|
|
141
|
+
system: input.system,
|
|
142
|
+
messages: input.messages,
|
|
143
|
+
tools: toAnthropicTools(request.tools),
|
|
144
|
+
tool_choice: toAnthropicToolChoice(request.toolChoice),
|
|
145
|
+
temperature: request.temperature,
|
|
146
|
+
max_tokens: request.maxOutputTokens ?? 2048,
|
|
147
|
+
metadata: toAnthropicMetadata(request.metadata),
|
|
148
|
+
stream: true,
|
|
149
|
+
};
|
|
150
|
+
const stream = await this.client.messages.create(params);
|
|
151
|
+
const textParts = [];
|
|
152
|
+
const toolBlocks = new Map();
|
|
153
|
+
let usage = { inputTokens: 0, outputTokens: 0, cacheReadInputTokens: 0, cacheWriteInputTokens: 0 };
|
|
154
|
+
let model = request.model;
|
|
155
|
+
let stopReason = null;
|
|
156
|
+
try {
|
|
157
|
+
for await (const event of stream) {
|
|
158
|
+
switch (event.type) {
|
|
159
|
+
case "message_start": {
|
|
160
|
+
model = event.message?.model ?? model;
|
|
161
|
+
const msgUsage = event.message?.usage;
|
|
162
|
+
usage = {
|
|
163
|
+
...usage,
|
|
164
|
+
inputTokens: msgUsage?.input_tokens ?? usage.inputTokens,
|
|
165
|
+
outputTokens: msgUsage?.output_tokens ?? usage.outputTokens,
|
|
166
|
+
cacheReadInputTokens: msgUsage?.cache_read_input_tokens ?? usage.cacheReadInputTokens,
|
|
167
|
+
cacheWriteInputTokens: msgUsage?.cache_creation_input_tokens ?? usage.cacheWriteInputTokens,
|
|
168
|
+
};
|
|
169
|
+
break;
|
|
170
|
+
}
|
|
171
|
+
case "message_delta": {
|
|
172
|
+
stopReason = event.delta?.stop_reason ?? stopReason;
|
|
173
|
+
usage = {
|
|
174
|
+
...usage,
|
|
175
|
+
outputTokens: event.usage?.output_tokens ?? usage.outputTokens,
|
|
176
|
+
};
|
|
177
|
+
break;
|
|
178
|
+
}
|
|
179
|
+
case "content_block_start": {
|
|
180
|
+
if (event.content_block?.type === "tool_use") {
|
|
181
|
+
toolBlocks.set(event.index, {
|
|
182
|
+
id: String(event.content_block.id ?? event.index),
|
|
183
|
+
name: String(event.content_block.name ?? ""),
|
|
184
|
+
inputText: "",
|
|
185
|
+
});
|
|
186
|
+
}
|
|
187
|
+
break;
|
|
188
|
+
}
|
|
189
|
+
case "content_block_delta": {
|
|
190
|
+
if (event.delta?.type === "text_delta") {
|
|
191
|
+
const delta = event.delta.text;
|
|
192
|
+
textParts.push(delta);
|
|
193
|
+
yield { type: "text-delta", delta };
|
|
194
|
+
}
|
|
195
|
+
if (event.delta?.type === "input_json_delta") {
|
|
196
|
+
const existing = toolBlocks.get(event.index);
|
|
197
|
+
if (existing) {
|
|
198
|
+
existing.inputText += event.delta.partial_json ?? "";
|
|
199
|
+
toolBlocks.set(event.index, existing);
|
|
200
|
+
}
|
|
201
|
+
}
|
|
202
|
+
break;
|
|
203
|
+
}
|
|
204
|
+
default:
|
|
205
|
+
break;
|
|
206
|
+
}
|
|
207
|
+
}
|
|
208
|
+
const parsedCalls = [...toolBlocks.values()].map((block) => {
|
|
209
|
+
let inputData = {};
|
|
210
|
+
if (block.inputText.length > 0) {
|
|
211
|
+
try {
|
|
212
|
+
inputData = JSON.parse(block.inputText);
|
|
213
|
+
}
|
|
214
|
+
catch {
|
|
215
|
+
inputData = { __raw: block.inputText };
|
|
216
|
+
}
|
|
217
|
+
}
|
|
218
|
+
return {
|
|
219
|
+
id: block.id,
|
|
220
|
+
name: block.name,
|
|
221
|
+
input: inputData,
|
|
222
|
+
};
|
|
223
|
+
});
|
|
224
|
+
yield {
|
|
225
|
+
type: "usage",
|
|
226
|
+
usage: {
|
|
227
|
+
inputTokens: usage.inputTokens,
|
|
228
|
+
outputTokens: usage.outputTokens,
|
|
229
|
+
cacheReadInputTokens: usage.cacheReadInputTokens,
|
|
230
|
+
cacheWriteInputTokens: usage.cacheWriteInputTokens,
|
|
231
|
+
},
|
|
232
|
+
};
|
|
233
|
+
for (const call of parsedCalls) {
|
|
234
|
+
yield { type: "tool-call", call };
|
|
235
|
+
}
|
|
236
|
+
const response = {
|
|
237
|
+
provider: this.name,
|
|
238
|
+
model,
|
|
239
|
+
text: textParts.join(""),
|
|
240
|
+
toolCalls: parsedCalls,
|
|
241
|
+
stopReason,
|
|
242
|
+
usage: {
|
|
243
|
+
inputTokens: usage.inputTokens,
|
|
244
|
+
outputTokens: usage.outputTokens,
|
|
245
|
+
cacheReadInputTokens: usage.cacheReadInputTokens,
|
|
246
|
+
cacheWriteInputTokens: usage.cacheWriteInputTokens,
|
|
247
|
+
},
|
|
248
|
+
};
|
|
249
|
+
yield { type: "done", response };
|
|
250
|
+
}
|
|
251
|
+
catch (error) {
|
|
252
|
+
const normalized = error instanceof Error ? error : new Error(String(error));
|
|
253
|
+
yield { type: "error", error: normalized };
|
|
254
|
+
throw normalized;
|
|
255
|
+
}
|
|
256
|
+
}
|
|
257
|
+
}
|
|
258
|
+
//# sourceMappingURL=anthropic.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../src/providers/anthropic.ts"],"names":[],"mappings":"AAAA,OAAO,SAAS,MAAM,mBAAmB,CAAC;AAU1C,SAAS,gBAAgB,CAAC,QAAuB;IAI/C,MAAM,WAAW,GAAa,EAAE,CAAC;IACjC,MAAM,SAAS,GAA6B,EAAE,CAAC;IAE/C,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC9B,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAClC,SAAS;QACX,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC5B,MAAM,UAAU,GAAmC;gBACjD,IAAI,EAAE,aAAa;gBACnB,WAAW,EAAE,OAAO,CAAC,UAAU,IAAI,WAAW;gBAC9C,OAAO,EAAE,OAAO,CAAC,OAAO;aACzB,CAAC;YACF,SAAS,CAAC,IAAI,CAAC;gBACb,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,CAAC,UAAU,CAAC;aACtB,CAAC,CAAC;YACH,SAAS;QACX,CAAC;QAED,wCAAwC;QACxC,IACE,OAAO,CAAC,IAAI,KAAK,WAAW;YAC5B,OAAO,CAAC,SAAS;YACjB,OAAO,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAC5B,CAAC;YACD,MAAM,MAAM,GAAkE,EAAE,CAAC;YACjF,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBACpB,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;YACvD,CAAC;YACD,KAAK,MAAM,EAAE,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;gBACnC,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,UAAU;oBAChB,EAAE,EAAE,EAAE,CAAC,EAAE;oBACT,IAAI,EAAE,EAAE,CAAC,IAAI;oBACb,KAAK,EAAE,EAAE,CAAC,KAAK;iBAChB,CAAC,CAAC;YACL,CAAC;YACD,SAAS,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;YACvD,SAAS;QACX,CAAC;QAED,SAAS,CAAC,IAAI,CAAC;YACb,IAAI,EAAE,OAAO,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM;YACzD,OAAO,EAAE,OAAO,CAAC,OAAO;SACzB,CAAC,CAAC;IACL,CAAC;IAED,OAAO;QACL,MAAM,EAAE,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS;QACrE,QAAQ,EAAE,SAAS;KACpB,CAAC;AACJ,CAAC;AAED,SAAS,gBAAgB,CAAC,KAA+B;IACvD,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,EAAkB,EAAE,CAAC,CAAC;QAC1C,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,WAAW,EAAE,IAAI,CAAC,WAAW;QAC7B,YAAY,EAAE,IAAI,CAAC,WAAyC;KAC7D,CAAC,CAAC,CAAC;AACN,CAAC;AAED,SAAS,qBAAqB,CAAC,MAAqC;IAClE,IAAI,MAAM,KAAK,UAAU;QAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;IAClD,IAAI,MAAM,KAAK,MAAM;QAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;IAC/C,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,mBAAmB,CAAC,QAAqC;IAChE,IAAI,CAAC,QAAQ;QAAE,OAAO,SAAS,CAAC;IAChC,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC;IAChC,IAAI,OAAO,MAAM,KAAK,QAAQ;QAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;IAC3D,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,OAAO,iBAAiB;IACnB,IAAI,GAAG,WAAoB,CAAC;IAEpB,MAAM,CAAY;IAEnC,YAAY,MAA8C;QACxD,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC;YAC1B,MAAM,EAAE,MAAM,EAAE,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB;YACvD,OAAO,EAAE,MAAM,EAAE,OAAO;SACzB,CAAC,CAAC;IACL,CAAC;IAED,aAAa,CAAC,KAAa;QACzB,wCAAwC;QACxC,IAAI,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC;YAAE,OAAO,IAAI,CAAC;QACzE,kDAAkD;QAClD,IAAI,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,UAAU,CAAC,SAAS,CAAC;YAAE,OAAO,KAAK,CAAC;QACnG,oEAAoE;QACpE,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAAwB;QACrC,MAAM,KAAK,GAAG,gBAAgB,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAEjD,MAAM,MAAM,GAA8C;YACxD,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,KAAK,EAAE,gBAAgB,CAAC,OAAO,CAAC,KAAK,CAAC;YACtC,WAAW,EAAE,qBAAqB,CAAC,OAAO,CAAC,UAAU,CAAC;YACtD,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,UAAU,EAAE,OAAO,CAAC,eAAe,IAAI,IAAI;YAC3C,QAAQ,EAAE,mBAAmB,CAAC,OAAO,CAAC,QAAQ,CAAC;SAChD,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAE3D,MAAM,SAAS,GAAa,EAAE,CAAC;QAC/B,MAAM,SAAS,GAAe,EAAE,CAAC;QAEjC,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACrC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC1B,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAC7B,CAAC;YAED,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC9B,SAAS,CAAC,IAAI,CAAC;oBACb,EAAE,EAAE,KAAK,CAAC,EAAE;oBACZ,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,KAAK,EAAE,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE,CAA4B;iBACtD,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO;YACL,QAAQ,EAAE,IAAI,CAAC,IAAI;YACnB,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,IAAI,EAAE,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;YACxB,SAAS;YACT,UAAU,EAAE,QAAQ,CAAC,WAAW,IAAI,IAAI;YACxC,KAAK,EAAE;gBACL,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC;gBAC7C,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC;gBAC/C,oBAAoB,EAAE,QAAQ,CAAC,KAAK,CAAC,uBAAuB,IAAI,CAAC;gBACjE,qBAAqB,EAAE,QAAQ,CAAC,KAAK,CAAC,2BAA2B,IAAI,CAAC;aACvE;YACD,GAAG,EAAE,QAAQ;SACd,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,OAAwB;QACpC,MAAM,KAAK,GAAG,gBAAgB,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAEjD,MAAM,MAAM,GAA2C;YACrD,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,KAAK,EAAE,gBAAgB,CAAC,OAAO,CAAC,KAAK,CAAC;YACtC,WAAW,EAAE,qBAAqB,CAAC,OAAO,CAAC,UAAU,CAAC;YACtD,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,UAAU,EAAE,OAAO,CAAC,eAAe,IAAI,IAAI;YAC3C,QAAQ,EAAE,mBAAmB,CAAC,OAAO,CAAC,QAAQ,CAAC;YAC/C,MAAM,EAAE,IAAI;SACb,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAEzD,MAAM,SAAS,GAAa,EAAE,CAAC;QAC/B,MAAM,UAAU,GAAG,IAAI,GAAG,EAA2D,CAAC;QACtF,IAAI,KAAK,GAAG,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,oBAAoB,EAAE,CAAC,EAAE,qBAAqB,EAAE,CAAC,EAAE,CAAC;QACnG,IAAI,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC1B,IAAI,UAAU,GAAkB,IAAI,CAAC;QAErC,IAAI,CAAC;YACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBACjC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;oBACnB,KAAK,eAAe,CAAC,CAAC,CAAC;wBACrB,KAAK,GAAG,KAAK,CAAC,OAAO,EAAE,KAAK,IAAI,KAAK,CAAC;wBACtC,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC;wBACtC,KAAK,GAAG;4BACN,GAAG,KAAK;4BACR,WAAW,EAAE,QAAQ,EAAE,YAAY,IAAI,KAAK,CAAC,WAAW;4BACxD,YAAY,EAAE,QAAQ,EAAE,aAAa,IAAI,KAAK,CAAC,YAAY;4BAC3D,oBAAoB,EAAE,QAAQ,EAAE,uBAAuB,IAAI,KAAK,CAAC,oBAAoB;4BACrF,qBAAqB,EAAE,QAAQ,EAAE,2BAA2B,IAAI,KAAK,CAAC,qBAAqB;yBAC5F,CAAC;wBACF,MAAM;oBACR,CAAC;oBACD,KAAK,eAAe,CAAC,CAAC,CAAC;wBACrB,UAAU,GAAG,KAAK,CAAC,KAAK,EAAE,WAAW,IAAI,UAAU,CAAC;wBACpD,KAAK,GAAG;4BACN,GAAG,KAAK;4BACR,YAAY,EAAE,KAAK,CAAC,KAAK,EAAE,aAAa,IAAI,KAAK,CAAC,YAAY;yBAC/D,CAAC;wBACF,MAAM;oBACR,CAAC;oBACD,KAAK,qBAAqB,CAAC,CAAC,CAAC;wBAC3B,IAAI,KAAK,CAAC,aAAa,EAAE,IAAI,KAAK,UAAU,EAAE,CAAC;4BAC7C,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE;gCAC1B,EAAE,EAAE,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE,IAAI,KAAK,CAAC,KAAK,CAAC;gCACjD,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,IAAI,EAAE,CAAC;gCAC5C,SAAS,EAAE,EAAE;6BACd,CAAC,CAAC;wBACL,CAAC;wBACD,MAAM;oBACR,CAAC;oBACD,KAAK,qBAAqB,CAAC,CAAC,CAAC;wBAC3B,IAAI,KAAK,CAAC,KAAK,EAAE,IAAI,KAAK,YAAY,EAAE,CAAC;4BACvC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC;4BAC/B,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;4BACtB,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC;wBACtC,CAAC;wBAED,IAAI,KAAK,CAAC,KAAK,EAAE,IAAI,KAAK,kBAAkB,EAAE,CAAC;4BAC7C,MAAM,QAAQ,GAAG,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;4BAC7C,IAAI,QAAQ,EAAE,CAAC;gCACb,QAAQ,CAAC,SAAS,IAAI,KAAK,CAAC,KAAK,CAAC,YAAY,IAAI,EAAE,CAAC;gCACrD,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;4BACxC,CAAC;wBACH,CAAC;wBACD,MAAM;oBACR,CAAC;oBACD;wBACE,MAAM;gBACV,CAAC;YACH,CAAC;YAED,MAAM,WAAW,GAAe,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;gBACrE,IAAI,SAAS,GAA4B,EAAE,CAAC;gBAE5C,IAAI,KAAK,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC/B,IAAI,CAAC;wBACH,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,CAA4B,CAAC;oBACrE,CAAC;oBAAC,MAAM,CAAC;wBACP,SAAS,GAAG,EAAE,KAAK,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC;oBACzC,CAAC;gBACH,CAAC;gBAED,OAAO;oBACL,EAAE,EAAE,KAAK,CAAC,EAAE;oBACZ,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,KAAK,EAAE,SAAS;iBACjB,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,MAAM;gBACJ,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACL,WAAW,EAAE,KAAK,CAAC,WAAW;oBAC9B,YAAY,EAAE,KAAK,CAAC,YAAY;oBAChC,oBAAoB,EAAE,KAAK,CAAC,oBAAoB;oBAChD,qBAAqB,EAAE,KAAK,CAAC,qBAAqB;iBACnD;aACF,CAAC;YAEF,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;gBAC/B,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;YACpC,CAAC;YAED,MAAM,QAAQ,GAAqB;gBACjC,QAAQ,EAAE,IAAI,CAAC,IAAI;gBACnB,KAAK;gBACL,IAAI,EAAE,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;gBACxB,SAAS,EAAE,WAAW;gBACtB,UAAU;gBACV,KAAK,EAAE;oBACL,WAAW,EAAE,KAAK,CAAC,WAAW;oBAC9B,YAAY,EAAE,KAAK,CAAC,YAAY;oBAChC,oBAAoB,EAAE,KAAK,CAAC,oBAAoB;oBAChD,qBAAqB,EAAE,KAAK,CAAC,qBAAqB;iBACnD;aACF,CAAC;YAEF,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC;QACnC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,UAAU,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YAC7E,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;YAC3C,MAAM,UAAU,CAAC;QACnB,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,42 @@
|
|
|
1
|
+
import type { RouteRequest, RouteResult } from "./base.js";
|
|
2
|
+
import type { CostTracker } from "./cost_tracker.js";
|
|
3
|
+
export type AuctionMode = "single" | "hybrid" | "ab_test" | "auto";
|
|
4
|
+
export interface AgentBid {
|
|
5
|
+
bidId: string;
|
|
6
|
+
model: string;
|
|
7
|
+
provider: string;
|
|
8
|
+
score: number;
|
|
9
|
+
confidence: number;
|
|
10
|
+
estimatedCostUsd: number;
|
|
11
|
+
estimatedLatencyMs: number;
|
|
12
|
+
createdAt: number;
|
|
13
|
+
reasons: string[];
|
|
14
|
+
}
|
|
15
|
+
export interface AuctionDecision {
|
|
16
|
+
mode: Exclude<AuctionMode, "auto">;
|
|
17
|
+
winner: RouteResult;
|
|
18
|
+
shadow?: RouteResult;
|
|
19
|
+
bids: AgentBid[];
|
|
20
|
+
rationale: string;
|
|
21
|
+
}
|
|
22
|
+
export interface CapabilityAuctionOptions {
|
|
23
|
+
hybridTriggerPercent?: number;
|
|
24
|
+
}
|
|
25
|
+
export declare class CapabilityAuctionEngine {
|
|
26
|
+
private readonly hybridTriggerPercent;
|
|
27
|
+
constructor(options?: CapabilityAuctionOptions);
|
|
28
|
+
decide(params: {
|
|
29
|
+
candidates: RouteResult[];
|
|
30
|
+
request: RouteRequest;
|
|
31
|
+
mode?: AuctionMode;
|
|
32
|
+
providerHealth?: Map<string, number>;
|
|
33
|
+
abKey?: string;
|
|
34
|
+
costTracker?: CostTracker;
|
|
35
|
+
now?: number;
|
|
36
|
+
}): AuctionDecision;
|
|
37
|
+
private buildRationale;
|
|
38
|
+
private buildHybridRationale;
|
|
39
|
+
private rankCandidates;
|
|
40
|
+
private shouldTriggerHybrid;
|
|
41
|
+
private createBid;
|
|
42
|
+
}
|
|
@@ -0,0 +1,190 @@
|
|
|
1
|
+
const DEFAULT_ESTIMATED_INPUT_TOKENS = 1000;
|
|
2
|
+
const DEFAULT_ESTIMATED_OUTPUT_TOKENS = 2000;
|
|
3
|
+
function hashString(input) {
|
|
4
|
+
let hash = 0;
|
|
5
|
+
for (let i = 0; i < input.length; i += 1) {
|
|
6
|
+
hash = (hash << 5) - hash + input.charCodeAt(i);
|
|
7
|
+
hash |= 0;
|
|
8
|
+
}
|
|
9
|
+
return Math.abs(hash);
|
|
10
|
+
}
|
|
11
|
+
function formatCost(usd) {
|
|
12
|
+
if (usd < 0.01)
|
|
13
|
+
return `$${usd.toFixed(4)}`;
|
|
14
|
+
return `$${usd.toFixed(2)}`;
|
|
15
|
+
}
|
|
16
|
+
function formatLatency(ms) {
|
|
17
|
+
if (ms < 1000)
|
|
18
|
+
return `${ms}ms`;
|
|
19
|
+
return `${(ms / 1000).toFixed(1)}s`;
|
|
20
|
+
}
|
|
21
|
+
function describeTaskIntensity(request) {
|
|
22
|
+
const caps = new Set(request.requiredCapabilities ?? []);
|
|
23
|
+
const estimatedInput = request.estimatedInputTokens ?? DEFAULT_ESTIMATED_INPUT_TOKENS;
|
|
24
|
+
const estimatedOutput = request.estimatedOutputTokens ?? DEFAULT_ESTIMATED_OUTPUT_TOKENS;
|
|
25
|
+
const estimatedTotal = estimatedInput + estimatedOutput;
|
|
26
|
+
const complexCaps = [
|
|
27
|
+
"architecture", "deep_reasoning", "security_analysis",
|
|
28
|
+
"api_design", "data_modeling", "architecture_design",
|
|
29
|
+
];
|
|
30
|
+
const moderateCaps = [
|
|
31
|
+
"tool_use", "code_refactoring", "performance_optimization",
|
|
32
|
+
"test_generation", "bug_diagnosis", "code_review",
|
|
33
|
+
];
|
|
34
|
+
if (complexCaps.some((c) => caps.has(c)) || estimatedTotal >= 9000) {
|
|
35
|
+
return "complex";
|
|
36
|
+
}
|
|
37
|
+
if (moderateCaps.some((c) => caps.has(c)) || estimatedTotal >= 4500) {
|
|
38
|
+
return "moderate";
|
|
39
|
+
}
|
|
40
|
+
return "light";
|
|
41
|
+
}
|
|
42
|
+
export class CapabilityAuctionEngine {
|
|
43
|
+
hybridTriggerPercent;
|
|
44
|
+
constructor(options) {
|
|
45
|
+
this.hybridTriggerPercent = options?.hybridTriggerPercent ?? 0.15;
|
|
46
|
+
}
|
|
47
|
+
decide(params) {
|
|
48
|
+
const now = params.now ?? Date.now();
|
|
49
|
+
const mode = params.mode ?? "auto";
|
|
50
|
+
if (params.candidates.length === 0) {
|
|
51
|
+
throw new Error("Auction requires at least one candidate.");
|
|
52
|
+
}
|
|
53
|
+
const ranked = this.rankCandidates(params.candidates, params.providerHealth);
|
|
54
|
+
const bids = ranked.map((entry, index) => this.createBid(entry, index, now, params.request, params.costTracker));
|
|
55
|
+
const winner = ranked[0];
|
|
56
|
+
const winnerBid = bids[0];
|
|
57
|
+
const runnerUp = ranked[1];
|
|
58
|
+
const caps = params.request.requiredCapabilities;
|
|
59
|
+
if (mode === "single") {
|
|
60
|
+
return {
|
|
61
|
+
mode: "single",
|
|
62
|
+
winner,
|
|
63
|
+
bids,
|
|
64
|
+
rationale: this.buildRationale("single", winner, winnerBid, ranked.length, params.request),
|
|
65
|
+
};
|
|
66
|
+
}
|
|
67
|
+
if (mode === "ab_test") {
|
|
68
|
+
const topTwo = ranked.slice(0, 2);
|
|
69
|
+
if (topTwo.length === 1) {
|
|
70
|
+
return {
|
|
71
|
+
mode: "ab_test",
|
|
72
|
+
winner: topTwo[0],
|
|
73
|
+
bids,
|
|
74
|
+
rationale: `A/B test requested, but only one model was available. ${winner.model.model} was used automatically.`,
|
|
75
|
+
};
|
|
76
|
+
}
|
|
77
|
+
const key = params.abKey ?? `${caps.join(",")}:${now}`;
|
|
78
|
+
const arm = hashString(key) % 2;
|
|
79
|
+
const selected = topTwo[arm];
|
|
80
|
+
const shadow = topTwo[(arm + 1) % 2];
|
|
81
|
+
const selectedBid = bids[ranked.indexOf(selected)];
|
|
82
|
+
return {
|
|
83
|
+
mode: "ab_test",
|
|
84
|
+
winner: selected,
|
|
85
|
+
shadow,
|
|
86
|
+
bids,
|
|
87
|
+
rationale: `A/B testing is active for this task. ${selected.model.model} was selected for this run (${formatCost(selectedBid.estimatedCostUsd)}, ~${formatLatency(selectedBid.estimatedLatencyMs)}), with ${shadow.model.model} tracked as the alternate.`,
|
|
88
|
+
};
|
|
89
|
+
}
|
|
90
|
+
const shouldHybrid = mode === "hybrid" || this.shouldTriggerHybrid(winner, runnerUp);
|
|
91
|
+
if (shouldHybrid && runnerUp) {
|
|
92
|
+
const runnerBid = bids[1];
|
|
93
|
+
const scoreDelta = ((winner.score - runnerUp.score) / winner.score * 100).toFixed(1);
|
|
94
|
+
return {
|
|
95
|
+
mode: "hybrid",
|
|
96
|
+
winner,
|
|
97
|
+
shadow: runnerUp,
|
|
98
|
+
bids,
|
|
99
|
+
rationale: this.buildHybridRationale(winner, winnerBid, runnerUp, runnerBid, scoreDelta, params.request),
|
|
100
|
+
};
|
|
101
|
+
}
|
|
102
|
+
return {
|
|
103
|
+
mode: "single",
|
|
104
|
+
winner,
|
|
105
|
+
bids,
|
|
106
|
+
rationale: this.buildRationale("auto", winner, winnerBid, ranked.length, params.request),
|
|
107
|
+
};
|
|
108
|
+
}
|
|
109
|
+
buildRationale(trigger, winner, bid, candidateCount, request) {
|
|
110
|
+
const intensity = describeTaskIntensity(request);
|
|
111
|
+
const intensityExplanation = intensity === "complex"
|
|
112
|
+
? "The task looks complex, so quality and reliability were prioritized over raw price."
|
|
113
|
+
: intensity === "moderate"
|
|
114
|
+
? "The task looks moderately complex, so a balanced quality/cost model was selected."
|
|
115
|
+
: "The task looks lightweight, so a faster lower-cost model was favored.";
|
|
116
|
+
const parts = [];
|
|
117
|
+
parts.push(intensityExplanation);
|
|
118
|
+
parts.push(`${winner.model.model} ranked highest among ${candidateCount} available options for this request.`);
|
|
119
|
+
parts.push(`Estimated run cost is ${formatCost(bid.estimatedCostUsd)} with about ${formatLatency(bid.estimatedLatencyMs)} latency.`);
|
|
120
|
+
if (winner.model.contextWindow) {
|
|
121
|
+
const ctx = winner.model.contextWindow >= 1000000
|
|
122
|
+
? `${(winner.model.contextWindow / 1000000).toFixed(0)}M`
|
|
123
|
+
: `${(winner.model.contextWindow / 1000).toFixed(0)}k`;
|
|
124
|
+
parts.push(`It also provides a ${ctx} token context window for this job.`);
|
|
125
|
+
}
|
|
126
|
+
return parts.join(". ") + ".";
|
|
127
|
+
}
|
|
128
|
+
buildHybridRationale(winner, winnerBid, runnerUp, runnerBid, scoreDelta, request) {
|
|
129
|
+
const intensity = describeTaskIntensity(request);
|
|
130
|
+
const complexityLine = intensity === "complex"
|
|
131
|
+
? "This task appears complex, so we kept a second high-quality option ready."
|
|
132
|
+
: "This task does not require the most expensive model, so we chose a value-focused winner and kept a close backup.";
|
|
133
|
+
const parts = [];
|
|
134
|
+
parts.push(complexityLine);
|
|
135
|
+
parts.push(`${runnerUp.model.model} scored within ${scoreDelta}% of ${winner.model.model}, so both were considered strong choices.`);
|
|
136
|
+
parts.push(`${winner.model.model} is expected around ${formatCost(winnerBid.estimatedCostUsd)} (~${formatLatency(winnerBid.estimatedLatencyMs)}).`);
|
|
137
|
+
parts.push(`${runnerUp.model.model} is expected around ${formatCost(runnerBid.estimatedCostUsd)} (~${formatLatency(runnerBid.estimatedLatencyMs)}).`);
|
|
138
|
+
parts.push("A backup model stays ready in case the first choice fails.");
|
|
139
|
+
return parts.join(". ") + ".";
|
|
140
|
+
}
|
|
141
|
+
// Post-TOPSIS health adjustment: TOPSIS produces the base score from model
|
|
142
|
+
// quality, capability fit, latency, cost, and context fit. Provider health
|
|
143
|
+
// is a runtime reliability signal (0 = down, 1 = healthy) applied as a
|
|
144
|
+
// damping multiplier: score * (0.7 + 0.3 * health). An unhealthy provider
|
|
145
|
+
// loses up to 30% of its TOPSIS score; a fully healthy provider keeps 100%.
|
|
146
|
+
rankCandidates(candidates, providerHealth) {
|
|
147
|
+
return [...candidates].sort((a, b) => {
|
|
148
|
+
const healthA = providerHealth?.get(a.model.provider) ?? 1;
|
|
149
|
+
const healthB = providerHealth?.get(b.model.provider) ?? 1;
|
|
150
|
+
const adjustedA = a.score * (0.7 + 0.3 * healthA);
|
|
151
|
+
const adjustedB = b.score * (0.7 + 0.3 * healthB);
|
|
152
|
+
return adjustedB - adjustedA;
|
|
153
|
+
});
|
|
154
|
+
}
|
|
155
|
+
shouldTriggerHybrid(winner, runnerUp) {
|
|
156
|
+
if (!runnerUp) {
|
|
157
|
+
return false;
|
|
158
|
+
}
|
|
159
|
+
const threshold = winner.score * (1 - this.hybridTriggerPercent);
|
|
160
|
+
return runnerUp.score >= threshold;
|
|
161
|
+
}
|
|
162
|
+
createBid(candidate, rank, now, request, costTracker) {
|
|
163
|
+
const latency = candidate.model.avgLatencyMs ?? 1500 + rank * 100;
|
|
164
|
+
const inputTokens = request.estimatedInputTokens ?? DEFAULT_ESTIMATED_INPUT_TOKENS;
|
|
165
|
+
const outputTokens = request.estimatedOutputTokens ?? DEFAULT_ESTIMATED_OUTPUT_TOKENS;
|
|
166
|
+
let estimatedCost = 0;
|
|
167
|
+
if (costTracker) {
|
|
168
|
+
estimatedCost = costTracker.estimate(candidate.model.model, {
|
|
169
|
+
inputTokens,
|
|
170
|
+
outputTokens,
|
|
171
|
+
});
|
|
172
|
+
}
|
|
173
|
+
if (estimatedCost === 0 && costTracker) {
|
|
174
|
+
// Try with provider-prefixed model name (openrouter uses these)
|
|
175
|
+
estimatedCost = costTracker.estimate(`${candidate.model.provider}/${candidate.model.model}`, { inputTokens, outputTokens });
|
|
176
|
+
}
|
|
177
|
+
return {
|
|
178
|
+
bidId: `${candidate.model.provider}:${candidate.model.model}:${now}:${rank}`,
|
|
179
|
+
model: candidate.model.model,
|
|
180
|
+
provider: candidate.model.provider,
|
|
181
|
+
score: candidate.score,
|
|
182
|
+
confidence: Math.max(0, Math.min(1, candidate.score)),
|
|
183
|
+
estimatedCostUsd: Number(estimatedCost.toFixed(6)),
|
|
184
|
+
estimatedLatencyMs: latency,
|
|
185
|
+
createdAt: now,
|
|
186
|
+
reasons: candidate.reasons,
|
|
187
|
+
};
|
|
188
|
+
}
|
|
189
|
+
}
|
|
190
|
+
//# sourceMappingURL=auction.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"auction.js","sourceRoot":"","sources":["../../src/providers/auction.ts"],"names":[],"mappings":"AA6BA,MAAM,8BAA8B,GAAG,IAAI,CAAC;AAC5C,MAAM,+BAA+B,GAAG,IAAI,CAAC;AAE7C,SAAS,UAAU,CAAC,KAAa;IAC/B,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;QACzC,IAAI,GAAG,CAAC,IAAI,IAAI,CAAC,CAAC,GAAG,IAAI,GAAG,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QAChD,IAAI,IAAI,CAAC,CAAC;IACZ,CAAC;IAED,OAAO,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;AACxB,CAAC;AAED,SAAS,UAAU,CAAC,GAAW;IAC7B,IAAI,GAAG,GAAG,IAAI;QAAE,OAAO,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;IAC5C,OAAO,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;AAC9B,CAAC;AAED,SAAS,aAAa,CAAC,EAAU;IAC/B,IAAI,EAAE,GAAG,IAAI;QAAE,OAAO,GAAG,EAAE,IAAI,CAAC;IAChC,OAAO,GAAG,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;AACtC,CAAC;AAED,SAAS,qBAAqB,CAAC,OAAqB;IAClD,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,oBAAoB,IAAI,EAAE,CAAC,CAAC;IACzD,MAAM,cAAc,GAAG,OAAO,CAAC,oBAAoB,IAAI,8BAA8B,CAAC;IACtF,MAAM,eAAe,GAAG,OAAO,CAAC,qBAAqB,IAAI,+BAA+B,CAAC;IACzF,MAAM,cAAc,GAAG,cAAc,GAAG,eAAe,CAAC;IAExD,MAAM,WAAW,GAAG;QAClB,cAAc,EAAE,gBAAgB,EAAE,mBAAmB;QACrD,YAAY,EAAE,eAAe,EAAE,qBAAqB;KACrD,CAAC;IAEF,MAAM,YAAY,GAAG;QACnB,UAAU,EAAE,kBAAkB,EAAE,0BAA0B;QAC1D,iBAAiB,EAAE,eAAe,EAAE,aAAa;KAClD,CAAC;IAEF,IAAI,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,cAAc,IAAI,IAAI,EAAE,CAAC;QACnE,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,cAAc,IAAI,IAAI,EAAE,CAAC;QACpE,OAAO,UAAU,CAAC;IACpB,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,OAAO,uBAAuB;IACjB,oBAAoB,CAAS;IAE9C,YAAY,OAAkC;QAC5C,IAAI,CAAC,oBAAoB,GAAG,OAAO,EAAE,oBAAoB,IAAI,IAAI,CAAC;IACpE,CAAC;IAED,MAAM,CAAC,MAQN;QACC,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;QACrC,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC;QAEnC,IAAI,MAAM,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnC,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC9D,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,cAAc,CAAC,CAAC;QAC7E,MAAM,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CACvC,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,WAAW,CAAC,CACtE,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QACzB,MAAM,SAAS,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1B,MAAM,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QAC3B,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC;QAEjD,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;YACtB,OAAO;gBACL,IAAI,EAAE,QAAQ;gBACd,MAAM;gBACN,IAAI;gBACJ,SAAS,EAAE,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC;aAC3F,CAAC;QACJ,CAAC;QAED,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAClC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACxB,OAAO;oBACL,IAAI,EAAE,SAAS;oBACf,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;oBACjB,IAAI;oBACJ,SAAS,EAAE,yDAAyD,MAAM,CAAC,KAAK,CAAC,KAAK,0BAA0B;iBACjH,CAAC;YACJ,CAAC;YAED,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,GAAG,EAAE,CAAC;YACvD,MAAM,GAAG,GAAG,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAChC,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;YAC7B,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YACrC,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC;YAEnD,OAAO;gBACL,IAAI,EAAE,SAAS;gBACf,MAAM,EAAE,QAAQ;gBAChB,MAAM;gBACN,IAAI;gBACJ,SAAS,EAAE,wCAAwC,QAAQ,CAAC,KAAK,CAAC,KAAK,+BAA+B,UAAU,CAAC,WAAW,CAAC,gBAAgB,CAAC,MAAM,aAAa,CAAC,WAAW,CAAC,kBAAkB,CAAC,WAAW,MAAM,CAAC,KAAK,CAAC,KAAK,4BAA4B;aAC3P,CAAC;QACJ,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,mBAAmB,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QACrF,IAAI,YAAY,IAAI,QAAQ,EAAE,CAAC;YAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;YAC1B,MAAM,UAAU,GAAG,CAAC,CAAC,MAAM,CAAC,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,KAAK,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YACrF,OAAO;gBACL,IAAI,EAAE,QAAQ;gBACd,MAAM;gBACN,MAAM,EAAE,QAAQ;gBAChB,IAAI;gBACJ,SAAS,EAAE,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,CAAC,OAAO,CAAC;aACzG,CAAC;QACJ,CAAC;QAED,OAAO;YACL,IAAI,EAAE,QAAQ;YACd,MAAM;YACN,IAAI;YACJ,SAAS,EAAE,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC;SACzF,CAAC;IACJ,CAAC;IAEO,cAAc,CACpB,OAAe,EACf,MAAmB,EACnB,GAAa,EACb,cAAsB,EACtB,OAAqB;QAErB,MAAM,SAAS,GAAG,qBAAqB,CAAC,OAAO,CAAC,CAAC;QACjD,MAAM,oBAAoB,GACxB,SAAS,KAAK,SAAS;YACrB,CAAC,CAAC,qFAAqF;YACvF,CAAC,CAAC,SAAS,KAAK,UAAU;gBACxB,CAAC,CAAC,mFAAmF;gBACrF,CAAC,CAAC,uEAAuE,CAAC;QAEhF,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACjC,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,yBAAyB,cAAc,sCAAsC,CAAC,CAAC;QAC/G,KAAK,CAAC,IAAI,CAAC,yBAAyB,UAAU,CAAC,GAAG,CAAC,gBAAgB,CAAC,eAAe,aAAa,CAAC,GAAG,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC;QAErI,IAAI,MAAM,CAAC,KAAK,CAAC,aAAa,EAAE,CAAC;YAC/B,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,aAAa,IAAI,OAAS;gBACjD,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,aAAa,GAAG,OAAS,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG;gBAC3D,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,aAAa,GAAG,IAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;YAC1D,KAAK,CAAC,IAAI,CAAC,sBAAsB,GAAG,qCAAqC,CAAC,CAAC;QAC7E,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC;IAChC,CAAC;IAEO,oBAAoB,CAC1B,MAAmB,EACnB,SAAmB,EACnB,QAAqB,EACrB,SAAmB,EACnB,UAAkB,EAClB,OAAqB;QAErB,MAAM,SAAS,GAAG,qBAAqB,CAAC,OAAO,CAAC,CAAC;QACjD,MAAM,cAAc,GAClB,SAAS,KAAK,SAAS;YACrB,CAAC,CAAC,2EAA2E;YAC7E,CAAC,CAAC,kHAAkH,CAAC;QAEzH,MAAM,KAAK,GAAa,EAAE,CAAC;QAE3B,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC3B,KAAK,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,KAAK,kBAAkB,UAAU,QAAQ,MAAM,CAAC,KAAK,CAAC,KAAK,2CAA2C,CAAC,CAAC;QACrI,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,uBAAuB,UAAU,CAAC,SAAS,CAAC,gBAAgB,CAAC,MAAM,aAAa,CAAC,SAAS,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;QACpJ,KAAK,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,KAAK,uBAAuB,UAAU,CAAC,SAAS,CAAC,gBAAgB,CAAC,MAAM,aAAa,CAAC,SAAS,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;QACtJ,KAAK,CAAC,IAAI,CAAC,4DAA4D,CAAC,CAAC;QAEzE,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC;IAChC,CAAC;IAED,2EAA2E;IAC3E,2EAA2E;IAC3E,uEAAuE;IACvE,0EAA0E;IAC1E,4EAA4E;IACpE,cAAc,CAAC,UAAyB,EAAE,cAAoC;QACpF,OAAO,CAAC,GAAG,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACnC,MAAM,OAAO,GAAG,cAAc,EAAE,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC3D,MAAM,OAAO,GAAG,cAAc,EAAE,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC3D,MAAM,SAAS,GAAG,CAAC,CAAC,KAAK,GAAG,CAAC,GAAG,GAAG,GAAG,GAAG,OAAO,CAAC,CAAC;YAClD,MAAM,SAAS,GAAG,CAAC,CAAC,KAAK,GAAG,CAAC,GAAG,GAAG,GAAG,GAAG,OAAO,CAAC,CAAC;YAClD,OAAO,SAAS,GAAG,SAAS,CAAC;QAC/B,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,mBAAmB,CAAC,MAAmB,EAAE,QAAsB;QACrE,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACjE,OAAO,QAAQ,CAAC,KAAK,IAAI,SAAS,CAAC;IACrC,CAAC;IAEO,SAAS,CACf,SAAsB,EACtB,IAAY,EACZ,GAAW,EACX,OAAqB,EACrB,WAAyB;QAEzB,MAAM,OAAO,GAAG,SAAS,CAAC,KAAK,CAAC,YAAY,IAAI,IAAI,GAAG,IAAI,GAAG,GAAG,CAAC;QAElE,MAAM,WAAW,GAAG,OAAO,CAAC,oBAAoB,IAAI,8BAA8B,CAAC;QACnF,MAAM,YAAY,GAAG,OAAO,CAAC,qBAAqB,IAAI,+BAA+B,CAAC;QAEtF,IAAI,aAAa,GAAG,CAAC,CAAC;QACtB,IAAI,WAAW,EAAE,CAAC;YAChB,aAAa,GAAG,WAAW,CAAC,QAAQ,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,EAAE;gBAC1D,WAAW;gBACX,YAAY;aACb,CAAC,CAAC;QACL,CAAC;QAED,IAAI,aAAa,KAAK,CAAC,IAAI,WAAW,EAAE,CAAC;YACvC,gEAAgE;YAChE,aAAa,GAAG,WAAW,CAAC,QAAQ,CAClC,GAAG,SAAS,CAAC,KAAK,CAAC,QAAQ,IAAI,SAAS,CAAC,KAAK,CAAC,KAAK,EAAE,EACtD,EAAE,WAAW,EAAE,YAAY,EAAE,CAC9B,CAAC;QACJ,CAAC;QAED,OAAO;YACL,KAAK,EAAE,GAAG,SAAS,CAAC,KAAK,CAAC,QAAQ,IAAI,SAAS,CAAC,KAAK,CAAC,KAAK,IAAI,GAAG,IAAI,IAAI,EAAE;YAC5E,KAAK,EAAE,SAAS,CAAC,KAAK,CAAC,KAAK;YAC5B,QAAQ,EAAE,SAAS,CAAC,KAAK,CAAC,QAAQ;YAClC,KAAK,EAAE,SAAS,CAAC,KAAK;YACtB,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,CAAC,KAAK,CAAC,CAAC;YACrD,gBAAgB,EAAE,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YAClD,kBAAkB,EAAE,OAAO;YAC3B,SAAS,EAAE,GAAG;YACd,OAAO,EAAE,SAAS,CAAC,OAAO;SAC3B,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1,103 @@
|
|
|
1
|
+
/// <reference types="node" resolution-mode="require"/>
|
|
2
|
+
export type ProviderName = "openai" | "anthropic" | "openrouter";
|
|
3
|
+
export type MessageRole = "system" | "user" | "assistant" | "tool";
|
|
4
|
+
export interface ChatMessage {
|
|
5
|
+
role: MessageRole;
|
|
6
|
+
content: string;
|
|
7
|
+
name?: string;
|
|
8
|
+
toolCallId?: string;
|
|
9
|
+
/** Present on assistant messages that invoke tools */
|
|
10
|
+
toolCalls?: ToolCall[];
|
|
11
|
+
}
|
|
12
|
+
export interface ToolDefinition {
|
|
13
|
+
name: string;
|
|
14
|
+
description?: string;
|
|
15
|
+
inputSchema: Record<string, unknown>;
|
|
16
|
+
}
|
|
17
|
+
export interface ToolCall {
|
|
18
|
+
id: string;
|
|
19
|
+
name: string;
|
|
20
|
+
input: Record<string, unknown>;
|
|
21
|
+
}
|
|
22
|
+
export interface TokenUsage {
|
|
23
|
+
inputTokens: number;
|
|
24
|
+
outputTokens: number;
|
|
25
|
+
cacheReadInputTokens?: number;
|
|
26
|
+
cacheWriteInputTokens?: number;
|
|
27
|
+
}
|
|
28
|
+
export interface ProviderResponse {
|
|
29
|
+
provider: ProviderName;
|
|
30
|
+
model: string;
|
|
31
|
+
text: string;
|
|
32
|
+
toolCalls: ToolCall[];
|
|
33
|
+
stopReason: string | null;
|
|
34
|
+
usage: TokenUsage;
|
|
35
|
+
raw?: unknown;
|
|
36
|
+
}
|
|
37
|
+
export type ProviderStreamEvent = {
|
|
38
|
+
type: "text-delta";
|
|
39
|
+
delta: string;
|
|
40
|
+
} | {
|
|
41
|
+
type: "tool-call";
|
|
42
|
+
call: ToolCall;
|
|
43
|
+
} | {
|
|
44
|
+
type: "usage";
|
|
45
|
+
usage: TokenUsage;
|
|
46
|
+
} | {
|
|
47
|
+
type: "done";
|
|
48
|
+
response: ProviderResponse;
|
|
49
|
+
} | {
|
|
50
|
+
type: "error";
|
|
51
|
+
error: Error;
|
|
52
|
+
};
|
|
53
|
+
export interface GenerateRequest {
|
|
54
|
+
model: string;
|
|
55
|
+
messages: ChatMessage[];
|
|
56
|
+
tools?: ToolDefinition[];
|
|
57
|
+
toolChoice?: "auto" | "required" | "none";
|
|
58
|
+
temperature?: number;
|
|
59
|
+
maxOutputTokens?: number;
|
|
60
|
+
metadata?: Record<string, string>;
|
|
61
|
+
}
|
|
62
|
+
export interface StreamRequest extends GenerateRequest {
|
|
63
|
+
signal?: AbortSignal;
|
|
64
|
+
}
|
|
65
|
+
export interface ModelProfile {
|
|
66
|
+
model: string;
|
|
67
|
+
provider: ProviderName;
|
|
68
|
+
capabilities: string[];
|
|
69
|
+
supportsStreaming: boolean;
|
|
70
|
+
supportsTools: boolean;
|
|
71
|
+
contextWindow?: number;
|
|
72
|
+
maxOutputTokens?: number;
|
|
73
|
+
avgLatencyMs?: number;
|
|
74
|
+
defaultScore?: number;
|
|
75
|
+
roleAffinity?: {
|
|
76
|
+
architect: number;
|
|
77
|
+
executor: number;
|
|
78
|
+
};
|
|
79
|
+
}
|
|
80
|
+
export interface ModelProvider {
|
|
81
|
+
readonly name: ProviderName;
|
|
82
|
+
supportsModel(model: string): boolean;
|
|
83
|
+
generate(request: GenerateRequest): Promise<ProviderResponse>;
|
|
84
|
+
stream(request: StreamRequest): AsyncIterable<ProviderStreamEvent>;
|
|
85
|
+
}
|
|
86
|
+
export interface RouteRequest {
|
|
87
|
+
requiredCapabilities: string[];
|
|
88
|
+
preferredProvider?: ProviderName;
|
|
89
|
+
excludedProviders?: string[];
|
|
90
|
+
excludedModels?: string[];
|
|
91
|
+
maxInputCostPer1k?: number;
|
|
92
|
+
maxOutputCostPer1k?: number;
|
|
93
|
+
maxLatencyMs?: number;
|
|
94
|
+
requireStreaming?: boolean;
|
|
95
|
+
requireTools?: boolean;
|
|
96
|
+
estimatedInputTokens?: number;
|
|
97
|
+
estimatedOutputTokens?: number;
|
|
98
|
+
}
|
|
99
|
+
export interface RouteResult {
|
|
100
|
+
model: ModelProfile;
|
|
101
|
+
score: number;
|
|
102
|
+
reasons: string[];
|
|
103
|
+
}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/providers/base.ts"],"names":[],"mappings":""}
|