@omnicross/core 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 +21 -0
- package/NOTICE +57 -0
- package/README.md +15 -0
- package/dist/ApiKeyPoolService-BmMkau07.d.cts +170 -0
- package/dist/ApiKeyPoolService-BmMkau07.d.ts +170 -0
- package/dist/ProviderProxy-f_8ziIhW.d.cts +120 -0
- package/dist/ProviderProxy-vjt8sQQk.d.ts +120 -0
- package/dist/SubscriptionAuthSource-Cr4fVEYY.d.cts +264 -0
- package/dist/SubscriptionAuthSource-D89zmiSS.d.ts +264 -0
- package/dist/auth/GeminiCodeAssistProjectResolver.cjs +218 -0
- package/dist/auth/GeminiCodeAssistProjectResolver.d.cts +68 -0
- package/dist/auth/GeminiCodeAssistProjectResolver.d.ts +68 -0
- package/dist/auth/GeminiCodeAssistProjectResolver.js +189 -0
- package/dist/completion/ApiKeyPoolService.cjs +331 -0
- package/dist/completion/ApiKeyPoolService.d.cts +2 -0
- package/dist/completion/ApiKeyPoolService.d.ts +2 -0
- package/dist/completion/ApiKeyPoolService.js +306 -0
- package/dist/completion.cjs +4027 -0
- package/dist/completion.d.cts +17 -0
- package/dist/completion.d.ts +17 -0
- package/dist/completion.js +3983 -0
- package/dist/index-BTSmc9Sm.d.ts +645 -0
- package/dist/index-DXazdTzZ.d.cts +645 -0
- package/dist/index.cjs +10428 -0
- package/dist/index.d.cts +128 -0
- package/dist/index.d.ts +128 -0
- package/dist/index.js +10339 -0
- package/dist/outbound-api/subscriptionRegistryPort.cjs +38 -0
- package/dist/outbound-api/subscriptionRegistryPort.d.cts +73 -0
- package/dist/outbound-api/subscriptionRegistryPort.d.ts +73 -0
- package/dist/outbound-api/subscriptionRegistryPort.js +12 -0
- package/dist/outbound-api.cjs +5264 -0
- package/dist/outbound-api.d.cts +320 -0
- package/dist/outbound-api.d.ts +320 -0
- package/dist/outbound-api.js +5218 -0
- package/dist/pipeline/SubscriptionAuthSource.cjs +131 -0
- package/dist/pipeline/SubscriptionAuthSource.d.cts +3 -0
- package/dist/pipeline/SubscriptionAuthSource.d.ts +3 -0
- package/dist/pipeline/SubscriptionAuthSource.js +103 -0
- package/dist/pipeline/SubscriptionAuthStrategy.cjs +18 -0
- package/dist/pipeline/SubscriptionAuthStrategy.d.cts +61 -0
- package/dist/pipeline/SubscriptionAuthStrategy.d.ts +61 -0
- package/dist/pipeline/SubscriptionAuthStrategy.js +0 -0
- package/dist/ports/gemini-code-assist-resolver.cjs +38 -0
- package/dist/ports/gemini-code-assist-resolver.d.cts +26 -0
- package/dist/ports/gemini-code-assist-resolver.d.ts +26 -0
- package/dist/ports/gemini-code-assist-resolver.js +12 -0
- package/dist/ports.cjs +18 -0
- package/dist/ports.d.cts +15 -0
- package/dist/ports.d.ts +15 -0
- package/dist/ports.js +0 -0
- package/dist/provider-proxy/ingress/providerProxyShared.cjs +2958 -0
- package/dist/provider-proxy/ingress/providerProxyShared.d.cts +77 -0
- package/dist/provider-proxy/ingress/providerProxyShared.d.ts +77 -0
- package/dist/provider-proxy/ingress/providerProxyShared.js +2925 -0
- package/dist/provider-proxy/matchText.cjs +73 -0
- package/dist/provider-proxy/matchText.d.cts +47 -0
- package/dist/provider-proxy/matchText.d.ts +47 -0
- package/dist/provider-proxy/matchText.js +45 -0
- package/dist/provider-proxy/types.cjs +18 -0
- package/dist/provider-proxy/types.d.cts +12 -0
- package/dist/provider-proxy/types.d.ts +12 -0
- package/dist/provider-proxy/types.js +0 -0
- package/dist/provider-proxy.cjs +4667 -0
- package/dist/provider-proxy.d.cts +69 -0
- package/dist/provider-proxy.d.ts +69 -0
- package/dist/provider-proxy.js +4636 -0
- package/dist/serializeError.cjs +82 -0
- package/dist/serializeError.d.cts +24 -0
- package/dist/serializeError.d.ts +24 -0
- package/dist/serializeError.js +57 -0
- package/dist/sse-parser.cjs +456 -0
- package/dist/sse-parser.d.cts +143 -0
- package/dist/sse-parser.d.ts +143 -0
- package/dist/sse-parser.js +430 -0
- package/dist/transformer/TransformerChainExecutor.cjs +321 -0
- package/dist/transformer/TransformerChainExecutor.d.cts +104 -0
- package/dist/transformer/TransformerChainExecutor.d.ts +104 -0
- package/dist/transformer/TransformerChainExecutor.js +294 -0
- package/dist/transformer/TransformerService.cjs +290 -0
- package/dist/transformer/TransformerService.d.cts +138 -0
- package/dist/transformer/TransformerService.d.ts +138 -0
- package/dist/transformer/TransformerService.js +265 -0
- package/dist/transformer/transformers/GeminiCodeAssistTransformer.cjs +1115 -0
- package/dist/transformer/transformers/GeminiCodeAssistTransformer.d.cts +102 -0
- package/dist/transformer/transformers/GeminiCodeAssistTransformer.d.ts +102 -0
- package/dist/transformer/transformers/GeminiCodeAssistTransformer.js +1085 -0
- package/dist/transformer/transformers/GeminiTransformer.cjs +1013 -0
- package/dist/transformer/transformers/GeminiTransformer.d.cts +70 -0
- package/dist/transformer/transformers/GeminiTransformer.d.ts +70 -0
- package/dist/transformer/transformers/GeminiTransformer.js +986 -0
- package/dist/transformer/transformers/OpenAIResponseTransformer.cjs +538 -0
- package/dist/transformer/transformers/OpenAIResponseTransformer.d.cts +53 -0
- package/dist/transformer/transformers/OpenAIResponseTransformer.d.ts +53 -0
- package/dist/transformer/transformers/OpenAIResponseTransformer.js +513 -0
- package/dist/transformer/transformers/OpenCodeGoTransformer.cjs +73 -0
- package/dist/transformer/transformers/OpenCodeGoTransformer.d.cts +51 -0
- package/dist/transformer/transformers/OpenCodeGoTransformer.d.ts +51 -0
- package/dist/transformer/transformers/OpenCodeGoTransformer.js +48 -0
- package/dist/transformer/types.cjs +18 -0
- package/dist/transformer/types.d.cts +405 -0
- package/dist/transformer/types.d.ts +405 -0
- package/dist/transformer/types.js +0 -0
- package/dist/transformer.cjs +3736 -0
- package/dist/transformer.d.cts +33 -0
- package/dist/transformer.d.ts +33 -0
- package/dist/transformer.js +3712 -0
- package/dist/types-CGGrKqC_.d.cts +142 -0
- package/dist/types-CbCN2NQP.d.ts +142 -0
- package/dist/types-DCzHkhJt.d.ts +467 -0
- package/dist/types-DZIQbgp0.d.cts +467 -0
- package/dist/usage-event-sink-BX7FE1NL.d.cts +59 -0
- package/dist/usage-event-sink-BX7FE1NL.d.ts +59 -0
- package/package.json +62 -0
|
@@ -0,0 +1,294 @@
|
|
|
1
|
+
// src/transformer/anthropicBetaInject.ts
|
|
2
|
+
import { isExtendedContextCapable } from "@omnicross/contracts/extended-context";
|
|
3
|
+
var EXTENDED_CONTEXT_BETA = "context-1m-2025-08-07";
|
|
4
|
+
var ANTHROPIC_BETA_HEADER = "anthropic-beta";
|
|
5
|
+
function injectExtendedContextBeta(headers, model, useExtendedContext) {
|
|
6
|
+
if (!useExtendedContext) return;
|
|
7
|
+
if (!isExtendedContextCapable(model)) return;
|
|
8
|
+
let existingValue = "";
|
|
9
|
+
for (const key of Object.keys(headers)) {
|
|
10
|
+
if (key.toLowerCase() === ANTHROPIC_BETA_HEADER) {
|
|
11
|
+
const v = headers[key];
|
|
12
|
+
if (typeof v === "string") existingValue = v;
|
|
13
|
+
if (key !== ANTHROPIC_BETA_HEADER) delete headers[key];
|
|
14
|
+
}
|
|
15
|
+
}
|
|
16
|
+
const parts = existingValue.split(",").map((s) => s.trim()).filter((s) => s.length > 0);
|
|
17
|
+
if (!parts.includes(EXTENDED_CONTEXT_BETA)) {
|
|
18
|
+
parts.push(EXTENDED_CONTEXT_BETA);
|
|
19
|
+
}
|
|
20
|
+
headers[ANTHROPIC_BETA_HEADER] = parts.join(",");
|
|
21
|
+
}
|
|
22
|
+
|
|
23
|
+
// src/transformer/TransformerChainExecutor.ts
|
|
24
|
+
var defaultLogger = {
|
|
25
|
+
debug: (msg, ...args) => console.debug(`[ChainExecutor] ${msg}`, ...args),
|
|
26
|
+
info: (msg, ...args) => console.info(`[ChainExecutor] ${msg}`, ...args),
|
|
27
|
+
warn: (msg, ...args) => console.warn(`[ChainExecutor] ${msg}`, ...args),
|
|
28
|
+
error: (msg, ...args) => console.error(`[ChainExecutor] ${msg}`, ...args)
|
|
29
|
+
};
|
|
30
|
+
var TransformerChainExecutor = class {
|
|
31
|
+
logger;
|
|
32
|
+
constructor(logger) {
|
|
33
|
+
this.logger = logger ?? defaultLogger;
|
|
34
|
+
}
|
|
35
|
+
/**
|
|
36
|
+
* Execute the request transformation chain
|
|
37
|
+
*
|
|
38
|
+
* @param request - Original request body
|
|
39
|
+
* @param provider - LLM provider configuration
|
|
40
|
+
* @param chain - Resolved transformer chain
|
|
41
|
+
* @param options - Execution options
|
|
42
|
+
* @returns Transformed request result
|
|
43
|
+
*/
|
|
44
|
+
async executeRequestChain(request, provider, chain, options = {}) {
|
|
45
|
+
const { endpointTransformer, headers, extendedContext } = options;
|
|
46
|
+
const context = {
|
|
47
|
+
logger: this.logger,
|
|
48
|
+
providerName: provider.name
|
|
49
|
+
};
|
|
50
|
+
let requestBody = request;
|
|
51
|
+
let config = {};
|
|
52
|
+
let bypass = false;
|
|
53
|
+
bypass = this.shouldBypassTransformers(
|
|
54
|
+
chain,
|
|
55
|
+
endpointTransformer,
|
|
56
|
+
requestBody
|
|
57
|
+
);
|
|
58
|
+
if (bypass) {
|
|
59
|
+
if (headers) {
|
|
60
|
+
const cleanHeaders = this.cleanHeaders(headers);
|
|
61
|
+
config.headers = cleanHeaders;
|
|
62
|
+
}
|
|
63
|
+
this.logger.debug("Bypass mode enabled - skipping transformations");
|
|
64
|
+
}
|
|
65
|
+
if (!bypass && endpointTransformer?.transformRequestOut) {
|
|
66
|
+
this.logger.debug("Executing transformRequestOut");
|
|
67
|
+
try {
|
|
68
|
+
const transformOut = await endpointTransformer.transformRequestOut(requestBody, context);
|
|
69
|
+
if (transformOut && typeof transformOut === "object") {
|
|
70
|
+
if ("body" in transformOut) {
|
|
71
|
+
requestBody = transformOut.body;
|
|
72
|
+
config = transformOut.config ?? {};
|
|
73
|
+
} else {
|
|
74
|
+
requestBody = transformOut;
|
|
75
|
+
}
|
|
76
|
+
}
|
|
77
|
+
} catch (error) {
|
|
78
|
+
this.logger.error(`transformRequestOut error: ${this.getErrorMessage(error)}`);
|
|
79
|
+
throw error;
|
|
80
|
+
}
|
|
81
|
+
}
|
|
82
|
+
if (!bypass && chain.providerTransformers.length > 0) {
|
|
83
|
+
this.logger.debug(`Executing ${chain.providerTransformers.length} provider transformers`);
|
|
84
|
+
for (const transformer of chain.providerTransformers) {
|
|
85
|
+
if (transformer.transformRequestIn) {
|
|
86
|
+
try {
|
|
87
|
+
const transformIn = await transformer.transformRequestIn(
|
|
88
|
+
requestBody,
|
|
89
|
+
provider,
|
|
90
|
+
context
|
|
91
|
+
);
|
|
92
|
+
if (transformIn && typeof transformIn === "object") {
|
|
93
|
+
if ("body" in transformIn) {
|
|
94
|
+
requestBody = transformIn.body;
|
|
95
|
+
config = { ...config, ...transformIn.config };
|
|
96
|
+
} else {
|
|
97
|
+
requestBody = transformIn;
|
|
98
|
+
}
|
|
99
|
+
}
|
|
100
|
+
} catch (error) {
|
|
101
|
+
this.logger.error(
|
|
102
|
+
`Provider transformer ${transformer.name} error: ${this.getErrorMessage(error)}`
|
|
103
|
+
);
|
|
104
|
+
throw error;
|
|
105
|
+
}
|
|
106
|
+
}
|
|
107
|
+
}
|
|
108
|
+
}
|
|
109
|
+
if (!bypass && chain.modelTransformers.length > 0) {
|
|
110
|
+
this.logger.debug(`Executing ${chain.modelTransformers.length} model transformers`);
|
|
111
|
+
for (const transformer of chain.modelTransformers) {
|
|
112
|
+
if (transformer.transformRequestIn) {
|
|
113
|
+
try {
|
|
114
|
+
const result = await transformer.transformRequestIn(
|
|
115
|
+
requestBody,
|
|
116
|
+
provider,
|
|
117
|
+
context
|
|
118
|
+
);
|
|
119
|
+
requestBody = result;
|
|
120
|
+
} catch (error) {
|
|
121
|
+
this.logger.error(
|
|
122
|
+
`Model transformer ${transformer.name} error: ${this.getErrorMessage(error)}`
|
|
123
|
+
);
|
|
124
|
+
throw error;
|
|
125
|
+
}
|
|
126
|
+
}
|
|
127
|
+
}
|
|
128
|
+
}
|
|
129
|
+
if (extendedContext?.enabled) {
|
|
130
|
+
if (!config.headers || typeof config.headers !== "object") {
|
|
131
|
+
config.headers = {};
|
|
132
|
+
}
|
|
133
|
+
injectExtendedContextBeta(
|
|
134
|
+
config.headers,
|
|
135
|
+
extendedContext.model,
|
|
136
|
+
true
|
|
137
|
+
);
|
|
138
|
+
}
|
|
139
|
+
return { requestBody, config, bypass };
|
|
140
|
+
}
|
|
141
|
+
/**
|
|
142
|
+
* Execute the response transformation chain
|
|
143
|
+
*
|
|
144
|
+
* @param request - Original request (for context)
|
|
145
|
+
* @param response - Response from provider
|
|
146
|
+
* @param provider - LLM provider configuration
|
|
147
|
+
* @param chain - Resolved transformer chain
|
|
148
|
+
* @param options - Execution options
|
|
149
|
+
* @returns Transformed response
|
|
150
|
+
*/
|
|
151
|
+
async executeResponseChain(request, response, provider, chain, options = {}) {
|
|
152
|
+
const { endpointTransformer } = options;
|
|
153
|
+
const context = {
|
|
154
|
+
logger: this.logger,
|
|
155
|
+
providerName: provider.name
|
|
156
|
+
};
|
|
157
|
+
let finalResponse = response;
|
|
158
|
+
const bypass = this.shouldBypassTransformers(chain, endpointTransformer, request);
|
|
159
|
+
if (bypass) {
|
|
160
|
+
this.logger.debug("Bypass mode - skipping response transformations");
|
|
161
|
+
return finalResponse;
|
|
162
|
+
}
|
|
163
|
+
if (chain.modelTransformers.length > 0) {
|
|
164
|
+
const reversedModelTransformers = [...chain.modelTransformers].reverse();
|
|
165
|
+
this.logger.debug(
|
|
166
|
+
`Executing ${reversedModelTransformers.length} model response transformers (reversed)`
|
|
167
|
+
);
|
|
168
|
+
for (const transformer of reversedModelTransformers) {
|
|
169
|
+
if (transformer.transformResponseOut) {
|
|
170
|
+
try {
|
|
171
|
+
finalResponse = await transformer.transformResponseOut(finalResponse, context);
|
|
172
|
+
} catch (error) {
|
|
173
|
+
this.logger.error(
|
|
174
|
+
`Model transformer ${transformer.name} response error: ${this.getErrorMessage(error)}`
|
|
175
|
+
);
|
|
176
|
+
throw error;
|
|
177
|
+
}
|
|
178
|
+
}
|
|
179
|
+
}
|
|
180
|
+
}
|
|
181
|
+
if (chain.providerTransformers.length > 0) {
|
|
182
|
+
const reversedProviderTransformers = [...chain.providerTransformers].reverse();
|
|
183
|
+
this.logger.debug(
|
|
184
|
+
`Executing ${reversedProviderTransformers.length} provider response transformers (reversed)`
|
|
185
|
+
);
|
|
186
|
+
for (const transformer of reversedProviderTransformers) {
|
|
187
|
+
if (transformer.transformResponseOut) {
|
|
188
|
+
try {
|
|
189
|
+
finalResponse = await transformer.transformResponseOut(finalResponse, context);
|
|
190
|
+
} catch (error) {
|
|
191
|
+
this.logger.error(
|
|
192
|
+
`Provider transformer ${transformer.name} response error: ${this.getErrorMessage(error)}`
|
|
193
|
+
);
|
|
194
|
+
throw error;
|
|
195
|
+
}
|
|
196
|
+
}
|
|
197
|
+
}
|
|
198
|
+
}
|
|
199
|
+
if (endpointTransformer?.transformResponseIn) {
|
|
200
|
+
this.logger.debug("Executing transformResponseIn");
|
|
201
|
+
try {
|
|
202
|
+
finalResponse = await endpointTransformer.transformResponseIn(finalResponse, context);
|
|
203
|
+
} catch (error) {
|
|
204
|
+
this.logger.error(`transformResponseIn error: ${this.getErrorMessage(error)}`);
|
|
205
|
+
throw error;
|
|
206
|
+
}
|
|
207
|
+
}
|
|
208
|
+
return finalResponse;
|
|
209
|
+
}
|
|
210
|
+
/**
|
|
211
|
+
* Execute authentication handler if available
|
|
212
|
+
*
|
|
213
|
+
* @param request - Request body
|
|
214
|
+
* @param provider - LLM provider
|
|
215
|
+
* @param endpointTransformer - Endpoint transformer with auth handler
|
|
216
|
+
* @param context - Transformer context
|
|
217
|
+
* @returns Auth result with potentially modified request and config
|
|
218
|
+
*/
|
|
219
|
+
async executeAuth(request, provider, endpointTransformer, context) {
|
|
220
|
+
let requestBody = request;
|
|
221
|
+
let config = {};
|
|
222
|
+
if (endpointTransformer?.auth) {
|
|
223
|
+
this.logger.debug("Executing auth handler");
|
|
224
|
+
try {
|
|
225
|
+
const auth = await endpointTransformer.auth(requestBody, provider, context);
|
|
226
|
+
if (auth && typeof auth === "object") {
|
|
227
|
+
if ("body" in auth) {
|
|
228
|
+
requestBody = auth.body;
|
|
229
|
+
const authConfig = auth.config;
|
|
230
|
+
if (authConfig) {
|
|
231
|
+
const headers = { ...config.headers ?? {}, ...authConfig.headers ?? {} };
|
|
232
|
+
delete headers["host"];
|
|
233
|
+
config = { ...config, ...authConfig, headers };
|
|
234
|
+
}
|
|
235
|
+
} else {
|
|
236
|
+
requestBody = auth;
|
|
237
|
+
}
|
|
238
|
+
}
|
|
239
|
+
} catch (error) {
|
|
240
|
+
this.logger.error(`Auth handler error: ${this.getErrorMessage(error)}`);
|
|
241
|
+
throw error;
|
|
242
|
+
}
|
|
243
|
+
}
|
|
244
|
+
return { requestBody, config };
|
|
245
|
+
}
|
|
246
|
+
/**
|
|
247
|
+
* Check if transformers should be bypassed (optimization)
|
|
248
|
+
*
|
|
249
|
+
* Bypass is enabled when:
|
|
250
|
+
* - Provider has only one transformer that matches the endpoint transformer
|
|
251
|
+
* - Model has no specific transformers or only the same endpoint transformer
|
|
252
|
+
*/
|
|
253
|
+
shouldBypassTransformers(chain, endpointTransformer, _request) {
|
|
254
|
+
if (!endpointTransformer?.name) {
|
|
255
|
+
return false;
|
|
256
|
+
}
|
|
257
|
+
const providerHasOnlyEndpoint = chain.providerTransformers.length === 1 && chain.providerTransformers[0]?.name === endpointTransformer.name;
|
|
258
|
+
const modelHasNoTransformers = chain.modelTransformers.length === 0;
|
|
259
|
+
const modelHasOnlyEndpoint = chain.modelTransformers.length === 1 && chain.modelTransformers[0]?.name === endpointTransformer.name;
|
|
260
|
+
return providerHasOnlyEndpoint && (modelHasNoTransformers || modelHasOnlyEndpoint);
|
|
261
|
+
}
|
|
262
|
+
/**
|
|
263
|
+
* Clean headers for pass-through
|
|
264
|
+
*/
|
|
265
|
+
cleanHeaders(headers) {
|
|
266
|
+
const result = {};
|
|
267
|
+
if (headers instanceof Headers) {
|
|
268
|
+
headers.forEach((value, key) => {
|
|
269
|
+
if (key.toLowerCase() !== "content-length") {
|
|
270
|
+
result[key] = value;
|
|
271
|
+
}
|
|
272
|
+
});
|
|
273
|
+
} else {
|
|
274
|
+
for (const [key, value] of Object.entries(headers)) {
|
|
275
|
+
if (key.toLowerCase() !== "content-length") {
|
|
276
|
+
result[key] = value;
|
|
277
|
+
}
|
|
278
|
+
}
|
|
279
|
+
}
|
|
280
|
+
return result;
|
|
281
|
+
}
|
|
282
|
+
/**
|
|
283
|
+
* Get error message from unknown error
|
|
284
|
+
*/
|
|
285
|
+
getErrorMessage(error) {
|
|
286
|
+
if (error instanceof Error) {
|
|
287
|
+
return error.message;
|
|
288
|
+
}
|
|
289
|
+
return String(error);
|
|
290
|
+
}
|
|
291
|
+
};
|
|
292
|
+
export {
|
|
293
|
+
TransformerChainExecutor
|
|
294
|
+
};
|
|
@@ -0,0 +1,290 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __defProp = Object.defineProperty;
|
|
3
|
+
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
|
|
4
|
+
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
5
|
+
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
6
|
+
var __export = (target, all) => {
|
|
7
|
+
for (var name in all)
|
|
8
|
+
__defProp(target, name, { get: all[name], enumerable: true });
|
|
9
|
+
};
|
|
10
|
+
var __copyProps = (to, from, except, desc) => {
|
|
11
|
+
if (from && typeof from === "object" || typeof from === "function") {
|
|
12
|
+
for (let key of __getOwnPropNames(from))
|
|
13
|
+
if (!__hasOwnProp.call(to, key) && key !== except)
|
|
14
|
+
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
|
|
15
|
+
}
|
|
16
|
+
return to;
|
|
17
|
+
};
|
|
18
|
+
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
|
|
19
|
+
|
|
20
|
+
// src/transformer/TransformerService.ts
|
|
21
|
+
var TransformerService_exports = {};
|
|
22
|
+
__export(TransformerService_exports, {
|
|
23
|
+
TransformerService: () => TransformerService
|
|
24
|
+
});
|
|
25
|
+
module.exports = __toCommonJS(TransformerService_exports);
|
|
26
|
+
var defaultLogger = {
|
|
27
|
+
debug: (msg, ...args) => console.debug(`[TransformerService] ${msg}`, ...args),
|
|
28
|
+
info: (msg, ...args) => console.info(`[TransformerService] ${msg}`, ...args),
|
|
29
|
+
warn: (msg, ...args) => console.warn(`[TransformerService] ${msg}`, ...args),
|
|
30
|
+
error: (msg, ...args) => console.error(`[TransformerService] ${msg}`, ...args)
|
|
31
|
+
};
|
|
32
|
+
var TransformerService = class {
|
|
33
|
+
transformers = /* @__PURE__ */ new Map();
|
|
34
|
+
logger;
|
|
35
|
+
// Performance optimization: Cache instantiated transformers
|
|
36
|
+
instanceCache = /* @__PURE__ */ new Map();
|
|
37
|
+
// Performance optimization: Cache endpoint transformer results
|
|
38
|
+
endpointTransformersCache = null;
|
|
39
|
+
noEndpointTransformersCache = null;
|
|
40
|
+
constructor(logger) {
|
|
41
|
+
this.logger = logger ?? defaultLogger;
|
|
42
|
+
}
|
|
43
|
+
/**
|
|
44
|
+
* Clear instance caches
|
|
45
|
+
*/
|
|
46
|
+
clearInstanceCache() {
|
|
47
|
+
this.instanceCache.clear();
|
|
48
|
+
this.endpointTransformersCache = null;
|
|
49
|
+
this.noEndpointTransformersCache = null;
|
|
50
|
+
}
|
|
51
|
+
/**
|
|
52
|
+
* Get cache statistics for diagnostics
|
|
53
|
+
*/
|
|
54
|
+
getCacheStats() {
|
|
55
|
+
return {
|
|
56
|
+
instanceCacheSize: this.instanceCache.size,
|
|
57
|
+
hasEndpointCache: this.endpointTransformersCache !== null
|
|
58
|
+
};
|
|
59
|
+
}
|
|
60
|
+
/**
|
|
61
|
+
* Register a transformer by name
|
|
62
|
+
* @param name - Unique transformer name
|
|
63
|
+
* @param transformer - Transformer instance or constructor
|
|
64
|
+
*/
|
|
65
|
+
registerTransformer(name, transformer) {
|
|
66
|
+
this.transformers.set(name, transformer);
|
|
67
|
+
this.clearInstanceCache();
|
|
68
|
+
const endpoint = "endPoint" in transformer ? transformer.endPoint : "prototype" in transformer && transformer.prototype?.endPoint ? transformer.prototype.endPoint : void 0;
|
|
69
|
+
this.logger.info(
|
|
70
|
+
`Registered transformer: ${name}${endpoint ? ` (endpoint: ${endpoint})` : " (no endpoint)"}`
|
|
71
|
+
);
|
|
72
|
+
}
|
|
73
|
+
/**
|
|
74
|
+
* Get a transformer by name
|
|
75
|
+
* @param name - Transformer name
|
|
76
|
+
* @returns Transformer instance/constructor or undefined
|
|
77
|
+
*/
|
|
78
|
+
getTransformer(name) {
|
|
79
|
+
return this.transformers.get(name);
|
|
80
|
+
}
|
|
81
|
+
/**
|
|
82
|
+
* Get all registered transformers
|
|
83
|
+
* @returns Map of all transformers
|
|
84
|
+
*/
|
|
85
|
+
getAllTransformers() {
|
|
86
|
+
return new Map(this.transformers);
|
|
87
|
+
}
|
|
88
|
+
/**
|
|
89
|
+
* Get transformers that have an endpoint defined
|
|
90
|
+
* Uses cache to avoid redundant traversal
|
|
91
|
+
* @returns Array of transformers with endpoints
|
|
92
|
+
*/
|
|
93
|
+
getTransformersWithEndpoint() {
|
|
94
|
+
if (this.endpointTransformersCache) {
|
|
95
|
+
return this.endpointTransformersCache;
|
|
96
|
+
}
|
|
97
|
+
const result = [];
|
|
98
|
+
this.transformers.forEach((transformer, name) => {
|
|
99
|
+
const instance = this.instantiateIfNeeded(transformer);
|
|
100
|
+
if (instance?.endPoint) {
|
|
101
|
+
result.push({ name, transformer: instance });
|
|
102
|
+
}
|
|
103
|
+
});
|
|
104
|
+
this.endpointTransformersCache = result;
|
|
105
|
+
return result;
|
|
106
|
+
}
|
|
107
|
+
/**
|
|
108
|
+
* Get transformers without endpoints
|
|
109
|
+
* Uses cache to avoid redundant traversal
|
|
110
|
+
* @returns Array of transformers without endpoints
|
|
111
|
+
*/
|
|
112
|
+
getTransformersWithoutEndpoint() {
|
|
113
|
+
if (this.noEndpointTransformersCache) {
|
|
114
|
+
return this.noEndpointTransformersCache;
|
|
115
|
+
}
|
|
116
|
+
const result = [];
|
|
117
|
+
this.transformers.forEach((transformer, name) => {
|
|
118
|
+
const instance = this.instantiateIfNeeded(transformer);
|
|
119
|
+
if (instance && !instance.endPoint) {
|
|
120
|
+
result.push({ name, transformer: instance });
|
|
121
|
+
}
|
|
122
|
+
});
|
|
123
|
+
this.noEndpointTransformersCache = result;
|
|
124
|
+
return result;
|
|
125
|
+
}
|
|
126
|
+
/**
|
|
127
|
+
* Remove a transformer by name
|
|
128
|
+
* @param name - Transformer name
|
|
129
|
+
* @returns true if removed, false if not found
|
|
130
|
+
*/
|
|
131
|
+
removeTransformer(name) {
|
|
132
|
+
const removed = this.transformers.delete(name);
|
|
133
|
+
if (removed) {
|
|
134
|
+
this.clearInstanceCache();
|
|
135
|
+
}
|
|
136
|
+
return removed;
|
|
137
|
+
}
|
|
138
|
+
/**
|
|
139
|
+
* Check if a transformer is registered
|
|
140
|
+
* @param name - Transformer name
|
|
141
|
+
* @returns true if registered
|
|
142
|
+
*/
|
|
143
|
+
hasTransformer(name) {
|
|
144
|
+
return this.transformers.has(name);
|
|
145
|
+
}
|
|
146
|
+
/**
|
|
147
|
+
* Initialize the service with default transformers
|
|
148
|
+
* @param defaultTransformers - Map or object of default transformers
|
|
149
|
+
*/
|
|
150
|
+
async initialize(defaultTransformers) {
|
|
151
|
+
try {
|
|
152
|
+
if (defaultTransformers) {
|
|
153
|
+
await this.registerDefaultTransformers(defaultTransformers);
|
|
154
|
+
}
|
|
155
|
+
this.logger.info(
|
|
156
|
+
`TransformerService initialized with ${this.transformers.size} transformers`
|
|
157
|
+
);
|
|
158
|
+
} catch (error) {
|
|
159
|
+
const errorMessage = error instanceof Error ? error.message : String(error);
|
|
160
|
+
this.logger.error(`TransformerService init error: ${errorMessage}`);
|
|
161
|
+
throw error;
|
|
162
|
+
}
|
|
163
|
+
}
|
|
164
|
+
/**
|
|
165
|
+
* Register default transformers from a map
|
|
166
|
+
* @param transformers - Map of transformer name to transformer
|
|
167
|
+
*/
|
|
168
|
+
async registerDefaultTransformers(transformers) {
|
|
169
|
+
for (const [key, TransformerClass] of Object.entries(transformers)) {
|
|
170
|
+
try {
|
|
171
|
+
if (this.isTransformerConstructor(TransformerClass)) {
|
|
172
|
+
const staticName = TransformerClass.TransformerName ?? key;
|
|
173
|
+
this.registerTransformer(staticName, TransformerClass);
|
|
174
|
+
} else {
|
|
175
|
+
const name = TransformerClass.name ?? key;
|
|
176
|
+
TransformerClass.logger = this.logger;
|
|
177
|
+
this.registerTransformer(name, TransformerClass);
|
|
178
|
+
}
|
|
179
|
+
} catch (error) {
|
|
180
|
+
const errorMessage = error instanceof Error ? error.message : String(error);
|
|
181
|
+
this.logger.error(`Failed to register transformer ${key}: ${errorMessage}`);
|
|
182
|
+
}
|
|
183
|
+
}
|
|
184
|
+
}
|
|
185
|
+
/**
|
|
186
|
+
* Resolve a transformer chain configuration to actual transformer instances
|
|
187
|
+
* @param chainConfig - Transformer chain configuration
|
|
188
|
+
* @param modelName - Model name for model-specific transformers
|
|
189
|
+
* @returns Resolved transformer chain with instances
|
|
190
|
+
*/
|
|
191
|
+
resolveTransformerChain(chainConfig, modelName) {
|
|
192
|
+
const result = {
|
|
193
|
+
providerTransformers: [],
|
|
194
|
+
modelTransformers: []
|
|
195
|
+
};
|
|
196
|
+
if (!chainConfig) {
|
|
197
|
+
return result;
|
|
198
|
+
}
|
|
199
|
+
if (chainConfig.use) {
|
|
200
|
+
result.providerTransformers = this.resolveTransformerReferences(chainConfig.use);
|
|
201
|
+
}
|
|
202
|
+
if (modelName && chainConfig[modelName]) {
|
|
203
|
+
const modelConfig = chainConfig[modelName];
|
|
204
|
+
if (Array.isArray(modelConfig)) {
|
|
205
|
+
result.modelTransformers = this.resolveTransformerReferences(modelConfig);
|
|
206
|
+
} else if (modelConfig && typeof modelConfig === "object" && "use" in modelConfig) {
|
|
207
|
+
result.modelTransformers = this.resolveTransformerReferences(modelConfig.use ?? []);
|
|
208
|
+
}
|
|
209
|
+
}
|
|
210
|
+
return result;
|
|
211
|
+
}
|
|
212
|
+
/**
|
|
213
|
+
* Resolve an array of transformer references to instances
|
|
214
|
+
* @param refs - Array of transformer references
|
|
215
|
+
* @returns Array of transformer instances
|
|
216
|
+
*/
|
|
217
|
+
resolveTransformerReferences(refs) {
|
|
218
|
+
const instances = [];
|
|
219
|
+
for (const ref of refs) {
|
|
220
|
+
const [name, options] = Array.isArray(ref) ? ref : [ref, void 0];
|
|
221
|
+
const transformer = this.getTransformer(name);
|
|
222
|
+
if (!transformer) {
|
|
223
|
+
this.logger.warn(`Transformer not found: ${name}`);
|
|
224
|
+
continue;
|
|
225
|
+
}
|
|
226
|
+
const instance = this.instantiateIfNeeded(transformer, options);
|
|
227
|
+
if (instance) {
|
|
228
|
+
instance.logger = this.logger;
|
|
229
|
+
instances.push(instance);
|
|
230
|
+
}
|
|
231
|
+
}
|
|
232
|
+
return instances;
|
|
233
|
+
}
|
|
234
|
+
/**
|
|
235
|
+
* Instantiate a transformer if it's a constructor
|
|
236
|
+
* Uses instance cache to avoid redundant instantiation
|
|
237
|
+
*
|
|
238
|
+
* @param transformer - Transformer instance or constructor
|
|
239
|
+
* @param options - Options to pass to constructor
|
|
240
|
+
* @returns Transformer instance
|
|
241
|
+
*/
|
|
242
|
+
instantiateIfNeeded(transformer, options) {
|
|
243
|
+
if (!this.isTransformerConstructor(transformer)) {
|
|
244
|
+
return transformer;
|
|
245
|
+
}
|
|
246
|
+
const constructorName = transformer.TransformerName || transformer.name || "anonymous";
|
|
247
|
+
const optionsHash = options ? JSON.stringify(options) : "";
|
|
248
|
+
const cacheKey = `${constructorName}:${optionsHash}`;
|
|
249
|
+
if (this.instanceCache.has(cacheKey)) {
|
|
250
|
+
return this.instanceCache.get(cacheKey);
|
|
251
|
+
}
|
|
252
|
+
try {
|
|
253
|
+
const instance = new transformer(options);
|
|
254
|
+
instance.logger = this.logger;
|
|
255
|
+
this.instanceCache.set(cacheKey, instance);
|
|
256
|
+
return instance;
|
|
257
|
+
} catch (error) {
|
|
258
|
+
const errorMessage = error instanceof Error ? error.message : String(error);
|
|
259
|
+
this.logger.error(`Failed to instantiate transformer: ${errorMessage}`);
|
|
260
|
+
return void 0;
|
|
261
|
+
}
|
|
262
|
+
}
|
|
263
|
+
/**
|
|
264
|
+
* Check if a value is a transformer constructor
|
|
265
|
+
* @param value - Value to check
|
|
266
|
+
* @returns true if it's a constructor
|
|
267
|
+
*/
|
|
268
|
+
isTransformerConstructor(value) {
|
|
269
|
+
return typeof value === "function" && value.prototype && (typeof value.prototype.transformRequestIn === "function" || typeof value.prototype.transformRequestOut === "function" || typeof value.prototype.transformResponseIn === "function" || typeof value.prototype.transformResponseOut === "function" || "TransformerName" in value);
|
|
270
|
+
}
|
|
271
|
+
/**
|
|
272
|
+
* Get list of all registered transformer names
|
|
273
|
+
* @returns Array of transformer names
|
|
274
|
+
*/
|
|
275
|
+
getTransformerNames() {
|
|
276
|
+
return Array.from(this.transformers.keys());
|
|
277
|
+
}
|
|
278
|
+
/**
|
|
279
|
+
* Clear all registered transformers
|
|
280
|
+
*/
|
|
281
|
+
clear() {
|
|
282
|
+
this.transformers.clear();
|
|
283
|
+
this.clearInstanceCache();
|
|
284
|
+
this.logger.info("All transformers cleared");
|
|
285
|
+
}
|
|
286
|
+
};
|
|
287
|
+
// Annotate the CommonJS export names for ESM import in node:
|
|
288
|
+
0 && (module.exports = {
|
|
289
|
+
TransformerService
|
|
290
|
+
});
|