diff --git a/include/vsag/constants.h b/include/vsag/constants.h index 9c5c20ba..8ba3bdb2 100644 --- a/include/vsag/constants.h +++ b/include/vsag/constants.h @@ -101,5 +101,7 @@ extern const char* const HGRAPH_USE_REORDER; extern const char* const HGRAPH_BASE_QUANTIZATION_TYPE; extern const char* const HGRAPH_GRAPH_MAX_DEGREE; extern const char* const HGRAPH_BUILD_EF_CONSTRUCTION; +extern const char* const HGRAPH_INIT_CAPACITY; +extern const char* const HGRAPH_BUILD_THREAD_COUNT; } // namespace vsag diff --git a/src/constants.cpp b/src/constants.cpp index 444b3f46..4c4190ca 100644 --- a/src/constants.cpp +++ b/src/constants.cpp @@ -102,5 +102,7 @@ const char* const HGRAPH_USE_REORDER = HGRAPH_USE_REORDER_KEY; const char* const HGRAPH_BASE_QUANTIZATION_TYPE = "base_quantization_type"; const char* const HGRAPH_GRAPH_MAX_DEGREE = "max_degree"; const char* const HGRAPH_BUILD_EF_CONSTRUCTION = "ef_construction"; +const char* const HGRAPH_INIT_CAPACITY = "hgraph_init_capacity"; +const char* const HGRAPH_BUILD_THREAD_COUNT = "build_thread_count"; }; // namespace vsag diff --git a/src/index/hgraph_zparameters.cpp b/src/index/hgraph_zparameters.cpp index 42f8ad21..a10a6ecd 100644 --- a/src/index/hgraph_zparameters.cpp +++ b/src/index/hgraph_zparameters.cpp @@ -27,7 +27,9 @@ const std::unordered_map> HGraphParameters {{HGRAPH_USE_REORDER, {HGRAPH_USE_REORDER_KEY}}, {HGRAPH_BASE_QUANTIZATION_TYPE, {HGRAPH_BASE_CODES_KEY, QUANTIZATION_TYPE_KEY}}, {HGRAPH_GRAPH_MAX_DEGREE, {HGRAPH_GRAPH_KEY, GRAPH_PARAMS_KEY, GRAPH_PARAM_MAX_DEGREE}}, - {HGRAPH_BUILD_EF_CONSTRUCTION, {BUILD_PARAMS_KEY, BUILD_EF_CONSTRUCTION}}}; + {HGRAPH_BUILD_EF_CONSTRUCTION, {BUILD_PARAMS_KEY, BUILD_EF_CONSTRUCTION}}, + {HGRAPH_INIT_CAPACITY, {HGRAPH_GRAPH_KEY, GRAPH_PARAMS_KEY, GRAPH_PARAM_INIT_MAX_CAPACITY}}, + {HGRAPH_BUILD_THREAD_COUNT, {BUILD_PARAMS_KEY, BUILD_THREAD_COUNT}}}; HGraphParameters::HGraphParameters(JsonType& hgraph_param, const IndexCommonParam& common_param) : common_param_(common_param) {