饭宝 发表于 2024-8-19 21:06:55

ES向量功能实战:向量搜刮(Docker部署ES及可视化ES-Head)

1 缘起

项目需要,向量搜刮使用ES,为了顺利使项目顺利交付,开始学习ES的稠密向量功能,本文即ES向量的实践:增删查改。ES从7.x版本支持向量功能,为测试ES向量功能,需要使用7.x及以上的版本。本文从ES环境搭建开始,完成ES向量功能实践,分享如下。
https://i-blog.csdnimg.cn/blog_migrate/b6ef3e9c49960ebc91e80d054837074f.png
2 Docker部署ES

2.1 配置ES config

# 打开系统配置文件
vim /etc/sysctl.conf
# 添加修改内容
vm.max_map_count=655360
# 生效
sudo systcl -p
2.2 拉取镜像

拉取指定版本ES,如8.12.2,
稠密向量功能是ES在7.x版本之后推出的功能。
docker pull docker.elastic.co/elasticsearch/elasticsearch:8.12.2
2.3 启动ES

docker run -dit \
--restart=always \
--name es01-8-12-2 \
-p 9200:9200 \
-p 9300:9300 \
-v /home/xindaqi/data/es-8-12-2/data:/usr/share/elasticsearch/data \
-v /home/xindaqi/data/es-8-12-2/logs:/usr/share/elasticsearch/logs \
-e ES_JAVA_OPS="-Xms512m -Xmx1g" \
-e discovery.type="single-node" \
-e ELASTIC_PASSWORD="admin-es" \
-m 1GB \
docker.elastic.co/elasticsearch/elasticsearch:8.12.2
2.4 测试连接ES



[*]cURL
curl --location 'http://localhost:9200
' \
--header 'Authorization: Basic ZWxhc3RpYzphZG1pbi1lcw=='


[*]欣赏器访问:
http://localhost:9200
用户名:elastic
暗码:admin-es
https://i-blog.csdnimg.cn/blog_migrate/0eb20a7e0fb60fe8524e5c448a359d28.png
2.5 部署ES-Head

2.5.1 拉取镜像&部署

# 拉取镜像
docker pull mobz/elasticsearch-head:5-alpine
先启动ES,复制vendor.js文件:
docker run -dit \
--restart=always \
--name es-head \
-p 9100:9100 \
docker.io/mobz/elasticsearch-head:5-alpine
# 命令行:复制ES-Head数据到本地
docker cp es-head:/usr/src/app/_site/vendor.js /home/xindaqi/data/es-head/vendor.js
# 修改内容:6886行,替换为:
application/json;charset=UTF-8
# 修改内容:7573行,替换为:
application/json;charset=UTF-8
启动ES-Head
# 自动启动
docker run -dit \
--restart=always \
--name es-head \
-p 9100:9100 \
-v /home/xindaqi/data/es-head/vendor.js:/usr/src/app/_site/vendor.js \
docker.io/mobz/elasticsearch-head:5-alpine
2.5.2 使用用户名暗码访问ES-head

http://172.22.75.234:9100/?auth_user=elastic&auth_password=admin-es

配置ES:http://172.22.75.234:9200
出现跨域,无法访问.
https://i-blog.csdnimg.cn/blog_migrate/c30727aaa22140083fa237757aadfeef.png#pic_center
2.5.3 配置ES

# 命令行
docker cp es01-8-12-2:/usr/share/elasticsearch/config /home/xindaqi/data/es-8-12-2
# 修改ES配置
vi /path/config/elasticsearch.yml
# 配置如下
network.host: 0.0.0.0
http.cors.enabled: true
http.cors.allow-origin: "*"
ES重新启动:
docker run -dit \
--restart=always \
--name es01-8-12-2 \
-p 9200:9200 \
-p 9300:9300 \
-v /home/xindaqi/data/es-8-12-2/data:/usr/share/elasticsearch/data \
-v /home/xindaqi/data/es-8-12-2/logs:/usr/share/elasticsearch/logs \
-v /home/xindaqi/data/es-8-12-2/config:/usr/share/elasticsearch/config \
-e ES_JAVA_OPS="-Xms512m -Xmx1g" \
-e discovery.type="single-node" \
-e ELASTIC_PASSWORD="admin-es" \
-m 1GB \
docker.elastic.co/elasticsearch/elasticsearch:8.12.2
https://i-blog.csdnimg.cn/blog_migrate/037b1712e7fc0ef6fc24a0645bdc30b5.png#pic_center
3 向量使用

ES向量功能使用了新的数据范例:dense_vector,创建索引时,需要指定存储向量属性的范例为dense_vector,同时指定向量维度,固然,创建索引时可以不指定dense_vector范例,ES会主动识别范例,需要注意的是,当向量为低维度数据时,会被识别为float,因此,当使用低维度的数据时,为保险起见,需要在创建索引时指定dense_vector范例。
3.1 新建向量索引

新建向量索引格式:
index-name用户指定的索引名称,用户自界说即可。
http://localhost:9200
/index-name 参数:
"field-name":{
    "type":"dense_vector",
    "dims": n
}
此中,filed-name为用户设定的属性名称,n为向量维度,其余为固定值。
样例如下:
curl --location --request PUT 'http://localhost:9200
/vector-test' \--header 'Content-Type: application/json' \--header 'Authorization: Basic ZWxhc3RpYzphZG1pbi1lcw==' \--data '{    "mappings": {      "properties": {            "dense_value":{                "type":"dense_vector",                "dims":5            },            "text": {                "type": "text"            },            "uid":{                "type":"keyword"            }      }    }}' 3.2 插入向量数据

插入向量数据需要和界说的向量维度雷同,否则插入失败。


