UNPKG

openai

Version:

The official TypeScript library for the OpenAI API

64 lines 2.66 kB
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