Elasticsearch

Elasticsearch 是一个分布式的、RESTful 的搜索和分析引擎，能够执行向量和词汇搜索。它建立在 Apache Lucene 库之上。

本笔记本展示了如何使用 Elasticsearch 的聊天消息历史功能。

设置 Elasticsearch

有两种主要方式来设置 Elasticsearch 实例：

1. Elastic Cloud。 Elastic Cloud 是一个托管的 Elasticsearch 服务。注册 免费试用。

2. 本地 Elasticsearch 安装。 通过在本地运行 Elasticsearch 开始使用。最简单的方法是使用官方的 Elasticsearch Docker 镜像。有关更多信息，请参见 Elasticsearch Docker 文档。

安装依赖

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

认证

如何获取默认 "elastic" 用户的密码

要获取默认 "elastic" 用户的 Elastic Cloud 密码：

1. 登录到 Elastic Cloud 控制台
2. 转到 "安全" > "用户"
3. 找到 "elastic" 用户并点击 "编辑"
4. 点击 "重置密码"
5. 按照提示重置密码

使用用户名/密码

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 密钥：

1. 登录到 Elastic Cloud 控制台
2. 打开 Kibana，并转到 Stack Management > API Keys
3. 点击 "Create API key"
4. 输入 API 密钥的名称，然后点击 "Create"

使用 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", "http://localhost:9200")

# 如果使用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("whats up?")

indexing message content='hi!' additional_kwargs={} example=False
indexing message content='whats up?' additional_kwargs={} example=False