FIX CI CD ERRORS AGAIN

langchain-ai · Nov 11, 2024 · ebc4732 · ebc4732
1 parent 73edb06
commit ebc4732
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 4 deletions.
diff --git a/libs/community/langchain_community/chat_message_histories/__init__.py b/libs/community/langchain_community/chat_message_histories/__init__.py
@@ -1,6 +1,5 @@
 """**Chat message history** stores a history of the message interactions in a chat.
 
-
 **Class hierarchy:**
 
 .. code-block::
@@ -40,6 +39,9 @@
     from langchain_community.chat_message_histories.firestore import (
         FirestoreChatMessageHistory,
     )
+    from langchain_community.chat_message_histories.falkordb import (
+        FalkorDBChatMessageHistory,
+    )
     from langchain_community.chat_message_histories.in_memory import (
         ChatMessageHistory,
     )
@@ -98,6 +100,7 @@
     "ElasticsearchChatMessageHistory",
     "FileChatMessageHistory",
     "FirestoreChatMessageHistory",
+    "FalkorDBChatMessageHistory",
     "MomentoChatMessageHistory",
     "MongoDBChatMessageHistory",
     "Neo4jChatMessageHistory",
@@ -124,6 +127,7 @@
     "ElasticsearchChatMessageHistory": "langchain_community.chat_message_histories.elasticsearch",  # noqa: E501
     "FileChatMessageHistory": "langchain_community.chat_message_histories.file",
     "FirestoreChatMessageHistory": "langchain_community.chat_message_histories.firestore",  # noqa: E501
+    "FalkorDBChatMessageHistory": "langchain_community.chat_message_histories.falkordb",
     "MomentoChatMessageHistory": "langchain_community.chat_message_histories.momento",
     "MongoDBChatMessageHistory": "langchain_community.chat_message_histories.mongodb",
     "Neo4jChatMessageHistory": "langchain_community.chat_message_histories.neo4j",

diff --git a/libs/community/poetry.lock b/libs/community/poetry.lock
diff --git a/libs/community/pyproject.toml b/libs/community/pyproject.toml
@@ -41,6 +41,7 @@ dataclasses-json = ">= 0.5.7, < 0.7"
 pydantic-settings = "^2.4.0"
 langsmith = "^0.1.125"
 httpx-sse = "^0.4.0"
+redis = "^5.2.0"
 [[tool.poetry.dependencies.numpy]]
 version = "^1"
 python = "<3.12"

diff --git a/...nity/tests/integration_tests/chat_message_histories/test_falkordb_chat_message_history.py b/...nity/tests/integration_tests/chat_message_histories/test_falkordb_chat_message_history.py
@@ -17,7 +17,9 @@
 
 from langchain_core.messages import AIMessage, HumanMessage
 
-from langchain_community.chat_message_histories.falkordb import FalkorDBChatMessageHistory
+from langchain_community.chat_message_histories.falkordb import (
+    FalkorDBChatMessageHistory,
+)
 from langchain_community.graphs import FalkorDBGraph