github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-30 10:23:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

[community]: Elasticsearch chat history encoding (#15055)

Browse Source 
- Added ensure_ascii property to ElasticsearchChatMessageHistory

<!-- Thank you for contributing to LangChain!

Please title your PR "<package>: <description>", where <package> is
whichever of langchain, community, core, experimental, etc. is being
modified.

Replace this entire comment with:
  - **Description:** a description of the change, 
  - **Issue:** the issue # it fixes if applicable,
  - **Dependencies:** any dependencies required for this change,
- **Twitter handle:** we announce bigger features on Twitter. If your PR
gets announced, and you'd like a mention, we'll gladly shout you out!

Please make sure your PR is passing linting and testing before
submitting. Run `make format`, `make lint` and `make test` from the root
of the package you've modified to check this locally.

See contribution guidelines for more information on how to write/run
tests, lint, etc: https://python.langchain.com/docs/contributing/

If you're adding a new integration, please include:
1. a test for the integration, preferably unit tests that do not rely on
network access,
2. an example notebook showing its use. It lives in
`docs/docs/integrations` directory.

If no one reviews your PR within a few days, please @-mention one of
@baskaryan, @eyurtsev, @hwchase17.
 -->

---------

Co-authored-by: Ivan Chetverikov <ivan.chetverikov@raftds.com>
Co-authored-by: Harrison Chase <hw.chase.17@gmail.com>
This commit is contained in:
Ivan 2023-12-23 00:21:34 +03:00 committed by GitHub
parent 9e492620d4
commit 59d4b80a92
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libs/community/langchain_community/chat_message_histories/elasticsearch.py
View File

@ -26,6 +26,7 @@ class ElasticsearchChatMessageHistory(BaseChatMessageHistory):
es_password: Password to use when connecting to Elasticsearch. es_password: Password to use when connecting to Elasticsearch.
es_api_key: API key to use when connecting to Elasticsearch. es_api_key: API key to use when connecting to Elasticsearch.
es_connection: Optional pre-existing Elasticsearch connection. es_connection: Optional pre-existing Elasticsearch connection.
esnsure_ascii: Used to escape ASCII symbols in json.dumps. Defaults to True.
index: Name of the index to use. index: Name of the index to use.
session_id: Arbitrary key that is used to store the messages session_id: Arbitrary key that is used to store the messages
of a single chat session. of a single chat session.
@ -42,9 +43,11 @@ class ElasticsearchChatMessageHistory(BaseChatMessageHistory):
es_user: Optional[str] = None, es_user: Optional[str] = None,
es_api_key: Optional[str] = None, es_api_key: Optional[str] = None,
es_password: Optional[str] = None, es_password: Optional[str] = None,
esnsure_ascii: Optional[bool] = True,
): ):
self.index: str = index self.index: str = index
self.session_id: str = session_id self.session_id: str = session_id
self.ensure_ascii: bool = esnsure_ascii
# Initialize Elasticsearch client from passed client arg or connection info # Initialize Elasticsearch client from passed client arg or connection info
if es_connection is not None: if es_connection is not None:
@ -172,7 +175,10 @@ class ElasticsearchChatMessageHistory(BaseChatMessageHistory):
document={ document={
"session_id": self.session_id, "session_id": self.session_id,
"created_at": round(time() * 1000), "created_at": round(time() * 1000),
"history": json.dumps(message_to_dict(message)), "history": json.dumps(
message_to_dict(message),
ensure_ascii=self.ensure_ascii,
),
}, },
refresh=True, refresh=True,
) )