Skip to main content

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

认证​

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

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

  1. 登录 Elastic Cloud 控制台

  2. 转到 "Security" > "Users"

  3. 找到 "elastic" 用户并点击 "Edit"

  4. 点击 "Reset password"

  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

Was this page helpful?


You can leave detailed feedback on GitHub.