openai
Version:
The official TypeScript library for the OpenAI API
64 lines • 2.66 kB
JavaScript
import { OpenAI } from "../../index.mjs";
import { OpenAIError } from "../../error.mjs";
import * as Core from "../../core.mjs";
import { OpenAIRealtimeEmitter, buildRealtimeURL } from "./internal-base.mjs";
export class OpenAIRealtimeWebSocket extends OpenAIRealtimeEmitter {
constructor(props, client) {
super();
const dangerouslyAllowBrowser = props.dangerouslyAllowBrowser ??
client?._options?.dangerouslyAllowBrowser ??
(client?.apiKey.startsWith('ek_') ? true : null);
if (!dangerouslyAllowBrowser && Core.isRunningInBrowser()) {
throw new OpenAIError("It looks like you're running in a browser-like environment.\n\nThis is disabled by default, as it risks exposing your secret API credentials to attackers.\n\nYou can avoid this error by creating an ephemeral session token:\nhttps://platform.openai.com/docs/api-reference/realtime-sessions\n");
}
client ?? (client = new OpenAI({ dangerouslyAllowBrowser }));
this.url = buildRealtimeURL({ baseURL: client.baseURL, model: props.model });
// @ts-ignore
this.socket = new WebSocket(this.url, [
'realtime',
`openai-insecure-api-key.${client.apiKey}`,
'openai-beta.realtime-v1',
]);
this.socket.addEventListener('message', (websocketEvent) => {
const event = (() => {
try {
return JSON.parse(websocketEvent.data.toString());
}
catch (err) {
this._onError(null, 'could not parse websocket event', err);
return null;
}
})();
if (event) {
this._emit('event', event);
if (event.type === 'error') {
this._onError(event);
}
else {
// @ts-expect-error TS isn't smart enough to get the relationship right here
this._emit(event.type, event);
}
}
});
this.socket.addEventListener('error', (event) => {
this._onError(null, event.message, null);
});
}
send(event) {
try {
this.socket.send(JSON.stringify(event));
}
catch (err) {
this._onError(null, 'could not send data', err);
}
}
close(props) {
try {
this.socket.close(props?.code ?? 1000, props?.reason ?? 'OK');
}
catch (err) {
this._onError(null, 'could not close the connection', err);
}
}
}
//# sourceMappingURL=websocket.mjs.map