diff --git a/docs/modules/indexes/document_loaders/examples/bilibili.ipynb b/docs/modules/indexes/document_loaders/examples/bilibili.ipynb index d9a1d07f..7663bbd7 100644 --- a/docs/modules/indexes/document_loaders/examples/bilibili.ipynb +++ b/docs/modules/indexes/document_loaders/examples/bilibili.ipynb @@ -5,7 +5,7 @@ "id": "66a7777e", "metadata": {}, "source": [ - "# Bilibili\n", + "# BiliBili\n", "\n", ">[Bilibili](https://www.bilibili.tv/) is one of the most beloved long-form video sites in China.\n", "\n", @@ -35,7 +35,7 @@ }, "outputs": [], "source": [ - "from langchain.document_loaders.bilibili import BiliBiliLoader" + "from langchain.document_loaders import BiliBiliLoader" ] }, { diff --git a/langchain/document_loaders/bilibili.py b/langchain/document_loaders/bilibili.py index 909b060a..962aa4df 100644 --- a/langchain/document_loaders/bilibili.py +++ b/langchain/document_loaders/bilibili.py @@ -60,11 +60,11 @@ class BiliBiliLoader(BaseLoader): raw_sub_titles = json.loads(result.content)["body"] raw_transcript = " ".join([c["content"] for c in raw_sub_titles]) - raw_transcript_with_meta_info = f""" - Video Title: {video_info['title']}, - description: {video_info['desc']}\n - Transcript: {raw_transcript} - """ + raw_transcript_with_meta_info = ( + f"Video Title: {video_info['title']}," + f"description: {video_info['desc']}\n\n" + f"Transcript: {raw_transcript}" + ) return raw_transcript_with_meta_info, video_info else: raw_transcript = "" diff --git a/tests/integration_tests/document_loaders/test_bilibili.py b/tests/integration_tests/document_loaders/test_bilibili.py index 191d7bd8..cf1affdf 100644 --- a/tests/integration_tests/document_loaders/test_bilibili.py +++ b/tests/integration_tests/document_loaders/test_bilibili.py @@ -1,4 +1,4 @@ -from langchain.document_loaders.bilibili import BiliBiliLoader +from langchain.document_loaders import BiliBiliLoader def test_bilibili_loader() -> None: