fix: distinguish error types in GetCentroid for untrained vs invalid bucket_id

- Use WRONG_STATUS when partition is not trained
- Use INVALID_ARGUMENT when bucket_id is out of range

Author: LHT129

src/algorithm/ivf_partition/gno_imi_partition.cpp

@@ -423,9 +423,12 @@ GNOIMIPartition::inner_joint_classify_datas(const float* datas,
 
 void
 GNOIMIPartition::GetCentroid(BucketIdType bucket_id, Vector<float>& centroid) {
-    if (!is_trained_ || bucket_id >= bucket_count_) {
+    if (!is_trained_) {
         throw VsagException(ErrorType::WRONG_STATUS, "Partition not trained");
     }
+    if (bucket_id >= bucket_count_) {
+        throw VsagException(ErrorType::INVALID_ARGUMENT, "Invalid bucket_id");
+    }
     auto bucket_id_s = bucket_id / bucket_count_t_;
     auto bucket_id_t = bucket_id % bucket_count_t_;
     FP32Add(data_centroids_s_.data() + bucket_id_s * dim_,

src/algorithm/ivf_partition/ivf_nearest_partition.cpp

@@ -163,9 +163,12 @@ IVFNearestPartition::factory_router_index(const IndexCommonParam& common_param)
 }
 void
 IVFNearestPartition::GetCentroid(BucketIdType bucket_id, Vector<float>& centroid) {
-    if (!is_trained_ || bucket_id >= bucket_count_) {
+    if (!is_trained_) {
         throw VsagException(ErrorType::WRONG_STATUS, "Partition not trained");
     }
+    if (bucket_id >= bucket_count_) {
+        throw VsagException(ErrorType::INVALID_ARGUMENT, "Invalid bucket_id");
+    }
     this->route_index_ptr_->GetCodeByInnerId(bucket_id, (uint8_t*)centroid.data());
 }