cost-katana 2.0.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/README.md +550 -0
- package/dist/analyzers/cost-analyzer.d.ts +28 -0
- package/dist/analyzers/cost-analyzer.d.ts.map +1 -0
- package/dist/analyzers/cost-analyzer.js +165 -0
- package/dist/analyzers/cost-analyzer.js.map +1 -0
- package/dist/analyzers/token-counter.d.ts +21 -0
- package/dist/analyzers/token-counter.d.ts.map +1 -0
- package/dist/analyzers/token-counter.js +100 -0
- package/dist/analyzers/token-counter.js.map +1 -0
- package/dist/analyzers/usage-tracker.d.ts +29 -0
- package/dist/analyzers/usage-tracker.d.ts.map +1 -0
- package/dist/analyzers/usage-tracker.js +142 -0
- package/dist/analyzers/usage-tracker.js.map +1 -0
- package/dist/config/default.d.ts +45 -0
- package/dist/config/default.d.ts.map +1 -0
- package/dist/config/default.js +103 -0
- package/dist/config/default.js.map +1 -0
- package/dist/config/model-types.d.ts +45 -0
- package/dist/config/model-types.d.ts.map +1 -0
- package/dist/config/model-types.js +101 -0
- package/dist/config/model-types.js.map +1 -0
- package/dist/config/pricing/anthropic.d.ts +3 -0
- package/dist/config/pricing/anthropic.d.ts.map +1 -0
- package/dist/config/pricing/anthropic.js +191 -0
- package/dist/config/pricing/anthropic.js.map +1 -0
- package/dist/config/pricing/aws-bedrock.d.ts +3 -0
- package/dist/config/pricing/aws-bedrock.d.ts.map +1 -0
- package/dist/config/pricing/aws-bedrock.js +747 -0
- package/dist/config/pricing/aws-bedrock.js.map +1 -0
- package/dist/config/pricing/cohere.d.ts +3 -0
- package/dist/config/pricing/cohere.d.ts.map +1 -0
- package/dist/config/pricing/cohere.js +371 -0
- package/dist/config/pricing/cohere.js.map +1 -0
- package/dist/config/pricing/deepseek.d.ts +3 -0
- package/dist/config/pricing/deepseek.d.ts.map +1 -0
- package/dist/config/pricing/deepseek.js +102 -0
- package/dist/config/pricing/deepseek.js.map +1 -0
- package/dist/config/pricing/google.d.ts +3 -0
- package/dist/config/pricing/google.d.ts.map +1 -0
- package/dist/config/pricing/google.js +671 -0
- package/dist/config/pricing/google.js.map +1 -0
- package/dist/config/pricing/grok.d.ts +3 -0
- package/dist/config/pricing/grok.d.ts.map +1 -0
- package/dist/config/pricing/grok.js +59 -0
- package/dist/config/pricing/grok.js.map +1 -0
- package/dist/config/pricing/index.d.ts +11 -0
- package/dist/config/pricing/index.d.ts.map +1 -0
- package/dist/config/pricing/index.js +24 -0
- package/dist/config/pricing/index.js.map +1 -0
- package/dist/config/pricing/meta.d.ts +3 -0
- package/dist/config/pricing/meta.d.ts.map +1 -0
- package/dist/config/pricing/meta.js +69 -0
- package/dist/config/pricing/meta.js.map +1 -0
- package/dist/config/pricing/mistral.d.ts +3 -0
- package/dist/config/pricing/mistral.d.ts.map +1 -0
- package/dist/config/pricing/mistral.js +690 -0
- package/dist/config/pricing/mistral.js.map +1 -0
- package/dist/config/pricing/openai.d.ts +3 -0
- package/dist/config/pricing/openai.d.ts.map +1 -0
- package/dist/config/pricing/openai.js +475 -0
- package/dist/config/pricing/openai.js.map +1 -0
- package/dist/config/pricing/xai.d.ts +3 -0
- package/dist/config/pricing/xai.d.ts.map +1 -0
- package/dist/config/pricing/xai.js +141 -0
- package/dist/config/pricing/xai.js.map +1 -0
- package/dist/config/pricing-data.d.ts +55 -0
- package/dist/config/pricing-data.d.ts.map +1 -0
- package/dist/config/pricing-data.js +179 -0
- package/dist/config/pricing-data.js.map +1 -0
- package/dist/feedback/client.d.ts +14 -0
- package/dist/feedback/client.d.ts.map +1 -0
- package/dist/feedback/client.js +95 -0
- package/dist/feedback/client.js.map +1 -0
- package/dist/feedback/index.d.ts +3 -0
- package/dist/feedback/index.d.ts.map +1 -0
- package/dist/feedback/index.js +21 -0
- package/dist/feedback/index.js.map +1 -0
- package/dist/gateway/client.d.ts +77 -0
- package/dist/gateway/client.d.ts.map +1 -0
- package/dist/gateway/client.js +630 -0
- package/dist/gateway/client.js.map +1 -0
- package/dist/gateway/index.d.ts +9 -0
- package/dist/gateway/index.d.ts.map +1 -0
- package/dist/gateway/index.js +66 -0
- package/dist/gateway/index.js.map +1 -0
- package/dist/index.d.ts +183 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +1438 -0
- package/dist/index.js.map +1 -0
- package/dist/optimizers/context-trimmer.d.ts +18 -0
- package/dist/optimizers/context-trimmer.d.ts.map +1 -0
- package/dist/optimizers/context-trimmer.js +221 -0
- package/dist/optimizers/context-trimmer.js.map +1 -0
- package/dist/optimizers/index.d.ts +6 -0
- package/dist/optimizers/index.d.ts.map +1 -0
- package/dist/optimizers/index.js +14 -0
- package/dist/optimizers/index.js.map +1 -0
- package/dist/optimizers/prompt-compressor.d.ts +15 -0
- package/dist/optimizers/prompt-compressor.d.ts.map +1 -0
- package/dist/optimizers/prompt-compressor.js +268 -0
- package/dist/optimizers/prompt-compressor.js.map +1 -0
- package/dist/optimizers/prompt-optimizer.d.ts +29 -0
- package/dist/optimizers/prompt-optimizer.d.ts.map +1 -0
- package/dist/optimizers/prompt-optimizer.js +381 -0
- package/dist/optimizers/prompt-optimizer.js.map +1 -0
- package/dist/optimizers/request-fusion.d.ts +34 -0
- package/dist/optimizers/request-fusion.d.ts.map +1 -0
- package/dist/optimizers/request-fusion.js +284 -0
- package/dist/optimizers/request-fusion.js.map +1 -0
- package/dist/optimizers/suggestion-engine.d.ts +29 -0
- package/dist/optimizers/suggestion-engine.d.ts.map +1 -0
- package/dist/optimizers/suggestion-engine.js +234 -0
- package/dist/optimizers/suggestion-engine.js.map +1 -0
- package/dist/providers/anthropic.d.ts +23 -0
- package/dist/providers/anthropic.d.ts.map +1 -0
- package/dist/providers/anthropic.js +108 -0
- package/dist/providers/anthropic.js.map +1 -0
- package/dist/providers/azure.d.ts +15 -0
- package/dist/providers/azure.d.ts.map +1 -0
- package/dist/providers/azure.js +115 -0
- package/dist/providers/azure.js.map +1 -0
- package/dist/providers/base.d.ts +20 -0
- package/dist/providers/base.d.ts.map +1 -0
- package/dist/providers/base.js +67 -0
- package/dist/providers/base.js.map +1 -0
- package/dist/providers/bedrock.d.ts +23 -0
- package/dist/providers/bedrock.d.ts.map +1 -0
- package/dist/providers/bedrock.js +247 -0
- package/dist/providers/bedrock.js.map +1 -0
- package/dist/providers/cohere.d.ts +17 -0
- package/dist/providers/cohere.d.ts.map +1 -0
- package/dist/providers/cohere.js +122 -0
- package/dist/providers/cohere.js.map +1 -0
- package/dist/providers/deepseek.d.ts +6 -0
- package/dist/providers/deepseek.d.ts.map +1 -0
- package/dist/providers/deepseek.js +17 -0
- package/dist/providers/deepseek.js.map +1 -0
- package/dist/providers/google.d.ts +17 -0
- package/dist/providers/google.d.ts.map +1 -0
- package/dist/providers/google.js +141 -0
- package/dist/providers/google.js.map +1 -0
- package/dist/providers/index.d.ts +13 -0
- package/dist/providers/index.d.ts.map +1 -0
- package/dist/providers/index.js +61 -0
- package/dist/providers/index.js.map +1 -0
- package/dist/providers/openai.d.ts +16 -0
- package/dist/providers/openai.d.ts.map +1 -0
- package/dist/providers/openai.js +194 -0
- package/dist/providers/openai.js.map +1 -0
- package/dist/trace/client.d.ts +48 -0
- package/dist/trace/client.d.ts.map +1 -0
- package/dist/trace/client.js +130 -0
- package/dist/trace/client.js.map +1 -0
- package/dist/trace/index.d.ts +7 -0
- package/dist/trace/index.d.ts.map +1 -0
- package/dist/trace/index.js +30 -0
- package/dist/trace/index.js.map +1 -0
- package/dist/trace/middleware.d.ts +70 -0
- package/dist/trace/middleware.d.ts.map +1 -0
- package/dist/trace/middleware.js +104 -0
- package/dist/trace/middleware.js.map +1 -0
- package/dist/trace/providers.d.ts +83 -0
- package/dist/trace/providers.d.ts.map +1 -0
- package/dist/trace/providers.js +340 -0
- package/dist/trace/providers.js.map +1 -0
- package/dist/trace/service.d.ts +59 -0
- package/dist/trace/service.d.ts.map +1 -0
- package/dist/trace/service.js +425 -0
- package/dist/trace/service.js.map +1 -0
- package/dist/trace/types.d.ts +102 -0
- package/dist/trace/types.d.ts.map +1 -0
- package/dist/trace/types.js +3 -0
- package/dist/trace/types.js.map +1 -0
- package/dist/types/failover.d.ts +29 -0
- package/dist/types/failover.d.ts.map +1 -0
- package/dist/types/failover.js +3 -0
- package/dist/types/failover.js.map +1 -0
- package/dist/types/feedback.d.ts +69 -0
- package/dist/types/feedback.d.ts.map +1 -0
- package/dist/types/feedback.js +3 -0
- package/dist/types/feedback.js.map +1 -0
- package/dist/types/gateway.d.ts +303 -0
- package/dist/types/gateway.d.ts.map +1 -0
- package/dist/types/gateway.js +3 -0
- package/dist/types/gateway.js.map +1 -0
- package/dist/types/index.d.ts +242 -0
- package/dist/types/index.d.ts.map +1 -0
- package/dist/types/index.js +37 -0
- package/dist/types/index.js.map +1 -0
- package/dist/types/models.d.ts +7 -0
- package/dist/types/models.d.ts.map +1 -0
- package/dist/types/models.js +3838 -0
- package/dist/types/models.js.map +1 -0
- package/dist/types/providers.d.ts +128 -0
- package/dist/types/providers.d.ts.map +1 -0
- package/dist/types/providers.js +10 -0
- package/dist/types/providers.js.map +1 -0
- package/dist/types/simplified.d.ts +65 -0
- package/dist/types/simplified.d.ts.map +1 -0
- package/dist/types/simplified.js +4 -0
- package/dist/types/simplified.js.map +1 -0
- package/dist/utils/logger.d.ts +35 -0
- package/dist/utils/logger.d.ts.map +1 -0
- package/dist/utils/logger.js +154 -0
- package/dist/utils/logger.js.map +1 -0
- package/dist/utils/pricing.d.ts +30 -0
- package/dist/utils/pricing.d.ts.map +1 -0
- package/dist/utils/pricing.js +148 -0
- package/dist/utils/pricing.js.map +1 -0
- package/dist/utils/validators.d.ts +10 -0
- package/dist/utils/validators.d.ts.map +1 -0
- package/dist/utils/validators.js +133 -0
- package/dist/utils/validators.js.map +1 -0
- package/package.json +92 -0
|
@@ -0,0 +1,108 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.AnthropicProvider = void 0;
|
|
4
|
+
const base_1 = require("./base");
|
|
5
|
+
const types_1 = require("../types");
|
|
6
|
+
class AnthropicProvider extends base_1.BaseProvider {
|
|
7
|
+
constructor(config) {
|
|
8
|
+
super({ ...config, provider: types_1.AIProvider.Anthropic });
|
|
9
|
+
if (!config.apiKey) {
|
|
10
|
+
throw new Error('Anthropic API key is required');
|
|
11
|
+
}
|
|
12
|
+
this.apiKey = config.apiKey;
|
|
13
|
+
this.apiUrl = config.endpoint || 'https://api.anthropic.com/v1';
|
|
14
|
+
}
|
|
15
|
+
async countTokens(text, _model) {
|
|
16
|
+
return Promise.resolve(Math.ceil(text.length / 3.5));
|
|
17
|
+
}
|
|
18
|
+
async makeRequest(request) {
|
|
19
|
+
const url = `${this.apiUrl}/messages`;
|
|
20
|
+
const body = this.formatRequestBody(request);
|
|
21
|
+
try {
|
|
22
|
+
const response = await fetch(url, {
|
|
23
|
+
method: 'POST',
|
|
24
|
+
headers: {
|
|
25
|
+
'Content-Type': 'application/json',
|
|
26
|
+
'x-api-key': this.apiKey,
|
|
27
|
+
'anthropic-version': '2023-06-01'
|
|
28
|
+
},
|
|
29
|
+
body: JSON.stringify(body)
|
|
30
|
+
});
|
|
31
|
+
if (!response.ok) {
|
|
32
|
+
let errorMsg = response.statusText;
|
|
33
|
+
try {
|
|
34
|
+
const error = await response.json();
|
|
35
|
+
if (typeof error === 'object' && error !== null && 'error' in error) {
|
|
36
|
+
const errObj = error;
|
|
37
|
+
errorMsg = errObj.error?.message || errorMsg;
|
|
38
|
+
}
|
|
39
|
+
}
|
|
40
|
+
catch (_err) {
|
|
41
|
+
console.debug('Error parsing response:', _err);
|
|
42
|
+
}
|
|
43
|
+
throw new Error(`Anthropic API error: ${errorMsg}`);
|
|
44
|
+
}
|
|
45
|
+
const data = (await response.json());
|
|
46
|
+
return this.formatResponse(data, request);
|
|
47
|
+
}
|
|
48
|
+
catch (error) {
|
|
49
|
+
if (error instanceof Error) {
|
|
50
|
+
throw error;
|
|
51
|
+
}
|
|
52
|
+
throw new Error('Failed to make Anthropic request');
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
formatRequestBody(request) {
|
|
56
|
+
const { model, messages, prompt, maxTokens, temperature, topP } = request;
|
|
57
|
+
const requestMessages = messages || [{ role: 'user', content: prompt || '' }];
|
|
58
|
+
return {
|
|
59
|
+
model,
|
|
60
|
+
messages: this.filterSystemMessages(requestMessages),
|
|
61
|
+
system: this.getSystemPrompt(requestMessages),
|
|
62
|
+
max_tokens: maxTokens,
|
|
63
|
+
temperature,
|
|
64
|
+
top_p: topP
|
|
65
|
+
};
|
|
66
|
+
}
|
|
67
|
+
getSystemPrompt(messages) {
|
|
68
|
+
return messages.find(m => m.role === 'system')?.content;
|
|
69
|
+
}
|
|
70
|
+
filterSystemMessages(messages) {
|
|
71
|
+
return messages.filter(m => m.role !== 'system');
|
|
72
|
+
}
|
|
73
|
+
formatResponse(response, request) {
|
|
74
|
+
const content = response.content?.[0]?.text ?? '';
|
|
75
|
+
return {
|
|
76
|
+
id: response.id,
|
|
77
|
+
model: request.model,
|
|
78
|
+
choices: [
|
|
79
|
+
{
|
|
80
|
+
message: {
|
|
81
|
+
role: 'assistant',
|
|
82
|
+
content
|
|
83
|
+
},
|
|
84
|
+
index: 0,
|
|
85
|
+
finishReason: response.stop_reason
|
|
86
|
+
}
|
|
87
|
+
],
|
|
88
|
+
usage: {
|
|
89
|
+
input_tokens: response.usage?.input_tokens ?? 0,
|
|
90
|
+
output_tokens: response.usage?.output_tokens ?? 0
|
|
91
|
+
}
|
|
92
|
+
};
|
|
93
|
+
}
|
|
94
|
+
parseUsage(usage) {
|
|
95
|
+
return {
|
|
96
|
+
input_tokens: usage.input_tokens ?? 0,
|
|
97
|
+
output_tokens: usage.output_tokens ?? 0
|
|
98
|
+
};
|
|
99
|
+
}
|
|
100
|
+
getPromptTokens(usage) {
|
|
101
|
+
return usage.input_tokens;
|
|
102
|
+
}
|
|
103
|
+
getCompletionTokens(usage) {
|
|
104
|
+
return usage.output_tokens;
|
|
105
|
+
}
|
|
106
|
+
}
|
|
107
|
+
exports.AnthropicProvider = AnthropicProvider;
|
|
108
|
+
//# sourceMappingURL=anthropic.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../src/providers/anthropic.ts"],"names":[],"mappings":";;;AAAA,iCAAsC;AACtC,oCAAsD;AA4BtD,MAAa,iBAAkB,SAAQ,mBAAY;IAIjD,YAAY,MAAsB;QAChC,KAAK,CAAC,EAAE,GAAG,MAAM,EAAE,QAAQ,EAAE,kBAAU,CAAC,SAAS,EAAE,CAAC,CAAC;QAErD,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;QACnD,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,QAAQ,IAAI,8BAA8B,CAAC;IAClE,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,IAAY,EAAE,MAAc;QAG5C,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,OAAwB;QACxC,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,WAAW,CAAC;QACtC,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAE7C,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBAChC,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,WAAW,EAAE,IAAI,CAAC,MAAM;oBACxB,mBAAmB,EAAE,YAAY;iBAClC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,IAAI,QAAQ,GAAG,QAAQ,CAAC,UAAU,CAAC;gBACnC,IAAI,CAAC;oBACH,MAAM,KAAK,GAAY,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;oBAC7C,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,IAAI,KAAK,EAAE,CAAC;wBACpE,MAAM,MAAM,GAAG,KAA+B,CAAC;wBAC/C,QAAQ,GAAG,MAAM,CAAC,KAAK,EAAE,OAAO,IAAI,QAAQ,CAAC;oBAC/C,CAAC;gBACH,CAAC;gBAAC,OAAO,IAAI,EAAE,CAAC;oBAEd,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,IAAI,CAAC,CAAC;gBACjD,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,wBAAwB,QAAQ,EAAE,CAAC,CAAC;YACtD,CAAC;YAED,MAAM,IAAI,GAAsB,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAsB,CAAC;YAC7E,OAAO,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAC5C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC3B,MAAM,KAAK,CAAC;YACd,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;IAEO,iBAAiB,CAAC,OAAwB;QAChD,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC;QAG1E,MAAM,eAAe,GAAG,QAAQ,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,IAAI,EAAE,EAAE,CAAC,CAAC;QAE9E,OAAO;YACL,KAAK;YACL,QAAQ,EAAE,IAAI,CAAC,oBAAoB,CAAC,eAAe,CAAC;YACpD,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC;YAC7C,UAAU,EAAE,SAAS;YACrB,WAAW;YACX,KAAK,EAAE,IAAI;SACZ,CAAC;IACJ,CAAC;IAEO,eAAe,CAAC,QAAmB;QACzC,OAAO,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,EAAE,OAAO,CAAC;IAC1D,CAAC;IAEO,oBAAoB,CAAC,QAAmB;QAC9C,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;IACnD,CAAC;IAEO,cAAc,CAAC,QAA2B,EAAE,OAAwB;QAC1E,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;QAClD,OAAO;YACL,EAAE,EAAE,QAAQ,CAAC,EAAE;YACf,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,OAAO,EAAE;gBACP;oBACE,OAAO,EAAE;wBACP,IAAI,EAAE,WAAW;wBACjB,OAAO;qBACR;oBACD,KAAK,EAAE,CAAC;oBACR,YAAY,EAAE,QAAQ,CAAC,WAAW;iBACnC;aACF;YACD,KAAK,EAAE;gBACL,YAAY,EAAE,QAAQ,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC;gBAC/C,aAAa,EAAE,QAAQ,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;aAClD;SACF,CAAC;IACJ,CAAC;IAED,UAAU,CAAC,KAA6B;QACtC,OAAO;YACL,YAAY,EAAE,KAAK,CAAC,YAAY,IAAI,CAAC;YACrC,aAAa,EAAE,KAAK,CAAC,aAAa,IAAI,CAAC;SACxC,CAAC;IACJ,CAAC;IAES,eAAe,CAAC,KAAe;QACvC,OAAQ,KAAwB,CAAC,YAAY,CAAC;IAChD,CAAC;IAES,mBAAmB,CAAC,KAAe;QAC3C,OAAQ,KAAwB,CAAC,aAAa,CAAC;IACjD,CAAC;CACF;AAzHD,8CAyHC"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import { BaseProvider } from './base';
|
|
2
|
+
import { ProviderConfig } from '../types';
|
|
3
|
+
import { ProviderRequest, ProviderResponse, OpenAIUsage, AnyUsage } from '../types/providers';
|
|
4
|
+
export declare class AzureProvider extends BaseProvider {
|
|
5
|
+
private apiKey;
|
|
6
|
+
private apiUrl;
|
|
7
|
+
private apiVersion;
|
|
8
|
+
constructor(config: ProviderConfig);
|
|
9
|
+
countTokens(text: string, _model: string): Promise<number>;
|
|
10
|
+
makeRequest(request: ProviderRequest): Promise<ProviderResponse>;
|
|
11
|
+
parseUsage(usage: unknown): OpenAIUsage;
|
|
12
|
+
protected getPromptTokens(usage: AnyUsage): number;
|
|
13
|
+
protected getCompletionTokens(usage: AnyUsage): number;
|
|
14
|
+
}
|
|
15
|
+
//# sourceMappingURL=azure.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"azure.d.ts","sourceRoot":"","sources":["../../src/providers/azure.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,cAAc,EAAc,MAAM,UAAU,CAAC;AACtD,OAAO,EAAE,eAAe,EAAE,gBAAgB,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAyB9F,qBAAa,aAAc,SAAQ,YAAY;IAC7C,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,UAAU,CAAS;gBAEf,MAAM,EAAE,cAAc;IAc5B,WAAW,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAU1D,WAAW,CAAC,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,gBAAgB,CAAC;IA0EtE,UAAU,CAAC,KAAK,EAAE,OAAO,GAAG,WAAW;IAYvC,SAAS,CAAC,eAAe,CAAC,KAAK,EAAE,QAAQ,GAAG,MAAM;IAIlD,SAAS,CAAC,mBAAmB,CAAC,KAAK,EAAE,QAAQ,GAAG,MAAM;CAGvD"}
|
|
@@ -0,0 +1,115 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.AzureProvider = void 0;
|
|
4
|
+
const base_1 = require("./base");
|
|
5
|
+
const types_1 = require("../types");
|
|
6
|
+
const gpt_3_encoder_1 = require("gpt-3-encoder");
|
|
7
|
+
class AzureProvider extends base_1.BaseProvider {
|
|
8
|
+
constructor(config) {
|
|
9
|
+
super({ ...config, provider: types_1.AIProvider.Azure });
|
|
10
|
+
if (!config.apiKey || !config.resourceName || !config.deploymentId || !config.apiVersion) {
|
|
11
|
+
throw new Error('Azure provider requires apiKey, resourceName, deploymentId, and apiVersion in config');
|
|
12
|
+
}
|
|
13
|
+
this.apiKey = config.apiKey;
|
|
14
|
+
this.apiVersion = config.apiVersion;
|
|
15
|
+
this.apiUrl = `https://${config.resourceName}.openai.azure.com/openai/deployments/${config.deploymentId}`;
|
|
16
|
+
}
|
|
17
|
+
async countTokens(text, _model) {
|
|
18
|
+
try {
|
|
19
|
+
const encoded = (0, gpt_3_encoder_1.encode)(text);
|
|
20
|
+
return Promise.resolve(encoded.length);
|
|
21
|
+
}
|
|
22
|
+
catch (error) {
|
|
23
|
+
console.warn('Token counting failed for Azure model:', error);
|
|
24
|
+
return Promise.resolve(Math.ceil(text.length / 4));
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
async makeRequest(request) {
|
|
28
|
+
const endpoint = request.messages ? '/chat/completions' : '/completions';
|
|
29
|
+
const url = `${this.apiUrl}${endpoint}?api-version=${this.apiVersion}`;
|
|
30
|
+
const body = {
|
|
31
|
+
model: request.model,
|
|
32
|
+
max_tokens: request.maxTokens,
|
|
33
|
+
temperature: request.temperature,
|
|
34
|
+
top_p: request.topP,
|
|
35
|
+
stop: request.stopSequences,
|
|
36
|
+
stream: false,
|
|
37
|
+
user: request.userId
|
|
38
|
+
};
|
|
39
|
+
if (request.messages) {
|
|
40
|
+
body.messages = request.messages;
|
|
41
|
+
}
|
|
42
|
+
else {
|
|
43
|
+
body.prompt = request.prompt;
|
|
44
|
+
}
|
|
45
|
+
try {
|
|
46
|
+
const response = await fetch(url, {
|
|
47
|
+
method: 'POST',
|
|
48
|
+
headers: {
|
|
49
|
+
'Content-Type': 'application/json',
|
|
50
|
+
'api-key': this.apiKey
|
|
51
|
+
},
|
|
52
|
+
body: JSON.stringify(body)
|
|
53
|
+
});
|
|
54
|
+
if (!response.ok) {
|
|
55
|
+
let errorMsg = response.statusText;
|
|
56
|
+
try {
|
|
57
|
+
const error = await response.json();
|
|
58
|
+
if (typeof error === 'object' && error !== null && 'error' in error) {
|
|
59
|
+
const errObj = error;
|
|
60
|
+
errorMsg = errObj.error?.message || errorMsg;
|
|
61
|
+
}
|
|
62
|
+
}
|
|
63
|
+
catch (_err) {
|
|
64
|
+
console.debug('Error parsing response:', _err);
|
|
65
|
+
}
|
|
66
|
+
throw new Error(`Azure API error: ${errorMsg}`);
|
|
67
|
+
}
|
|
68
|
+
const data = await response.json();
|
|
69
|
+
const apiData = data;
|
|
70
|
+
return {
|
|
71
|
+
id: apiData.id,
|
|
72
|
+
model: apiData.model,
|
|
73
|
+
choices: Array.isArray(apiData.choices)
|
|
74
|
+
? apiData.choices.map((choice) => ({
|
|
75
|
+
text: choice.text,
|
|
76
|
+
message: choice.message,
|
|
77
|
+
index: choice.index,
|
|
78
|
+
finishReason: choice.finish_reason
|
|
79
|
+
}))
|
|
80
|
+
: [],
|
|
81
|
+
usage: apiData.usage,
|
|
82
|
+
created: apiData.created,
|
|
83
|
+
metadata: {
|
|
84
|
+
object: apiData.object,
|
|
85
|
+
system_fingerprint: apiData.system_fingerprint
|
|
86
|
+
}
|
|
87
|
+
};
|
|
88
|
+
}
|
|
89
|
+
catch (error) {
|
|
90
|
+
if (error instanceof Error) {
|
|
91
|
+
throw error;
|
|
92
|
+
}
|
|
93
|
+
throw new Error('Failed to make Azure request');
|
|
94
|
+
}
|
|
95
|
+
}
|
|
96
|
+
parseUsage(usage) {
|
|
97
|
+
if (typeof usage === 'object' && usage !== null) {
|
|
98
|
+
const u = usage;
|
|
99
|
+
return {
|
|
100
|
+
prompt_tokens: u.prompt_tokens ?? 0,
|
|
101
|
+
completion_tokens: u.completion_tokens ?? 0,
|
|
102
|
+
total_tokens: u.total_tokens ?? 0
|
|
103
|
+
};
|
|
104
|
+
}
|
|
105
|
+
return { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 };
|
|
106
|
+
}
|
|
107
|
+
getPromptTokens(usage) {
|
|
108
|
+
return usage.prompt_tokens;
|
|
109
|
+
}
|
|
110
|
+
getCompletionTokens(usage) {
|
|
111
|
+
return usage.completion_tokens;
|
|
112
|
+
}
|
|
113
|
+
}
|
|
114
|
+
exports.AzureProvider = AzureProvider;
|
|
115
|
+
//# sourceMappingURL=azure.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"azure.js","sourceRoot":"","sources":["../../src/providers/azure.ts"],"names":[],"mappings":";;;AAAA,iCAAsC;AACtC,oCAAsD;AAEtD,iDAAuC;AAwBvC,MAAa,aAAc,SAAQ,mBAAY;IAK7C,YAAY,MAAsB;QAChC,KAAK,CAAC,EAAE,GAAG,MAAM,EAAE,QAAQ,EAAE,kBAAU,CAAC,KAAK,EAAE,CAAC,CAAC;QAEjD,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;YACzF,MAAM,IAAI,KAAK,CACb,sFAAsF,CACvF,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;QACpC,IAAI,CAAC,MAAM,GAAG,WAAW,MAAM,CAAC,YAAY,wCAAwC,MAAM,CAAC,YAAY,EAAE,CAAC;IAC5G,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,IAAY,EAAE,MAAc;QAC5C,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAA,sBAAM,EAAC,IAAI,CAAC,CAAC;YAC7B,OAAO,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACzC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,IAAI,CAAC,wCAAwC,EAAE,KAAK,CAAC,CAAC;YAC9D,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,OAAwB;QACxC,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,cAAc,CAAC;QACzE,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,MAAM,GAAG,QAAQ,gBAAgB,IAAI,CAAC,UAAU,EAAE,CAAC;QAEvE,MAAM,IAAI,GAAwB;YAChC,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,UAAU,EAAE,OAAO,CAAC,SAAS;YAC7B,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,KAAK,EAAE,OAAO,CAAC,IAAI;YACnB,IAAI,EAAE,OAAO,CAAC,aAAa;YAC3B,MAAM,EAAE,KAAK;YACb,IAAI,EAAE,OAAO,CAAC,MAAM;SACrB,CAAC;QAEF,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACrB,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC/B,CAAC;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBAChC,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,SAAS,EAAE,IAAI,CAAC,MAAM;iBACvB;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC3B,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,IAAI,QAAQ,GAAG,QAAQ,CAAC,UAAU,CAAC;gBACnC,IAAI,CAAC;oBACH,MAAM,KAAK,GAAY,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;oBAC7C,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,IAAI,KAAK,EAAE,CAAC;wBACpE,MAAM,MAAM,GAAG,KAA4B,CAAC;wBAC5C,QAAQ,GAAG,MAAM,CAAC,KAAK,EAAE,OAAO,IAAI,QAAQ,CAAC;oBAC/C,CAAC;gBACH,CAAC;gBAAC,OAAO,IAAI,EAAE,CAAC;oBAEd,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,IAAI,CAAC,CAAC;gBACjD,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,oBAAoB,QAAQ,EAAE,CAAC,CAAC;YAClD,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnC,MAAM,OAAO,GAAG,IAAqB,CAAC;YAEtC,OAAO;gBACL,EAAE,EAAE,OAAO,CAAC,EAAE;gBACd,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC;oBACrC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAW,EAAE,EAAE,CAAC,CAAC;wBACpC,IAAI,EAAE,MAAM,CAAC,IAAI;wBACjB,OAAO,EAAE,MAAM,CAAC,OAAO;wBACvB,KAAK,EAAE,MAAM,CAAC,KAAK;wBACnB,YAAY,EAAE,MAAM,CAAC,aAAa;qBACnC,CAAC,CAAC;oBACL,CAAC,CAAC,EAAE;gBACN,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,QAAQ,EAAE;oBACR,MAAM,EAAE,OAAO,CAAC,MAAM;oBACtB,kBAAkB,EAAE,OAAO,CAAC,kBAAkB;iBAC/C;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC3B,MAAM,KAAK,CAAC;YACd,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAClD,CAAC;IACH,CAAC;IAED,UAAU,CAAC,KAAc;QACvB,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;YAChD,MAAM,CAAC,GAAG,KAA6B,CAAC;YACxC,OAAO;gBACL,aAAa,EAAE,CAAC,CAAC,aAAa,IAAI,CAAC;gBACnC,iBAAiB,EAAE,CAAC,CAAC,iBAAiB,IAAI,CAAC;gBAC3C,YAAY,EAAE,CAAC,CAAC,YAAY,IAAI,CAAC;aAClC,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,aAAa,EAAE,CAAC,EAAE,iBAAiB,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC;IACrE,CAAC;IAES,eAAe,CAAC,KAAe;QACvC,OAAQ,KAAqB,CAAC,aAAa,CAAC;IAC9C,CAAC;IAES,mBAAmB,CAAC,KAAe;QAC3C,OAAQ,KAAqB,CAAC,iBAAiB,CAAC;IAClD,CAAC;CACF;AA1HD,sCA0HC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import { AIProvider, ProviderConfig, UsageMetadata, CostEstimate } from '../types';
|
|
2
|
+
import { ProviderRequest, ProviderResponse, ProviderModel, AnyUsage, Message } from '../types/providers';
|
|
3
|
+
export declare abstract class BaseProvider {
|
|
4
|
+
protected config: ProviderConfig;
|
|
5
|
+
protected provider: AIProvider;
|
|
6
|
+
constructor(config: ProviderConfig);
|
|
7
|
+
abstract countTokens(text: string, model: string): Promise<number>;
|
|
8
|
+
abstract makeRequest(request: ProviderRequest): Promise<ProviderResponse>;
|
|
9
|
+
abstract parseUsage(usage: any): AnyUsage;
|
|
10
|
+
estimateCost(prompt: string, model: string, expectedCompletion?: number): Promise<CostEstimate>;
|
|
11
|
+
trackUsage(request: ProviderRequest, response: ProviderResponse, startTime: number): UsageMetadata;
|
|
12
|
+
protected abstract getPromptTokens(usage: AnyUsage): number;
|
|
13
|
+
protected abstract getCompletionTokens(usage: AnyUsage): number;
|
|
14
|
+
protected messagesToPrompt(messages: Message[]): string;
|
|
15
|
+
getProvider(): AIProvider;
|
|
16
|
+
getConfig(): ProviderConfig;
|
|
17
|
+
validateModel(modelId: string): boolean;
|
|
18
|
+
getSupportedModels(): ProviderModel[];
|
|
19
|
+
}
|
|
20
|
+
//# sourceMappingURL=base.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../src/providers/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AACnF,OAAO,EACL,eAAe,EACf,gBAAgB,EAChB,aAAa,EACb,QAAQ,EACR,OAAO,EACR,MAAM,oBAAoB,CAAC;AAK5B,8BAAsB,YAAY;IAChC,SAAS,CAAC,MAAM,EAAE,cAAc,CAAC;IACjC,SAAS,CAAC,QAAQ,EAAE,UAAU,CAAC;gBAEnB,MAAM,EAAE,cAAc;IAKlC,QAAQ,CAAC,WAAW,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAClE,QAAQ,CAAC,WAAW,CAAC,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,gBAAgB,CAAC;IACzE,QAAQ,CAAC,UAAU,CAAC,KAAK,EAAE,GAAG,GAAG,QAAQ;IAEnC,YAAY,CAChB,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,MAAM,EACb,kBAAkB,CAAC,EAAE,MAAM,GAC1B,OAAO,CAAC,YAAY,CAAC;IAYxB,UAAU,CACR,OAAO,EAAE,eAAe,EACxB,QAAQ,EAAE,gBAAgB,EAC1B,SAAS,EAAE,MAAM,GAChB,aAAa;IAoChB,SAAS,CAAC,QAAQ,CAAC,eAAe,CAAC,KAAK,EAAE,QAAQ,GAAG,MAAM;IAC3D,SAAS,CAAC,QAAQ,CAAC,mBAAmB,CAAC,KAAK,EAAE,QAAQ,GAAG,MAAM;IAE/D,SAAS,CAAC,gBAAgB,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,MAAM;IAIvD,WAAW,IAAI,UAAU;IAIzB,SAAS,IAAI,cAAc;IAI3B,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAKvC,kBAAkB,IAAI,aAAa,EAAE;CAGtC"}
|
|
@@ -0,0 +1,67 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.BaseProvider = void 0;
|
|
4
|
+
const models_1 = require("../types/models");
|
|
5
|
+
const pricing_1 = require("../utils/pricing");
|
|
6
|
+
const uuid_1 = require("uuid");
|
|
7
|
+
class BaseProvider {
|
|
8
|
+
constructor(config) {
|
|
9
|
+
this.config = config;
|
|
10
|
+
this.provider = config.provider;
|
|
11
|
+
}
|
|
12
|
+
async estimateCost(prompt, model, expectedCompletion) {
|
|
13
|
+
const modelInfo = (0, models_1.getModelById)(model);
|
|
14
|
+
if (!modelInfo) {
|
|
15
|
+
throw new Error(`Unknown model: ${model}`);
|
|
16
|
+
}
|
|
17
|
+
const promptTokens = await this.countTokens(prompt, model);
|
|
18
|
+
const completionTokens = expectedCompletion || Math.floor(promptTokens * 0.5);
|
|
19
|
+
return (0, pricing_1.calculateCost)(promptTokens, completionTokens, modelInfo, this.config.customPricing);
|
|
20
|
+
}
|
|
21
|
+
trackUsage(request, response, startTime) {
|
|
22
|
+
const endTime = Date.now();
|
|
23
|
+
const usage = this.parseUsage(response.usage);
|
|
24
|
+
const model = (0, models_1.getModelById)(request.model);
|
|
25
|
+
if (!model) {
|
|
26
|
+
throw new Error(`Unknown model: ${request.model}`);
|
|
27
|
+
}
|
|
28
|
+
const promptTokens = this.getPromptTokens(usage);
|
|
29
|
+
const completionTokens = this.getCompletionTokens(usage);
|
|
30
|
+
const totalTokens = promptTokens + completionTokens;
|
|
31
|
+
const cost = (0, pricing_1.calculateCost)(promptTokens, completionTokens, model, this.config.customPricing);
|
|
32
|
+
const prompt = request.prompt || (request.messages ? this.messagesToPrompt(request.messages) : '');
|
|
33
|
+
const completion = response.choices[0]?.text || response.choices[0]?.message?.content || '';
|
|
34
|
+
return {
|
|
35
|
+
provider: this.provider,
|
|
36
|
+
model: request.model,
|
|
37
|
+
promptTokens,
|
|
38
|
+
completionTokens,
|
|
39
|
+
totalTokens,
|
|
40
|
+
estimatedCost: cost.totalCost,
|
|
41
|
+
prompt,
|
|
42
|
+
completion,
|
|
43
|
+
responseTime: endTime - startTime,
|
|
44
|
+
tags: request.metadata?.tags,
|
|
45
|
+
sessionId: request.metadata?.sessionId || (0, uuid_1.v4)(),
|
|
46
|
+
projectId: request.metadata?.projectId
|
|
47
|
+
};
|
|
48
|
+
}
|
|
49
|
+
messagesToPrompt(messages) {
|
|
50
|
+
return messages.map(msg => `${msg.role}: ${msg.content}`).join('\n\n');
|
|
51
|
+
}
|
|
52
|
+
getProvider() {
|
|
53
|
+
return this.provider;
|
|
54
|
+
}
|
|
55
|
+
getConfig() {
|
|
56
|
+
return this.config;
|
|
57
|
+
}
|
|
58
|
+
validateModel(modelId) {
|
|
59
|
+
const model = (0, models_1.getModelById)(modelId);
|
|
60
|
+
return model !== undefined && model.provider === this.provider;
|
|
61
|
+
}
|
|
62
|
+
getSupportedModels() {
|
|
63
|
+
return Object.values(models_1.MODELS).filter(model => model && model.provider === this.provider);
|
|
64
|
+
}
|
|
65
|
+
}
|
|
66
|
+
exports.BaseProvider = BaseProvider;
|
|
67
|
+
//# sourceMappingURL=base.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/providers/base.ts"],"names":[],"mappings":";;;AAQA,4CAAuD;AACvD,8CAAiD;AACjD,+BAAoC;AAEpC,MAAsB,YAAY;IAIhC,YAAY,MAAsB;QAChC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;IAClC,CAAC;IAMD,KAAK,CAAC,YAAY,CAChB,MAAc,EACd,KAAa,EACb,kBAA2B;QAE3B,MAAM,SAAS,GAAG,IAAA,qBAAY,EAAC,KAAK,CAAC,CAAC;QACtC,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,kBAAkB,KAAK,EAAE,CAAC,CAAC;QAC7C,CAAC;QAED,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAC3D,MAAM,gBAAgB,GAAG,kBAAkB,IAAI,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,GAAG,CAAC,CAAC;QAE9E,OAAO,IAAA,uBAAa,EAAC,YAAY,EAAE,gBAAgB,EAAE,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;IAC7F,CAAC;IAED,UAAU,CACR,OAAwB,EACxB,QAA0B,EAC1B,SAAiB;QAEjB,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC3B,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAC9C,MAAM,KAAK,GAAG,IAAA,qBAAY,EAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAE1C,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,kBAAkB,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;QACrD,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QACjD,MAAM,gBAAgB,GAAG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;QACzD,MAAM,WAAW,GAAG,YAAY,GAAG,gBAAgB,CAAC;QAEpD,MAAM,IAAI,GAAG,IAAA,uBAAa,EAAC,YAAY,EAAE,gBAAgB,EAAE,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QAE7F,MAAM,MAAM,GACV,OAAO,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAEtF,MAAM,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;QAE5F,OAAO;YACL,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,YAAY;YACZ,gBAAgB;YAChB,WAAW;YACX,aAAa,EAAE,IAAI,CAAC,SAAS;YAC7B,MAAM;YACN,UAAU;YACV,YAAY,EAAE,OAAO,GAAG,SAAS;YACjC,IAAI,EAAE,OAAO,CAAC,QAAQ,EAAE,IAAI;YAC5B,SAAS,EAAE,OAAO,CAAC,QAAQ,EAAE,SAAS,IAAI,IAAA,SAAM,GAAE;YAClD,SAAS,EAAE,OAAO,CAAC,QAAQ,EAAE,SAAS;SACvC,CAAC;IACJ,CAAC;IAKS,gBAAgB,CAAC,QAAmB;QAC5C,OAAO,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,IAAI,KAAK,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACzE,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,aAAa,CAAC,OAAe;QAC3B,MAAM,KAAK,GAAG,IAAA,qBAAY,EAAC,OAAO,CAAC,CAAC;QACpC,OAAO,KAAK,KAAK,SAAS,IAAI,KAAK,CAAC,QAAQ,KAAK,IAAI,CAAC,QAAQ,CAAC;IACjE,CAAC;IAED,kBAAkB;QAChB,OAAO,MAAM,CAAC,MAAM,CAAC,eAAM,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,IAAI,KAAK,CAAC,QAAQ,KAAK,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1F,CAAC;CACF;AA5FD,oCA4FC"}
|
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
import { BaseProvider } from './base';
|
|
2
|
+
import { ProviderConfig } from '../types';
|
|
3
|
+
import { ProviderRequest, ProviderResponse, BedrockUsage, AnyUsage } from '../types/providers';
|
|
4
|
+
export declare class BedrockProvider extends BaseProvider {
|
|
5
|
+
private client;
|
|
6
|
+
private region;
|
|
7
|
+
constructor(config: ProviderConfig);
|
|
8
|
+
private getCredentials;
|
|
9
|
+
countTokens(text: string, model: string): Promise<number>;
|
|
10
|
+
makeRequest(request: ProviderRequest): Promise<ProviderResponse>;
|
|
11
|
+
private formatRequestBody;
|
|
12
|
+
private formatClaudeMessages;
|
|
13
|
+
private formatResponse;
|
|
14
|
+
parseUsage(usage: any): BedrockUsage;
|
|
15
|
+
protected getPromptTokens(usage: AnyUsage): number;
|
|
16
|
+
protected getCompletionTokens(usage: AnyUsage): number;
|
|
17
|
+
listFoundationModels(): Promise<string[]>;
|
|
18
|
+
invokeModelWithOptimization(request: ProviderRequest, optimizationModel?: string): Promise<{
|
|
19
|
+
response: ProviderResponse;
|
|
20
|
+
optimization?: any;
|
|
21
|
+
}>;
|
|
22
|
+
}
|
|
23
|
+
//# sourceMappingURL=bedrock.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"bedrock.d.ts","sourceRoot":"","sources":["../../src/providers/bedrock.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,cAAc,EAAc,MAAM,UAAU,CAAC;AACtD,OAAO,EAAE,eAAe,EAAE,gBAAgB,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAQ/F,qBAAa,eAAgB,SAAQ,YAAY;IAC/C,OAAO,CAAC,MAAM,CAAuB;IACrC,OAAO,CAAC,MAAM,CAAS;gBAEX,MAAM,EAAE,cAAc;IAalC,OAAO,CAAC,cAAc;IAoBhB,WAAW,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAezD,WAAW,CAAC,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,gBAAgB,CAAC;IA6BtE,OAAO,CAAC,iBAAiB;IAyCzB,OAAO,CAAC,oBAAoB;IAe5B,OAAO,CAAC,cAAc;IAsEtB,UAAU,CAAC,KAAK,EAAE,GAAG,GAAG,YAAY;IAWpC,SAAS,CAAC,eAAe,CAAC,KAAK,EAAE,QAAQ,GAAG,MAAM;IAIlD,SAAS,CAAC,mBAAmB,CAAC,KAAK,EAAE,QAAQ,GAAG,MAAM;IAKtD,oBAAoB,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAYnC,2BAA2B,CAC/B,OAAO,EAAE,eAAe,EACxB,iBAAiB,GAAE,MAAkD,GACpE,OAAO,CAAC;QAAE,QAAQ,EAAE,gBAAgB,CAAC;QAAC,YAAY,CAAC,EAAE,GAAG,CAAA;KAAE,CAAC;CA2C/D"}
|
|
@@ -0,0 +1,247 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.BedrockProvider = void 0;
|
|
4
|
+
const base_1 = require("./base");
|
|
5
|
+
const types_1 = require("../types");
|
|
6
|
+
const client_bedrock_runtime_1 = require("@aws-sdk/client-bedrock-runtime");
|
|
7
|
+
const credential_providers_1 = require("@aws-sdk/credential-providers");
|
|
8
|
+
class BedrockProvider extends base_1.BaseProvider {
|
|
9
|
+
constructor(config) {
|
|
10
|
+
super({ ...config, provider: types_1.AIProvider.AWSBedrock });
|
|
11
|
+
this.region = config.region || 'us-east-1';
|
|
12
|
+
const credentials = this.getCredentials(config);
|
|
13
|
+
this.client = new client_bedrock_runtime_1.BedrockRuntimeClient({
|
|
14
|
+
region: this.region,
|
|
15
|
+
credentials
|
|
16
|
+
});
|
|
17
|
+
}
|
|
18
|
+
getCredentials(config) {
|
|
19
|
+
const bedrockConfig = config.optimization?.bedrockConfig;
|
|
20
|
+
if (bedrockConfig && bedrockConfig.credentials) {
|
|
21
|
+
return {
|
|
22
|
+
accessKeyId: bedrockConfig.credentials.accessKeyId,
|
|
23
|
+
secretAccessKey: bedrockConfig.credentials.secretAccessKey,
|
|
24
|
+
sessionToken: bedrockConfig.credentials.sessionToken
|
|
25
|
+
};
|
|
26
|
+
}
|
|
27
|
+
try {
|
|
28
|
+
return (0, credential_providers_1.fromEnv)();
|
|
29
|
+
}
|
|
30
|
+
catch {
|
|
31
|
+
return (0, credential_providers_1.fromIni)({ profile: 'default' });
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
async countTokens(text, model) {
|
|
35
|
+
if (model.includes('claude')) {
|
|
36
|
+
return Promise.resolve(Math.ceil(text.length / 3.5));
|
|
37
|
+
}
|
|
38
|
+
else if (model.includes('titan')) {
|
|
39
|
+
return Promise.resolve(Math.ceil(text.length / 4));
|
|
40
|
+
}
|
|
41
|
+
else {
|
|
42
|
+
return Promise.resolve(Math.ceil(text.length / 4));
|
|
43
|
+
}
|
|
44
|
+
}
|
|
45
|
+
async makeRequest(request) {
|
|
46
|
+
const body = this.formatRequestBody(request);
|
|
47
|
+
const input = {
|
|
48
|
+
modelId: request.model,
|
|
49
|
+
contentType: 'application/json',
|
|
50
|
+
accept: 'application/json',
|
|
51
|
+
body: new TextEncoder().encode(JSON.stringify(body))
|
|
52
|
+
};
|
|
53
|
+
try {
|
|
54
|
+
const command = new client_bedrock_runtime_1.InvokeModelCommand(input);
|
|
55
|
+
const response = await this.client.send(command);
|
|
56
|
+
if (!response.body) {
|
|
57
|
+
throw new Error('Empty response from Bedrock');
|
|
58
|
+
}
|
|
59
|
+
const responseBody = JSON.parse(new TextDecoder().decode(response.body));
|
|
60
|
+
return this.formatResponse(responseBody, request);
|
|
61
|
+
}
|
|
62
|
+
catch (error) {
|
|
63
|
+
if (error instanceof Error) {
|
|
64
|
+
throw new Error(`Bedrock API error: ${error.message}`);
|
|
65
|
+
}
|
|
66
|
+
throw new Error('Failed to make Bedrock request');
|
|
67
|
+
}
|
|
68
|
+
}
|
|
69
|
+
formatRequestBody(request) {
|
|
70
|
+
const { model, prompt, messages, maxTokens, temperature, topP, stopSequences } = request;
|
|
71
|
+
if (model.includes('claude')) {
|
|
72
|
+
const formattedPrompt = messages
|
|
73
|
+
? this.formatClaudeMessages(messages)
|
|
74
|
+
: `\n\nHuman: ${prompt}\n\nAssistant:`;
|
|
75
|
+
return {
|
|
76
|
+
prompt: formattedPrompt,
|
|
77
|
+
max_tokens_to_sample: maxTokens || 1000,
|
|
78
|
+
temperature: temperature || 0.7,
|
|
79
|
+
top_p: topP || 1,
|
|
80
|
+
stop_sequences: stopSequences || ['\n\nHuman:'],
|
|
81
|
+
anthropic_version: 'bedrock-2023-05-31'
|
|
82
|
+
};
|
|
83
|
+
}
|
|
84
|
+
else if (model.includes('titan')) {
|
|
85
|
+
return {
|
|
86
|
+
inputText: prompt || this.messagesToPrompt(messages || []),
|
|
87
|
+
textGenerationConfig: {
|
|
88
|
+
maxTokenCount: maxTokens || 1000,
|
|
89
|
+
temperature: temperature || 0.7,
|
|
90
|
+
topP: topP || 1,
|
|
91
|
+
stopSequences: stopSequences
|
|
92
|
+
}
|
|
93
|
+
};
|
|
94
|
+
}
|
|
95
|
+
else {
|
|
96
|
+
return {
|
|
97
|
+
prompt: prompt || this.messagesToPrompt(messages || []),
|
|
98
|
+
max_tokens: maxTokens || 1000,
|
|
99
|
+
temperature: temperature || 0.7,
|
|
100
|
+
top_p: topP || 1,
|
|
101
|
+
stop: stopSequences
|
|
102
|
+
};
|
|
103
|
+
}
|
|
104
|
+
}
|
|
105
|
+
formatClaudeMessages(messages) {
|
|
106
|
+
let prompt = '';
|
|
107
|
+
for (const message of messages) {
|
|
108
|
+
if (message.role === 'system') {
|
|
109
|
+
prompt += `${message.content}\n\n`;
|
|
110
|
+
}
|
|
111
|
+
else if (message.role === 'user') {
|
|
112
|
+
prompt += `Human: ${message.content}\n\n`;
|
|
113
|
+
}
|
|
114
|
+
else if (message.role === 'assistant') {
|
|
115
|
+
prompt += `Assistant: ${message.content}\n\n`;
|
|
116
|
+
}
|
|
117
|
+
}
|
|
118
|
+
prompt += 'Assistant:';
|
|
119
|
+
return prompt;
|
|
120
|
+
}
|
|
121
|
+
formatResponse(response, request) {
|
|
122
|
+
const { model } = request;
|
|
123
|
+
if (model.includes('claude')) {
|
|
124
|
+
const claudeResponse = response;
|
|
125
|
+
return {
|
|
126
|
+
id: claudeResponse.id ?? `bedrock-${Date.now()}`,
|
|
127
|
+
model,
|
|
128
|
+
choices: [
|
|
129
|
+
{
|
|
130
|
+
text: claudeResponse.completion ?? '',
|
|
131
|
+
index: 0,
|
|
132
|
+
finishReason: claudeResponse.stop_reason ?? 'stop'
|
|
133
|
+
}
|
|
134
|
+
],
|
|
135
|
+
usage: {
|
|
136
|
+
inputTokens: claudeResponse.usage?.input_tokens ?? 0,
|
|
137
|
+
outputTokens: claudeResponse.usage?.output_tokens ?? 0,
|
|
138
|
+
totalTokens: (claudeResponse.usage?.input_tokens ?? 0) + (claudeResponse.usage?.output_tokens ?? 0)
|
|
139
|
+
},
|
|
140
|
+
metadata: claudeResponse.metrics
|
|
141
|
+
};
|
|
142
|
+
}
|
|
143
|
+
else if (model.includes('titan')) {
|
|
144
|
+
const result = Array.isArray(response.results) && response.results.length > 0
|
|
145
|
+
? response.results[0]
|
|
146
|
+
: response;
|
|
147
|
+
return {
|
|
148
|
+
id: `bedrock-${Date.now()}`,
|
|
149
|
+
model,
|
|
150
|
+
choices: [
|
|
151
|
+
{
|
|
152
|
+
text: result.outputText ?? result.text ?? '',
|
|
153
|
+
index: 0,
|
|
154
|
+
finishReason: result.completionReason ?? 'stop'
|
|
155
|
+
}
|
|
156
|
+
],
|
|
157
|
+
usage: {
|
|
158
|
+
inputTokens: result.tokenCount ?? 0,
|
|
159
|
+
outputTokens: result.tokenCount ?? 0,
|
|
160
|
+
totalTokens: (result.tokenCount ?? 0) * 2
|
|
161
|
+
}
|
|
162
|
+
};
|
|
163
|
+
}
|
|
164
|
+
else {
|
|
165
|
+
const defaultResponse = response;
|
|
166
|
+
return {
|
|
167
|
+
id: defaultResponse.id ?? `bedrock-${Date.now()}`,
|
|
168
|
+
model,
|
|
169
|
+
choices: [
|
|
170
|
+
{
|
|
171
|
+
text: defaultResponse.text ?? defaultResponse.completion ?? defaultResponse.output ?? '',
|
|
172
|
+
index: 0,
|
|
173
|
+
finishReason: 'stop'
|
|
174
|
+
}
|
|
175
|
+
],
|
|
176
|
+
usage: {
|
|
177
|
+
inputTokens: 0,
|
|
178
|
+
outputTokens: 0,
|
|
179
|
+
totalTokens: 0
|
|
180
|
+
}
|
|
181
|
+
};
|
|
182
|
+
}
|
|
183
|
+
}
|
|
184
|
+
parseUsage(usage) {
|
|
185
|
+
const usageAsAny = usage;
|
|
186
|
+
return {
|
|
187
|
+
inputTokens: usageAsAny.inputTokens ?? usageAsAny.input_tokens ?? 0,
|
|
188
|
+
outputTokens: usageAsAny.outputTokens ?? usageAsAny.output_tokens ?? 0,
|
|
189
|
+
totalTokens: usageAsAny.totalTokens ?? usageAsAny.total_tokens ?? 0,
|
|
190
|
+
invocationLatency: usageAsAny.invocationLatency,
|
|
191
|
+
firstByteLatency: usageAsAny.firstByteLatency
|
|
192
|
+
};
|
|
193
|
+
}
|
|
194
|
+
getPromptTokens(usage) {
|
|
195
|
+
return usage.inputTokens;
|
|
196
|
+
}
|
|
197
|
+
getCompletionTokens(usage) {
|
|
198
|
+
return usage.outputTokens;
|
|
199
|
+
}
|
|
200
|
+
listFoundationModels() {
|
|
201
|
+
return Promise.resolve([
|
|
202
|
+
'anthropic.claude-sonnet-4-20250514-v1:0',
|
|
203
|
+
'anthropic.claude-3-opus-20240229-v1:0',
|
|
204
|
+
'anthropic.claude-3-haiku-20240307-v1:0',
|
|
205
|
+
'amazon.titan-text-express-v1',
|
|
206
|
+
'amazon.titan-text-lite-v1'
|
|
207
|
+
]);
|
|
208
|
+
}
|
|
209
|
+
async invokeModelWithOptimization(request, optimizationModel = 'anthropic.claude-sonnet-4-20250514-v1:0') {
|
|
210
|
+
const optimizationPrompt = `Analyze this prompt and suggest optimizations to reduce tokens while maintaining effectiveness:
|
|
211
|
+
|
|
212
|
+
Original prompt: "${request.prompt || this.messagesToPrompt(request.messages || [])}"
|
|
213
|
+
|
|
214
|
+
Provide:
|
|
215
|
+
1. Optimized version of the prompt
|
|
216
|
+
2. Estimated token reduction percentage
|
|
217
|
+
3. Key improvements made
|
|
218
|
+
|
|
219
|
+
Format as JSON.`;
|
|
220
|
+
const optimizationRequest = {
|
|
221
|
+
model: optimizationModel,
|
|
222
|
+
prompt: optimizationPrompt,
|
|
223
|
+
maxTokens: 500,
|
|
224
|
+
temperature: 0.3
|
|
225
|
+
};
|
|
226
|
+
try {
|
|
227
|
+
const optimizationResponse = await this.makeRequest(optimizationRequest);
|
|
228
|
+
const optimizationText = optimizationResponse.choices[0]?.text || '';
|
|
229
|
+
let optimization;
|
|
230
|
+
try {
|
|
231
|
+
optimization = JSON.parse(optimizationText);
|
|
232
|
+
}
|
|
233
|
+
catch {
|
|
234
|
+
optimization = { suggestions: optimizationText };
|
|
235
|
+
}
|
|
236
|
+
const response = await this.makeRequest(request);
|
|
237
|
+
return { response, optimization };
|
|
238
|
+
}
|
|
239
|
+
catch (error) {
|
|
240
|
+
console.warn('Optimization failed, using regular request:', error);
|
|
241
|
+
const response = await this.makeRequest(request);
|
|
242
|
+
return { response };
|
|
243
|
+
}
|
|
244
|
+
}
|
|
245
|
+
}
|
|
246
|
+
exports.BedrockProvider = BedrockProvider;
|
|
247
|
+
//# sourceMappingURL=bedrock.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"bedrock.js","sourceRoot":"","sources":["../../src/providers/bedrock.ts"],"names":[],"mappings":";;;AAAA,iCAAsC;AACtC,oCAAsD;AAEtD,4EAIyC;AACzC,wEAAiE;AAEjE,MAAa,eAAgB,SAAQ,mBAAY;IAI/C,YAAY,MAAsB;QAChC,KAAK,CAAC,EAAE,GAAG,MAAM,EAAE,QAAQ,EAAE,kBAAU,CAAC,UAAU,EAAE,CAAC,CAAC;QAEtD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,WAAW,CAAC;QAG3C,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAChD,IAAI,CAAC,MAAM,GAAG,IAAI,6CAAoB,CAAC;YACrC,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,WAAW;SACZ,CAAC,CAAC;IACL,CAAC;IAEO,cAAc,CAAC,MAAsB;QAC3C,MAAM,aAAa,GAAG,MAAM,CAAC,YAAY,EAAE,aAAa,CAAC;QAEzD,IAAI,aAAa,IAAI,aAAa,CAAC,WAAW,EAAE,CAAC;YAE/C,OAAO;gBACL,WAAW,EAAE,aAAa,CAAC,WAAW,CAAC,WAAW;gBAClD,eAAe,EAAE,aAAa,CAAC,WAAW,CAAC,eAAe;gBAC1D,YAAY,EAAE,aAAa,CAAC,WAAW,CAAC,YAAY;aACrD,CAAC;QACJ,CAAC;QAGD,IAAI,CAAC;YACH,OAAO,IAAA,8BAAO,GAAE,CAAC;QACnB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAA,8BAAO,EAAC,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,IAAY,EAAE,KAAa;QAG3C,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YAE7B,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC;QACvD,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAEnC,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;QACrD,CAAC;aAAM,CAAC;YAEN,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,OAAwB;QACxC,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAE7C,MAAM,KAAK,GAA4B;YACrC,OAAO,EAAE,OAAO,CAAC,KAAK;YACtB,WAAW,EAAE,kBAAkB;YAC/B,MAAM,EAAE,kBAAkB;YAC1B,IAAI,EAAE,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;SACrD,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,2CAAkB,CAAC,KAAK,CAAC,CAAC;YAC9C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAEjD,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACnB,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;YACjD,CAAC;YAED,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;YAEzE,OAAO,IAAI,CAAC,cAAc,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;QACpD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC3B,MAAM,IAAI,KAAK,CAAC,sBAAsB,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YACzD,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;IAEO,iBAAiB,CAAC,OAAwB;QAChD,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,WAAW,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,OAAO,CAAC;QAGzF,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YAE7B,MAAM,eAAe,GAAG,QAAQ;gBAC9B,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,QAAQ,CAAC;gBACrC,CAAC,CAAC,cAAc,MAAM,gBAAgB,CAAC;YAEzC,OAAO;gBACL,MAAM,EAAE,eAAe;gBACvB,oBAAoB,EAAE,SAAS,IAAI,IAAI;gBACvC,WAAW,EAAE,WAAW,IAAI,GAAG;gBAC/B,KAAK,EAAE,IAAI,IAAI,CAAC;gBAChB,cAAc,EAAE,aAAa,IAAI,CAAC,YAAY,CAAC;gBAC/C,iBAAiB,EAAE,oBAAoB;aACxC,CAAC;QACJ,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAEnC,OAAO;gBACL,SAAS,EAAE,MAAM,IAAI,IAAI,CAAC,gBAAgB,CAAC,QAAQ,IAAI,EAAE,CAAC;gBAC1D,oBAAoB,EAAE;oBACpB,aAAa,EAAE,SAAS,IAAI,IAAI;oBAChC,WAAW,EAAE,WAAW,IAAI,GAAG;oBAC/B,IAAI,EAAE,IAAI,IAAI,CAAC;oBACf,aAAa,EAAE,aAAa;iBAC7B;aACF,CAAC;QACJ,CAAC;aAAM,CAAC;YAEN,OAAO;gBACL,MAAM,EAAE,MAAM,IAAI,IAAI,CAAC,gBAAgB,CAAC,QAAQ,IAAI,EAAE,CAAC;gBACvD,UAAU,EAAE,SAAS,IAAI,IAAI;gBAC7B,WAAW,EAAE,WAAW,IAAI,GAAG;gBAC/B,KAAK,EAAE,IAAI,IAAI,CAAC;gBAChB,IAAI,EAAE,aAAa;aACpB,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,oBAAoB,CAAC,QAAkD;QAC7E,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC9B,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,MAAM,CAAC;YACrC,CAAC;iBAAM,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACnC,MAAM,IAAI,UAAU,OAAO,CAAC,OAAO,MAAM,CAAC;YAC5C,CAAC;iBAAM,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBACxC,MAAM,IAAI,cAAc,OAAO,CAAC,OAAO,MAAM,CAAC;YAChD,CAAC;QACH,CAAC;QACD,MAAM,IAAI,YAAY,CAAC;QACvB,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,cAAc,CACpB,QAAiC,EACjC,OAAwB;QAExB,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC;QAE1B,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC7B,MAAM,cAAc,GAAG,QAA0B,CAAC;YAClD,OAAO;gBACL,EAAE,EAAE,cAAc,CAAC,EAAE,IAAI,WAAW,IAAI,CAAC,GAAG,EAAE,EAAE;gBAChD,KAAK;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,cAAc,CAAC,UAAU,IAAI,EAAE;wBACrC,KAAK,EAAE,CAAC;wBACR,YAAY,EAAE,cAAc,CAAC,WAAW,IAAI,MAAM;qBACnD;iBACF;gBACD,KAAK,EAAE;oBACL,WAAW,EAAE,cAAc,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC;oBACpD,YAAY,EAAE,cAAc,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;oBACtD,WAAW,EACT,CAAC,cAAc,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC,CAAC;iBACzF;gBACD,QAAQ,EAAE,cAAc,CAAC,OAAO;aACjC,CAAC;QACJ,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YACnC,MAAM,MAAM,GACV,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,QAAQ,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC;gBAC5D,CAAC,CAAE,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAiB;gBACtC,CAAC,CAAE,QAAwB,CAAC;YAChC,OAAO;gBACL,EAAE,EAAE,WAAW,IAAI,CAAC,GAAG,EAAE,EAAE;gBAC3B,KAAK;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,IAAI,IAAI,EAAE;wBAC5C,KAAK,EAAE,CAAC;wBACR,YAAY,EAAE,MAAM,CAAC,gBAAgB,IAAI,MAAM;qBAChD;iBACF;gBACD,KAAK,EAAE;oBACL,WAAW,EAAE,MAAM,CAAC,UAAU,IAAI,CAAC;oBACnC,YAAY,EAAE,MAAM,CAAC,UAAU,IAAI,CAAC;oBACpC,WAAW,EAAE,CAAC,MAAM,CAAC,UAAU,IAAI,CAAC,CAAC,GAAG,CAAC;iBAC1C;aACF,CAAC;QACJ,CAAC;aAAM,CAAC;YAEN,MAAM,eAAe,GAAG,QAA2B,CAAC;YACpD,OAAO;gBACL,EAAE,EAAE,eAAe,CAAC,EAAE,IAAI,WAAW,IAAI,CAAC,GAAG,EAAE,EAAE;gBACjD,KAAK;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EACF,eAAe,CAAC,IAAI,IAAI,eAAe,CAAC,UAAU,IAAI,eAAe,CAAC,MAAM,IAAI,EAAE;wBACpF,KAAK,EAAE,CAAC;wBACR,YAAY,EAAE,MAAM;qBACrB;iBACF;gBACD,KAAK,EAAE;oBACL,WAAW,EAAE,CAAC;oBACd,YAAY,EAAE,CAAC;oBACf,WAAW,EAAE,CAAC;iBACf;aACF,CAAC;QACJ,CAAC;IACH,CAAC;IAED,UAAU,CAAC,KAAU;QACnB,MAAM,UAAU,GAAG,KAAK,CAAC;QACzB,OAAO;YACL,WAAW,EAAE,UAAU,CAAC,WAAW,IAAI,UAAU,CAAC,YAAY,IAAI,CAAC;YACnE,YAAY,EAAE,UAAU,CAAC,YAAY,IAAI,UAAU,CAAC,aAAa,IAAI,CAAC;YACtE,WAAW,EAAE,UAAU,CAAC,WAAW,IAAI,UAAU,CAAC,YAAY,IAAI,CAAC;YACnE,iBAAiB,EAAE,UAAU,CAAC,iBAAiB;YAC/C,gBAAgB,EAAE,UAAU,CAAC,gBAAgB;SAC9C,CAAC;IACJ,CAAC;IAES,eAAe,CAAC,KAAe;QACvC,OAAQ,KAAsB,CAAC,WAAW,CAAC;IAC7C,CAAC;IAES,mBAAmB,CAAC,KAAe;QAC3C,OAAQ,KAAsB,CAAC,YAAY,CAAC;IAC9C,CAAC;IAGD,oBAAoB;QAGlB,OAAO,OAAO,CAAC,OAAO,CAAC;YACrB,yCAAyC;YACzC,uCAAuC;YACvC,wCAAwC;YACxC,8BAA8B;YAC9B,2BAA2B;SAC5B,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,2BAA2B,CAC/B,OAAwB,EACxB,oBAA4B,yCAAyC;QAGrE,MAAM,kBAAkB,GAAG;;oBAEX,OAAO,CAAC,MAAM,IAAI,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC;;;;;;;gBAOnE,CAAC;QAEb,MAAM,mBAAmB,GAAoB;YAC3C,KAAK,EAAE,iBAAiB;YACxB,MAAM,EAAE,kBAAkB;YAC1B,SAAS,EAAE,GAAG;YACd,WAAW,EAAE,GAAG;SACjB,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,mBAAmB,CAAC,CAAC;YACzE,MAAM,gBAAgB,GAAG,oBAAoB,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;YAGrE,IAAI,YAAiB,CAAC;YACtB,IAAI,CAAC;gBACH,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;YAC9C,CAAC;YAAC,MAAM,CAAC;gBACP,YAAY,GAAG,EAAE,WAAW,EAAE,gBAAgB,EAAE,CAAC;YACnD,CAAC;YAGD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YAEjD,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC;QACpC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAEf,OAAO,CAAC,IAAI,CAAC,6CAA6C,EAAE,KAAK,CAAC,CAAC;YACnE,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YACjD,OAAO,EAAE,QAAQ,EAAE,CAAC;QACtB,CAAC;IACH,CAAC;CACF;AA7RD,0CA6RC"}
|