Elasticsearch
Elasticsearch 是一个分布式、RESTful 的搜索和分析引擎,能够执行向量和词法搜索。它建立在 Apache Lucene 库之上。
本笔记本展示了如何将聊天消息历史功能与 Elasticsearch
一起使用。
设置 Elasticsearch
有两种主要方法可以设置 Elasticsearch 实例:
-
Elastic Cloud。 Elastic Cloud 是一项托管的 Elasticsearch 服务。注册一个免费试用。
-
本地 Elasticsearch 安装。 通过在本地运行 Elasticsearch 来开始使用它。最简单的方法是使用官方的 Elasticsearch Docker 镜像。有关更多信息,请参阅Elasticsearch Docker 文档。
安装依赖
%pip install --upgrade --quiet elasticsearch langchain langchain-community
认证
如何获取默认“elastic”用户的密码
要获取默认“elastic”用户的 Elastic Cloud 密码:
- 登录到 Elastic Cloud 控制台
- 转到“安全”>“用户”
- 找到“elastic”用户并点击“编辑”
- 点击“重置密码”
- 按照提示重置密码
使用用户名/密码
es_username = os.environ.get("ES_USERNAME", "elastic")
es_password = os.environ.get("ES_PASSWORD", "change me...")
history = ElasticsearchChatMessageHistory(
es_url=es_url,
es_user=es_username,
es_password=es_password,
index="test-history",
session_id="test-session"
)
如何获取 API 密钥
要获取 API 密钥:
- 登录到 Elastic Cloud 控制台
- 打开
Kibana
并转到 Stack Management > API Keys - 点击“创建 API 密钥”
- 输入 API 密钥的名称并点击“创建”
使用 API 密钥
es_api_key = os.environ.get("ES_API_KEY")
history = ElasticsearchChatMessageHistory(
es_api_key=es_api_key,
index="test-history",
session_id="test-session"
)
初始化 Elasticsearch 客户端和聊天消息历史
import os
from langchain_community.chat_message_histories import (
ElasticsearchChatMessageHistory,
)
es_url = os.environ.get("ES_URL", "https://:9200")
# If using Elastic Cloud:
# es_cloud_id = os.environ.get("ES_CLOUD_ID")
# Note: see Authentication section for various authentication methods
history = ElasticsearchChatMessageHistory(
es_url=es_url, index="test-history", session_id="test-session"
)
使用聊天消息历史
history.add_user_message("hi!")
history.add_ai_message("whats up?")
indexing message content='hi!' additional_kwargs={} example=False
indexing message content='whats up?' additional_kwargs={} example=False