@skyold/provider-adapters 0.1.0-beta.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 +201 -0
- package/README.md +59 -0
- package/README.zh-CN.md +15 -0
- package/dist/anthropic-messages/adapter.d.ts +3 -0
- package/dist/anthropic-messages/adapter.d.ts.map +1 -0
- package/dist/anthropic-messages/adapter.js +50 -0
- package/dist/anthropic-messages/adapter.js.map +1 -0
- package/dist/anthropic-messages/codec.d.ts +7 -0
- package/dist/anthropic-messages/codec.d.ts.map +1 -0
- package/dist/anthropic-messages/codec.js +581 -0
- package/dist/anthropic-messages/codec.js.map +1 -0
- package/dist/anthropic-messages/sse.d.ts +7 -0
- package/dist/anthropic-messages/sse.d.ts.map +1 -0
- package/dist/anthropic-messages/sse.js +77 -0
- package/dist/anthropic-messages/sse.js.map +1 -0
- package/dist/builtin-registry.d.ts +17 -0
- package/dist/builtin-registry.d.ts.map +1 -0
- package/dist/builtin-registry.js +110 -0
- package/dist/builtin-registry.js.map +1 -0
- package/dist/image-response-json.d.ts +36 -0
- package/dist/image-response-json.d.ts.map +1 -0
- package/dist/image-response-json.js +193 -0
- package/dist/image-response-json.js.map +1 -0
- package/dist/index.d.ts +24 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +24 -0
- package/dist/index.js.map +1 -0
- package/dist/openai-chat/adapter.d.ts +3 -0
- package/dist/openai-chat/adapter.d.ts.map +1 -0
- package/dist/openai-chat/adapter.js +83 -0
- package/dist/openai-chat/adapter.js.map +1 -0
- package/dist/openai-chat/codec.d.ts +7 -0
- package/dist/openai-chat/codec.d.ts.map +1 -0
- package/dist/openai-chat/codec.js +584 -0
- package/dist/openai-chat/codec.js.map +1 -0
- package/dist/openai-chat/sse.d.ts +3 -0
- package/dist/openai-chat/sse.d.ts.map +1 -0
- package/dist/openai-chat/sse.js +64 -0
- package/dist/openai-chat/sse.js.map +1 -0
- package/dist/openai-embeddings/adapter.d.ts +3 -0
- package/dist/openai-embeddings/adapter.d.ts.map +1 -0
- package/dist/openai-embeddings/adapter.js +90 -0
- package/dist/openai-embeddings/adapter.js.map +1 -0
- package/dist/openai-embeddings/codec.d.ts +7 -0
- package/dist/openai-embeddings/codec.d.ts.map +1 -0
- package/dist/openai-embeddings/codec.js +120 -0
- package/dist/openai-embeddings/codec.js.map +1 -0
- package/dist/openai-images/adapter.d.ts +4 -0
- package/dist/openai-images/adapter.d.ts.map +1 -0
- package/dist/openai-images/adapter.js +103 -0
- package/dist/openai-images/adapter.js.map +1 -0
- package/dist/openai-images/codec.d.ts +24 -0
- package/dist/openai-images/codec.d.ts.map +1 -0
- package/dist/openai-images/codec.js +979 -0
- package/dist/openai-images/codec.js.map +1 -0
- package/dist/openai-responses/adapter.d.ts +3 -0
- package/dist/openai-responses/adapter.d.ts.map +1 -0
- package/dist/openai-responses/adapter.js +102 -0
- package/dist/openai-responses/adapter.js.map +1 -0
- package/dist/openai-responses/codec.d.ts +7 -0
- package/dist/openai-responses/codec.d.ts.map +1 -0
- package/dist/openai-responses/codec.js +1086 -0
- package/dist/openai-responses/codec.js.map +1 -0
- package/dist/openai-responses/sse.d.ts +7 -0
- package/dist/openai-responses/sse.d.ts.map +1 -0
- package/dist/openai-responses/sse.js +82 -0
- package/dist/openai-responses/sse.js.map +1 -0
- package/dist/ppio-video/adapter.d.ts +37 -0
- package/dist/ppio-video/adapter.d.ts.map +1 -0
- package/dist/ppio-video/adapter.js +155 -0
- package/dist/ppio-video/adapter.js.map +1 -0
- package/dist/ppio-video/codec.d.ts +19 -0
- package/dist/ppio-video/codec.d.ts.map +1 -0
- package/dist/ppio-video/codec.js +172 -0
- package/dist/ppio-video/codec.js.map +1 -0
- package/dist/ppio-video/estimator.d.ts +21 -0
- package/dist/ppio-video/estimator.d.ts.map +1 -0
- package/dist/ppio-video/estimator.js +23 -0
- package/dist/ppio-video/estimator.js.map +1 -0
- package/dist/registry.d.ts +15 -0
- package/dist/registry.d.ts.map +1 -0
- package/dist/registry.js +36 -0
- package/dist/registry.js.map +1 -0
- package/dist/transport-profile.d.ts +17 -0
- package/dist/transport-profile.d.ts.map +1 -0
- package/dist/transport-profile.js +36 -0
- package/dist/transport-profile.js.map +1 -0
- package/dist/volc-embeddings/adapter.d.ts +4 -0
- package/dist/volc-embeddings/adapter.d.ts.map +1 -0
- package/dist/volc-embeddings/adapter.js +92 -0
- package/dist/volc-embeddings/adapter.js.map +1 -0
- package/dist/volc-embeddings/codec.d.ts +8 -0
- package/dist/volc-embeddings/codec.d.ts.map +1 -0
- package/dist/volc-embeddings/codec.js +196 -0
- package/dist/volc-embeddings/codec.js.map +1 -0
- package/dist/volc-images/adapter.d.ts +3 -0
- package/dist/volc-images/adapter.d.ts.map +1 -0
- package/dist/volc-images/adapter.js +96 -0
- package/dist/volc-images/adapter.js.map +1 -0
- package/dist/volc-images/codec.d.ts +13 -0
- package/dist/volc-images/codec.d.ts.map +1 -0
- package/dist/volc-images/codec.js +581 -0
- package/dist/volc-images/codec.js.map +1 -0
- package/dist/volc-video/adapter.d.ts +40 -0
- package/dist/volc-video/adapter.d.ts.map +1 -0
- package/dist/volc-video/adapter.js +165 -0
- package/dist/volc-video/adapter.js.map +1 -0
- package/dist/volc-video/codec.d.ts +26 -0
- package/dist/volc-video/codec.d.ts.map +1 -0
- package/dist/volc-video/codec.js +203 -0
- package/dist/volc-video/codec.js.map +1 -0
- package/dist/volc-video/estimator.d.ts +18 -0
- package/dist/volc-video/estimator.d.ts.map +1 -0
- package/dist/volc-video/estimator.js +21 -0
- package/dist/volc-video/estimator.js.map +1 -0
- package/package.json +61 -0
|
@@ -0,0 +1,64 @@
|
|
|
1
|
+
export const PROVIDER_STREAM_INVALID = 'PROVIDER_STREAM_INVALID';
|
|
2
|
+
const MAX_SSE_BUFFER_BYTES = 1_048_576;
|
|
3
|
+
const MAX_SSE_FRAME_BYTES = 1_048_576;
|
|
4
|
+
const encoder = new TextEncoder();
|
|
5
|
+
export async function* parseOpenAiSseDataFrames(body) {
|
|
6
|
+
const decoder = new TextDecoder('utf-8', { fatal: true });
|
|
7
|
+
let buffered = '';
|
|
8
|
+
let dataLines = [];
|
|
9
|
+
let dataBytes = 0;
|
|
10
|
+
const consumeLine = (line) => {
|
|
11
|
+
if (line === '') {
|
|
12
|
+
if (dataLines.length === 0)
|
|
13
|
+
return undefined;
|
|
14
|
+
const data = dataLines.join('\n');
|
|
15
|
+
dataLines = [];
|
|
16
|
+
dataBytes = 0;
|
|
17
|
+
return data;
|
|
18
|
+
}
|
|
19
|
+
if (line.startsWith(':'))
|
|
20
|
+
return undefined;
|
|
21
|
+
if (!line.startsWith('data:'))
|
|
22
|
+
throw new Error(PROVIDER_STREAM_INVALID);
|
|
23
|
+
const value = line.slice(5);
|
|
24
|
+
const data = value.startsWith(' ') ? value.slice(1) : value;
|
|
25
|
+
const nextDataBytes = dataBytes + (dataLines.length === 0 ? 0 : 1) + byteLength(data);
|
|
26
|
+
if (nextDataBytes > MAX_SSE_FRAME_BYTES)
|
|
27
|
+
throw new Error(PROVIDER_STREAM_INVALID);
|
|
28
|
+
dataLines.push(data);
|
|
29
|
+
dataBytes = nextDataBytes;
|
|
30
|
+
return undefined;
|
|
31
|
+
};
|
|
32
|
+
try {
|
|
33
|
+
for await (const chunk of body) {
|
|
34
|
+
buffered += decoder.decode(chunk, { stream: true });
|
|
35
|
+
if (byteLength(buffered) > MAX_SSE_BUFFER_BYTES)
|
|
36
|
+
throw new Error(PROVIDER_STREAM_INVALID);
|
|
37
|
+
let newline = buffered.indexOf('\n');
|
|
38
|
+
while (newline >= 0) {
|
|
39
|
+
const rawLine = buffered.slice(0, newline);
|
|
40
|
+
buffered = buffered.slice(newline + 1);
|
|
41
|
+
const frame = consumeLine(rawLine.endsWith('\r') ? rawLine.slice(0, -1) : rawLine);
|
|
42
|
+
if (frame !== undefined)
|
|
43
|
+
yield frame;
|
|
44
|
+
newline = buffered.indexOf('\n');
|
|
45
|
+
}
|
|
46
|
+
}
|
|
47
|
+
buffered += decoder.decode();
|
|
48
|
+
if (buffered.length > 0)
|
|
49
|
+
throw new Error(PROVIDER_STREAM_INVALID);
|
|
50
|
+
if (dataLines.length === 1 && dataLines[0] === '[DONE]') {
|
|
51
|
+
yield '[DONE]';
|
|
52
|
+
return;
|
|
53
|
+
}
|
|
54
|
+
if (dataLines.length > 0)
|
|
55
|
+
throw new Error(PROVIDER_STREAM_INVALID);
|
|
56
|
+
}
|
|
57
|
+
catch {
|
|
58
|
+
throw new Error(PROVIDER_STREAM_INVALID);
|
|
59
|
+
}
|
|
60
|
+
}
|
|
61
|
+
function byteLength(value) {
|
|
62
|
+
return encoder.encode(value).byteLength;
|
|
63
|
+
}
|
|
64
|
+
//# sourceMappingURL=sse.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sse.js","sourceRoot":"","sources":["../../src/openai-chat/sse.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,uBAAuB,GAAG,yBAAyB,CAAC;AAEjE,MAAM,oBAAoB,GAAG,SAAS,CAAC;AACvC,MAAM,mBAAmB,GAAG,SAAS,CAAC;AACtC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;AAElC,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC,wBAAwB,CAC7C,IAA+B;IAE/B,MAAM,OAAO,GAAG,IAAI,WAAW,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IAC1D,IAAI,QAAQ,GAAG,EAAE,CAAC;IAClB,IAAI,SAAS,GAAa,EAAE,CAAC;IAC7B,IAAI,SAAS,GAAG,CAAC,CAAC;IAElB,MAAM,WAAW,GAAG,CAAC,IAAY,EAAsB,EAAE;QACvD,IAAI,IAAI,KAAK,EAAE,EAAE,CAAC;YAChB,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,SAAS,CAAC;YAC7C,MAAM,IAAI,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,SAAS,GAAG,EAAE,CAAC;YACf,SAAS,GAAG,CAAC,CAAC;YACd,OAAO,IAAI,CAAC;QACd,CAAC;QACD,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;YAAE,OAAO,SAAS,CAAC;QAC3C,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QACxE,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5B,MAAM,IAAI,GAAG,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;QAC5D,MAAM,aAAa,GAAG,SAAS,GAAG,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;QACtF,IAAI,aAAa,GAAG,mBAAmB;YAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAClF,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrB,SAAS,GAAG,aAAa,CAAC;QAC1B,OAAO,SAAS,CAAC;IACnB,CAAC,CAAC;IAEF,IAAI,CAAC;QACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,EAAE,CAAC;YAC/B,QAAQ,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YACpD,IAAI,UAAU,CAAC,QAAQ,CAAC,GAAG,oBAAoB;gBAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;YAC1F,IAAI,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YACrC,OAAO,OAAO,IAAI,CAAC,EAAE,CAAC;gBACpB,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;gBAC3C,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC;gBACvC,MAAM,KAAK,GAAG,WAAW,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBACnF,IAAI,KAAK,KAAK,SAAS;oBAAE,MAAM,KAAK,CAAC;gBACrC,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YACnC,CAAC;QACH,CAAC;QAED,QAAQ,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QAC7B,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAClE,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,IAAI,SAAS,CAAC,CAAC,CAAC,KAAK,QAAQ,EAAE,CAAC;YACxD,MAAM,QAAQ,CAAC;YACf,OAAO;QACT,CAAC;QACD,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;IACrE,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;IAC3C,CAAC;AACH,CAAC;AAED,SAAS,UAAU,CAAC,KAAa;IAC/B,OAAO,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,UAAU,CAAC;AAC1C,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"adapter.d.ts","sourceRoot":"","sources":["../../src/openai-embeddings/adapter.ts"],"names":[],"mappings":"AAUA,OAAO,KAAK,EAAE,gCAAgC,EAAE,MAAM,gBAAgB,CAAC;AAKvE,wBAAgB,oCAAoC,IAAI,gCAAgC,CAWvF"}
|
|
@@ -0,0 +1,90 @@
|
|
|
1
|
+
import { parseEmbeddingCapabilityV1, validateOpenAiEmbeddingsCapability, } from '@skyold/model-protocol';
|
|
2
|
+
import { openAiEmbeddingsCodec } from './codec.js';
|
|
3
|
+
const PROTOCOL_FAMILY = 'openai-embeddings/v1';
|
|
4
|
+
export function createOpenAiEmbeddingsAdapterFactory() {
|
|
5
|
+
return {
|
|
6
|
+
id: 'openai-embeddings',
|
|
7
|
+
protocolFamily: PROTOCOL_FAMILY,
|
|
8
|
+
bind(binding) {
|
|
9
|
+
if (binding.dispatchProtocolFamily !== PROTOCOL_FAMILY) {
|
|
10
|
+
throw new Error('OPENAI_EMBEDDINGS_BINDING_MISMATCH');
|
|
11
|
+
}
|
|
12
|
+
return createAdapter(binding);
|
|
13
|
+
},
|
|
14
|
+
};
|
|
15
|
+
}
|
|
16
|
+
function createAdapter(binding) {
|
|
17
|
+
return {
|
|
18
|
+
id: 'openai-embeddings',
|
|
19
|
+
supports(capability) {
|
|
20
|
+
return supportsCapability(capability, binding.providerModel);
|
|
21
|
+
},
|
|
22
|
+
async start(input) {
|
|
23
|
+
if (!this.supports(input.capability) || !supportsTask(input.task, input.capability)) {
|
|
24
|
+
throw new Error('OPENAI_EMBEDDINGS_CAPABILITY_UNSUPPORTED');
|
|
25
|
+
}
|
|
26
|
+
const body = openAiEmbeddingsCodec.encodeProviderRequest(input.task, binding.providerModel);
|
|
27
|
+
let response;
|
|
28
|
+
try {
|
|
29
|
+
response = await binding.transport.sendOnce({
|
|
30
|
+
method: 'POST',
|
|
31
|
+
path: '/v1/embeddings',
|
|
32
|
+
headers: { 'content-type': 'application/json' },
|
|
33
|
+
body,
|
|
34
|
+
signal: input.signal,
|
|
35
|
+
});
|
|
36
|
+
}
|
|
37
|
+
catch {
|
|
38
|
+
throw new Error('PROVIDER_TRANSPORT_FAILED');
|
|
39
|
+
}
|
|
40
|
+
return openAiEmbeddingsCodec.decodeProviderStart(response, input.task, input.capability);
|
|
41
|
+
},
|
|
42
|
+
};
|
|
43
|
+
}
|
|
44
|
+
function supportsCapability(capability, providerModel) {
|
|
45
|
+
const facts = parseEmbeddingCapabilityV1(capability.embeddingCapability);
|
|
46
|
+
return (capability.active &&
|
|
47
|
+
capability.model === providerModel &&
|
|
48
|
+
capability.taskKind === 'embedding' &&
|
|
49
|
+
capability.protocolFamily === PROTOCOL_FAMILY &&
|
|
50
|
+
capability.inputModalities.length === 1 &&
|
|
51
|
+
capability.inputModalities[0] === 'text' &&
|
|
52
|
+
capability.outputModalities.length === 1 &&
|
|
53
|
+
capability.outputModalities[0] === 'vector' &&
|
|
54
|
+
capability.executionModes.length === 1 &&
|
|
55
|
+
capability.executionModes[0] === 'sync' &&
|
|
56
|
+
facts?.compatibilitySpace.provider === capability.provider &&
|
|
57
|
+
facts.compatibilitySpace.model === capability.model);
|
|
58
|
+
}
|
|
59
|
+
function supportsTask(task, capability) {
|
|
60
|
+
if (task.protocolFamily !== PROTOCOL_FAMILY ||
|
|
61
|
+
task.dispatchProtocolFamily !== PROTOCOL_FAMILY ||
|
|
62
|
+
task.taskKind !== 'embedding' ||
|
|
63
|
+
task.executionMode !== 'sync' ||
|
|
64
|
+
task.inputModalities.length !== 1 ||
|
|
65
|
+
task.inputModalities[0] !== 'text' ||
|
|
66
|
+
task.outputModalities.length !== 1 ||
|
|
67
|
+
task.outputModalities[0] !== 'vector') {
|
|
68
|
+
return false;
|
|
69
|
+
}
|
|
70
|
+
const decoded = {
|
|
71
|
+
ingressProtocolFamily: PROTOCOL_FAMILY,
|
|
72
|
+
model: task.model,
|
|
73
|
+
taskKind: task.taskKind,
|
|
74
|
+
executionMode: task.executionMode,
|
|
75
|
+
inputModalities: task.inputModalities,
|
|
76
|
+
outputModalities: task.outputModalities,
|
|
77
|
+
requiresFunctionTools: false,
|
|
78
|
+
requiresReasoning: false,
|
|
79
|
+
normalizedInput: task.input,
|
|
80
|
+
providerOptions: task.providerOptions,
|
|
81
|
+
};
|
|
82
|
+
try {
|
|
83
|
+
validateOpenAiEmbeddingsCapability(decoded, capability);
|
|
84
|
+
return true;
|
|
85
|
+
}
|
|
86
|
+
catch {
|
|
87
|
+
return false;
|
|
88
|
+
}
|
|
89
|
+
}
|
|
90
|
+
//# sourceMappingURL=adapter.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"adapter.js","sourceRoot":"","sources":["../../src/openai-embeddings/adapter.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,0BAA0B,EAC1B,kCAAkC,GAOnC,MAAM,wBAAwB,CAAC;AAEhC,OAAO,EAAE,qBAAqB,EAAE,MAAM,YAAY,CAAC;AAEnD,MAAM,eAAe,GAAG,sBAAsB,CAAC;AAE/C,MAAM,UAAU,oCAAoC;IAClD,OAAO;QACL,EAAE,EAAE,mBAAmB;QACvB,cAAc,EAAE,eAAe;QAC/B,IAAI,CAAC,OAAO;YACV,IAAI,OAAO,CAAC,sBAAsB,KAAK,eAAe,EAAE,CAAC;gBACvD,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;YACxD,CAAC;YACD,OAAO,aAAa,CAAC,OAAO,CAAC,CAAC;QAChC,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,OAAoC;IACzD,OAAO;QACL,EAAE,EAAE,mBAAmB;QACvB,QAAQ,CAAC,UAAU;YACjB,OAAO,kBAAkB,CAAC,UAAU,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC;QAC/D,CAAC;QACD,KAAK,CAAC,KAAK,CAAC,KAA8B;YACxC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC;gBACpF,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;YAC9D,CAAC;YACD,MAAM,IAAI,GAAG,qBAAqB,CAAC,qBAAqB,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC;YAC5F,IAAI,QAAQ,CAAC;YACb,IAAI,CAAC;gBACH,QAAQ,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC;oBAC1C,MAAM,EAAE,MAAM;oBACd,IAAI,EAAE,gBAAgB;oBACtB,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;oBAC/C,IAAI;oBACJ,MAAM,EAAE,KAAK,CAAC,MAAM;iBACrB,CAAC,CAAC;YACL,CAAC;YAAC,MAAM,CAAC;gBACP,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;YAC/C,CAAC;YACD,OAAO,qBAAqB,CAAC,mBAAmB,CAAC,QAAQ,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;QAC3F,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CAAC,UAA8B,EAAE,aAAqB;IAC/E,MAAM,KAAK,GAAG,0BAA0B,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC;IACzE,OAAO,CACL,UAAU,CAAC,MAAM;QACjB,UAAU,CAAC,KAAK,KAAK,aAAa;QAClC,UAAU,CAAC,QAAQ,KAAK,WAAW;QACnC,UAAU,CAAC,cAAc,KAAK,eAAe;QAC7C,UAAU,CAAC,eAAe,CAAC,MAAM,KAAK,CAAC;QACvC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,KAAK,MAAM;QACxC,UAAU,CAAC,gBAAgB,CAAC,MAAM,KAAK,CAAC;QACxC,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,KAAK,QAAQ;QAC3C,UAAU,CAAC,cAAc,CAAC,MAAM,KAAK,CAAC;QACtC,UAAU,CAAC,cAAc,CAAC,CAAC,CAAC,KAAK,MAAM;QACvC,KAAK,EAAE,kBAAkB,CAAC,QAAQ,KAAK,UAAU,CAAC,QAAQ;QAC1D,KAAK,CAAC,kBAAkB,CAAC,KAAK,KAAK,UAAU,CAAC,KAAK,CACpD,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,IAAoB,EAAE,UAA8B;IACxE,IACE,IAAI,CAAC,cAAc,KAAK,eAAe;QACvC,IAAI,CAAC,sBAAsB,KAAK,eAAe;QAC/C,IAAI,CAAC,QAAQ,KAAK,WAAW;QAC7B,IAAI,CAAC,aAAa,KAAK,MAAM;QAC7B,IAAI,CAAC,eAAe,CAAC,MAAM,KAAK,CAAC;QACjC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,KAAK,MAAM;QAClC,IAAI,CAAC,gBAAgB,CAAC,MAAM,KAAK,CAAC;QAClC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,KAAK,QAAQ,EACrC,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,OAAO,GAAkB;QAC7B,qBAAqB,EAAE,eAAe;QACtC,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,QAAQ,EAAE,IAAI,CAAC,QAAQ;QACvB,aAAa,EAAE,IAAI,CAAC,aAAa;QACjC,eAAe,EAAE,IAAI,CAAC,eAAe;QACrC,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;QACvC,qBAAqB,EAAE,KAAK;QAC5B,iBAAiB,EAAE,KAAK;QACxB,eAAe,EAAE,IAAI,CAAC,KAAK;QAC3B,eAAe,EAAE,IAAI,CAAC,eAAe;KACtC,CAAC;IACF,IAAI,CAAC;QACH,kCAAkC,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QACxD,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
import { type ProviderCapability, type ProviderExecutionStartV2, type ProviderTransportResponse, type TaskEnvelopeV2 } from '@skyold/model-protocol';
|
|
2
|
+
export declare const openAiEmbeddingsCodec: {
|
|
3
|
+
protocolFamily: string;
|
|
4
|
+
encodeProviderRequest(task: TaskEnvelopeV2, providerModel: string): Uint8Array;
|
|
5
|
+
decodeProviderStart(response: ProviderTransportResponse, task: TaskEnvelopeV2, capability: ProviderCapability): Promise<ProviderExecutionStartV2>;
|
|
6
|
+
};
|
|
7
|
+
//# sourceMappingURL=codec.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"codec.d.ts","sourceRoot":"","sources":["../../src/openai-embeddings/codec.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,kBAAkB,EACvB,KAAK,wBAAwB,EAE7B,KAAK,yBAAyB,EAC9B,KAAK,cAAc,EACpB,MAAM,wBAAwB,CAAC;AAMhC,eAAO,MAAM,qBAAqB;;gCAGJ,cAAc,iBAAiB,MAAM,GAAG,UAAU;kCAsBlE,yBAAyB,QAC7B,cAAc,cACR,kBAAkB,GAC7B,OAAO,CAAC,wBAAwB,CAAC;CAwCrC,CAAC"}
|
|
@@ -0,0 +1,120 @@
|
|
|
1
|
+
import { normalizeOpenAiEmbeddingResponse, } from '@skyold/model-protocol';
|
|
2
|
+
const PROTOCOL_FAMILY = 'openai-embeddings/v1';
|
|
3
|
+
const encoder = new TextEncoder();
|
|
4
|
+
const decoder = new TextDecoder();
|
|
5
|
+
export const openAiEmbeddingsCodec = {
|
|
6
|
+
protocolFamily: PROTOCOL_FAMILY,
|
|
7
|
+
encodeProviderRequest(task, providerModel) {
|
|
8
|
+
if (task.protocolFamily !== PROTOCOL_FAMILY ||
|
|
9
|
+
task.dispatchProtocolFamily !== PROTOCOL_FAMILY ||
|
|
10
|
+
task.taskKind !== 'embedding' ||
|
|
11
|
+
task.executionMode !== 'sync') {
|
|
12
|
+
throw new Error('OPENAI_EMBEDDINGS_TASK_UNSUPPORTED');
|
|
13
|
+
}
|
|
14
|
+
const facts = requestFacts(task);
|
|
15
|
+
return encoder.encode(JSON.stringify({
|
|
16
|
+
model: providerModel,
|
|
17
|
+
input: facts.input,
|
|
18
|
+
...(facts.dimensions === undefined ? {} : { dimensions: facts.dimensions }),
|
|
19
|
+
...(facts.encodingFormat === undefined ? {} : { encoding_format: facts.encodingFormat }),
|
|
20
|
+
}));
|
|
21
|
+
},
|
|
22
|
+
async decodeProviderStart(response, task, capability) {
|
|
23
|
+
const responseMetadata = metadata(response);
|
|
24
|
+
if (response.status < 200 || response.status >= 300) {
|
|
25
|
+
return failedStart(responseMetadata, response.status);
|
|
26
|
+
}
|
|
27
|
+
try {
|
|
28
|
+
if (!(response.body instanceof Uint8Array)) {
|
|
29
|
+
throw new Error('PROVIDER_RESPONSE_INVALID');
|
|
30
|
+
}
|
|
31
|
+
const facts = requestFacts(task);
|
|
32
|
+
const embeddingFacts = capability.embeddingCapability;
|
|
33
|
+
if (!embeddingFacts)
|
|
34
|
+
throw new Error('PROVIDER_RESPONSE_INVALID');
|
|
35
|
+
const normalized = normalizeOpenAiEmbeddingResponse(parseJson(response.body), {
|
|
36
|
+
inputCount: facts.input.length,
|
|
37
|
+
dimensions: facts.dimensions ?? embeddingFacts.nativeDimensions,
|
|
38
|
+
encodingFormat: facts.encodingFormat ?? 'float',
|
|
39
|
+
capability: embeddingFacts,
|
|
40
|
+
});
|
|
41
|
+
return {
|
|
42
|
+
kind: 'sync',
|
|
43
|
+
acceptance: { state: 'accepted' },
|
|
44
|
+
responseMetadata,
|
|
45
|
+
terminal: {
|
|
46
|
+
status: 'succeeded',
|
|
47
|
+
acceptanceState: 'accepted',
|
|
48
|
+
retryable: false,
|
|
49
|
+
result: {
|
|
50
|
+
taskKind: 'embedding',
|
|
51
|
+
output: normalized.output,
|
|
52
|
+
usage: normalized.usage,
|
|
53
|
+
},
|
|
54
|
+
usage: normalized.usage,
|
|
55
|
+
},
|
|
56
|
+
};
|
|
57
|
+
}
|
|
58
|
+
catch {
|
|
59
|
+
return invalidStart(responseMetadata);
|
|
60
|
+
}
|
|
61
|
+
},
|
|
62
|
+
};
|
|
63
|
+
function requestFacts(task) {
|
|
64
|
+
const input = task.input['input'];
|
|
65
|
+
const dimensions = task.input['dimensions'];
|
|
66
|
+
const encodingFormat = task.input['encodingFormat'];
|
|
67
|
+
if (!Array.isArray(input) ||
|
|
68
|
+
input.length === 0 ||
|
|
69
|
+
(dimensions !== undefined &&
|
|
70
|
+
(typeof dimensions !== 'number' || !Number.isSafeInteger(dimensions) || dimensions <= 0)) ||
|
|
71
|
+
(encodingFormat !== undefined && encodingFormat !== 'float' && encodingFormat !== 'base64')) {
|
|
72
|
+
throw new Error('OPENAI_EMBEDDINGS_REQUEST_INVALID');
|
|
73
|
+
}
|
|
74
|
+
return {
|
|
75
|
+
input,
|
|
76
|
+
...(dimensions === undefined ? {} : { dimensions }),
|
|
77
|
+
...(encodingFormat === undefined ? {} : { encodingFormat }),
|
|
78
|
+
};
|
|
79
|
+
}
|
|
80
|
+
function invalidStart(responseMetadata) {
|
|
81
|
+
return {
|
|
82
|
+
kind: 'sync',
|
|
83
|
+
acceptance: { state: 'accepted' },
|
|
84
|
+
responseMetadata,
|
|
85
|
+
terminal: {
|
|
86
|
+
status: 'failed',
|
|
87
|
+
acceptanceState: 'accepted',
|
|
88
|
+
failureOrigin: 'provider',
|
|
89
|
+
failureCode: 'PROVIDER_RESPONSE_INVALID',
|
|
90
|
+
retryable: false,
|
|
91
|
+
},
|
|
92
|
+
};
|
|
93
|
+
}
|
|
94
|
+
function failedStart(responseMetadata, status) {
|
|
95
|
+
const rateLimited = status === 429;
|
|
96
|
+
return {
|
|
97
|
+
kind: 'sync',
|
|
98
|
+
acceptance: { state: 'not-accepted' },
|
|
99
|
+
responseMetadata,
|
|
100
|
+
terminal: {
|
|
101
|
+
status: 'failed',
|
|
102
|
+
acceptanceState: 'not-accepted',
|
|
103
|
+
failureOrigin: 'provider',
|
|
104
|
+
failureCode: rateLimited ? 'PROVIDER_RATE_LIMITED' : 'PROVIDER_REQUEST_FAILED',
|
|
105
|
+
retryable: rateLimited || status >= 500,
|
|
106
|
+
},
|
|
107
|
+
};
|
|
108
|
+
}
|
|
109
|
+
function parseJson(body) {
|
|
110
|
+
return JSON.parse(decoder.decode(body));
|
|
111
|
+
}
|
|
112
|
+
function metadata(response) {
|
|
113
|
+
const providerRequestId = response.headers['x-request-id'] ?? response.headers['openai-request-id'];
|
|
114
|
+
return {
|
|
115
|
+
headersReceived: true,
|
|
116
|
+
statusCode: response.status,
|
|
117
|
+
...(providerRequestId ? { providerRequestId } : {}),
|
|
118
|
+
};
|
|
119
|
+
}
|
|
120
|
+
//# sourceMappingURL=codec.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"codec.js","sourceRoot":"","sources":["../../src/openai-embeddings/codec.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,gCAAgC,GAMjC,MAAM,wBAAwB,CAAC;AAEhC,MAAM,eAAe,GAAG,sBAAsB,CAAC;AAC/C,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;AAClC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;AAElC,MAAM,CAAC,MAAM,qBAAqB,GAAG;IACnC,cAAc,EAAE,eAAe;IAE/B,qBAAqB,CAAC,IAAoB,EAAE,aAAqB;QAC/D,IACE,IAAI,CAAC,cAAc,KAAK,eAAe;YACvC,IAAI,CAAC,sBAAsB,KAAK,eAAe;YAC/C,IAAI,CAAC,QAAQ,KAAK,WAAW;YAC7B,IAAI,CAAC,aAAa,KAAK,MAAM,EAC7B,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;QACxD,CAAC;QAED,MAAM,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;QACjC,OAAO,OAAO,CAAC,MAAM,CACnB,IAAI,CAAC,SAAS,CAAC;YACb,KAAK,EAAE,aAAa;YACpB,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,GAAG,CAAC,KAAK,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,CAAC;YAC3E,GAAG,CAAC,KAAK,CAAC,cAAc,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,KAAK,CAAC,cAAc,EAAE,CAAC;SACzF,CAAC,CACH,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,mBAAmB,CACvB,QAAmC,EACnC,IAAoB,EACpB,UAA8B;QAE9B,MAAM,gBAAgB,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC5C,IAAI,QAAQ,CAAC,MAAM,GAAG,GAAG,IAAI,QAAQ,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC;YACpD,OAAO,WAAW,CAAC,gBAAgB,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;QACxD,CAAC;QAED,IAAI,CAAC;YACH,IAAI,CAAC,CAAC,QAAQ,CAAC,IAAI,YAAY,UAAU,CAAC,EAAE,CAAC;gBAC3C,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;YAC/C,CAAC;YACD,MAAM,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;YACjC,MAAM,cAAc,GAAG,UAAU,CAAC,mBAAmB,CAAC;YACtD,IAAI,CAAC,cAAc;gBAAE,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;YAClE,MAAM,UAAU,GAAG,gCAAgC,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE;gBAC5E,UAAU,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM;gBAC9B,UAAU,EAAE,KAAK,CAAC,UAAU,IAAI,cAAc,CAAC,gBAAgB;gBAC/D,cAAc,EAAE,KAAK,CAAC,cAAc,IAAI,OAAO;gBAC/C,UAAU,EAAE,cAAc;aAC3B,CAAC,CAAC;YAEH,OAAO;gBACL,IAAI,EAAE,MAAM;gBACZ,UAAU,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE;gBACjC,gBAAgB;gBAChB,QAAQ,EAAE;oBACR,MAAM,EAAE,WAAW;oBACnB,eAAe,EAAE,UAAU;oBAC3B,SAAS,EAAE,KAAK;oBAChB,MAAM,EAAE;wBACN,QAAQ,EAAE,WAAW;wBACrB,MAAM,EAAE,UAAU,CAAC,MAAM;wBACzB,KAAK,EAAE,UAAU,CAAC,KAAK;qBACxB;oBACD,KAAK,EAAE,UAAU,CAAC,KAAK;iBACxB;aACF,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,YAAY,CAAC,gBAAgB,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;CACF,CAAC;AAQF,SAAS,YAAY,CAAC,IAAoB;IACxC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAClC,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IAC5C,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;IACpD,IACE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QACrB,KAAK,CAAC,MAAM,KAAK,CAAC;QAClB,CAAC,UAAU,KAAK,SAAS;YACvB,CAAC,OAAO,UAAU,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,UAAU,CAAC,IAAI,UAAU,IAAI,CAAC,CAAC,CAAC;QAC3F,CAAC,cAAc,KAAK,SAAS,IAAI,cAAc,KAAK,OAAO,IAAI,cAAc,KAAK,QAAQ,CAAC,EAC3F,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;IACvD,CAAC;IACD,OAAO;QACL,KAAK;QACL,GAAG,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC;QACnD,GAAG,CAAC,cAAc,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,CAAC;KAC5D,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,gBAA0C;IAC9D,OAAO;QACL,IAAI,EAAE,MAAM;QACZ,UAAU,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE;QACjC,gBAAgB;QAChB,QAAQ,EAAE;YACR,MAAM,EAAE,QAAQ;YAChB,eAAe,EAAE,UAAU;YAC3B,aAAa,EAAE,UAAU;YACzB,WAAW,EAAE,2BAA2B;YACxC,SAAS,EAAE,KAAK;SACjB;KACF,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAClB,gBAA0C,EAC1C,MAAc;IAEd,MAAM,WAAW,GAAG,MAAM,KAAK,GAAG,CAAC;IACnC,OAAO;QACL,IAAI,EAAE,MAAM;QACZ,UAAU,EAAE,EAAE,KAAK,EAAE,cAAc,EAAE;QACrC,gBAAgB;QAChB,QAAQ,EAAE;YACR,MAAM,EAAE,QAAQ;YAChB,eAAe,EAAE,cAAc;YAC/B,aAAa,EAAE,UAAU;YACzB,WAAW,EAAE,WAAW,CAAC,CAAC,CAAC,uBAAuB,CAAC,CAAC,CAAC,yBAAyB;YAC9E,SAAS,EAAE,WAAW,IAAI,MAAM,IAAI,GAAG;SACxC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,SAAS,CAAC,IAAgB;IACjC,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;AAC1C,CAAC;AAED,SAAS,QAAQ,CAAC,QAAmC;IACnD,MAAM,iBAAiB,GACrB,QAAQ,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,QAAQ,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;IAC5E,OAAO;QACL,eAAe,EAAE,IAAI;QACrB,UAAU,EAAE,QAAQ,CAAC,MAAM;QAC3B,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACpD,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,4 @@
|
|
|
1
|
+
import type { RegisteredProviderAdapterFactory } from '../registry.js';
|
|
2
|
+
import { type OpenAiImagesEditDiagnostics } from './codec.js';
|
|
3
|
+
export declare function createOpenAiImagesAdapterFactory(diagnostics?: OpenAiImagesEditDiagnostics): RegisteredProviderAdapterFactory;
|
|
4
|
+
//# sourceMappingURL=adapter.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"adapter.d.ts","sourceRoot":"","sources":["../../src/openai-images/adapter.ts"],"names":[],"mappings":"AAQA,OAAO,KAAK,EAAE,gCAAgC,EAAE,MAAM,gBAAgB,CAAC;AACvE,OAAO,EAAqB,KAAK,2BAA2B,EAAE,MAAM,YAAY,CAAC;AAIjF,wBAAgB,gCAAgC,CAC9C,WAAW,CAAC,EAAE,2BAA2B,GACxC,gCAAgC,CAWlC"}
|
|
@@ -0,0 +1,103 @@
|
|
|
1
|
+
import { parseImageGenerationCapabilityV1, parseImageTaskCapabilityDeclarationV2, } from '@skyold/model-protocol';
|
|
2
|
+
import { openAiImagesCodec } from './codec.js';
|
|
3
|
+
const PROTOCOL_FAMILY = 'openai-images/v1';
|
|
4
|
+
export function createOpenAiImagesAdapterFactory(diagnostics) {
|
|
5
|
+
return {
|
|
6
|
+
id: 'openai-images',
|
|
7
|
+
protocolFamily: PROTOCOL_FAMILY,
|
|
8
|
+
bind(binding) {
|
|
9
|
+
if (binding.dispatchProtocolFamily !== PROTOCOL_FAMILY) {
|
|
10
|
+
throw new Error('OPENAI_IMAGES_BINDING_MISMATCH');
|
|
11
|
+
}
|
|
12
|
+
return createAdapter(binding, diagnostics);
|
|
13
|
+
},
|
|
14
|
+
};
|
|
15
|
+
}
|
|
16
|
+
function createAdapter(binding, diagnostics) {
|
|
17
|
+
return {
|
|
18
|
+
id: 'openai-images',
|
|
19
|
+
supports(capability) {
|
|
20
|
+
return supportsCapability(capability, binding.providerModel);
|
|
21
|
+
},
|
|
22
|
+
async start(input) {
|
|
23
|
+
if (!this.supports(input.capability) ||
|
|
24
|
+
!openAiImagesCodec.supportsTask(input.task, input.capability)) {
|
|
25
|
+
throw new Error('OPENAI_IMAGES_CAPABILITY_UNSUPPORTED');
|
|
26
|
+
}
|
|
27
|
+
const operation = input.task.input['operation'];
|
|
28
|
+
const editRequest = operation === 'edit'
|
|
29
|
+
? await openAiImagesCodec.encodeEditProviderRequest(input.task, binding.providerModel, input.capability, input.preparedMedia, input.signal, diagnostics)
|
|
30
|
+
: undefined;
|
|
31
|
+
const body = editRequest?.body ??
|
|
32
|
+
openAiImagesCodec.encodeProviderRequest(input.task, binding.providerModel, input.capability);
|
|
33
|
+
const responseBodyPolicy = openAiImagesCodec.responseBodyPolicy(input.task, input.capability);
|
|
34
|
+
if (operation === 'edit')
|
|
35
|
+
diagnostics?.onBeforeSend?.();
|
|
36
|
+
if (input.signal.aborted)
|
|
37
|
+
throw new Error('OPENAI_IMAGES_MEDIA_ABORTED');
|
|
38
|
+
let response;
|
|
39
|
+
try {
|
|
40
|
+
response = await binding.transport.sendOnce({
|
|
41
|
+
method: 'POST',
|
|
42
|
+
path: operation === 'edit' ? '/v1/images/edits' : '/v1/images/generations',
|
|
43
|
+
headers: {
|
|
44
|
+
'content-type': operation === 'edit'
|
|
45
|
+
? `multipart/form-data; boundary=${editRequest.boundary}`
|
|
46
|
+
: 'application/json',
|
|
47
|
+
},
|
|
48
|
+
body,
|
|
49
|
+
signal: input.signal,
|
|
50
|
+
responseBodyPolicy,
|
|
51
|
+
});
|
|
52
|
+
}
|
|
53
|
+
catch {
|
|
54
|
+
return {
|
|
55
|
+
kind: 'sync',
|
|
56
|
+
acceptance: { state: 'unknown' },
|
|
57
|
+
responseMetadata: { headersReceived: false },
|
|
58
|
+
terminal: {
|
|
59
|
+
status: 'failed',
|
|
60
|
+
acceptanceState: 'unknown',
|
|
61
|
+
failureOrigin: 'transport',
|
|
62
|
+
failureCode: 'PROVIDER_ACCEPTANCE_UNKNOWN',
|
|
63
|
+
retryable: false,
|
|
64
|
+
},
|
|
65
|
+
};
|
|
66
|
+
}
|
|
67
|
+
return openAiImagesCodec.decodeProviderStart(response, input.task, input.capability);
|
|
68
|
+
},
|
|
69
|
+
};
|
|
70
|
+
}
|
|
71
|
+
function supportsCapability(capability, providerModel) {
|
|
72
|
+
const facts = parseImageGenerationCapabilityV1(capability.imageGenerationCapability);
|
|
73
|
+
const isV1 = capability.operations === undefined &&
|
|
74
|
+
capability.imageEditingCapability === undefined &&
|
|
75
|
+
capability.inputModalities.length === 1 &&
|
|
76
|
+
capability.inputModalities[0] === 'text';
|
|
77
|
+
const isV2 = parseImageTaskCapabilityDeclarationV2({
|
|
78
|
+
schemaVersion: 2,
|
|
79
|
+
taskKind: capability.taskKind,
|
|
80
|
+
protocolFamily: capability.protocolFamily,
|
|
81
|
+
operations: capability.operations,
|
|
82
|
+
inputModalities: capability.inputModalities,
|
|
83
|
+
outputModalities: capability.outputModalities,
|
|
84
|
+
executionModes: capability.executionModes,
|
|
85
|
+
supportsCallback: false,
|
|
86
|
+
supportsCancel: false,
|
|
87
|
+
imageGenerationCapability: capability.imageGenerationCapability,
|
|
88
|
+
...(capability.imageEditingCapability === undefined
|
|
89
|
+
? {}
|
|
90
|
+
: { imageEditingCapability: capability.imageEditingCapability }),
|
|
91
|
+
}) !== null;
|
|
92
|
+
return (capability.active &&
|
|
93
|
+
capability.model === providerModel &&
|
|
94
|
+
capability.taskKind === 'image_generation' &&
|
|
95
|
+
capability.protocolFamily === PROTOCOL_FAMILY &&
|
|
96
|
+
(isV1 || isV2) &&
|
|
97
|
+
capability.outputModalities.length === 1 &&
|
|
98
|
+
capability.outputModalities[0] === 'image' &&
|
|
99
|
+
capability.executionModes.length === 1 &&
|
|
100
|
+
capability.executionModes[0] === 'sync' &&
|
|
101
|
+
facts?.supportedPolicies.includes('transparent') === true);
|
|
102
|
+
}
|
|
103
|
+
//# sourceMappingURL=adapter.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"adapter.js","sourceRoot":"","sources":["../../src/openai-images/adapter.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,gCAAgC,EAChC,qCAAqC,GAKtC,MAAM,wBAAwB,CAAC;AAEhC,OAAO,EAAE,iBAAiB,EAAoC,MAAM,YAAY,CAAC;AAEjF,MAAM,eAAe,GAAG,kBAAkB,CAAC;AAE3C,MAAM,UAAU,gCAAgC,CAC9C,WAAyC;IAEzC,OAAO;QACL,EAAE,EAAE,eAAe;QACnB,cAAc,EAAE,eAAe;QAC/B,IAAI,CAAC,OAAO;YACV,IAAI,OAAO,CAAC,sBAAsB,KAAK,eAAe,EAAE,CAAC;gBACvD,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;YACpD,CAAC;YACD,OAAO,aAAa,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QAC7C,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CACpB,OAAoC,EACpC,WAAoD;IAEpD,OAAO;QACL,EAAE,EAAE,eAAe;QACnB,QAAQ,CAAC,UAAU;YACjB,OAAO,kBAAkB,CAAC,UAAU,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC;QAC/D,CAAC;QACD,KAAK,CAAC,KAAK,CAAC,KAA8B;YACxC,IACE,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC;gBAChC,CAAC,iBAAiB,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,UAAU,CAAC,EAC7D,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;YAC1D,CAAC;YACD,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAChD,MAAM,WAAW,GACf,SAAS,KAAK,MAAM;gBAClB,CAAC,CAAC,MAAM,iBAAiB,CAAC,yBAAyB,CAC/C,KAAK,CAAC,IAAI,EACV,OAAO,CAAC,aAAa,EACrB,KAAK,CAAC,UAAU,EAChB,KAAK,CAAC,aAAa,EACnB,KAAK,CAAC,MAAM,EACZ,WAAW,CACZ;gBACH,CAAC,CAAC,SAAS,CAAC;YAChB,MAAM,IAAI,GACR,WAAW,EAAE,IAAI;gBACjB,iBAAiB,CAAC,qBAAqB,CACrC,KAAK,CAAC,IAAI,EACV,OAAO,CAAC,aAAa,EACrB,KAAK,CAAC,UAAU,CACjB,CAAC;YACJ,MAAM,kBAAkB,GAAG,iBAAiB,CAAC,kBAAkB,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;YAC9F,IAAI,SAAS,KAAK,MAAM;gBAAE,WAAW,EAAE,YAAY,EAAE,EAAE,CAAC;YACxD,IAAI,KAAK,CAAC,MAAM,CAAC,OAAO;gBAAE,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;YACzE,IAAI,QAAQ,CAAC;YACb,IAAI,CAAC;gBACH,QAAQ,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC;oBAC1C,MAAM,EAAE,MAAM;oBACd,IAAI,EAAE,SAAS,KAAK,MAAM,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,wBAAwB;oBAC1E,OAAO,EAAE;wBACP,cAAc,EACZ,SAAS,KAAK,MAAM;4BAClB,CAAC,CAAC,iCAAiC,WAAY,CAAC,QAAQ,EAAE;4BAC1D,CAAC,CAAC,kBAAkB;qBACzB;oBACD,IAAI;oBACJ,MAAM,EAAE,KAAK,CAAC,MAAM;oBACpB,kBAAkB;iBACnB,CAAC,CAAC;YACL,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO;oBACL,IAAI,EAAE,MAAM;oBACZ,UAAU,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE;oBAChC,gBAAgB,EAAE,EAAE,eAAe,EAAE,KAAK,EAAE;oBAC5C,QAAQ,EAAE;wBACR,MAAM,EAAE,QAAQ;wBAChB,eAAe,EAAE,SAAS;wBAC1B,aAAa,EAAE,WAAW;wBAC1B,WAAW,EAAE,6BAA6B;wBAC1C,SAAS,EAAE,KAAK;qBACjB;iBACF,CAAC;YACJ,CAAC;YACD,OAAO,iBAAiB,CAAC,mBAAmB,CAAC,QAAQ,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;QACvF,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CACzB,UAAuC,EACvC,aAAqB;IAErB,MAAM,KAAK,GAAG,gCAAgC,CAAC,UAAU,CAAC,yBAAyB,CAAC,CAAC;IACrF,MAAM,IAAI,GACR,UAAU,CAAC,UAAU,KAAK,SAAS;QACnC,UAAU,CAAC,sBAAsB,KAAK,SAAS;QAC/C,UAAU,CAAC,eAAe,CAAC,MAAM,KAAK,CAAC;QACvC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,KAAK,MAAM,CAAC;IAC3C,MAAM,IAAI,GACR,qCAAqC,CAAC;QACpC,aAAa,EAAE,CAAC;QAChB,QAAQ,EAAE,UAAU,CAAC,QAAQ;QAC7B,cAAc,EAAE,UAAU,CAAC,cAAc;QACzC,UAAU,EAAE,UAAU,CAAC,UAAU;QACjC,eAAe,EAAE,UAAU,CAAC,eAAe;QAC3C,gBAAgB,EAAE,UAAU,CAAC,gBAAgB;QAC7C,cAAc,EAAE,UAAU,CAAC,cAAc;QACzC,gBAAgB,EAAE,KAAK;QACvB,cAAc,EAAE,KAAK;QACrB,yBAAyB,EAAE,UAAU,CAAC,yBAAyB;QAC/D,GAAG,CAAC,UAAU,CAAC,sBAAsB,KAAK,SAAS;YACjD,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,EAAE,sBAAsB,EAAE,UAAU,CAAC,sBAAsB,EAAE,CAAC;KACnE,CAAC,KAAK,IAAI,CAAC;IACd,OAAO,CACL,UAAU,CAAC,MAAM;QACjB,UAAU,CAAC,KAAK,KAAK,aAAa;QAClC,UAAU,CAAC,QAAQ,KAAK,kBAAkB;QAC1C,UAAU,CAAC,cAAc,KAAK,eAAe;QAC7C,CAAC,IAAI,IAAI,IAAI,CAAC;QACd,UAAU,CAAC,gBAAgB,CAAC,MAAM,KAAK,CAAC;QACxC,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,KAAK,OAAO;QAC1C,UAAU,CAAC,cAAc,CAAC,MAAM,KAAK,CAAC;QACtC,UAAU,CAAC,cAAc,CAAC,CAAC,CAAC,KAAK,MAAM;QACvC,KAAK,EAAE,iBAAiB,CAAC,QAAQ,CAAC,aAAa,CAAC,KAAK,IAAI,CAC1D,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,24 @@
|
|
|
1
|
+
import { type AdapterProviderCapabilityV2, type PreparedMediaExecutionInputV2, type ProviderExecutionStartV2, type ProviderTransportResponse, type TaskEnvelopeV2 } from '@skyold/model-protocol';
|
|
2
|
+
export interface OpenAiImagesEditDiagnostics {
|
|
3
|
+
readonly onAfterFinalMediaChunk?: () => void;
|
|
4
|
+
readonly onAfterMediaDigest?: () => void;
|
|
5
|
+
readonly onAfterBoundaryDigest?: () => void;
|
|
6
|
+
readonly onBeforeSend?: () => void;
|
|
7
|
+
}
|
|
8
|
+
export declare const openAiImagesCodec: {
|
|
9
|
+
protocolFamily: string;
|
|
10
|
+
supportsTask(task: TaskEnvelopeV2, capability: AdapterProviderCapabilityV2): boolean;
|
|
11
|
+
encodeProviderRequest(task: TaskEnvelopeV2, providerModel: string, capability: AdapterProviderCapabilityV2): Uint8Array;
|
|
12
|
+
encodeEditProviderRequest(task: TaskEnvelopeV2, providerModel: string, capability: AdapterProviderCapabilityV2, preparedMedia: PreparedMediaExecutionInputV2 | undefined, signal: AbortSignal, diagnostics?: OpenAiImagesEditDiagnostics): Promise<{
|
|
13
|
+
readonly body: Uint8Array;
|
|
14
|
+
readonly boundary: string;
|
|
15
|
+
}>;
|
|
16
|
+
responseBodyPolicy(task: TaskEnvelopeV2, capability: AdapterProviderCapabilityV2): {
|
|
17
|
+
mode: "bounded-success-stream";
|
|
18
|
+
maxBytes: number;
|
|
19
|
+
};
|
|
20
|
+
decodeProviderStart(response: ProviderTransportResponse, task: TaskEnvelopeV2, capability: AdapterProviderCapabilityV2, diagnostics?: {
|
|
21
|
+
readonly onParserRetainedItems?: (count: number) => void;
|
|
22
|
+
}): Promise<ProviderExecutionStartV2>;
|
|
23
|
+
};
|
|
24
|
+
//# sourceMappingURL=codec.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"codec.d.ts","sourceRoot":"","sources":["../../src/openai-images/codec.ts"],"names":[],"mappings":"AAAA,OAAO,EAOL,KAAK,2BAA2B,EAShC,KAAK,6BAA6B,EAElC,KAAK,wBAAwB,EAE7B,KAAK,yBAAyB,EAC9B,KAAK,cAAc,EACpB,MAAM,wBAAwB,CAAC;AAwBhC,MAAM,WAAW,2BAA2B;IAC1C,QAAQ,CAAC,sBAAsB,CAAC,EAAE,MAAM,IAAI,CAAC;IAC7C,QAAQ,CAAC,kBAAkB,CAAC,EAAE,MAAM,IAAI,CAAC;IACzC,QAAQ,CAAC,qBAAqB,CAAC,EAAE,MAAM,IAAI,CAAC;IAC5C,QAAQ,CAAC,YAAY,CAAC,EAAE,MAAM,IAAI,CAAC;CACpC;AAED,eAAO,MAAM,iBAAiB;;uBAGT,cAAc,cAAc,2BAA2B,GAAG,OAAO;gCAU5E,cAAc,iBACL,MAAM,cACT,2BAA2B,GACtC,UAAU;oCAmBL,cAAc,iBACL,MAAM,cACT,2BAA2B,iBACxB,6BAA6B,GAAG,SAAS,UAChD,WAAW,gBACL,2BAA2B,GACxC,OAAO,CAAC;QAAE,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAC;QAAC,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC;6BAoC3C,cAAc,cAAc,2BAA2B;;;;kCAQpE,yBAAyB,QAC7B,cAAc,cACR,2BAA2B,gBACzB;QACZ,QAAQ,CAAC,qBAAqB,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;KAC1D,GACA,OAAO,CAAC,wBAAwB,CAAC;CA8CrC,CAAC"}
|