[*]格式:
http://172.22.75.234:9200/index-name/_doc/id
此中,index-name为用户指定的索引名称,id为数据id。


[*]样例如下:
curl --location 'http://172.22.75.234:9200/vector-test/_doc/5' \
--header 'Content-Type: application/json' \
--header 'Authorization: Basic ZWxhc3RpYzphZG1pbi1lcw==' \
--data '{
    "uid": "5",
    "text": "新增的数据5",
    "dense_value": [
      0.6369616873214543,
      0.2697867137638703,
      0.04097352393619469,
      0.016527635528529094,
      0.8132702392002724
    ]
}'
更新向量数据



[*]格式:
http://172.22.75.234:9200/index-name/_update/id
此中,index-name为用户指定的索引名称,id为数据id。


[*]样例如下:
curl --location 'http://172.22.75.234:9200/vector-test/_update/1' \
--header 'Content-Type: application/json' \
--header 'Authorization: Basic ZWxhc3RpYzphZG1pbi1lcw==' \
--data '{
    "doc": {
      "uid": "1",
      "text": "新增的数据11",
      "dense_value": [
            0.6369616873214543,
            0.2697867137638703,
            0.04097352393619469,
            0.016527635528529094,
            0.8132702392002724
      ]
    }
}'
3.3 查询向量数据

3.3.1 相似度查询

相似度查询使用knn,相似度计算使用:cosine


[*]格式:
http://172.22.75.234:9200/index-name/_search

此中,index-name为用户指定的索引名称。
{
    "knn":{
      "field":"filed-name",
      "query_vector":,
      "k": n,
      "num_candidates": m
    }
此中,field-name为用户界说的向量属性,n为查询的前n个相似数据(分数从大->小分列),v1为向量值,维度与界说的索引维度同等,否则会报错,m为候选数据数目(等看了官方文档再更新)。


[*]样例如下:
curl --location 'http://172.22.75.234:9200/vector-test/_search' \
--header 'Content-Type: application/json' \
--header 'Authorization: Basic ZWxhc3RpYzphZG1pbi1lcw==' \
--data '{
    "knn":{
      "field":"dense_value",
      "query_vector":[
                        0.6369616873214543,
                        0.2697867137638703,
                        0.04097352393619469,
                        0.016527635528529094,
                        0.8132702392002724
                  ],
      "k": 3,
      "num_candidates": 100
    },
    "_source":{
      "excludes":["dense_value"],
      "includes":["text", "uid"]
    }
}'
3.3.2 条件查询

根据指定条件筛选数据,应用于只使用精准过滤的情况。


[*]格式:
http://172.22.75.234:9200/index-name/_search

此中,index-name为用户指定的索引名称,id为数据id。
{
    "query": {
      "bool": {
            "filter": [
                {
                  "terms": {
                        "field-name":
                  }
                }
            ]
      }
    }
}
此中,filed-name为用户自界说的属性名称,v1为属性的值。


[*]样例如下:
curl --location 'http://172.22.75.234:9200/vector-test/_search' \
--header 'Content-Type: application/json' \
--header 'Authorization: Basic ZWxhc3RpYzphZG1pbi1lcw==' \
--data '{
    "query": {
      "bool": {
            "filter": [
                {
                  "terms": {
                        "uid": ["0","1"]
                  }
                }
            ]
      }
    },
    "_source": {
      "includes": ["text","uid"],
      "excludes": [
            "dense_value"
      ]
    }
}'
4 小结

(1)Docker部署ES时通过ELASTIC_PASSWORD配置暗码;
(2)ES7.x之后的版本支持稠密向量功能;
(3)创建使用稠密向量的索引时,需要指定属性范例:dense_vector和向量维度;
(4)向量维度固定后,后续使用向量进行使用(插入数据/插叙数据/更新数据)时必须保持一直,否则报错。
5 Q&A

5.1 ES-Head无法访问ES



[*]缘故原由:
ES配置了用户名和暗码。
[*]方案:
使用用户名和暗码访问ES-Head,再配置ES。
访问ES-Head格式:
http://172.22.75.234:9100/?auth_user=elastic&auth_password=admin-es
在ES-Head中配置ES即可。
https://i-blog.csdnimg.cn/blog_migrate/608c50cd4d498fc1e754bedf082078c8.png#pic_center
5.2 ES-Head无法查询数据

{
“error”: “Content-Type header is not supported”,
“status”: 406
}
https://i-blog.csdnimg.cn/blog_migrate/5d0dbe0fce2322f9d54e5b77735eee54.png


[*]缘故原由
ES8对请求头限制。
[*]方案
修改vendor.js
# 命令行:复制ES-Head数据到本地
docker cp es-head:/usr/src/app/_site/vendor.js /home/xindaqi/data/es-head/vendor.js
# 修改内容:6886行,替换为:
application/json;charset=UTF-8
# 修改内容:7573行,替换为:
application/json;charset=UTF-8
启动ES-Head
# 自动启动
docker run -dit \
--restart=always \
--name es-head \
-p 9100:9100 \
-v /home/xindaqi/data/es-head/vendor.js:/usr/src/app/_site/vendor.js \
docker.io/mobz/elasticsearch-head:5-alpine
https://i-blog.csdnimg.cn/blog_migrate/f2485a5bfb55bc97035eef59e8a302a9.png#pic_center
https://i-blog.csdnimg.cn/blog_migrate/20b064d0d035e2a283d38b4efed3a192.png#pic_center
解决:
https://i-blog.csdnimg.cn/blog_migrate/037a5d80bcd46e334e47da43682b529d.png#pic_center

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
页: [1]
查看完整版本: ES向量功能实战:向量搜刮(Docker部署ES及可视化ES-Head)