theorum 0.1.3 → 0.1.4

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.
Files changed (187) hide show
  1. package/README.md +30 -9
  2. package/deno.json +57 -0
  3. package/deno.lock +469 -0
  4. package/{esm/mod.d.ts → mod.ts} +57 -15
  5. package/package.json +35 -44
  6. package/src/cli/commands/profile.ts +57 -0
  7. package/src/cli/commands/run.ts +66 -0
  8. package/src/cli/commands/test.ts +195 -0
  9. package/src/cli/index.ts +161 -0
  10. package/src/cli/matrix/fixtures.ts +96 -0
  11. package/src/cli/matrix/synthesizer.ts +240 -0
  12. package/src/guardrails/error.ts +142 -0
  13. package/src/guardrails/injection.ts +254 -0
  14. package/src/guardrails/mod.ts +40 -0
  15. package/src/guardrails/quota.ts +87 -0
  16. package/src/guardrails/sanitize.ts +165 -0
  17. package/src/guardrails/sensitive.ts +96 -0
  18. package/src/host/mint-trace.ts +54 -0
  19. package/{esm/src/host/mod.js → src/host/mod.ts} +11 -3
  20. package/src/host/reply.ts +32 -0
  21. package/src/kernel/engine/assert.ts +29 -0
  22. package/src/kernel/engine/boundary.ts +75 -0
  23. package/src/kernel/engine/delta.ts +440 -0
  24. package/src/kernel/engine/hash.ts +11 -0
  25. package/src/kernel/engine/record.ts +8 -0
  26. package/src/kernel/engine/repair.ts +56 -0
  27. package/src/kernel/engine/runner/gates.ts +338 -0
  28. package/src/kernel/engine/runner/mod.ts +135 -0
  29. package/src/kernel/engine/runner/state.ts +31 -0
  30. package/src/kernel/engine/runner/steps.ts +183 -0
  31. package/src/kernel/engine/runner/stream.ts +83 -0
  32. package/src/kernel/engine/runner/tokens.ts +47 -0
  33. package/src/kernel/engine/runner/tools.ts +266 -0
  34. package/{esm/src/kernel/engine/runner.d.ts → src/kernel/engine/runner.ts} +2 -1
  35. package/src/kernel/engine/tree.ts +18 -0
  36. package/src/kernel/mod.ts +39 -0
  37. package/src/kernel/registry/attachments.ts +195 -0
  38. package/src/kernel/registry/catalog.ts +181 -0
  39. package/src/kernel/registry/ingress.ts +223 -0
  40. package/src/kernel/registry/profiles.ts +165 -0
  41. package/src/kernel/registry/provider-request.ts +31 -0
  42. package/src/kernel/registry/resolve.ts +290 -0
  43. package/src/kernel/registry/schemas.ts +29 -0
  44. package/src/kernel/registry/tools.ts +45 -0
  45. package/{esm/src/kernel/registry/vault.d.ts → src/kernel/registry/vault.ts} +17 -2
  46. package/src/kernel/types.ts +576 -0
  47. package/{esm/src/observability/mod.js → src/observability/mod.ts} +11 -2
  48. package/src/observability/spans.ts +70 -0
  49. package/src/observability/trace-attach.ts +96 -0
  50. package/src/observability/trace-record.ts +254 -0
  51. package/src/observability/trace-usage.ts +36 -0
  52. package/src/observability/trace.ts +143 -0
  53. package/src/presets/google.ts +147 -0
  54. package/src/presets/mod.ts +24 -0
  55. package/src/providers/create-provider.ts +64 -0
  56. package/src/providers/gemini-tape.ts +59 -0
  57. package/src/providers/google-tap.ts +57 -0
  58. package/src/providers/interactions.ts +182 -0
  59. package/src/providers/keys.ts +182 -0
  60. package/{esm/src/providers/mod.js → src/providers/mod.ts} +4 -2
  61. package/{esm/src/providers/openrouter-mod.js → src/providers/openrouter-mod.ts} +7 -3
  62. package/src/providers/openrouter-payload.ts +232 -0
  63. package/src/providers/openrouter.ts +758 -0
  64. package/src/providers/pcm.ts +37 -0
  65. package/src/providers/provider.ts +206 -0
  66. package/src/providers/speech.ts +180 -0
  67. package/src/providers/sse.ts +59 -0
  68. package/esm/_dnt.polyfills.d.ts +0 -110
  69. package/esm/_dnt.polyfills.js +0 -144
  70. package/esm/_dnt.shims.d.ts +0 -5
  71. package/esm/_dnt.shims.js +0 -61
  72. package/esm/mod.js +0 -49
  73. package/esm/package.json +0 -3
  74. package/esm/src/cli/commands/profile.d.ts +0 -2
  75. package/esm/src/cli/commands/profile.js +0 -53
  76. package/esm/src/cli/commands/run.d.ts +0 -10
  77. package/esm/src/cli/commands/run.js +0 -55
  78. package/esm/src/cli/commands/test.d.ts +0 -15
  79. package/esm/src/cli/commands/test.js +0 -140
  80. package/esm/src/cli/index.d.ts +0 -4
  81. package/esm/src/cli/index.js +0 -148
  82. package/esm/src/cli/matrix/fixtures.d.ts +0 -13
  83. package/esm/src/cli/matrix/fixtures.js +0 -77
  84. package/esm/src/cli/matrix/synthesizer.d.ts +0 -29
  85. package/esm/src/cli/matrix/synthesizer.js +0 -187
  86. package/esm/src/guardrails/error.d.ts +0 -33
  87. package/esm/src/guardrails/error.js +0 -114
  88. package/esm/src/guardrails/injection.d.ts +0 -12
  89. package/esm/src/guardrails/injection.js +0 -220
  90. package/esm/src/guardrails/mod.d.ts +0 -14
  91. package/esm/src/guardrails/mod.js +0 -14
  92. package/esm/src/guardrails/sanitize.d.ts +0 -22
  93. package/esm/src/guardrails/sanitize.js +0 -133
  94. package/esm/src/guardrails/sensitive.d.ts +0 -12
  95. package/esm/src/guardrails/sensitive.js +0 -88
  96. package/esm/src/host/mint-trace.d.ts +0 -27
  97. package/esm/src/host/mint-trace.js +0 -33
  98. package/esm/src/host/mod.d.ts +0 -13
  99. package/esm/src/host/reply.d.ts +0 -14
  100. package/esm/src/host/reply.js +0 -27
  101. package/esm/src/kernel/engine/boundary.d.ts +0 -10
  102. package/esm/src/kernel/engine/boundary.js +0 -55
  103. package/esm/src/kernel/engine/delta.d.ts +0 -8
  104. package/esm/src/kernel/engine/delta.js +0 -389
  105. package/esm/src/kernel/engine/hash.d.ts +0 -1
  106. package/esm/src/kernel/engine/hash.js +0 -9
  107. package/esm/src/kernel/engine/record.d.ts +0 -2
  108. package/esm/src/kernel/engine/record.js +0 -7
  109. package/esm/src/kernel/engine/repair.d.ts +0 -7
  110. package/esm/src/kernel/engine/repair.js +0 -38
  111. package/esm/src/kernel/engine/runner/gates.d.ts +0 -4
  112. package/esm/src/kernel/engine/runner/gates.js +0 -206
  113. package/esm/src/kernel/engine/runner/mod.d.ts +0 -14
  114. package/esm/src/kernel/engine/runner/mod.js +0 -103
  115. package/esm/src/kernel/engine/runner/state.d.ts +0 -16
  116. package/esm/src/kernel/engine/runner/state.js +0 -8
  117. package/esm/src/kernel/engine/runner/steps.d.ts +0 -15
  118. package/esm/src/kernel/engine/runner/steps.js +0 -119
  119. package/esm/src/kernel/engine/runner/stream.d.ts +0 -11
  120. package/esm/src/kernel/engine/runner/stream.js +0 -65
  121. package/esm/src/kernel/engine/runner/tokens.d.ts +0 -3
  122. package/esm/src/kernel/engine/runner/tokens.js +0 -38
  123. package/esm/src/kernel/engine/runner/tools.d.ts +0 -13
  124. package/esm/src/kernel/engine/runner/tools.js +0 -197
  125. package/esm/src/kernel/engine/runner.js +0 -6
  126. package/esm/src/kernel/engine/tree.d.ts +0 -2
  127. package/esm/src/kernel/engine/tree.js +0 -17
  128. package/esm/src/kernel/mod.d.ts +0 -18
  129. package/esm/src/kernel/mod.js +0 -16
  130. package/esm/src/kernel/registry/attachments.d.ts +0 -16
  131. package/esm/src/kernel/registry/attachments.js +0 -156
  132. package/esm/src/kernel/registry/catalog.d.ts +0 -31
  133. package/esm/src/kernel/registry/catalog.js +0 -135
  134. package/esm/src/kernel/registry/ingress.d.ts +0 -13
  135. package/esm/src/kernel/registry/ingress.js +0 -163
  136. package/esm/src/kernel/registry/profiles.d.ts +0 -34
  137. package/esm/src/kernel/registry/profiles.js +0 -120
  138. package/esm/src/kernel/registry/provider-request.d.ts +0 -4
  139. package/esm/src/kernel/registry/provider-request.js +0 -25
  140. package/esm/src/kernel/registry/resolve.d.ts +0 -19
  141. package/esm/src/kernel/registry/resolve.js +0 -217
  142. package/esm/src/kernel/registry/schemas.d.ts +0 -14
  143. package/esm/src/kernel/registry/schemas.js +0 -23
  144. package/esm/src/kernel/registry/tools.d.ts +0 -12
  145. package/esm/src/kernel/registry/tools.js +0 -36
  146. package/esm/src/kernel/registry/vault.js +0 -18
  147. package/esm/src/kernel/types.d.ts +0 -521
  148. package/esm/src/kernel/types.js +0 -10
  149. package/esm/src/observability/mod.d.ts +0 -12
  150. package/esm/src/observability/spans.d.ts +0 -16
  151. package/esm/src/observability/spans.js +0 -56
  152. package/esm/src/observability/trace-attach.d.ts +0 -16
  153. package/esm/src/observability/trace-attach.js +0 -67
  154. package/esm/src/observability/trace-record.d.ts +0 -112
  155. package/esm/src/observability/trace-record.js +0 -140
  156. package/esm/src/observability/trace-usage.d.ts +0 -3
  157. package/esm/src/observability/trace-usage.js +0 -32
  158. package/esm/src/observability/trace.d.ts +0 -23
  159. package/esm/src/observability/trace.js +0 -121
  160. package/esm/src/presets/google.d.ts +0 -50
  161. package/esm/src/presets/google.js +0 -96
  162. package/esm/src/presets/mod.d.ts +0 -11
  163. package/esm/src/presets/mod.js +0 -10
  164. package/esm/src/providers/create-provider.d.ts +0 -29
  165. package/esm/src/providers/create-provider.js +0 -38
  166. package/esm/src/providers/gemini-tape.d.ts +0 -2
  167. package/esm/src/providers/gemini-tape.js +0 -46
  168. package/esm/src/providers/google-tap.d.ts +0 -3
  169. package/esm/src/providers/google-tap.js +0 -48
  170. package/esm/src/providers/interactions.d.ts +0 -5
  171. package/esm/src/providers/interactions.js +0 -152
  172. package/esm/src/providers/keys.d.ts +0 -19
  173. package/esm/src/providers/keys.js +0 -129
  174. package/esm/src/providers/mod.d.ts +0 -12
  175. package/esm/src/providers/openrouter-mod.d.ts +0 -13
  176. package/esm/src/providers/openrouter-payload.d.ts +0 -32
  177. package/esm/src/providers/openrouter-payload.js +0 -179
  178. package/esm/src/providers/openrouter.d.ts +0 -15
  179. package/esm/src/providers/openrouter.js +0 -589
  180. package/esm/src/providers/pcm.d.ts +0 -7
  181. package/esm/src/providers/pcm.js +0 -35
  182. package/esm/src/providers/provider.d.ts +0 -15
  183. package/esm/src/providers/provider.js +0 -166
  184. package/esm/src/providers/speech.d.ts +0 -23
  185. package/esm/src/providers/speech.js +0 -125
  186. package/esm/src/providers/sse.d.ts +0 -7
  187. package/esm/src/providers/sse.js +0 -53
