aidp-python-client 1.0.0__py3-none-any.whl

aidp_python_client/aidataplatform_dp/models/create_spark_cluster_details.py

@@ -0,0 +1,243 @@
+# coding: utf-8
+# Copyright (c) 2026, Oracle and/or its affiliates.  All rights reserved.
+
+
+from .create_cluster_details import CreateClusterDetails
+from oci.util import formatted_flat_dict, NONE_SENTINEL, value_allowed_none_or_none_sentinel  # noqa: F401
+from oci.decorators import init_model_state_from_kwargs
+
+
+@init_model_state_from_kwargs
+class CreateSparkClusterDetails(CreateClusterDetails):
+    """
+    Spark cluster details for creation.
+    """
+
+    def __init__(self, **kwargs):
+        """
+        Initializes a new CreateSparkClusterDetails object with values from keyword arguments. The default value of the :py:attr:`~oci.aidataplatform_dp.models.CreateSparkClusterDetails.type` attribute
+        of this class is ``USER`` and it should not be changed.
+        The following keyword arguments are supported (corresponding to the getters/setters of this class):
+
+        :param type:
+            The value to assign to the type property of this CreateSparkClusterDetails.
+            Allowed values for this property are: "USER", "AGENT_FLOW_COMPUTE"
+        :type type: str
+
+        :param display_name:
+            The value to assign to the display_name property of this CreateSparkClusterDetails.
+        :type display_name: str
+
+        :param description:
+            The value to assign to the description property of this CreateSparkClusterDetails.
+        :type description: str
+
+        :param driver_config:
+            The value to assign to the driver_config property of this CreateSparkClusterDetails.
+        :type driver_config: oci.aidataplatform_dp.models.DriverConfig
+
+        :param node_type:
+            The value to assign to the node_type property of this CreateSparkClusterDetails.
+        :type node_type: str
+
+        :param worker_config:
+            The value to assign to the worker_config property of this CreateSparkClusterDetails.
+        :type worker_config: oci.aidataplatform_dp.models.WorkerConfig
+
+        :param cluster_runtime_config:
+            The value to assign to the cluster_runtime_config property of this CreateSparkClusterDetails.
+        :type cluster_runtime_config: oci.aidataplatform_dp.models.ClusterRuntimeConfig
+
+        :param logging_config:
+            The value to assign to the logging_config property of this CreateSparkClusterDetails.
+        :type logging_config: oci.aidataplatform_dp.models.LoggingConfig
+
+        :param auto_termination_minutes:
+            The value to assign to the auto_termination_minutes property of this CreateSparkClusterDetails.
+        :type auto_termination_minutes: int
+
+        :param attach_to_notebook_config:
+            The value to assign to the attach_to_notebook_config property of this CreateSparkClusterDetails.
+        :type attach_to_notebook_config: oci.aidataplatform_dp.models.AttachToNotebookConfig
+
+        :param subscription:
+            The value to assign to the subscription property of this CreateSparkClusterDetails.
+        :type subscription: oci.aidataplatform_dp.models.SubscriptionDetails
+
+        """
+        self.swagger_types = {
+            'type': 'str',
+            'display_name': 'str',
+            'description': 'str',
+            'driver_config': 'DriverConfig',
+            'node_type': 'str',
+            'worker_config': 'WorkerConfig',
+            'cluster_runtime_config': 'ClusterRuntimeConfig',
+            'logging_config': 'LoggingConfig',
+            'auto_termination_minutes': 'int',
+            'attach_to_notebook_config': 'AttachToNotebookConfig',
+            'subscription': 'SubscriptionDetails'
+        }
+
+        self.attribute_map = {
+            'type': 'type',
+            'display_name': 'displayName',
+            'description': 'description',
+            'driver_config': 'driverConfig',
+            'node_type': 'nodeType',
+            'worker_config': 'workerConfig',
+            'cluster_runtime_config': 'clusterRuntimeConfig',
+            'logging_config': 'loggingConfig',
+            'auto_termination_minutes': 'autoTerminationMinutes',
+            'attach_to_notebook_config': 'attachToNotebookConfig',
+            'subscription': 'subscription'
+        }
+
+        self._type = None
+        self._display_name = None
+        self._description = None
+        self._driver_config = None
+        self._node_type = None
+        self._worker_config = None
+        self._cluster_runtime_config = None
+        self._logging_config = None
+        self._auto_termination_minutes = None
+        self._attach_to_notebook_config = None
+        self._subscription = None
+        self._type = 'USER'
+
+    @property
+    def worker_config(self):
+        """
+        Gets the worker_config of this CreateSparkClusterDetails.
+
+        :return: The worker_config of this CreateSparkClusterDetails.
+        :rtype: oci.aidataplatform_dp.models.WorkerConfig
+        """
+        return self._worker_config
+
+    @worker_config.setter
+    def worker_config(self, worker_config):
+        """
+        Sets the worker_config of this CreateSparkClusterDetails.
+
+        :param worker_config: The worker_config of this CreateSparkClusterDetails.
+        :type: oci.aidataplatform_dp.models.WorkerConfig
+        """
+        self._worker_config = worker_config
+
+    @property
+    def cluster_runtime_config(self):
+        """
+        Gets the cluster_runtime_config of this CreateSparkClusterDetails.
+
+        :return: The cluster_runtime_config of this CreateSparkClusterDetails.
+        :rtype: oci.aidataplatform_dp.models.ClusterRuntimeConfig
+        """
+        return self._cluster_runtime_config
+
+    @cluster_runtime_config.setter
+    def cluster_runtime_config(self, cluster_runtime_config):
+        """
+        Sets the cluster_runtime_config of this CreateSparkClusterDetails.
+
+        :param cluster_runtime_config: The cluster_runtime_config of this CreateSparkClusterDetails.
+        :type: oci.aidataplatform_dp.models.ClusterRuntimeConfig
+        """
+        self._cluster_runtime_config = cluster_runtime_config
+
+    @property
+    def logging_config(self):
+        """
+        Gets the logging_config of this CreateSparkClusterDetails.
+
+        :return: The logging_config of this CreateSparkClusterDetails.
+        :rtype: oci.aidataplatform_dp.models.LoggingConfig
+        """
+        return self._logging_config
+
+    @logging_config.setter
+    def logging_config(self, logging_config):
+        """
+        Sets the logging_config of this CreateSparkClusterDetails.
+
+        :param logging_config: The logging_config of this CreateSparkClusterDetails.
+        :type: oci.aidataplatform_dp.models.LoggingConfig
+        """
+        self._logging_config = logging_config
+
+    @property
+    def auto_termination_minutes(self):
+        """
+        Gets the auto_termination_minutes of this CreateSparkClusterDetails.
+        Optional timeout value in minutes used to automatically stop idle compute clusters.
+
+
+        :return: The auto_termination_minutes of this CreateSparkClusterDetails.
+        :rtype: int
+        """
+        return self._auto_termination_minutes
+
+    @auto_termination_minutes.setter
+    def auto_termination_minutes(self, auto_termination_minutes):
+        """
+        Sets the auto_termination_minutes of this CreateSparkClusterDetails.
+        Optional timeout value in minutes used to automatically stop idle compute clusters.
+
+
+        :param auto_termination_minutes: The auto_termination_minutes of this CreateSparkClusterDetails.
+        :type: int
+        """
+        self._auto_termination_minutes = auto_termination_minutes
+
+    @property
+    def attach_to_notebook_config(self):
+        """
+        Gets the attach_to_notebook_config of this CreateSparkClusterDetails.
+
+        :return: The attach_to_notebook_config of this CreateSparkClusterDetails.
+        :rtype: oci.aidataplatform_dp.models.AttachToNotebookConfig
+        """
+        return self._attach_to_notebook_config
+
+    @attach_to_notebook_config.setter
+    def attach_to_notebook_config(self, attach_to_notebook_config):
+        """
+        Sets the attach_to_notebook_config of this CreateSparkClusterDetails.
+
+        :param attach_to_notebook_config: The attach_to_notebook_config of this CreateSparkClusterDetails.
+        :type: oci.aidataplatform_dp.models.AttachToNotebookConfig
+        """
+        self._attach_to_notebook_config = attach_to_notebook_config
+
+    @property
+    def subscription(self):
+        """
+        Gets the subscription of this CreateSparkClusterDetails.
+
+        :return: The subscription of this CreateSparkClusterDetails.
+        :rtype: oci.aidataplatform_dp.models.SubscriptionDetails
+        """
+        return self._subscription
+
+    @subscription.setter
+    def subscription(self, subscription):
+        """
+        Sets the subscription of this CreateSparkClusterDetails.
+
+        :param subscription: The subscription of this CreateSparkClusterDetails.
+        :type: oci.aidataplatform_dp.models.SubscriptionDetails
+        """
+        self._subscription = subscription
+
+    def __repr__(self):
+        return formatted_flat_dict(self)
+
+    def __eq__(self, other):
+        if other is None:
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other):
+        return not self == other

