assemblyai 2.0.2 → 3.0.0

package/README.md

@@ -40,7 +40,7 @@ bun add assemblyai
 Import the AssemblyAI package and create an AssemblyAI object with your API key:
 
 ```javascript
-import AssemblyAI from "assemblyai";
+import { AssemblyAI } from "assemblyai";
 
 const client = new AssemblyAI({
   apiKey: process.env.ASSEMBLYAI_API_KEY,
@@ -94,7 +94,7 @@ const transcript = await client.transcripts.get(transcript.id)
 
 ## List transcripts
 
-This will return a paged list of transcripts that you have transcript.
+This will return a page of transcripts that you have transcript.
 
 ```javascript
 const page = await client.transcripts.list()

package/dist/index.d.ts

@@ -1,4 +1,2 @@
-import { AssemblyAI } from "./services";
-export * from "./services";
 export type * from "./types";
-export default AssemblyAI;
+export * from "./services";

package/dist/index.esm.js

@@ -1,7 +1,7 @@
 import axios, { isAxiosError } from 'axios';
 import WebSocket from 'ws';
 import { Writable } from 'stream';
-import { readFile } from 'fs/promises';
+import fs from 'fs';
 
 function createAxiosClient(params) {
     const client = axios.create({
@@ -369,15 +369,23 @@ class TranscriptService extends BaseService {
             return res.data;
         });
     }
-    // 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";
+            let query;
+            if (typeof parameters === "string") {
+                url = parameters;
+            }
+            else if (parameters) {
+                query = parameters;
+            }
+            const { data } = yield this.client.get(url, {
+                params: query,
+            });
             for (const transcriptListItem of data.transcripts) {
                 transcriptListItem.created = new Date(transcriptListItem.created);
                 if (transcriptListItem.completed) {
@@ -478,13 +486,17 @@ 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 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.client.post("/v2/upload", fileData, {
                 headers: {
                     "Content-Type": "application/octet-stream",
                 },
@@ -509,18 +521,4 @@ class AssemblyAI {
     }
 }
 
-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);
-
-export { AssemblyAI, FileService, LemurService, RealtimeService, RealtimeServiceFactory, TranscriptService, AssemblyAI as default };
+export { AssemblyAI, FileService, LemurService, RealtimeService, RealtimeServiceFactory, TranscriptService };

package/dist/index.js

@@ -1,11 +1,9 @@
 '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');
+var fs = require('fs');
 
 function createAxiosClient(params) {
     const client = axios.create({
@@ -373,15 +371,23 @@ class TranscriptService extends BaseService {
             return res.data;
         });
     }
-    // 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";
+            let query;
+            if (typeof parameters === "string") {
+                url = parameters;
+            }
+            else if (parameters) {
+                query = parameters;
+            }
+            const { data } = yield this.client.get(url, {
+                params: query,
+            });
             for (const transcriptListItem of data.transcripts) {
                 transcriptListItem.created = new Date(transcriptListItem.created);
                 if (transcriptListItem.completed) {
@@ -482,13 +488,17 @@ 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.client.post("/v2/upload", fileData, {
                 headers: {
                     "Content-Type": "application/octet-stream",
                 },
@@ -513,24 +523,9 @@ class AssemblyAI {
     }
 }
 
-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/files/index.d.ts

@@ -1,9 +1,10 @@
 import { BaseService } from "@/services/base";
+import { FileUploadParameters } from "@/types";
 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/transcripts/index.d.ts

@@ -1,5 +1,5 @@
 import { BaseService } from "@/services/base";
-import { ParagraphsResponse, SentencesResponse, Transcript, TranscriptList, CreateTranscriptParameters, CreateTranscriptOptions, Createable, Deletable, Listable, Retrieveable, SubtitleFormat, RedactedAudioResponse, WordSearchResponse } from "@/types";
+import { ParagraphsResponse, SentencesResponse, Transcript, TranscriptList, CreateTranscriptParameters, CreateTranscriptOptions, Createable, Deletable, Listable, Retrieveable, SubtitleFormat, RedactedAudioResponse, TranscriptListParameters, WordSearchResponse } from "@/types";
 import { AxiosInstance } from "axios";
 import { FileService } from "../files";
 export declare class TranscriptService extends BaseService implements Createable<Transcript, CreateTranscriptParameters, CreateTranscriptOptions>, Retrieveable<Transcript>, Deletable<Transcript>, Listable<TranscriptList> {
@@ -20,11 +20,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.

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

@@ -0,0 +1,5 @@
+/// <reference types="node" />
+/// <reference types="node" />
+type FileUploadParameters = string | FileUploadData;
+type FileUploadData = NodeJS.ReadableStream | ReadableStream | Buffer | 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";

package/dist/types/openapi.generated.d.ts

@@ -620,6 +620,27 @@ export type TranscriptListItem = {
     resource_url: string;
     status: TranscriptStatus;
 };
+export type TranscriptListParameters = {
+    /** @description Get transcripts that were created after this transcript ID */
+    after_id?: string;
+    /** @description Get transcripts that were created before this transcript ID */
+    before_id?: string;
+    /**
+     * Format: date
+     * @description Only get transcripts created on this date
+     */
+    created_on?: string;
+    /**
+     * Format: int64
+     * @description Maximum amount of transcripts to retrieve
+     * @default 10
+     */
+    limit?: number;
+    /** @description Filter by transcript status */
+    status?: TranscriptStatus;
+    /** @description Only get throttled transcripts, overrides the status filter */
+    throttled_only?: boolean;
+};
 export type TranscriptParagraph = {
     /** Format: double */
     confidence: number;

package/dist/utils/axios.d.ts

@@ -1,3 +1,3 @@
-import { BaseServiceParams } from "../.";
+import { BaseServiceParams } from "@/types";
 export declare function createAxiosClient(params: BaseServiceParams): import("axios").AxiosInstance;
 export declare function throwApiError(error: unknown): Promise<never>;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "assemblyai",
-  "version": "2.0.2",
+  "version": "3.0.0",
   "description": "The AssemblyAI Node.js SDK provides an easy-to-use interface for interacting with the AssemblyAI API, which supports async and real-time transcription, as well as the latest LeMUR models.",
   "main": "dist/index.js",
   "module": "dist/index.esm.js",

package/src/index.ts

@@ -1,7 +1,2 @@
-import * as services from "./services";
-import { AssemblyAI } from "./services";
-export * from "./services";
 export type * from "./types";
-export default AssemblyAI;
-class AssemblyAIExports extends AssemblyAI {}
-module.exports = Object.assign(AssemblyAIExports, services);
+export * from "./services";

package/src/services/files/index.ts

@@ -1,21 +1,29 @@
-import { readFile } from "fs/promises";
+// import the fs module instead if specific named exports
+// to keep the assemblyai module more compatible. Some fs polyfills don't include `createReadStream`.
+import fs from "fs";
 import { BaseService } from "@/services/base";
-import { UploadedFile } from "@/types";
+import { UploadedFile, FileUploadParameters, FileUploadData } from "@/types";
 
 export 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.
    */
-  async upload(path: string): Promise<string> {
-    const file = await readFile(path);
+  async upload(input: FileUploadParameters): Promise<string> {
+    let fileData: FileUploadData;
+    if (typeof input === "string") fileData = fs.createReadStream(input);
+    else fileData = input;
 
-    const { data } = await this.client.post<UploadedFile>("/v2/upload", file, {
-      headers: {
-        "Content-Type": "application/octet-stream",
-      },
-    });
+    const { data } = await this.client.post<UploadedFile>(
+      "/v2/upload",
+      fileData,
+      {
+        headers: {
+          "Content-Type": "application/octet-stream",
+        },
+      }
+    );
 
     return data.upload_url;
   }

package/src/services/transcripts/index.ts

@@ -12,6 +12,7 @@ import {
   Retrieveable,
   SubtitleFormat,
   RedactedAudioResponse,
+  TranscriptListParameters,
   WordSearchResponse,
 } from "@/types";
 import { AxiosInstance } from "axios";
@@ -87,16 +88,23 @@ export class TranscriptService
     return res.data;
   }
 
-  // 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.
    */
-  async list(nextUrl?: string | null): Promise<TranscriptList> {
-    const { data } = await this.client.get<TranscriptList>(
-      nextUrl ?? "/v2/transcript"
-    );
+  async list(
+    parameters?: TranscriptListParameters | string
+  ): Promise<TranscriptList> {
+    let url = "/v2/transcript";
+    let query: TranscriptListParameters | undefined;
+    if (typeof parameters === "string") {
+      url = parameters;
+    } else if (parameters) {
+      query = parameters;
+    }
+    const { data } = await this.client.get<TranscriptList>(url, {
+      params: query,
+    });
     for (const transcriptListItem of data.transcripts) {
       transcriptListItem.created = new Date(transcriptListItem.created);
       if (transcriptListItem.completed) {

package/src/types/files/index.ts

@@ -0,0 +1,10 @@
+type FileUploadParameters = string | FileUploadData;
+type FileUploadData =
+  | NodeJS.ReadableStream
+  | ReadableStream
+  | Buffer
+  | ArrayBufferView
+  | ArrayBufferLike
+  | Uint8Array;
+
+export type { FileUploadParameters, FileUploadData };

package/src/types/index.ts

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

package/src/types/openapi.generated.ts

@@ -762,6 +762,28 @@ export type TranscriptListItem = {
   status: TranscriptStatus;
 };
 
+export type TranscriptListParameters = {
+  /** @description Get transcripts that were created after this transcript ID */
+  after_id?: string;
+  /** @description Get transcripts that were created before this transcript ID */
+  before_id?: string;
+  /**
+   * Format: date
+   * @description Only get transcripts created on this date
+   */
+  created_on?: string;
+  /**
+   * Format: int64
+   * @description Maximum amount of transcripts to retrieve
+   * @default 10
+   */
+  limit?: number;
+  /** @description Filter by transcript status */
+  status?: TranscriptStatus;
+  /** @description Only get throttled transcripts, overrides the status filter */
+  throttled_only?: boolean;
+};
+
 export type TranscriptParagraph = {
   /** Format: double */
   confidence: number;

package/src/utils/axios.ts

@@ -1,5 +1,5 @@
 import axios, { isAxiosError } from "axios";
-import { BaseServiceParams } from "../.";
+import { BaseServiceParams } from "@/types";
 
 export function createAxiosClient(params: BaseServiceParams) {
   const client = axios.create({