Update README.md

langchain-ai · Feb 21, 2024 · c50176a · c50176a
1 parent 35420b7
commit c50176a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/docs/docs/integrations/providers/astradb.mdx b/docs/docs/integrations/providers/astradb.mdx
@@ -28,7 +28,7 @@ Learn more in the [example notebook](/docs/integrations/vectorstores/astradb).
 ## Chat message history
 
 ```python
-from langchain_community.chat_message_histories import AstraDBChatMessageHistory
+from langchain_astradb import AstraDBChatMessageHistory
 message_history = AstraDBChatMessageHistory(
     session_id="test-session",
     api_endpoint="...",

diff --git a/libs/partners/astradb/README.md b/libs/partners/astradb/README.md
@@ -20,7 +20,7 @@ pip install langchain-astradb
 ### Vector Store
 
 ```python
-from langchain_astradb.vectorstores import AstraDBVectorStore
+from langchain_astradb import AstraDBVectorStore
 
 my_store = AstraDBVectorStore(
   embedding=my_embeddings,
@@ -30,6 +30,17 @@ my_store = AstraDBVectorStore(
 )
 ```
 
+### Chat message history
+
+```python
+from langchain_astradb import AstraDBChatMessageHistory
+message_history = AstraDBChatMessageHistory(
+    session_id="test-session",
+    api_endpoint="...",
+    token="...",
+)
+```
+
 ## Reference
 
 See the [LangChain docs page](https://python.langchain.com/docs/integrations/providers/astradb) for a more detailed listing.