aidp_python_client/aidataplatform_dp/models/create_sql_tool_details.py

@@ -0,0 +1,128 @@
+# coding: utf-8
+# Copyright (c) 2026, Oracle and/or its affiliates.  All rights reserved.
+
+
+from .create_tool_details import CreateToolDetails
+from oci.util import formatted_flat_dict, NONE_SENTINEL, value_allowed_none_or_none_sentinel  # noqa: F401
+from oci.decorators import init_model_state_from_kwargs
+
+
+@init_model_state_from_kwargs
+class CreateSqlToolDetails(CreateToolDetails):
+    """
+    Details to create a SQL Tool
+    """
+
+    def __init__(self, **kwargs):
+        """
+        Initializes a new CreateSqlToolDetails object with values from keyword arguments. The default value of the :py:attr:`~oci.aidataplatform_dp.models.CreateSqlToolDetails.tool_type` attribute
+        of this class is ``SQL`` and it should not be changed.
+        The following keyword arguments are supported (corresponding to the getters/setters of this class):
+
+        :param display_name:
+            The value to assign to the display_name property of this CreateSqlToolDetails.
+        :type display_name: str
+
+        :param description:
+            The value to assign to the description property of this CreateSqlToolDetails.
+        :type description: str
+
+        :param tool_type:
+            The value to assign to the tool_type property of this CreateSqlToolDetails.
+            Allowed values for this property are: "RAG", "SQL", "PROMPT", "NL2SQL", "MCP", "CUSTOM", "HTTP"
+        :type tool_type: str
+
+        :param properties:
+            The value to assign to the properties property of this CreateSqlToolDetails.
+        :type properties: dict(str, object)
+
+        :param input_schema:
+            The value to assign to the input_schema property of this CreateSqlToolDetails.
+        :type input_schema: dict(str, object)
+
+        :param tool_config:
+            The value to assign to the tool_config property of this CreateSqlToolDetails.
+        :type tool_config: oci.aidataplatform_dp.models.SqlToolConfiguration
+
+        """
+        self.swagger_types = {
+            'display_name': 'str',
+            'description': 'str',
+            'tool_type': 'str',
+            'properties': 'dict(str, object)',
+            'input_schema': 'dict(str, object)',
+            'tool_config': 'SqlToolConfiguration'
+        }
+
+        self.attribute_map = {
+            'display_name': 'displayName',
+            'description': 'description',
+            'tool_type': 'toolType',
+            'properties': 'properties',
+            'input_schema': 'inputSchema',
+            'tool_config': 'toolConfig'
+        }
+
+        self._display_name = None
+        self._description = None
+        self._tool_type = None
+        self._properties = None
+        self._input_schema = None
+        self._tool_config = None
+        self._tool_type = 'SQL'
+
+    @property
+    def input_schema(self):
+        """
+        Gets the input_schema of this CreateSqlToolDetails.
+        The list of properties in the inputSchema, along with the default value and description of each property
+
+
+        :return: The input_schema of this CreateSqlToolDetails.
+        :rtype: dict(str, object)
+        """
+        return self._input_schema
+
+    @input_schema.setter
+    def input_schema(self, input_schema):
+        """
+        Sets the input_schema of this CreateSqlToolDetails.
+        The list of properties in the inputSchema, along with the default value and description of each property
+
+
+        :param input_schema: The input_schema of this CreateSqlToolDetails.
+        :type: dict(str, object)
+        """
+        self._input_schema = input_schema
+
+    @property
+    def tool_config(self):
+        """
+        Gets the tool_config of this CreateSqlToolDetails.
+
+        :return: The tool_config of this CreateSqlToolDetails.
+        :rtype: oci.aidataplatform_dp.models.SqlToolConfiguration
+        """
+        return self._tool_config
+
+    @tool_config.setter
+    def tool_config(self, tool_config):
+        """
+        Sets the tool_config of this CreateSqlToolDetails.
+
+        :param tool_config: The tool_config of this CreateSqlToolDetails.
+        :type: oci.aidataplatform_dp.models.SqlToolConfiguration
+        """
+        self._tool_config = tool_config
+
+    def __repr__(self):
+        return formatted_flat_dict(self)
+
+    def __eq__(self, other):
+        if other is None:
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other):
+        return not self == other

