From 6bedfdf25aed14c87e18daebcc4dee4ccb7f777c Mon Sep 17 00:00:00 2001
From: Patrick Loeber <50772274+patrickloeber@users.noreply.github.com>
Date: Thu, 24 Aug 2023 16:24:53 +0200
Subject: [PATCH] Fix docs for AssemblyAIAudioTranscriptLoader (shorter import
 path) (#9687)

Uses the shorter import path

`from langchain.document_loaders import` instead of the full path
`from langchain.document_loaders.assemblyai`

Applies those changes to the docs and the unit test.

See #9667 that adds this new loader.
---
 docs/extras/integrations/document_loaders/assemblyai.ipynb | 7 ++-----
 .../tests/unit_tests/document_loaders/test_assemblyai.py   | 6 ++----
 2 files changed, 4 insertions(+), 9 deletions(-)

diff --git a/docs/extras/integrations/document_loaders/assemblyai.ipynb b/docs/extras/integrations/document_loaders/assemblyai.ipynb
index 33fdef929d8..80d60df48b5 100644
--- a/docs/extras/integrations/document_loaders/assemblyai.ipynb
+++ b/docs/extras/integrations/document_loaders/assemblyai.ipynb
@@ -53,7 +53,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.document_loaders.assemblyai import AssemblyAIAudioTranscriptLoader\n",
+    "from langchain.document_loaders import AssemblyAIAudioTranscriptLoader\n",
     "\n",
     "audio_file = \"https://storage.googleapis.com/aai-docs-samples/nbc.mp3\"\n",
     "# or a local file path: audio_file = \"./nbc.mp3\"\n",
@@ -148,10 +148,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.document_loaders.assemblyai import (\n",
-    "    AssemblyAIAudioTranscriptLoader,\n",
-    "    TranscriptFormat,\n",
-    ")\n",
+    "from langchain.document_loaders.assemblyai import TranscriptFormat\n",
     "\n",
     "loader = AssemblyAIAudioTranscriptLoader(\n",
     "    file_path=\"./your_file.mp3\",\n",
diff --git a/libs/langchain/tests/unit_tests/document_loaders/test_assemblyai.py b/libs/langchain/tests/unit_tests/document_loaders/test_assemblyai.py
index a9b6112e7b1..550a2c0d1aa 100644
--- a/libs/langchain/tests/unit_tests/document_loaders/test_assemblyai.py
+++ b/libs/langchain/tests/unit_tests/document_loaders/test_assemblyai.py
@@ -1,10 +1,8 @@
 import pytest
 from pytest_mock import MockerFixture
 
-from langchain.document_loaders.assemblyai import (
-    AssemblyAIAudioTranscriptLoader,
-    TranscriptFormat,
-)
+from langchain.document_loaders import AssemblyAIAudioTranscriptLoader
+from langchain.document_loaders.assemblyai import TranscriptFormat
 
 
 @pytest.mark.requires("assemblyai")