diff --git a/tests/python_client/base/utility_wrapper.py b/tests/python_client/base/utility_wrapper.py
index c78a434a22..5f542c20db 100644
--- a/tests/python_client/base/utility_wrapper.py
+++ b/tests/python_client/base/utility_wrapper.py
@@ -14,6 +14,14 @@ class ApiUtilityWrapper:
 
     ut = utility
 
+    def get_query_segment_info(self, collection_name, timeout=None, using="default", check_task=None, check_items=None):
+        timeout = TIMEOUT if timeout is None else timeout
+        func_name = sys._getframe().f_code.co_name
+        res, is_succ = api_request([self.ut.get_query_segment_info, collection_name, timeout, using])
+        check_result = ResponseChecker(res, func_name, check_task, check_items, is_succ,
+                                        collection_name=collection_name, timeout=timeout, using=using).run()
+        return res, check_result
+
     def loading_progress(self, collection_name, partition_names=None,
                          using="default", check_task=None, check_items=None):
         func_name = sys._getframe().f_code.co_name