@langchain/community/dist/document_loaders/web/assemblyai.js

Third-party integrations for LangChain.js

import { __exportAll } from "../../_virtual/_rolldown/runtime.js";
import { getEnvironmentVariable } from "@langchain/core/utils/env";
import { Document } from "@langchain/core/documents";
import { BaseDocumentLoader } from "@langchain/core/document_loaders/base";
import { AssemblyAI } from "assemblyai";
//#region src/document_loaders/web/assemblyai.ts
var assemblyai_exports = /* @__PURE__ */ __exportAll({
	AudioSubtitleLoader: () => AudioSubtitleLoader,
	AudioTranscriptLoader: () => AudioTranscriptLoader,
	AudioTranscriptParagraphsLoader: () => AudioTranscriptParagraphsLoader,
	AudioTranscriptSentencesLoader: () => AudioTranscriptSentencesLoader
});
const defaultOptions = { userAgent: { integration: {
	name: "LangChainJS",
	version: "1.0.1"
} } };
/**
* Base class for AssemblyAI loaders.
*/
var AssemblyAILoader = class extends BaseDocumentLoader {
	client;
	/**
	* Create a new AssemblyAI loader.
	* @param assemblyAIOptions The options to configure the AssemblyAI loader.
	* Configure the `assemblyAIOptions.apiKey` with your AssemblyAI API key, or configure it as the `ASSEMBLYAI_API_KEY` environment variable.
	*/
	constructor(assemblyAIOptions) {
		super();
		let options = assemblyAIOptions;
		if (!options) options = {};
		if (!options.apiKey) options.apiKey = getEnvironmentVariable("ASSEMBLYAI_API_KEY");
		if (!options.apiKey) throw new Error("No AssemblyAI API key provided");
		this.client = new AssemblyAI({
			...defaultOptions,
			...options
		});
	}
};
var CreateTranscriptLoader = class extends AssemblyAILoader {
	transcribeParams;
	transcriptId;
	/**
	* Transcribe audio or retrieve an existing transcript by its ID.
	* @param params The parameters to transcribe audio, or the ID of the transcript to retrieve.
	* @param assemblyAIOptions The options to configure the AssemblyAI loader.
	* Configure the `assemblyAIOptions.apiKey` with your AssemblyAI API key, or configure it as the `ASSEMBLYAI_API_KEY` environment variable.
	*/
	constructor(params, assemblyAIOptions) {
		super(assemblyAIOptions);
		if (typeof params === "string") this.transcriptId = params;
		else this.transcribeParams = params;
	}
	async transcribeOrGetTranscript() {
		if (this.transcriptId) return await this.client.transcripts.get(this.transcriptId);
		if (this.transcribeParams) {
			let transcribeParams;
			if ("audio_url" in this.transcribeParams) transcribeParams = {
				...this.transcribeParams,
				audio: this.transcribeParams.audio_url
			};
			else transcribeParams = this.transcribeParams;
			return await this.client.transcripts.transcribe(transcribeParams);
		} else throw new Error("No transcript ID or transcribe parameters provided");
	}
};
/**
* Transcribe audio and load the transcript as a document using AssemblyAI.
*/
var AudioTranscriptLoader = class extends CreateTranscriptLoader {
	/**
	* Transcribe audio and load the transcript as a document using AssemblyAI.
	* @returns A promise that resolves to a single document containing the transcript text
	* as the page content, and the transcript object as the metadata.
	*/
	async load() {
		const transcript = await this.transcribeOrGetTranscript();
		return [new Document({
			pageContent: transcript.text,
			metadata: transcript
		})];
	}
};
/**
* Transcribe audio and load the paragraphs of the transcript, creating a document for each paragraph.
*/
var AudioTranscriptParagraphsLoader = class extends CreateTranscriptLoader {
	/**
	* Transcribe audio and load the paragraphs of the transcript, creating a document for each paragraph.
	* @returns A promise that resolves to an array of documents, each containing a paragraph of the transcript.
	*/
	async load() {
		const transcript = await this.transcribeOrGetTranscript();
		return (await this.client.transcripts.paragraphs(transcript.id)).paragraphs.map((p) => new Document({
			pageContent: p.text,
			metadata: p
		}));
	}
};
/**
* Transcribe audio and load the sentences of the transcript, creating a document for each sentence.
*/
var AudioTranscriptSentencesLoader = class extends CreateTranscriptLoader {
	/**
	* Transcribe audio and load the sentences of the transcript, creating a document for each sentence.
	* @returns A promise that resolves to an array of documents, each containing a sentence of the transcript.
	*/
	async load() {
		const transcript = await this.transcribeOrGetTranscript();
		return (await this.client.transcripts.sentences(transcript.id)).sentences.map((p) => new Document({
			pageContent: p.text,
			metadata: p
		}));
	}
};
/**
* Transcribe audio and load subtitles for the transcript as `srt` or `vtt` format.
*/
var AudioSubtitleLoader = class extends CreateTranscriptLoader {
	/**
	* Create a new AudioSubtitleLoader.
	* @param params The parameters to transcribe audio, or the ID of the transcript to retrieve.
	* @param subtitleFormat The format of the subtitles, either `srt` or `vtt`.
	* @param assemblyAIOptions The options to configure the AssemblyAI loader.
	* Configure the `assemblyAIOptions.apiKey` with your AssemblyAI API key, or configure it as the `ASSEMBLYAI_API_KEY` environment variable.
	*/
	constructor(params, subtitleFormat = "srt", assemblyAIOptions) {
		super(params, assemblyAIOptions);
		this.subtitleFormat = subtitleFormat;
		this.subtitleFormat = subtitleFormat;
	}
	/**
	* Transcribe audio and load subtitles for the transcript as `srt` or `vtt` format.
	* @returns A promise that resolves a document containing the subtitles as the page content.
	*/
	async load() {
		const transcript = await this.transcribeOrGetTranscript();
		return [new Document({ pageContent: await this.client.transcripts.subtitles(transcript.id, this.subtitleFormat) })];
	}
};
//#endregion
export { AudioSubtitleLoader, AudioTranscriptLoader, AudioTranscriptParagraphsLoader, AudioTranscriptSentencesLoader, assemblyai_exports };

//# sourceMappingURL=assemblyai.js.map