From f2a7dda4bdcdff3008b384548b234ab165ebb8f6 Mon Sep 17 00:00:00 2001
From: Christophe Bornet <cbornet@hotmail.com>
Date: Fri, 15 Mar 2024 23:57:25 +0100
Subject: [PATCH] community[patch]: Use langchain-astradb for AstraDB doc
 loader (#19071)

Co-authored-by: Bagatur <baskaryan@gmail.com>
Co-authored-by: Bagatur <22008038+baskaryan@users.noreply.github.com>
---
 docs/docs/integrations/providers/astradb.mdx        | 13 ++++---------
 .../langchain_community/document_loaders/astradb.py |  6 ++++++
 2 files changed, 10 insertions(+), 9 deletions(-)

diff --git a/docs/docs/integrations/providers/astradb.mdx b/docs/docs/integrations/providers/astradb.mdx
index f1e9cf0705069..bda862139579e 100644
--- a/docs/docs/integrations/providers/astradb.mdx
+++ b/docs/docs/integrations/providers/astradb.mdx
@@ -10,12 +10,7 @@ See a [tutorial provided by DataStax](https://docs.datastax.com/en/astra/astra-d
 
 Install the following Python package:
 ```bash
-pip install "langchain-astradb>=0.0.1"
-```
-
-Some old integrations require the `astrapy` package:
-```bash
-pip install "astrapy>=0.7.1"
+pip install "langchain-astradb>=0.1.0"
 ```
 
 Get the [connection secrets](https://docs.datastax.com/en/astra/astra-db-vector/get-started/quickstart.html).
@@ -92,7 +87,7 @@ Learn more in the [example notebook](/docs/integrations/memory/astradb_chat_mess
 ## Document loader
 
 ```python
-from langchain_community.document_loaders import AstraDBLoader
+from langchain_astradb import AstraDBLoader
 
 loader = AstraDBLoader(
     collection_name="my_collection",
@@ -129,7 +124,7 @@ Learn more in the [example notebook](/docs/integrations/retrievers/self_query/as
 ## Store
 
 ```python
-from langchain_community.storage import AstraDBStore
+from langchain_astradb import AstraDBStore
 
 store = AstraDBStore(
     collection_name="my_kv_store",
@@ -143,7 +138,7 @@ Learn more in the [example notebook](/docs/integrations/stores/astradb#astradbst
 ## Byte Store
 
 ```python
-from langchain_community.storage import AstraDBByteStore
+from langchain_astradb import AstraDBByteStore
 
 store = AstraDBByteStore(
     collection_name="my_kv_store",
diff --git a/libs/community/langchain_community/document_loaders/astradb.py b/libs/community/langchain_community/document_loaders/astradb.py
index 5f1b82cc429e8..2898f2eaec284 100644
--- a/libs/community/langchain_community/document_loaders/astradb.py
+++ b/libs/community/langchain_community/document_loaders/astradb.py
@@ -13,6 +13,7 @@
     Optional,
 )
 
+from langchain_core._api.deprecation import deprecated
 from langchain_core.documents import Document
 
 from langchain_community.document_loaders.base import BaseLoader
@@ -24,6 +25,11 @@
 logger = logging.getLogger(__name__)
 
 
+@deprecated(
+    since="0.0.29",
+    removal="0.2.0",
+    alternative_import="langchain_astradb.AstraDBLoader",
+)
 class AstraDBLoader(BaseLoader):
     def __init__(
         self,