Fixed linting and tests

langchain-ai · Sep 3, 2024 · 75e0065 · 75e0065
1 parent 5040082
commit 75e0065
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/libs/community/langchain_community/chat_message_histories/__init__.py b/libs/community/langchain_community/chat_message_histories/__init__.py
@@ -131,7 +131,7 @@
     "MomentoChatMessageHistory": "langchain_community.chat_message_histories.momento",
     "MongoDBChatMessageHistory": "langchain_community.chat_message_histories.mongodb",
     "Neo4jChatMessageHistory": "langchain_community.chat_message_histories.neo4j",
-    "OpenSearchChatMessageHistory": "langchain_community.chat_message_histories.opensearch",
+    "OpenSearchChatMessageHistory": "langchain_community.chat_message_histories.opensearch",  # noqa: E501
     "PostgresChatMessageHistory": "langchain_community.chat_message_histories.postgres",
     "RedisChatMessageHistory": "langchain_community.chat_message_histories.redis",
     "RocksetChatMessageHistory": "langchain_community.chat_message_histories.rocksetdb",

diff --git a/libs/community/langchain_community/chat_message_histories/opensearch.py b/libs/community/langchain_community/chat_message_histories/opensearch.py
@@ -1,14 +1,11 @@
 import json
 import logging
 from time import time
-from typing import TYPE_CHECKING, List
+from typing import List
 
 from langchain_core.chat_history import BaseChatMessageHistory
 from langchain_core.messages import BaseMessage, message_to_dict, messages_from_dict
 
-if TYPE_CHECKING:
-    from opensearchpy import OpenSearch
-
 logger = logging.getLogger(__name__)
 
 DEFAULT_INDEX_NAME = "chat-history"

diff --git a/libs/community/tests/unit_tests/chat_message_histories/test_imports.py b/libs/community/tests/unit_tests/chat_message_histories/test_imports.py
@@ -12,6 +12,7 @@
     "MomentoChatMessageHistory",
     "MongoDBChatMessageHistory",
     "Neo4jChatMessageHistory",
+    "OpenSearchChatMessageHistory",
     "PostgresChatMessageHistory",
     "RedisChatMessageHistory",
     "RocksetChatMessageHistory",