assemblyai 2.0.2 → 3.0.1

package/dist/index.js

@@ -1,27 +1,8 @@
 'use strict';
 
-Object.defineProperty(exports, '__esModule', { value: true });
-
-var axios = require('axios');
 var WebSocket = require('ws');
-var stream = require('stream');
-var promises = require('fs/promises');
-
-function createAxiosClient(params) {
-    const client = axios.create({
-        baseURL: params.baseUrl,
-        headers: { Authorization: params.apiKey },
-    });
-    client.interceptors.response.use(undefined, throwApiError);
-    return client;
-}
-function throwApiError(error) {
-    var _a, _b;
-    if (axios.isAxiosError(error) && ((_b = (_a = error.response) === null || _a === void 0 ? void 0 : _a.data) === null || _b === void 0 ? void 0 : _b.error)) {
-        return Promise.reject(new Error(error.response.data.error));
-    }
-    return Promise.reject(error);
-}
+var Stream = require('stream');
+var fs = require('fs');
 
 /******************************************************************************
 Copyright (c) Microsoft Corporation.
@@ -61,36 +42,70 @@ typeof SuppressedError === "function" ? SuppressedError : function (error, suppr
 class BaseService {
     /**
      * Create a new service.
-     * @param params The AxiosInstance to send HTTP requests to the API.
+     * @param params The parameters to use for the service.
      */
-    constructor(client) {
-        this.client = client;
+    constructor(params) {
+        this.params = params;
+    }
+    fetch(input, init) {
+        var _a;
+        return __awaiter(this, void 0, void 0, function* () {
+            init = init !== null && init !== void 0 ? init : {};
+            init.headers = (_a = init.headers) !== null && _a !== void 0 ? _a : {};
+            init.headers = Object.assign({ Authorization: this.params.apiKey, "Content-Type": "application/json" }, init.headers);
+            if (!input.startsWith("http"))
+                input = this.params.baseUrl + input;
+            const response = yield fetch(input, init);
+            if (response.status >= 400) {
+                let json;
+                const text = yield response.text();
+                if (text) {
+                    try {
+                        json = JSON.parse(text);
+                    }
+                    catch (_b) {
+                        /* empty */
+                    }
+                    if (json === null || json === void 0 ? void 0 : json.error)
+                        throw new Error(json.error);
+                    throw new Error(text);
+                }
+                throw new Error(`HTTP Error: ${response.status} ${response.statusText}`);
+            }
+            return response;
+        });
+    }
+    fetchJson(input, init) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const response = yield this.fetch(input, init);
+            return response.json();
+        });
     }
 }
 
 class LemurService extends BaseService {
     summary(params) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const { data } = yield this.client.post("/lemur/v3/generate/summary", params);
-            return data;
+        return this.fetchJson("/lemur/v3/generate/summary", {
+            method: "POST",
+            body: JSON.stringify(params),
         });
     }
     questionAnswer(params) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const { data } = yield this.client.post("/lemur/v3/generate/question-answer", params);
-            return data;
+        return this.fetchJson("/lemur/v3/generate/question-answer", {
+            method: "POST",
+            body: JSON.stringify(params),
         });
     }
     actionItems(params) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const { data } = yield this.client.post("/lemur/v3/generate/action-items", params);
-            return data;
+        return this.fetchJson("/lemur/v3/generate/action-items", {
+            method: "POST",
+            body: JSON.stringify(params),
         });
     }
     task(params) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const { data } = yield this.client.post("/lemur/v3/generate/task", params);
-            return data;
+        return this.fetchJson("/lemur/v3/generate/task", {
+            method: "POST",
+            body: JSON.stringify(params),
         });
     }
     /**
@@ -98,9 +113,8 @@ class LemurService extends BaseService {
      * @param id ID of the LeMUR request
      */
     purgeRequestData(id) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const { data } = yield this.client.delete(`/lemur/v3/${id}`);
-            return data;
+        return this.fetchJson(`/lemur/v3/${id}`, {
+            method: "DELETE",
         });
     }
 }
