diff --git a/tests/benchmark/milvus_benchmark/update.py b/tests/benchmark/milvus_benchmark/update.py
index 81c17a9f99..98154a5568 100644
--- a/tests/benchmark/milvus_benchmark/update.py
+++ b/tests/benchmark/milvus_benchmark/update.py
@@ -105,6 +105,20 @@ def get_latest_tag(limit=200):
     return latest_tag
 
 
+def get_image_tag():
+    url = ""
+    headers = {"accept": "application/json"}
+    try:
+        rep = requests.get(url, headers=headers)
+        data = json.loads(rep.text)
+        tag_name = data[0]["tags"][0]["name"]
+        print("[benchmark update] The image name used is %s" % str(tag_name))
+        return tag_name
+    except:
+        print("Can not get the tag list")
+        return "master-latest"
+
+
 def parse_server_tag(server_tag):
     """ paser server tag from server config"""
     # tag format: "8c"/"8c16m"/"8c16m1g"