@ai-ecoverse/pi-audio 1.3.0 → 1.5.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.
@@ -0,0 +1,91 @@
1
+ import { VoiceRequestError } from "../errors.js";
2
+ import { SpeechEventStream } from "../speech.js";
3
+ import { priceAudio } from "../usage.js";
4
+ import {
5
+ applyPayload,
6
+ asVoiceError,
7
+ endpoint,
8
+ isOpenRouter,
9
+ isRecord,
10
+ voiceFetch
11
+ } from "./wire.js";
12
+ function sameFormat(left, right) {
13
+ return left.encoding === right.encoding && left.sampleRate === right.sampleRate && left.channels === right.channels;
14
+ }
15
+ function resolveSpeechFormat(model, requested) {
16
+ const fallback = model.outputFormats[0];
17
+ if (!fallback) throw new Error("No output format");
18
+ if (!requested) return fallback;
19
+ const match = model.outputFormats.find((format) => sameFormat(format, requested));
20
+ if (!match) throw new Error(`Unsupported speech format ${requested.encoding}`);
21
+ return match;
22
+ }
23
+ function speechResponseFormat(encoding, openRouter) {
24
+ if (encoding === "pcm16") return "pcm";
25
+ if (encoding === "mp3") return "mp3";
26
+ if (openRouter) throw new Error(`OpenRouter speech cannot return ${encoding}`);
27
+ if (encoding === "wav" || encoding === "opus" || encoding === "flac") return encoding;
28
+ throw new Error(`OpenAI speech cannot return ${encoding}`);
29
+ }
30
+ function outputSeconds(format, byteLength) {
31
+ if (format.encoding !== "pcm16") return void 0;
32
+ const width = format.sampleRate * format.channels * 2;
33
+ return width > 0 ? byteLength / width : 0;
34
+ }
35
+ async function readText(text, signal) {
36
+ if (typeof text === "string") return text;
37
+ let combined = "";
38
+ for await (const part of text) {
39
+ if (signal?.aborted) throw new VoiceRequestError("aborted", "aborted");
40
+ combined += part;
41
+ }
42
+ return combined;
43
+ }
44
+ function speakOpenAI(model, input, options) {
45
+ const stream = new SpeechEventStream();
46
+ void pumpSpeech(model, input, options, stream);
47
+ return stream;
48
+ }
49
+ async function pumpSpeech(model, input, options, stream) {
50
+ try {
51
+ if (options?.signal?.aborted) throw new DOMException("aborted", "AbortError");
52
+ const format = resolveSpeechFormat(model, input.format);
53
+ const openRouter = isOpenRouter(model);
54
+ const responseFormat = speechResponseFormat(format.encoding, openRouter);
55
+ const text = await readText(input.text, options?.signal);
56
+ const payload = {
57
+ model: model.id,
58
+ input: text,
59
+ voice: input.voice,
60
+ response_format: responseFormat
61
+ };
62
+ if (input.instructions !== void 0) payload.instructions = input.instructions;
63
+ const body = await applyPayload(payload, model, options);
64
+ if (!isRecord(body)) throw new Error("Speech payload was not JSON");
65
+ const response = await voiceFetch(
66
+ model,
67
+ options,
68
+ endpoint(model.baseUrl, "audio/speech"),
69
+ JSON.stringify(body),
70
+ "application/json"
71
+ );
72
+ const data = new Uint8Array(await response.arrayBuffer());
73
+ const seconds = outputSeconds(format, data.length);
74
+ const meters = { characters: text.length };
75
+ if (seconds !== void 0) meters.outputSeconds = seconds;
76
+ stream.push({ type: "start", format });
77
+ stream.push({
78
+ type: "audio",
79
+ chunk: { data, format, timestamp: 0 },
80
+ span: { itemId: model.id, start: 0, end: text.length }
81
+ });
82
+ stream.push({ type: "done", usage: priceAudio(model.audioCost, meters) });
83
+ } catch (error) {
84
+ const voice = asVoiceError(error);
85
+ stream.push({ type: "error", reason: voice.reason, error: voice.error });
86
+ }
87
+ }
88
+ export {
89
+ speakOpenAI
90
+ };
91
+ //# sourceMappingURL=speak.js.map
@@ -0,0 +1,7 @@
1
+ {
2
+ "version": 3,
3
+ "sources": ["../../src/openai/speak.ts"],
4
+ "sourcesContent": ["import type { ProviderRequestOptions } from '@earendil-works/pi-ai';\nimport type { AudioFormat } from '../audio.ts';\nimport { VoiceRequestError } from '../errors.ts';\nimport type { SpeechModel } from '../models.ts';\nimport { SpeechEventStream, type SpeechInput } from '../speech.ts';\nimport { type AudioCounters, priceAudio } from '../usage.ts';\nimport {\n applyPayload,\n asVoiceError,\n endpoint,\n isOpenRouter,\n isRecord,\n voiceFetch,\n} from './wire.ts';\n\nfunction sameFormat(left: AudioFormat, right: AudioFormat): boolean {\n return (\n left.encoding === right.encoding &&\n left.sampleRate === right.sampleRate &&\n left.channels === right.channels\n );\n}\n\nfunction resolveSpeechFormat(\n model: SpeechModel<string>,\n requested: AudioFormat | undefined\n): AudioFormat {\n const fallback = model.outputFormats[0];\n if (!fallback) throw new Error('No output format');\n if (!requested) return fallback;\n const match = model.outputFormats.find((format) => sameFormat(format, requested));\n if (!match) throw new Error(`Unsupported speech format ${requested.encoding}`);\n return match;\n}\n\nfunction speechResponseFormat(encoding: AudioFormat['encoding'], openRouter: boolean): string {\n if (encoding === 'pcm16') return 'pcm';\n if (encoding === 'mp3') return 'mp3';\n if (openRouter) throw new Error(`OpenRouter speech cannot return ${encoding}`);\n if (encoding === 'wav' || encoding === 'opus' || encoding === 'flac') return encoding;\n throw new Error(`OpenAI speech cannot return ${encoding}`);\n}\n\nfunction outputSeconds(format: AudioFormat, byteLength: number): number | undefined {\n if (format.encoding !== 'pcm16') return undefined;\n const width = format.sampleRate * format.channels * 2;\n return width > 0 ? byteLength / width : 0;\n}\n\nasync function readText(\n text: SpeechInput['text'],\n signal: AbortSignal | undefined\n): Promise<string> {\n if (typeof text === 'string') return text;\n let combined = '';\n for await (const part of text) {\n if (signal?.aborted) throw new VoiceRequestError('aborted', 'aborted');\n combined += part;\n }\n return combined;\n}\n\nexport function speakOpenAI(\n model: SpeechModel<string>,\n input: SpeechInput,\n options?: ProviderRequestOptions\n): SpeechEventStream {\n const stream = new SpeechEventStream();\n void pumpSpeech(model, input, options, stream);\n return stream;\n}\n\nasync function pumpSpeech(\n model: SpeechModel<string>,\n input: SpeechInput,\n options: ProviderRequestOptions | undefined,\n stream: SpeechEventStream\n) {\n try {\n if (options?.signal?.aborted) throw new DOMException('aborted', 'AbortError');\n const format = resolveSpeechFormat(model, input.format);\n const openRouter = isOpenRouter(model);\n const responseFormat = speechResponseFormat(format.encoding, openRouter);\n const text = await readText(input.text, options?.signal);\n const payload: Record<string, unknown> = {\n model: model.id,\n input: text,\n voice: input.voice,\n response_format: responseFormat,\n };\n if (input.instructions !== undefined) payload.instructions = input.instructions;\n const body = await applyPayload(payload, model, options);\n if (!isRecord(body)) throw new Error('Speech payload was not JSON');\n const response = await voiceFetch(\n model,\n options,\n endpoint(model.baseUrl, 'audio/speech'),\n JSON.stringify(body),\n 'application/json'\n );\n const data = new Uint8Array(await response.arrayBuffer());\n const seconds = outputSeconds(format, data.length);\n const meters: AudioCounters = { characters: text.length };\n if (seconds !== undefined) meters.outputSeconds = seconds;\n stream.push({ type: 'start', format });\n stream.push({\n type: 'audio',\n chunk: { data, format, timestamp: 0 },\n span: { itemId: model.id, start: 0, end: text.length },\n });\n stream.push({ type: 'done', usage: priceAudio(model.audioCost, meters) });\n } catch (error) {\n const voice = asVoiceError(error);\n stream.push({ type: 'error', reason: voice.reason, error: voice.error });\n }\n}\n"],
5
+ "mappings": "AAEA,SAAS,yBAAyB;AAElC,SAAS,yBAA2C;AACpD,SAA6B,kBAAkB;AAC/C;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAEP,SAAS,WAAW,MAAmB,OAA6B;AAClE,SACE,KAAK,aAAa,MAAM,YACxB,KAAK,eAAe,MAAM,cAC1B,KAAK,aAAa,MAAM;AAE5B;AAEA,SAAS,oBACP,OACA,WACa;AACb,QAAM,WAAW,MAAM,cAAc,CAAC;AACtC,MAAI,CAAC,SAAU,OAAM,IAAI,MAAM,kBAAkB;AACjD,MAAI,CAAC,UAAW,QAAO;AACvB,QAAM,QAAQ,MAAM,cAAc,KAAK,CAAC,WAAW,WAAW,QAAQ,SAAS,CAAC;AAChF,MAAI,CAAC,MAAO,OAAM,IAAI,MAAM,6BAA6B,UAAU,QAAQ,EAAE;AAC7E,SAAO;AACT;AAEA,SAAS,qBAAqB,UAAmC,YAA6B;AAC5F,MAAI,aAAa,QAAS,QAAO;AACjC,MAAI,aAAa,MAAO,QAAO;AAC/B,MAAI,WAAY,OAAM,IAAI,MAAM,mCAAmC,QAAQ,EAAE;AAC7E,MAAI,aAAa,SAAS,aAAa,UAAU,aAAa,OAAQ,QAAO;AAC7E,QAAM,IAAI,MAAM,+BAA+B,QAAQ,EAAE;AAC3D;AAEA,SAAS,cAAc,QAAqB,YAAwC;AAClF,MAAI,OAAO,aAAa,QAAS,QAAO;AACxC,QAAM,QAAQ,OAAO,aAAa,OAAO,WAAW;AACpD,SAAO,QAAQ,IAAI,aAAa,QAAQ;AAC1C;AAEA,eAAe,SACb,MACA,QACiB;AACjB,MAAI,OAAO,SAAS,SAAU,QAAO;AACrC,MAAI,WAAW;AACf,mBAAiB,QAAQ,MAAM;AAC7B,QAAI,QAAQ,QAAS,OAAM,IAAI,kBAAkB,WAAW,SAAS;AACrE,gBAAY;AAAA,EACd;AACA,SAAO;AACT;AAEO,SAAS,YACd,OACA,OACA,SACmB;AACnB,QAAM,SAAS,IAAI,kBAAkB;AACrC,OAAK,WAAW,OAAO,OAAO,SAAS,MAAM;AAC7C,SAAO;AACT;AAEA,eAAe,WACb,OACA,OACA,SACA,QACA;AACA,MAAI;AACF,QAAI,SAAS,QAAQ,QAAS,OAAM,IAAI,aAAa,WAAW,YAAY;AAC5E,UAAM,SAAS,oBAAoB,OAAO,MAAM,MAAM;AACtD,UAAM,aAAa,aAAa,KAAK;AACrC,UAAM,iBAAiB,qBAAqB,OAAO,UAAU,UAAU;AACvE,UAAM,OAAO,MAAM,SAAS,MAAM,MAAM,SAAS,MAAM;AACvD,UAAM,UAAmC;AAAA,MACvC,OAAO,MAAM;AAAA,MACb,OAAO;AAAA,MACP,OAAO,MAAM;AAAA,MACb,iBAAiB;AAAA,IACnB;AACA,QAAI,MAAM,iBAAiB,OAAW,SAAQ,eAAe,MAAM;AACnE,UAAM,OAAO,MAAM,aAAa,SAAS,OAAO,OAAO;AACvD,QAAI,CAAC,SAAS,IAAI,EAAG,OAAM,IAAI,MAAM,6BAA6B;AAClE,UAAM,WAAW,MAAM;AAAA,MACrB;AAAA,MACA;AAAA,MACA,SAAS,MAAM,SAAS,cAAc;AAAA,MACtC,KAAK,UAAU,IAAI;AAAA,MACnB;AAAA,IACF;AACA,UAAM,OAAO,IAAI,WAAW,MAAM,SAAS,YAAY,CAAC;AACxD,UAAM,UAAU,cAAc,QAAQ,KAAK,MAAM;AACjD,UAAM,SAAwB,EAAE,YAAY,KAAK,OAAO;AACxD,QAAI,YAAY,OAAW,QAAO,gBAAgB;AAClD,WAAO,KAAK,EAAE,MAAM,SAAS,OAAO,CAAC;AACrC,WAAO,KAAK;AAAA,MACV,MAAM;AAAA,MACN,OAAO,EAAE,MAAM,QAAQ,WAAW,EAAE;AAAA,MACpC,MAAM,EAAE,QAAQ,MAAM,IAAI,OAAO,GAAG,KAAK,KAAK,OAAO;AAAA,IACvD,CAAC;AACD,WAAO,KAAK,EAAE,MAAM,QAAQ,OAAO,WAAW,MAAM,WAAW,MAAM,EAAE,CAAC;AAAA,EAC1E,SAAS,OAAO;AACd,UAAM,QAAQ,aAAa,KAAK;AAChC,WAAO,KAAK,EAAE,MAAM,SAAS,QAAQ,MAAM,QAAQ,OAAO,MAAM,MAAM,CAAC;AAAA,EACzE;AACF;",
6
+ "names": []
7
+ }
@@ -0,0 +1,112 @@
1
+ import { VoiceRequestError } from "../errors.js";
2
+ import { priceAudio } from "../usage.js";
3
+ import {
4
+ applyPayload,
5
+ asVoiceError,
6
+ audioFile,
7
+ audioPayload,
8
+ encodeMultipart,
9
+ endpoint,
10
+ isOpenRouter,
11
+ isRecord,
12
+ multipartType,
13
+ numberField,
14
+ readAudio,
15
+ readJson,
16
+ requestSignal,
17
+ stringField,
18
+ voiceFetch
19
+ } from "./wire.js";
20
+ function durationSeconds(record) {
21
+ const usage = record.usage;
22
+ if (isRecord(usage)) {
23
+ const seconds = numberField(usage, "seconds");
24
+ if (seconds !== void 0) return seconds;
25
+ }
26
+ const duration = numberField(record, "duration");
27
+ return duration === void 0 ? 0 : duration;
28
+ }
29
+ function fieldText(segment) {
30
+ const text = stringField(segment, "text");
31
+ return text === void 0 ? "" : text;
32
+ }
33
+ function fieldNumber(segment, key) {
34
+ const value = numberField(segment, key);
35
+ return value === void 0 ? 0 : value;
36
+ }
37
+ function segmentFrom(segment) {
38
+ const speaker = stringField(segment, "speaker");
39
+ const mapped = {
40
+ text: fieldText(segment),
41
+ start: fieldNumber(segment, "start"),
42
+ end: fieldNumber(segment, "end")
43
+ };
44
+ if (speaker !== void 0) mapped.speaker = speaker;
45
+ return mapped;
46
+ }
47
+ function transcriptFrom(body, model, seconds) {
48
+ if (!isRecord(body)) throw new Error("Transcription response had no text");
49
+ const text = stringField(body, "text");
50
+ if (text === void 0) throw new Error("Transcription response had no text");
51
+ const language = stringField(body, "language");
52
+ const segments = Array.isArray(body.segments) ? body.segments.filter(isRecord).map(segmentFrom) : void 0;
53
+ return {
54
+ text,
55
+ ...language !== void 0 ? { language } : {},
56
+ ...segments !== void 0 ? { segments } : {},
57
+ usage: priceAudio(model.audioCost, { inputSeconds: seconds })
58
+ };
59
+ }
60
+ function fileFrom(loaded) {
61
+ const named = audioFile(loaded.mimeType);
62
+ return { data: loaded.data, mimeType: loaded.mimeType, ...named };
63
+ }
64
+ async function transcribeOpenAI(model, input, options) {
65
+ try {
66
+ if (options?.signal?.aborted) throw new DOMException("aborted", "AbortError");
67
+ const fetchImpl = options?.fetch ?? globalThis.fetch;
68
+ const loaded = await readAudio(input, fetchImpl, requestSignal(options));
69
+ const file = fileFrom(loaded);
70
+ const payload = audioPayload(file, isOpenRouter(model), model.id);
71
+ const body = await applyPayload(payload, model, options);
72
+ if (!isRecord(body)) throw new Error("Transcription payload was not JSON");
73
+ const url = endpoint(model.baseUrl, "audio/transcriptions");
74
+ const response = await postTranscription(model, options, url, body);
75
+ const json = await readJson(response);
76
+ const seconds = isRecord(json) ? durationSeconds(json) : 0;
77
+ return transcriptFrom(json, model, seconds);
78
+ } catch (error) {
79
+ const voice = asVoiceError(error);
80
+ throw new VoiceRequestError(voice.reason, voice.error);
81
+ }
82
+ }
83
+ async function postTranscription(model, options, url, body) {
84
+ const inline = body.input_audio;
85
+ if (isRecord(inline)) {
86
+ return voiceFetch(model, options, url, JSON.stringify(body), "application/json");
87
+ }
88
+ const file = body.file;
89
+ if (!isRecord(file) || !(file.data instanceof Uint8Array) || typeof file.mimeType !== "string") {
90
+ throw new Error("Transcription payload was not JSON");
91
+ }
92
+ const named = audioFile(file.mimeType);
93
+ const fields = {};
94
+ const modelId = stringField(body, "model");
95
+ fields.model = modelId === void 0 ? model.id : modelId;
96
+ const encoded = encodeMultipart(
97
+ {
98
+ data: file.data,
99
+ mimeType: file.mimeType,
100
+ filename: typeof file.filename === "string" ? file.filename : named.filename,
101
+ format: named.format
102
+ },
103
+ fields
104
+ );
105
+ const copy = new ArrayBuffer(encoded.byteLength);
106
+ new Uint8Array(copy).set(encoded);
107
+ return voiceFetch(model, options, url, copy, multipartType());
108
+ }
109
+ export {
110
+ transcribeOpenAI
111
+ };
112
+ //# sourceMappingURL=transcribe.js.map
@@ -0,0 +1,7 @@
1
+ {
2
+ "version": 3,
3
+ "sources": ["../../src/openai/transcribe.ts"],
4
+ "sourcesContent": ["import type { ProviderRequestOptions } from '@earendil-works/pi-ai';\nimport { VoiceRequestError } from '../errors.ts';\nimport type { TranscriptionModel } from '../models.ts';\nimport type { AudioInput, Transcript, TranscriptSegment } from '../transcription.ts';\nimport { priceAudio } from '../usage.ts';\nimport {\n type AudioFile,\n applyPayload,\n asVoiceError,\n audioFile,\n audioPayload,\n encodeMultipart,\n endpoint,\n isOpenRouter,\n isRecord,\n multipartType,\n numberField,\n readAudio,\n readJson,\n requestSignal,\n stringField,\n voiceFetch,\n} from './wire.ts';\n\nfunction durationSeconds(record: Record<string, unknown>): number {\n const usage = record.usage;\n if (isRecord(usage)) {\n const seconds = numberField(usage, 'seconds');\n if (seconds !== undefined) return seconds;\n }\n const duration = numberField(record, 'duration');\n return duration === undefined ? 0 : duration;\n}\n\nfunction fieldText(segment: Record<string, unknown>): string {\n const text = stringField(segment, 'text');\n return text === undefined ? '' : text;\n}\n\nfunction fieldNumber(segment: Record<string, unknown>, key: string): number {\n const value = numberField(segment, key);\n return value === undefined ? 0 : value;\n}\n\nfunction segmentFrom(segment: Record<string, unknown>): TranscriptSegment {\n const speaker = stringField(segment, 'speaker');\n const mapped: TranscriptSegment = {\n text: fieldText(segment),\n start: fieldNumber(segment, 'start'),\n end: fieldNumber(segment, 'end'),\n };\n if (speaker !== undefined) mapped.speaker = speaker;\n return mapped;\n}\n\nfunction transcriptFrom(\n body: unknown,\n model: TranscriptionModel<string>,\n seconds: number\n): Transcript {\n if (!isRecord(body)) throw new Error('Transcription response had no text');\n const text = stringField(body, 'text');\n if (text === undefined) throw new Error('Transcription response had no text');\n const language = stringField(body, 'language');\n const segments = Array.isArray(body.segments)\n ? body.segments.filter(isRecord).map(segmentFrom)\n : undefined;\n return {\n text,\n ...(language !== undefined ? { language } : {}),\n ...(segments !== undefined ? { segments } : {}),\n usage: priceAudio(model.audioCost, { inputSeconds: seconds }),\n };\n}\n\nfunction fileFrom(loaded: { data: Uint8Array; mimeType: string }): AudioFile {\n const named = audioFile(loaded.mimeType);\n return { data: loaded.data, mimeType: loaded.mimeType, ...named };\n}\n\nexport async function transcribeOpenAI(\n model: TranscriptionModel<string>,\n input: AudioInput,\n options?: ProviderRequestOptions\n): Promise<Transcript> {\n try {\n if (options?.signal?.aborted) throw new DOMException('aborted', 'AbortError');\n const fetchImpl = options?.fetch ?? globalThis.fetch;\n const loaded = await readAudio(input, fetchImpl, requestSignal(options));\n const file = fileFrom(loaded);\n const payload = audioPayload(file, isOpenRouter(model), model.id);\n const body = await applyPayload(payload, model, options);\n if (!isRecord(body)) throw new Error('Transcription payload was not JSON');\n const url = endpoint(model.baseUrl, 'audio/transcriptions');\n const response = await postTranscription(model, options, url, body);\n const json = await readJson(response);\n const seconds = isRecord(json) ? durationSeconds(json) : 0;\n return transcriptFrom(json, model, seconds);\n } catch (error) {\n const voice = asVoiceError(error);\n throw new VoiceRequestError(voice.reason, voice.error);\n }\n}\n\nasync function postTranscription(\n model: TranscriptionModel<string>,\n options: ProviderRequestOptions | undefined,\n url: string,\n body: Record<string, unknown>\n): Promise<Response> {\n const inline = body.input_audio;\n if (isRecord(inline)) {\n return voiceFetch(model, options, url, JSON.stringify(body), 'application/json');\n }\n const file = body.file;\n if (!isRecord(file) || !(file.data instanceof Uint8Array) || typeof file.mimeType !== 'string') {\n throw new Error('Transcription payload was not JSON');\n }\n const named = audioFile(file.mimeType);\n const fields: Record<string, string> = {};\n const modelId = stringField(body, 'model');\n fields.model = modelId === undefined ? model.id : modelId;\n const encoded = encodeMultipart(\n {\n data: file.data,\n mimeType: file.mimeType,\n filename: typeof file.filename === 'string' ? file.filename : named.filename,\n format: named.format,\n },\n fields\n );\n const copy = new ArrayBuffer(encoded.byteLength);\n new Uint8Array(copy).set(encoded);\n return voiceFetch(model, options, url, copy, multipartType());\n}\n"],
5
+ "mappings": "AACA,SAAS,yBAAyB;AAGlC,SAAS,kBAAkB;AAC3B;AAAA,EAEE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAEP,SAAS,gBAAgB,QAAyC;AAChE,QAAM,QAAQ,OAAO;AACrB,MAAI,SAAS,KAAK,GAAG;AACnB,UAAM,UAAU,YAAY,OAAO,SAAS;AAC5C,QAAI,YAAY,OAAW,QAAO;AAAA,EACpC;AACA,QAAM,WAAW,YAAY,QAAQ,UAAU;AAC/C,SAAO,aAAa,SAAY,IAAI;AACtC;AAEA,SAAS,UAAU,SAA0C;AAC3D,QAAM,OAAO,YAAY,SAAS,MAAM;AACxC,SAAO,SAAS,SAAY,KAAK;AACnC;AAEA,SAAS,YAAY,SAAkC,KAAqB;AAC1E,QAAM,QAAQ,YAAY,SAAS,GAAG;AACtC,SAAO,UAAU,SAAY,IAAI;AACnC;AAEA,SAAS,YAAY,SAAqD;AACxE,QAAM,UAAU,YAAY,SAAS,SAAS;AAC9C,QAAM,SAA4B;AAAA,IAChC,MAAM,UAAU,OAAO;AAAA,IACvB,OAAO,YAAY,SAAS,OAAO;AAAA,IACnC,KAAK,YAAY,SAAS,KAAK;AAAA,EACjC;AACA,MAAI,YAAY,OAAW,QAAO,UAAU;AAC5C,SAAO;AACT;AAEA,SAAS,eACP,MACA,OACA,SACY;AACZ,MAAI,CAAC,SAAS,IAAI,EAAG,OAAM,IAAI,MAAM,oCAAoC;AACzE,QAAM,OAAO,YAAY,MAAM,MAAM;AACrC,MAAI,SAAS,OAAW,OAAM,IAAI,MAAM,oCAAoC;AAC5E,QAAM,WAAW,YAAY,MAAM,UAAU;AAC7C,QAAM,WAAW,MAAM,QAAQ,KAAK,QAAQ,IACxC,KAAK,SAAS,OAAO,QAAQ,EAAE,IAAI,WAAW,IAC9C;AACJ,SAAO;AAAA,IACL;AAAA,IACA,GAAI,aAAa,SAAY,EAAE,SAAS,IAAI,CAAC;AAAA,IAC7C,GAAI,aAAa,SAAY,EAAE,SAAS,IAAI,CAAC;AAAA,IAC7C,OAAO,WAAW,MAAM,WAAW,EAAE,cAAc,QAAQ,CAAC;AAAA,EAC9D;AACF;AAEA,SAAS,SAAS,QAA2D;AAC3E,QAAM,QAAQ,UAAU,OAAO,QAAQ;AACvC,SAAO,EAAE,MAAM,OAAO,MAAM,UAAU,OAAO,UAAU,GAAG,MAAM;AAClE;AAEA,eAAsB,iBACpB,OACA,OACA,SACqB;AACrB,MAAI;AACF,QAAI,SAAS,QAAQ,QAAS,OAAM,IAAI,aAAa,WAAW,YAAY;AAC5E,UAAM,YAAY,SAAS,SAAS,WAAW;AAC/C,UAAM,SAAS,MAAM,UAAU,OAAO,WAAW,cAAc,OAAO,CAAC;AACvE,UAAM,OAAO,SAAS,MAAM;AAC5B,UAAM,UAAU,aAAa,MAAM,aAAa,KAAK,GAAG,MAAM,EAAE;AAChE,UAAM,OAAO,MAAM,aAAa,SAAS,OAAO,OAAO;AACvD,QAAI,CAAC,SAAS,IAAI,EAAG,OAAM,IAAI,MAAM,oCAAoC;AACzE,UAAM,MAAM,SAAS,MAAM,SAAS,sBAAsB;AAC1D,UAAM,WAAW,MAAM,kBAAkB,OAAO,SAAS,KAAK,IAAI;AAClE,UAAM,OAAO,MAAM,SAAS,QAAQ;AACpC,UAAM,UAAU,SAAS,IAAI,IAAI,gBAAgB,IAAI,IAAI;AACzD,WAAO,eAAe,MAAM,OAAO,OAAO;AAAA,EAC5C,SAAS,OAAO;AACd,UAAM,QAAQ,aAAa,KAAK;AAChC,UAAM,IAAI,kBAAkB,MAAM,QAAQ,MAAM,KAAK;AAAA,EACvD;AACF;AAEA,eAAe,kBACb,OACA,SACA,KACA,MACmB;AACnB,QAAM,SAAS,KAAK;AACpB,MAAI,SAAS,MAAM,GAAG;AACpB,WAAO,WAAW,OAAO,SAAS,KAAK,KAAK,UAAU,IAAI,GAAG,kBAAkB;AAAA,EACjF;AACA,QAAM,OAAO,KAAK;AAClB,MAAI,CAAC,SAAS,IAAI,KAAK,EAAE,KAAK,gBAAgB,eAAe,OAAO,KAAK,aAAa,UAAU;AAC9F,UAAM,IAAI,MAAM,oCAAoC;AAAA,EACtD;AACA,QAAM,QAAQ,UAAU,KAAK,QAAQ;AACrC,QAAM,SAAiC,CAAC;AACxC,QAAM,UAAU,YAAY,MAAM,OAAO;AACzC,SAAO,QAAQ,YAAY,SAAY,MAAM,KAAK;AAClD,QAAM,UAAU;AAAA,IACd;AAAA,MACE,MAAM,KAAK;AAAA,MACX,UAAU,KAAK;AAAA,MACf,UAAU,OAAO,KAAK,aAAa,WAAW,KAAK,WAAW,MAAM;AAAA,MACpE,QAAQ,MAAM;AAAA,IAChB;AAAA,IACA;AAAA,EACF;AACA,QAAM,OAAO,IAAI,YAAY,QAAQ,UAAU;AAC/C,MAAI,WAAW,IAAI,EAAE,IAAI,OAAO;AAChC,SAAO,WAAW,OAAO,SAAS,KAAK,MAAM,cAAc,CAAC;AAC9D;",
6
+ "names": []
7
+ }
@@ -0,0 +1,201 @@
1
+ import { encodeAudioBase64 } from "../convert.js";
2
+ import { VoiceRequestError } from "../errors.js";
3
+ const boundary = "pi-audio-boundary";
4
+ const mimeFiles = {
5
+ "audio/wav": { format: "wav", filename: "audio.wav" },
6
+ "audio/wave": { format: "wav", filename: "audio.wav" },
7
+ "audio/x-wav": { format: "wav", filename: "audio.wav" },
8
+ "audio/mpeg": { format: "mp3", filename: "audio.mp3" },
9
+ "audio/mp3": { format: "mp3", filename: "audio.mp3" },
10
+ "audio/flac": { format: "flac", filename: "audio.flac" },
11
+ "audio/ogg": { format: "ogg", filename: "audio.ogg" },
12
+ "audio/opus": { format: "ogg", filename: "audio.ogg" },
13
+ "audio/webm": { format: "webm", filename: "audio.webm" },
14
+ "audio/mp4": { format: "m4a", filename: "audio.m4a" },
15
+ "audio/m4a": { format: "m4a", filename: "audio.m4a" },
16
+ "audio/aac": { format: "aac", filename: "audio.aac" },
17
+ "audio/pcm": { format: "pcm", filename: "audio.pcm" }
18
+ };
19
+ function isOpenRouter(model) {
20
+ return model.provider === "openrouter" || model.baseUrl.includes("openrouter.ai");
21
+ }
22
+ function endpoint(baseUrl, path) {
23
+ return `${baseUrl.replace(/\/$/, "")}/${path}`;
24
+ }
25
+ function realtimeUrl(baseUrl) {
26
+ const url = new URL(baseUrl);
27
+ url.protocol = url.protocol === "https:" ? "wss:" : "ws:";
28
+ url.pathname = `${url.pathname.replace(/\/$/, "")}/realtime`;
29
+ url.search = "intent=transcription";
30
+ return url.toString();
31
+ }
32
+ function audioFile(mimeType) {
33
+ const found = mimeFiles[mimeType.toLowerCase()];
34
+ if (!found) throw new Error(`Unsupported audio type ${mimeType}`);
35
+ return found;
36
+ }
37
+ function isRecord(value) {
38
+ return typeof value === "object" && value !== null;
39
+ }
40
+ function stringField(record, key) {
41
+ const value = record[key];
42
+ return typeof value === "string" ? value : void 0;
43
+ }
44
+ function numberField(record, key) {
45
+ const value = record[key];
46
+ return typeof value === "number" ? value : void 0;
47
+ }
48
+ function asVoiceError(error) {
49
+ if (error instanceof VoiceRequestError) return { reason: error.reason, error: error.message };
50
+ if (error instanceof Error && error.name === "AbortError")
51
+ return { reason: "aborted", error: "aborted" };
52
+ if (error instanceof Error) return { reason: "error", error: error.message };
53
+ return { reason: "error", error: String(error) };
54
+ }
55
+ function headerValue(headers, name) {
56
+ if (!headers) return void 0;
57
+ const expected = name.toLowerCase();
58
+ for (const [key, value] of Object.entries(headers)) {
59
+ if (key.toLowerCase() === expected && value !== null && value.trim() !== "") return value;
60
+ }
61
+ return void 0;
62
+ }
63
+ function hasAuth(headers) {
64
+ return headerValue(headers, "authorization") !== void 0 || headerValue(headers, "cf-aig-authorization") !== void 0;
65
+ }
66
+ function requestHeaders(model, options) {
67
+ const headers = { ...model.headers };
68
+ const extra = options?.headers;
69
+ if (extra) {
70
+ for (const [key, value] of Object.entries(extra)) {
71
+ if (value === null) delete headers[key];
72
+ else headers[key] = value;
73
+ }
74
+ }
75
+ if (!hasAuth(headers)) {
76
+ const apiKey = options?.apiKey;
77
+ if (!apiKey) throw new Error(`No API key for provider: ${model.provider}`);
78
+ headers.Authorization = `Bearer ${apiKey}`;
79
+ }
80
+ return headers;
81
+ }
82
+ function requestSignal(options) {
83
+ const signals = [];
84
+ if (options?.signal) signals.push(options.signal);
85
+ const timeout = options?.timeoutMs;
86
+ if (timeout !== void 0 && timeout > 0) signals.push(AbortSignal.timeout(timeout));
87
+ if (signals.length === 0) return void 0;
88
+ if (signals.length === 1) return signals[0];
89
+ return AbortSignal.any(signals);
90
+ }
91
+ function encodeMultipart(file, fields) {
92
+ const chunks = [];
93
+ const encoder = new TextEncoder();
94
+ const write = (value) => chunks.push(encoder.encode(value));
95
+ write(
96
+ `--${boundary}\r
97
+ Content-Disposition: form-data; name="file"; filename="${file.filename}"\r
98
+ Content-Type: ${file.mimeType}\r
99
+ \r
100
+ `
101
+ );
102
+ chunks.push(file.data);
103
+ write("\r\n");
104
+ for (const [name, value] of Object.entries(fields)) {
105
+ write(`--${boundary}\r
106
+ Content-Disposition: form-data; name="${name}"\r
107
+ \r
108
+ ${value}\r
109
+ `);
110
+ }
111
+ write(`--${boundary}--\r
112
+ `);
113
+ const body = new Uint8Array(chunks.reduce((sum, chunk) => sum + chunk.length, 0));
114
+ let offset = 0;
115
+ for (const chunk of chunks) {
116
+ body.set(chunk, offset);
117
+ offset += chunk.length;
118
+ }
119
+ return body;
120
+ }
121
+ function multipartType() {
122
+ return `multipart/form-data; boundary=${boundary}`;
123
+ }
124
+ async function readAudio(input, fetchImpl, signal) {
125
+ if (!("url" in input)) return { data: input.data, mimeType: input.mimeType };
126
+ const response = await fetchImpl(input.url, { signal });
127
+ if (!response.ok) throw new Error(`Audio URL failed: HTTP ${response.status}`);
128
+ const mime = response.headers.get("content-type");
129
+ const mimeType = mime === null || mime.trim() === "" ? "audio/wav" : mime.split(";")[0].trim();
130
+ return { data: new Uint8Array(await response.arrayBuffer()), mimeType };
131
+ }
132
+ function audioPayload(file, openRouter, modelId) {
133
+ if (openRouter) {
134
+ return {
135
+ model: modelId,
136
+ input_audio: { data: encodeAudioBase64(file.data), format: file.format }
137
+ };
138
+ }
139
+ return { model: modelId, file };
140
+ }
141
+ async function applyPayload(payload, model, options) {
142
+ const hook = options?.onPayload;
143
+ const replaced = await hook?.(payload, model);
144
+ return replaced === void 0 ? payload : replaced;
145
+ }
146
+ function headerRecord(headers) {
147
+ const record = {};
148
+ headers.forEach((value, key) => {
149
+ record[key] = value;
150
+ });
151
+ return record;
152
+ }
153
+ async function voiceFetch(model, options, url, body, contentType) {
154
+ const headers = requestHeaders(model, options);
155
+ headers["Content-Type"] = contentType;
156
+ const fetchImpl = options?.fetch ?? globalThis.fetch;
157
+ const response = await fetchImpl(url, {
158
+ method: "POST",
159
+ headers,
160
+ body,
161
+ signal: requestSignal(options)
162
+ });
163
+ await options?.onResponse?.(
164
+ { status: response.status, headers: headerRecord(response.headers) },
165
+ model
166
+ );
167
+ if (!response.ok) {
168
+ const text = await response.text();
169
+ throw new Error(text === "" ? `HTTP ${response.status}` : text);
170
+ }
171
+ return response;
172
+ }
173
+ async function readJson(response) {
174
+ const text = await response.text();
175
+ try {
176
+ return JSON.parse(text);
177
+ } catch {
178
+ throw new Error("Transcription response was not JSON");
179
+ }
180
+ }
181
+ export {
182
+ applyPayload,
183
+ asVoiceError,
184
+ audioFile,
185
+ audioPayload,
186
+ encodeMultipart,
187
+ endpoint,
188
+ hasAuth,
189
+ isOpenRouter,
190
+ isRecord,
191
+ multipartType,
192
+ numberField,
193
+ readAudio,
194
+ readJson,
195
+ realtimeUrl,
196
+ requestHeaders,
197
+ requestSignal,
198
+ stringField,
199
+ voiceFetch
200
+ };
201
+ //# sourceMappingURL=wire.js.map
@@ -0,0 +1,7 @@
1
+ {
2
+ "version": 3,
3
+ "sources": ["../../src/openai/wire.ts"],
4
+ "sourcesContent": ["import type { ProviderRequestOptions } from '@earendil-works/pi-ai';\nimport { encodeAudioBase64 } from '../convert.ts';\nimport { VoiceRequestError } from '../errors.ts';\n\nconst boundary = 'pi-audio-boundary';\n\nconst mimeFiles: Record<string, { format: string; filename: string }> = {\n 'audio/wav': { format: 'wav', filename: 'audio.wav' },\n 'audio/wave': { format: 'wav', filename: 'audio.wav' },\n 'audio/x-wav': { format: 'wav', filename: 'audio.wav' },\n 'audio/mpeg': { format: 'mp3', filename: 'audio.mp3' },\n 'audio/mp3': { format: 'mp3', filename: 'audio.mp3' },\n 'audio/flac': { format: 'flac', filename: 'audio.flac' },\n 'audio/ogg': { format: 'ogg', filename: 'audio.ogg' },\n 'audio/opus': { format: 'ogg', filename: 'audio.ogg' },\n 'audio/webm': { format: 'webm', filename: 'audio.webm' },\n 'audio/mp4': { format: 'm4a', filename: 'audio.m4a' },\n 'audio/m4a': { format: 'm4a', filename: 'audio.m4a' },\n 'audio/aac': { format: 'aac', filename: 'audio.aac' },\n 'audio/pcm': { format: 'pcm', filename: 'audio.pcm' },\n};\n\nexport interface AudioFile {\n data: Uint8Array;\n mimeType: string;\n filename: string;\n format: string;\n}\n\nexport interface JsonBody {\n [key: string]: unknown;\n}\n\ninterface HeaderModel {\n provider: string;\n headers?: Record<string, string>;\n}\n\nexport function isOpenRouter(model: { provider: string; baseUrl: string }): boolean {\n return model.provider === 'openrouter' || model.baseUrl.includes('openrouter.ai');\n}\n\nexport function endpoint(baseUrl: string, path: string): string {\n return `${baseUrl.replace(/\\/$/, '')}/${path}`;\n}\n\nexport function realtimeUrl(baseUrl: string): string {\n const url = new URL(baseUrl);\n url.protocol = url.protocol === 'https:' ? 'wss:' : 'ws:';\n url.pathname = `${url.pathname.replace(/\\/$/, '')}/realtime`;\n url.search = 'intent=transcription';\n return url.toString();\n}\n\nexport function audioFile(mimeType: string): { format: string; filename: string } {\n const found = mimeFiles[mimeType.toLowerCase()];\n if (!found) throw new Error(`Unsupported audio type ${mimeType}`);\n return found;\n}\n\nexport function isRecord(value: unknown): value is Record<string, unknown> {\n return typeof value === 'object' && value !== null;\n}\n\nexport function stringField(record: Record<string, unknown>, key: string): string | undefined {\n const value = record[key];\n return typeof value === 'string' ? value : undefined;\n}\n\nexport function numberField(record: Record<string, unknown>, key: string): number | undefined {\n const value = record[key];\n return typeof value === 'number' ? value : undefined;\n}\n\nexport function asVoiceError(error: unknown): { reason: 'aborted' | 'error'; error: string } {\n if (error instanceof VoiceRequestError) return { reason: error.reason, error: error.message };\n if (error instanceof Error && error.name === 'AbortError')\n return { reason: 'aborted', error: 'aborted' };\n if (error instanceof Error) return { reason: 'error', error: error.message };\n return { reason: 'error', error: String(error) };\n}\n\nfunction headerValue(\n headers: Record<string, string | null> | undefined,\n name: string\n): string | undefined {\n if (!headers) return undefined;\n const expected = name.toLowerCase();\n for (const [key, value] of Object.entries(headers)) {\n if (key.toLowerCase() === expected && value !== null && value.trim() !== '') return value;\n }\n return undefined;\n}\n\nexport function hasAuth(headers: Record<string, string | null> | undefined): boolean {\n return (\n headerValue(headers, 'authorization') !== undefined ||\n headerValue(headers, 'cf-aig-authorization') !== undefined\n );\n}\n\nexport function requestHeaders(\n model: HeaderModel,\n options: ProviderRequestOptions | undefined\n): Record<string, string> {\n const headers: Record<string, string> = { ...model.headers };\n const extra = options?.headers;\n if (extra) {\n for (const [key, value] of Object.entries(extra)) {\n if (value === null) delete headers[key];\n else headers[key] = value;\n }\n }\n if (!hasAuth(headers)) {\n const apiKey = options?.apiKey;\n if (!apiKey) throw new Error(`No API key for provider: ${model.provider}`);\n headers.Authorization = `Bearer ${apiKey}`;\n }\n return headers;\n}\n\nexport function requestSignal(\n options: ProviderRequestOptions | undefined\n): AbortSignal | undefined {\n const signals: AbortSignal[] = [];\n if (options?.signal) signals.push(options.signal);\n const timeout = options?.timeoutMs;\n if (timeout !== undefined && timeout > 0) signals.push(AbortSignal.timeout(timeout));\n if (signals.length === 0) return undefined;\n if (signals.length === 1) return signals[0];\n return AbortSignal.any(signals);\n}\n\nexport function encodeMultipart(file: AudioFile, fields: Record<string, string>): Uint8Array {\n const chunks: Uint8Array[] = [];\n const encoder = new TextEncoder();\n const write = (value: string) => chunks.push(encoder.encode(value));\n write(\n `--${boundary}\\r\\nContent-Disposition: form-data; name=\"file\"; filename=\"${file.filename}\"\\r\\nContent-Type: ${file.mimeType}\\r\\n\\r\\n`\n );\n chunks.push(file.data);\n write('\\r\\n');\n for (const [name, value] of Object.entries(fields)) {\n write(`--${boundary}\\r\\nContent-Disposition: form-data; name=\"${name}\"\\r\\n\\r\\n${value}\\r\\n`);\n }\n write(`--${boundary}--\\r\\n`);\n const body = new Uint8Array(chunks.reduce((sum, chunk) => sum + chunk.length, 0));\n let offset = 0;\n for (const chunk of chunks) {\n body.set(chunk, offset);\n offset += chunk.length;\n }\n return body;\n}\n\nexport function multipartType(): string {\n return `multipart/form-data; boundary=${boundary}`;\n}\n\nexport async function readAudio(\n input: { data: Uint8Array; mimeType: string } | { url: string },\n fetchImpl: typeof fetch,\n signal: AbortSignal | undefined\n): Promise<{ data: Uint8Array; mimeType: string }> {\n if (!('url' in input)) return { data: input.data, mimeType: input.mimeType };\n const response = await fetchImpl(input.url, { signal });\n if (!response.ok) throw new Error(`Audio URL failed: HTTP ${response.status}`);\n const mime = response.headers.get('content-type');\n const mimeType = mime === null || mime.trim() === '' ? 'audio/wav' : mime.split(';')[0].trim();\n return { data: new Uint8Array(await response.arrayBuffer()), mimeType };\n}\n\nexport function audioPayload(file: AudioFile, openRouter: boolean, modelId: string): JsonBody {\n if (openRouter) {\n return {\n model: modelId,\n input_audio: { data: encodeAudioBase64(file.data), format: file.format },\n };\n }\n return { model: modelId, file };\n}\n\nexport async function applyPayload(\n payload: unknown,\n model: unknown,\n options: ProviderRequestOptions | undefined\n): Promise<unknown> {\n const hook = options?.onPayload as\n | ((body: unknown, voiceModel: unknown) => unknown | undefined | Promise<unknown | undefined>)\n | undefined;\n const replaced = await hook?.(payload, model);\n return replaced === undefined ? payload : replaced;\n}\n\nfunction headerRecord(headers: Headers): Record<string, string> {\n const record: Record<string, string> = {};\n headers.forEach((value, key) => {\n record[key] = value;\n });\n return record;\n}\n\nexport async function voiceFetch(\n model: HeaderModel & { id: string },\n options: ProviderRequestOptions | undefined,\n url: string,\n body: BodyInit,\n contentType: string\n): Promise<Response> {\n const headers = requestHeaders(model, options);\n headers['Content-Type'] = contentType;\n const fetchImpl = options?.fetch ?? globalThis.fetch;\n const response = await fetchImpl(url, {\n method: 'POST',\n headers,\n body,\n signal: requestSignal(options),\n });\n await options?.onResponse?.(\n { status: response.status, headers: headerRecord(response.headers) },\n model as never\n );\n if (!response.ok) {\n const text = await response.text();\n throw new Error(text === '' ? `HTTP ${response.status}` : text);\n }\n return response;\n}\n\nexport async function readJson(response: Response): Promise<unknown> {\n const text = await response.text();\n try {\n return JSON.parse(text);\n } catch {\n throw new Error('Transcription response was not JSON');\n }\n}\n"],
5
+ "mappings": "AACA,SAAS,yBAAyB;AAClC,SAAS,yBAAyB;AAElC,MAAM,WAAW;AAEjB,MAAM,YAAkE;AAAA,EACtE,aAAa,EAAE,QAAQ,OAAO,UAAU,YAAY;AAAA,EACpD,cAAc,EAAE,QAAQ,OAAO,UAAU,YAAY;AAAA,EACrD,eAAe,EAAE,QAAQ,OAAO,UAAU,YAAY;AAAA,EACtD,cAAc,EAAE,QAAQ,OAAO,UAAU,YAAY;AAAA,EACrD,aAAa,EAAE,QAAQ,OAAO,UAAU,YAAY;AAAA,EACpD,cAAc,EAAE,QAAQ,QAAQ,UAAU,aAAa;AAAA,EACvD,aAAa,EAAE,QAAQ,OAAO,UAAU,YAAY;AAAA,EACpD,cAAc,EAAE,QAAQ,OAAO,UAAU,YAAY;AAAA,EACrD,cAAc,EAAE,QAAQ,QAAQ,UAAU,aAAa;AAAA,EACvD,aAAa,EAAE,QAAQ,OAAO,UAAU,YAAY;AAAA,EACpD,aAAa,EAAE,QAAQ,OAAO,UAAU,YAAY;AAAA,EACpD,aAAa,EAAE,QAAQ,OAAO,UAAU,YAAY;AAAA,EACpD,aAAa,EAAE,QAAQ,OAAO,UAAU,YAAY;AACtD;AAkBO,SAAS,aAAa,OAAuD;AAClF,SAAO,MAAM,aAAa,gBAAgB,MAAM,QAAQ,SAAS,eAAe;AAClF;AAEO,SAAS,SAAS,SAAiB,MAAsB;AAC9D,SAAO,GAAG,QAAQ,QAAQ,OAAO,EAAE,CAAC,IAAI,IAAI;AAC9C;AAEO,SAAS,YAAY,SAAyB;AACnD,QAAM,MAAM,IAAI,IAAI,OAAO;AAC3B,MAAI,WAAW,IAAI,aAAa,WAAW,SAAS;AACpD,MAAI,WAAW,GAAG,IAAI,SAAS,QAAQ,OAAO,EAAE,CAAC;AACjD,MAAI,SAAS;AACb,SAAO,IAAI,SAAS;AACtB;AAEO,SAAS,UAAU,UAAwD;AAChF,QAAM,QAAQ,UAAU,SAAS,YAAY,CAAC;AAC9C,MAAI,CAAC,MAAO,OAAM,IAAI,MAAM,0BAA0B,QAAQ,EAAE;AAChE,SAAO;AACT;AAEO,SAAS,SAAS,OAAkD;AACzE,SAAO,OAAO,UAAU,YAAY,UAAU;AAChD;AAEO,SAAS,YAAY,QAAiC,KAAiC;AAC5F,QAAM,QAAQ,OAAO,GAAG;AACxB,SAAO,OAAO,UAAU,WAAW,QAAQ;AAC7C;AAEO,SAAS,YAAY,QAAiC,KAAiC;AAC5F,QAAM,QAAQ,OAAO,GAAG;AACxB,SAAO,OAAO,UAAU,WAAW,QAAQ;AAC7C;AAEO,SAAS,aAAa,OAAgE;AAC3F,MAAI,iBAAiB,kBAAmB,QAAO,EAAE,QAAQ,MAAM,QAAQ,OAAO,MAAM,QAAQ;AAC5F,MAAI,iBAAiB,SAAS,MAAM,SAAS;AAC3C,WAAO,EAAE,QAAQ,WAAW,OAAO,UAAU;AAC/C,MAAI,iBAAiB,MAAO,QAAO,EAAE,QAAQ,SAAS,OAAO,MAAM,QAAQ;AAC3E,SAAO,EAAE,QAAQ,SAAS,OAAO,OAAO,KAAK,EAAE;AACjD;AAEA,SAAS,YACP,SACA,MACoB;AACpB,MAAI,CAAC,QAAS,QAAO;AACrB,QAAM,WAAW,KAAK,YAAY;AAClC,aAAW,CAAC,KAAK,KAAK,KAAK,OAAO,QAAQ,OAAO,GAAG;AAClD,QAAI,IAAI,YAAY,MAAM,YAAY,UAAU,QAAQ,MAAM,KAAK,MAAM,GAAI,QAAO;AAAA,EACtF;AACA,SAAO;AACT;AAEO,SAAS,QAAQ,SAA6D;AACnF,SACE,YAAY,SAAS,eAAe,MAAM,UAC1C,YAAY,SAAS,sBAAsB,MAAM;AAErD;AAEO,SAAS,eACd,OACA,SACwB;AACxB,QAAM,UAAkC,EAAE,GAAG,MAAM,QAAQ;AAC3D,QAAM,QAAQ,SAAS;AACvB,MAAI,OAAO;AACT,eAAW,CAAC,KAAK,KAAK,KAAK,OAAO,QAAQ,KAAK,GAAG;AAChD,UAAI,UAAU,KAAM,QAAO,QAAQ,GAAG;AAAA,UACjC,SAAQ,GAAG,IAAI;AAAA,IACtB;AAAA,EACF;AACA,MAAI,CAAC,QAAQ,OAAO,GAAG;AACrB,UAAM,SAAS,SAAS;AACxB,QAAI,CAAC,OAAQ,OAAM,IAAI,MAAM,4BAA4B,MAAM,QAAQ,EAAE;AACzE,YAAQ,gBAAgB,UAAU,MAAM;AAAA,EAC1C;AACA,SAAO;AACT;AAEO,SAAS,cACd,SACyB;AACzB,QAAM,UAAyB,CAAC;AAChC,MAAI,SAAS,OAAQ,SAAQ,KAAK,QAAQ,MAAM;AAChD,QAAM,UAAU,SAAS;AACzB,MAAI,YAAY,UAAa,UAAU,EAAG,SAAQ,KAAK,YAAY,QAAQ,OAAO,CAAC;AACnF,MAAI,QAAQ,WAAW,EAAG,QAAO;AACjC,MAAI,QAAQ,WAAW,EAAG,QAAO,QAAQ,CAAC;AAC1C,SAAO,YAAY,IAAI,OAAO;AAChC;AAEO,SAAS,gBAAgB,MAAiB,QAA4C;AAC3F,QAAM,SAAuB,CAAC;AAC9B,QAAM,UAAU,IAAI,YAAY;AAChC,QAAM,QAAQ,CAAC,UAAkB,OAAO,KAAK,QAAQ,OAAO,KAAK,CAAC;AAClE;AAAA,IACE,KAAK,QAAQ;AAAA,yDAA8D,KAAK,QAAQ;AAAA,gBAAsB,KAAK,QAAQ;AAAA;AAAA;AAAA,EAC7H;AACA,SAAO,KAAK,KAAK,IAAI;AACrB,QAAM,MAAM;AACZ,aAAW,CAAC,MAAM,KAAK,KAAK,OAAO,QAAQ,MAAM,GAAG;AAClD,UAAM,KAAK,QAAQ;AAAA,wCAA6C,IAAI;AAAA;AAAA,EAAY,KAAK;AAAA,CAAM;AAAA,EAC7F;AACA,QAAM,KAAK,QAAQ;AAAA,CAAQ;AAC3B,QAAM,OAAO,IAAI,WAAW,OAAO,OAAO,CAAC,KAAK,UAAU,MAAM,MAAM,QAAQ,CAAC,CAAC;AAChF,MAAI,SAAS;AACb,aAAW,SAAS,QAAQ;AAC1B,SAAK,IAAI,OAAO,MAAM;AACtB,cAAU,MAAM;AAAA,EAClB;AACA,SAAO;AACT;AAEO,SAAS,gBAAwB;AACtC,SAAO,iCAAiC,QAAQ;AAClD;AAEA,eAAsB,UACpB,OACA,WACA,QACiD;AACjD,MAAI,EAAE,SAAS,OAAQ,QAAO,EAAE,MAAM,MAAM,MAAM,UAAU,MAAM,SAAS;AAC3E,QAAM,WAAW,MAAM,UAAU,MAAM,KAAK,EAAE,OAAO,CAAC;AACtD,MAAI,CAAC,SAAS,GAAI,OAAM,IAAI,MAAM,0BAA0B,SAAS,MAAM,EAAE;AAC7E,QAAM,OAAO,SAAS,QAAQ,IAAI,cAAc;AAChD,QAAM,WAAW,SAAS,QAAQ,KAAK,KAAK,MAAM,KAAK,cAAc,KAAK,MAAM,GAAG,EAAE,CAAC,EAAE,KAAK;AAC7F,SAAO,EAAE,MAAM,IAAI,WAAW,MAAM,SAAS,YAAY,CAAC,GAAG,SAAS;AACxE;AAEO,SAAS,aAAa,MAAiB,YAAqB,SAA2B;AAC5F,MAAI,YAAY;AACd,WAAO;AAAA,MACL,OAAO;AAAA,MACP,aAAa,EAAE,MAAM,kBAAkB,KAAK,IAAI,GAAG,QAAQ,KAAK,OAAO;AAAA,IACzE;AAAA,EACF;AACA,SAAO,EAAE,OAAO,SAAS,KAAK;AAChC;AAEA,eAAsB,aACpB,SACA,OACA,SACkB;AAClB,QAAM,OAAO,SAAS;AAGtB,QAAM,WAAW,MAAM,OAAO,SAAS,KAAK;AAC5C,SAAO,aAAa,SAAY,UAAU;AAC5C;AAEA,SAAS,aAAa,SAA0C;AAC9D,QAAM,SAAiC,CAAC;AACxC,UAAQ,QAAQ,CAAC,OAAO,QAAQ;AAC9B,WAAO,GAAG,IAAI;AAAA,EAChB,CAAC;AACD,SAAO;AACT;AAEA,eAAsB,WACpB,OACA,SACA,KACA,MACA,aACmB;AACnB,QAAM,UAAU,eAAe,OAAO,OAAO;AAC7C,UAAQ,cAAc,IAAI;AAC1B,QAAM,YAAY,SAAS,SAAS,WAAW;AAC/C,QAAM,WAAW,MAAM,UAAU,KAAK;AAAA,IACpC,QAAQ;AAAA,IACR;AAAA,IACA;AAAA,IACA,QAAQ,cAAc,OAAO;AAAA,EAC/B,CAAC;AACD,QAAM,SAAS;AAAA,IACb,EAAE,QAAQ,SAAS,QAAQ,SAAS,aAAa,SAAS,OAAO,EAAE;AAAA,IACnE;AAAA,EACF;AACA,MAAI,CAAC,SAAS,IAAI;AAChB,UAAM,OAAO,MAAM,SAAS,KAAK;AACjC,UAAM,IAAI,MAAM,SAAS,KAAK,QAAQ,SAAS,MAAM,KAAK,IAAI;AAAA,EAChE;AACA,SAAO;AACT;AAEA,eAAsB,SAAS,UAAsC;AACnE,QAAM,OAAO,MAAM,SAAS,KAAK;AACjC,MAAI;AACF,WAAO,KAAK,MAAM,IAAI;AAAA,EACxB,QAAQ;AACN,UAAM,IAAI,MAAM,qCAAqC;AAAA,EACvD;AACF;",
6
+ "names": []
7
+ }
package/dist/registry.js CHANGED
@@ -1,3 +1,5 @@
1
+ import { speakInSentences } from "./chunk-speech.js";
2
+ import { listenWithVad } from "./vad-listen.js";
1
3
  const transcriptionApis = /* @__PURE__ */ new Map();
