如何使用 Elasticsearch 作为向量数据库

本文由 简悦 SimpRead 转码， 原文地址 blog.csdn.net

在今天的文章中，我们将很快地通过 Docker 来快速地设置 Elasticsearch 及 Kibana，并设置 Elasticsearch 为向量搜索。

拉取 Docker 镜像

1.  docker pull docker.elastic.co/elasticsearch/elasticsearch:8.12.2
2.  docker pull docker.elastic.co/kibana/kibana:8.12.2

启动 Elasticsearch 及 Kibana 容器

1.  docker network create elastic

3.  docker run -d --name elasticsearch --net elastic -p 9200:9200 -p 9300:9300 -m 1GB -e "discovery.type=single-node" -e "ELASTIC_PASSWORD=password" docker.elastic.co/elasticsearch/elasticsearch:8.12.2

5.  docker run -d --name kibana --net elastic -p 5601:5601 docker.elastic.co/kibana/kibana:8.12.2

1.  $ docker run -d --name elasticsearch --net elastic -p 9200:9200 -p 9300:9300 -m 1GB -e "discovery.type=single-node" -e "ELASTIC_PASSWORD=password" docker.elastic.co/elasticsearch/elasticsearch:8.12.2
2.  39dc9085f239edb3c963de4fb122f0ec02f78a6311abd8297cf046c025cd2618

1.  $ docker run -d --name kibana --net elastic -p 5601:5601 docker.elastic.co/kibana/kibana:8.12.2
2.  2766a300b3fd165f793f5f47b55748b2e9d4b016ea78b5c23565442e2c4cdfb5

在上面，我们指定了 elasic 超级用户的密码为 password。这在下面将要使用到。

验证容器是否已启动并正在运行：

1.  $ docker ps
2.  CONTAINER ID   IMAGE                                                  COMMAND                  CREATED              STATUS              PORTS                                            NAMES
3.  2766a300b3fd   docker.elastic.co/kibana/kibana:8.12.2                 "/bin/tini -- /usr/l..."   About a minute ago   Up About a minute   0.0.0.0:5601->5601/tcp                           kibana
4.  39dc9085f239   docker.elastic.co/elasticsearch/elasticsearch:8.12.2   "/bin/tini -- /usr/l..."   3 minutes ago        Up 3 minutes        0.0.0.0:9200->9200/tcp, 0.0.0.0:9300->9300/tcp   elasticsearch

从上面我们可以看到 Elasticsarch 及 Kibana 已经完全运行起来了。我们可以在浏览器中进行验证：

docker exec -it elasticsearch /bin/bash

docker logs -f kibana

1.  $ docker logs -f kibana
2.  Kibana is currently running with legacy OpenSSL providers enabled! For details and instructions on how to disable see https://www.elastic.co/guide/en/kibana/8.12/production.html#openssl-legacy-provider
3.  {"log.level":"info","@timestamp":"2024-03-22T01:28:37.598Z","log.logger":"elastic-apm-node","ecs.version":"8.10.0","agentVersion":"4.2.0","env":{"pid":7,"proctitle":"/usr/share/kibana/bin/../node/bin/node","os":"linux 6.4.16-linuxkit","arch":"arm64","host":"2766a300b3fd","timezone":"UTC+00","runtime":"Node.js v18.18.2"},"config":{"active":{"source":"start","value":true},"breakdownMetrics":{"source":"start","value":false},"captureBody":{"source":"start","value":"off","commonName":"capture_body"},"captureHeaders":{"source":"start","value":false},"centralConfig":{"source":"start","value":false},"contextPropagationOnly":{"source":"start","value":true},"environment":{"source":"start","value":"production"},"globalLabels":{"source":"start","value":[["git_rev","f5bd489c5ff9c676c4f861c42da6ea99ae350832"]],"sourceValue":{"git_rev":"f5bd489c5ff9c676c4f861c42da6ea99ae350832"}},"logLevel":{"source":"default","value":"info","commonName":"log_level"},"metricsInterval":{"source":"start","value":120,"sourceValue":"120s"},"serverUrl":{"source":"start","value":"https://kibana-cloud-apm.apm.us-east-1.aws.found.io/","commonName":"server_url"},"transactionSampleRate":{"source":"start","value":0.1,"commonName":"transaction_sample_rate"},"captureSpanStackTraces":{"source":"start","sourceValue":false},"secretToken":{"source":"start","value":"[REDACTED]","commonName":"secret_token"},"serviceName":{"source":"start","value":"kibana","commonName":"service_name"},"serviceVersion":{"source":"start","value":"8.12.2","commonName":"service_version"}},"activationMethod":"require","message":"Elastic APM Node.js Agent v4.2.0"}
4.  [2024-03-22T01:28:38.276+00:00][INFO ][root] Kibana is starting
5.  [2024-03-22T01:28:38.320+00:00][INFO ][node] Kibana process configured with roles: [background_tasks, ui]
6.  [2024-03-22T01:28:42.183+00:00][INFO ][plugins-service] Plugin "cloudChat" is disabled.
7.  [2024-03-22T01:28:42.192+00:00][INFO ][plugins-service] Plugin "cloudExperiments" is disabled.
8.  [2024-03-22T01:28:42.193+00:00][INFO ][plugins-service] Plugin "cloudFullStory" is disabled.
9.  [2024-03-22T01:28:42.501+00:00][INFO ][plugins-service] Plugin "profilingDataAccess" is disabled.
10.  [2024-03-22T01:28:42.501+00:00][INFO ][plugins-service] Plugin "profiling" is disabled.
11.  [2024-03-22T01:28:42.587+00:00][INFO ][plugins-service] Plugin "securitySolutionServerless" is disabled.
12.  [2024-03-22T01:28:42.587+00:00][INFO ][plugins-service] Plugin "serverless" is disabled.
13.  [2024-03-22T01:28:42.587+00:00][INFO ][plugins-service] Plugin "serverlessObservability" is disabled.
14.  [2024-03-22T01:28:42.587+00:00][INFO ][plugins-service] Plugin "serverlessSearch" is disabled.
15.  [2024-03-22T01:28:42.929+00:00][INFO ][http.server.Preboot] http server running at http://0.0.0.0:5601
16.  [2024-03-22T01:28:42.996+00:00][INFO ][plugins-system.preboot] Setting up [1] plugins: [interactiveSetup]
17.  [2024-03-22T01:28:43.004+00:00][INFO ][preboot] "interactiveSetup" plugin is holding setup: Validating Elasticsearch connection configuration...
18.  [2024-03-22T01:28:43.019+00:00][INFO ][root] Holding setup until preboot stage is completed.

