@a5c-ai/transport-adapter 5.1.1-staging.00ceebd28cf2
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/README.md +54 -0
- package/dist/__tests__/codecs.test.d.ts +1 -0
- package/dist/__tests__/codecs.test.d.ts.map +1 -0
- package/dist/__tests__/codecs.test.js +698 -0
- package/dist/__tests__/codecs.test.js.map +1 -0
- package/dist/__tests__/openai-engine.test.d.ts +1 -0
- package/dist/__tests__/openai-engine.test.d.ts.map +1 -0
- package/dist/__tests__/openai-engine.test.js +89 -0
- package/dist/__tests__/openai-engine.test.js.map +1 -0
- package/dist/bin/adapters-proxy.d.ts +2 -0
- package/dist/bin/adapters-proxy.d.ts.map +1 -0
- package/dist/bin/adapters-proxy.js +5 -0
- package/dist/bin/adapters-proxy.js.map +1 -0
- package/dist/bin/adapters-transport-proxy.d.ts +2 -0
- package/dist/bin/adapters-transport-proxy.d.ts.map +1 -0
- package/dist/bin/adapters-transport-proxy.js +57 -0
- package/dist/bin/adapters-transport-proxy.js.map +1 -0
- package/dist/codec.d.ts +36 -0
- package/dist/codec.d.ts.map +1 -0
- package/dist/codec.js +2 -0
- package/dist/codec.js.map +1 -0
- package/dist/codecs/anthropic.d.ts +12 -0
- package/dist/codecs/anthropic.d.ts.map +1 -0
- package/dist/codecs/anthropic.js +185 -0
- package/dist/codecs/anthropic.js.map +1 -0
- package/dist/codecs/bedrock.d.ts +12 -0
- package/dist/codecs/bedrock.d.ts.map +1 -0
- package/dist/codecs/bedrock.js +175 -0
- package/dist/codecs/bedrock.js.map +1 -0
- package/dist/codecs/google.d.ts +12 -0
- package/dist/codecs/google.d.ts.map +1 -0
- package/dist/codecs/google.js +176 -0
- package/dist/codecs/google.js.map +1 -0
- package/dist/codecs/index.d.ts +30 -0
- package/dist/codecs/index.d.ts.map +1 -0
- package/dist/codecs/index.js +115 -0
- package/dist/codecs/index.js.map +1 -0
- package/dist/codecs/openai-chat.d.ts +12 -0
- package/dist/codecs/openai-chat.d.ts.map +1 -0
- package/dist/codecs/openai-chat.js +191 -0
- package/dist/codecs/openai-chat.js.map +1 -0
- package/dist/codecs/openai-responses.d.ts +12 -0
- package/dist/codecs/openai-responses.d.ts.map +1 -0
- package/dist/codecs/openai-responses.js +229 -0
- package/dist/codecs/openai-responses.js.map +1 -0
- package/dist/config.d.ts +8 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +86 -0
- package/dist/config.js.map +1 -0
- package/dist/engines/anthropic.d.ts +7 -0
- package/dist/engines/anthropic.d.ts.map +1 -0
- package/dist/engines/anthropic.js +232 -0
- package/dist/engines/anthropic.js.map +1 -0
- package/dist/engines/google.d.ts +17 -0
- package/dist/engines/google.d.ts.map +1 -0
- package/dist/engines/google.js +232 -0
- package/dist/engines/google.js.map +1 -0
- package/dist/engines/index.d.ts +3 -0
- package/dist/engines/index.d.ts.map +1 -0
- package/dist/engines/index.js +3 -0
- package/dist/engines/index.js.map +1 -0
- package/dist/engines/openai.d.ts +12 -0
- package/dist/engines/openai.d.ts.map +1 -0
- package/dist/engines/openai.js +253 -0
- package/dist/engines/openai.js.map +1 -0
- package/dist/index.d.ts +22 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +19 -0
- package/dist/index.js.map +1 -0
- package/dist/runtime.d.ts +19 -0
- package/dist/runtime.d.ts.map +1 -0
- package/dist/runtime.js +56 -0
- package/dist/runtime.js.map +1 -0
- package/dist/server.d.ts +4 -0
- package/dist/server.d.ts.map +1 -0
- package/dist/server.js +1428 -0
- package/dist/server.js.map +1 -0
- package/dist/types.d.ts +109 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +11 -0
- package/dist/types.js.map +1 -0
- package/package.json +99 -0
|
@@ -0,0 +1,175 @@
|
|
|
1
|
+
function textFromContent(content) {
|
|
2
|
+
if (typeof content === 'string') {
|
|
3
|
+
return content;
|
|
4
|
+
}
|
|
5
|
+
if (!Array.isArray(content)) {
|
|
6
|
+
return '';
|
|
7
|
+
}
|
|
8
|
+
return content
|
|
9
|
+
.map((part) => {
|
|
10
|
+
if (typeof part === 'string') {
|
|
11
|
+
return part;
|
|
12
|
+
}
|
|
13
|
+
if (part
|
|
14
|
+
&& typeof part === 'object'
|
|
15
|
+
&& 'text' in part
|
|
16
|
+
&& typeof part.text === 'string') {
|
|
17
|
+
return part.text;
|
|
18
|
+
}
|
|
19
|
+
return '';
|
|
20
|
+
})
|
|
21
|
+
.filter(Boolean)
|
|
22
|
+
.join(' ');
|
|
23
|
+
}
|
|
24
|
+
function normalizeMessages(raw) {
|
|
25
|
+
if (!Array.isArray(raw)) {
|
|
26
|
+
return [];
|
|
27
|
+
}
|
|
28
|
+
return raw.map((entry) => {
|
|
29
|
+
const record = entry;
|
|
30
|
+
return {
|
|
31
|
+
role: typeof record.role === 'string' ? record.role : 'user',
|
|
32
|
+
content: textFromContent(record.content),
|
|
33
|
+
};
|
|
34
|
+
});
|
|
35
|
+
}
|
|
36
|
+
function toolEntries(raw) {
|
|
37
|
+
if (!raw || typeof raw !== 'object') {
|
|
38
|
+
return undefined;
|
|
39
|
+
}
|
|
40
|
+
const config = raw;
|
|
41
|
+
return Array.isArray(config.tools) ? config.tools : undefined;
|
|
42
|
+
}
|
|
43
|
+
export class BedrockConverseCodec {
|
|
44
|
+
transportId = 'bedrock-converse';
|
|
45
|
+
capabilities = {
|
|
46
|
+
supportsTools: true,
|
|
47
|
+
supportsStreaming: true,
|
|
48
|
+
supportsTokenCounting: true,
|
|
49
|
+
costTracking: true,
|
|
50
|
+
toolSchemaFormat: 'anthropic',
|
|
51
|
+
};
|
|
52
|
+
decodeRequest(body) {
|
|
53
|
+
return {
|
|
54
|
+
model: typeof body.modelId === 'string'
|
|
55
|
+
? body.modelId
|
|
56
|
+
: typeof body.model === 'string'
|
|
57
|
+
? body.model
|
|
58
|
+
: 'mock-model',
|
|
59
|
+
transport: this.transportId,
|
|
60
|
+
messages: normalizeMessages(body.messages),
|
|
61
|
+
tools: toolEntries(body.toolConfig),
|
|
62
|
+
toolChoice: body.toolConfig && typeof body.toolConfig === 'object'
|
|
63
|
+
? body.toolConfig.toolChoice
|
|
64
|
+
: undefined,
|
|
65
|
+
stream: body.stream === true,
|
|
66
|
+
raw: body,
|
|
67
|
+
};
|
|
68
|
+
}
|
|
69
|
+
encodeResult(result) {
|
|
70
|
+
return {
|
|
71
|
+
output: {
|
|
72
|
+
message: {
|
|
73
|
+
role: 'assistant',
|
|
74
|
+
content: result.text ? [{ text: result.text }] : [],
|
|
75
|
+
},
|
|
76
|
+
},
|
|
77
|
+
stopReason: result.finishReason === 'stop' ? 'end_turn' : result.finishReason,
|
|
78
|
+
usage: {
|
|
79
|
+
inputTokens: result.usage.promptTokens,
|
|
80
|
+
outputTokens: result.usage.completionTokens,
|
|
81
|
+
totalTokens: result.usage.totalTokens,
|
|
82
|
+
},
|
|
83
|
+
};
|
|
84
|
+
}
|
|
85
|
+
encodeStreamChunk(event) {
|
|
86
|
+
if (event.type === 'text-delta' && event.text) {
|
|
87
|
+
return `${JSON.stringify({
|
|
88
|
+
contentBlockDelta: {
|
|
89
|
+
delta: { text: event.text },
|
|
90
|
+
contentBlockIndex: 0,
|
|
91
|
+
},
|
|
92
|
+
})}\n`;
|
|
93
|
+
}
|
|
94
|
+
if (event.type === 'tool-call') {
|
|
95
|
+
return `${JSON.stringify({
|
|
96
|
+
contentBlockStart: {
|
|
97
|
+
start: {
|
|
98
|
+
toolUse: {
|
|
99
|
+
toolUseId: event.id,
|
|
100
|
+
name: event.name,
|
|
101
|
+
},
|
|
102
|
+
},
|
|
103
|
+
contentBlockIndex: 1,
|
|
104
|
+
},
|
|
105
|
+
})}\n${JSON.stringify({
|
|
106
|
+
contentBlockDelta: {
|
|
107
|
+
delta: { toolUse: { input: event.arguments } },
|
|
108
|
+
contentBlockIndex: 1,
|
|
109
|
+
},
|
|
110
|
+
})}\n`;
|
|
111
|
+
}
|
|
112
|
+
if (event.type === 'done') {
|
|
113
|
+
return `${JSON.stringify({
|
|
114
|
+
messageStop: { stopReason: event.finishReason === 'stop' ? 'end_turn' : event.finishReason },
|
|
115
|
+
metadata: event.usage
|
|
116
|
+
? {
|
|
117
|
+
usage: {
|
|
118
|
+
inputTokens: event.usage.promptTokens,
|
|
119
|
+
outputTokens: event.usage.completionTokens,
|
|
120
|
+
totalTokens: event.usage.totalTokens,
|
|
121
|
+
},
|
|
122
|
+
}
|
|
123
|
+
: undefined,
|
|
124
|
+
})}\n`;
|
|
125
|
+
}
|
|
126
|
+
return '';
|
|
127
|
+
}
|
|
128
|
+
normalizeTools(tools) {
|
|
129
|
+
const result = [];
|
|
130
|
+
for (const tool of tools) {
|
|
131
|
+
if (!tool || typeof tool !== 'object') {
|
|
132
|
+
continue;
|
|
133
|
+
}
|
|
134
|
+
const record = tool;
|
|
135
|
+
const spec = record.toolSpec && typeof record.toolSpec === 'object'
|
|
136
|
+
? record.toolSpec
|
|
137
|
+
: record;
|
|
138
|
+
if (typeof spec.name !== 'string') {
|
|
139
|
+
continue;
|
|
140
|
+
}
|
|
141
|
+
const inputSchema = spec.inputSchema && typeof spec.inputSchema === 'object'
|
|
142
|
+
? spec.inputSchema
|
|
143
|
+
: undefined;
|
|
144
|
+
result.push({
|
|
145
|
+
name: spec.name,
|
|
146
|
+
description: typeof spec.description === 'string' ? spec.description : undefined,
|
|
147
|
+
parameters: inputSchema?.json && typeof inputSchema.json === 'object'
|
|
148
|
+
? inputSchema.json
|
|
149
|
+
: inputSchema,
|
|
150
|
+
});
|
|
151
|
+
}
|
|
152
|
+
return result;
|
|
153
|
+
}
|
|
154
|
+
denormalizeTools(tools) {
|
|
155
|
+
return tools.map((tool) => ({
|
|
156
|
+
toolSpec: {
|
|
157
|
+
name: tool.name,
|
|
158
|
+
description: tool.description,
|
|
159
|
+
inputSchema: { json: tool.parameters ?? { type: 'object', properties: {} } },
|
|
160
|
+
},
|
|
161
|
+
}));
|
|
162
|
+
}
|
|
163
|
+
extractCostRecord(result) {
|
|
164
|
+
if (!result.usage) {
|
|
165
|
+
return undefined;
|
|
166
|
+
}
|
|
167
|
+
return {
|
|
168
|
+
inputTokens: result.usage.promptTokens,
|
|
169
|
+
outputTokens: result.usage.completionTokens,
|
|
170
|
+
model: result.model,
|
|
171
|
+
provider: 'bedrock',
|
|
172
|
+
};
|
|
173
|
+
}
|
|
174
|
+
}
|
|
175
|
+
//# sourceMappingURL=bedrock.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"bedrock.js","sourceRoot":"","sources":["../../src/codecs/bedrock.ts"],"names":[],"mappings":"AAYA,SAAS,eAAe,CAAC,OAAgB;IACvC,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAChC,OAAO,OAAO,CAAC;IACjB,CAAC;IACD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC5B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,OAAO,OAAO;SACX,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACZ,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC7B,OAAO,IAAI,CAAC;QACd,CAAC;QACD,IACE,IAAI;eACD,OAAO,IAAI,KAAK,QAAQ;eACxB,MAAM,IAAI,IAAI;eACd,OAAQ,IAA2B,CAAC,IAAI,KAAK,QAAQ,EACxD,CAAC;YACD,OAAQ,IAAyB,CAAC,IAAI,CAAC;QACzC,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC;SACD,MAAM,CAAC,OAAO,CAAC;SACf,IAAI,CAAC,GAAG,CAAC,CAAC;AACf,CAAC;AAED,SAAS,iBAAiB,CAAC,GAAY;IACrC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;QACxB,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,OAAO,GAAG,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;QACvB,MAAM,MAAM,GAAG,KAA8C,CAAC;QAC9D,OAAO;YACL,IAAI,EAAE,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM;YAC5D,OAAO,EAAE,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC;SACzC,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,WAAW,CAAC,GAAY;IAC/B,IAAI,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;QACpC,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,MAAM,GAAG,GAA0B,CAAC;IAC1C,OAAO,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;AAChE,CAAC;AAED,MAAM,OAAO,oBAAoB;IACtB,WAAW,GAAG,kBAAkB,CAAC;IAEjC,YAAY,GAAsB;QACzC,aAAa,EAAE,IAAI;QACnB,iBAAiB,EAAE,IAAI;QACvB,qBAAqB,EAAE,IAAI;QAC3B,YAAY,EAAE,IAAI;QAClB,gBAAgB,EAAE,WAAW;KAC9B,CAAC;IAEF,aAAa,CAAC,IAA6B;QACzC,OAAO;YACL,KAAK,EAAE,OAAO,IAAI,CAAC,OAAO,KAAK,QAAQ;gBACrC,CAAC,CAAC,IAAI,CAAC,OAAO;gBACd,CAAC,CAAC,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ;oBAC9B,CAAC,CAAC,IAAI,CAAC,KAAK;oBACZ,CAAC,CAAC,YAAY;YAClB,SAAS,EAAE,IAAI,CAAC,WAAW;YAC3B,QAAQ,EAAE,iBAAiB,CAAC,IAAI,CAAC,QAAQ,CAAC;YAC1C,KAAK,EAAE,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC;YACnC,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,OAAO,IAAI,CAAC,UAAU,KAAK,QAAQ;gBAChE,CAAC,CAAE,IAAI,CAAC,UAAuC,CAAC,UAAU;gBAC1D,CAAC,CAAC,SAAS;YACb,MAAM,EAAE,IAAI,CAAC,MAAM,KAAK,IAAI;YAC5B,GAAG,EAAE,IAAI;SACV,CAAC;IACJ,CAAC;IAED,YAAY,CAAC,MAAwB;QACnC,OAAO;YACL,MAAM,EAAE;gBACN,OAAO,EAAE;oBACP,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE;iBACpD;aACF;YACD,UAAU,EAAE,MAAM,CAAC,YAAY,KAAK,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,YAAY;YAC7E,KAAK,EAAE;gBACL,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,YAAY;gBACtC,YAAY,EAAE,MAAM,CAAC,KAAK,CAAC,gBAAgB;gBAC3C,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,WAAW;aACtC;SACF,CAAC;IACJ,CAAC;IAED,iBAAiB,CAAC,KAA4B;QAC5C,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;YAC9C,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC;gBACvB,iBAAiB,EAAE;oBACjB,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE;oBAC3B,iBAAiB,EAAE,CAAC;iBACrB;aACF,CAAC,IAAI,CAAC;QACT,CAAC;QAED,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC/B,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC;gBACvB,iBAAiB,EAAE;oBACjB,KAAK,EAAE;wBACL,OAAO,EAAE;4BACP,SAAS,EAAE,KAAK,CAAC,EAAE;4BACnB,IAAI,EAAE,KAAK,CAAC,IAAI;yBACjB;qBACF;oBACD,iBAAiB,EAAE,CAAC;iBACrB;aACF,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC;gBACpB,iBAAiB,EAAE;oBACjB,KAAK,EAAE,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,SAAS,EAAE,EAAE;oBAC9C,iBAAiB,EAAE,CAAC;iBACrB;aACF,CAAC,IAAI,CAAC;QACT,CAAC;QAED,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC1B,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC;gBACvB,WAAW,EAAE,EAAE,UAAU,EAAE,KAAK,CAAC,YAAY,KAAK,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC,YAAY,EAAE;gBAC5F,QAAQ,EAAE,KAAK,CAAC,KAAK;oBACnB,CAAC,CAAC;wBACE,KAAK,EAAE;4BACL,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,YAAY;4BACrC,YAAY,EAAE,KAAK,CAAC,KAAK,CAAC,gBAAgB;4BAC1C,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,WAAW;yBACrC;qBACF;oBACH,CAAC,CAAC,SAAS;aACd,CAAC,IAAI,CAAC;QACT,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,cAAc,CAAC,KAAgB;QAC7B,MAAM,MAAM,GAA+B,EAAE,CAAC;QAC9C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACtC,SAAS;YACX,CAAC;YACD,MAAM,MAAM,GAAG,IAA+B,CAAC;YAC/C,MAAM,IAAI,GAAG,MAAM,CAAC,QAAQ,IAAI,OAAO,MAAM,CAAC,QAAQ,KAAK,QAAQ;gBACjE,CAAC,CAAC,MAAM,CAAC,QAAmC;gBAC5C,CAAC,CAAC,MAAM,CAAC;YACX,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAClC,SAAS;YACX,CAAC;YACD,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,IAAI,OAAO,IAAI,CAAC,WAAW,KAAK,QAAQ;gBAC1E,CAAC,CAAC,IAAI,CAAC,WAAsC;gBAC7C,CAAC,CAAC,SAAS,CAAC;YACd,MAAM,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,WAAW,EAAE,OAAO,IAAI,CAAC,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS;gBAChF,UAAU,EAAE,WAAW,EAAE,IAAI,IAAI,OAAO,WAAW,CAAC,IAAI,KAAK,QAAQ;oBACnE,CAAC,CAAC,WAAW,CAAC,IAA+B;oBAC7C,CAAC,CAAC,WAAW;aAChB,CAAC,CAAC;QACL,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,gBAAgB,CAAC,KAAiC;QAChD,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1B,QAAQ,EAAE;gBACR,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,WAAW,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,UAAU,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE,EAAE;aAC7E;SACF,CAAC,CAAC,CAAC;IACN,CAAC;IAED,iBAAiB,CAAC,MAAwB;QACxC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YAClB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,OAAO;YACL,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,YAAY;YACtC,YAAY,EAAE,MAAM,CAAC,KAAK,CAAC,gBAAgB;YAC3C,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,QAAQ,EAAE,SAAS;SACpB,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import type { CompletionRequest, CompletionResult, CompletionStreamEvent } from '../types.js';
|
|
2
|
+
import type { TransportCodec, CodecCapabilities, NormalizedToolDefinition, NormalizedCostRecord } from '../codec.js';
|
|
3
|
+
export declare class GoogleCodec implements TransportCodec {
|
|
4
|
+
readonly transportId = "google";
|
|
5
|
+
readonly capabilities: CodecCapabilities;
|
|
6
|
+
decodeRequest(body: Record<string, unknown>): CompletionRequest;
|
|
7
|
+
encodeResult(result: CompletionResult): unknown;
|
|
8
|
+
encodeStreamChunk(event: CompletionStreamEvent): string;
|
|
9
|
+
normalizeTools(tools: unknown[]): NormalizedToolDefinition[];
|
|
10
|
+
denormalizeTools(tools: NormalizedToolDefinition[]): unknown[];
|
|
11
|
+
extractCostRecord(result: CompletionResult): NormalizedCostRecord | undefined;
|
|
12
|
+
}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"google.d.ts","sourceRoot":"","sources":["../../src/codecs/google.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,iBAAiB,EACjB,gBAAgB,EAChB,qBAAqB,EACtB,MAAM,aAAa,CAAC;AAErB,OAAO,KAAK,EACV,cAAc,EACd,iBAAiB,EACjB,wBAAwB,EACxB,oBAAoB,EACrB,MAAM,aAAa,CAAC;AAsErB,qBAAa,WAAY,YAAW,cAAc;IAChD,QAAQ,CAAC,WAAW,YAAY;IAEhC,QAAQ,CAAC,YAAY,EAAE,iBAAiB,CAMtC;IAEF,aAAa,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,iBAAiB;IAmC/D,YAAY,CAAC,MAAM,EAAE,gBAAgB,GAAG,OAAO;IAsB/C,iBAAiB,CAAC,KAAK,EAAE,qBAAqB,GAAG,MAAM;IAkBvD,cAAc,CAAC,KAAK,EAAE,OAAO,EAAE,GAAG,wBAAwB,EAAE;IA2C5D,gBAAgB,CAAC,KAAK,EAAE,wBAAwB,EAAE,GAAG,OAAO,EAAE;IAY9D,iBAAiB,CACf,MAAM,EAAE,gBAAgB,GACvB,oBAAoB,GAAG,SAAS;CAWpC"}
|
|
@@ -0,0 +1,176 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Parse Google AI "parts" array into a plain string.
|
|
3
|
+
*/
|
|
4
|
+
function parsePartsContent(parts) {
|
|
5
|
+
if (!Array.isArray(parts)) {
|
|
6
|
+
return '';
|
|
7
|
+
}
|
|
8
|
+
return parts
|
|
9
|
+
.map((part) => {
|
|
10
|
+
if (typeof part === 'string') {
|
|
11
|
+
return part;
|
|
12
|
+
}
|
|
13
|
+
if (part &&
|
|
14
|
+
typeof part === 'object' &&
|
|
15
|
+
'text' in part &&
|
|
16
|
+
typeof part.text === 'string') {
|
|
17
|
+
return part.text;
|
|
18
|
+
}
|
|
19
|
+
return '';
|
|
20
|
+
})
|
|
21
|
+
.filter(Boolean)
|
|
22
|
+
.join(' ');
|
|
23
|
+
}
|
|
24
|
+
/**
|
|
25
|
+
* Map a Google role string to the normalised internal role.
|
|
26
|
+
* Google uses "model" where other providers use "assistant".
|
|
27
|
+
*/
|
|
28
|
+
function mapGoogleRole(role) {
|
|
29
|
+
if (typeof role !== 'string') {
|
|
30
|
+
return 'user';
|
|
31
|
+
}
|
|
32
|
+
return role === 'model' ? 'assistant' : role;
|
|
33
|
+
}
|
|
34
|
+
function normalizeMessages(raw) {
|
|
35
|
+
if (!Array.isArray(raw)) {
|
|
36
|
+
return [];
|
|
37
|
+
}
|
|
38
|
+
return raw.map((entry) => {
|
|
39
|
+
const record = entry;
|
|
40
|
+
return {
|
|
41
|
+
role: mapGoogleRole(record.role),
|
|
42
|
+
content: parsePartsContent(record.parts),
|
|
43
|
+
};
|
|
44
|
+
});
|
|
45
|
+
}
|
|
46
|
+
export class GoogleCodec {
|
|
47
|
+
transportId = 'google';
|
|
48
|
+
capabilities = {
|
|
49
|
+
supportsTools: true,
|
|
50
|
+
supportsStreaming: true,
|
|
51
|
+
supportsTokenCounting: true,
|
|
52
|
+
costTracking: true,
|
|
53
|
+
toolSchemaFormat: 'google',
|
|
54
|
+
};
|
|
55
|
+
decodeRequest(body) {
|
|
56
|
+
const messages = normalizeMessages(body.contents);
|
|
57
|
+
// Extract tools from body.tools[].functionDeclarations
|
|
58
|
+
let tools;
|
|
59
|
+
if (Array.isArray(body.tools)) {
|
|
60
|
+
const declarations = [];
|
|
61
|
+
for (const toolEntry of body.tools) {
|
|
62
|
+
if (Array.isArray(toolEntry.functionDeclarations)) {
|
|
63
|
+
declarations.push(...toolEntry.functionDeclarations.map((decl) => ({
|
|
64
|
+
...decl,
|
|
65
|
+
parameters: decl.parameters ?? decl.parametersJsonSchema,
|
|
66
|
+
})));
|
|
67
|
+
}
|
|
68
|
+
}
|
|
69
|
+
if (declarations.length > 0) {
|
|
70
|
+
tools = declarations;
|
|
71
|
+
}
|
|
72
|
+
}
|
|
73
|
+
const model = typeof body.model === 'string' ? body.model : 'mock-model';
|
|
74
|
+
return {
|
|
75
|
+
model,
|
|
76
|
+
transport: this.transportId,
|
|
77
|
+
messages,
|
|
78
|
+
tools,
|
|
79
|
+
stream: false,
|
|
80
|
+
raw: body,
|
|
81
|
+
};
|
|
82
|
+
}
|
|
83
|
+
encodeResult(result) {
|
|
84
|
+
return {
|
|
85
|
+
candidates: [
|
|
86
|
+
{
|
|
87
|
+
content: {
|
|
88
|
+
role: 'model',
|
|
89
|
+
parts: [{ text: result.text }],
|
|
90
|
+
},
|
|
91
|
+
finishReason: result.finishReason === 'stop'
|
|
92
|
+
? 'STOP'
|
|
93
|
+
: result.finishReason,
|
|
94
|
+
},
|
|
95
|
+
],
|
|
96
|
+
usageMetadata: {
|
|
97
|
+
promptTokenCount: result.usage.promptTokens,
|
|
98
|
+
candidatesTokenCount: result.usage.completionTokens,
|
|
99
|
+
totalTokenCount: result.usage.totalTokens,
|
|
100
|
+
},
|
|
101
|
+
};
|
|
102
|
+
}
|
|
103
|
+
encodeStreamChunk(event) {
|
|
104
|
+
if (event.type === 'text-delta' && event.text) {
|
|
105
|
+
return (JSON.stringify({
|
|
106
|
+
candidates: [
|
|
107
|
+
{
|
|
108
|
+
content: {
|
|
109
|
+
parts: [{ text: event.text }],
|
|
110
|
+
role: 'model',
|
|
111
|
+
},
|
|
112
|
+
},
|
|
113
|
+
],
|
|
114
|
+
}) + '\n');
|
|
115
|
+
}
|
|
116
|
+
return '';
|
|
117
|
+
}
|
|
118
|
+
normalizeTools(tools) {
|
|
119
|
+
const result = [];
|
|
120
|
+
for (const entry of tools) {
|
|
121
|
+
if (!entry || typeof entry !== 'object') {
|
|
122
|
+
continue;
|
|
123
|
+
}
|
|
124
|
+
const record = entry;
|
|
125
|
+
// Wrapper form: { functionDeclarations: [...] }
|
|
126
|
+
if (Array.isArray(record.functionDeclarations)) {
|
|
127
|
+
for (const decl of record.functionDeclarations) {
|
|
128
|
+
result.push({
|
|
129
|
+
name: decl.name,
|
|
130
|
+
description: decl.description,
|
|
131
|
+
parameters: decl.parameters ?? decl.parametersJsonSchema,
|
|
132
|
+
});
|
|
133
|
+
}
|
|
134
|
+
continue;
|
|
135
|
+
}
|
|
136
|
+
// Direct functionDeclaration object with name
|
|
137
|
+
if (typeof record.name === 'string') {
|
|
138
|
+
result.push({
|
|
139
|
+
name: record.name,
|
|
140
|
+
description: typeof record.description === 'string'
|
|
141
|
+
? record.description
|
|
142
|
+
: undefined,
|
|
143
|
+
parameters: record.parameters && typeof record.parameters === 'object'
|
|
144
|
+
? record.parameters
|
|
145
|
+
: record.parametersJsonSchema && typeof record.parametersJsonSchema === 'object'
|
|
146
|
+
? record.parametersJsonSchema
|
|
147
|
+
: undefined,
|
|
148
|
+
});
|
|
149
|
+
}
|
|
150
|
+
}
|
|
151
|
+
return result;
|
|
152
|
+
}
|
|
153
|
+
denormalizeTools(tools) {
|
|
154
|
+
return [
|
|
155
|
+
{
|
|
156
|
+
functionDeclarations: tools.map((t) => ({
|
|
157
|
+
name: t.name,
|
|
158
|
+
description: t.description,
|
|
159
|
+
parameters: t.parameters,
|
|
160
|
+
})),
|
|
161
|
+
},
|
|
162
|
+
];
|
|
163
|
+
}
|
|
164
|
+
extractCostRecord(result) {
|
|
165
|
+
if (!result.usage) {
|
|
166
|
+
return undefined;
|
|
167
|
+
}
|
|
168
|
+
return {
|
|
169
|
+
inputTokens: result.usage.promptTokens,
|
|
170
|
+
outputTokens: result.usage.completionTokens,
|
|
171
|
+
model: result.model,
|
|
172
|
+
provider: 'google',
|
|
173
|
+
};
|
|
174
|
+
}
|
|
175
|
+
}
|
|
176
|
+
//# sourceMappingURL=google.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"google.js","sourceRoot":"","sources":["../../src/codecs/google.ts"],"names":[],"mappings":"AAaA;;GAEG;AACH,SAAS,iBAAiB,CAAC,KAAc;IACvC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QAC1B,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,OAAO,KAAK;SACT,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACZ,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC7B,OAAO,IAAI,CAAC;QACd,CAAC;QACD,IACE,IAAI;YACJ,OAAO,IAAI,KAAK,QAAQ;YACxB,MAAM,IAAI,IAAI;YACd,OAAQ,IAA2B,CAAC,IAAI,KAAK,QAAQ,EACrD,CAAC;YACD,OAAQ,IAAyB,CAAC,IAAI,CAAC;QACzC,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC;SACD,MAAM,CAAC,OAAO,CAAC;SACf,IAAI,CAAC,GAAG,CAAC,CAAC;AACf,CAAC;AAED;;;GAGG;AACH,SAAS,aAAa,CAAC,IAAa;IAClC,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC7B,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,OAAO,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC;AAC/C,CAAC;AAkBD,SAAS,iBAAiB,CACxB,GAAY;IAEZ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;QACxB,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,OAAO,GAAG,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;QACvB,MAAM,MAAM,GAAG,KAA2B,CAAC;QAC3C,OAAO;YACL,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC;YAChC,OAAO,EAAE,iBAAiB,CAAC,MAAM,CAAC,KAAK,CAAC;SACzC,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,OAAO,WAAW;IACb,WAAW,GAAG,QAAQ,CAAC;IAEvB,YAAY,GAAsB;QACzC,aAAa,EAAE,IAAI;QACnB,iBAAiB,EAAE,IAAI;QACvB,qBAAqB,EAAE,IAAI;QAC3B,YAAY,EAAE,IAAI;QAClB,gBAAgB,EAAE,QAAQ;KAC3B,CAAC;IAEF,aAAa,CAAC,IAA6B;QACzC,MAAM,QAAQ,GAAG,iBAAiB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAElD,uDAAuD;QACvD,IAAI,KAA4B,CAAC;QACjC,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;YAC9B,MAAM,YAAY,GAAgC,EAAE,CAAC;YACrD,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,KAA0B,EAAE,CAAC;gBACxD,IAAI,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,oBAAoB,CAAC,EAAE,CAAC;oBAClD,YAAY,CAAC,IAAI,CACf,GAAG,SAAS,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;wBAC/C,GAAG,IAAI;wBACP,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,oBAAoB;qBACzD,CAAC,CAAC,CACJ,CAAC;gBACJ,CAAC;YACH,CAAC;YACD,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC5B,KAAK,GAAG,YAAY,CAAC;YACvB,CAAC;QACH,CAAC;QAED,MAAM,KAAK,GACT,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,YAAY,CAAC;QAE7D,OAAO;YACL,KAAK;YACL,SAAS,EAAE,IAAI,CAAC,WAAW;YAC3B,QAAQ;YACR,KAAK;YACL,MAAM,EAAE,KAAK;YACb,GAAG,EAAE,IAAI;SACV,CAAC;IACJ,CAAC;IAED,YAAY,CAAC,MAAwB;QACnC,OAAO;YACL,UAAU,EAAE;gBACV;oBACE,OAAO,EAAE;wBACP,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC;qBAC/B;oBACD,YAAY,EACV,MAAM,CAAC,YAAY,KAAK,MAAM;wBAC5B,CAAC,CAAC,MAAM;wBACR,CAAC,CAAC,MAAM,CAAC,YAAY;iBAC1B;aACF;YACD,aAAa,EAAE;gBACb,gBAAgB,EAAE,MAAM,CAAC,KAAK,CAAC,YAAY;gBAC3C,oBAAoB,EAAE,MAAM,CAAC,KAAK,CAAC,gBAAgB;gBACnD,eAAe,EAAE,MAAM,CAAC,KAAK,CAAC,WAAW;aAC1C;SACF,CAAC;IACJ,CAAC;IAED,iBAAiB,CAAC,KAA4B;QAC5C,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;YAC9C,OAAO,CACL,IAAI,CAAC,SAAS,CAAC;gBACb,UAAU,EAAE;oBACV;wBACE,OAAO,EAAE;4BACP,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;4BAC7B,IAAI,EAAE,OAAO;yBACd;qBACF;iBACF;aACF,CAAC,GAAG,IAAI,CACV,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,cAAc,CAAC,KAAgB;QAC7B,MAAM,MAAM,GAA+B,EAAE,CAAC;QAE9C,KAAK,MAAM,KAAK,IAAI,KAAK,EAAE,CAAC;YAC1B,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;gBACxC,SAAS;YACX,CAAC;YAED,MAAM,MAAM,GAAG,KAAgC,CAAC;YAEhD,gDAAgD;YAChD,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,oBAAoB,CAAC,EAAE,CAAC;gBAC/C,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,oBAAmD,EAAE,CAAC;oBAC9E,MAAM,CAAC,IAAI,CAAC;wBACV,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,WAAW,EAAE,IAAI,CAAC,WAAW;wBAC7B,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,oBAAoB;qBACzD,CAAC,CAAC;gBACL,CAAC;gBACD,SAAS;YACX,CAAC;YAED,8CAA8C;YAC9C,IAAI,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACpC,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,MAAM,CAAC,IAAI;oBACjB,WAAW,EACT,OAAO,MAAM,CAAC,WAAW,KAAK,QAAQ;wBACpC,CAAC,CAAC,MAAM,CAAC,WAAW;wBACpB,CAAC,CAAC,SAAS;oBACf,UAAU,EACR,MAAM,CAAC,UAAU,IAAI,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ;wBACxD,CAAC,CAAE,MAAM,CAAC,UAAsC;wBAChD,CAAC,CAAC,MAAM,CAAC,oBAAoB,IAAI,OAAO,MAAM,CAAC,oBAAoB,KAAK,QAAQ;4BAC9E,CAAC,CAAE,MAAM,CAAC,oBAAgD;4BAC5D,CAAC,CAAC,SAAS;iBAChB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,gBAAgB,CAAC,KAAiC;QAChD,OAAO;YACL;gBACE,oBAAoB,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBACtC,IAAI,EAAE,CAAC,CAAC,IAAI;oBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;oBAC1B,UAAU,EAAE,CAAC,CAAC,UAAU;iBACzB,CAAC,CAAC;aACJ;SACF,CAAC;IACJ,CAAC;IAED,iBAAiB,CACf,MAAwB;QAExB,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YAClB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,OAAO;YACL,WAAW,EAAE,MAAM,CAAC,KAAK,CAAC,YAAY;YACtC,YAAY,EAAE,MAAM,CAAC,KAAK,CAAC,gBAAgB;YAC3C,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,QAAQ,EAAE,QAAQ;SACnB,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1,30 @@
|
|
|
1
|
+
import type { CodecCapabilities, TransportCodec } from '../codec.js';
|
|
2
|
+
import type { CompletionUsage } from '../types.js';
|
|
3
|
+
import type { TransportId } from '../types.js';
|
|
4
|
+
type ToolSchemaFormat = CodecCapabilities['toolSchemaFormat'];
|
|
5
|
+
export interface RegisterCodecOptions {
|
|
6
|
+
aliases?: TransportId[];
|
|
7
|
+
override?: boolean;
|
|
8
|
+
}
|
|
9
|
+
export interface TransportDescriptorLike {
|
|
10
|
+
transportId?: string;
|
|
11
|
+
runtimeId?: string;
|
|
12
|
+
runtimeKind?: string;
|
|
13
|
+
id?: string;
|
|
14
|
+
codecCapabilities?: Partial<CodecCapabilities>;
|
|
15
|
+
}
|
|
16
|
+
export declare function registerCodec(codec: TransportCodec, options?: RegisterCodecOptions): readonly TransportId[];
|
|
17
|
+
export declare function listRegisteredCodecs(): readonly TransportCodec[];
|
|
18
|
+
/**
|
|
19
|
+
* Look up a TransportCodec by transport identifier.
|
|
20
|
+
* Returns `undefined` when no codec has been registered for the given id.
|
|
21
|
+
*/
|
|
22
|
+
export declare function getCodec(transportId: TransportId): TransportCodec | undefined;
|
|
23
|
+
export declare function getCodecForDescriptor(descriptor: TransportDescriptorLike): TransportCodec | undefined;
|
|
24
|
+
export declare function convertTools(tools: unknown[], from: ToolSchemaFormat, to: ToolSchemaFormat): unknown[];
|
|
25
|
+
export declare function normalizeUsage(raw: unknown): CompletionUsage;
|
|
26
|
+
export { AnthropicCodec } from './anthropic.js';
|
|
27
|
+
export { BedrockConverseCodec } from './bedrock.js';
|
|
28
|
+
export { GoogleCodec } from './google.js';
|
|
29
|
+
export { OpenAiChatCodec } from './openai-chat.js';
|
|
30
|
+
export { OpenAiResponsesCodec } from './openai-responses.js';
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/codecs/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAA4B,cAAc,EAAE,MAAM,aAAa,CAAC;AAC/F,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AACnD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAU/C,KAAK,gBAAgB,GAAG,iBAAiB,CAAC,kBAAkB,CAAC,CAAC;AAE9D,MAAM,WAAW,oBAAoB;IACnC,OAAO,CAAC,EAAE,WAAW,EAAE,CAAC;IACxB,QAAQ,CAAC,EAAE,OAAO,CAAC;CACpB;AAED,MAAM,WAAW,uBAAuB;IACtC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,iBAAiB,CAAC,EAAE,OAAO,CAAC,iBAAiB,CAAC,CAAC;CAChD;AA8BD,wBAAgB,aAAa,CAC3B,KAAK,EAAE,cAAc,EACrB,OAAO,GAAE,oBAAyB,GACjC,SAAS,WAAW,EAAE,CAiBxB;AAED,wBAAgB,oBAAoB,IAAI,SAAS,cAAc,EAAE,CAEhE;AAQD;;;GAGG;AACH,wBAAgB,QAAQ,CAAC,WAAW,EAAE,WAAW,GAAG,cAAc,GAAG,SAAS,CAE7E;AAED,wBAAgB,qBAAqB,CACnC,UAAU,EAAE,uBAAuB,GAClC,cAAc,GAAG,SAAS,CAU5B;AAYD,wBAAgB,YAAY,CAC1B,KAAK,EAAE,OAAO,EAAE,EAChB,IAAI,EAAE,gBAAgB,EACtB,EAAE,EAAE,gBAAgB,GACnB,OAAO,EAAE,CAcX;AAYD,wBAAgB,cAAc,CAAC,GAAG,EAAE,OAAO,GAAG,eAAe,CAwB5D;AAED,OAAO,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAChD,OAAO,EAAE,oBAAoB,EAAE,MAAM,cAAc,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACnD,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC"}
|
|
@@ -0,0 +1,115 @@
|
|
|
1
|
+
import { AnthropicCodec } from './anthropic.js';
|
|
2
|
+
import { BedrockConverseCodec } from './bedrock.js';
|
|
3
|
+
import { GoogleCodec } from './google.js';
|
|
4
|
+
import { OpenAiChatCodec } from './openai-chat.js';
|
|
5
|
+
import { OpenAiResponsesCodec } from './openai-responses.js';
|
|
6
|
+
const codecRegistry = new Map();
|
|
7
|
+
function normalizeTransportId(transportId) {
|
|
8
|
+
return transportId
|
|
9
|
+
.replace(/^transport-runtime:/, '')
|
|
10
|
+
.replace(/^model-transport:/, '')
|
|
11
|
+
.replace(/-messages$/, '')
|
|
12
|
+
.replace(/-completions$/, '')
|
|
13
|
+
.replace(/-converse$/, '')
|
|
14
|
+
.replace(/^bedrock$/, 'bedrock-converse')
|
|
15
|
+
.replace(/^openai-chat$/, 'openai-chat')
|
|
16
|
+
.replace(/^openai-responses$/, 'openai-responses')
|
|
17
|
+
.replace(/^anthropic$/, 'anthropic')
|
|
18
|
+
.replace(/^google$/, 'google');
|
|
19
|
+
}
|
|
20
|
+
function codecIds(codec, aliases = []) {
|
|
21
|
+
return Array.from(new Set([codec.transportId, ...aliases]
|
|
22
|
+
.map((id) => id.trim())
|
|
23
|
+
.filter(Boolean)
|
|
24
|
+
.flatMap((id) => {
|
|
25
|
+
const normalized = normalizeTransportId(id);
|
|
26
|
+
return normalized === id ? [id] : [id, normalized];
|
|
27
|
+
})));
|
|
28
|
+
}
|
|
29
|
+
export function registerCodec(codec, options = {}) {
|
|
30
|
+
const ids = codecIds(codec, options.aliases);
|
|
31
|
+
if (ids.length === 0) {
|
|
32
|
+
throw new Error('Cannot register codec without a transport id.');
|
|
33
|
+
}
|
|
34
|
+
for (const id of ids) {
|
|
35
|
+
const existing = codecRegistry.get(id);
|
|
36
|
+
if (existing && existing !== codec && options.override !== true) {
|
|
37
|
+
throw new Error(`Codec already registered for transport id ${id}.`);
|
|
38
|
+
}
|
|
39
|
+
}
|
|
40
|
+
for (const id of ids) {
|
|
41
|
+
codecRegistry.set(id, codec);
|
|
42
|
+
}
|
|
43
|
+
return ids;
|
|
44
|
+
}
|
|
45
|
+
export function listRegisteredCodecs() {
|
|
46
|
+
return Array.from(new Set(codecRegistry.values()));
|
|
47
|
+
}
|
|
48
|
+
registerCodec(new AnthropicCodec());
|
|
49
|
+
registerCodec(new GoogleCodec(), { aliases: ['vertex-native'] });
|
|
50
|
+
registerCodec(new OpenAiChatCodec(), { aliases: ['azure-foundry', 'openai-chat-completions'] });
|
|
51
|
+
registerCodec(new OpenAiResponsesCodec());
|
|
52
|
+
registerCodec(new BedrockConverseCodec(), { aliases: ['bedrock'] });
|
|
53
|
+
/**
|
|
54
|
+
* Look up a TransportCodec by transport identifier.
|
|
55
|
+
* Returns `undefined` when no codec has been registered for the given id.
|
|
56
|
+
*/
|
|
57
|
+
export function getCodec(transportId) {
|
|
58
|
+
return codecRegistry.get(transportId) ?? codecRegistry.get(normalizeTransportId(transportId));
|
|
59
|
+
}
|
|
60
|
+
export function getCodecForDescriptor(descriptor) {
|
|
61
|
+
const candidate = descriptor.transportId
|
|
62
|
+
?? descriptor.runtimeId
|
|
63
|
+
?? descriptor.runtimeKind
|
|
64
|
+
?? descriptor.id;
|
|
65
|
+
if (!candidate) {
|
|
66
|
+
return undefined;
|
|
67
|
+
}
|
|
68
|
+
return getCodec(candidate);
|
|
69
|
+
}
|
|
70
|
+
function requiredCodec(format) {
|
|
71
|
+
const codec = [...codecRegistry.values()].find((candidate) => candidate.capabilities.toolSchemaFormat === format);
|
|
72
|
+
if (!codec) {
|
|
73
|
+
throw new Error(`No codec registered for tool schema format ${format}.`);
|
|
74
|
+
}
|
|
75
|
+
return codec;
|
|
76
|
+
}
|
|
77
|
+
export function convertTools(tools, from, to) {
|
|
78
|
+
if (from === to) {
|
|
79
|
+
return tools;
|
|
80
|
+
}
|
|
81
|
+
if (from === 'none' || to === 'none') {
|
|
82
|
+
return [];
|
|
83
|
+
}
|
|
84
|
+
const source = requiredCodec(from);
|
|
85
|
+
const target = requiredCodec(to);
|
|
86
|
+
const normalized = source.normalizeTools
|
|
87
|
+
? source.normalizeTools(tools)
|
|
88
|
+
: [];
|
|
89
|
+
return target.denormalizeTools ? target.denormalizeTools(normalized) : normalized;
|
|
90
|
+
}
|
|
91
|
+
function numberField(record, ...keys) {
|
|
92
|
+
for (const key of keys) {
|
|
93
|
+
const value = record[key];
|
|
94
|
+
if (typeof value === 'number' && Number.isFinite(value)) {
|
|
95
|
+
return value;
|
|
96
|
+
}
|
|
97
|
+
}
|
|
98
|
+
return 0;
|
|
99
|
+
}
|
|
100
|
+
export function normalizeUsage(raw) {
|
|
101
|
+
const record = raw && typeof raw === 'object'
|
|
102
|
+
? raw
|
|
103
|
+
: {};
|
|
104
|
+
const promptTokens = numberField(record, 'promptTokens', 'prompt_tokens', 'inputTokens', 'input_tokens', 'promptTokenCount');
|
|
105
|
+
const completionTokens = numberField(record, 'completionTokens', 'completion_tokens', 'outputTokens', 'output_tokens', 'candidatesTokenCount');
|
|
106
|
+
const totalTokens = numberField(record, 'totalTokens', 'total_tokens', 'totalTokenCount')
|
|
107
|
+
|| promptTokens + completionTokens;
|
|
108
|
+
return { promptTokens, completionTokens, totalTokens };
|
|
109
|
+
}
|
|
110
|
+
export { AnthropicCodec } from './anthropic.js';
|
|
111
|
+
export { BedrockConverseCodec } from './bedrock.js';
|
|
112
|
+
export { GoogleCodec } from './google.js';
|
|
113
|
+
export { OpenAiChatCodec } from './openai-chat.js';
|
|
114
|
+
export { OpenAiResponsesCodec } from './openai-responses.js';
|
|
115
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/codecs/index.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAChD,OAAO,EAAE,oBAAoB,EAAE,MAAM,cAAc,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACnD,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAE7D,MAAM,aAAa,GAAG,IAAI,GAAG,EAA+B,CAAC;AAiB7D,SAAS,oBAAoB,CAAC,WAAmB;IAC/C,OAAO,WAAW;SACf,OAAO,CAAC,qBAAqB,EAAE,EAAE,CAAC;SAClC,OAAO,CAAC,mBAAmB,EAAE,EAAE,CAAC;SAChC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC;SACzB,OAAO,CAAC,eAAe,EAAE,EAAE,CAAC;SAC5B,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC;SACzB,OAAO,CAAC,WAAW,EAAE,kBAAkB,CAAC;SACxC,OAAO,CAAC,eAAe,EAAE,aAAa,CAAC;SACvC,OAAO,CAAC,oBAAoB,EAAE,kBAAkB,CAAC;SACjD,OAAO,CAAC,aAAa,EAAE,WAAW,CAAC;SACnC,OAAO,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;AACnC,CAAC;AAED,SAAS,QAAQ,CAAC,KAAqB,EAAE,UAAyB,EAAE;IAClE,OAAO,KAAK,CAAC,IAAI,CACf,IAAI,GAAG,CACL,CAAC,KAAK,CAAC,WAAW,EAAE,GAAG,OAAO,CAAC;SAC5B,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;SACtB,MAAM,CAAC,OAAO,CAAC;SACf,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE;QACd,MAAM,UAAU,GAAG,oBAAoB,CAAC,EAAE,CAAC,CAAC;QAC5C,OAAO,UAAU,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;IACrD,CAAC,CAAC,CACL,CACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,aAAa,CAC3B,KAAqB,EACrB,UAAgC,EAAE;IAElC,MAAM,GAAG,GAAG,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;IAC7C,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACrB,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;IACnE,CAAC;IAED,KAAK,MAAM,EAAE,IAAI,GAAG,EAAE,CAAC;QACrB,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACvC,IAAI,QAAQ,IAAI,QAAQ,KAAK,KAAK,IAAI,OAAO,CAAC,QAAQ,KAAK,IAAI,EAAE,CAAC;YAChE,MAAM,IAAI,KAAK,CAAC,6CAA6C,EAAE,GAAG,CAAC,CAAC;QACtE,CAAC;IACH,CAAC;IAED,KAAK,MAAM,EAAE,IAAI,GAAG,EAAE,CAAC;QACrB,aAAa,CAAC,GAAG,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;IAC/B,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,UAAU,oBAAoB;IAClC,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;AACrD,CAAC;AAED,aAAa,CAAC,IAAI,cAAc,EAAE,CAAC,CAAC;AACpC,aAAa,CAAC,IAAI,WAAW,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;AACjE,aAAa,CAAC,IAAI,eAAe,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,eAAe,EAAE,yBAAyB,CAAC,EAAE,CAAC,CAAC;AAChG,aAAa,CAAC,IAAI,oBAAoB,EAAE,CAAC,CAAC;AAC1C,aAAa,CAAC,IAAI,oBAAoB,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;AAEpE;;;GAGG;AACH,MAAM,UAAU,QAAQ,CAAC,WAAwB;IAC/C,OAAO,aAAa,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,aAAa,CAAC,GAAG,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,CAAC;AAChG,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,UAAmC;IAEnC,MAAM,SAAS,GAAG,UAAU,CAAC,WAAW;WACnC,UAAU,CAAC,SAAS;WACpB,UAAU,CAAC,WAAW;WACtB,UAAU,CAAC,EAAE,CAAC;IACnB,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,QAAQ,CAAC,SAAS,CAAC,CAAC;AAC7B,CAAC;AAED,SAAS,aAAa,CAAC,MAAwB;IAC7C,MAAM,KAAK,GAAG,CAAC,GAAG,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAC5C,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,YAAY,CAAC,gBAAgB,KAAK,MAAM,CAClE,CAAC;IACF,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,IAAI,KAAK,CAAC,8CAA8C,MAAM,GAAG,CAAC,CAAC;IAC3E,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,YAAY,CAC1B,KAAgB,EAChB,IAAsB,EACtB,EAAoB;IAEpB,IAAI,IAAI,KAAK,EAAE,EAAE,CAAC;QAChB,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,IAAI,KAAK,MAAM,IAAI,EAAE,KAAK,MAAM,EAAE,CAAC;QACrC,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;IACnC,MAAM,MAAM,GAAG,aAAa,CAAC,EAAE,CAAC,CAAC;IACjC,MAAM,UAAU,GAA+B,MAAM,CAAC,cAAc;QAClE,CAAC,CAAC,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC;QAC9B,CAAC,CAAC,EAAE,CAAC;IACP,OAAO,MAAM,CAAC,gBAAgB,CAAC,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC;AACpF,CAAC;AAED,SAAS,WAAW,CAAC,MAA+B,EAAE,GAAG,IAAc;IACrE,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;QAC1B,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACxD,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IACD,OAAO,CAAC,CAAC;AACX,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,GAAY;IACzC,MAAM,MAAM,GAAG,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ;QAC3C,CAAC,CAAC,GAA8B;QAChC,CAAC,CAAC,EAAE,CAAC;IACP,MAAM,YAAY,GAAG,WAAW,CAC9B,MAAM,EACN,cAAc,EACd,eAAe,EACf,aAAa,EACb,cAAc,EACd,kBAAkB,CACnB,CAAC;IACF,MAAM,gBAAgB,GAAG,WAAW,CAClC,MAAM,EACN,kBAAkB,EAClB,mBAAmB,EACnB,cAAc,EACd,eAAe,EACf,sBAAsB,CACvB,CAAC;IACF,MAAM,WAAW,GAAG,WAAW,CAAC,MAAM,EAAE,aAAa,EAAE,cAAc,EAAE,iBAAiB,CAAC;WACpF,YAAY,GAAG,gBAAgB,CAAC;IAErC,OAAO,EAAE,YAAY,EAAE,gBAAgB,EAAE,WAAW,EAAE,CAAC;AACzD,CAAC;AAED,OAAO,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAChD,OAAO,EAAE,oBAAoB,EAAE,MAAM,cAAc,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACnD,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import type { CompletionRequest, CompletionResult, CompletionStreamEvent } from '../types.js';
|
|
2
|
+
import type { TransportCodec, CodecCapabilities, NormalizedToolDefinition, NormalizedCostRecord } from '../codec.js';
|
|
3
|
+
export declare class OpenAiChatCodec implements TransportCodec {
|
|
4
|
+
readonly transportId = "openai-chat";
|
|
5
|
+
readonly capabilities: CodecCapabilities;
|
|
6
|
+
decodeRequest(body: Record<string, unknown>): CompletionRequest;
|
|
7
|
+
encodeResult(result: CompletionResult): unknown;
|
|
8
|
+
encodeStreamChunk(event: CompletionStreamEvent): string;
|
|
9
|
+
normalizeTools(tools: unknown[]): NormalizedToolDefinition[];
|
|
10
|
+
denormalizeTools(tools: NormalizedToolDefinition[]): unknown[];
|
|
11
|
+
extractCostRecord(result: CompletionResult): NormalizedCostRecord | undefined;
|
|
12
|
+
}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai-chat.d.ts","sourceRoot":"","sources":["../../src/codecs/openai-chat.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,iBAAiB,EACjB,gBAAgB,EAChB,qBAAqB,EACtB,MAAM,aAAa,CAAC;AAErB,OAAO,KAAK,EACV,cAAc,EACd,iBAAiB,EACjB,wBAAwB,EACxB,oBAAoB,EACrB,MAAM,aAAa,CAAC;AAqErB,qBAAa,eAAgB,YAAW,cAAc;IACpD,QAAQ,CAAC,WAAW,iBAAiB;IAErC,QAAQ,CAAC,YAAY,EAAE,iBAAiB,CAMtC;IAEF,aAAa,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,iBAAiB;IAmB/D,YAAY,CAAC,MAAM,EAAE,gBAAgB,GAAG,OAAO;IA6B/C,iBAAiB,CAAC,KAAK,EAAE,qBAAqB,GAAG,MAAM;IAmCvD,cAAc,CAAC,KAAK,EAAE,OAAO,EAAE,GAAG,wBAAwB,EAAE;IAwC5D,gBAAgB,CAAC,KAAK,EAAE,wBAAwB,EAAE,GAAG,OAAO,EAAE;IAW9D,iBAAiB,CAAC,MAAM,EAAE,gBAAgB,GAAG,oBAAoB,GAAG,SAAS;CAY9E"}
|