2
4
  const speechApis = /* @__PURE__ */ new Map();
3
5
  const realtimeApis = /* @__PURE__ */ new Map();
@@ -33,11 +35,13 @@ function transcribe(model, input, options) {
33
35
  function listen(model, audio, options) {
34
36
  const listenModel = transcriptionApis.get(model.api)?.provider.listen;
35
37
  if (!listenModel) throw missing(model.api);
38
+ if (model.turnDetection === "none") return listenWithVad(listenModel, model, audio, options);
36
39
  return listenModel(model, audio, options);
37
40
  }
38
41
  function speak(model, input, options) {
39
42
  const speakModel = speechApis.get(model.api)?.provider.speak;
40
43
  if (!speakModel) throw missing(model.api);
44
+ if (!model.textStreaming) return speakInSentences(speakModel, model, input, options);
41
45
  return speakModel(model, input, options);
42
46
  }
43
47
  function connectRealtime(model, context, options) {
@@ -1,7 +1,7 @@
1
1
  {
2
2
  "version": 3,
3
3
  "sources": ["../src/registry.ts"],
4
- "sourcesContent": ["import type { ProviderRequestOptions } from '@earendil-works/pi-ai';\nimport type { AudioSource } from './audio.ts';\nimport type { RealtimeModel, SpeechModel, TranscriptionModel } from './models.ts';\nimport type {\n RealtimeApiProvider,\n RealtimeContext,\n RealtimeOptions,\n RealtimeSession,\n} from './realtime.ts';\nimport type { SpeechApiProvider, SpeechEventStream, SpeechInput } from './speech.ts';\nimport type {\n AudioInput,\n ListenOptions,\n Transcript,\n TranscriptionApiProvider,\n TranscriptionEventStream,\n} from './transcription.ts';\n\ninterface TranscriptionEntry {\n provider: TranscriptionApiProvider;\n}\n\ninterface SpeechEntry {\n provider: SpeechApiProvider;\n}\n\ninterface RealtimeEntry {\n provider: RealtimeApiProvider;\n}\n\nconst transcriptionApis = new Map<string, TranscriptionEntry>();\nconst speechApis = new Map<string, SpeechEntry>();\nconst realtimeApis = new Map<string, RealtimeEntry>();\n\nfunction missing(api: string): Error {\n return new Error(`No API provider registered for api: ${api}`);\n}\n\nexport function registerTranscriptionApiProvider(provider: TranscriptionApiProvider): () => void {\n const entry = { provider };\n transcriptionApis.set(provider.api, entry);\n return () => {\n if (transcriptionApis.get(provider.api) === entry) transcriptionApis.delete(provider.api);\n };\n}\n\nexport function registerSpeechApiProvider(provider: SpeechApiProvider): () => void {\n const entry = { provider };\n speechApis.set(provider.api, entry);\n return () => {\n if (speechApis.get(provider.api) === entry) speechApis.delete(provider.api);\n };\n}\n\nexport function registerRealtimeApiProvider(provider: RealtimeApiProvider): () => void {\n const entry = { provider };\n realtimeApis.set(provider.api, entry);\n return () => {\n if (realtimeApis.get(provider.api) === entry) realtimeApis.delete(provider.api);\n };\n}\n\nexport function transcribe(\n model: TranscriptionModel<string>,\n input: AudioInput,\n options?: ProviderRequestOptions\n): Promise<Transcript> {\n const transcribeModel = transcriptionApis.get(model.api)?.provider.transcribe;\n if (!transcribeModel) return Promise.reject(missing(model.api));\n return transcribeModel(model, input, options);\n}\n\nexport function listen(\n model: TranscriptionModel<string>,\n audio: AudioSource,\n options?: ListenOptions\n): TranscriptionEventStream {\n const listenModel = transcriptionApis.get(model.api)?.provider.listen;\n if (!listenModel) throw missing(model.api);\n return listenModel(model, audio, options);\n}\n\nexport function speak(\n model: SpeechModel<string>,\n input: SpeechInput,\n options?: ProviderRequestOptions\n): SpeechEventStream {\n const speakModel = speechApis.get(model.api)?.provider.speak;\n if (!speakModel) throw missing(model.api);\n return speakModel(model, input, options);\n}\n\nexport function connectRealtime(\n model: RealtimeModel<string>,\n context: RealtimeContext,\n options?: RealtimeOptions\n): Promise<RealtimeSession> {\n const connect = realtimeApis.get(model.api)?.provider.connect;\n if (!connect) return Promise.reject(missing(model.api));\n return connect(model, context, options);\n}\n"],
5
- "mappings": "AA8BA,MAAM,oBAAoB,oBAAI,IAAgC;AAC9D,MAAM,aAAa,oBAAI,IAAyB;AAChD,MAAM,eAAe,oBAAI,IAA2B;AAEpD,SAAS,QAAQ,KAAoB;AACnC,SAAO,IAAI,MAAM,uCAAuC,GAAG,EAAE;AAC/D;AAEO,SAAS,iCAAiC,UAAgD;AAC/F,QAAM,QAAQ,EAAE,SAAS;AACzB,oBAAkB,IAAI,SAAS,KAAK,KAAK;AACzC,SAAO,MAAM;AACX,QAAI,kBAAkB,IAAI,SAAS,GAAG,MAAM,MAAO,mBAAkB,OAAO,SAAS,GAAG;AAAA,EAC1F;AACF;AAEO,SAAS,0BAA0B,UAAyC;AACjF,QAAM,QAAQ,EAAE,SAAS;AACzB,aAAW,IAAI,SAAS,KAAK,KAAK;AAClC,SAAO,MAAM;AACX,QAAI,WAAW,IAAI,SAAS,GAAG,MAAM,MAAO,YAAW,OAAO,SAAS,GAAG;AAAA,EAC5E;AACF;AAEO,SAAS,4BAA4B,UAA2C;AACrF,QAAM,QAAQ,EAAE,SAAS;AACzB,eAAa,IAAI,SAAS,KAAK,KAAK;AACpC,SAAO,MAAM;AACX,QAAI,aAAa,IAAI,SAAS,GAAG,MAAM,MAAO,cAAa,OAAO,SAAS,GAAG;AAAA,EAChF;AACF;AAEO,SAAS,WACd,OACA,OACA,SACqB;AACrB,QAAM,kBAAkB,kBAAkB,IAAI,MAAM,GAAG,GAAG,SAAS;AACnE,MAAI,CAAC,gBAAiB,QAAO,QAAQ,OAAO,QAAQ,MAAM,GAAG,CAAC;AAC9D,SAAO,gBAAgB,OAAO,OAAO,OAAO;AAC9C;AAEO,SAAS,OACd,OACA,OACA,SAC0B;AAC1B,QAAM,cAAc,kBAAkB,IAAI,MAAM,GAAG,GAAG,SAAS;AAC/D,MAAI,CAAC,YAAa,OAAM,QAAQ,MAAM,GAAG;AACzC,SAAO,YAAY,OAAO,OAAO,OAAO;AAC1C;AAEO,SAAS,MACd,OACA,OACA,SACmB;AACnB,QAAM,aAAa,WAAW,IAAI,MAAM,GAAG,GAAG,SAAS;AACvD,MAAI,CAAC,WAAY,OAAM,QAAQ,MAAM,GAAG;AACxC,SAAO,WAAW,OAAO,OAAO,OAAO;AACzC;AAEO,SAAS,gBACd,OACA,SACA,SAC0B;AAC1B,QAAM,UAAU,aAAa,IAAI,MAAM,GAAG,GAAG,SAAS;AACtD,MAAI,CAAC,QAAS,QAAO,QAAQ,OAAO,QAAQ,MAAM,GAAG,CAAC;AACtD,SAAO,QAAQ,OAAO,SAAS,OAAO;AACxC;",
4
+ "sourcesContent": ["import type { ProviderRequestOptions } from '@earendil-works/pi-ai';\nimport type { AudioSource } from './audio.ts';\nimport { speakInSentences } from './chunk-speech.ts';\nimport type { RealtimeModel, SpeechModel, TranscriptionModel } from './models.ts';\nimport type {\n RealtimeApiProvider,\n RealtimeContext,\n RealtimeOptions,\n RealtimeSession,\n} from './realtime.ts';\nimport type { SpeechApiProvider, SpeechEventStream, SpeechInput } from './speech.ts';\nimport type {\n AudioInput,\n ListenOptions,\n Transcript,\n TranscriptionApiProvider,\n TranscriptionEventStream,\n} from './transcription.ts';\nimport { listenWithVad } from './vad-listen.ts';\n\ninterface TranscriptionEntry {\n provider: TranscriptionApiProvider;\n}\n\ninterface SpeechEntry {\n provider: SpeechApiProvider;\n}\n\ninterface RealtimeEntry {\n provider: RealtimeApiProvider;\n}\n\nconst transcriptionApis = new Map<string, TranscriptionEntry>();\nconst speechApis = new Map<string, SpeechEntry>();\nconst realtimeApis = new Map<string, RealtimeEntry>();\n\nfunction missing(api: string): Error {\n return new Error(`No API provider registered for api: ${api}`);\n}\n\nexport function registerTranscriptionApiProvider(provider: TranscriptionApiProvider): () => void {\n const entry = { provider };\n transcriptionApis.set(provider.api, entry);\n return () => {\n if (transcriptionApis.get(provider.api) === entry) transcriptionApis.delete(provider.api);\n };\n}\n\nexport function registerSpeechApiProvider(provider: SpeechApiProvider): () => void {\n const entry = { provider };\n speechApis.set(provider.api, entry);\n return () => {\n if (speechApis.get(provider.api) === entry) speechApis.delete(provider.api);\n };\n}\n\nexport function registerRealtimeApiProvider(provider: RealtimeApiProvider): () => void {\n const entry = { provider };\n realtimeApis.set(provider.api, entry);\n return () => {\n if (realtimeApis.get(provider.api) === entry) realtimeApis.delete(provider.api);\n };\n}\n\nexport function transcribe(\n model: TranscriptionModel<string>,\n input: AudioInput,\n options?: ProviderRequestOptions\n): Promise<Transcript> {\n const transcribeModel = transcriptionApis.get(model.api)?.provider.transcribe;\n if (!transcribeModel) return Promise.reject(missing(model.api));\n return transcribeModel(model, input, options);\n}\n\nexport function listen(\n model: TranscriptionModel<string>,\n audio: AudioSource,\n options?: ListenOptions\n): TranscriptionEventStream {\n const listenModel = transcriptionApis.get(model.api)?.provider.listen;\n if (!listenModel) throw missing(model.api);\n if (model.turnDetection === 'none') return listenWithVad(listenModel, model, audio, options);\n return listenModel(model, audio, options);\n}\n\nexport function speak(\n model: SpeechModel<string>,\n input: SpeechInput,\n options?: ProviderRequestOptions\n): SpeechEventStream {\n const speakModel = speechApis.get(model.api)?.provider.speak;\n if (!speakModel) throw missing(model.api);\n if (!model.textStreaming) return speakInSentences(speakModel, model, input, options);\n return speakModel(model, input, options);\n}\n\nexport function connectRealtime(\n model: RealtimeModel<string>,\n context: RealtimeContext,\n options?: RealtimeOptions\n): Promise<RealtimeSession> {\n const connect = realtimeApis.get(model.api)?.provider.connect;\n if (!connect) return Promise.reject(missing(model.api));\n return connect(model, context, options);\n}\n"],
5
+ "mappings": "AAEA,SAAS,wBAAwB;AAgBjC,SAAS,qBAAqB;AAc9B,MAAM,oBAAoB,oBAAI,IAAgC;AAC9D,MAAM,aAAa,oBAAI,IAAyB;AAChD,MAAM,eAAe,oBAAI,IAA2B;AAEpD,SAAS,QAAQ,KAAoB;AACnC,SAAO,IAAI,MAAM,uCAAuC,GAAG,EAAE;AAC/D;AAEO,SAAS,iCAAiC,UAAgD;AAC/F,QAAM,QAAQ,EAAE,SAAS;AACzB,oBAAkB,IAAI,SAAS,KAAK,KAAK;AACzC,SAAO,MAAM;AACX,QAAI,kBAAkB,IAAI,SAAS,GAAG,MAAM,MAAO,mBAAkB,OAAO,SAAS,GAAG;AAAA,EAC1F;AACF;AAEO,SAAS,0BAA0B,UAAyC;AACjF,QAAM,QAAQ,EAAE,SAAS;AACzB,aAAW,IAAI,SAAS,KAAK,KAAK;AAClC,SAAO,MAAM;AACX,QAAI,WAAW,IAAI,SAAS,GAAG,MAAM,MAAO,YAAW,OAAO,SAAS,GAAG;AAAA,EAC5E;AACF;AAEO,SAAS,4BAA4B,UAA2C;AACrF,QAAM,QAAQ,EAAE,SAAS;AACzB,eAAa,IAAI,SAAS,KAAK,KAAK;AACpC,SAAO,MAAM;AACX,QAAI,aAAa,IAAI,SAAS,GAAG,MAAM,MAAO,cAAa,OAAO,SAAS,GAAG;AAAA,EAChF;AACF;AAEO,SAAS,WACd,OACA,OACA,SACqB;AACrB,QAAM,kBAAkB,kBAAkB,IAAI,MAAM,GAAG,GAAG,SAAS;AACnE,MAAI,CAAC,gBAAiB,QAAO,QAAQ,OAAO,QAAQ,MAAM,GAAG,CAAC;AAC9D,SAAO,gBAAgB,OAAO,OAAO,OAAO;AAC9C;AAEO,SAAS,OACd,OACA,OACA,SAC0B;AAC1B,QAAM,cAAc,kBAAkB,IAAI,MAAM,GAAG,GAAG,SAAS;AAC/D,MAAI,CAAC,YAAa,OAAM,QAAQ,MAAM,GAAG;AACzC,MAAI,MAAM,kBAAkB,OAAQ,QAAO,cAAc,aAAa,OAAO,OAAO,OAAO;AAC3F,SAAO,YAAY,OAAO,OAAO,OAAO;AAC1C;AAEO,SAAS,MACd,OACA,OACA,SACmB;AACnB,QAAM,aAAa,WAAW,IAAI,MAAM,GAAG,GAAG,SAAS;AACvD,MAAI,CAAC,WAAY,OAAM,QAAQ,MAAM,GAAG;AACxC,MAAI,CAAC,MAAM,cAAe,QAAO,iBAAiB,YAAY,OAAO,OAAO,OAAO;AACnF,SAAO,WAAW,OAAO,OAAO,OAAO;AACzC;AAEO,SAAS,gBACd,OACA,SACA,SAC0B;AAC1B,QAAM,UAAU,aAAa,IAAI,MAAM,GAAG,GAAG,SAAS;AACtD,MAAI,CAAC,QAAS,QAAO,QAAQ,OAAO,QAAQ,MAAM,GAAG,CAAC;AACtD,SAAO,QAAQ,OAAO,SAAS,OAAO;AACxC;",
6
6
  "names": []
7
7
  }
@@ -0,0 +1,88 @@
1
+ const abbreviations = /* @__PURE__ */ new Set([
2
+ "mr",
3
+ "mrs",
4
+ "ms",
5
+ "dr",
6
+ "prof",
7
+ "sr",
8
+ "jr",
9
+ "st",
10
+ "vs",
11
+ "etc",
12
+ "eg",
13
+ "ie",
14
+ "fig",
15
+ "no"
16
+ ]);
17
+ function wordBefore(text, index) {
18
+ let start = index;
19
+ while (start > 0 && /[A-Za-z]/.test(text[start - 1] ?? "")) start -= 1;
20
+ return text.slice(start, index).toLowerCase();
21
+ }
22
+ function boundary(text, index) {
23
+ const ch = text[index];
24
+ if (ch === "\n") return true;
25
+ if (ch !== "." && ch !== "!" && ch !== "?" && ch !== "\u2026") return false;
26
+ const previous = text[index - 1];
27
+ const next = text[index + 1];
28
+ if ((ch === "." || ch === "\u2026") && previous !== void 0 && next !== void 0 && previous >= "0" && previous <= "9" && next >= "0" && next <= "9") {
29
+ return false;
30
+ }
31
+ if (ch === "." && next === ".") return false;
32
+ if (ch === ".") {
33
+ const word = wordBefore(text, index);
34
+ if (abbreviations.has(word) || /^[A-Za-z]$/.test(word)) return false;
35
+ }
36
+ return true;
37
+ }
38
+ function splitSentences(buffer, flush) {
39
+ const sentences = [];
40
+ let start = 0;
41
+ let index = 0;
42
+ while (index < buffer.length) {
43
+ if (!boundary(buffer, index)) {
44
+ index += 1;
45
+ continue;
46
+ }
47
+ const ch = buffer[index];
48
+ const next = index + 1;
49
+ if (ch !== "\n" && next >= buffer.length && !flush) break;
50
+ if (ch !== "\n" && next < buffer.length && !/\s/.test(buffer[next] ?? "")) {
51
+ index += 1;
52
+ continue;
53
+ }
54
+ let end = next;
55
+ while (end < buffer.length && /\s/.test(buffer[end] ?? "")) end += 1;
56
+ const sentence = buffer.slice(start, end);
57
+ if (sentence.trim().length === 0) {
58
+ index = Math.max(end, index + 1);
59
+ continue;
60
+ }
61
+ sentences.push(sentence);
62
+ start = end;
63
+ index = end;
64
+ }
65
+ const rest = buffer.slice(start);
66
+ if (!flush) return { sentences, rest };
67
+ if (rest.length > 0) sentences.push(rest);
68
+ return { sentences, rest: "" };
69
+ }
70
+ async function* sentencesOf(text) {
71
+ if (typeof text === "string") {
72
+ for (const sentence of splitSentences(text, true).sentences) yield sentence;
73
+ return;
74
+ }
75
+ let buffer = "";
76
+ for await (const part of text) {
77
+ buffer += part;
78
+ const split = splitSentences(buffer, false);
79
+ buffer = split.rest;
80
+ for (const sentence of split.sentences) yield sentence;
81
+ }
82
+ for (const sentence of splitSentences(buffer, true).sentences) yield sentence;
83
+ }
84
+ export {
85
+ sentencesOf,
86
+ splitSentences
87
+ };
88
+ //# sourceMappingURL=sentences.js.map
@@ -0,0 +1,7 @@
1
+ {
2
+ "version": 3,
3
+ "sources": ["../src/sentences.ts"],
4
+ "sourcesContent": ["const abbreviations = new Set([\n 'mr',\n 'mrs',\n 'ms',\n 'dr',\n 'prof',\n 'sr',\n 'jr',\n 'st',\n 'vs',\n 'etc',\n 'eg',\n 'ie',\n 'fig',\n 'no',\n]);\n\nfunction wordBefore(text: string, index: number): string {\n let start = index;\n while (start > 0 && /[A-Za-z]/.test(text[start - 1] ?? '')) start -= 1;\n return text.slice(start, index).toLowerCase();\n}\n\nfunction boundary(text: string, index: number): boolean {\n const ch = text[index];\n if (ch === '\\n') return true;\n if (ch !== '.' && ch !== '!' && ch !== '?' && ch !== '\u2026') return false;\n const previous = text[index - 1];\n const next = text[index + 1];\n if (\n (ch === '.' || ch === '\u2026') &&\n previous !== undefined &&\n next !== undefined &&\n previous >= '0' &&\n previous <= '9' &&\n next >= '0' &&\n next <= '9'\n ) {\n return false;\n }\n if (ch === '.' && next === '.') return false;\n if (ch === '.') {\n const word = wordBefore(text, index);\n if (abbreviations.has(word) || /^[A-Za-z]$/.test(word)) return false;\n }\n return true;\n}\n\nexport function splitSentences(\n buffer: string,\n flush: boolean\n): { sentences: string[]; rest: string } {\n const sentences: string[] = [];\n let start = 0;\n let index = 0;\n while (index < buffer.length) {\n if (!boundary(buffer, index)) {\n index += 1;\n continue;\n }\n const ch = buffer[index];\n const next = index + 1;\n if (ch !== '\\n' && next >= buffer.length && !flush) break;\n if (ch !== '\\n' && next < buffer.length && !/\\s/.test(buffer[next] ?? '')) {\n index += 1;\n continue;\n }\n let end = next;\n while (end < buffer.length && /\\s/.test(buffer[end] ?? '')) end += 1;\n const sentence = buffer.slice(start, end);\n if (sentence.trim().length === 0) {\n index = Math.max(end, index + 1);\n continue;\n }\n sentences.push(sentence);\n start = end;\n index = end;\n }\n const rest = buffer.slice(start);\n if (!flush) return { sentences, rest };\n if (rest.length > 0) sentences.push(rest);\n return { sentences, rest: '' };\n}\n\nexport async function* sentencesOf(text: string | AsyncIterable<string>): AsyncGenerator<string> {\n if (typeof text === 'string') {\n for (const sentence of splitSentences(text, true).sentences) yield sentence;\n return;\n }\n let buffer = '';\n for await (const part of text) {\n buffer += part;\n const split = splitSentences(buffer, false);\n buffer = split.rest;\n for (const sentence of split.sentences) yield sentence;\n }\n for (const sentence of splitSentences(buffer, true).sentences) yield sentence;\n}\n"],
5
+ "mappings": "AAAA,MAAM,gBAAgB,oBAAI,IAAI;AAAA,EAC5B;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF,CAAC;AAED,SAAS,WAAW,MAAc,OAAuB;AACvD,MAAI,QAAQ;AACZ,SAAO,QAAQ,KAAK,WAAW,KAAK,KAAK,QAAQ,CAAC,KAAK,EAAE,EAAG,UAAS;AACrE,SAAO,KAAK,MAAM,OAAO,KAAK,EAAE,YAAY;AAC9C;AAEA,SAAS,SAAS,MAAc,OAAwB;AACtD,QAAM,KAAK,KAAK,KAAK;AACrB,MAAI,OAAO,KAAM,QAAO;AACxB,MAAI,OAAO,OAAO,OAAO,OAAO,OAAO,OAAO,OAAO,SAAK,QAAO;AACjE,QAAM,WAAW,KAAK,QAAQ,CAAC;AAC/B,QAAM,OAAO,KAAK,QAAQ,CAAC;AAC3B,OACG,OAAO,OAAO,OAAO,aACtB,aAAa,UACb,SAAS,UACT,YAAY,OACZ,YAAY,OACZ,QAAQ,OACR,QAAQ,KACR;AACA,WAAO;AAAA,EACT;AACA,MAAI,OAAO,OAAO,SAAS,IAAK,QAAO;AACvC,MAAI,OAAO,KAAK;AACd,UAAM,OAAO,WAAW,MAAM,KAAK;AACnC,QAAI,cAAc,IAAI,IAAI,KAAK,aAAa,KAAK,IAAI,EAAG,QAAO;AAAA,EACjE;AACA,SAAO;AACT;AAEO,SAAS,eACd,QACA,OACuC;AACvC,QAAM,YAAsB,CAAC;AAC7B,MAAI,QAAQ;AACZ,MAAI,QAAQ;AACZ,SAAO,QAAQ,OAAO,QAAQ;AAC5B,QAAI,CAAC,SAAS,QAAQ,KAAK,GAAG;AAC5B,eAAS;AACT;AAAA,IACF;AACA,UAAM,KAAK,OAAO,KAAK;AACvB,UAAM,OAAO,QAAQ;AACrB,QAAI,OAAO,QAAQ,QAAQ,OAAO,UAAU,CAAC,MAAO;AACpD,QAAI,OAAO,QAAQ,OAAO,OAAO,UAAU,CAAC,KAAK,KAAK,OAAO,IAAI,KAAK,EAAE,GAAG;AACzE,eAAS;AACT;AAAA,IACF;AACA,QAAI,MAAM;AACV,WAAO,MAAM,OAAO,UAAU,KAAK,KAAK,OAAO,GAAG,KAAK,EAAE,EAAG,QAAO;AACnE,UAAM,WAAW,OAAO,MAAM,OAAO,GAAG;AACxC,QAAI,SAAS,KAAK,EAAE,WAAW,GAAG;AAChC,cAAQ,KAAK,IAAI,KAAK,QAAQ,CAAC;AAC/B;AAAA,IACF;AACA,cAAU,KAAK,QAAQ;AACvB,YAAQ;AACR,YAAQ;AAAA,EACV;AACA,QAAM,OAAO,OAAO,MAAM,KAAK;AAC/B,MAAI,CAAC,MAAO,QAAO,EAAE,WAAW,KAAK;AACrC,MAAI,KAAK,SAAS,EAAG,WAAU,KAAK,IAAI;AACxC,SAAO,EAAE,WAAW,MAAM,GAAG;AAC/B;AAEA,gBAAuB,YAAY,MAA8D;AAC/F,MAAI,OAAO,SAAS,UAAU;AAC5B,eAAW,YAAY,eAAe,MAAM,IAAI,EAAE,UAAW,OAAM;AACnE;AAAA,EACF;AACA,MAAI,SAAS;AACb,mBAAiB,QAAQ,MAAM;AAC7B,cAAU;AACV,UAAM,QAAQ,eAAe,QAAQ,KAAK;AAC1C,aAAS,MAAM;AACf,eAAW,YAAY,MAAM,UAAW,OAAM;AAAA,EAChD;AACA,aAAW,YAAY,eAAe,QAAQ,IAAI,EAAE,UAAW,OAAM;AACvE;",
6
+ "names": []
7
+ }