openai
Version:
The official TypeScript library for the OpenAI API
62 lines • 1.98 kB
JavaScript
// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
import * as WS from 'ws';
import { ResponsesEmitter, buildURL } from "./internal-base.mjs";
export class ResponsesWS extends ResponsesEmitter {
constructor(client, options) {
super();
this.client = client;
this.url = buildURL(client);
this.socket = new WS.WebSocket(this.url, {
...options,
headers: {
...this.authHeaders(),
...options?.headers,
},
});
this.socket.on('message', (wsEvent) => {
const event = (() => {
try {
return JSON.parse(wsEvent.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-ignore TS isn't smart enough to get the relationship right here
this._emit(event.type, event);
}
}
});
this.socket.on('error', (err) => {
this._onError(null, err.message, err);
});
}
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);
}
}
authHeaders() {
return { Authorization: `Bearer ${this.client.apiKey}` };
return {};
}
}
//# sourceMappingURL=ws.mjs.map