From 8ac5d8b8e443725eb4fa45f56ce3fb65bc73fd4f Mon Sep 17 00:00:00 2001 From: kingtroga Date: Mon, 11 Nov 2024 15:22:36 +0100 Subject: [PATCH] Fix again --- .../integrations/memory/falkordb_chat_message_history.ipynb | 4 +++- .../langchain_community/chat_message_histories/__init__.py | 6 +++--- .../test_falkordb_chat_message_history.py | 3 ++- 3 files changed, 8 insertions(+), 5 deletions(-) diff --git a/docs/docs/integrations/memory/falkordb_chat_message_history.ipynb b/docs/docs/integrations/memory/falkordb_chat_message_history.ipynb index 583ec76755d72..b0883184c06e2 100644 --- a/docs/docs/integrations/memory/falkordb_chat_message_history.ipynb +++ b/docs/docs/integrations/memory/falkordb_chat_message_history.ipynb @@ -30,7 +30,9 @@ "metadata": {}, "outputs": [], "source": [ - "from langchain_community.chat_message_histories.falkordb import FalkorDBChatMessageHistory\n", + "from langchain_community.chat_message_histories.falkordb import (\n", + " FalkorDBChatMessageHistory,\n", + ")\n", "\n", "history = FalkorDBChatMessageHistory(host=host, port=port, session_id=\"session_id_1\")\n", "\n", diff --git a/libs/community/langchain_community/chat_message_histories/__init__.py b/libs/community/langchain_community/chat_message_histories/__init__.py index 7e91d7b0f0235..914cf289c4647 100644 --- a/libs/community/langchain_community/chat_message_histories/__init__.py +++ b/libs/community/langchain_community/chat_message_histories/__init__.py @@ -33,15 +33,15 @@ from langchain_community.chat_message_histories.elasticsearch import ( ElasticsearchChatMessageHistory, ) + from langchain_community.chat_message_histories.falkordb import ( + FalkorDBChatMessageHistory, + ) from langchain_community.chat_message_histories.file import ( FileChatMessageHistory, ) 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, ) diff --git a/libs/community/tests/integration_tests/chat_message_histories/test_falkordb_chat_message_history.py b/libs/community/tests/integration_tests/chat_message_histories/test_falkordb_chat_message_history.py index 38bd34014bf36..5482b527c5fee 100644 --- a/libs/community/tests/integration_tests/chat_message_histories/test_falkordb_chat_message_history.py +++ b/libs/community/tests/integration_tests/chat_message_histories/test_falkordb_chat_message_history.py @@ -18,7 +18,8 @@ from langchain_core.messages import AIMessage, HumanMessage from langchain_community.chat_message_histories.falkordb import ( - FalkorDBChatMessageHistory) + FalkorDBChatMessageHistory, +) from langchain_community.graphs import FalkorDBGraph