fix: Use StrEnum instead of str, Enum (#722)

Signed-off-by: yangxuan <xuan.yang@zilliz.com>

Author: XuanYang-cn

vectordb_bench/backend/clients/api.py

@@ -1,13 +1,13 @@
 from abc import ABC, abstractmethod
 from contextlib import contextmanager
-from enum import Enum
+from enum import StrEnum
 
 from pydantic import BaseModel, SecretStr, validator
 
 from vectordb_bench.backend.filter import Filter, FilterOp
 
 
-class MetricType(str, Enum):
+class MetricType(StrEnum):
     L2 = "L2"
     COSINE = "COSINE"
     IP = "IP"
@@ -16,7 +16,7 @@ class MetricType(str, Enum):
     JACCARD = "JACCARD"
 
 
-class IndexType(str, Enum):
+class IndexType(StrEnum):
     HNSW = "HNSW"
     HNSW_SQ = "HNSW_SQ"
     HNSW_BQ = "HNSW_BQ"
@@ -47,7 +47,7 @@ class IndexType(str, Enum):
     NONE = "NONE"
 
 
-class SQType(str, Enum):
+class SQType(StrEnum):
     SQ6 = "SQ6"
     SQ8 = "SQ8"
     BF16 = "BF16"

vectordb_bench/backend/clients/elastic_cloud/config.py

@@ -1,4 +1,4 @@
-from enum import Enum
+from enum import StrEnum
 
 from pydantic import BaseModel, SecretStr
 
@@ -16,7 +16,7 @@ def to_dict(self) -> dict:
         }
 
 
-class ESElementType(str, Enum):
+class ESElementType(StrEnum):
     float = "float"  # 4 byte
     byte = "byte"  # 1 byte, -128 to 127
 

vectordb_bench/backend/clients/tencent_elasticsearch/config.py

@@ -1,4 +1,4 @@
-from enum import Enum
+from enum import StrEnum
 
 from pydantic import BaseModel, SecretStr
 
@@ -20,7 +20,7 @@ def to_dict(self) -> dict:
         }
 
 
-class ESElementType(str, Enum):
+class ESElementType(StrEnum):
     float = "float"  # 4 byte
     byte = "byte"  # 1 byte, -128 to 127