add ca cert field to the bulk import functions

Author: Abdullah Ahmed

Diff for: pymilvus/bulk_writer/bulk_import.py

@@ -45,11 +45,23 @@ def _handle_response(url: str, res: json):
 
 
 def _post_request(
-    url: str, api_key: str, params: {}, timeout: int = 20, **kwargs
+    url: str, api_key: str, params: {}, timeout: int = 20, cert_path: Optional[str] = None, **kwargs
 ) -> requests.Response:
+    """Send a POST request with optional CA certificate verification
+
+    Args:
+        url (str): The endpoint URL
+        api_key (str): API key for authentication
+        params (dict): JSON parameters for the request
+        timeout (int): Timeout for the request
+        cert_path (str, optional): Path to the CA certificate file
+
+    Returns:
+        requests.Response: Response object.
+    """
     try:
         resp = requests.post(
-            url=url, headers=_http_headers(api_key), json=params, timeout=timeout, **kwargs
+            url=url, headers=_http_headers(api_key), json=params, timeout=timeout, verify=cert_path, **kwargs
         )
         if resp.status_code != 200:
             _throw(f"Failed to post url: {url}, status code: {resp.status_code}")
@@ -85,6 +97,7 @@ def bulk_import(
     api_key: str = "",
     access_key: str = "",
     secret_key: str = "",
+    cert_path: Optional[str] = None,
     **kwargs,
 ) -> requests.Response:
     """call bulkinsert restful interface to import files
@@ -103,6 +116,7 @@ def bulk_import(
         api_key (str): API key to authenticate your requests.
         access_key (str): access key to access the object storage
         secret_key (str): secret key to access the object storage
+        cert_path (str, optional): path to the CA certificate file
 
     Returns:
         response of the restful interface
@@ -121,13 +135,18 @@ def bulk_import(
         "secretKey": secret_key,
     }
 
-    resp = _post_request(url=request_url, api_key=api_key, params=params, **kwargs)
+    resp = _post_request(url=request_url, api_key=api_key, params=params, cert_path=cert_path, **kwargs)
     _handle_response(request_url, resp.json())
     return resp
 
 
 def get_import_progress(
-    url: str, job_id: str, cluster_id: str = "", api_key: str = "", **kwargs
+    url: str, 
+    job_id: str, 
+    cluster_id: str = "", 
+    api_key: str = "", 
+    cert_path: Optional[str] = None, 
+    **kwargs
 ) -> requests.Response:
     """get job progress
 
@@ -136,6 +155,7 @@ def get_import_progress(
         job_id (str): a job id
         cluster_id (str): id of a milvus instance(for cloud)
         api_key (str): API key to authenticate your requests.
+        cert_path (str, optional): path to the CA certificate file
 
     Returns:
         response of the restful interface
@@ -147,7 +167,7 @@ def get_import_progress(
         "clusterId": cluster_id,
     }
 
-    resp = _post_request(url=request_url, api_key=api_key, params=params, **kwargs)
+    resp = _post_request(url=request_url, api_key=api_key, params=params, cert_path=cert_path, **kwargs)
     _handle_response(request_url, resp.json())
     return resp
 
@@ -159,6 +179,7 @@ def list_import_jobs(
     api_key: str = "",
     page_size: int = 10,
     current_page: int = 1,
+    cert_path: Optional[str] = None,
     **kwargs,
 ) -> requests.Response:
     """list jobs in a cluster
@@ -170,6 +191,7 @@ def list_import_jobs(
         api_key (str): API key to authenticate your requests.
         page_size (int): pagination size
         current_page (int): pagination
+        cert_path (str, optional): path to the CA certificate file
 
     Returns:
         response of the restful interface
@@ -183,6 +205,6 @@ def list_import_jobs(
         "currentPage": current_page,
     }
 
-    resp = _post_request(url=request_url, api_key=api_key, params=params, **kwargs)
+    resp = _post_request(url=request_url, api_key=api_key, params=params, cert_path=cert_path, **kwargs)
     _handle_response(request_url, resp.json())
     return resp