21.  i Kibana has not been configured.

23.  Go to http://0.0.0.0:5601/?code=897018 to get started.

28.  Your verification code is:  897 018 

我们把上面的 enrollment token 及 verification code 填入下面的方框里：

注意：由于一些原因，在上面显示的地址不是 localhost，而是电脑上的另外一个地址，比如 172.18.0.2:9200。这个并不影响我们的配置。

这样我们就成功地登录了。

创建索引

现在，让我们创建 "movies" 索引。 我们将使用 text-embedding-3-small 模型来生成 title 字段的向量嵌入并将其存储为 title_embedding。 该模型生成长度为 1536 的嵌入。因此我们需要将 title_embedding 字段映射指定为具有 1536 维的密集向量。

1.  PUT /movies
2.  {
3.    "mappings": {
4.      "properties": {
5.        "title": {
6.          "type": "text"
7.        },
8.        "genre": {
9.          "type": "keyword"
10.        },
11.        "release_year": {
12.          "type": "integer"
13.        },
14.        "title_embedding": {
15.          "type": "dense_vector",
16.          "dims": 1536
17.        }
18.      }
19.    }
20.  }

让我们使用 Elasticsearch Python 客户端插入一些文档。

Python 客户端需要 ssl_assert_fingerprint 才能连接到 Elasticsearch。 让我们使用以下命令来获取它：

openssl s_client -connect localhost:9200 -servername localhost -showcerts </dev/null 2>/dev/null | openssl x509 -fingerprint -sha256 -noout -in /dev/stdin

1.  $ openssl s_client -connect localhost:9200 -servername localhost -showcerts </dev/null 2>/dev/null | openssl x509 -fingerprint -sha256 -noout -in /dev/stdin

3.  sha256 Fingerprint=20:67:39:6C:33:C0:D6:AC:E2:E3:A5:2E:56:6C:57:4F:91:DC:41:4D:99:9B:7F:0F:1C:20:AD:E2:20:FE:1E:1B

写入文档到 Elasticsearch

现在我们可以在电影索引中插入一些文档。

我们现在 terminal 中打入如下的命令：

export OPENAI_API_KEY="YourOpenAiKey"

在上面，请填入自己申请的 OpenAI key。

请按照下面的命令来安装所需要的包：

pip3 install elasticsearch python-dotenv

我们创建如下的 python 应用：

write_index.py

1.  from elasticsearch import Elasticsearch
2.  from openai import OpenAI
3.  import os

5.  OPENAI_API_KEY= os.getenv("OPENAI_API_KEY")

7.  es = Elasticsearch(
8.      "https://localhost:9200",
9.      ssl_assert_fingerprint='20:67:39:6C:33:C0:D6:AC:E2:E3:A5:2E:56:6C:57:4F:91:DC:41:4D:99:9B:7F:0F:1C:20:AD:E2:20:FE:1E:1B',
10.      basic_auth=("elastic", "password")
11.  )

13.  openai = OpenAI(api_key=OPENAI_API_KEY)

