Fix hybrid search rescore (#205)

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
milvus-io · Aug 8, 2024 · 77ef99b · 77ef99b
1 parent 3515d66
commit 77ef99b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/hybrid_search_task.cpp b/src/hybrid_search_task.cpp
@@ -188,7 +188,7 @@ HyBridSearchTask::ProcessSearch(
         search_request.set_nq(req.nq());
         std::string metric;
         CHECK_STATUS(GetMetricType(search_request, &metric), "");
-        re_scorers_[0]->SetMetricType(metric);
+        re_scorers_[i]->SetMetricType(metric);
         search_requests->push_back(search_request);
     }
     return Status::Ok();

diff --git a/src/server.cpp b/src/server.cpp
@@ -59,7 +59,7 @@ main(int argc, char** argv) {
     std::string work_dir = argv[1];
     std::string address = argv[2];
     std::string log_level = "ERROR";
-    if (argc == 4) {
+    if (argc >= 4) {
         log_level = argv[3];
     }
     if (log_level == "INFO") {