@autonsh/cli 0.18.0 → 0.19.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/dist/chat.d.ts +1 -1
- package/dist/chat.js +166 -120
- package/dist/chat.js.map +1 -1
- package/dist/completion.d.ts +2 -0
- package/dist/completion.js +379 -0
- package/dist/completion.js.map +1 -0
- package/dist/config-validation.d.ts +10 -0
- package/dist/config-validation.js +101 -0
- package/dist/config-validation.js.map +1 -0
- package/dist/config.d.ts +9 -3
- package/dist/config.js +23 -11
- package/dist/config.js.map +1 -1
- package/dist/doctor.d.ts +18 -0
- package/dist/doctor.js +409 -0
- package/dist/doctor.js.map +1 -0
- package/dist/errors.d.ts +22 -0
- package/dist/errors.js +60 -0
- package/dist/errors.js.map +1 -0
- package/dist/index.js +62 -64
- package/dist/index.js.map +1 -1
- package/dist/input.d.ts +43 -0
- package/dist/input.js +288 -0
- package/dist/input.js.map +1 -0
- package/dist/llm/factory.d.ts +5 -0
- package/dist/llm/factory.js +54 -0
- package/dist/llm/factory.js.map +1 -0
- package/dist/llm/index.d.ts +9 -0
- package/dist/llm/index.js +11 -0
- package/dist/llm/index.js.map +1 -0
- package/dist/llm/provider.d.ts +27 -0
- package/dist/llm/provider.js +98 -0
- package/dist/llm/provider.js.map +1 -0
- package/dist/llm/providers/anthropic.d.ts +23 -0
- package/dist/llm/providers/anthropic.js +198 -0
- package/dist/llm/providers/anthropic.js.map +1 -0
- package/dist/llm/providers/google.d.ts +31 -0
- package/dist/llm/providers/google.js +190 -0
- package/dist/llm/providers/google.js.map +1 -0
- package/dist/llm/providers/ollama.d.ts +21 -0
- package/dist/llm/providers/ollama.js +157 -0
- package/dist/llm/providers/ollama.js.map +1 -0
- package/dist/llm/providers/openai-compatible.d.ts +22 -0
- package/dist/llm/providers/openai-compatible.js +175 -0
- package/dist/llm/providers/openai-compatible.js.map +1 -0
- package/dist/llm/providers/openai.d.ts +20 -0
- package/dist/llm/providers/openai.js +167 -0
- package/dist/llm/providers/openai.js.map +1 -0
- package/dist/llm/registry.d.ts +19 -0
- package/dist/llm/registry.js +421 -0
- package/dist/llm/registry.js.map +1 -0
- package/dist/llm/types.d.ts +174 -0
- package/dist/llm/types.js +3 -0
- package/dist/llm/types.js.map +1 -0
- package/dist/login.d.ts +1 -1
- package/dist/login.js +14 -14
- package/dist/login.js.map +1 -1
- package/dist/mcp/client.d.ts +36 -0
- package/dist/mcp/client.js +199 -0
- package/dist/mcp/client.js.map +1 -0
- package/dist/mcp/config.d.ts +10 -0
- package/dist/mcp/config.js +43 -0
- package/dist/mcp/config.js.map +1 -0
- package/dist/mcp/index.d.ts +4 -0
- package/dist/mcp/index.js +6 -0
- package/dist/mcp/index.js.map +1 -0
- package/dist/mcp/types.d.ts +104 -0
- package/dist/mcp/types.js +3 -0
- package/dist/mcp/types.js.map +1 -0
- package/dist/model.d.ts +3 -14
- package/dist/model.js +131 -91
- package/dist/model.js.map +1 -1
- package/dist/ollama.d.ts +10 -0
- package/dist/ollama.js +11 -0
- package/dist/ollama.js.map +1 -0
- package/dist/repomap.d.ts +37 -0
- package/dist/repomap.js +431 -0
- package/dist/repomap.js.map +1 -0
- package/dist/skills/index.d.ts +3 -0
- package/dist/skills/index.js +5 -0
- package/dist/skills/index.js.map +1 -0
- package/dist/skills/registry.d.ts +16 -0
- package/dist/skills/registry.js +128 -0
- package/dist/skills/registry.js.map +1 -0
- package/dist/skills/types.d.ts +48 -0
- package/dist/skills/types.js +3 -0
- package/dist/skills/types.js.map +1 -0
- package/dist/statusline.d.ts +21 -0
- package/dist/statusline.js +68 -0
- package/dist/statusline.js.map +1 -0
- package/dist/streaming.d.ts +25 -0
- package/dist/streaming.js +91 -0
- package/dist/streaming.js.map +1 -0
- package/dist/theme.d.ts +52 -0
- package/dist/theme.js +92 -0
- package/dist/theme.js.map +1 -0
- package/dist/tools/builtin.d.ts +11 -0
- package/dist/tools/builtin.js +491 -0
- package/dist/tools/builtin.js.map +1 -0
- package/dist/tools/index.d.ts +3 -0
- package/dist/tools/index.js +5 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/types.d.ts +186 -0
- package/dist/tools/types.js +29 -0
- package/dist/tools/types.js.map +1 -0
- package/dist/ui.js +34 -30
- package/dist/ui.js.map +1 -1
- package/dist/verbose.d.ts +20 -0
- package/dist/verbose.js +75 -0
- package/dist/verbose.js.map +1 -0
- package/package.json +11 -2
- package/templates/api-json/task.yaml +0 -2
- package/templates/coding/task.yaml +1 -1
- package/templates/cron-digester/task.yaml +0 -2
- package/templates/harness/task.yaml +0 -11
- package/templates/multi-agent/agent.ts +7 -8
- package/templates/multi-agent/task.yaml +1 -3
- package/templates/research/task.yaml +0 -28
- package/templates/support/task.yaml +0 -2
- package/templates/webhook-echo/task.yaml +0 -2
- package/templates/with-approval/task.yaml +0 -2
|
@@ -0,0 +1,175 @@
|
|
|
1
|
+
// openai-compatible.ts — OpenAI-compatible provider (Groq, DeepSeek, OpenRouter, etc.)
|
|
2
|
+
import { BaseProvider } from "../provider.js";
|
|
3
|
+
export class OpenAICompatibleProvider extends BaseProvider {
|
|
4
|
+
disablePromptCacheKey;
|
|
5
|
+
constructor(config) {
|
|
6
|
+
super(config);
|
|
7
|
+
this.disablePromptCacheKey = config.kind === "openai-compatible";
|
|
8
|
+
}
|
|
9
|
+
defaultBaseUrl() {
|
|
10
|
+
return "https://api.openai.com/v1";
|
|
11
|
+
}
|
|
12
|
+
chatCompletionsEndpoint() {
|
|
13
|
+
return `${this.baseUrl}/chat/completions`;
|
|
14
|
+
}
|
|
15
|
+
modelsEndpoint() {
|
|
16
|
+
return `${this.baseUrl}/models`;
|
|
17
|
+
}
|
|
18
|
+
buildRequestBody(request, stream) {
|
|
19
|
+
const messages = this.buildMessages(request);
|
|
20
|
+
const tools = this.mapTools(request);
|
|
21
|
+
const body = {
|
|
22
|
+
model: request.model || this.model,
|
|
23
|
+
messages,
|
|
24
|
+
stream,
|
|
25
|
+
stream_options: stream ? { include_usage: true } : undefined,
|
|
26
|
+
};
|
|
27
|
+
if (tools)
|
|
28
|
+
body.tools = tools;
|
|
29
|
+
if (request.toolChoice)
|
|
30
|
+
body.tool_choice = request.toolChoice;
|
|
31
|
+
if (request.generation) {
|
|
32
|
+
const gen = request.generation;
|
|
33
|
+
if (gen.temperature !== undefined)
|
|
34
|
+
body.temperature = gen.temperature;
|
|
35
|
+
if (gen.topP !== undefined)
|
|
36
|
+
body.top_p = gen.topP;
|
|
37
|
+
if (gen.maxTokens !== undefined)
|
|
38
|
+
body.max_tokens = gen.maxTokens;
|
|
39
|
+
if (gen.stopSequences?.length)
|
|
40
|
+
body.stop = gen.stopSequences;
|
|
41
|
+
if (gen.reasoningEffort)
|
|
42
|
+
body.reasoning_effort = this.mapReasoningEffort(gen.reasoningEffort);
|
|
43
|
+
if (gen.responseFormat)
|
|
44
|
+
body.response_format = gen.responseFormat;
|
|
45
|
+
}
|
|
46
|
+
if (this.disablePromptCacheKey) {
|
|
47
|
+
body.disable_prompt_cache_key = true;
|
|
48
|
+
}
|
|
49
|
+
if (request.providerOptions?.openai) {
|
|
50
|
+
Object.assign(body, request.providerOptions.openai);
|
|
51
|
+
}
|
|
52
|
+
return body;
|
|
53
|
+
}
|
|
54
|
+
parseResponse(data) {
|
|
55
|
+
const choice = data.choices?.[0];
|
|
56
|
+
const message = choice?.message;
|
|
57
|
+
const usage = this.mapUsage(data);
|
|
58
|
+
let content = message?.content ?? "";
|
|
59
|
+
const toolCalls = (message?.tool_calls ?? []).map((tc) => ({
|
|
60
|
+
id: tc.id,
|
|
61
|
+
type: "function",
|
|
62
|
+
function: { name: tc.function.name, arguments: tc.function.arguments },
|
|
63
|
+
}));
|
|
64
|
+
if (toolCalls.length > 0) {
|
|
65
|
+
content = toolCalls.map((tc) => ({
|
|
66
|
+
type: "tool_result",
|
|
67
|
+
toolResult: { toolCallId: tc.id, content: tc.function.arguments },
|
|
68
|
+
}));
|
|
69
|
+
}
|
|
70
|
+
return {
|
|
71
|
+
id: data.id ?? crypto.randomUUID(),
|
|
72
|
+
model: data.model ?? this.model,
|
|
73
|
+
content,
|
|
74
|
+
toolCalls: toolCalls.length > 0 ? toolCalls : undefined,
|
|
75
|
+
usage,
|
|
76
|
+
stopReason: this.mapStopReason(data),
|
|
77
|
+
costUsd: this.estimateCost(usage),
|
|
78
|
+
};
|
|
79
|
+
}
|
|
80
|
+
parseStreamChunk(data) {
|
|
81
|
+
const lines = String(data)
|
|
82
|
+
.split("\n")
|
|
83
|
+
.filter((l) => l.startsWith("data: "))
|
|
84
|
+
.map((l) => l.slice(6))
|
|
85
|
+
.filter((l) => l !== "[DONE]");
|
|
86
|
+
const chunks = [];
|
|
87
|
+
for (const line of lines) {
|
|
88
|
+
try {
|
|
89
|
+
const parsed = JSON.parse(line);
|
|
90
|
+
const choice = parsed.choices?.[0];
|
|
91
|
+
if (!choice)
|
|
92
|
+
continue;
|
|
93
|
+
const delta = choice.delta;
|
|
94
|
+
if (delta?.content) {
|
|
95
|
+
chunks.push({ type: "content", content: delta.content });
|
|
96
|
+
}
|
|
97
|
+
if (delta?.tool_calls) {
|
|
98
|
+
for (const tc of delta.tool_calls) {
|
|
99
|
+
if (tc.index === 0 && tc.function?.name) {
|
|
100
|
+
chunks.push({
|
|
101
|
+
type: "tool_call",
|
|
102
|
+
toolCall: {
|
|
103
|
+
id: tc.id ?? crypto.randomUUID(),
|
|
104
|
+
type: "function",
|
|
105
|
+
function: { name: tc.function.name, arguments: tc.function.arguments ?? "" },
|
|
106
|
+
},
|
|
107
|
+
});
|
|
108
|
+
}
|
|
109
|
+
else if (tc.function?.arguments) {
|
|
110
|
+
chunks.push({
|
|
111
|
+
type: "tool_call_delta",
|
|
112
|
+
toolCallDelta: { index: tc.index, arguments: tc.function.arguments },
|
|
113
|
+
});
|
|
114
|
+
}
|
|
115
|
+
}
|
|
116
|
+
}
|
|
117
|
+
if (parsed.usage) {
|
|
118
|
+
chunks.push({ type: "usage", usage: this.mapUsage(parsed) });
|
|
119
|
+
}
|
|
120
|
+
if (choice.finish_reason) {
|
|
121
|
+
chunks.push({ type: "complete", stopReason: choice.finish_reason });
|
|
122
|
+
}
|
|
123
|
+
}
|
|
124
|
+
catch {
|
|
125
|
+
// ignore parse errors
|
|
126
|
+
}
|
|
127
|
+
}
|
|
128
|
+
return chunks;
|
|
129
|
+
}
|
|
130
|
+
buildMessages(request) {
|
|
131
|
+
const messages = [];
|
|
132
|
+
if (request.system) {
|
|
133
|
+
messages.push({ role: "system", content: request.system });
|
|
134
|
+
}
|
|
135
|
+
for (const msg of request.messages) {
|
|
136
|
+
messages.push(this.mapMessage(msg));
|
|
137
|
+
}
|
|
138
|
+
return messages;
|
|
139
|
+
}
|
|
140
|
+
mapMessage(msg) {
|
|
141
|
+
if (typeof msg.content === "string") {
|
|
142
|
+
return { role: msg.role, content: msg.content };
|
|
143
|
+
}
|
|
144
|
+
return { role: msg.role, content: msg.content };
|
|
145
|
+
}
|
|
146
|
+
mapTools(request) {
|
|
147
|
+
if (!request.tools?.length)
|
|
148
|
+
return undefined;
|
|
149
|
+
return request.tools.map((t) => ({
|
|
150
|
+
type: "function",
|
|
151
|
+
function: { name: t.name, description: t.description, parameters: t.parameters },
|
|
152
|
+
}));
|
|
153
|
+
}
|
|
154
|
+
mapUsage(data) {
|
|
155
|
+
return {
|
|
156
|
+
inputTokens: data?.usage?.prompt_tokens ?? 0,
|
|
157
|
+
outputTokens: data?.usage?.completion_tokens ?? 0,
|
|
158
|
+
cachedInputTokens: data?.usage?.prompt_tokens_details?.cached_tokens ?? 0,
|
|
159
|
+
totalTokens: data?.usage?.total_tokens ?? 0,
|
|
160
|
+
};
|
|
161
|
+
}
|
|
162
|
+
mapStopReason(data) {
|
|
163
|
+
const choice = data?.choices?.[0];
|
|
164
|
+
return choice?.finish_reason ?? "end_turn";
|
|
165
|
+
}
|
|
166
|
+
estimateCost(usage) {
|
|
167
|
+
const inputRate = 0.50;
|
|
168
|
+
const outputRate = 1.50;
|
|
169
|
+
return (usage.inputTokens / 1e6) * inputRate + (usage.outputTokens / 1e6) * outputRate;
|
|
170
|
+
}
|
|
171
|
+
}
|
|
172
|
+
export function createOpenAICompatibleProvider(config) {
|
|
173
|
+
return new OpenAICompatibleProvider(config);
|
|
174
|
+
}
|
|
175
|
+
//# sourceMappingURL=openai-compatible.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai-compatible.js","sourceRoot":"","sources":["../../../src/llm/providers/openai-compatible.ts"],"names":[],"mappings":"AAAA,uFAAuF;AAEvF,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAW9C,MAAM,OAAO,wBAAyB,SAAQ,YAAY;IAC/C,qBAAqB,CAAU;IAExC,YAAY,MAAsB;QAChC,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,qBAAqB,GAAG,MAAM,CAAC,IAAI,KAAK,mBAAmB,CAAC;IACnE,CAAC;IAEkB,cAAc;QAC/B,OAAO,2BAA2B,CAAC;IACrC,CAAC;IAEkB,uBAAuB;QACxC,OAAO,GAAG,IAAI,CAAC,OAAO,mBAAmB,CAAC;IAC5C,CAAC;IAEkB,cAAc;QAC/B,OAAO,GAAG,IAAI,CAAC,OAAO,SAAS,CAAC;IAClC,CAAC;IAEkB,gBAAgB,CAAC,OAAmB,EAAE,MAAe;QACtE,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAErC,MAAM,IAAI,GAA4B;YACpC,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK;YAClC,QAAQ;YACR,MAAM;YACN,cAAc,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS;SAC7D,CAAC;QAEF,IAAI,KAAK;YAAE,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QAC9B,IAAI,OAAO,CAAC,UAAU;YAAE,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,UAAU,CAAC;QAC9D,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YACvB,MAAM,GAAG,GAAG,OAAO,CAAC,UAAU,CAAC;YAC/B,IAAI,GAAG,CAAC,WAAW,KAAK,SAAS;gBAAE,IAAI,CAAC,WAAW,GAAG,GAAG,CAAC,WAAW,CAAC;YACtE,IAAI,GAAG,CAAC,IAAI,KAAK,SAAS;gBAAE,IAAI,CAAC,KAAK,GAAG,GAAG,CAAC,IAAI,CAAC;YAClD,IAAI,GAAG,CAAC,SAAS,KAAK,SAAS;gBAAE,IAAI,CAAC,UAAU,GAAG,GAAG,CAAC,SAAS,CAAC;YACjE,IAAI,GAAG,CAAC,aAAa,EAAE,MAAM;gBAAE,IAAI,CAAC,IAAI,GAAG,GAAG,CAAC,aAAa,CAAC;YAC7D,IAAI,GAAG,CAAC,eAAe;gBAAE,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;YAC9F,IAAI,GAAG,CAAC,cAAc;gBAAE,IAAI,CAAC,eAAe,GAAG,GAAG,CAAC,cAAc,CAAC;QACpE,CAAC;QAED,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;YAC/B,IAAI,CAAC,wBAAwB,GAAG,IAAI,CAAC;QACvC,CAAC;QAED,IAAI,OAAO,CAAC,eAAe,EAAE,MAAM,EAAE,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QACtD,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAEkB,aAAa,CAAC,IAAa;QAC5C,MAAM,MAAM,GAAI,IAAY,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,OAAO,GAAG,MAAM,EAAE,OAAO,CAAC;QAChC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAElC,IAAI,OAAO,GAAsB,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;QACxD,MAAM,SAAS,GAAe,CAAC,OAAO,EAAE,UAAU,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAO,EAAE,EAAE,CAAC,CAAC;YAC1E,EAAE,EAAE,EAAE,CAAC,EAAE;YACT,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC,QAAQ,CAAC,SAAS,EAAE;SACvE,CAAC,CAAC,CAAC;QAEJ,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,OAAO,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBAC/B,IAAI,EAAE,aAAa;gBACnB,UAAU,EAAE,EAAE,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,OAAO,EAAE,EAAE,CAAC,QAAQ,CAAC,SAAS,EAAE;aAClE,CAAC,CAAC,CAAC;QACN,CAAC;QAED,OAAO;YACL,EAAE,EAAG,IAAY,CAAC,EAAE,IAAI,MAAM,CAAC,UAAU,EAAE;YAC3C,KAAK,EAAG,IAAY,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK;YACxC,OAAO;YACP,SAAS,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;YACvD,KAAK;YACL,UAAU,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;YACpC,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC;SAClC,CAAC;IACJ,CAAC;IAEkB,gBAAgB,CAAC,IAAa;QAC/C,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC;aACvB,KAAK,CAAC,IAAI,CAAC;aACX,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;aACrC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;aACtB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC;QAEjC,MAAM,MAAM,GAAkB,EAAE,CAAC;QACjC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAChC,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;gBACnC,IAAI,CAAC,MAAM;oBAAE,SAAS;gBAEtB,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;gBAC3B,IAAI,KAAK,EAAE,OAAO,EAAE,CAAC;oBACnB,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC3D,CAAC;gBACD,IAAI,KAAK,EAAE,UAAU,EAAE,CAAC;oBACtB,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;wBAClC,IAAI,EAAE,CAAC,KAAK,KAAK,CAAC,IAAI,EAAE,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC;4BACxC,MAAM,CAAC,IAAI,CAAC;gCACV,IAAI,EAAE,WAAW;gCACjB,QAAQ,EAAE;oCACR,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,MAAM,CAAC,UAAU,EAAE;oCAChC,IAAI,EAAE,UAAU;oCAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC,QAAQ,CAAC,SAAS,IAAI,EAAE,EAAE;iCAC7E;6BACF,CAAC,CAAC;wBACL,CAAC;6BAAM,IAAI,EAAE,CAAC,QAAQ,EAAE,SAAS,EAAE,CAAC;4BAClC,MAAM,CAAC,IAAI,CAAC;gCACV,IAAI,EAAE,iBAAiB;gCACvB,aAAa,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,EAAE,CAAC,QAAQ,CAAC,SAAS,EAAE;6BACrE,CAAC,CAAC;wBACL,CAAC;oBACH,CAAC;gBACH,CAAC;gBACD,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;oBACjB,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBAC/D,CAAC;gBACD,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;oBACzB,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC;gBACtE,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,sBAAsB;YACxB,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAES,aAAa,CAAC,OAAmB;QACzC,MAAM,QAAQ,GAAa,EAAE,CAAC;QAC9B,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAC7D,CAAC;QACD,KAAK,MAAM,GAAG,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACnC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;QACtC,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAES,UAAU,CAAC,GAAiD;QACpE,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YACpC,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;QAClD,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;IAClD,CAAC;IAES,QAAQ,CAAC,OAAmB;QACpC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM;YAAE,OAAO,SAAS,CAAC;QAC7C,OAAO,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC/B,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC,CAAC,UAAU,EAAE;SACjF,CAAC,CAAC,CAAC;IACN,CAAC;IAES,QAAQ,CAAC,IAAa;QAC9B,OAAO;YACL,WAAW,EAAG,IAAY,EAAE,KAAK,EAAE,aAAa,IAAI,CAAC;YACrD,YAAY,EAAG,IAAY,EAAE,KAAK,EAAE,iBAAiB,IAAI,CAAC;YAC1D,iBAAiB,EAAG,IAAY,EAAE,KAAK,EAAE,qBAAqB,EAAE,aAAa,IAAI,CAAC;YAClF,WAAW,EAAG,IAAY,EAAE,KAAK,EAAE,YAAY,IAAI,CAAC;SACrD,CAAC;IACJ,CAAC;IAES,aAAa,CAAC,IAAa;QACnC,MAAM,MAAM,GAAI,IAAY,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QAC3C,OAAO,MAAM,EAAE,aAAa,IAAI,UAAU,CAAC;IAC7C,CAAC;IAEO,YAAY,CAAC,KAAY;QAC/B,MAAM,SAAS,GAAG,IAAI,CAAC;QACvB,MAAM,UAAU,GAAG,IAAI,CAAC;QACxB,OAAO,CAAC,KAAK,CAAC,WAAW,GAAG,GAAG,CAAC,GAAG,SAAS,GAAG,CAAC,KAAK,CAAC,YAAY,GAAG,GAAG,CAAC,GAAG,UAAU,CAAC;IACzF,CAAC;CACF;AAED,MAAM,UAAU,8BAA8B,CAAC,MAAsB;IACnE,OAAO,IAAI,wBAAwB,CAAC,MAAM,CAAC,CAAC;AAC9C,CAAC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import { BaseProvider } from "../provider.js";
|
|
2
|
+
import type { ProviderConfig, LLMRequest, LLMResponse, StreamChunk, Usage } from "../types.js";
|
|
3
|
+
export declare class OpenAIProvider extends BaseProvider {
|
|
4
|
+
protected defaultBaseUrl(): string;
|
|
5
|
+
protected chatCompletionsEndpoint(): string;
|
|
6
|
+
protected modelsEndpoint(): string;
|
|
7
|
+
protected buildRequestBody(request: LLMRequest, stream: boolean): object;
|
|
8
|
+
protected parseResponse(data: unknown): LLMResponse;
|
|
9
|
+
protected parseStreamChunk(data: unknown): StreamChunk[];
|
|
10
|
+
protected buildMessages(request: LLMRequest): object[];
|
|
11
|
+
protected mapMessage(msg: {
|
|
12
|
+
role: string;
|
|
13
|
+
content: string | object[];
|
|
14
|
+
}): object;
|
|
15
|
+
protected mapTools(request: LLMRequest): object[] | undefined;
|
|
16
|
+
protected mapUsage(data: unknown): Usage;
|
|
17
|
+
protected mapStopReason(data: unknown): LLMResponse["stopReason"];
|
|
18
|
+
private estimateCost;
|
|
19
|
+
}
|
|
20
|
+
export declare function createOpenAIProvider(config: ProviderConfig): OpenAIProvider;
|
|
@@ -0,0 +1,167 @@
|
|
|
1
|
+
// openai.ts — OpenAI provider
|
|
2
|
+
import { BaseProvider } from "../provider.js";
|
|
3
|
+
export class OpenAIProvider extends BaseProvider {
|
|
4
|
+
defaultBaseUrl() {
|
|
5
|
+
return "https://api.openai.com/v1";
|
|
6
|
+
}
|
|
7
|
+
chatCompletionsEndpoint() {
|
|
8
|
+
return `${this.baseUrl}/chat/completions`;
|
|
9
|
+
}
|
|
10
|
+
modelsEndpoint() {
|
|
11
|
+
return `${this.baseUrl}/models`;
|
|
12
|
+
}
|
|
13
|
+
buildRequestBody(request, stream) {
|
|
14
|
+
const messages = this.buildMessages(request);
|
|
15
|
+
const tools = this.mapTools(request);
|
|
16
|
+
const body = {
|
|
17
|
+
model: request.model || this.model,
|
|
18
|
+
messages,
|
|
19
|
+
stream,
|
|
20
|
+
stream_options: stream ? { include_usage: true } : undefined,
|
|
21
|
+
};
|
|
22
|
+
if (tools)
|
|
23
|
+
body.tools = tools;
|
|
24
|
+
if (request.toolChoice)
|
|
25
|
+
body.tool_choice = request.toolChoice;
|
|
26
|
+
if (request.generation) {
|
|
27
|
+
const gen = request.generation;
|
|
28
|
+
if (gen.temperature !== undefined)
|
|
29
|
+
body.temperature = gen.temperature;
|
|
30
|
+
if (gen.topP !== undefined)
|
|
31
|
+
body.top_p = gen.topP;
|
|
32
|
+
if (gen.maxTokens !== undefined)
|
|
33
|
+
body.max_tokens = gen.maxTokens;
|
|
34
|
+
if (gen.stopSequences?.length)
|
|
35
|
+
body.stop = gen.stopSequences;
|
|
36
|
+
if (gen.reasoningEffort)
|
|
37
|
+
body.reasoning_effort = this.mapReasoningEffort(gen.reasoningEffort);
|
|
38
|
+
if (gen.responseFormat)
|
|
39
|
+
body.response_format = gen.responseFormat;
|
|
40
|
+
}
|
|
41
|
+
if (request.providerOptions?.openai) {
|
|
42
|
+
Object.assign(body, request.providerOptions.openai);
|
|
43
|
+
}
|
|
44
|
+
return body;
|
|
45
|
+
}
|
|
46
|
+
parseResponse(data) {
|
|
47
|
+
const choice = data.choices?.[0];
|
|
48
|
+
const message = choice?.message;
|
|
49
|
+
const usage = this.mapUsage(data);
|
|
50
|
+
let content = message?.content ?? "";
|
|
51
|
+
const toolCalls = (message?.tool_calls ?? []).map((tc) => ({
|
|
52
|
+
id: tc.id,
|
|
53
|
+
type: "function",
|
|
54
|
+
function: { name: tc.function.name, arguments: tc.function.arguments },
|
|
55
|
+
}));
|
|
56
|
+
if (toolCalls.length > 0) {
|
|
57
|
+
content = toolCalls.map((tc) => ({
|
|
58
|
+
type: "tool_result",
|
|
59
|
+
toolResult: { toolCallId: tc.id, content: tc.function.arguments },
|
|
60
|
+
}));
|
|
61
|
+
}
|
|
62
|
+
return {
|
|
63
|
+
id: data.id ?? crypto.randomUUID(),
|
|
64
|
+
model: data.model ?? this.model,
|
|
65
|
+
content,
|
|
66
|
+
toolCalls: toolCalls.length > 0 ? toolCalls : undefined,
|
|
67
|
+
usage,
|
|
68
|
+
stopReason: this.mapStopReason(data),
|
|
69
|
+
costUsd: this.estimateCost(usage),
|
|
70
|
+
};
|
|
71
|
+
}
|
|
72
|
+
parseStreamChunk(data) {
|
|
73
|
+
const lines = String(data)
|
|
74
|
+
.split("\n")
|
|
75
|
+
.filter((l) => l.startsWith("data: "))
|
|
76
|
+
.map((l) => l.slice(6))
|
|
77
|
+
.filter((l) => l !== "[DONE]");
|
|
78
|
+
const chunks = [];
|
|
79
|
+
for (const line of lines) {
|
|
80
|
+
try {
|
|
81
|
+
const parsed = JSON.parse(line);
|
|
82
|
+
const choice = parsed.choices?.[0];
|
|
83
|
+
if (!choice)
|
|
84
|
+
continue;
|
|
85
|
+
const delta = choice.delta;
|
|
86
|
+
if (delta?.content) {
|
|
87
|
+
chunks.push({ type: "content", content: delta.content });
|
|
88
|
+
}
|
|
89
|
+
if (delta?.tool_calls) {
|
|
90
|
+
for (const tc of delta.tool_calls) {
|
|
91
|
+
if (tc.index === 0 && tc.function?.name) {
|
|
92
|
+
chunks.push({
|
|
93
|
+
type: "tool_call",
|
|
94
|
+
toolCall: {
|
|
95
|
+
id: tc.id ?? crypto.randomUUID(),
|
|
96
|
+
type: "function",
|
|
97
|
+
function: { name: tc.function.name, arguments: tc.function.arguments ?? "" },
|
|
98
|
+
},
|
|
99
|
+
});
|
|
100
|
+
}
|
|
101
|
+
else if (tc.function?.arguments) {
|
|
102
|
+
chunks.push({
|
|
103
|
+
type: "tool_call_delta",
|
|
104
|
+
toolCallDelta: { index: tc.index, arguments: tc.function.arguments },
|
|
105
|
+
});
|
|
106
|
+
}
|
|
107
|
+
}
|
|
108
|
+
}
|
|
109
|
+
if (parsed.usage) {
|
|
110
|
+
chunks.push({ type: "usage", usage: this.mapUsage(parsed) });
|
|
111
|
+
}
|
|
112
|
+
if (choice.finish_reason) {
|
|
113
|
+
chunks.push({ type: "complete", stopReason: choice.finish_reason });
|
|
114
|
+
}
|
|
115
|
+
}
|
|
116
|
+
catch {
|
|
117
|
+
// ignore parse errors
|
|
118
|
+
}
|
|
119
|
+
}
|
|
120
|
+
return chunks;
|
|
121
|
+
}
|
|
122
|
+
buildMessages(request) {
|
|
123
|
+
const messages = [];
|
|
124
|
+
if (request.system) {
|
|
125
|
+
messages.push({ role: "system", content: request.system });
|
|
126
|
+
}
|
|
127
|
+
for (const msg of request.messages) {
|
|
128
|
+
messages.push(this.mapMessage(msg));
|
|
129
|
+
}
|
|
130
|
+
return messages;
|
|
131
|
+
}
|
|
132
|
+
mapMessage(msg) {
|
|
133
|
+
if (typeof msg.content === "string") {
|
|
134
|
+
return { role: msg.role, content: msg.content };
|
|
135
|
+
}
|
|
136
|
+
return { role: msg.role, content: msg.content };
|
|
137
|
+
}
|
|
138
|
+
mapTools(request) {
|
|
139
|
+
if (!request.tools?.length)
|
|
140
|
+
return undefined;
|
|
141
|
+
return request.tools.map((t) => ({
|
|
142
|
+
type: "function",
|
|
143
|
+
function: { name: t.name, description: t.description, parameters: t.parameters },
|
|
144
|
+
}));
|
|
145
|
+
}
|
|
146
|
+
mapUsage(data) {
|
|
147
|
+
return {
|
|
148
|
+
inputTokens: data?.usage?.prompt_tokens ?? 0,
|
|
149
|
+
outputTokens: data?.usage?.completion_tokens ?? 0,
|
|
150
|
+
cachedInputTokens: data?.usage?.prompt_tokens_details?.cached_tokens ?? 0,
|
|
151
|
+
totalTokens: data?.usage?.total_tokens ?? 0,
|
|
152
|
+
};
|
|
153
|
+
}
|
|
154
|
+
mapStopReason(data) {
|
|
155
|
+
const choice = data?.choices?.[0];
|
|
156
|
+
return choice?.finish_reason ?? "end_turn";
|
|
157
|
+
}
|
|
158
|
+
estimateCost(usage) {
|
|
159
|
+
const inputRate = 2.50;
|
|
160
|
+
const outputRate = 10.00;
|
|
161
|
+
return (usage.inputTokens / 1e6) * inputRate + (usage.outputTokens / 1e6) * outputRate;
|
|
162
|
+
}
|
|
163
|
+
}
|
|
164
|
+
export function createOpenAIProvider(config) {
|
|
165
|
+
return new OpenAIProvider(config);
|
|
166
|
+
}
|
|
167
|
+
//# sourceMappingURL=openai.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../../src/llm/providers/openai.ts"],"names":[],"mappings":"AAAA,8BAA8B;AAE9B,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAW9C,MAAM,OAAO,cAAe,SAAQ,YAAY;IAC3B,cAAc;QAC/B,OAAO,2BAA2B,CAAC;IACrC,CAAC;IAEkB,uBAAuB;QACxC,OAAO,GAAG,IAAI,CAAC,OAAO,mBAAmB,CAAC;IAC5C,CAAC;IAEkB,cAAc;QAC/B,OAAO,GAAG,IAAI,CAAC,OAAO,SAAS,CAAC;IAClC,CAAC;IAEkB,gBAAgB,CAAC,OAAmB,EAAE,MAAe;QACtE,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAErC,MAAM,IAAI,GAA4B;YACpC,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK;YAClC,QAAQ;YACR,MAAM;YACN,cAAc,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS;SAC7D,CAAC;QAEF,IAAI,KAAK;YAAE,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QAC9B,IAAI,OAAO,CAAC,UAAU;YAAE,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,UAAU,CAAC;QAC9D,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YACvB,MAAM,GAAG,GAAG,OAAO,CAAC,UAAU,CAAC;YAC/B,IAAI,GAAG,CAAC,WAAW,KAAK,SAAS;gBAAE,IAAI,CAAC,WAAW,GAAG,GAAG,CAAC,WAAW,CAAC;YACtE,IAAI,GAAG,CAAC,IAAI,KAAK,SAAS;gBAAE,IAAI,CAAC,KAAK,GAAG,GAAG,CAAC,IAAI,CAAC;YAClD,IAAI,GAAG,CAAC,SAAS,KAAK,SAAS;gBAAE,IAAI,CAAC,UAAU,GAAG,GAAG,CAAC,SAAS,CAAC;YACjE,IAAI,GAAG,CAAC,aAAa,EAAE,MAAM;gBAAE,IAAI,CAAC,IAAI,GAAG,GAAG,CAAC,aAAa,CAAC;YAC7D,IAAI,GAAG,CAAC,eAAe;gBAAE,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;YAC9F,IAAI,GAAG,CAAC,cAAc;gBAAE,IAAI,CAAC,eAAe,GAAG,GAAG,CAAC,cAAc,CAAC;QACpE,CAAC;QACD,IAAI,OAAO,CAAC,eAAe,EAAE,MAAM,EAAE,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QACtD,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAEkB,aAAa,CAAC,IAAa;QAC5C,MAAM,MAAM,GAAI,IAAY,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,OAAO,GAAG,MAAM,EAAE,OAAO,CAAC;QAChC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAElC,IAAI,OAAO,GAAsB,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;QACxD,MAAM,SAAS,GAAe,CAAC,OAAO,EAAE,UAAU,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAO,EAAE,EAAE,CAAC,CAAC;YAC1E,EAAE,EAAE,EAAE,CAAC,EAAE;YACT,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC,QAAQ,CAAC,SAAS,EAAE;SACvE,CAAC,CAAC,CAAC;QAEJ,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,OAAO,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBAC/B,IAAI,EAAE,aAAa;gBACnB,UAAU,EAAE,EAAE,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,OAAO,EAAE,EAAE,CAAC,QAAQ,CAAC,SAAS,EAAE;aAClE,CAAC,CAAC,CAAC;QACN,CAAC;QAED,OAAO;YACL,EAAE,EAAG,IAAY,CAAC,EAAE,IAAI,MAAM,CAAC,UAAU,EAAE;YAC3C,KAAK,EAAG,IAAY,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK;YACxC,OAAO;YACP,SAAS,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;YACvD,KAAK;YACL,UAAU,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;YACpC,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC;SAClC,CAAC;IACJ,CAAC;IAEkB,gBAAgB,CAAC,IAAa;QAC/C,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC;aACvB,KAAK,CAAC,IAAI,CAAC;aACX,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;aACrC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;aACtB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC;QAEjC,MAAM,MAAM,GAAkB,EAAE,CAAC;QACjC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAChC,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;gBACnC,IAAI,CAAC,MAAM;oBAAE,SAAS;gBAEtB,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;gBAC3B,IAAI,KAAK,EAAE,OAAO,EAAE,CAAC;oBACnB,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC3D,CAAC;gBACD,IAAI,KAAK,EAAE,UAAU,EAAE,CAAC;oBACtB,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;wBAClC,IAAI,EAAE,CAAC,KAAK,KAAK,CAAC,IAAI,EAAE,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC;4BACxC,MAAM,CAAC,IAAI,CAAC;gCACV,IAAI,EAAE,WAAW;gCACjB,QAAQ,EAAE;oCACR,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,MAAM,CAAC,UAAU,EAAE;oCAChC,IAAI,EAAE,UAAU;oCAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC,QAAQ,CAAC,SAAS,IAAI,EAAE,EAAE;iCAC7E;6BACF,CAAC,CAAC;wBACL,CAAC;6BAAM,IAAI,EAAE,CAAC,QAAQ,EAAE,SAAS,EAAE,CAAC;4BAClC,MAAM,CAAC,IAAI,CAAC;gCACV,IAAI,EAAE,iBAAiB;gCACvB,aAAa,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,EAAE,CAAC,QAAQ,CAAC,SAAS,EAAE;6BACrE,CAAC,CAAC;wBACL,CAAC;oBACH,CAAC;gBACH,CAAC;gBACD,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;oBACjB,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBAC/D,CAAC;gBACD,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;oBACzB,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC;gBACtE,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,sBAAsB;YACxB,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAES,aAAa,CAAC,OAAmB;QACzC,MAAM,QAAQ,GAAa,EAAE,CAAC;QAC9B,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAC7D,CAAC;QACD,KAAK,MAAM,GAAG,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACnC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;QACtC,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAES,UAAU,CAAC,GAAiD;QACpE,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YACpC,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;QAClD,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;IAClD,CAAC;IAES,QAAQ,CAAC,OAAmB;QACpC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM;YAAE,OAAO,SAAS,CAAC;QAC7C,OAAO,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC/B,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC,CAAC,UAAU,EAAE;SACjF,CAAC,CAAC,CAAC;IACN,CAAC;IAES,QAAQ,CAAC,IAAa;QAC9B,OAAO;YACL,WAAW,EAAG,IAAY,EAAE,KAAK,EAAE,aAAa,IAAI,CAAC;YACrD,YAAY,EAAG,IAAY,EAAE,KAAK,EAAE,iBAAiB,IAAI,CAAC;YAC1D,iBAAiB,EAAG,IAAY,EAAE,KAAK,EAAE,qBAAqB,EAAE,aAAa,IAAI,CAAC;YAClF,WAAW,EAAG,IAAY,EAAE,KAAK,EAAE,YAAY,IAAI,CAAC;SACrD,CAAC;IACJ,CAAC;IAES,aAAa,CAAC,IAAa;QACnC,MAAM,MAAM,GAAI,IAAY,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QAC3C,OAAO,MAAM,EAAE,aAAa,IAAI,UAAU,CAAC;IAC7C,CAAC;IAEO,YAAY,CAAC,KAAY;QAC/B,MAAM,SAAS,GAAG,IAAI,CAAC;QACvB,MAAM,UAAU,GAAG,KAAK,CAAC;QACzB,OAAO,CAAC,KAAK,CAAC,WAAW,GAAG,GAAG,CAAC,GAAG,SAAS,GAAG,CAAC,KAAK,CAAC,YAAY,GAAG,GAAG,CAAC,GAAG,UAAU,CAAC;IACzF,CAAC;CACF;AAED,MAAM,UAAU,oBAAoB,CAAC,MAAsB;IACzD,OAAO,IAAI,cAAc,CAAC,MAAM,CAAC,CAAC;AACpC,CAAC"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
import type { ModelEntry, ModelCapability, ReasoningEffort } from "./types.js";
|
|
2
|
+
export declare class ModelRegistry {
|
|
3
|
+
private entries;
|
|
4
|
+
private models;
|
|
5
|
+
private patterns;
|
|
6
|
+
constructor(entries?: ModelEntry[]);
|
|
7
|
+
registerEntry(entry: ModelEntry): void;
|
|
8
|
+
get(id: string): ModelEntry | undefined;
|
|
9
|
+
getAll(): ModelEntry[];
|
|
10
|
+
resolve(query: string): ModelEntry | undefined;
|
|
11
|
+
resolveWithFallback(query: string): ModelEntry;
|
|
12
|
+
getUpgradeTarget(entry: ModelEntry): ModelEntry | undefined;
|
|
13
|
+
getReasoningEfforts(modelId: string): ReasoningEffort[];
|
|
14
|
+
supportsCapability(modelId: string, capability: ModelCapability): boolean;
|
|
15
|
+
estimateCost(modelId: string, inputTokens: number, outputTokens: number, cachedInputTokens?: number): number;
|
|
16
|
+
}
|
|
17
|
+
export declare const defaultRegistry: ModelRegistry;
|
|
18
|
+
export declare function getDefaultRegistry(): ModelRegistry;
|
|
19
|
+
export declare function registerModel(entry: ModelEntry): void;
|