From 48682ee9b1c0e6c5f77faa6ab3cea9173b68d360 Mon Sep 17 00:00:00 2001
From: hust-xing <chind_dyx@163.com>
Date: Mon, 16 Dec 2024 14:32:51 +0800
Subject: [PATCH] add aliyun Opensearch requirements

---
 README.md      | 1 +
 pyproject.toml | 3 +++
 2 files changed, 4 insertions(+)

diff --git a/README.md b/README.md
index 893c0497..a642abf0 100644
--- a/README.md
+++ b/README.md
@@ -41,6 +41,7 @@ All the database client supported
 | memorydb                 | `pip install vectordb-bench[memorydb]`      |
 | chromadb                 | `pip install vectordb-bench[chromadb]`      |
 | awsopensearch            | `pip install vectordb-bench[opensearch]` |
+| aliyun_opensearch        | `pip install vectordb-bench[aliyun_opensearch]` |
 
 ### Run
 
diff --git a/pyproject.toml b/pyproject.toml
index 6ad8e23e..8363aa4f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -65,6 +65,8 @@ all = [
     "opensearch-dsl",
     "opensearch-py",
     "memorydb",
+    "alibabacloud_ha3engine_vector",
+    "alibabacloud_searchengine20211025",
 ]
 
 qdrant          = [ "qdrant-client" ]
@@ -81,6 +83,7 @@ redis           = [ "redis" ]
 memorydb        = [ "memorydb" ]
 chromadb        = [ "chromadb" ]
 opensearch      = [ "opensearch-py" ]
+aliyun_opensearch = [ "alibabacloud_ha3engine_vector", "alibabacloud_searchengine20211025"]
 
 [project.urls]
 "repository" = "https://github.com/zilliztech/VectorDBBench"