@@ -149,12 +163,11 @@ class RealtimeError extends Error {
 const defaultRealtimeUrl = "wss://api.assemblyai.com/v2/realtime/ws";
 class RealtimeService {
     constructor(params) {
-        var _a, _b, _c;
+        var _a, _b;
         this.listeners = {};
         this.realtimeUrl = (_a = params.realtimeUrl) !== null && _a !== void 0 ? _a : defaultRealtimeUrl;
         this.sampleRate = (_b = params.sampleRate) !== null && _b !== void 0 ? _b : 16000;
         this.wordBoost = params.wordBoost;
-        this.realtimeUrl = (_c = params.realtimeUrl) !== null && _c !== void 0 ? _c : defaultRealtimeUrl;
         if ("apiKey" in params)
             this.apiKey = params.apiKey;
         if ("token" in params)
@@ -262,13 +275,13 @@ class RealtimeService {
         this.socket.send(JSON.stringify(payload));
     }
     stream() {
-        const stream$1 = new stream.Writable({
+        const stream = new Stream.Writable({
             write: (chunk, encoding, next) => {
                 this.sendAudio(chunk);
                 next();
             },
         });
-        return stream$1;
+        return stream;
     }
     close(waitForSessionTermination = true) {
         return __awaiter(this, void 0, void 0, function* () {
@@ -295,30 +308,33 @@ class RealtimeService {
     }
 }
 
-class RealtimeServiceFactory {
-    constructor(client, params) {
-        this.client = client;
-        this.params = params;
+class RealtimeServiceFactory extends BaseService {
+    constructor(params) {
+        super(params);
+        this.rtFactoryParams = params;
     }
     createService(params) {
         if (!params)
-            params = { apiKey: this.params.apiKey };
+            params = { apiKey: this.rtFactoryParams.apiKey };
         else if (!("token" in params) && !params.apiKey) {
-            params.apiKey = this.params.apiKey;
+            params.apiKey = this.rtFactoryParams.apiKey;
         }
         return new RealtimeService(params);
     }
     createTemporaryToken(params) {
         return __awaiter(this, void 0, void 0, function* () {
-            const response = yield this.client.post("/v2/realtime/token", params);
-            return response.data.token;
+            const data = yield this.fetchJson("/v2/realtime/token", {
+                method: "POST",
+                body: JSON.stringify(params),
+            });
+            return data.token;
         });
     }
 }
 
 class TranscriptService extends BaseService {
-    constructor(client, files) {
-        super(client);
+    constructor(params, files) {
+        super(params);
         this.files = files;
     }
     /**
@@ -335,11 +351,14 @@ class TranscriptService extends BaseService {
                 const uploadUrl = yield this.files.upload(path);
                 params.audio_url = uploadUrl;
             }
-            const res = yield this.client.post("/v2/transcript", params);
+            const data = yield this.fetchJson("/v2/transcript", {
+                method: "POST",
+                body: JSON.stringify(params),
+            });
             if ((_a = options === null || options === void 0 ? void 0 : options.poll) !== null && _a !== void 0 ? _a : true) {
-                return yield this.poll(res.data.id, options);
+                return yield this.poll(data.id, options);
             }
-            return res.data;
+            return data;
         });
     }
     poll(transcriptId, options) {
@@ -368,20 +387,28 @@ class TranscriptService extends BaseService {
      * @returns A promise that resolves to the transcript.
      */
     get(id) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const res = yield this.client.get(`/v2/transcript/${id}`);
-            return res.data;
-        });
+        return this.fetchJson(`/v2/transcript/${id}`);
     }
-    // TODO: add options overload to support list querystring parameters
     /**
-     * Retrieves a paged list of transcript listings.
-     * @param nextUrl The URL to retrieve the transcript list from. If not provided, the first page will be retrieved.
-     * @returns
+     * Retrieves a page of transcript listings.
+     * @param parameters The parameters to filter the transcript list by, or the URL to retrieve the transcript list from.
      */
-    list(nextUrl) {
+    list(parameters) {
         return __awaiter(this, void 0, void 0, function* () {
-            const { data } = yield this.client.get(nextUrl !== null && nextUrl !== void 0 ? nextUrl : "/v2/transcript");
+            let url = "/v2/transcript";
+            if (typeof parameters === "string") {
+                url = parameters;
+            }
+            else if (parameters) {
+                url = `${url}?${new URLSearchParams(Object.keys(parameters).map((key) => {
+                    var _a;
+                    return [
+                        key,
+                        ((_a = parameters[key]) === null || _a === void 0 ? void 0 : _a.toString()) || "",
+                    ];
+                }))}`;
+            }
+            const data = yield this.fetchJson(url);
             for (const transcriptListItem of data.transcripts) {
                 transcriptListItem.created = new Date(transcriptListItem.created);
                 if (transcriptListItem.completed) {
@@ -397,27 +424,17 @@ class TranscriptService extends BaseService {
      * @returns A promise that resolves to the transcript.
      */
     delete(id) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const res = yield this.client.delete(`/v2/transcript/${id}`);
-            return res.data;
-        });
+        return this.fetchJson(`/v2/transcript/${id}`, { method: "DELETE" });
     }
     /**
      * Search through the transcript for a specific set of keywords.
      * You can search for individual words, numbers, or phrases containing up to five words or numbers.
      * @param id The identifier of the transcript.
-     * @param id Keywords to search for.
+     * @param words Keywords to search for.
      * @return A promise that resolves to the sentences.
      */
     wordSearch(id, words) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const { data } = yield this.client.get(`/v2/transcript/${id}/word-search`, {
-                params: {
-                    words: JSON.stringify(words),
-                },
-            });
-            return data;
-        });
+        return this.fetchJson(`/v2/transcript/${id}/word-search?words=${JSON.stringify(words)}`);
     }
     /**
      * Retrieve all sentences of a transcript.
@@ -425,10 +442,7 @@ class TranscriptService extends BaseService {
      * @return A promise that resolves to the sentences.
      */
     sentences(id) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const { data } = yield this.client.get(`/v2/transcript/${id}/sentences`);
-            return data;
-        });
+        return this.fetchJson(`/v2/transcript/${id}/sentences`);
     }
     /**
      * Retrieve all paragraphs of a transcript.
@@ -436,10 +450,7 @@ class TranscriptService extends BaseService {
      * @return A promise that resolves to the paragraphs.
      */
     paragraphs(id) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const { data } = yield this.client.get(`/v2/transcript/${id}/paragraphs`);
-            return data;
-        });
+        return this.fetchJson(`/v2/transcript/${id}/paragraphs`);
     }
     /**
      * Retrieve subtitles of a transcript.
@@ -449,8 +460,8 @@ class TranscriptService extends BaseService {
      */
     subtitles(id, format = "srt") {
         return __awaiter(this, void 0, void 0, function* () {
-            const { data } = yield this.client.get(`/v2/transcript/${id}/${format}`);
-            return data;
+            const response = yield this.fetch(`/v2/transcript/${id}/${format}`);
+            return yield response.text();
         });
     }
     /**
@@ -459,10 +470,7 @@ class TranscriptService extends BaseService {
      * @return A promise that resolves to the subtitles text.
      */
     redactions(id) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const { data } = yield this.client.get(`/v2/transcript/${id}/redacted-audio`);
-            return data;
-        });
+        return this.fetchJson(`/v2/transcript/${id}/redacted-audio`);
     }
 }
 function getPath(path) {
@@ -482,55 +490,49 @@ function getPath(path) {
 class FileService extends BaseService {
     /**
      * Upload a local file to AssemblyAI.
-     * @param path The local file to upload.
+     * @param input The local file path to upload, or a stream or buffer of the file to upload.
      * @return A promise that resolves to the uploaded file URL.
      */
-    upload(path) {
+    upload(input) {
         return __awaiter(this, void 0, void 0, function* () {
-            const file = yield promises.readFile(path);
-            const { data } = yield this.client.post("/v2/upload", file, {
+            let fileData;
+            if (typeof input === "string")
+                fileData = fs.createReadStream(input);
+            else
+                fileData = input;
+            const data = yield this.fetchJson("/v2/upload", {
+                method: "POST",
+                body: fileData,
                 headers: {
                     "Content-Type": "application/octet-stream",
                 },
+                duplex: "half",
             });
             return data.upload_url;
         });
     }
 }
 
+const defaultBaseUrl = "https://api.assemblyai.com";
 class AssemblyAI {
     /**
      * Create a new AssemblyAI client.
      * @param params The parameters for the service, including the API key and base URL, if any.
      */
     constructor(params) {
-        params.baseUrl = params.baseUrl || "https://api.assemblyai.com";
-        const client = createAxiosClient(params);
-        this.files = new FileService(client);
-        this.transcripts = new TranscriptService(client, this.files);
-        this.lemur = new LemurService(client);
-        this.realtime = new RealtimeServiceFactory(client, params);
+        params.baseUrl = params.baseUrl || defaultBaseUrl;
+        if (params.baseUrl && params.baseUrl.endsWith("/"))
+            params.baseUrl = params.baseUrl.slice(0, -1);
+        this.files = new FileService(params);
+        this.transcripts = new TranscriptService(params, this.files);
+        this.lemur = new LemurService(params);
+        this.realtime = new RealtimeServiceFactory(params);
     }
 }
 
-var services = /*#__PURE__*/Object.freeze({
-  __proto__: null,
-  AssemblyAI: AssemblyAI,
-  FileService: FileService,
-  LemurService: LemurService,
-  RealtimeService: RealtimeService,
-  RealtimeServiceFactory: RealtimeServiceFactory,
-  TranscriptService: TranscriptService
-});
-
-class AssemblyAIExports extends AssemblyAI {
-}
-module.exports = Object.assign(AssemblyAIExports, services);
-
 exports.AssemblyAI = AssemblyAI;
 exports.FileService = FileService;
 exports.LemurService = LemurService;
 exports.RealtimeService = RealtimeService;
 exports.RealtimeServiceFactory = RealtimeServiceFactory;
 exports.TranscriptService = TranscriptService;
-exports.default = AssemblyAI;

package/dist/services/base.d.ts

@@ -1,12 +1,14 @@
-import { AxiosInstance } from "axios";
+import { BaseServiceParams } from "..";
 /**
  * Base class for services that communicate with the API.
  */
 export declare abstract class BaseService {
-    protected client: AxiosInstance;
+    private params;
     /**
      * Create a new service.
-     * @param params The AxiosInstance to send HTTP requests to the API.
+     * @param params The parameters to use for the service.
      */
-    constructor(client: AxiosInstance);
+    constructor(params: BaseServiceParams);
+    protected fetch(input: string, init?: RequestInit | undefined): Promise<Response>;
+    protected fetchJson<T>(input: string, init?: RequestInit | undefined): Promise<T>;
 }

package/dist/services/files/index.d.ts

@@ -1,9 +1,10 @@
-import { BaseService } from "@/services/base";
+import { BaseService } from "../base";
+import { FileUploadParameters } from "../..";
 export declare class FileService extends BaseService {
     /**
      * Upload a local file to AssemblyAI.
-     * @param path The local file to upload.
+     * @param input The local file path to upload, or a stream or buffer of the file to upload.
      * @return A promise that resolves to the uploaded file URL.
      */
-    upload(path: string): Promise<string>;
+    upload(input: FileUploadParameters): Promise<string>;
 }

package/dist/services/index.d.ts

@@ -1,4 +1,4 @@
-import { BaseServiceParams } from "@/types";
+import { BaseServiceParams } from "..";
 import { LemurService } from "./lemur";
 import { RealtimeService, RealtimeServiceFactory } from "./realtime";
 import { TranscriptService } from "./transcripts";

package/dist/services/lemur/index.d.ts

@@ -1,5 +1,5 @@
-import { LemurSummaryParameters, LemurActionItemsParameters, LemurQuestionAnswerParameters, LemurTaskParameters, LemurSummaryResponse, LemurQuestionAnswerResponse, LemurActionItemsResponse, LemurTaskResponse, PurgeLemurRequestDataResponse } from "@/types";
-import { BaseService } from "@/services/base";
+import { LemurSummaryParameters, LemurActionItemsParameters, LemurQuestionAnswerParameters, LemurTaskParameters, LemurSummaryResponse, LemurQuestionAnswerResponse, LemurActionItemsResponse, LemurTaskResponse, PurgeLemurRequestDataResponse } from "../..";
+import { BaseService } from "../base";
 export declare class LemurService extends BaseService {
     summary(params: LemurSummaryParameters): Promise<LemurSummaryResponse>;
     questionAnswer(params: LemurQuestionAnswerParameters): Promise<LemurQuestionAnswerResponse>;

package/dist/services/realtime/factory.d.ts

@@ -1,10 +1,9 @@
-import { BaseServiceParams, RealtimeTokenParams, CreateRealtimeServiceParams } from "@/types";
-import { AxiosInstance } from "axios";
+import { BaseServiceParams, RealtimeTokenParams, CreateRealtimeServiceParams } from "../..";
 import { RealtimeService } from "./service";
-export declare class RealtimeServiceFactory {
-    private client;
-    private params;
-    constructor(client: AxiosInstance, params: BaseServiceParams);
+import { BaseService } from "../base";
+export declare class RealtimeServiceFactory extends BaseService {
+    private rtFactoryParams;
+    constructor(params: BaseServiceParams);
     createService(params?: CreateRealtimeServiceParams): RealtimeService;
     createTemporaryToken(params: RealtimeTokenParams): Promise<string>;
 }

package/dist/services/realtime/index.d.ts

@@ -1,2 +1,2 @@
-export * from "@/services/realtime/factory";
-export * from "@/services/realtime/service";
+export * from "./factory";
+export * from "./service";

package/dist/services/realtime/service.d.ts

@@ -1,6 +1,5 @@
 /// <reference types="node" />
-import { RealtimeServiceParams, RealtimeTranscript, PartialTranscript, FinalTranscript, SessionBeginsEventData } from "@/types";
-import { Writable } from "stream";
+import { RealtimeServiceParams, RealtimeTranscript, PartialTranscript, FinalTranscript, SessionBeginsEventData } from "../..";
 export declare class RealtimeService {
     private realtimeUrl;
     private sampleRate;
@@ -20,6 +19,6 @@ export declare class RealtimeService {
     on(event: "close", listener: (code: number, reason: string) => void): void;
     connect(): Promise<SessionBeginsEventData>;
     sendAudio(audio: ArrayBuffer): void;
-    stream(): Writable;
+    stream(): NodeJS.WritableStream;
     close(waitForSessionTermination?: boolean): Promise<void>;
 }

package/dist/services/transcripts/index.d.ts

@@ -1,10 +1,9 @@
-import { BaseService } from "@/services/base";
-import { ParagraphsResponse, SentencesResponse, Transcript, TranscriptList, CreateTranscriptParameters, CreateTranscriptOptions, Createable, Deletable, Listable, Retrieveable, SubtitleFormat, RedactedAudioResponse, WordSearchResponse } from "@/types";
-import { AxiosInstance } from "axios";
+import { BaseService } from "../base";
+import { ParagraphsResponse, SentencesResponse, Transcript, TranscriptList, CreateTranscriptParameters, CreateTranscriptOptions, Createable, Deletable, Listable, Retrieveable, SubtitleFormat, RedactedAudioResponse, TranscriptListParameters, WordSearchResponse, BaseServiceParams } from "../..";
 import { FileService } from "../files";
 export declare class TranscriptService extends BaseService implements Createable<Transcript, CreateTranscriptParameters, CreateTranscriptOptions>, Retrieveable<Transcript>, Deletable<Transcript>, Listable<TranscriptList> {
     private files;
-    constructor(client: AxiosInstance, files: FileService);
+    constructor(params: BaseServiceParams, files: FileService);
     /**
      * Create a transcript.
      * @param params The parameters to create a transcript.
@@ -20,11 +19,10 @@ export declare class TranscriptService extends BaseService implements Createable
      */
     get(id: string): Promise<Transcript>;
     /**
-     * Retrieves a paged list of transcript listings.
-     * @param nextUrl The URL to retrieve the transcript list from. If not provided, the first page will be retrieved.
-     * @returns
+     * Retrieves a page of transcript listings.
+     * @param parameters The parameters to filter the transcript list by, or the URL to retrieve the transcript list from.
      */
-    list(nextUrl?: string | null): Promise<TranscriptList>;
+    list(parameters?: TranscriptListParameters | string): Promise<TranscriptList>;
     /**
      * Delete a transcript
      * @param id The identifier of the transcript.
@@ -35,7 +33,7 @@ export declare class TranscriptService extends BaseService implements Createable
      * Search through the transcript for a specific set of keywords.
      * You can search for individual words, numbers, or phrases containing up to five words or numbers.
      * @param id The identifier of the transcript.
-     * @param id Keywords to search for.
+     * @param words Keywords to search for.
      * @return A promise that resolves to the sentences.
      */
     wordSearch(id: string, words: string[]): Promise<WordSearchResponse>;

package/dist/types/files/index.d.ts

@@ -0,0 +1,4 @@
+/// <reference types="node" />
+type FileUploadParameters = string | FileUploadData;
+type FileUploadData = NodeJS.ReadableStream | ReadableStream | Blob | BufferSource | ArrayBufferView | ArrayBufferLike | Uint8Array;
+export type { FileUploadParameters, FileUploadData };

package/dist/types/index.d.ts

@@ -1,3 +1,4 @@
+export type * from "./files";
 export type * from "./transcripts";
 export type * from "./realtime";
 export type * from "./services";