diff --git a/.changeset/shy-pots-marry.md b/.changeset/shy-pots-marry.md new file mode 100644 index 0000000000000000000000000000000000000000..1b1774d8f2177d78b83de3914311aa622ae0dbfa --- /dev/null +++ b/.changeset/shy-pots-marry.md @@ -0,0 +1,6 @@ +--- +"llamaindex": patch +"llamaindex-loader-example": patch +--- + +fix: update user agent in AssemblyAI diff --git a/examples/readers/package.json b/examples/readers/package.json index cd649b84312d589ae8757fed80c30ee1c42d0e7d..eb545b85796e6a0146f778fc2840bc1d8448833b 100644 --- a/examples/readers/package.json +++ b/examples/readers/package.json @@ -11,6 +11,7 @@ "start:pdf": "node --import tsx ./src/pdf.ts", "start:llamaparse": "node --import tsx ./src/llamaparse.ts", "start:notion": "node --import tsx ./src/notion.ts", + "start:assemblyai": "node --import tsx ./src/assemblyai.ts", "start:llamaparse-dir": "node --import tsx ./src/simple-directory-reader-with-llamaparse.ts", "start:llamaparse-json": "node --import tsx ./src/llamaparse-json.ts", "start:discord": "node --import tsx ./src/discord.ts" diff --git a/packages/llamaindex/src/readers/AssemblyAIReader.ts b/packages/llamaindex/src/readers/AssemblyAIReader.ts index 18d2b54604fee1cae35f05bc747b4f285d82a8a2..74184cb8fd4ae06fcefd225f7a930e86346de891 100644 --- a/packages/llamaindex/src/readers/AssemblyAIReader.ts +++ b/packages/llamaindex/src/readers/AssemblyAIReader.ts @@ -11,7 +11,14 @@ import { AssemblyAI } from "assemblyai"; import type { BaseReader } from "./type.js"; type AssemblyAIOptions = Partial<BaseServiceParams>; - +const defaultOptions = { + userAgent: { + integration: { + name: "LlamaIndexTS", + version: "1.0.1", + }, + }, +}; /** * Base class for AssemblyAI Readers. */ @@ -37,7 +44,10 @@ abstract class AssemblyAIReader implements BaseReader { ); } - this.client = new AssemblyAI(options as BaseServiceParams); + this.client = new AssemblyAI({ + ...defaultOptions, + ...options, + } as BaseServiceParams); } abstract loadData(params: TranscribeParams | string): Promise<Document[]>;