@holokai/holo-sdk 1.4.1
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 +264 -0
- package/dist/client/builder.d.ts +82 -0
- package/dist/client/builder.d.ts.map +1 -0
- package/dist/client/builder.js +207 -0
- package/dist/client/builder.js.map +1 -0
- package/dist/client/client.d.ts +148 -0
- package/dist/client/client.d.ts.map +1 -0
- package/dist/client/client.js +309 -0
- package/dist/client/client.js.map +1 -0
- package/dist/client/errors.d.ts +20 -0
- package/dist/client/errors.d.ts.map +1 -0
- package/dist/client/errors.js +36 -0
- package/dist/client/errors.js.map +1 -0
- package/dist/client/index.d.ts +11 -0
- package/dist/client/index.d.ts.map +1 -0
- package/dist/client/index.js +8 -0
- package/dist/client/index.js.map +1 -0
- package/dist/client/merge.d.ts +28 -0
- package/dist/client/merge.d.ts.map +1 -0
- package/dist/client/merge.js +185 -0
- package/dist/client/merge.js.map +1 -0
- package/dist/client/output.d.ts +26 -0
- package/dist/client/output.d.ts.map +1 -0
- package/dist/client/output.js +75 -0
- package/dist/client/output.js.map +1 -0
- package/dist/client/runner.d.ts +92 -0
- package/dist/client/runner.d.ts.map +1 -0
- package/dist/client/runner.js +121 -0
- package/dist/client/runner.js.map +1 -0
- package/dist/client/sse.d.ts +3 -0
- package/dist/client/sse.d.ts.map +1 -0
- package/dist/client/sse.js +72 -0
- package/dist/client/sse.js.map +1 -0
- package/dist/client/stream.d.ts +60 -0
- package/dist/client/stream.d.ts.map +1 -0
- package/dist/client/stream.js +137 -0
- package/dist/client/stream.js.map +1 -0
- package/dist/client/transport.d.ts +42 -0
- package/dist/client/transport.d.ts.map +1 -0
- package/dist/client/transport.js +129 -0
- package/dist/client/transport.js.map +1 -0
- package/dist/client/types.d.ts +41 -0
- package/dist/client/types.d.ts.map +1 -0
- package/dist/client/types.js +2 -0
- package/dist/client/types.js.map +1 -0
- package/dist/core/async-event-queue.d.ts +11 -0
- package/dist/core/async-event-queue.d.ts.map +1 -0
- package/dist/core/async-event-queue.js +38 -0
- package/dist/core/async-event-queue.js.map +1 -0
- package/dist/core/binary.d.ts +3 -0
- package/dist/core/binary.d.ts.map +1 -0
- package/dist/core/binary.js +11 -0
- package/dist/core/binary.js.map +1 -0
- package/dist/core/errors.d.ts +11 -0
- package/dist/core/errors.d.ts.map +1 -0
- package/dist/core/errors.js +24 -0
- package/dist/core/errors.js.map +1 -0
- package/dist/core/headers.d.ts +3 -0
- package/dist/core/headers.d.ts.map +1 -0
- package/dist/core/headers.js +42 -0
- package/dist/core/headers.js.map +1 -0
- package/dist/core/index.d.ts +13 -0
- package/dist/core/index.d.ts.map +1 -0
- package/dist/core/index.js +13 -0
- package/dist/core/index.js.map +1 -0
- package/dist/core/logger.d.ts +37 -0
- package/dist/core/logger.d.ts.map +1 -0
- package/dist/core/logger.js +73 -0
- package/dist/core/logger.js.map +1 -0
- package/dist/core/pick-defined.d.ts +9 -0
- package/dist/core/pick-defined.d.ts.map +1 -0
- package/dist/core/pick-defined.js +79 -0
- package/dist/core/pick-defined.js.map +1 -0
- package/dist/core/safe-parse.d.ts +2 -0
- package/dist/core/safe-parse.d.ts.map +1 -0
- package/dist/core/safe-parse.js +11 -0
- package/dist/core/safe-parse.js.map +1 -0
- package/dist/core/sanitize.object.d.ts +2 -0
- package/dist/core/sanitize.object.d.ts.map +1 -0
- package/dist/core/sanitize.object.js +10 -0
- package/dist/core/sanitize.object.js.map +1 -0
- package/dist/core/stable-id.d.ts +2 -0
- package/dist/core/stable-id.d.ts.map +1 -0
- package/dist/core/stable-id.js +11 -0
- package/dist/core/stable-id.js.map +1 -0
- package/dist/core/stringify-any.d.ts +2 -0
- package/dist/core/stringify-any.d.ts.map +1 -0
- package/dist/core/stringify-any.js +4 -0
- package/dist/core/stringify-any.js.map +1 -0
- package/dist/core/time-utils.d.ts +2 -0
- package/dist/core/time-utils.d.ts.map +1 -0
- package/dist/core/time-utils.js +2 -0
- package/dist/core/time-utils.js.map +1 -0
- package/dist/core/tokenizer.d.ts +2 -0
- package/dist/core/tokenizer.d.ts.map +1 -0
- package/dist/core/tokenizer.js +12 -0
- package/dist/core/tokenizer.js.map +1 -0
- package/dist/holo/content.d.ts +9 -0
- package/dist/holo/content.d.ts.map +1 -0
- package/dist/holo/content.js +22 -0
- package/dist/holo/content.js.map +1 -0
- package/dist/holo/factories/index.d.ts +2 -0
- package/dist/holo/factories/index.d.ts.map +1 -0
- package/dist/holo/factories/index.js +2 -0
- package/dist/holo/factories/index.js.map +1 -0
- package/dist/holo/factories/request-factory.d.ts +3 -0
- package/dist/holo/factories/request-factory.d.ts.map +1 -0
- package/dist/holo/factories/request-factory.js +8 -0
- package/dist/holo/factories/request-factory.js.map +1 -0
- package/dist/holo/index.d.ts +3 -0
- package/dist/holo/index.d.ts.map +1 -0
- package/dist/holo/index.js +3 -0
- package/dist/holo/index.js.map +1 -0
- package/dist/index.d.ts +5 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +5 -0
- package/dist/index.js.map +1 -0
- package/dist/matching/index.d.ts +3 -0
- package/dist/matching/index.d.ts.map +1 -0
- package/dist/matching/index.js +2 -0
- package/dist/matching/index.js.map +1 -0
- package/dist/matching/normalize.d.ts +19 -0
- package/dist/matching/normalize.d.ts.map +1 -0
- package/dist/matching/normalize.js +169 -0
- package/dist/matching/normalize.js.map +1 -0
- package/dist/notification/index.d.ts +4 -0
- package/dist/notification/index.d.ts.map +1 -0
- package/dist/notification/index.js +4 -0
- package/dist/notification/index.js.map +1 -0
- package/dist/notification/service.d.ts +8 -0
- package/dist/notification/service.d.ts.map +1 -0
- package/dist/notification/service.js +16 -0
- package/dist/notification/service.js.map +1 -0
- package/dist/notification/store.d.ts +2 -0
- package/dist/notification/store.d.ts.map +1 -0
- package/dist/notification/store.js +2 -0
- package/dist/notification/store.js.map +1 -0
- package/dist/notification/types.d.ts +11 -0
- package/dist/notification/types.d.ts.map +1 -0
- package/dist/notification/types.js +63 -0
- package/dist/notification/types.js.map +1 -0
- package/dist/plugin/base.d.ts +29 -0
- package/dist/plugin/base.d.ts.map +1 -0
- package/dist/plugin/base.js +105 -0
- package/dist/plugin/base.js.map +1 -0
- package/dist/plugin/errors.d.ts +9 -0
- package/dist/plugin/errors.d.ts.map +1 -0
- package/dist/plugin/errors.js +15 -0
- package/dist/plugin/errors.js.map +1 -0
- package/dist/plugin/index.d.ts +5 -0
- package/dist/plugin/index.d.ts.map +1 -0
- package/dist/plugin/index.js +5 -0
- package/dist/plugin/index.js.map +1 -0
- package/dist/plugin/loader.d.ts +14 -0
- package/dist/plugin/loader.d.ts.map +1 -0
- package/dist/plugin/loader.js +67 -0
- package/dist/plugin/loader.js.map +1 -0
- package/dist/plugin/normalize.d.ts +3 -0
- package/dist/plugin/normalize.d.ts.map +1 -0
- package/dist/plugin/normalize.js +160 -0
- package/dist/plugin/normalize.js.map +1 -0
- package/dist/provider/auditor/base.d.ts +47 -0
- package/dist/provider/auditor/base.d.ts.map +1 -0
- package/dist/provider/auditor/base.js +254 -0
- package/dist/provider/auditor/base.js.map +1 -0
- package/dist/provider/auditor/index.d.ts +2 -0
- package/dist/provider/auditor/index.d.ts.map +1 -0
- package/dist/provider/auditor/index.js +2 -0
- package/dist/provider/auditor/index.js.map +1 -0
- package/dist/provider/base.d.ts +36 -0
- package/dist/provider/base.d.ts.map +1 -0
- package/dist/provider/base.js +185 -0
- package/dist/provider/base.js.map +1 -0
- package/dist/provider/event-processor.d.ts +13 -0
- package/dist/provider/event-processor.d.ts.map +1 -0
- package/dist/provider/event-processor.js +22 -0
- package/dist/provider/event-processor.js.map +1 -0
- package/dist/provider/index.d.ts +7 -0
- package/dist/provider/index.d.ts.map +1 -0
- package/dist/provider/index.js +7 -0
- package/dist/provider/index.js.map +1 -0
- package/dist/provider/translator/base.d.ts +15 -0
- package/dist/provider/translator/base.d.ts.map +1 -0
- package/dist/provider/translator/base.js +33 -0
- package/dist/provider/translator/base.js.map +1 -0
- package/dist/provider/translator/index.d.ts +3 -0
- package/dist/provider/translator/index.d.ts.map +1 -0
- package/dist/provider/translator/index.js +3 -0
- package/dist/provider/translator/index.js.map +1 -0
- package/dist/provider/translator/stream.d.ts +24 -0
- package/dist/provider/translator/stream.d.ts.map +1 -0
- package/dist/provider/translator/stream.js +50 -0
- package/dist/provider/translator/stream.js.map +1 -0
- package/dist/provider/wire/base.d.ts +39 -0
- package/dist/provider/wire/base.d.ts.map +1 -0
- package/dist/provider/wire/base.js +106 -0
- package/dist/provider/wire/base.js.map +1 -0
- package/dist/provider/wire/holo.wire.adapter.d.ts +33 -0
- package/dist/provider/wire/holo.wire.adapter.d.ts.map +1 -0
- package/dist/provider/wire/holo.wire.adapter.js +220 -0
- package/dist/provider/wire/holo.wire.adapter.js.map +1 -0
- package/dist/provider/wire/index.d.ts +3 -0
- package/dist/provider/wire/index.d.ts.map +1 -0
- package/dist/provider/wire/index.js +3 -0
- package/dist/provider/wire/index.js.map +1 -0
- package/dist/provider/worker-request-factory.d.ts +23 -0
- package/dist/provider/worker-request-factory.d.ts.map +1 -0
- package/dist/provider/worker-request-factory.js +38 -0
- package/dist/provider/worker-request-factory.js.map +1 -0
- package/package.json +106 -0
|
@@ -0,0 +1,185 @@
|
|
|
1
|
+
import { ProtocolCapability } from "@holokai/holo-types/entities";
|
|
2
|
+
import { AsyncEventQueue, ClassLogger, countTokens, pickDefined } from "../core";
|
|
3
|
+
export class BaseProvider extends ClassLogger {
|
|
4
|
+
id;
|
|
5
|
+
name;
|
|
6
|
+
plugin;
|
|
7
|
+
_config;
|
|
8
|
+
auditor;
|
|
9
|
+
translator;
|
|
10
|
+
responseFactory;
|
|
11
|
+
models = {};
|
|
12
|
+
client;
|
|
13
|
+
constructor(id, name, plugin, _config) {
|
|
14
|
+
super();
|
|
15
|
+
this.id = id;
|
|
16
|
+
this.name = name;
|
|
17
|
+
this.plugin = plugin;
|
|
18
|
+
this._config = _config;
|
|
19
|
+
this.client = this.createClient();
|
|
20
|
+
this.auditor = this.createAuditor();
|
|
21
|
+
this.translator = this.createTranslator();
|
|
22
|
+
this.responseFactory = this.createResponseFactory();
|
|
23
|
+
}
|
|
24
|
+
async auditRequest(workerRequest) {
|
|
25
|
+
return this.auditor.auditRequest(workerRequest);
|
|
26
|
+
}
|
|
27
|
+
async auditResponse(workerEnvelope, providerEvent) {
|
|
28
|
+
return this.auditor.auditResponse(workerEnvelope, providerEvent);
|
|
29
|
+
}
|
|
30
|
+
async processWorkerRequest(request, _opts) {
|
|
31
|
+
const q = new AsyncEventQueue();
|
|
32
|
+
const { requestId, protocol, payload, httpRequestDetails } = request;
|
|
33
|
+
const requestPayload = request.isHoloNative
|
|
34
|
+
? await this.translatePayload(protocol?.capability, payload, protocol?.name)
|
|
35
|
+
: payload;
|
|
36
|
+
let fullText = "";
|
|
37
|
+
const metrics = {
|
|
38
|
+
startTime: Date.now(),
|
|
39
|
+
timeToFirstToken: 0,
|
|
40
|
+
inputTokens: 0,
|
|
41
|
+
outputTokens: 0,
|
|
42
|
+
totalTokens: 0,
|
|
43
|
+
totalProcessingTime: 0,
|
|
44
|
+
};
|
|
45
|
+
const push = this.createEventPusher(q, requestId);
|
|
46
|
+
const query = { ...(httpRequestDetails?.query ?? {}) };
|
|
47
|
+
if (request.isHoloNative && request.isStreaming) {
|
|
48
|
+
query.alt = 'sse';
|
|
49
|
+
query.stream = 'true';
|
|
50
|
+
}
|
|
51
|
+
const ctx = pickDefined({
|
|
52
|
+
protocol,
|
|
53
|
+
headers: httpRequestDetails?.headers,
|
|
54
|
+
query,
|
|
55
|
+
emitStreamEvent: (event) => push({ type: "stream_event", event }),
|
|
56
|
+
emitTextDelta: (text) => {
|
|
57
|
+
if (metrics.timeToFirstToken === 0) {
|
|
58
|
+
metrics.firstTime = Date.now();
|
|
59
|
+
metrics.timeToFirstToken = metrics.firstTime - metrics.startTime;
|
|
60
|
+
}
|
|
61
|
+
if (text != null) {
|
|
62
|
+
fullText += text;
|
|
63
|
+
push({ type: "text_delta", text });
|
|
64
|
+
}
|
|
65
|
+
},
|
|
66
|
+
emitToolCallDelta: (index, delta) => push({ type: "tool_call_delta", index, delta }),
|
|
67
|
+
});
|
|
68
|
+
let runner;
|
|
69
|
+
try {
|
|
70
|
+
runner = await this.createRequestRunner(requestPayload, ctx);
|
|
71
|
+
}
|
|
72
|
+
catch (e) {
|
|
73
|
+
metrics.endTime = Date.now();
|
|
74
|
+
metrics.totalProcessingTime = metrics.endTime - metrics.startTime;
|
|
75
|
+
metrics.outputTokens = countTokens(fullText);
|
|
76
|
+
const handledError = await this.handleError(e);
|
|
77
|
+
push({
|
|
78
|
+
type: "error",
|
|
79
|
+
error: handledError,
|
|
80
|
+
text: e.message,
|
|
81
|
+
status: this.getErrorStatus(e, handledError),
|
|
82
|
+
headers: e?.headers,
|
|
83
|
+
metrics,
|
|
84
|
+
acc: fullText
|
|
85
|
+
});
|
|
86
|
+
q.end();
|
|
87
|
+
return q;
|
|
88
|
+
}
|
|
89
|
+
// IMPORTANT: do not await here — return q immediately for streaming
|
|
90
|
+
void (async () => {
|
|
91
|
+
try {
|
|
92
|
+
const final = await runner.start();
|
|
93
|
+
if (metrics.timeToFirstToken === 0) {
|
|
94
|
+
metrics.firstTime = Date.now();
|
|
95
|
+
metrics.timeToFirstToken = metrics.firstTime - metrics.startTime;
|
|
96
|
+
}
|
|
97
|
+
metrics.endTime = Date.now();
|
|
98
|
+
metrics.totalProcessingTime = metrics.endTime - metrics.startTime;
|
|
99
|
+
metrics.outputTokens = countTokens(fullText);
|
|
100
|
+
if (request.isHoloNative && protocol?.capability === ProtocolCapability.EMBED && this.translator.toHoloEmbedResponse) {
|
|
101
|
+
const normalized = await this.translator.toHoloEmbedResponse(final);
|
|
102
|
+
push({ type: "done", message: normalized, text: '', metrics });
|
|
103
|
+
}
|
|
104
|
+
else {
|
|
105
|
+
if (!fullText && request.isHoloNative && final) {
|
|
106
|
+
try {
|
|
107
|
+
const translated = await this.translator.toHoloResponse(final);
|
|
108
|
+
const extractedText = (translated.output ?? [])
|
|
109
|
+
.filter((m) => m.role === 'assistant')
|
|
110
|
+
.map((m) => typeof m.content === 'string' ? m.content : '')
|
|
111
|
+
.join('');
|
|
112
|
+
if (extractedText)
|
|
113
|
+
fullText = extractedText;
|
|
114
|
+
}
|
|
115
|
+
catch { /* best-effort fallback */
|
|
116
|
+
}
|
|
117
|
+
}
|
|
118
|
+
const holoResponse = (request.isHoloNative && !request.isStreaming) ? {
|
|
119
|
+
id: requestId,
|
|
120
|
+
model: await this.getModelNameFromRequest(requestPayload) ?? '',
|
|
121
|
+
output: fullText ? [{ role: 'assistant', content: fullText }] : [],
|
|
122
|
+
created: Date.now(),
|
|
123
|
+
finish_reason: this.auditor.mapFinishReason(final, protocol?.name),
|
|
124
|
+
usage: this.auditor.mapUsage(final, protocol?.name),
|
|
125
|
+
} : undefined;
|
|
126
|
+
push({
|
|
127
|
+
type: "done",
|
|
128
|
+
message: final,
|
|
129
|
+
text: fullText,
|
|
130
|
+
holoResponse,
|
|
131
|
+
metrics
|
|
132
|
+
});
|
|
133
|
+
}
|
|
134
|
+
}
|
|
135
|
+
catch (e) {
|
|
136
|
+
metrics.endTime = Date.now();
|
|
137
|
+
metrics.totalProcessingTime = metrics.endTime - metrics.startTime;
|
|
138
|
+
metrics.outputTokens = countTokens(fullText);
|
|
139
|
+
const handledError = await this.handleError(e);
|
|
140
|
+
push({
|
|
141
|
+
type: "error",
|
|
142
|
+
error: handledError,
|
|
143
|
+
text: e.message,
|
|
144
|
+
status: this.getErrorStatus(e, handledError),
|
|
145
|
+
headers: e?.headers,
|
|
146
|
+
metrics
|
|
147
|
+
});
|
|
148
|
+
}
|
|
149
|
+
finally {
|
|
150
|
+
q.end();
|
|
151
|
+
}
|
|
152
|
+
})();
|
|
153
|
+
return q;
|
|
154
|
+
}
|
|
155
|
+
getErrorStatus(e, handledError) {
|
|
156
|
+
return e?.status ?? e?.statusCode ?? handledError?.error?.code;
|
|
157
|
+
}
|
|
158
|
+
async translatePayload(capability, payload, _protocolName) {
|
|
159
|
+
switch (capability) {
|
|
160
|
+
case ProtocolCapability.GENERATE:
|
|
161
|
+
if (this.translator.fromHoloGenerateRequest) {
|
|
162
|
+
return this.translator.fromHoloGenerateRequest(payload);
|
|
163
|
+
}
|
|
164
|
+
break;
|
|
165
|
+
case ProtocolCapability.EMBED:
|
|
166
|
+
if (this.translator.fromHoloEmbedRequest) {
|
|
167
|
+
return this.translator.fromHoloEmbedRequest(payload);
|
|
168
|
+
}
|
|
169
|
+
break;
|
|
170
|
+
case ProtocolCapability.METRICS:
|
|
171
|
+
if (this.translator.fromHoloCountTokensRequest) {
|
|
172
|
+
return this.translator.fromHoloCountTokensRequest(payload);
|
|
173
|
+
}
|
|
174
|
+
break;
|
|
175
|
+
}
|
|
176
|
+
return this.translator.fromHoloRequest(payload);
|
|
177
|
+
}
|
|
178
|
+
createEventPusher(q, requestId) {
|
|
179
|
+
let seq = 0;
|
|
180
|
+
return (ev) => {
|
|
181
|
+
q.push({ ...ev, requestId, seq: seq++, ts: Date.now() });
|
|
182
|
+
};
|
|
183
|
+
}
|
|
184
|
+
}
|
|
185
|
+
//# sourceMappingURL=base.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/provider/base.ts"],"names":[],"mappings":"AAWA,OAAO,EAAC,kBAAkB,EAAoC,MAAM,8BAA8B,CAAC;AACnG,OAAO,EAAC,eAAe,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAC,MAAM,SAAS,CAAC;AAG/E,MAAM,OAAgB,YAAsE,SAAQ,WAAW;IAQvF;IACA;IACA;IACG;IAVP,OAAO,CAAW;IAClB,UAAU,CAAsB;IAChC,eAAe,CAAmB;IACxC,MAAM,GAA8B,EAAE,CAAC;IAC9B,MAAM,CAAiB;IAE1C,YACoB,EAAU,EACV,IAAY,EACZ,MAAuB,EACpB,OAAY;QAE/B,KAAK,EAAE,CAAC;QALQ,OAAE,GAAF,EAAE,CAAQ;QACV,SAAI,GAAJ,IAAI,CAAQ;QACZ,WAAM,GAAN,MAAM,CAAiB;QACpB,YAAO,GAAP,OAAO,CAAK;QAI/B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QAClC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;QACpC,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC1C,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,qBAAqB,EAAE,CAAC;IACxD,CAAC;IAMD,KAAK,CAAC,YAAY,CAAC,aAAgC;QAC/C,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;IACpD,CAAC;IAED,KAAK,CAAC,aAAa,CACf,cAAsC,EACtC,aAA4B;QAE5B,OAAO,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;IACrE,CAAC;IAED,KAAK,CAAC,oBAAoB,CACtB,OAA0B,EAC1B,KAAgC;QAEhC,MAAM,CAAC,GAAG,IAAI,eAAe,EAAiB,CAAC;QAE/C,MAAM,EAAC,SAAS,EAAE,QAAQ,EAAE,OAAO,EAAE,kBAAkB,EAAC,GAAG,OAAO,CAAC;QACnE,MAAM,cAAc,GAAG,OAAO,CAAC,YAAY;YACvC,CAAC,CAAC,MAAM,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,UAAU,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,CAAmB;YAC9F,CAAC,CAAC,OAAyB,CAAC;QAEhC,IAAI,QAAQ,GAAG,EAAE,CAAC;QAElB,MAAM,OAAO,GAAyB;YAClC,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,gBAAgB,EAAE,CAAC;YACnB,WAAW,EAAE,CAAC;YACd,YAAY,EAAE,CAAC;YACf,WAAW,EAAE,CAAC;YACd,mBAAmB,EAAE,CAAC;SACzB,CAAC;QAEF,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;QAElD,MAAM,KAAK,GAAG,EAAC,GAAG,CAAC,kBAAkB,EAAE,KAAK,IAAI,EAAE,CAAC,EAAC,CAAC;QACrD,IAAI,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;YAC9C,KAAK,CAAC,GAAG,GAAG,KAAK,CAAC;YAClB,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC;QAC1B,CAAC;QAED,MAAM,GAAG,GAAG,WAAW,CAAC;YACpB,QAAQ;YACR,OAAO,EAAE,kBAAkB,EAAE,OAAO;YACpC,KAAK;YACL,eAAe,EAAE,CAAC,KAAU,EAAE,EAAE,CAC5B,IAAI,CAAC,EAAC,IAAI,EAAE,cAAc,EAAE,KAAK,EAAkB,CAAC;YACxD,aAAa,EAAE,CAAC,IAAoB,EAAE,EAAE;gBACpC,IAAI,OAAO,CAAC,gBAAgB,KAAK,CAAC,EAAE,CAAC;oBACjC,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBAC/B,OAAO,CAAC,gBAAgB,GAAG,OAAO,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;gBACrE,CAAC;gBACD,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;oBACf,QAAQ,IAAI,IAAI,CAAC;oBACjB,IAAI,CAAC,EAAC,IAAI,EAAE,YAAY,EAAE,IAAI,EAAkB,CAAC,CAAC;gBACtD,CAAC;YACL,CAAC;YACD,iBAAiB,EAAE,CAAC,KAAa,EAAE,KAA+D,EAAE,EAAE,CAClG,IAAI,CAAC,EAAC,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,KAAK,EAAkB,CAAC;SACrE,CAAoB,CAAC;QAEtB,IAAI,MAA6B,CAAC;QAElC,IAAI,CAAC;YACD,MAAM,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC;QACjE,CAAC;QAAC,OAAO,CAAM,EAAE,CAAC;YACd,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,OAAO,CAAC,mBAAmB,GAAG,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC,SAAS,CAAC;YAClE,OAAO,CAAC,YAAY,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;YAC7C,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;YAC/C,IAAI,CAAC;gBACD,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,YAAY;gBACnB,IAAI,EAAG,CAAW,CAAC,OAAO;gBAC1B,MAAM,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC,EAAE,YAAY,CAAC;gBAC5C,OAAO,EAAE,CAAC,EAAE,OAAO;gBACnB,OAAO;gBACP,GAAG,EAAE,QAAQ;aACC,CAAC,CAAC;YACpB,CAAC,CAAC,GAAG,EAAE,CAAC;YACR,OAAO,CAAC,CAAC;QACb,CAAC;QAED,oEAAoE;QACpE,KAAK,CAAC,KAAK,IAAI,EAAE;YACb,IAAI,CAAC;gBACD,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;gBACnC,IAAI,OAAO,CAAC,gBAAgB,KAAK,CAAC,EAAE,CAAC;oBACjC,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBAC/B,OAAO,CAAC,gBAAgB,GAAG,OAAO,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;gBACrE,CAAC;gBACD,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAC7B,OAAO,CAAC,mBAAmB,GAAG,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC,SAAS,CAAC;gBAClE,OAAO,CAAC,YAAY,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;gBAE7C,IAAI,OAAO,CAAC,YAAY,IAAI,QAAQ,EAAE,UAAU,KAAK,kBAAkB,CAAC,KAAK,IAAI,IAAI,CAAC,UAAU,CAAC,mBAAmB,EAAE,CAAC;oBACnH,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;oBACpE,IAAI,CAAC,EAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,EAAE,OAAO,EAAkB,CAAC,CAAC;gBAClF,CAAC;qBAAM,CAAC;oBACJ,IAAI,CAAC,QAAQ,IAAI,OAAO,CAAC,YAAY,IAAI,KAAK,EAAE,CAAC;wBAC7C,IAAI,CAAC;4BACD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;4BAC/D,MAAM,aAAa,GAAG,CAAC,UAAU,CAAC,MAAM,IAAI,EAAE,CAAC;iCAC1C,MAAM,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC;iCAC1C,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;iCAC/D,IAAI,CAAC,EAAE,CAAC,CAAC;4BACd,IAAI,aAAa;gCAAE,QAAQ,GAAG,aAAa,CAAC;wBAChD,CAAC;wBAAC,MAAM,CAAC,CAAC,0BAA0B;wBACpC,CAAC;oBACL,CAAC;oBAED,MAAM,YAAY,GAA6B,CAAC,OAAO,CAAC,YAAY,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;wBAC5F,EAAE,EAAE,SAAS;wBACb,KAAK,EAAE,MAAM,IAAI,CAAC,uBAAuB,CAAC,cAAc,CAAC,IAAI,EAAE;wBAC/D,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAC,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAC,CAAC,CAAC,CAAC,CAAC,EAAE;wBAChE,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE;wBACnB,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,KAAK,EAAE,QAAQ,EAAE,IAAI,CAAC;wBAClE,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,EAAE,QAAQ,EAAE,IAAI,CAAC;qBACtD,CAAC,CAAC,CAAC,SAAS,CAAC;oBAEd,IAAI,CAAC;wBACD,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,KAAK;wBACd,IAAI,EAAE,QAAQ;wBACd,YAAY;wBACZ,OAAO;qBACO,CAAC,CAAC;gBACxB,CAAC;YACL,CAAC;YAAC,OAAO,CAAM,EAAE,CAAC;gBACd,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAC7B,OAAO,CAAC,mBAAmB,GAAG,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC,SAAS,CAAC;gBAClE,OAAO,CAAC,YAAY,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;gBAC7C,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;gBAC/C,IAAI,CAAC;oBACD,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,YAAY;oBACnB,IAAI,EAAG,CAAW,CAAC,OAAO;oBAC1B,MAAM,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC,EAAE,YAAY,CAAC;oBAC5C,OAAO,EAAE,CAAC,EAAE,OAAO;oBACnB,OAAO;iBACO,CAAC,CAAC;YACxB,CAAC;oBAAS,CAAC;gBACP,CAAC,CAAC,GAAG,EAAE,CAAC;YACZ,CAAC;QACL,CAAC,CAAC,EAAE,CAAC;QAEL,OAAO,CAAC,CAAC;IACb,CAAC;IAcS,cAAc,CAAC,CAAM,EAAE,YAAiB;QAC9C,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC,EAAE,UAAU,IAAI,YAAY,EAAE,KAAK,EAAE,IAAI,CAAC;IACnE,CAAC;IAOS,KAAK,CAAC,gBAAgB,CAAC,UAA0C,EAAE,OAAY,EAAE,aAAsB;QAC7G,QAAQ,UAAU,EAAE,CAAC;YACjB,KAAK,kBAAkB,CAAC,QAAQ;gBAC5B,IAAI,IAAI,CAAC,UAAU,CAAC,uBAAuB,EAAE,CAAC;oBAC1C,OAAO,IAAI,CAAC,UAAU,CAAC,uBAAuB,CAAC,OAAO,CAAC,CAAC;gBAC5D,CAAC;gBACD,MAAM;YACV,KAAK,kBAAkB,CAAC,KAAK;gBACzB,IAAI,IAAI,CAAC,UAAU,CAAC,oBAAoB,EAAE,CAAC;oBACvC,OAAO,IAAI,CAAC,UAAU,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;gBACzD,CAAC;gBACD,MAAM;YACV,KAAK,kBAAkB,CAAC,OAAO;gBAC3B,IAAI,IAAI,CAAC,UAAU,CAAC,0BAA0B,EAAE,CAAC;oBAC7C,OAAO,IAAI,CAAC,UAAU,CAAC,0BAA0B,CAAC,OAAO,CAAC,CAAC;gBAC/D,CAAC;gBACD,MAAM;QACd,CAAC;QACD,OAAO,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;IACpD,CAAC;IAEO,iBAAiB,CAAC,CAAiC,EAAE,SAAiB;QAC1E,IAAI,GAAG,GAAG,CAAC,CAAC;QACZ,OAAO,CAAC,EAAmD,EAAE,EAAE;YAC3D,CAAC,CAAC,IAAI,CAAC,EAAC,GAAG,EAAE,EAAE,SAAS,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE,EAAkB,CAAC,CAAC;QAC5E,CAAC,CAAC;IACN,CAAC;CACJ"}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
import { IWireAdapter, ProviderDoneEvent, ProviderErrorEvent, ProviderEvent } from '@holokai/holo-types/provider';
|
|
2
|
+
import type { HoloWorkerRequest, WorkerResponseEnvelope } from '@holokai/holo-types/worker';
|
|
3
|
+
export interface ProviderEventResult {
|
|
4
|
+
text: string | undefined;
|
|
5
|
+
final?: HoloWorkerRequest;
|
|
6
|
+
}
|
|
7
|
+
export declare function processProviderEvents(events: AsyncIterable<ProviderEvent>, wireAdapter: IWireAdapter, workerRequest: HoloWorkerRequest, envelope: WorkerResponseEnvelope, publisher: {
|
|
8
|
+
publishChunk(sourceId: string, requestId: string, data: object): Promise<void>;
|
|
9
|
+
publishFinal(final: HoloWorkerRequest & {
|
|
10
|
+
providerEvent: ProviderDoneEvent | ProviderErrorEvent;
|
|
11
|
+
}): Promise<void>;
|
|
12
|
+
}): Promise<ProviderEventResult>;
|
|
13
|
+
//# sourceMappingURL=event-processor.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"event-processor.d.ts","sourceRoot":"","sources":["../../src/provider/event-processor.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,YAAY,EACZ,iBAAiB,EACjB,kBAAkB,EAClB,aAAa,EAEhB,MAAM,8BAA8B,CAAC;AACtC,OAAO,KAAK,EAAC,iBAAiB,EAAE,sBAAsB,EAAC,MAAM,4BAA4B,CAAC;AAE1F,MAAM,WAAW,mBAAmB;IAChC,IAAI,EAAE,MAAM,GAAG,SAAS,CAAC;IACzB,KAAK,CAAC,EAAE,iBAAiB,CAAC;CAC7B;AAED,wBAAsB,qBAAqB,CACvC,MAAM,EAAE,aAAa,CAAC,aAAa,CAAC,EACpC,WAAW,EAAE,YAAY,EACzB,aAAa,EAAE,iBAAiB,EAChC,QAAQ,EAAE,sBAAsB,EAChC,SAAS,EAAE;IACP,YAAY,CAAC,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC/E,YAAY,CAAC,KAAK,EAAE,iBAAiB,GAAG;QACpC,aAAa,EAAE,iBAAiB,GAAG,kBAAkB,CAAA;KACxD,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;CACpB,GACF,OAAO,CAAC,mBAAmB,CAAC,CAiB9B"}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import { ProviderEventType } from '@holokai/holo-types/provider';
|
|
2
|
+
export async function processProviderEvents(events, wireAdapter, workerRequest, envelope, publisher) {
|
|
3
|
+
const result = { text: '' };
|
|
4
|
+
for await (const evt of events) {
|
|
5
|
+
if (evt.type === ProviderEventType.TEXT_DELTA)
|
|
6
|
+
result.text += evt.text;
|
|
7
|
+
for (const chunk of await wireAdapter.fromProviderEvent(evt)) {
|
|
8
|
+
await publisher.publishChunk(envelope.source_id, envelope.request_id, chunk);
|
|
9
|
+
}
|
|
10
|
+
if (evt.type === ProviderEventType.DONE || evt.type === ProviderEventType.ERROR) {
|
|
11
|
+
result.text = evt.type === ProviderEventType.DONE ? evt.text : evt.acc;
|
|
12
|
+
const final = { ...workerRequest, providerEvent: evt };
|
|
13
|
+
if (envelope.worker_id)
|
|
14
|
+
final.workerId = envelope.worker_id;
|
|
15
|
+
result.final = final;
|
|
16
|
+
await publisher.publishFinal(final);
|
|
17
|
+
break;
|
|
18
|
+
}
|
|
19
|
+
}
|
|
20
|
+
return result;
|
|
21
|
+
}
|
|
22
|
+
//# sourceMappingURL=event-processor.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"event-processor.js","sourceRoot":"","sources":["../../src/provider/event-processor.ts"],"names":[],"mappings":"AAAA,OAAO,EAKH,iBAAiB,EACpB,MAAM,8BAA8B,CAAC;AAQtC,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACvC,MAAoC,EACpC,WAAyB,EACzB,aAAgC,EAChC,QAAgC,EAChC,SAKC;IAED,MAAM,MAAM,GAAwB,EAAC,IAAI,EAAE,EAAE,EAAC,CAAC;IAC/C,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QAC7B,IAAI,GAAG,CAAC,IAAI,KAAK,iBAAiB,CAAC,UAAU;YAAE,MAAM,CAAC,IAAI,IAAI,GAAG,CAAC,IAAI,CAAC;QACvE,KAAK,MAAM,KAAK,IAAI,MAAM,WAAW,CAAC,iBAAiB,CAAC,GAAG,CAAC,EAAE,CAAC;YAC3D,MAAM,SAAS,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,EAAE,QAAQ,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QACjF,CAAC;QACD,IAAI,GAAG,CAAC,IAAI,KAAK,iBAAiB,CAAC,IAAI,IAAI,GAAG,CAAC,IAAI,KAAK,iBAAiB,CAAC,KAAK,EAAE,CAAC;YAC9E,MAAM,CAAC,IAAI,GAAG,GAAG,CAAC,IAAI,KAAK,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC;YACvE,MAAM,KAAK,GAAG,EAAC,GAAG,aAAa,EAAE,aAAa,EAAE,GAAG,EAAC,CAAC;YACrD,IAAI,QAAQ,CAAC,SAAS;gBAAE,KAAK,CAAC,QAAQ,GAAG,QAAQ,CAAC,SAAS,CAAC;YAC5D,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC;YACrB,MAAM,SAAS,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;YACpC,MAAM;QACV,CAAC;IACL,CAAC;IACD,OAAO,MAAM,CAAC;AAClB,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/provider/index.ts"],"names":[],"mappings":"AAAA,cAAc,WAAW,CAAC;AAC1B,cAAc,cAAc,CAAC;AAC7B,cAAc,QAAQ,CAAC;AACvB,cAAc,QAAQ,CAAC;AACvB,cAAc,mBAAmB,CAAC;AAClC,cAAc,0BAA0B,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/provider/index.ts"],"names":[],"mappings":"AAAA,cAAc,WAAW,CAAC;AAC1B,cAAc,cAAc,CAAC;AAC7B,cAAc,QAAQ,CAAC;AACvB,cAAc,QAAQ,CAAC;AACvB,cAAc,mBAAmB,CAAC;AAClC,cAAc,0BAA0B,CAAC"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import type { TranslateOptions } from "@holokai/holo-types/provider";
|
|
2
|
+
import { ClassLogger } from "../../core";
|
|
3
|
+
export declare abstract class BaseTranslator<THolo, TProvider> extends ClassLogger {
|
|
4
|
+
protected abstract holoDefaults: any;
|
|
5
|
+
protected abstract providerDefaults: any;
|
|
6
|
+
translate<TSource, TTarget>(source: TSource, options: TranslateOptions): Promise<Partial<TTarget>>;
|
|
7
|
+
translateArray<TSource, TTarget>(items: TSource[] | undefined, options: TranslateOptions): Promise<TTarget[]>;
|
|
8
|
+
fromHolo(source: THolo, validateTarget?: boolean): Promise<Partial<TProvider>>;
|
|
9
|
+
fromHoloArray(items: THolo[] | undefined, validateTarget?: boolean): Promise<TProvider[]>;
|
|
10
|
+
toHolo(target: TProvider, validateTarget?: boolean): Promise<Partial<THolo>>;
|
|
11
|
+
toHoloArray(items: TProvider[] | undefined, validateTarget?: boolean): Promise<THolo[]>;
|
|
12
|
+
protected abstract fromHoloImpl(source: THolo): Promise<Partial<TProvider>>;
|
|
13
|
+
protected abstract toHoloImpl(source: TProvider): Promise<Partial<THolo>>;
|
|
14
|
+
}
|
|
15
|
+
//# sourceMappingURL=base.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../src/provider/translator/base.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,gBAAgB,EAAC,MAAM,8BAA8B,CAAC;AACnE,OAAO,EAAC,WAAW,EAAC,MAAM,YAAY,CAAC;AAGvC,8BAAsB,cAAc,CAAC,KAAK,EAAE,SAAS,CAAE,SAAQ,WAAW;IAEtE,SAAS,CAAC,QAAQ,CAAC,YAAY,EAAE,GAAG,CAAC;IACrC,SAAS,CAAC,QAAQ,CAAC,gBAAgB,EAAE,GAAG,CAAC;IAEnC,SAAS,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAelG,cAAc,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,SAAS,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IAS7G,QAAQ,CAAC,MAAM,EAAE,KAAK,EAAE,cAAc,UAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAI5E,aAAa,CAAC,KAAK,EAAE,KAAK,EAAE,GAAG,SAAS,EAAE,cAAc,UAAQ,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAIvF,MAAM,CAAC,MAAM,EAAE,SAAS,EAAE,cAAc,UAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAI1E,WAAW,CAAC,KAAK,EAAE,SAAS,EAAE,GAAG,SAAS,EAAE,cAAc,UAAQ,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC;IAI3F,SAAS,CAAC,QAAQ,CAAC,YAAY,CAAC,MAAM,EAAE,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAE3E,SAAS,CAAC,QAAQ,CAAC,UAAU,CAAC,MAAM,EAAE,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;CAC5E"}
|
|
@@ -0,0 +1,33 @@
|
|
|
1
|
+
import { ClassLogger } from "../../core";
|
|
2
|
+
export class BaseTranslator extends ClassLogger {
|
|
3
|
+
async translate(source, options) {
|
|
4
|
+
const { fromHolo = true } = options;
|
|
5
|
+
const defaults = fromHolo ? this.providerDefaults : this.holoDefaults;
|
|
6
|
+
const partial = fromHolo
|
|
7
|
+
? (await this.fromHoloImpl(source))
|
|
8
|
+
: (await this.toHoloImpl(source));
|
|
9
|
+
return {
|
|
10
|
+
...(defaults ?? {}),
|
|
11
|
+
...(partial ?? {}),
|
|
12
|
+
};
|
|
13
|
+
}
|
|
14
|
+
async translateArray(items, options) {
|
|
15
|
+
if (!items?.length)
|
|
16
|
+
return [];
|
|
17
|
+
const results = await Promise.all(items.map(item => this.translate(item, options))); // Remove .bind(this)
|
|
18
|
+
return results.filter(item => Object.keys(item).length > 0);
|
|
19
|
+
}
|
|
20
|
+
async fromHolo(source, validateTarget = false) {
|
|
21
|
+
return await this.translate(source, { fromHolo: true, validateTarget });
|
|
22
|
+
}
|
|
23
|
+
async fromHoloArray(items, validateTarget = false) {
|
|
24
|
+
return await this.translateArray(items, { fromHolo: true, validateTarget });
|
|
25
|
+
}
|
|
26
|
+
async toHolo(target, validateTarget = false) {
|
|
27
|
+
return await this.translate(target, { fromHolo: false, validateTarget });
|
|
28
|
+
}
|
|
29
|
+
async toHoloArray(items, validateTarget = false) {
|
|
30
|
+
return await this.translateArray(items, { fromHolo: false, validateTarget });
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
//# sourceMappingURL=base.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../src/provider/translator/base.ts"],"names":[],"mappings":"AACA,OAAO,EAAC,WAAW,EAAC,MAAM,YAAY,CAAC;AAGvC,MAAM,OAAgB,cAAiC,SAAQ,WAAW;IAKtE,KAAK,CAAC,SAAS,CAAmB,MAAe,EAAE,OAAyB;QACxE,MAAM,EAAC,QAAQ,GAAG,IAAI,EAAC,GAAG,OAAO,CAAC;QAClC,MAAM,QAAQ,GAAG,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC;QAEtE,MAAM,OAAO,GAAqB,QAAQ;YACtC,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,YAAY,CAAC,MAA0B,CAAC,CAAgC;YACtF,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,UAAU,CAAC,MAA8B,CAAC,CAAgC,CAAC;QAE7F,OAAO;YACH,GAAG,CAAC,QAAQ,IAAI,EAAE,CAAC;YACnB,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC;SACrB,CAAC;IACN,CAAC;IAGD,KAAK,CAAC,cAAc,CAAmB,KAA4B,EAAE,OAAyB;QAC1F,IAAI,CAAC,KAAK,EAAE,MAAM;YAAE,OAAO,EAAE,CAAC;QAE9B,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAC/C,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,qBAAqB;QAE1D,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,CAAc,CAAC;IAC7E,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,MAAa,EAAE,cAAc,GAAG,KAAK;QAChD,OAAO,MAAM,IAAI,CAAC,SAAS,CAAmB,MAAM,EAAE,EAAC,QAAQ,EAAE,IAAI,EAAE,cAAc,EAAC,CAAC,CAAC;IAC5F,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,KAA0B,EAAE,cAAc,GAAG,KAAK;QAClE,OAAO,MAAM,IAAI,CAAC,cAAc,CAAmB,KAAK,EAAE,EAAC,QAAQ,EAAE,IAAI,EAAE,cAAc,EAAC,CAAC,CAAC;IAChG,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAiB,EAAE,cAAc,GAAG,KAAK;QAClD,OAAO,MAAM,IAAI,CAAC,SAAS,CAAmB,MAAM,EAAE,EAAC,QAAQ,EAAE,KAAK,EAAE,cAAc,EAAC,CAAC,CAAC;IAC7F,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,KAA8B,EAAE,cAAc,GAAG,KAAK;QACpE,OAAO,MAAM,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,EAAC,QAAQ,EAAE,KAAK,EAAE,cAAc,EAAC,CAAC,CAAC;IAC/E,CAAC;CAKJ"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/provider/translator/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAC;AACzB,cAAc,QAAQ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/provider/translator/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAC;AACzB,cAAc,QAAQ,CAAC"}
|
|
@@ -0,0 +1,24 @@
|
|
|
1
|
+
import type { TranslateOptions } from "@holokai/holo-types/provider";
|
|
2
|
+
export declare abstract class StreamTranslator<THolo, TProvider> {
|
|
3
|
+
/** Validators for single streaming chunks (not full messages) */
|
|
4
|
+
protected abstract providerDefaults: Partial<TProvider>;
|
|
5
|
+
protected abstract holoDefaults: Partial<THolo>;
|
|
6
|
+
/**
|
|
7
|
+
* Translate a single streaming chunk to 0..N target chunks.
|
|
8
|
+
* - Validates the source (as a complete chunk of its type).
|
|
9
|
+
* - Does NOT merge defaults (deltas are partial by design).
|
|
10
|
+
* - Optionally validates each emitted target chunk and strips undeclared keys.
|
|
11
|
+
*/
|
|
12
|
+
translateMany<TSource, TTarget>(source: TSource, options: TranslateOptions): Promise<Partial<TTarget>[]>;
|
|
13
|
+
/**
|
|
14
|
+
* Batch helper: many sources → flattened output stream.
|
|
15
|
+
*/
|
|
16
|
+
translateManyArray<TSource, TTarget>(items: TSource[] | undefined, options: TranslateOptions): Promise<Partial<TTarget>[]>;
|
|
17
|
+
fromHoloMany(source: THolo, validateTarget?: boolean): Promise<Partial<TProvider>[]>;
|
|
18
|
+
fromHoloManyArray(items: THolo[] | undefined, validateTarget?: boolean): Promise<Partial<TProvider>[]>;
|
|
19
|
+
toHoloMany(target: TProvider, validateTarget?: boolean): Promise<Partial<THolo>[]>;
|
|
20
|
+
toHoloManyArray(items: TProvider[] | undefined, validateTarget?: boolean): Promise<Partial<THolo>[]>;
|
|
21
|
+
protected abstract fromHoloManyImpl(source: THolo): Promise<Partial<TProvider>[]>;
|
|
22
|
+
protected abstract toHoloManyImpl(source: TProvider): Promise<Partial<THolo>[]>;
|
|
23
|
+
}
|
|
24
|
+
//# sourceMappingURL=stream.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stream.d.ts","sourceRoot":"","sources":["../../../src/provider/translator/stream.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAC,gBAAgB,EAAC,MAAM,8BAA8B,CAAC;AAEnE,8BAAsB,gBAAgB,CAAC,KAAK,EAAE,SAAS;IACnD,iEAAiE;IACjE,SAAS,CAAC,QAAQ,CAAC,gBAAgB,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;IACxD,SAAS,CAAC,QAAQ,CAAC,YAAY,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;IAEhD;;;;;OAKG;IACG,aAAa,CAAC,OAAO,EAAE,OAAO,EAChC,MAAM,EAAE,OAAO,EACf,OAAO,EAAE,gBAAgB,GAC1B,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;IA0B9B;;OAEG;IACG,kBAAkB,CAAC,OAAO,EAAE,OAAO,EACrC,KAAK,EAAE,OAAO,EAAE,GAAG,SAAS,EAC5B,OAAO,EAAE,gBAAgB,GAC1B,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;IAOxB,YAAY,CAAC,MAAM,EAAE,KAAK,EAAE,cAAc,UAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;IAIlF,iBAAiB,CAAC,KAAK,EAAE,KAAK,EAAE,GAAG,SAAS,EAAE,cAAc,UAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;IAIpG,UAAU,CAAC,MAAM,EAAE,SAAS,EAAE,cAAc,UAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;IAIhF,eAAe,CAAC,KAAK,EAAE,SAAS,EAAE,GAAG,SAAS,EAAE,cAAc,UAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;IAKxG,SAAS,CAAC,QAAQ,CAAC,gBAAgB,CAAC,MAAM,EAAE,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;IAEjF,SAAS,CAAC,QAAQ,CAAC,cAAc,CAAC,MAAM,EAAE,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;CAClF"}
|
|
@@ -0,0 +1,50 @@
|
|
|
1
|
+
export class StreamTranslator {
|
|
2
|
+
/**
|
|
3
|
+
* Translate a single streaming chunk to 0..N target chunks.
|
|
4
|
+
* - Validates the source (as a complete chunk of its type).
|
|
5
|
+
* - Does NOT merge defaults (deltas are partial by design).
|
|
6
|
+
* - Optionally validates each emitted target chunk and strips undeclared keys.
|
|
7
|
+
*/
|
|
8
|
+
async translateMany(source, options) {
|
|
9
|
+
const { fromHolo = true } = options;
|
|
10
|
+
const defaults = (fromHolo ? this.providerDefaults : this.holoDefaults) ?? {};
|
|
11
|
+
// 2) Impl: single → many
|
|
12
|
+
const emitted = fromHolo
|
|
13
|
+
? await this.fromHoloManyImpl(source)
|
|
14
|
+
: await this.toHoloManyImpl(source);
|
|
15
|
+
if (!Array.isArray(emitted) || emitted.length === 0)
|
|
16
|
+
return [];
|
|
17
|
+
// 3) Apply defaults per item
|
|
18
|
+
const withDefaults = emitted.map(item => ({ ...defaults, ...(item ?? {}) }));
|
|
19
|
+
const cleaned = [];
|
|
20
|
+
for (const item of withDefaults) {
|
|
21
|
+
const obj = item;
|
|
22
|
+
if (obj && Object.keys(obj).length > 0)
|
|
23
|
+
cleaned.push(obj);
|
|
24
|
+
}
|
|
25
|
+
return cleaned;
|
|
26
|
+
}
|
|
27
|
+
/**
|
|
28
|
+
* Batch helper: many sources → flattened output stream.
|
|
29
|
+
*/
|
|
30
|
+
async translateManyArray(items, options) {
|
|
31
|
+
if (!items?.length)
|
|
32
|
+
return [];
|
|
33
|
+
const batches = await Promise.all(items.map(item => this.translateMany(item, options)));
|
|
34
|
+
return batches.flat().filter(o => o && Object.keys(o).length > 0);
|
|
35
|
+
}
|
|
36
|
+
// Convenience wrappers
|
|
37
|
+
async fromHoloMany(source, validateTarget = false) {
|
|
38
|
+
return this.translateMany(source, { fromHolo: true, validateTarget });
|
|
39
|
+
}
|
|
40
|
+
async fromHoloManyArray(items, validateTarget = false) {
|
|
41
|
+
return this.translateManyArray(items, { fromHolo: true, validateTarget });
|
|
42
|
+
}
|
|
43
|
+
async toHoloMany(target, validateTarget = false) {
|
|
44
|
+
return this.translateMany(target, { fromHolo: false, validateTarget });
|
|
45
|
+
}
|
|
46
|
+
async toHoloManyArray(items, validateTarget = false) {
|
|
47
|
+
return this.translateManyArray(items, { fromHolo: false, validateTarget });
|
|
48
|
+
}
|
|
49
|
+
}
|
|
50
|
+
//# sourceMappingURL=stream.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stream.js","sourceRoot":"","sources":["../../../src/provider/translator/stream.ts"],"names":[],"mappings":"AAEA,MAAM,OAAgB,gBAAgB;IAKlC;;;;;OAKG;IACH,KAAK,CAAC,aAAa,CACf,MAAe,EACf,OAAyB;QAEzB,MAAM,EACF,QAAQ,GAAG,IAAI,EAClB,GAAG,OAAO,CAAC;QAEZ,MAAM,QAAQ,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC;QAE9E,yBAAyB;QACzB,MAAM,OAAO,GAAuB,QAAQ;YACxC,CAAC,CAAC,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAA0B,CAAkC;YAC1F,CAAC,CAAC,MAAM,IAAI,CAAC,cAAc,CAAC,MAA8B,CAAkC,CAAC;QAEjG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAE/D,6BAA6B;QAC7B,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,EAAC,GAAG,QAAQ,EAAE,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC,EAAC,CAAC,CAAC,CAAC;QAE3E,MAAM,OAAO,GAAuB,EAAE,CAAC;QACvC,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;YAC9B,MAAM,GAAG,GAAG,IAAwB,CAAC;YACrC,IAAI,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,MAAM,GAAG,CAAC;gBAAE,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC9D,CAAC;QAED,OAAO,OAAO,CAAC;IACnB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,kBAAkB,CACpB,KAA4B,EAC5B,OAAyB;QAEzB,IAAI,CAAC,KAAK,EAAE,MAAM;YAAE,OAAO,EAAE,CAAC;QAC9B,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,aAAa,CAAmB,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;QAC1G,OAAO,OAAO,CAAC,IAAI,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACtE,CAAC;IAED,uBAAuB;IACvB,KAAK,CAAC,YAAY,CAAC,MAAa,EAAE,cAAc,GAAG,KAAK;QACpD,OAAO,IAAI,CAAC,aAAa,CAAmB,MAAM,EAAE,EAAC,QAAQ,EAAE,IAAI,EAAE,cAAc,EAAC,CAAC,CAAC;IAC1F,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,KAA0B,EAAE,cAAc,GAAG,KAAK;QACtE,OAAO,IAAI,CAAC,kBAAkB,CAAmB,KAAK,EAAE,EAAC,QAAQ,EAAE,IAAI,EAAE,cAAc,EAAC,CAAC,CAAC;IAC9F,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,MAAiB,EAAE,cAAc,GAAG,KAAK;QACtD,OAAO,IAAI,CAAC,aAAa,CAAmB,MAAM,EAAE,EAAC,QAAQ,EAAE,KAAK,EAAE,cAAc,EAAC,CAAC,CAAC;IAC3F,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,KAA8B,EAAE,cAAc,GAAG,KAAK;QACxE,OAAO,IAAI,CAAC,kBAAkB,CAAmB,KAAK,EAAE,EAAC,QAAQ,EAAE,KAAK,EAAE,cAAc,EAAC,CAAC,CAAC;IAC/F,CAAC;CAMJ"}
|
|
@@ -0,0 +1,39 @@
|
|
|
1
|
+
import { ClassLogger } from '../../core';
|
|
2
|
+
import type { IWireAdapter, ProviderEvent, WireChunk } from '@holokai/holo-types/provider';
|
|
3
|
+
export declare abstract class BaseWireAdapter extends ClassLogger implements IWireAdapter {
|
|
4
|
+
readonly requestId: string;
|
|
5
|
+
readonly isStreaming: boolean;
|
|
6
|
+
wireSeq: number;
|
|
7
|
+
constructor(requestId: string, isStreaming: boolean);
|
|
8
|
+
fromProviderEvent(ev: ProviderEvent): Promise<WireChunk[]>;
|
|
9
|
+
abstract formatWire(data: any): string;
|
|
10
|
+
protected nonStreamingHeaders(): Record<string, string>;
|
|
11
|
+
protected streamingHeaders(): Record<string, string>;
|
|
12
|
+
protected defaultHeadersForFirst(ev: ProviderEvent): Record<string, string>;
|
|
13
|
+
protected chunkify(ev: ProviderEvent, bodyFn: (ev: any) => Promise<string>, done?: true, options?: {
|
|
14
|
+
fullText?: string;
|
|
15
|
+
status?: number;
|
|
16
|
+
headers?: Record<string, string>;
|
|
17
|
+
}): Promise<WireChunk>;
|
|
18
|
+
protected fromNonStreaming(ev: ProviderEvent): Promise<WireChunk[]>;
|
|
19
|
+
protected fromStreaming(ev: ProviderEvent): Promise<WireChunk[]>;
|
|
20
|
+
protected onStreamEvent(ev: Extract<ProviderEvent, {
|
|
21
|
+
type: 'stream_event';
|
|
22
|
+
}>): Promise<WireChunk[]>;
|
|
23
|
+
protected onDoneStreaming(ev: Extract<ProviderEvent, {
|
|
24
|
+
type: 'done';
|
|
25
|
+
}>): Promise<WireChunk[]>;
|
|
26
|
+
protected onErrorStreaming(ev: Extract<ProviderEvent, {
|
|
27
|
+
type: 'error';
|
|
28
|
+
}>): Promise<WireChunk[]>;
|
|
29
|
+
protected defaultDoneStreamFormatter(ev: Extract<ProviderEvent, {
|
|
30
|
+
type: 'done';
|
|
31
|
+
}>): Promise<string>;
|
|
32
|
+
protected defaultStreamErrorFormatter(ev: Extract<ProviderEvent, {
|
|
33
|
+
type: 'error';
|
|
34
|
+
}>): Promise<string>;
|
|
35
|
+
protected defaultStreamEventFormatter(ev: Extract<ProviderEvent, {
|
|
36
|
+
type: 'stream_event';
|
|
37
|
+
}>): Promise<string>;
|
|
38
|
+
}
|
|
39
|
+
//# sourceMappingURL=base.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../src/provider/wire/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,WAAW,EAA4B,MAAM,YAAY,CAAC;AAClE,OAAO,KAAK,EAAC,YAAY,EAAE,aAAa,EAAE,SAAS,EAAC,MAAM,8BAA8B,CAAC;AAEzF,8BAAsB,eAAgB,SAAQ,WAAY,YAAW,YAAY;aAIzD,SAAS,EAAE,MAAM;aACjB,WAAW,EAAE,OAAO;IAJxC,OAAO,SAAK;gBAGQ,SAAS,EAAE,MAAM,EACjB,WAAW,EAAE,OAAO;IAKlC,iBAAiB,CAAC,EAAE,EAAE,aAAa,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAKhE,QAAQ,CAAC,UAAU,CAAC,IAAI,EAAE,GAAG,GAAG,MAAM;IAGtC,SAAS,CAAC,mBAAmB,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAIvD,SAAS,CAAC,gBAAgB,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAQpD,SAAS,CAAC,sBAAsB,CAAC,EAAE,EAAE,aAAa,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;cAI3D,QAAQ,CAAC,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,GAAG,KAAK,OAAO,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,EAAE;QACrG,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;KACnC,GAAG,OAAO,CAAC,SAAS,CAAC;cAiCN,gBAAgB,CAAC,EAAE,EAAE,aAAa,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;cAezD,aAAa,CAAC,EAAE,EAAE,aAAa,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;cActD,aAAa,CAAC,EAAE,EAAE,OAAO,CAAC,aAAa,EAAE;QAAE,IAAI,EAAE,cAAc,CAAA;KAAE,CAAC,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;cAIzF,eAAe,CAAC,EAAE,EAAE,OAAO,CAAC,aAAa,EAAE;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;cAInF,gBAAgB,CAAC,EAAE,EAAE,OAAO,CAAC,aAAa,EAAE;QAAE,IAAI,EAAE,OAAO,CAAA;KAAE,CAAC,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;cAQrF,0BAA0B,CAAC,EAAE,EAAE,OAAO,CAAC,aAAa,EAAE;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;cAIzF,2BAA2B,CAAC,EAAE,EAAE,OAAO,CAAC,aAAa,EAAE;QAAE,IAAI,EAAE,OAAO,CAAA;KAAE,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;cAI3F,2BAA2B,CAAC,EAAE,EAAE,OAAO,CAAC,aAAa,EAAE;QAAE,IAAI,EAAE,cAAc,CAAA;KAAE,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;CAGrH"}
|
|
@@ -0,0 +1,106 @@
|
|
|
1
|
+
import { ClassLogger, pickDefined, stringifyAny } from '../../core';
|
|
2
|
+
export class BaseWireAdapter extends ClassLogger {
|
|
3
|
+
requestId;
|
|
4
|
+
isStreaming;
|
|
5
|
+
wireSeq = 0;
|
|
6
|
+
constructor(requestId, isStreaming) {
|
|
7
|
+
super();
|
|
8
|
+
this.requestId = requestId;
|
|
9
|
+
this.isStreaming = isStreaming;
|
|
10
|
+
}
|
|
11
|
+
async fromProviderEvent(ev) {
|
|
12
|
+
if (!this.isStreaming)
|
|
13
|
+
return this.fromNonStreaming(ev);
|
|
14
|
+
return this.fromStreaming(ev);
|
|
15
|
+
}
|
|
16
|
+
// --- defaults ---
|
|
17
|
+
nonStreamingHeaders() {
|
|
18
|
+
return { 'Content-Type': 'application/json' };
|
|
19
|
+
}
|
|
20
|
+
streamingHeaders() {
|
|
21
|
+
return {
|
|
22
|
+
'Content-Type': 'text/event-stream',
|
|
23
|
+
'Cache-Control': 'no-cache',
|
|
24
|
+
'Connection': 'keep-alive',
|
|
25
|
+
};
|
|
26
|
+
}
|
|
27
|
+
defaultHeadersForFirst(ev) {
|
|
28
|
+
return (ev.type === 'error' || !this.isStreaming) ? this.nonStreamingHeaders() : this.streamingHeaders();
|
|
29
|
+
}
|
|
30
|
+
async chunkify(ev, bodyFn, done, options) {
|
|
31
|
+
const isFirst = this.wireSeq == 0;
|
|
32
|
+
const seq = this.wireSeq++;
|
|
33
|
+
const body = await bodyFn(ev);
|
|
34
|
+
const fullText = options?.fullText;
|
|
35
|
+
const chunk = pickDefined({
|
|
36
|
+
fullText,
|
|
37
|
+
requestId: this.requestId,
|
|
38
|
+
seq,
|
|
39
|
+
eventSeq: ev.seq,
|
|
40
|
+
body,
|
|
41
|
+
done,
|
|
42
|
+
});
|
|
43
|
+
if (isFirst) {
|
|
44
|
+
const status = options?.status ?? (ev.type === 'error' ? (ev.status ?? 400) : 200);
|
|
45
|
+
const headers = options?.headers ??
|
|
46
|
+
(ev.type === 'error' && ev.headers ? ev.headers : undefined) ?? this.defaultHeadersForFirst(ev);
|
|
47
|
+
chunk.status = status;
|
|
48
|
+
chunk.headers = headers;
|
|
49
|
+
}
|
|
50
|
+
else {
|
|
51
|
+
// generally do NOT send headers/status after first chunk; allow explicit override if needed
|
|
52
|
+
if (options?.status !== undefined)
|
|
53
|
+
chunk.status = options.status;
|
|
54
|
+
if (options?.headers)
|
|
55
|
+
chunk.headers = options.headers;
|
|
56
|
+
}
|
|
57
|
+
return chunk;
|
|
58
|
+
}
|
|
59
|
+
// --- non-streaming ---
|
|
60
|
+
async fromNonStreaming(ev) {
|
|
61
|
+
if (ev.type === 'done') {
|
|
62
|
+
return [await this.chunkify(ev, async (ev) => stringifyAny(ev.message), true)];
|
|
63
|
+
}
|
|
64
|
+
if (ev.type === 'error') {
|
|
65
|
+
return [await this.chunkify(ev, async (ev) => stringifyAny(ev.error), true, pickDefined({ status: ev.status ?? 400, headers: ev.headers }))];
|
|
66
|
+
}
|
|
67
|
+
return [];
|
|
68
|
+
}
|
|
69
|
+
// --- streaming ---
|
|
70
|
+
async fromStreaming(ev) {
|
|
71
|
+
switch (ev.type) {
|
|
72
|
+
case 'stream_event':
|
|
73
|
+
return this.onStreamEvent(ev);
|
|
74
|
+
case 'done':
|
|
75
|
+
return this.onDoneStreaming(ev);
|
|
76
|
+
case 'error':
|
|
77
|
+
return this.onErrorStreaming(ev);
|
|
78
|
+
case 'text_delta':
|
|
79
|
+
default:
|
|
80
|
+
return [];
|
|
81
|
+
}
|
|
82
|
+
}
|
|
83
|
+
async onStreamEvent(ev) {
|
|
84
|
+
return [await this.chunkify(ev, this.defaultStreamEventFormatter.bind(this))];
|
|
85
|
+
}
|
|
86
|
+
async onDoneStreaming(ev) {
|
|
87
|
+
return [await this.chunkify(ev, this.defaultDoneStreamFormatter.bind(this), true, { fullText: ev.text })];
|
|
88
|
+
}
|
|
89
|
+
async onErrorStreaming(ev) {
|
|
90
|
+
if (this.wireSeq === 0) {
|
|
91
|
+
return await this.fromNonStreaming(ev);
|
|
92
|
+
}
|
|
93
|
+
// Mid-stream: default behavior (provider adapters should usually override)
|
|
94
|
+
return [await this.chunkify(ev, this.defaultStreamErrorFormatter.bind(this), true)];
|
|
95
|
+
}
|
|
96
|
+
async defaultDoneStreamFormatter(ev) {
|
|
97
|
+
return this.formatWire(ev.message);
|
|
98
|
+
}
|
|
99
|
+
async defaultStreamErrorFormatter(ev) {
|
|
100
|
+
return stringifyAny(ev.error);
|
|
101
|
+
}
|
|
102
|
+
async defaultStreamEventFormatter(ev) {
|
|
103
|
+
return this.formatWire(ev.event);
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
//# sourceMappingURL=base.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../src/provider/wire/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAAC,WAAW,EAAE,WAAW,EAAE,YAAY,EAAC,MAAM,YAAY,CAAC;AAGlE,MAAM,OAAgB,eAAgB,SAAQ,WAAW;IAIjC;IACA;IAJpB,OAAO,GAAG,CAAC,CAAC;IAEZ,YACoB,SAAiB,EACjB,WAAoB;QAEpC,KAAK,EAAE,CAAC;QAHQ,cAAS,GAAT,SAAS,CAAQ;QACjB,gBAAW,GAAX,WAAW,CAAS;IAGxC,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,EAAiB;QACrC,IAAI,CAAC,IAAI,CAAC,WAAW;YAAE,OAAO,IAAI,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;QACxD,OAAO,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;IAClC,CAAC;IAID,mBAAmB;IACT,mBAAmB;QACzB,OAAO,EAAC,cAAc,EAAE,kBAAkB,EAAC,CAAC;IAChD,CAAC;IAES,gBAAgB;QACtB,OAAO;YACH,cAAc,EAAE,mBAAmB;YACnC,eAAe,EAAE,UAAU;YAC3B,YAAY,EAAE,YAAY;SAC7B,CAAC;IACN,CAAC;IAES,sBAAsB,CAAC,EAAiB;QAC9C,OAAO,CAAC,EAAE,CAAC,IAAI,KAAK,OAAO,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;IAC7G,CAAC;IAES,KAAK,CAAC,QAAQ,CAAC,EAAiB,EAAE,MAAoC,EAAE,IAAW,EAAE,OAI9F;QACG,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,IAAI,CAAC,CAAC;QAClC,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;QAC3B,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,EAAE,CAAC,CAAC;QAC9B,MAAM,QAAQ,GAAG,OAAO,EAAE,QAAQ,CAAC;QAEnC,MAAM,KAAK,GAAG,WAAW,CAAC;YACtB,QAAQ;YACR,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,GAAG;YACH,QAAQ,EAAE,EAAE,CAAC,GAAG;YAChB,IAAI;YACJ,IAAI;SACP,CAAc,CAAC;QAEhB,IAAI,OAAO,EAAE,CAAC;YACV,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,CAAC,EAAE,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YACnF,MAAM,OAAO,GACT,OAAO,EAAE,OAAO;gBAChB,CAAC,EAAE,CAAC,IAAI,KAAK,OAAO,IAAI,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,IAAI,CAAC,sBAAsB,CAAC,EAAE,CAAC,CAAC;YAEpG,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC;YACtB,KAAK,CAAC,OAAO,GAAG,OAAO,CAAC;QAC5B,CAAC;aAAM,CAAC;YACJ,4FAA4F;YAC5F,IAAI,OAAO,EAAE,MAAM,KAAK,SAAS;gBAAE,KAAK,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;YACjE,IAAI,OAAO,EAAE,OAAO;gBAAE,KAAK,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAC1D,CAAC;QAED,OAAO,KAAK,CAAC;IACjB,CAAC;IAED,wBAAwB;IACd,KAAK,CAAC,gBAAgB,CAAC,EAAiB;QAC9C,IAAI,EAAE,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACrB,OAAO,CAAC,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,KAAK,EAAE,EAErC,EAAE,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;QAC5C,CAAC;QACD,IAAI,EAAE,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YACtB,OAAO,CAAC,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,KAAK,EAAE,EAErC,EAAE,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,WAAW,CAAC,EAAC,MAAM,EAAE,EAAE,CAAC,MAAM,IAAI,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,OAAO,EAAC,CAAC,CAAC,CAAC,CAAC;QACxG,CAAC;QACD,OAAO,EAAE,CAAC;IACd,CAAC;IAED,oBAAoB;IACV,KAAK,CAAC,aAAa,CAAC,EAAiB;QAC3C,QAAQ,EAAE,CAAC,IAAI,EAAE,CAAC;YACd,KAAK,cAAc;gBACf,OAAO,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;YAClC,KAAK,MAAM;gBACP,OAAO,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;YACpC,KAAK,OAAO;gBACR,OAAO,IAAI,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;YACrC,KAAK,YAAY,CAAC;YAClB;gBACI,OAAO,EAAE,CAAC;QAClB,CAAC;IACL,CAAC;IAES,KAAK,CAAC,aAAa,CAAC,EAAoD;QAC9E,OAAO,CAAC,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClF,CAAC;IAES,KAAK,CAAC,eAAe,CAAC,EAA4C;QACxE,OAAO,CAAC,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,IAAI,CAAC,0BAA0B,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,EAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,EAAC,CAAC,CAAC,CAAC;IAC5G,CAAC;IAES,KAAK,CAAC,gBAAgB,CAAC,EAA6C;QAC1E,IAAI,IAAI,CAAC,OAAO,KAAK,CAAC,EAAE,CAAC;YACrB,OAAO,MAAM,IAAI,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;QAC3C,CAAC;QACD,2EAA2E;QAC3E,OAAO,CAAC,MAAM,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;IACxF,CAAC;IAES,KAAK,CAAC,0BAA0B,CAAC,EAA4C;QACnF,OAAO,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;IACvC,CAAC;IAES,KAAK,CAAC,2BAA2B,CAAC,EAA6C;QACrF,OAAO,YAAY,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAES,KAAK,CAAC,2BAA2B,CAAC,EAAoD;QAC5F,OAAO,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;IACrC,CAAC;CACJ"}
|
|
@@ -0,0 +1,33 @@
|
|
|
1
|
+
import type { ProviderEvent, WireChunk } from '@holokai/holo-types/provider';
|
|
2
|
+
import type { HoloStreamEvent } from '@holokai/holo-types/holo';
|
|
3
|
+
import type { ProtocolCapability } from '@holokai/holo-types/entities';
|
|
4
|
+
import { BaseWireAdapter } from './base';
|
|
5
|
+
export declare class HoloWireAdapter extends BaseWireAdapter {
|
|
6
|
+
private textAccumulator;
|
|
7
|
+
private toolCalls;
|
|
8
|
+
private responseId?;
|
|
9
|
+
private model?;
|
|
10
|
+
private firstEvent;
|
|
11
|
+
private readonly capability;
|
|
12
|
+
constructor(requestId: string, isStreaming: boolean, initialModel?: string, capability?: ProtocolCapability);
|
|
13
|
+
formatWire(data: string | HoloStreamEvent): string;
|
|
14
|
+
emitCreated(id: string, model: string): HoloStreamEvent;
|
|
15
|
+
emitTextDelta(index: number, text: string): HoloStreamEvent;
|
|
16
|
+
emitReasoningDelta(index: number, text: string): HoloStreamEvent;
|
|
17
|
+
emitToolCallDelta(index: number, delta: {
|
|
18
|
+
id?: string;
|
|
19
|
+
name?: string;
|
|
20
|
+
arguments_delta?: string;
|
|
21
|
+
}): HoloStreamEvent;
|
|
22
|
+
emitHoloEvent(ev: ProviderEvent, holoEvent: HoloStreamEvent): Promise<WireChunk[]>;
|
|
23
|
+
protected fromNonStreaming(ev: ProviderEvent): Promise<WireChunk[]>;
|
|
24
|
+
protected fromStreaming(ev: ProviderEvent): Promise<WireChunk[]>;
|
|
25
|
+
protected onDoneStreaming(ev: Extract<ProviderEvent, {
|
|
26
|
+
type: 'done';
|
|
27
|
+
}>): Promise<WireChunk[]>;
|
|
28
|
+
protected onErrorStreaming(ev: Extract<ProviderEvent, {
|
|
29
|
+
type: 'error';
|
|
30
|
+
}>): Promise<WireChunk[]>;
|
|
31
|
+
private buildFinalResponse;
|
|
32
|
+
}
|
|
33
|
+
//# sourceMappingURL=holo.wire.adapter.d.ts.map
|