diff --git a/pymilvus/client/check.py b/pymilvus/client/check.py index e6948af26..009e1358f 100644 --- a/pymilvus/client/check.py +++ b/pymilvus/client/check.py @@ -128,6 +128,26 @@ def is_legal_nprobe(nprobe: Any) -> bool: return isinstance(nprobe, int) +def is_legal_itopk_size(itopk_size: Any) -> bool: + return isinstance(itopk_size, int) + + +def is_legal_search_width(search_width: Any) -> bool: + return isinstance(search_width, int) + + +def is_legal_min_iterations(min_iterations: Any) -> bool: + return isinstance(min_iterations, int) + + +def is_legal_max_iterations(max_iterations: Any) -> bool: + return isinstance(max_iterations, int) + + +def is_legal_team_size(team_size: Any) -> bool: + return isinstance(team_size, int) + + def is_legal_cmd(cmd: Any) -> bool: return cmd and isinstance(cmd, str) @@ -298,6 +318,11 @@ def __init__(self) -> None: "properties": is_legal_collection_properties, "replica_number": is_legal_replica_number, "resource_group_name": is_legal_table_name, + "itopk_size": is_legal_itopk_size, + "search_width": is_legal_search_width, + "min_iterations": is_legal_min_iterations, + "max_iterations": is_legal_max_iterations, + "team_size": is_legal_team_size, } def check(self, key: str, value: Callable): diff --git a/pymilvus/client/utils.py b/pymilvus/client/utils.py index 4e29c9dd6..a1b9a593e 100644 --- a/pymilvus/client/utils.py +++ b/pymilvus/client/utils.py @@ -23,6 +23,7 @@ "BIN_IVF_FLAT", "DISKANN", "AUTOINDEX", + "GPU_CAGRA", ] valid_index_params_keys = [ @@ -33,6 +34,8 @@ "efConstruction", "PQM", "n_trees", + "intermediate_graph_degree", + "graph_degree", ] valid_binary_index_types = [