diff --git a/docs/docs/integrations/providers/astradb.mdx b/docs/docs/integrations/providers/astradb.mdx index 8cc4788cc2..9f553c63fa 100644 --- a/docs/docs/integrations/providers/astradb.mdx +++ b/docs/docs/integrations/providers/astradb.mdx @@ -105,7 +105,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", api_endpoint="...", @@ -118,7 +118,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", api_endpoint="...", diff --git a/libs/partners/astradb/README.md b/libs/partners/astradb/README.md index a4c2dc84e1..e14f4cfe92 100644 --- a/libs/partners/astradb/README.md +++ b/libs/partners/astradb/README.md @@ -30,6 +30,28 @@ my_store = AstraDBVectorStore( ) ``` +### Store + +```python +from langchain_astradb import AstraDBStore +store = AstraDBStore( + collection_name="my_kv_store", + api_endpoint="...", + token="..." +) +``` + +### Byte Store + +```python +from langchain_astradb import AstraDBByteStore +store = AstraDBByteStore( + collection_name="my_kv_store", + api_endpoint="...", + token="..." +) +``` + ## Reference See the [LangChain docs page](https://python.langchain.com/docs/integrations/providers/astradb) for a more detailed listing.