Create external connector and local connector

opensearch_py_ml/ml_commons/ml_commons_client.py

@@ -600,3 +600,63 @@ def delete_task(self, task_id: str) -> object:
             method="DELETE",
             url=API_URL,
         )
+
+    def create_connector(self, connector_payload: dict) -> dict:
+        """
+        This method creates a connector in the OpenSearch cluster using the ml-common plugin's API.
+
+        :param connector_payload: Dictionary containing the details of the connector.
+        :type connector_payload: dict
+        :return: API response
+        :rtype: dict
+        """
+        API_URL = f"{ML_BASE_URI}/connectors/_create"
+        return self._client.transport.perform_request(
+            method="POST",
+            url=API_URL,
+            body=connector_payload,
+        )
+
+    def delete_connector(self, connector_id: str) -> dict:
+        """
+        This method deletes a specific connector using its ID.
+
+        :param connector_id: ID of the connector to be deleted.
+        :type connector_id: str
+        :return: API response
+        :rtype: dict
+        """
+        API_URL = f"{ML_BASE_URI}/connectors/{connector_id}"
+        return self._client.transport.perform_request(
+            method="DELETE",
+            url=API_URL,
+        )
+
+    def list_connectors(self) -> dict:
+        """
+        This method lists all connectors in the OpenSearch cluster.
+
+        :return: API response containing a list of connectors.
+        :rtype: dict
+        """
+        API_URL = f"{ML_BASE_URI}/connectors"
+        return self._client.transport.perform_request(
+            method="GET",
+            url=API_URL,
+        )
+
+    def search_connectors(self, search_payload: dict) -> dict:
+        """
+        This method searches for connectors based on specific criteria.
+
+        :param search_payload: Dictionary containing search criteria.
+        :type search_payload: dict
+        :return: API response containing search results.
+        :rtype: dict
+        """
+        API_URL = f"{ML_BASE_URI}/connectors/_search"
+        return self._client.transport.perform_request(
+            method="POST",
+            url=API_URL,
+            body=search_payload,
+        )

setup.py

@@ -87,6 +87,7 @@
         "pandas>=1.5,<3",
         "matplotlib>=3.6.0,<4",
         "numpy>=1.24.0,<2",
+        "deprecated",
     ],
     python_requires=">=3.8",
     package_data={"opensearch_py_ml": ["py.typed"]},

tests/ml_commons/test_ml_commons_client.py

@@ -6,10 +6,12 @@
 # GitHub history for details.
 
 import os
+import subprocess
 import shutil
 import time
 from json import JSONDecodeError
 from os.path import exists
+import sys
 
 import pytest
 from opensearchpy import OpenSearch, helpers
@@ -385,7 +387,7 @@ def test_DEPRECATED_integration_model_train_upload_full_cycle():
                 except:  # noqa: E722
                     raised = True
                 assert raised == False, "Raised Exception in deleting model"
-
+    
 
 def test_integration_model_train_register_full_cycle():
     # first training the model with small epoch
@@ -510,6 +512,8 @@ def test_integration_model_train_register_full_cycle():
                     raised = True
                 assert raised == False, "Raised Exception in deleting model"
 
+
+
 
 def test_search():
     # Search task cases