aidp_python_client/aidataplatform_dp/models/create_sql_tool_node_details.py

@@ -0,0 +1,208 @@
+# coding: utf-8
+# Copyright (c) 2026, Oracle and/or its affiliates.  All rights reserved.
+
+
+from .create_agent_flow_node_details import CreateAgentFlowNodeDetails
+from oci.util import formatted_flat_dict, NONE_SENTINEL, value_allowed_none_or_none_sentinel  # noqa: F401
+from oci.decorators import init_model_state_from_kwargs
+
+
+@init_model_state_from_kwargs
+class CreateSqlToolNodeDetails(CreateAgentFlowNodeDetails):
+    """
+    Details to create a SQL Tool Node in an Agent Flow
+    """
+
+    def __init__(self, **kwargs):
+        """
+        Initializes a new CreateSqlToolNodeDetails object with values from keyword arguments. The default value of the :py:attr:`~oci.aidataplatform_dp.models.CreateSqlToolNodeDetails.type` attribute
+        of this class is ``SQL_TOOL`` and it should not be changed.
+        The following keyword arguments are supported (corresponding to the getters/setters of this class):
+
+        :param type:
+            The value to assign to the type property of this CreateSqlToolNodeDetails.
+            Allowed values for this property are: "START_NODE", "GUARDRAILS", "AGENT", "SUPERVISOR_AGENT", "NESTED_AGENT_FLOW", "EXTERNAL_AGENT", "HUMAN_IN_THE_LOOP", "SQL_TOOL", "PROMPT_TOOL", "RAG_TOOL", "MCP_TOOL", "HTTP_TOOL", "CUSTOM_TOOL"
+        :type type: str
+
+        :param node_type:
+            The value to assign to the node_type property of this CreateSqlToolNodeDetails.
+        :type node_type: str
+
+        :param name:
+            The value to assign to the name property of this CreateSqlToolNodeDetails.
+        :type name: str
+
+        :param description:
+            The value to assign to the description property of this CreateSqlToolNodeDetails.
+        :type description: str
+
+        :param position_x:
+            The value to assign to the position_x property of this CreateSqlToolNodeDetails.
+        :type position_x: float
+
+        :param position_y:
+            The value to assign to the position_y property of this CreateSqlToolNodeDetails.
+        :type position_y: float
+
+        :param is_expanded:
+            The value to assign to the is_expanded property of this CreateSqlToolNodeDetails.
+        :type is_expanded: bool
+
+        :param parent_node_id:
+            The value to assign to the parent_node_id property of this CreateSqlToolNodeDetails.
+        :type parent_node_id: str
+
+        :param src_node_id:
+            The value to assign to the src_node_id property of this CreateSqlToolNodeDetails.
+        :type src_node_id: str
+
+        :param configuration:
+            The value to assign to the configuration property of this CreateSqlToolNodeDetails.
+        :type configuration: dict(str, object)
+
+        :param node_type_id:
+            The value to assign to the node_type_id property of this CreateSqlToolNodeDetails.
+        :type node_type_id: str
+
+        :param tool_key:
+            The value to assign to the tool_key property of this CreateSqlToolNodeDetails.
+        :type tool_key: str
+
+        :param input_schema:
+            The value to assign to the input_schema property of this CreateSqlToolNodeDetails.
+        :type input_schema: dict(str, object)
+
+        :param tool_config:
+            The value to assign to the tool_config property of this CreateSqlToolNodeDetails.
+        :type tool_config: oci.aidataplatform_dp.models.SqlToolConfiguration
+
+        """
+        self.swagger_types = {
+            'type': 'str',
+            'node_type': 'str',
+            'name': 'str',
+            'description': 'str',
+            'position_x': 'float',
+            'position_y': 'float',
+            'is_expanded': 'bool',
+            'parent_node_id': 'str',
+            'src_node_id': 'str',
+            'configuration': 'dict(str, object)',
+            'node_type_id': 'str',
+            'tool_key': 'str',
+            'input_schema': 'dict(str, object)',
+            'tool_config': 'SqlToolConfiguration'
+        }
+
+        self.attribute_map = {
+            'type': 'type',
+            'node_type': 'nodeType',
+            'name': 'name',
+            'description': 'description',
+            'position_x': 'positionX',
+            'position_y': 'positionY',
+            'is_expanded': 'isExpanded',
+            'parent_node_id': 'parentNodeId',
+            'src_node_id': 'srcNodeId',
+            'configuration': 'configuration',
+            'node_type_id': 'nodeTypeId',
+            'tool_key': 'toolKey',
+            'input_schema': 'inputSchema',
+            'tool_config': 'toolConfig'
+        }
+
+        self._type = None
+        self._node_type = None
+        self._name = None
+        self._description = None
+        self._position_x = None
+        self._position_y = None
+        self._is_expanded = None
+        self._parent_node_id = None
+        self._src_node_id = None
+        self._configuration = None
+        self._node_type_id = None
+        self._tool_key = None
+        self._input_schema = None
+        self._tool_config = None
+        self._type = 'SQL_TOOL'
+
+    @property
+    def tool_key(self):
+        """
+        Gets the tool_key of this CreateSqlToolNodeDetails.
+        The unique identifier (key) of the saved AI tool
+
+
+        :return: The tool_key of this CreateSqlToolNodeDetails.
+        :rtype: str
+        """
+        return self._tool_key
+
+    @tool_key.setter
+    def tool_key(self, tool_key):
+        """
+        Sets the tool_key of this CreateSqlToolNodeDetails.
+        The unique identifier (key) of the saved AI tool
+
+
+        :param tool_key: The tool_key of this CreateSqlToolNodeDetails.
+        :type: str
+        """
+        self._tool_key = tool_key
+
+    @property
+    def input_schema(self):
+        """
+        Gets the input_schema of this CreateSqlToolNodeDetails.
+        The list of properties in the inputSchema, along with the default value and description of each property
+
+
+        :return: The input_schema of this CreateSqlToolNodeDetails.
+        :rtype: dict(str, object)
+        """
+        return self._input_schema
+
+    @input_schema.setter
+    def input_schema(self, input_schema):
+        """
+        Sets the input_schema of this CreateSqlToolNodeDetails.
+        The list of properties in the inputSchema, along with the default value and description of each property
+
+
+        :param input_schema: The input_schema of this CreateSqlToolNodeDetails.
+        :type: dict(str, object)
+        """
+        self._input_schema = input_schema
+
+    @property
+    def tool_config(self):
+        """
+        Gets the tool_config of this CreateSqlToolNodeDetails.
+
+        :return: The tool_config of this CreateSqlToolNodeDetails.
+        :rtype: oci.aidataplatform_dp.models.SqlToolConfiguration
+        """
+        return self._tool_config
+
+    @tool_config.setter
+    def tool_config(self, tool_config):
+        """
+        Sets the tool_config of this CreateSqlToolNodeDetails.
+
+        :param tool_config: The tool_config of this CreateSqlToolNodeDetails.
+        :type: oci.aidataplatform_dp.models.SqlToolConfiguration
+        """
+        self._tool_config = tool_config
+
+    def __repr__(self):
+        return formatted_flat_dict(self)
+
+    def __eq__(self, other):
+        if other is None:
+            return False
+
+        return self.__dict__ == other.__dict__
+
+    def __ne__(self, other):
+        return not self == other