From 77ef99bf83060cc41ba27e7b593930a8ccf5fbdb Mon Sep 17 00:00:00 2001
From: junjiejiangjjj <junjie.jiang@zilliz.com>
Date: Thu, 8 Aug 2024 20:32:36 +0800
Subject: [PATCH] Fix hybrid search rescore (#205)

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
---
 src/hybrid_search_task.cpp | 2 +-
 src/server.cpp             | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/hybrid_search_task.cpp b/src/hybrid_search_task.cpp
index e0c6e4f..b3b46d9 100644
--- 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
index 8dc6723..dc1108c 100644
--- 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") {