@@ -1,166 +0,0 @@
1
- /**
2
- * Google Interactions provider adapter.
3
- *
4
- * This adapter converts THEORUM's provider-neutral request into the Google
5
- * Interactions wire format and streams normalized `TurnEvent` objects.
6
- * Speech-role turns use `response_format: audio` + `speech_config` (same
7
- * transport as chat/image).
8
- *
9
- * @module
10
- */
11
- import { publicError, TheorumError } from '../guardrails/error.js';
12
- import { eventsFromComplete, eventsFromDelta, extractTokenEvent, groundingFromEvent, tryStructured, } from '../kernel/engine/delta.js';
13
- import { tapFetch } from './google-tap.js';
14
- import { toInteractionsBody } from './interactions.js';
15
- import { fetchGemini } from './keys.js';
16
- import { wrapPcmAsWav } from './pcm.js';
17
- import { INTERACTIONS_URL, takeSsePayloads } from './sse.js';
18
- const HTTP_OK = 200;
19
- function base64ToBytes(data) {
20
- const bin = atob(data);
21
- const out = new Uint8Array(bin.length);
22
- for (let i = 0; i < bin.length; i++)
23
- out[i] = bin.charCodeAt(i);
24
- return out;
25
- }
26
- function bytesToBase64(bytes) {
27
- let bin = '';
28
- for (let i = 0; i < bytes.length; i++)
29
- bin += String.fromCharCode(bytes[i]);
30
- return btoa(bin);
31
- }
32
- function isRawPcmMime(mime) {
33
- const lower = mime.toLowerCase();
34
- return lower === 'audio/pcm' || lower === 'audio/l16' || lower === 'audio/raw';
35
- }
36
- /** Google TTS returns raw PCM; wrap as WAV for hosts (matches OpenRouter pcm path). */
37
- function normalizeSpeechMedia(event, speech) {
38
- if (!(speech && event.type === 'media' && event.media)) {
39
- return event;
40
- }
41
- const { mimeType, data } = event.media;
42
- if (!isRawPcmMime(mimeType)) {
43
- return event;
44
- }
45
- const wav = wrapPcmAsWav(base64ToBytes(data));
46
- return {
47
- type: 'media',
48
- media: { mimeType: 'audio/wav', data: bytesToBase64(wav) },
49
- };
50
- }
51
- async function* readSseRecords(res) {
52
- if (!res.body) {
53
- throw new TheorumError('empty Gemini stream');
54
- }
55
- const reader = res.body.getReader();
56
- const decoder = new TextDecoder();
57
- yield* pumpSse(reader, decoder, '', '');
58
- }
59
- async function* pumpSse(reader, decoder, buffer, pendingEvent) {
60
- const { done, value } = await reader.read();
61
- if (done) {
62
- return;
63
- }
64
- const next = buffer + decoder.decode(value, { stream: true });
65
- const taken = takeSsePayloads(next, pendingEvent);
66
- for (const payload of taken.payloads) {
67
- yield payload;
68
- }
69
- yield* pumpSse(reader, decoder, taken.rest, taken.pendingEvent);
70
- }
71
- function eventType(event) {
72
- return String(event.event_type ?? event.type ?? '');
73
- }
74
- function isDeltaEvent(kind) {
75
- return kind === 'content.delta' || kind === 'step.delta';
76
- }
77
- function isCompleteEvent(kind) {
78
- return (kind === 'interaction.complete' ||
79
- kind === 'interaction.completed' ||
80
- kind.startsWith('interaction.'));
81
- }
82
- function foldDeltaPayload(event, acc) {
83
- const deltaEvents = eventsFromDelta(event.delta);
84
- for (const item of deltaEvents) {
85
- if (item.type === 'text' && item.text) {
86
- acc.text += item.text;
87
- }
88
- }
89
- return deltaEvents;
90
- }
91
- function foldPayload(event, acc) {
92
- const kind = eventType(event);
93
- const events = [];
94
- if (isDeltaEvent(kind)) {
95
- events.push(...foldDeltaPayload(event, acc));
96
- }
97
- else if (isCompleteEvent(kind)) {
98
- events.push(...eventsFromComplete(event, acc.text.length > 0));
99
- }
100
- // Also check if usage was included on the event directly
101
- if (!events.some((e) => e.type === 'tokens')) {
102
- const tokenEvent = extractTokenEvent(event);
103
- if (tokenEvent) {
104
- events.push(tokenEvent);
105
- }
106
- }
107
- const groundingEvent = groundingFromEvent(event);
108
- if (groundingEvent && !events.some((e) => e.type === 'grounding')) {
109
- events.push(groundingEvent);
110
- }
111
- return events;
112
- }
113
- function withTap(req, transport) {
114
- return {
115
- ...transport,
116
- fetch: tapFetch(req.tapGemini, transport.fetch),
117
- };
118
- }
119
- async function* streamComplete(req, transport) {
120
- if (!req.geminiBucket) {
121
- yield {
122
- type: 'error',
123
- error: publicError('missing Gemini vault bucket for Interactions'),
124
- };
125
- return;
126
- }
127
- const res = await fetchGemini(INTERACTIONS_URL, { method: 'POST', body: JSON.stringify(toInteractionsBody(req)) }, req.geminiBucket, withTap(req, transport));
128
- if (res.status !== HTTP_OK) {
129
- const errorBody = await res.text().catch(() => '');
130
- yield {
131
- type: 'error',
132
- error: publicError(`Gemini HTTP ${String(res.status)}: ${errorBody}`),
133
- };
134
- return;
135
- }
136
- const acc = { text: '' };
137
- const speech = Boolean(req.speech);
138
- for await (const payload of readSseRecords(res)) {
139
- req.tapGemini?.(payload);
140
- const events = foldPayload(payload, acc);
141
- for (const event of events) {
142
- yield normalizeSpeechMedia(event, speech);
143
- }
144
- }
145
- if (req.structured && acc.text) {
146
- const structured = tryStructured(acc.text);
147
- if (structured) {
148
- yield structured;
149
- }
150
- }
151
- }
152
- async function* streamGuarded(req, transport) {
153
- try {
154
- yield* streamComplete(req, transport);
155
- }
156
- catch (err) {
157
- yield { type: 'error', error: publicError(err) };
158
- }
159
- }
160
- /** Create a `ModelProvider` backed by Google Interactions streaming. */
161
- function createInteractionsProvider(transport) {
162
- return {
163
- complete: (req) => streamGuarded(req, transport),
164
- };
165
- }
166
- export { createInteractionsProvider };
@@ -1,23 +0,0 @@
1
- /**
2
- * OpenAI-compatible `/audio/speech` transport (internal).
3
- *
4
- * Hosts use `createProvider(profile, { openRouter })` — this module is selected
5
- * when the profile is an openAi speech role. Not a separate public door.
6
- *
7
- * @module
8
- */
9
- import type { ModelProvider, ProviderCompleteRequest, TurnEvent } from '../kernel/types.js';
10
- /** Credentials for the openAi speech path (same shape as OpenRouter chat config + voice). */
11
- export interface SpeechProviderConfig {
12
- apiKey?: string;
13
- /** Fallback TTS voice when the profile does not pin `outputs.speech.voice`. */
14
- voice?: string;
15
- baseUrl?: string;
16
- siteUrl?: string;
17
- siteName?: string;
18
- fetch?: typeof fetch;
19
- }
20
- declare function streamSpeech(req: ProviderCompleteRequest, config?: SpeechProviderConfig): AsyncGenerator<TurnEvent>;
21
- /** Internal ModelProvider for openAi speech roles. */
22
- declare function createSpeechProvider(config?: SpeechProviderConfig): ModelProvider;
23
- export { createSpeechProvider, streamSpeech };
@@ -1,125 +0,0 @@
1
- /**
2
- * OpenAI-compatible `/audio/speech` transport (internal).
3
- *
4
- * Hosts use `createProvider(profile, { openRouter })` — this module is selected
5
- * when the profile is an openAi speech role. Not a separate public door.
6
- *
7
- * @module
8
- */
9
- import { publicError } from '../guardrails/error.js';
10
- import { wrapPcmAsWav } from './pcm.js';
11
- const HTTP_OK = 200;
12
- function bytesToBase64(bytes) {
13
- let bin = '';
14
- for (let i = 0; i < bytes.length; i++)
15
- bin += String.fromCharCode(bytes[i]);
16
- return btoa(bin);
17
- }
18
- function extractInputText(input) {
19
- return input
20
- .filter((p) => p.type === 'text')
21
- .map((p) => (p.type === 'text' ? p.text : ''))
22
- .join(' ')
23
- .trim();
24
- }
25
- function resolveSpeechWireModel(req) {
26
- if (req.openRouterId) {
27
- return req.openRouterId;
28
- }
29
- if (req.apiId.includes('/')) {
30
- return req.apiId;
31
- }
32
- if (req.apiId) {
33
- return `google/${req.apiId}`;
34
- }
35
- return String(req.model);
36
- }
37
- function buildHeaders(apiKey, config) {
38
- const headers = {
39
- Authorization: `Bearer ${apiKey}`,
40
- 'Content-Type': 'application/json',
41
- };
42
- if (config.siteUrl) {
43
- headers['HTTP-Referer'] = config.siteUrl;
44
- }
45
- if (config.siteName) {
46
- headers['X-Title'] = config.siteName;
47
- }
48
- return headers;
49
- }
50
- function buildPayload(req, text, speech, configVoice) {
51
- const voice = speech?.voice ?? configVoice;
52
- const format = speech?.format ?? 'pcm';
53
- const payload = {
54
- model: resolveSpeechWireModel(req),
55
- input: text,
56
- response_format: format,
57
- };
58
- if (voice) {
59
- payload.voice = voice;
60
- }
61
- return payload;
62
- }
63
- async function requestSpeech(apiKey, text, req, config) {
64
- const fetchFn = config.fetch ?? fetch;
65
- const baseUrl = config.baseUrl?.replace(/\/+$/, '') ?? 'https://openrouter.ai/api/v1';
66
- const url = `${baseUrl}/audio/speech`;
67
- return await fetchFn(url, {
68
- method: 'POST',
69
- headers: buildHeaders(apiKey, config),
70
- body: JSON.stringify(buildPayload(req, text, req.speech, config.voice)),
71
- });
72
- }
73
- function* yieldSpeechSuccess(rawBytes, text, format) {
74
- let mediaMime = 'audio/mpeg';
75
- let mediaBytes = rawBytes;
76
- if (format === 'pcm') {
77
- mediaMime = 'audio/wav';
78
- mediaBytes = wrapPcmAsWav(rawBytes);
79
- }
80
- yield {
81
- type: 'media',
82
- media: { mimeType: mediaMime, data: bytesToBase64(mediaBytes) },
83
- };
84
- const inputTokens = Math.max(1, Math.round(text.length / 4));
85
- const outputTokens = Math.max(1, Math.round(rawBytes.length / 100));
86
- yield {
87
- type: 'tokens',
88
- tokens: { input: inputTokens, output: outputTokens, total: inputTokens + outputTokens },
89
- };
90
- yield { type: 'done' };
91
- }
92
- async function* streamSpeech(req, config = {}) {
93
- const apiKey = config.apiKey?.trim() || undefined;
94
- if (!apiKey) {
95
- yield { type: 'error', error: publicError('missing API key for speech') };
96
- return;
97
- }
98
- const text = extractInputText(req.input);
99
- if (!text) {
100
- yield { type: 'error', error: publicError('empty text for speech') };
101
- return;
102
- }
103
- const res = await requestSpeech(apiKey, text, req, config);
104
- if (res.status !== HTTP_OK) {
105
- yield { type: 'error', error: publicError(`Speech HTTP ${String(res.status)}`) };
106
- return;
107
- }
108
- const arrayBuffer = await res.arrayBuffer();
109
- const rawBytes = new Uint8Array(arrayBuffer);
110
- if (rawBytes.length === 0) {
111
- yield { type: 'error', error: publicError('no audio returned from speech') };
112
- return;
113
- }
114
- const format = req.speech?.format ?? 'pcm';
115
- for (const ev of yieldSpeechSuccess(rawBytes, text, format)) {
116
- yield ev;
117
- }
118
- }
119
- /** Internal ModelProvider for openAi speech roles. */
120
- function createSpeechProvider(config = {}) {
121
- return {
122
- complete: (req) => streamSpeech(req, config),
123
- };
124
- }
125
- export { createSpeechProvider, streamSpeech };
@@ -1,7 +0,0 @@
1
- declare const INTERACTIONS_URL = "https://generativelanguage.googleapis.com/v1beta/interactions?alt=sse";
2
- declare function takeSsePayloads(buffer: string, pendingEvent?: string): {
3
- rest: string;
4
- payloads: Record<string, unknown>[];
5
- pendingEvent: string;
6
- };
7
- export { INTERACTIONS_URL, takeSsePayloads };
@@ -1,53 +0,0 @@
1
- const INTERACTIONS_URL = 'https://generativelanguage.googleapis.com/v1beta/interactions?alt=sse';
2
- const DATA_PREFIX = 'data: ';
3
- const EVENT_PREFIX = 'event: ';
4
- const DONE = '[DONE]';
5
- function asObject(parsed) {
6
- if (parsed && typeof parsed === 'object' && !Array.isArray(parsed)) {
7
- return parsed;
8
- }
9
- return undefined;
10
- }
11
- function dataRecord(raw, sseEvent) {
12
- const data = raw.slice(DATA_PREFIX.length).trim();
13
- const row = {};
14
- if (sseEvent) {
15
- row.sseEvent = sseEvent;
16
- }
17
- if (!data || data === DONE) {
18
- row.eventType = 'sse_done';
19
- return row;
20
- }
21
- try {
22
- const parsed = JSON.parse(data);
23
- const obj = asObject(parsed);
24
- if (obj) {
25
- return { ...obj, ...row };
26
- }
27
- row.eventType = 'sse_unparsed';
28
- row.data = parsed;
29
- return row;
30
- }
31
- catch {
32
- row.eventType = 'sse_unparsed';
33
- row.data = data;
34
- return row;
35
- }
36
- }
37
- function takeSsePayloads(buffer, pendingEvent = '') {
38
- const payloads = [];
39
- const chunks = buffer.split('\n');
40
- const rest = chunks.pop() ?? '';
41
- let sseEvent = pendingEvent;
42
- for (const line of chunks) {
43
- if (line.startsWith(EVENT_PREFIX)) {
44
- sseEvent = line.slice(EVENT_PREFIX.length).trim();
45
- }
46
- else if (line.startsWith(DATA_PREFIX)) {
47
- payloads.push(dataRecord(line, sseEvent));
48
- sseEvent = '';
49
- }
50
- }
51
- return { rest, payloads, pendingEvent: sseEvent };
52
- }
53
- export { INTERACTIONS_URL, takeSsePayloads };