From 3d5ba0873c3e2e38e7dac29665724881d8a6521e Mon Sep 17 00:00:00 2001
From: Niels Swimberghe <3382717+Swimburger@users.noreply.github.com>
Date: Wed, 24 Jul 2024 03:50:35 -0400
Subject: [PATCH] fix: update user agent in AssemblyAI (#1039)
Co-authored-by: Alex Yang <himself65@outlook.com>
---
.changeset/shy-pots-marry.md | 6 ++++++
examples/readers/package.json | 1 +
.../llamaindex/src/readers/AssemblyAIReader.ts | 14 ++++++++++++--
3 files changed, 19 insertions(+), 2 deletions(-)
create mode 100644 .changeset/shy-pots-marry.md
diff --git a/.changeset/shy-pots-marry.md b/.changeset/shy-pots-marry.md
new file mode 100644
index 000000000..1b1774d8f
--- /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 cd649b843..eb545b857 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 18d2b5460..74184cb8f 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[]>;
--
GitLab