15.  movies = [
16.      {"title": "Inception", "genre": "Sci-Fi", "release_year": 2010},
17.      {"title": "The Shawshank Redemption", "genre": "Drama", "release_year": 1994},
18.      {"title": "The Godfather", "genre": "Crime", "release_year": 1972},
19.      {"title": "Pulp Fiction", "genre": "Crime", "release_year": 1994},
20.      {"title": "Forrest Gump", "genre": "Drama", "release_year": 1994}
21.  ]

23.  # Indexing movies
24.  for movie in movies:
25.      movie['title_embedding'] = openai.embeddings.create(
26.          input=[movie['title']], model='text-embedding-3-small'
27.      ).data[0].embedding

29.      es.index(index="movies", document=movie)

我们使用如下的命令来运行脚本：

python3 write_index.py

我们可以在 Kibana 中进行查看：

GET movies/_search

搜索索引

比方说，我们想要搜索与片名《godfather》紧密匹配的电影。 我们可以使用K最近邻（KNN）算法来搜索相关文档。 我们会将搜索限制为仅显示 1 个最接近的匹配结果。

首先我们需要获得单词 Godfather 的向量表示：

1.  vector_value = openai_client.embeddings.create(
2.          input=["Godfather"], model='text-embedding-3-small'
3.      ).data[0].embedding

现在我们可以搜索电影索引来获取与片名《Godfather》紧密匹配的电影。 在我们的例子中，它应该与标题为《Godfather》的电影文档匹配。

1.  query_string = {
2.      "field": "title_embedding",
3.      "query_vector": vector_value,
4.      "k": 1,
5.      "num_candidates": 100
6.  }

8.  results = es_client.search(index="movies", knn=query_string, source_includes=["title", "genre", "release_year"])

10.  print(results['hits']['hits'])

完整的 Python 应用如下：

search_index.py

1.  from elasticsearch import Elasticsearch
2.  from openai import OpenAI
3.  import os

5.  OPENAI_API_KEY= os.getenv("OPENAI_API_KEY")

7.  es = Elasticsearch(
8.      "https://localhost:9200",
9.      ssl_assert_fingerprint='20:67:39:6C:33:C0:D6:AC:E2:E3:A5:2E:56:6C:57:4F:91:DC:41:4D:99:9B:7F:0F:1C:20:AD:E2:20:FE:1E:1B',
10.      basic_auth=("elastic", "password")
11.  )

13.  openai = OpenAI(api_key=OPENAI_API_KEY)

15.  vector_value = openai.embeddings.create(
16.          input=["Godfather"], model='text-embedding-3-small'
17.      ).data[0].embedding

19.  query_string = {
20.      "field": "title_embedding",
21.      "query_vector": vector_value,
22.      "k": 1,
23.      "num_candidates": 100
24.  }

26.  results = es.search(index="movies", knn=query_string, source_includes=["title", "genre", "release_year"])

28.  print(results['hits']['hits'])

运行上面的代码：

1.  $ python3 search_index.py 
2.  [{'_index': 'movies', '_id': 'koeTZI4BvK48CYytTCuI', '_score': 0.8956262, '_source': {'title': 'The Godfather', 'genre': 'Crime', 'release_year': 1972}}]

很显然，它找到了 Godfather 这个文档。

很多开发者可能想问，我们是不是也可以使用中文来进行搜索呢？

我们尝试如下的代码：

search_index.py

1.  from elasticsearch import Elasticsearch
2.  from openai import OpenAI
3.  import os

5.  OPENAI_API_KEY= os.getenv("OPENAI_API_KEY")

7.  es = Elasticsearch(
8.      "https://localhost:9200",
9.      ssl_assert_fingerprint='20:67:39:6C:33:C0:D6:AC:E2:E3:A5:2E:56:6C:57:4F:91:DC:41:4D:99:9B:7F:0F:1C:20:AD:E2:20:FE:1E:1B',
10.      basic_auth=("elastic", "password")
11.  )

13.  openai = OpenAI(api_key=OPENAI_API_KEY)

15.  vector_value = openai.embeddings.create(
16.          input=["教父"], model='text-embedding-3-small'
17.      ).data[0].embedding

19.  query_string = {
20.      "field": "title_embedding",
21.      "query_vector": vector_value,
22.      "k": 1,
23.      "num_candidates": 100
24.  }

26.  results = es.search(index="movies", knn=query_string, source_includes=["title", "genre", "release_year"])

28.  print(results['hits']['hits'])

在上面的代码中，我们使用 "教父" 而不是 Godfather。运行上面的代码，它显示：

1.  $ python3 search_index.py
2.  [{'_index': 'movies', '_id': 'koeTZI4BvK48CYytTCuI', '_score': 0.6547822, '_source': {'title': 'The Godfather', 'genre': 'Crime', 'release_year': 1972}}]

很显然，它也同样找到 Godfather 这个电影。它说明这个大语言模型支持多语言的搜索。