@xemahq/llm-client-nest 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/LICENSE +176 -0
- package/README.md +105 -0
- package/dist/index.d.ts +14 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +70 -0
- package/dist/index.js.map +1 -0
- package/dist/lib/dialect.d.ts +15 -0
- package/dist/lib/dialect.d.ts.map +1 -0
- package/dist/lib/dialect.js +63 -0
- package/dist/lib/dialect.js.map +1 -0
- package/dist/lib/errors.d.ts +23 -0
- package/dist/lib/errors.d.ts.map +1 -0
- package/dist/lib/errors.js +47 -0
- package/dist/lib/errors.js.map +1 -0
- package/dist/lib/gateway-session-minter.d.ts +20 -0
- package/dist/lib/gateway-session-minter.d.ts.map +1 -0
- package/dist/lib/gateway-session-minter.js +112 -0
- package/dist/lib/gateway-session-minter.js.map +1 -0
- package/dist/lib/gen-ai.d.ts +45 -0
- package/dist/lib/gen-ai.d.ts.map +1 -0
- package/dist/lib/gen-ai.js +126 -0
- package/dist/lib/gen-ai.js.map +1 -0
- package/dist/lib/json-readers.d.ts +6 -0
- package/dist/lib/json-readers.d.ts.map +1 -0
- package/dist/lib/json-readers.js +32 -0
- package/dist/lib/json-readers.js.map +1 -0
- package/dist/lib/llm-client.d.ts +46 -0
- package/dist/lib/llm-client.d.ts.map +1 -0
- package/dist/lib/llm-client.js +341 -0
- package/dist/lib/llm-client.js.map +1 -0
- package/dist/lib/provider-address.d.ts +3 -0
- package/dist/lib/provider-address.d.ts.map +1 -0
- package/dist/lib/provider-address.js +18 -0
- package/dist/lib/provider-address.js.map +1 -0
- package/dist/lib/session-cache.d.ts +21 -0
- package/dist/lib/session-cache.d.ts.map +1 -0
- package/dist/lib/session-cache.js +65 -0
- package/dist/lib/session-cache.js.map +1 -0
- package/dist/lib/stream.d.ts +18 -0
- package/dist/lib/stream.d.ts.map +1 -0
- package/dist/lib/stream.js +185 -0
- package/dist/lib/stream.js.map +1 -0
- package/dist/lib/structured.d.ts +22 -0
- package/dist/lib/structured.d.ts.map +1 -0
- package/dist/lib/structured.js +62 -0
- package/dist/lib/structured.js.map +1 -0
- package/dist/lib/wire.d.ts +45 -0
- package/dist/lib/wire.d.ts.map +1 -0
- package/dist/lib/wire.js +7 -0
- package/dist/lib/wire.js.map +1 -0
- package/dist/nest/llm-client.module.d.ts +15 -0
- package/dist/nest/llm-client.module.d.ts.map +1 -0
- package/dist/nest/llm-client.module.js +61 -0
- package/dist/nest/llm-client.module.js.map +1 -0
- package/dist/nest/typed-client-mint-transport.d.ts +15 -0
- package/dist/nest/typed-client-mint-transport.d.ts.map +1 -0
- package/dist/nest/typed-client-mint-transport.js +32 -0
- package/dist/nest/typed-client-mint-transport.js.map +1 -0
- package/package.json +67 -0
|
@@ -0,0 +1,341 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.LlmClient = exports.DEFAULT_CALL_TIMEOUT_MS = void 0;
|
|
4
|
+
const observability_nest_1 = require("@xemahq/observability-nest");
|
|
5
|
+
const dialect_1 = require("./dialect");
|
|
6
|
+
const errors_1 = require("./errors");
|
|
7
|
+
const gen_ai_1 = require("./gen-ai");
|
|
8
|
+
const json_readers_1 = require("./json-readers");
|
|
9
|
+
const stream_1 = require("./stream");
|
|
10
|
+
const structured_1 = require("./structured");
|
|
11
|
+
exports.DEFAULT_CALL_TIMEOUT_MS = 120_000;
|
|
12
|
+
class LlmClient {
|
|
13
|
+
minter;
|
|
14
|
+
constructor(minter) {
|
|
15
|
+
this.minter = minter;
|
|
16
|
+
}
|
|
17
|
+
async chat(context, request) {
|
|
18
|
+
const { model } = context;
|
|
19
|
+
const providerName = (0, gen_ai_1.genAiProviderName)(model.apiType, model.providerSlug);
|
|
20
|
+
return this.instrument(gen_ai_1.GenAiOperation.Chat, model, providerName, { stream: false, maxTokens: request.maxTokens }, async (span) => {
|
|
21
|
+
const session = await this.minter.acquire(model, context);
|
|
22
|
+
const dialect = (0, dialect_1.selectDialect)(model.apiType);
|
|
23
|
+
const { url, headers } = (0, dialect_1.chatEndpoint)(dialect, session);
|
|
24
|
+
const body = buildChatBody(dialect, request, model);
|
|
25
|
+
const response = await this.send(url, headers, body, request.timeoutMs);
|
|
26
|
+
const result = parseChatResponse(dialect, await readJson(response));
|
|
27
|
+
span.setAttributes({
|
|
28
|
+
...(result.model === undefined
|
|
29
|
+
? {}
|
|
30
|
+
: { [gen_ai_1.ATTR_GEN_AI_RESPONSE_MODEL]: result.model }),
|
|
31
|
+
...(result.responseId === undefined
|
|
32
|
+
? {}
|
|
33
|
+
: { [gen_ai_1.ATTR_GEN_AI_RESPONSE_ID]: result.responseId }),
|
|
34
|
+
...(result.finishReason === undefined
|
|
35
|
+
? {}
|
|
36
|
+
: { [gen_ai_1.ATTR_GEN_AI_RESPONSE_FINISH_REASONS]: [result.finishReason] }),
|
|
37
|
+
...usageAttributes(result.usage),
|
|
38
|
+
});
|
|
39
|
+
return { result, usage: result.usage, responseModel: result.model };
|
|
40
|
+
});
|
|
41
|
+
}
|
|
42
|
+
async chatStructured(context, request, spec) {
|
|
43
|
+
const jsonSchema = (0, structured_1.structuredOutputJsonSchema)(spec);
|
|
44
|
+
const { model } = context;
|
|
45
|
+
const providerName = (0, gen_ai_1.genAiProviderName)(model.apiType, model.providerSlug);
|
|
46
|
+
return this.instrument(gen_ai_1.GenAiOperation.Chat, model, providerName, { stream: false, maxTokens: request.maxTokens }, async (span) => {
|
|
47
|
+
const session = await this.minter.acquire(model, context);
|
|
48
|
+
const dialect = (0, dialect_1.selectDialect)(model.apiType);
|
|
49
|
+
const { url, headers } = (0, dialect_1.chatEndpoint)(dialect, session);
|
|
50
|
+
const response = await this.send(url, headers, {
|
|
51
|
+
...buildChatBody(dialect, request, model),
|
|
52
|
+
...(0, structured_1.structuredOutputBodyFields)(dialect, spec, jsonSchema),
|
|
53
|
+
}, request.timeoutMs);
|
|
54
|
+
const completion = parseChatResponse(dialect, await readJson(response));
|
|
55
|
+
span.setAttributes({
|
|
56
|
+
...(completion.model === undefined
|
|
57
|
+
? {}
|
|
58
|
+
: { [gen_ai_1.ATTR_GEN_AI_RESPONSE_MODEL]: completion.model }),
|
|
59
|
+
...(completion.responseId === undefined
|
|
60
|
+
? {}
|
|
61
|
+
: { [gen_ai_1.ATTR_GEN_AI_RESPONSE_ID]: completion.responseId }),
|
|
62
|
+
...usageAttributes(completion.usage),
|
|
63
|
+
});
|
|
64
|
+
return {
|
|
65
|
+
result: (0, structured_1.parseStructuredOutput)(spec, completion.content),
|
|
66
|
+
usage: completion.usage,
|
|
67
|
+
responseModel: completion.model,
|
|
68
|
+
};
|
|
69
|
+
});
|
|
70
|
+
}
|
|
71
|
+
async *chatStream(context, request) {
|
|
72
|
+
const { model } = context;
|
|
73
|
+
const providerName = (0, gen_ai_1.genAiProviderName)(model.apiType, model.providerSlug);
|
|
74
|
+
const startedAt = Date.now();
|
|
75
|
+
const span = (0, observability_nest_1.getTracer)().startSpan((0, gen_ai_1.genAiSpanName)(gen_ai_1.GenAiOperation.Chat, model.modelId), {
|
|
76
|
+
attributes: {
|
|
77
|
+
[gen_ai_1.ATTR_GEN_AI_OPERATION_NAME]: gen_ai_1.GenAiOperation.Chat,
|
|
78
|
+
[gen_ai_1.ATTR_GEN_AI_PROVIDER_NAME]: providerName,
|
|
79
|
+
[gen_ai_1.ATTR_GEN_AI_REQUEST_MODEL]: model.modelId,
|
|
80
|
+
[gen_ai_1.ATTR_GEN_AI_REQUEST_STREAM]: true,
|
|
81
|
+
...(model.temperature === undefined
|
|
82
|
+
? {}
|
|
83
|
+
: { [gen_ai_1.ATTR_GEN_AI_REQUEST_TEMPERATURE]: model.temperature }),
|
|
84
|
+
...(request.maxTokens === undefined
|
|
85
|
+
? {}
|
|
86
|
+
: { [gen_ai_1.ATTR_GEN_AI_REQUEST_MAX_TOKENS]: request.maxTokens }),
|
|
87
|
+
},
|
|
88
|
+
});
|
|
89
|
+
let usage = {};
|
|
90
|
+
let responseModel;
|
|
91
|
+
let errorType;
|
|
92
|
+
try {
|
|
93
|
+
const session = await this.minter.acquire(model, context);
|
|
94
|
+
const dialect = (0, dialect_1.selectDialect)(model.apiType);
|
|
95
|
+
const { url, headers } = (0, dialect_1.chatEndpoint)(dialect, session);
|
|
96
|
+
const response = await this.send(url, headers, { ...buildChatBody(dialect, request, model), stream: true }, request.timeoutMs);
|
|
97
|
+
if (response.body === null) {
|
|
98
|
+
throw new errors_1.LlmClientError(errors_1.LlmClientFailure.GatewayResponseMalformed, 'The LLM gateway returned a 2xx streaming response with no body.');
|
|
99
|
+
}
|
|
100
|
+
for await (const event of (0, stream_1.chatStreamEvents)(dialect, response.body)) {
|
|
101
|
+
if (event.kind === 'done') {
|
|
102
|
+
usage = event.result.usage;
|
|
103
|
+
responseModel = event.result.model;
|
|
104
|
+
span.setAttributes({
|
|
105
|
+
...(event.result.model === undefined
|
|
106
|
+
? {}
|
|
107
|
+
: { [gen_ai_1.ATTR_GEN_AI_RESPONSE_MODEL]: event.result.model }),
|
|
108
|
+
...(event.result.responseId === undefined
|
|
109
|
+
? {}
|
|
110
|
+
: { [gen_ai_1.ATTR_GEN_AI_RESPONSE_ID]: event.result.responseId }),
|
|
111
|
+
...(event.result.finishReason === undefined
|
|
112
|
+
? {}
|
|
113
|
+
: {
|
|
114
|
+
[gen_ai_1.ATTR_GEN_AI_RESPONSE_FINISH_REASONS]: [
|
|
115
|
+
event.result.finishReason,
|
|
116
|
+
],
|
|
117
|
+
}),
|
|
118
|
+
...usageAttributes(event.result.usage),
|
|
119
|
+
});
|
|
120
|
+
}
|
|
121
|
+
yield event;
|
|
122
|
+
}
|
|
123
|
+
}
|
|
124
|
+
catch (err) {
|
|
125
|
+
errorType = (0, gen_ai_1.errorTypeOf)(err);
|
|
126
|
+
if (err instanceof Error) {
|
|
127
|
+
span.recordException(err);
|
|
128
|
+
}
|
|
129
|
+
span.setAttributes({ [gen_ai_1.ATTR_ERROR_TYPE]: errorType });
|
|
130
|
+
throw err;
|
|
131
|
+
}
|
|
132
|
+
finally {
|
|
133
|
+
span.end();
|
|
134
|
+
(0, gen_ai_1.recordGenAiMetrics)({
|
|
135
|
+
operation: gen_ai_1.GenAiOperation.Chat,
|
|
136
|
+
providerName,
|
|
137
|
+
requestModel: model.modelId,
|
|
138
|
+
responseModel,
|
|
139
|
+
inputTokens: usage.inputTokens,
|
|
140
|
+
outputTokens: usage.outputTokens,
|
|
141
|
+
durationMs: Date.now() - startedAt,
|
|
142
|
+
errorType,
|
|
143
|
+
});
|
|
144
|
+
}
|
|
145
|
+
}
|
|
146
|
+
async embed(context, input, options) {
|
|
147
|
+
const { model } = context;
|
|
148
|
+
const providerName = (0, gen_ai_1.genAiProviderName)(model.apiType, model.providerSlug);
|
|
149
|
+
return this.instrument(gen_ai_1.GenAiOperation.Embeddings, model, providerName, { stream: false }, async (span) => {
|
|
150
|
+
const session = await this.minter.acquire(model, context);
|
|
151
|
+
const { url, headers } = (0, dialect_1.embeddingsEndpoint)(session);
|
|
152
|
+
const response = await this.send(url, headers, { model: model.modelId, input: [...input] }, options?.timeoutMs);
|
|
153
|
+
const result = parseEmbedResponse(await readJson(response));
|
|
154
|
+
const dimensions = result.embeddings[0]?.length;
|
|
155
|
+
span.setAttributes({
|
|
156
|
+
...(result.model === undefined
|
|
157
|
+
? {}
|
|
158
|
+
: { [gen_ai_1.ATTR_GEN_AI_RESPONSE_MODEL]: result.model }),
|
|
159
|
+
...(dimensions === undefined
|
|
160
|
+
? {}
|
|
161
|
+
: { [gen_ai_1.ATTR_GEN_AI_EMBEDDINGS_DIMENSION_COUNT]: dimensions }),
|
|
162
|
+
...usageAttributes(result.usage),
|
|
163
|
+
});
|
|
164
|
+
return { result, usage: result.usage, responseModel: result.model };
|
|
165
|
+
});
|
|
166
|
+
}
|
|
167
|
+
async instrument(operation, model, providerName, request, run) {
|
|
168
|
+
const startedAt = Date.now();
|
|
169
|
+
try {
|
|
170
|
+
const { result, usage, responseModel } = await (0, observability_nest_1.withSpan)((0, gen_ai_1.genAiSpanName)(operation, model.modelId), run, {
|
|
171
|
+
attributes: {
|
|
172
|
+
[gen_ai_1.ATTR_GEN_AI_OPERATION_NAME]: operation,
|
|
173
|
+
[gen_ai_1.ATTR_GEN_AI_PROVIDER_NAME]: providerName,
|
|
174
|
+
[gen_ai_1.ATTR_GEN_AI_REQUEST_MODEL]: model.modelId,
|
|
175
|
+
[gen_ai_1.ATTR_GEN_AI_REQUEST_STREAM]: request.stream,
|
|
176
|
+
...(model.temperature === undefined
|
|
177
|
+
? {}
|
|
178
|
+
: { [gen_ai_1.ATTR_GEN_AI_REQUEST_TEMPERATURE]: model.temperature }),
|
|
179
|
+
...(request.maxTokens === undefined
|
|
180
|
+
? {}
|
|
181
|
+
: { [gen_ai_1.ATTR_GEN_AI_REQUEST_MAX_TOKENS]: request.maxTokens }),
|
|
182
|
+
},
|
|
183
|
+
});
|
|
184
|
+
(0, gen_ai_1.recordGenAiMetrics)({
|
|
185
|
+
operation,
|
|
186
|
+
providerName,
|
|
187
|
+
requestModel: model.modelId,
|
|
188
|
+
responseModel,
|
|
189
|
+
inputTokens: usage.inputTokens,
|
|
190
|
+
outputTokens: usage.outputTokens,
|
|
191
|
+
durationMs: Date.now() - startedAt,
|
|
192
|
+
});
|
|
193
|
+
return result;
|
|
194
|
+
}
|
|
195
|
+
catch (err) {
|
|
196
|
+
(0, gen_ai_1.recordGenAiMetrics)({
|
|
197
|
+
operation,
|
|
198
|
+
providerName,
|
|
199
|
+
requestModel: model.modelId,
|
|
200
|
+
durationMs: Date.now() - startedAt,
|
|
201
|
+
errorType: (0, gen_ai_1.errorTypeOf)(err),
|
|
202
|
+
});
|
|
203
|
+
throw err;
|
|
204
|
+
}
|
|
205
|
+
}
|
|
206
|
+
async send(url, headers, body, timeoutMs) {
|
|
207
|
+
let response;
|
|
208
|
+
try {
|
|
209
|
+
response = await fetch(url, {
|
|
210
|
+
method: 'POST',
|
|
211
|
+
headers,
|
|
212
|
+
body: JSON.stringify(body),
|
|
213
|
+
signal: AbortSignal.timeout(timeoutMs ?? exports.DEFAULT_CALL_TIMEOUT_MS),
|
|
214
|
+
});
|
|
215
|
+
}
|
|
216
|
+
catch (cause) {
|
|
217
|
+
throw new errors_1.LlmClientError(errors_1.LlmClientFailure.GatewayCallFailed, `The LLM gateway call did not complete: ${cause instanceof Error ? cause.message : String(cause)}`, { cause });
|
|
218
|
+
}
|
|
219
|
+
if (!response.ok) {
|
|
220
|
+
throw new errors_1.LlmClientError(errors_1.LlmClientFailure.GatewayCallFailed, `The LLM gateway returned ${response.status}: ${await (0, errors_1.readDetail)(response)}`, { status: response.status });
|
|
221
|
+
}
|
|
222
|
+
return response;
|
|
223
|
+
}
|
|
224
|
+
}
|
|
225
|
+
exports.LlmClient = LlmClient;
|
|
226
|
+
async function readJson(response) {
|
|
227
|
+
try {
|
|
228
|
+
return await response.json();
|
|
229
|
+
}
|
|
230
|
+
catch (cause) {
|
|
231
|
+
throw new errors_1.LlmClientError(errors_1.LlmClientFailure.GatewayResponseMalformed, 'The LLM gateway returned a 2xx body that is not JSON.', { cause });
|
|
232
|
+
}
|
|
233
|
+
}
|
|
234
|
+
function usageAttributes(usage) {
|
|
235
|
+
return {
|
|
236
|
+
...(usage.inputTokens === undefined
|
|
237
|
+
? {}
|
|
238
|
+
: { [gen_ai_1.ATTR_GEN_AI_USAGE_INPUT_TOKENS]: usage.inputTokens }),
|
|
239
|
+
...(usage.outputTokens === undefined
|
|
240
|
+
? {}
|
|
241
|
+
: { [gen_ai_1.ATTR_GEN_AI_USAGE_OUTPUT_TOKENS]: usage.outputTokens }),
|
|
242
|
+
};
|
|
243
|
+
}
|
|
244
|
+
function buildChatBody(dialect, request, model) {
|
|
245
|
+
if (dialect === dialect_1.GatewayDialect.AnthropicNative) {
|
|
246
|
+
const system = request.messages
|
|
247
|
+
.filter((m) => m.role === 'system')
|
|
248
|
+
.map((m) => m.content)
|
|
249
|
+
.join('\n\n');
|
|
250
|
+
return {
|
|
251
|
+
model: model.modelId,
|
|
252
|
+
...(system === '' ? {} : { system }),
|
|
253
|
+
messages: request.messages
|
|
254
|
+
.filter((m) => m.role !== 'system')
|
|
255
|
+
.map((m) => ({ role: m.role, content: m.content })),
|
|
256
|
+
max_tokens: requireMaxTokens(request.maxTokens),
|
|
257
|
+
...(model.temperature === undefined ? {} : { temperature: model.temperature }),
|
|
258
|
+
...(model.reasoningEffort === undefined
|
|
259
|
+
? {}
|
|
260
|
+
: { output_config: { effort: model.reasoningEffort } }),
|
|
261
|
+
};
|
|
262
|
+
}
|
|
263
|
+
return {
|
|
264
|
+
model: model.modelId,
|
|
265
|
+
messages: request.messages.map((m) => ({ role: m.role, content: m.content })),
|
|
266
|
+
...(model.temperature === undefined ? {} : { temperature: model.temperature }),
|
|
267
|
+
...(model.reasoningEffort === undefined
|
|
268
|
+
? {}
|
|
269
|
+
: { reasoning_effort: model.reasoningEffort }),
|
|
270
|
+
...(request.maxTokens === undefined ? {} : { max_tokens: request.maxTokens }),
|
|
271
|
+
};
|
|
272
|
+
}
|
|
273
|
+
function requireMaxTokens(maxTokens) {
|
|
274
|
+
if (maxTokens === undefined) {
|
|
275
|
+
throw new errors_1.LlmClientError(errors_1.LlmClientFailure.GatewayCallFailed, 'The Anthropic native surface requires `maxTokens`; this client will not ' +
|
|
276
|
+
"invent a platform default for somebody else's model budget.");
|
|
277
|
+
}
|
|
278
|
+
return maxTokens;
|
|
279
|
+
}
|
|
280
|
+
function parseChatResponse(dialect, body) {
|
|
281
|
+
const root = (0, json_readers_1.asRecord)(body, 'chat response');
|
|
282
|
+
if (dialect === dialect_1.GatewayDialect.AnthropicNative) {
|
|
283
|
+
const content = (0, json_readers_1.asArray)(root['content'], 'content')
|
|
284
|
+
.map((part) => {
|
|
285
|
+
const rec = part;
|
|
286
|
+
return typeof rec['text'] === 'string' ? rec['text'] : '';
|
|
287
|
+
})
|
|
288
|
+
.join('');
|
|
289
|
+
const usage = (0, json_readers_1.asRecordOrUndefined)(root['usage']);
|
|
290
|
+
return {
|
|
291
|
+
content,
|
|
292
|
+
model: typeof root['model'] === 'string' ? root['model'] : undefined,
|
|
293
|
+
responseId: typeof root['id'] === 'string' ? root['id'] : undefined,
|
|
294
|
+
finishReason: typeof root['stop_reason'] === 'string' ? root['stop_reason'] : undefined,
|
|
295
|
+
usage: {
|
|
296
|
+
inputTokens: (0, json_readers_1.numberOrUndefined)(usage?.['input_tokens']),
|
|
297
|
+
outputTokens: (0, json_readers_1.numberOrUndefined)(usage?.['output_tokens']),
|
|
298
|
+
},
|
|
299
|
+
};
|
|
300
|
+
}
|
|
301
|
+
const choice = (0, json_readers_1.asArray)(root['choices'], 'choices')[0];
|
|
302
|
+
const message = (0, json_readers_1.asRecordOrUndefined)((0, json_readers_1.asRecordOrUndefined)(choice)?.['message']);
|
|
303
|
+
const content = message?.['content'];
|
|
304
|
+
if (typeof content !== 'string') {
|
|
305
|
+
throw new errors_1.LlmClientError(errors_1.LlmClientFailure.GatewayResponseMalformed, 'The LLM gateway returned a completion with no `choices[0].message.content`.');
|
|
306
|
+
}
|
|
307
|
+
const usage = (0, json_readers_1.asRecordOrUndefined)(root['usage']);
|
|
308
|
+
return {
|
|
309
|
+
content,
|
|
310
|
+
model: typeof root['model'] === 'string' ? root['model'] : undefined,
|
|
311
|
+
responseId: typeof root['id'] === 'string' ? root['id'] : undefined,
|
|
312
|
+
finishReason: typeof (0, json_readers_1.asRecordOrUndefined)(choice)?.['finish_reason'] === 'string'
|
|
313
|
+
? (0, json_readers_1.asRecordOrUndefined)(choice)?.['finish_reason']
|
|
314
|
+
: undefined,
|
|
315
|
+
usage: {
|
|
316
|
+
inputTokens: (0, json_readers_1.numberOrUndefined)(usage?.['prompt_tokens']),
|
|
317
|
+
outputTokens: (0, json_readers_1.numberOrUndefined)(usage?.['completion_tokens']),
|
|
318
|
+
},
|
|
319
|
+
};
|
|
320
|
+
}
|
|
321
|
+
function parseEmbedResponse(body) {
|
|
322
|
+
const root = (0, json_readers_1.asRecord)(body, 'embeddings response');
|
|
323
|
+
const rows = (0, json_readers_1.asArray)(root['data'], 'data');
|
|
324
|
+
const embeddings = rows.map((row) => {
|
|
325
|
+
const vector = (0, json_readers_1.asRecordOrUndefined)(row)?.['embedding'];
|
|
326
|
+
if (!Array.isArray(vector) || vector.some((v) => typeof v !== 'number')) {
|
|
327
|
+
throw new errors_1.LlmClientError(errors_1.LlmClientFailure.GatewayResponseMalformed, 'The LLM gateway returned an embeddings row whose `embedding` is not a ' +
|
|
328
|
+
'numeric vector.');
|
|
329
|
+
}
|
|
330
|
+
return vector;
|
|
331
|
+
});
|
|
332
|
+
const usage = (0, json_readers_1.asRecordOrUndefined)(root['usage']);
|
|
333
|
+
return {
|
|
334
|
+
embeddings,
|
|
335
|
+
model: typeof root['model'] === 'string' ? root['model'] : undefined,
|
|
336
|
+
usage: {
|
|
337
|
+
inputTokens: (0, json_readers_1.numberOrUndefined)(usage?.['prompt_tokens']),
|
|
338
|
+
},
|
|
339
|
+
};
|
|
340
|
+
}
|
|
341
|
+
//# sourceMappingURL=llm-client.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"llm-client.js","sourceRoot":"","sources":["../../src/lib/llm-client.ts"],"names":[],"mappings":";;;AAAA,mEAA4E;AAE5E,uCAA4F;AAC5F,qCAAwE;AACxE,qCAmBkB;AAClB,iDAKwB;AACxB,qCAAkE;AAClE,6CAKsB;AAmBT,QAAA,uBAAuB,GAAG,OAAO,CAAC;AAgG/C,MAAa,SAAS;IACgB;IAApC,YAAoC,MAA4B;QAA5B,WAAM,GAAN,MAAM,CAAsB;IAAG,CAAC;IAG7D,KAAK,CAAC,IAAI,CACf,OAAuB,EACvB,OAAoB;QAEpB,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC;QAC1B,MAAM,YAAY,GAAG,IAAA,0BAAiB,EAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,CAAC;QAC1E,OAAO,IAAI,CAAC,UAAU,CACpB,uBAAc,CAAC,IAAI,EACnB,KAAK,EACL,YAAY,EACZ,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,EAC/C,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;YAC1D,MAAM,OAAO,GAAG,IAAA,uBAAa,EAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC7C,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,IAAA,sBAAY,EAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YACxD,MAAM,IAAI,GAAG,aAAa,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;YACpD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;YACxE,MAAM,MAAM,GAAG,iBAAiB,CAAC,OAAO,EAAE,MAAM,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;YACpE,IAAI,CAAC,aAAa,CAAC;gBACjB,GAAG,CAAC,MAAM,CAAC,KAAK,KAAK,SAAS;oBAC5B,CAAC,CAAC,EAAE;oBACJ,CAAC,CAAC,EAAE,CAAC,mCAA0B,CAAC,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;gBACnD,GAAG,CAAC,MAAM,CAAC,UAAU,KAAK,SAAS;oBACjC,CAAC,CAAC,EAAE;oBACJ,CAAC,CAAC,EAAE,CAAC,gCAAuB,CAAC,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC;gBACrD,GAAG,CAAC,MAAM,CAAC,YAAY,KAAK,SAAS;oBACnC,CAAC,CAAC,EAAE;oBACJ,CAAC,CAAC,EAAE,CAAC,4CAAmC,CAAC,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC;gBACrE,GAAG,eAAe,CAAC,MAAM,CAAC,KAAK,CAAC;aACjC,CAAC,CAAC;YACH,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,aAAa,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;QACtE,CAAC,CACF,CAAC;IACJ,CAAC;IAmBM,KAAK,CAAC,cAAc,CACzB,OAAuB,EACvB,OAAoB,EACpB,IAA6B;QAE7B,MAAM,UAAU,GAAG,IAAA,uCAA0B,EAAC,IAAI,CAAC,CAAC;QACpD,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC;QAC1B,MAAM,YAAY,GAAG,IAAA,0BAAiB,EAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,CAAC;QAC1E,OAAO,IAAI,CAAC,UAAU,CACpB,uBAAc,CAAC,IAAI,EACnB,KAAK,EACL,YAAY,EACZ,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,EAC/C,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;YAC1D,MAAM,OAAO,GAAG,IAAA,uBAAa,EAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC7C,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,IAAA,sBAAY,EAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YACxD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,IAAI,CAC9B,GAAG,EACH,OAAO,EACP;gBACE,GAAG,aAAa,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC;gBACzC,GAAG,IAAA,uCAA0B,EAAC,OAAO,EAAE,IAAI,EAAE,UAAU,CAAC;aACzD,EACD,OAAO,CAAC,SAAS,CAClB,CAAC;YACF,MAAM,UAAU,GAAG,iBAAiB,CAAC,OAAO,EAAE,MAAM,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;YACxE,IAAI,CAAC,aAAa,CAAC;gBACjB,GAAG,CAAC,UAAU,CAAC,KAAK,KAAK,SAAS;oBAChC,CAAC,CAAC,EAAE;oBACJ,CAAC,CAAC,EAAE,CAAC,mCAA0B,CAAC,EAAE,UAAU,CAAC,KAAK,EAAE,CAAC;gBACvD,GAAG,CAAC,UAAU,CAAC,UAAU,KAAK,SAAS;oBACrC,CAAC,CAAC,EAAE;oBACJ,CAAC,CAAC,EAAE,CAAC,gCAAuB,CAAC,EAAE,UAAU,CAAC,UAAU,EAAE,CAAC;gBACzD,GAAG,eAAe,CAAC,UAAU,CAAC,KAAK,CAAC;aACrC,CAAC,CAAC;YAKH,OAAO;gBACL,MAAM,EAAE,IAAA,kCAAqB,EAAC,IAAI,EAAE,UAAU,CAAC,OAAO,CAAC;gBACvD,KAAK,EAAE,UAAU,CAAC,KAAK;gBACvB,aAAa,EAAE,UAAU,CAAC,KAAK;aAChC,CAAC;QACJ,CAAC,CACF,CAAC;IACJ,CAAC;IA6BM,KAAK,CAAC,CAAC,UAAU,CACtB,OAAuB,EACvB,OAAoB;QAEpB,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC;QAC1B,MAAM,YAAY,GAAG,IAAA,0BAAiB,EAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,CAAC;QAC1E,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,IAAI,GAAG,IAAA,8BAAS,GAAE,CAAC,SAAS,CAChC,IAAA,sBAAa,EAAC,uBAAc,CAAC,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,EACjD;YACE,UAAU,EAAE;gBACV,CAAC,mCAA0B,CAAC,EAAE,uBAAc,CAAC,IAAI;gBACjD,CAAC,kCAAyB,CAAC,EAAE,YAAY;gBACzC,CAAC,kCAAyB,CAAC,EAAE,KAAK,CAAC,OAAO;gBAC1C,CAAC,mCAA0B,CAAC,EAAE,IAAI;gBAClC,GAAG,CAAC,KAAK,CAAC,WAAW,KAAK,SAAS;oBACjC,CAAC,CAAC,EAAE;oBACJ,CAAC,CAAC,EAAE,CAAC,wCAA+B,CAAC,EAAE,KAAK,CAAC,WAAW,EAAE,CAAC;gBAC7D,GAAG,CAAC,OAAO,CAAC,SAAS,KAAK,SAAS;oBACjC,CAAC,CAAC,EAAE;oBACJ,CAAC,CAAC,EAAE,CAAC,uCAA8B,CAAC,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC;aAC7D;SACF,CACF,CAAC;QACF,IAAI,KAAK,GAAa,EAAE,CAAC;QACzB,IAAI,aAAiC,CAAC;QACtC,IAAI,SAA6B,CAAC;QAClC,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;YAC1D,MAAM,OAAO,GAAG,IAAA,uBAAa,EAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC7C,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,IAAA,sBAAY,EAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YACxD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,IAAI,CAC9B,GAAG,EACH,OAAO,EAIP,EAAE,GAAG,aAAa,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,EAC3D,OAAO,CAAC,SAAS,CAClB,CAAC;YACF,IAAI,QAAQ,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;gBAC3B,MAAM,IAAI,uBAAc,CACtB,yBAAgB,CAAC,wBAAwB,EACzC,iEAAiE,CAClE,CAAC;YACJ,CAAC;YACD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAA,yBAAgB,EAAC,OAAO,EAAE,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;gBACnE,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBAC1B,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC;oBAC3B,aAAa,GAAG,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC;oBACnC,IAAI,CAAC,aAAa,CAAC;wBACjB,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,KAAK,SAAS;4BAClC,CAAC,CAAC,EAAE;4BACJ,CAAC,CAAC,EAAE,CAAC,mCAA0B,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;wBACzD,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,UAAU,KAAK,SAAS;4BACvC,CAAC,CAAC,EAAE;4BACJ,CAAC,CAAC,EAAE,CAAC,gCAAuB,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;wBAC3D,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY,KAAK,SAAS;4BACzC,CAAC,CAAC,EAAE;4BACJ,CAAC,CAAC;gCACE,CAAC,4CAAmC,CAAC,EAAE;oCACrC,KAAK,CAAC,MAAM,CAAC,YAAY;iCAC1B;6BACF,CAAC;wBACN,GAAG,eAAe,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC;qBACvC,CAAC,CAAC;gBACL,CAAC;gBACD,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,SAAS,GAAG,IAAA,oBAAW,EAAC,GAAG,CAAC,CAAC;YAC7B,IAAI,GAAG,YAAY,KAAK,EAAE,CAAC;gBACzB,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;YAC5B,CAAC;YAMD,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,wBAAe,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;YACrD,MAAM,GAAG,CAAC;QACZ,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,IAAA,2BAAkB,EAAC;gBACjB,SAAS,EAAE,uBAAc,CAAC,IAAI;gBAC9B,YAAY;gBACZ,YAAY,EAAE,KAAK,CAAC,OAAO;gBAC3B,aAAa;gBACb,WAAW,EAAE,KAAK,CAAC,WAAW;gBAC9B,YAAY,EAAE,KAAK,CAAC,YAAY;gBAChC,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;gBAClC,SAAS;aACV,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAGM,KAAK,CAAC,KAAK,CAChB,OAAuB,EACvB,KAAwB,EACxB,OAAyC;QAEzC,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC;QAC1B,MAAM,YAAY,GAAG,IAAA,0BAAiB,EAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,YAAY,CAAC,CAAC;QAC1E,OAAO,IAAI,CAAC,UAAU,CACpB,uBAAc,CAAC,UAAU,EACzB,KAAK,EACL,YAAY,EACZ,EAAE,MAAM,EAAE,KAAK,EAAE,EACjB,KAAK,EAAE,IAAI,EAAE,EAAE;YACb,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;YAC1D,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,IAAA,4BAAkB,EAAC,OAAO,CAAC,CAAC;YACrD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,IAAI,CAC9B,GAAG,EACH,OAAO,EACP,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC,GAAG,KAAK,CAAC,EAAE,EAC3C,OAAO,EAAE,SAAS,CACnB,CAAC;YACF,MAAM,MAAM,GAAG,kBAAkB,CAAC,MAAM,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;YAC5D,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC;YAChD,IAAI,CAAC,aAAa,CAAC;gBACjB,GAAG,CAAC,MAAM,CAAC,KAAK,KAAK,SAAS;oBAC5B,CAAC,CAAC,EAAE;oBACJ,CAAC,CAAC,EAAE,CAAC,mCAA0B,CAAC,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;gBACnD,GAAG,CAAC,UAAU,KAAK,SAAS;oBAC1B,CAAC,CAAC,EAAE;oBACJ,CAAC,CAAC,EAAE,CAAC,+CAAsC,CAAC,EAAE,UAAU,EAAE,CAAC;gBAC7D,GAAG,eAAe,CAAC,MAAM,CAAC,KAAK,CAAC;aACjC,CAAC,CAAC;YACH,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,aAAa,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;QACtE,CAAC,CACF,CAAC;IACJ,CAAC;IAUO,KAAK,CAAC,UAAU,CACtB,SAAyB,EACzB,KAAoB,EACpB,YAAoB,EACpB,OAGC,EACD,GAIE;QAEF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,IAAI,CAAC;YACH,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,aAAa,EAAE,GAAG,MAAM,IAAA,6BAAQ,EACrD,IAAA,sBAAa,EAAC,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,EACvC,GAAG,EACH;gBACE,UAAU,EAAE;oBACV,CAAC,mCAA0B,CAAC,EAAE,SAAS;oBACvC,CAAC,kCAAyB,CAAC,EAAE,YAAY;oBACzC,CAAC,kCAAyB,CAAC,EAAE,KAAK,CAAC,OAAO;oBAC1C,CAAC,mCAA0B,CAAC,EAAE,OAAO,CAAC,MAAM;oBAQ5C,GAAG,CAAC,KAAK,CAAC,WAAW,KAAK,SAAS;wBACjC,CAAC,CAAC,EAAE;wBACJ,CAAC,CAAC,EAAE,CAAC,wCAA+B,CAAC,EAAE,KAAK,CAAC,WAAW,EAAE,CAAC;oBAC7D,GAAG,CAAC,OAAO,CAAC,SAAS,KAAK,SAAS;wBACjC,CAAC,CAAC,EAAE;wBACJ,CAAC,CAAC,EAAE,CAAC,uCAA8B,CAAC,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC;iBAC7D;aACF,CACF,CAAC;YACF,IAAA,2BAAkB,EAAC;gBACjB,SAAS;gBACT,YAAY;gBACZ,YAAY,EAAE,KAAK,CAAC,OAAO;gBAC3B,aAAa;gBACb,WAAW,EAAE,KAAK,CAAC,WAAW;gBAC9B,YAAY,EAAE,KAAK,CAAC,YAAY;gBAChC,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;aACnC,CAAC,CAAC;YACH,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAA,2BAAkB,EAAC;gBACjB,SAAS;gBACT,YAAY;gBACZ,YAAY,EAAE,KAAK,CAAC,OAAO;gBAC3B,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;gBAClC,SAAS,EAAE,IAAA,oBAAW,EAAC,GAAG,CAAC;aAC5B,CAAC,CAAC;YACH,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,IAAI,CAChB,GAAW,EACX,OAA+B,EAC/B,IAAa,EACb,SAA6B;QAE7B,IAAI,QAAkB,CAAC;QACvB,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBAC1B,MAAM,EAAE,MAAM;gBACd,OAAO;gBACP,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;gBAC1B,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,SAAS,IAAI,+BAAuB,CAAC;aAClE,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,uBAAc,CACtB,yBAAgB,CAAC,iBAAiB,EAClC,0CACE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CACvD,EAAE,EACF,EAAE,KAAK,EAAE,CACV,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YAIjB,MAAM,IAAI,uBAAc,CACtB,yBAAgB,CAAC,iBAAiB,EAClC,4BAA4B,QAAQ,CAAC,MAAM,KAAK,MAAM,IAAA,mBAAU,EAAC,QAAQ,CAAC,EAAE,EAC5E,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,EAAE,CAC5B,CAAC;QACJ,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF;AApXD,8BAoXC;AAED,KAAK,UAAU,QAAQ,CAAC,QAAkB;IACxC,IAAI,CAAC;QACH,OAAO,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAC/B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,uBAAc,CACtB,yBAAgB,CAAC,wBAAwB,EACzC,uDAAuD,EACvD,EAAE,KAAK,EAAE,CACV,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CAAC,KAAe;IACtC,OAAO;QACL,GAAG,CAAC,KAAK,CAAC,WAAW,KAAK,SAAS;YACjC,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,EAAE,CAAC,uCAA8B,CAAC,EAAE,KAAK,CAAC,WAAW,EAAE,CAAC;QAC5D,GAAG,CAAC,KAAK,CAAC,YAAY,KAAK,SAAS;YAClC,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,EAAE,CAAC,wCAA+B,CAAC,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC;KAC/D,CAAC;AACJ,CAAC;AA0BD,SAAS,aAAa,CACpB,OAAuB,EACvB,OAAoB,EACpB,KAAoB;IAMpB,IAAI,OAAO,KAAK,wBAAc,CAAC,eAAe,EAAE,CAAC;QAC/C,MAAM,MAAM,GAAG,OAAO,CAAC,QAAQ;aAC5B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;aAClC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;aACrB,IAAI,CAAC,MAAM,CAAC,CAAC;QAChB,OAAO;YACL,KAAK,EAAE,KAAK,CAAC,OAAO;YAGpB,GAAG,CAAC,MAAM,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC;YACpC,QAAQ,EAAE,OAAO,CAAC,QAAQ;iBACvB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;iBAClC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;YAIrD,UAAU,EAAE,gBAAgB,CAAC,OAAO,CAAC,SAAS,CAAC;YAC/C,GAAG,CAAC,KAAK,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE,CAAC;YAC9E,GAAG,CAAC,KAAK,CAAC,eAAe,KAAK,SAAS;gBACrC,CAAC,CAAC,EAAE;gBACJ,CAAC,CAAC,EAAE,aAAa,EAAE,EAAE,MAAM,EAAE,KAAK,CAAC,eAAe,EAAE,EAAE,CAAC;SAC1D,CAAC;IACJ,CAAC;IACD,OAAO;QACL,KAAK,EAAE,KAAK,CAAC,OAAO;QACpB,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;QAC7E,GAAG,CAAC,KAAK,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE,CAAC;QAC9E,GAAG,CAAC,KAAK,CAAC,eAAe,KAAK,SAAS;YACrC,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,EAAE,gBAAgB,EAAE,KAAK,CAAC,eAAe,EAAE,CAAC;QAChD,GAAG,CAAC,OAAO,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC;KAC9E,CAAC;AACJ,CAAC;AAED,SAAS,gBAAgB,CAAC,SAA6B;IACrD,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;QAC5B,MAAM,IAAI,uBAAc,CACtB,yBAAgB,CAAC,iBAAiB,EAClC,0EAA0E;YACxE,6DAA6D,CAChE,CAAC;IACJ,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,iBAAiB,CACxB,OAAuB,EACvB,IAAa;IAEb,MAAM,IAAI,GAAG,IAAA,uBAAQ,EAAC,IAAI,EAAE,eAAe,CAAC,CAAC;IAC7C,IAAI,OAAO,KAAK,wBAAc,CAAC,eAAe,EAAE,CAAC;QAC/C,MAAM,OAAO,GAAG,IAAA,sBAAO,EAAC,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,CAAC;aAChD,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YACZ,MAAM,GAAG,GAAG,IAA+B,CAAC;YAC5C,OAAO,OAAO,GAAG,CAAC,MAAM,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC5D,CAAC,CAAC;aACD,IAAI,CAAC,EAAE,CAAC,CAAC;QACZ,MAAM,KAAK,GAAG,IAAA,kCAAmB,EAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;QACjD,OAAO;YACL,OAAO;YACP,KAAK,EAAE,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS;YACpE,UAAU,EAAE,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;YACnE,YAAY,EACV,OAAO,IAAI,CAAC,aAAa,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,SAAS;YAC3E,KAAK,EAAE;gBACL,WAAW,EAAE,IAAA,gCAAiB,EAAC,KAAK,EAAE,CAAC,cAAc,CAAC,CAAC;gBACvD,YAAY,EAAE,IAAA,gCAAiB,EAAC,KAAK,EAAE,CAAC,eAAe,CAAC,CAAC;aAC1D;SACF,CAAC;IACJ,CAAC;IACD,MAAM,MAAM,GAAG,IAAA,sBAAO,EAAC,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;IACtD,MAAM,OAAO,GAAG,IAAA,kCAAmB,EACjC,IAAA,kCAAmB,EAAC,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,CACzC,CAAC;IACF,MAAM,OAAO,GAAG,OAAO,EAAE,CAAC,SAAS,CAAC,CAAC;IACrC,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAChC,MAAM,IAAI,uBAAc,CACtB,yBAAgB,CAAC,wBAAwB,EACzC,6EAA6E,CAC9E,CAAC;IACJ,CAAC;IACD,MAAM,KAAK,GAAG,IAAA,kCAAmB,EAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;IACjD,OAAO;QACL,OAAO;QACP,KAAK,EAAE,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS;QACpE,UAAU,EAAE,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;QACnE,YAAY,EACV,OAAO,IAAA,kCAAmB,EAAC,MAAM,CAAC,EAAE,CAAC,eAAe,CAAC,KAAK,QAAQ;YAChE,CAAC,CAAE,IAAA,kCAAmB,EAAC,MAAM,CAAC,EAAE,CAAC,eAAe,CAAY;YAC5D,CAAC,CAAC,SAAS;QACf,KAAK,EAAE;YACL,WAAW,EAAE,IAAA,gCAAiB,EAAC,KAAK,EAAE,CAAC,eAAe,CAAC,CAAC;YACxD,YAAY,EAAE,IAAA,gCAAiB,EAAC,KAAK,EAAE,CAAC,mBAAmB,CAAC,CAAC;SAC9D;KACF,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CAAC,IAAa;IACvC,MAAM,IAAI,GAAG,IAAA,uBAAQ,EAAC,IAAI,EAAE,qBAAqB,CAAC,CAAC;IACnD,MAAM,IAAI,GAAG,IAAA,sBAAO,EAAC,IAAI,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC,CAAC;IAC3C,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QAClC,MAAM,MAAM,GAAG,IAAA,kCAAmB,EAAC,GAAG,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC;QACvD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,EAAE,CAAC;YACxE,MAAM,IAAI,uBAAc,CACtB,yBAAgB,CAAC,wBAAwB,EACzC,wEAAwE;gBACtE,iBAAiB,CACpB,CAAC;QACJ,CAAC;QACD,OAAO,MAAkB,CAAC;IAC5B,CAAC,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAA,kCAAmB,EAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;IACjD,OAAO;QACL,UAAU;QACV,KAAK,EAAE,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS;QACpE,KAAK,EAAE;YACL,WAAW,EAAE,IAAA,gCAAiB,EAAC,KAAK,EAAE,CAAC,eAAe,CAAC,CAAC;SACzD;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"provider-address.d.ts","sourceRoot":"","sources":["../../src/lib/provider-address.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,QAAQ,CAAC;AAwB5C,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,aAAa,GAAG,MAAM,CAepE"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.resolveBrokerProviderId = resolveBrokerProviderId;
|
|
4
|
+
const errors_1 = require("./errors");
|
|
5
|
+
function resolveBrokerProviderId(model) {
|
|
6
|
+
const slug = model.providerSlug?.trim();
|
|
7
|
+
if (slug !== undefined && slug !== '') {
|
|
8
|
+
return slug;
|
|
9
|
+
}
|
|
10
|
+
const id = model.providerId?.trim();
|
|
11
|
+
if (id !== undefined && id !== '') {
|
|
12
|
+
return id;
|
|
13
|
+
}
|
|
14
|
+
throw new errors_1.LlmClientError(errors_1.LlmClientFailure.ProviderAddressMissing, `The resolved model "${model.modelId}" carried no usable provider address ` +
|
|
15
|
+
'(neither providerSlug nor the registry providerId); refusing to mint a ' +
|
|
16
|
+
'gateway token against an ambiguous provider.');
|
|
17
|
+
}
|
|
18
|
+
//# sourceMappingURL=provider-address.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"provider-address.js","sourceRoot":"","sources":["../../src/lib/provider-address.ts"],"names":[],"mappings":";;AA0BA,0DAeC;AAzCD,qCAA4D;AA0B5D,SAAgB,uBAAuB,CAAC,KAAoB;IAC1D,MAAM,IAAI,GAAG,KAAK,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC;IACxC,IAAI,IAAI,KAAK,SAAS,IAAI,IAAI,KAAK,EAAE,EAAE,CAAC;QACtC,OAAO,IAAI,CAAC;IACd,CAAC;IACD,MAAM,EAAE,GAAG,KAAK,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC;IACpC,IAAI,EAAE,KAAK,SAAS,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;QAClC,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,MAAM,IAAI,uBAAc,CACtB,yBAAgB,CAAC,sBAAsB,EACvC,uBAAuB,KAAK,CAAC,OAAO,uCAAuC;QACzE,yEAAyE;QACzE,8CAA8C,CACjD,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
import type { GatewaySession } from './wire';
|
|
2
|
+
export declare const GRANT_REFRESH_BUFFER_MS = 60000;
|
|
3
|
+
export declare const MAX_CACHED_GRANTS = 512;
|
|
4
|
+
export declare function grantCacheKey(parts: {
|
|
5
|
+
readonly orgId: string;
|
|
6
|
+
readonly projectId?: string | undefined;
|
|
7
|
+
readonly providerId: string;
|
|
8
|
+
readonly modelId: string;
|
|
9
|
+
}): string;
|
|
10
|
+
export declare class GrantCache {
|
|
11
|
+
private readonly maxEntries;
|
|
12
|
+
private readonly now;
|
|
13
|
+
private readonly entries;
|
|
14
|
+
private readonly inFlight;
|
|
15
|
+
constructor(maxEntries?: number, now?: () => number);
|
|
16
|
+
getOrLoad(key: string, load: () => Promise<GatewaySession>): Promise<GatewaySession>;
|
|
17
|
+
invalidate(key: string): void;
|
|
18
|
+
get size(): number;
|
|
19
|
+
private store;
|
|
20
|
+
}
|
|
21
|
+
//# sourceMappingURL=session-cache.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session-cache.d.ts","sourceRoot":"","sources":["../../src/lib/session-cache.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,QAAQ,CAAC;AAW7C,eAAO,MAAM,uBAAuB,QAAS,CAAC;AAS9C,eAAO,MAAM,iBAAiB,MAAM,CAAC;AAqCrC,wBAAgB,aAAa,CAAC,KAAK,EAAE;IACnC,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,SAAS,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACxC,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;CAC1B,GAAG,MAAM,CAIT;AAcD,qBAAa,UAAU;IAKnB,OAAO,CAAC,QAAQ,CAAC,UAAU;IAC3B,OAAO,CAAC,QAAQ,CAAC,GAAG;IALtB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAiC;IACzD,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAA8C;gBAGpD,UAAU,GAAE,MAA0B,EACtC,GAAG,GAAE,MAAM,MAAiB;IAWxC,SAAS,CACd,GAAG,EAAE,MAAM,EACX,IAAI,EAAE,MAAM,OAAO,CAAC,cAAc,CAAC,GAClC,OAAO,CAAC,cAAc,CAAC;IA0BnB,UAAU,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;IAKpC,IAAW,IAAI,IAAI,MAAM,CAExB;IAED,OAAO,CAAC,KAAK;CAuBd"}
|
|
@@ -0,0 +1,65 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.GrantCache = exports.MAX_CACHED_GRANTS = exports.GRANT_REFRESH_BUFFER_MS = void 0;
|
|
4
|
+
exports.grantCacheKey = grantCacheKey;
|
|
5
|
+
exports.GRANT_REFRESH_BUFFER_MS = 60_000;
|
|
6
|
+
exports.MAX_CACHED_GRANTS = 512;
|
|
7
|
+
function grantCacheKey(parts) {
|
|
8
|
+
return [parts.orgId, parts.projectId ?? '', parts.providerId, parts.modelId]
|
|
9
|
+
.map((segment) => `${segment.length}:${segment}`)
|
|
10
|
+
.join('');
|
|
11
|
+
}
|
|
12
|
+
class GrantCache {
|
|
13
|
+
maxEntries;
|
|
14
|
+
now;
|
|
15
|
+
entries = new Map();
|
|
16
|
+
inFlight = new Map();
|
|
17
|
+
constructor(maxEntries = exports.MAX_CACHED_GRANTS, now = Date.now) {
|
|
18
|
+
this.maxEntries = maxEntries;
|
|
19
|
+
this.now = now;
|
|
20
|
+
}
|
|
21
|
+
getOrLoad(key, load) {
|
|
22
|
+
const cached = this.entries.get(key);
|
|
23
|
+
if (cached !== undefined && cached.expiresAtMs > this.now()) {
|
|
24
|
+
return Promise.resolve(cached.session);
|
|
25
|
+
}
|
|
26
|
+
const pending = this.inFlight.get(key);
|
|
27
|
+
if (pending !== undefined) {
|
|
28
|
+
return pending;
|
|
29
|
+
}
|
|
30
|
+
const started = load()
|
|
31
|
+
.then((session) => {
|
|
32
|
+
this.store(key, session);
|
|
33
|
+
return session;
|
|
34
|
+
})
|
|
35
|
+
.finally(() => {
|
|
36
|
+
this.inFlight.delete(key);
|
|
37
|
+
});
|
|
38
|
+
this.inFlight.set(key, started);
|
|
39
|
+
return started;
|
|
40
|
+
}
|
|
41
|
+
invalidate(key) {
|
|
42
|
+
this.entries.delete(key);
|
|
43
|
+
}
|
|
44
|
+
get size() {
|
|
45
|
+
return this.entries.size;
|
|
46
|
+
}
|
|
47
|
+
store(key, session) {
|
|
48
|
+
const lifetimeMs = session.expiresIn * 1000 - exports.GRANT_REFRESH_BUFFER_MS;
|
|
49
|
+
if (lifetimeMs <= 0) {
|
|
50
|
+
return;
|
|
51
|
+
}
|
|
52
|
+
if (this.entries.size >= this.maxEntries && !this.entries.has(key)) {
|
|
53
|
+
const oldest = this.entries.keys().next();
|
|
54
|
+
if (oldest.done !== true) {
|
|
55
|
+
this.entries.delete(oldest.value);
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
this.entries.set(key, {
|
|
59
|
+
session,
|
|
60
|
+
expiresAtMs: this.now() + lifetimeMs,
|
|
61
|
+
});
|
|
62
|
+
}
|
|
63
|
+
}
|
|
64
|
+
exports.GrantCache = GrantCache;
|
|
65
|
+
//# sourceMappingURL=session-cache.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session-cache.js","sourceRoot":"","sources":["../../src/lib/session-cache.ts"],"names":[],"mappings":";;;AAyDA,sCASC;AAvDY,QAAA,uBAAuB,GAAG,MAAM,CAAC;AASjC,QAAA,iBAAiB,GAAG,GAAG,CAAC;AAqCrC,SAAgB,aAAa,CAAC,KAK7B;IACC,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,SAAS,IAAI,EAAE,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,OAAO,CAAC;SACzE,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,IAAI,OAAO,EAAE,CAAC;SAChD,IAAI,CAAC,EAAE,CAAC,CAAC;AACd,CAAC;AAcD,MAAa,UAAU;IAKF;IACA;IALF,OAAO,GAAG,IAAI,GAAG,EAAsB,CAAC;IACxC,QAAQ,GAAG,IAAI,GAAG,EAAmC,CAAC;IAEvE,YACmB,aAAqB,yBAAiB,EACtC,MAAoB,IAAI,CAAC,GAAG;QAD5B,eAAU,GAAV,UAAU,CAA4B;QACtC,QAAG,GAAH,GAAG,CAAyB;IAC5C,CAAC;IAUG,SAAS,CACd,GAAW,EACX,IAAmC;QAEnC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACrC,IAAI,MAAM,KAAK,SAAS,IAAI,MAAM,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;YAC5D,OAAO,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACzC,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACvC,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;YAC1B,OAAO,OAAO,CAAC;QACjB,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,EAAE;aACnB,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE;YAChB,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;YACzB,OAAO,OAAO,CAAC;QACjB,CAAC,CAAC;aACD,OAAO,CAAC,GAAG,EAAE;YACZ,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QACL,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QAChC,OAAO,OAAO,CAAC;IACjB,CAAC;IAOM,UAAU,CAAC,GAAW;QAC3B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IAC3B,CAAC;IAGD,IAAW,IAAI;QACb,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;IAC3B,CAAC;IAEO,KAAK,CAAC,GAAW,EAAE,OAAuB;QAChD,MAAM,UAAU,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,GAAG,+BAAuB,CAAC;QAItE,IAAI,UAAU,IAAI,CAAC,EAAE,CAAC;YACpB,OAAO;QACT,CAAC;QACD,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,IAAI,IAAI,CAAC,UAAU,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YAKnE,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;YAC1C,IAAI,MAAM,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;gBACzB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACpC,CAAC;QACH,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE;YACpB,OAAO;YACP,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,UAAU;SACrC,CAAC,CAAC;IACL,CAAC;CACF;AA9ED,gCA8EC"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
import { GatewayDialect } from './dialect';
|
|
2
|
+
import type { ChatResult } from './llm-client';
|
|
3
|
+
export type ChatStreamEvent = {
|
|
4
|
+
readonly kind: 'delta';
|
|
5
|
+
readonly text: string;
|
|
6
|
+
} | {
|
|
7
|
+
readonly kind: 'done';
|
|
8
|
+
readonly result: ChatResult;
|
|
9
|
+
};
|
|
10
|
+
export declare function sseDataLines(body: ReadableStream<Uint8Array>): AsyncGenerator<string>;
|
|
11
|
+
interface StreamAccumulator {
|
|
12
|
+
push(payload: string): string | undefined | null;
|
|
13
|
+
finish(): ChatResult;
|
|
14
|
+
}
|
|
15
|
+
export declare function accumulatorFor(dialect: GatewayDialect): StreamAccumulator;
|
|
16
|
+
export declare function chatStreamEvents(dialect: GatewayDialect, body: ReadableStream<Uint8Array>): AsyncGenerator<ChatStreamEvent, void>;
|
|
17
|
+
export {};
|
|
18
|
+
//# sourceMappingURL=stream.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stream.d.ts","sourceRoot":"","sources":["../../src/lib/stream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAQ3C,OAAO,KAAK,EAAE,UAAU,EAAY,MAAM,cAAc,CAAC;AAgBzD,MAAM,MAAM,eAAe,GACvB;IAAE,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC;IAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GACjD;IAAE,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAAC,QAAQ,CAAC,MAAM,EAAE,UAAU,CAAA;CAAE,CAAC;AAmB3D,wBAAuB,YAAY,CACjC,IAAI,EAAE,cAAc,CAAC,UAAU,CAAC,GAC/B,cAAc,CAAC,MAAM,CAAC,CAgCxB;AAaD,UAAU,iBAAiB;IACzB,IAAI,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,GAAG,IAAI,CAAC;IACjD,MAAM,IAAI,UAAU,CAAC;CACtB;AA8JD,wBAAgB,cAAc,CAAC,OAAO,EAAE,cAAc,GAAG,iBAAiB,CAOzE;AAUD,wBAAuB,gBAAgB,CACrC,OAAO,EAAE,cAAc,EACvB,IAAI,EAAE,cAAc,CAAC,UAAU,CAAC,GAC/B,cAAc,CAAC,eAAe,EAAE,IAAI,CAAC,CAwBvC"}
|