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

aidp_python_client/aidataplatform_dp/models/nested_agent_flow_node.py

@@ -0,0 +1,243 @@
+# coding: utf-8
+# Copyright (c) 2026, Oracle and/or its affiliates.  All rights reserved.
+
+
+from .agent_flow_node import AgentFlowNode
+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 NestedAgentFlowNode(AgentFlowNode):
+    """
+    An agent node representing a nested Agent Flow
+    """
+
+    def __init__(self, **kwargs):
+        """
+        Initializes a new NestedAgentFlowNode object with values from keyword arguments. The default value of the :py:attr:`~oci.aidataplatform_dp.models.NestedAgentFlowNode.type` attribute
+        of this class is ``NESTED_AGENT_FLOW`` 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 NestedAgentFlowNode.
+            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 NestedAgentFlowNode.
+        :type node_type: str
+
+        :param name:
+            The value to assign to the name property of this NestedAgentFlowNode.
+        :type name: str
+
+        :param description:
+            The value to assign to the description property of this NestedAgentFlowNode.
+        :type description: str
+
+        :param position_x:
+            The value to assign to the position_x property of this NestedAgentFlowNode.
+        :type position_x: float
+
+        :param position_y:
+            The value to assign to the position_y property of this NestedAgentFlowNode.
+        :type position_y: float
+
+        :param is_expanded:
+            The value to assign to the is_expanded property of this NestedAgentFlowNode.
+        :type is_expanded: bool
+
+        :param parent_node_id:
+            The value to assign to the parent_node_id property of this NestedAgentFlowNode.
+        :type parent_node_id: str
+
+        :param configuration:
+            The value to assign to the configuration property of this NestedAgentFlowNode.
+        :type configuration: dict(str, object)
+
+        :param node_type_id:
+            The value to assign to the node_type_id property of this NestedAgentFlowNode.
+        :type node_type_id: str
+
+        :param key:
+            The value to assign to the key property of this NestedAgentFlowNode.
+        :type key: str
+
+        :param time_created:
+            The value to assign to the time_created property of this NestedAgentFlowNode.
+        :type time_created: datetime
+
+        :param time_updated:
+            The value to assign to the time_updated property of this NestedAgentFlowNode.
+        :type time_updated: datetime
+
+        :param inputs:
+            The value to assign to the inputs property of this NestedAgentFlowNode.
+        :type inputs: list[oci.aidataplatform_dp.models.NodeInput]
+
+        :param outputs:
+            The value to assign to the outputs property of this NestedAgentFlowNode.
+        :type outputs: list[oci.aidataplatform_dp.models.NodeOutput]
+
+        :param validation_errors:
+            The value to assign to the validation_errors property of this NestedAgentFlowNode.
+        :type validation_errors: list[oci.aidataplatform_dp.models.ValidationError]
+
+        :param instructions:
+            The value to assign to the instructions property of this NestedAgentFlowNode.
+        :type instructions: str
+
+        :param memory:
+            The value to assign to the memory property of this NestedAgentFlowNode.
+        :type memory: oci.aidataplatform_dp.models.MemoryConfiguration
+
+        :param nested_agent_flow_config:
+            The value to assign to the nested_agent_flow_config property of this NestedAgentFlowNode.
+        :type nested_agent_flow_config: dict(str, str)
+
+        """
+        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',
+            'configuration': 'dict(str, object)',
+            'node_type_id': 'str',
+            'key': 'str',
+            'time_created': 'datetime',
+            'time_updated': 'datetime',
+            'inputs': 'list[NodeInput]',
+            'outputs': 'list[NodeOutput]',
+            'validation_errors': 'list[ValidationError]',
+            'instructions': 'str',
+            'memory': 'MemoryConfiguration',
+            'nested_agent_flow_config': 'dict(str, str)'
+        }
+
+        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',
+            'configuration': 'configuration',
+            'node_type_id': 'nodeTypeId',
+            'key': 'key',
+            'time_created': 'timeCreated',
+            'time_updated': 'timeUpdated',
+            'inputs': 'inputs',
+            'outputs': 'outputs',
+            'validation_errors': 'validationErrors',
+            'instructions': 'instructions',
+            'memory': 'memory',
+            'nested_agent_flow_config': 'nestedAgentFlowConfig'
+        }
+
+        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._configuration = None
+        self._node_type_id = None
+        self._key = None
+        self._time_created = None
+        self._time_updated = None
+        self._inputs = None
+        self._outputs = None
+        self._validation_errors = None
+        self._instructions = None
+        self._memory = None
+        self._nested_agent_flow_config = None
+        self._type = 'NESTED_AGENT_FLOW'
+
+    @property
+    def instructions(self):
+        """
+        Gets the instructions of this NestedAgentFlowNode.
+        System prompt written by the flow developer defining the agent\u2019s goal(s) and what tools the agent has access to.
+
+
+        :return: The instructions of this NestedAgentFlowNode.
+        :rtype: str
+        """
+        return self._instructions
+
+    @instructions.setter
+    def instructions(self, instructions):
+        """
+        Sets the instructions of this NestedAgentFlowNode.
+        System prompt written by the flow developer defining the agent\u2019s goal(s) and what tools the agent has access to.
+
+
+        :param instructions: The instructions of this NestedAgentFlowNode.
+        :type: str
+        """
+        self._instructions = instructions
+
+    @property
+    def memory(self):
+        """
+        Gets the memory of this NestedAgentFlowNode.
+
+        :return: The memory of this NestedAgentFlowNode.
+        :rtype: oci.aidataplatform_dp.models.MemoryConfiguration
+        """
+        return self._memory
+
+    @memory.setter
+    def memory(self, memory):
+        """
+        Sets the memory of this NestedAgentFlowNode.
+
+        :param memory: The memory of this NestedAgentFlowNode.
+        :type: oci.aidataplatform_dp.models.MemoryConfiguration
+        """
+        self._memory = memory
+
+    @property
+    def nested_agent_flow_config(self):
+        """
+        Gets the nested_agent_flow_config of this NestedAgentFlowNode.
+        Extra configuration for the nested agent flow node.
+
+
+        :return: The nested_agent_flow_config of this NestedAgentFlowNode.
+        :rtype: dict(str, str)
+        """
+        return self._nested_agent_flow_config
+
+    @nested_agent_flow_config.setter
+    def nested_agent_flow_config(self, nested_agent_flow_config):
+        """
+        Sets the nested_agent_flow_config of this NestedAgentFlowNode.
+        Extra configuration for the nested agent flow node.
+
+
+        :param nested_agent_flow_config: The nested_agent_flow_config of this NestedAgentFlowNode.
+        :type: dict(str, str)
+        """
+        self._nested_agent_flow_config = nested_agent_flow_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/new_cluster_configuration.py

@@ -0,0 +1,190 @@
+# coding: utf-8
+# Copyright (c) 2026, Oracle and/or its affiliates.  All rights reserved.
+
+
+
+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 NewClusterConfiguration(object):
+    """
+    The cluster configuration to create a new cluster.
+    """
+
+    def __init__(self, **kwargs):
+        """
+        Initializes a new NewClusterConfiguration object with values from keyword arguments.
+        The following keyword arguments are supported (corresponding to the getters/setters of this class):
+
+        :param num_workers:
+            The value to assign to the num_workers property of this NewClusterConfiguration.
+        :type num_workers: int
+
+        :param auto_scale:
+            The value to assign to the auto_scale property of this NewClusterConfiguration.
+        :type auto_scale: oci.aidataplatform_dp.models.AutoScale
+
+        :param cluster_name:
+            The value to assign to the cluster_name property of this NewClusterConfiguration.
+        :type cluster_name: str
+
+        :param spark_version:
+            The value to assign to the spark_version property of this NewClusterConfiguration.
+        :type spark_version: str
+
+        :param spark_conf:
+            The value to assign to the spark_conf property of this NewClusterConfiguration.
+        :type spark_conf: str
+
+        """
+        self.swagger_types = {
+            'num_workers': 'int',
+            'auto_scale': 'AutoScale',
+            'cluster_name': 'str',
+            'spark_version': 'str',
+            'spark_conf': 'str'
+        }
+
+        self.attribute_map = {
+            'num_workers': 'numWorkers',
+            'auto_scale': 'autoScale',
+            'cluster_name': 'clusterName',
+            'spark_version': 'sparkVersion',
+            'spark_conf': 'sparkConf'
+        }
+
+        self._num_workers = None
+        self._auto_scale = None
+        self._cluster_name = None
+        self._spark_version = None
+        self._spark_conf = None
+
+    @property
+    def num_workers(self):
+        """
+        Gets the num_workers of this NewClusterConfiguration.
+        Number of worker nodes configured for this cluster.
+
+
+        :return: The num_workers of this NewClusterConfiguration.
+        :rtype: int
+        """
+        return self._num_workers
+
+    @num_workers.setter
+    def num_workers(self, num_workers):
+        """
+        Sets the num_workers of this NewClusterConfiguration.
+        Number of worker nodes configured for this cluster.
+
+
+        :param num_workers: The num_workers of this NewClusterConfiguration.
+        :type: int
+        """
+        self._num_workers = num_workers
+
+    @property
+    def auto_scale(self):
+        """
+        Gets the auto_scale of this NewClusterConfiguration.
+
+        :return: The auto_scale of this NewClusterConfiguration.
+        :rtype: oci.aidataplatform_dp.models.AutoScale
+        """
+        return self._auto_scale
+
+    @auto_scale.setter
+    def auto_scale(self, auto_scale):
+        """
+        Sets the auto_scale of this NewClusterConfiguration.
+
+        :param auto_scale: The auto_scale of this NewClusterConfiguration.
+        :type: oci.aidataplatform_dp.models.AutoScale
+        """
+        self._auto_scale = auto_scale
+
+    @property
+    def cluster_name(self):
+        """
+        Gets the cluster_name of this NewClusterConfiguration.
+        A unique name for the job cluster.
+
+
+        :return: The cluster_name of this NewClusterConfiguration.
+        :rtype: str
+        """
+        return self._cluster_name
+
+    @cluster_name.setter
+    def cluster_name(self, cluster_name):
+        """
+        Sets the cluster_name of this NewClusterConfiguration.
+        A unique name for the job cluster.
+
+
+        :param cluster_name: The cluster_name of this NewClusterConfiguration.
+        :type: str
+        """
+        self._cluster_name = cluster_name
+
+    @property
+    def spark_version(self):
+        """
+        Gets the spark_version of this NewClusterConfiguration.
+        The Spark version used to run the application.
+
+
+        :return: The spark_version of this NewClusterConfiguration.
+        :rtype: str
+        """
+        return self._spark_version
+
+    @spark_version.setter
+    def spark_version(self, spark_version):
+        """
+        Sets the spark_version of this NewClusterConfiguration.
+        The Spark version used to run the application.
+
+
+        :param spark_version: The spark_version of this NewClusterConfiguration.
+        :type: str
+        """
+        self._spark_version = spark_version
+
+    @property
+    def spark_conf(self):
+        """
+        Gets the spark_conf of this NewClusterConfiguration.
+        The spark configuration in key-value pairs.
+
+
+        :return: The spark_conf of this NewClusterConfiguration.
+        :rtype: str
+        """
+        return self._spark_conf
+
+    @spark_conf.setter
+    def spark_conf(self, spark_conf):
+        """
+        Sets the spark_conf of this NewClusterConfiguration.
+        The spark configuration in key-value pairs.
+
+
+        :param spark_conf: The spark_conf of this NewClusterConfiguration.
+        :type: str
+        """
+        self._spark_conf = spark_conf
+
+    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/nl_to_sql_tool.py

@@ -0,0 +1,180 @@
+# coding: utf-8
+# Copyright (c) 2026, Oracle and/or its affiliates.  All rights reserved.
+
+
+from .tool import Tool
+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 NlToSqlTool(Tool):
+    """
+    Derived model for Natural Language (NL) to SQL Tool
+    """
+
+    def __init__(self, **kwargs):
+        """
+        Initializes a new NlToSqlTool object with values from keyword arguments. The default value of the :py:attr:`~oci.aidataplatform_dp.models.NlToSqlTool.tool_type` attribute
+        of this class is ``NL2SQL`` and it should not be changed.
+        The following keyword arguments are supported (corresponding to the getters/setters of this class):
+
+        :param key:
+            The value to assign to the key property of this NlToSqlTool.
+        :type key: str
+
+        :param display_name:
+            The value to assign to the display_name property of this NlToSqlTool.
+        :type display_name: str
+
+        :param workspace_key:
+            The value to assign to the workspace_key property of this NlToSqlTool.
+        :type workspace_key: str
+
+        :param description:
+            The value to assign to the description property of this NlToSqlTool.
+        :type description: str
+
+        :param tool_type:
+            The value to assign to the tool_type property of this NlToSqlTool.
+            Allowed values for this property are: "RAG", "SQL", "PROMPT", "NL2SQL", "MCP", "CUSTOM", "HTTP"
+        :type tool_type: str
+
+        :param position_x:
+            The value to assign to the position_x property of this NlToSqlTool.
+        :type position_x: float
+
+        :param position_y:
+            The value to assign to the position_y property of this NlToSqlTool.
+        :type position_y: float
+
+        :param properties:
+            The value to assign to the properties property of this NlToSqlTool.
+        :type properties: dict(str, object)
+
+        :param time_created:
+            The value to assign to the time_created property of this NlToSqlTool.
+        :type time_created: datetime
+
+        :param time_updated:
+            The value to assign to the time_updated property of this NlToSqlTool.
+        :type time_updated: datetime
+
+        :param created_by:
+            The value to assign to the created_by property of this NlToSqlTool.
+        :type created_by: str
+
+        :param updated_by:
+            The value to assign to the updated_by property of this NlToSqlTool.
+        :type updated_by: str
+
+        :param input_schema:
+            The value to assign to the input_schema property of this NlToSqlTool.
+        :type input_schema: oci.aidataplatform_dp.models.NlToSqlToolInputSchema
+
+        :param tool_config:
+            The value to assign to the tool_config property of this NlToSqlTool.
+        :type tool_config: oci.aidataplatform_dp.models.NlToSqlToolConfiguration
+
+        """
+        self.swagger_types = {
+            'key': 'str',
+            'display_name': 'str',
+            'workspace_key': 'str',
+            'description': 'str',
+            'tool_type': 'str',
+            'position_x': 'float',
+            'position_y': 'float',
+            'properties': 'dict(str, object)',
+            'time_created': 'datetime',
+            'time_updated': 'datetime',
+            'created_by': 'str',
+            'updated_by': 'str',
+            'input_schema': 'NlToSqlToolInputSchema',
+            'tool_config': 'NlToSqlToolConfiguration'
+        }
+
+        self.attribute_map = {
+            'key': 'key',
+            'display_name': 'displayName',
+            'workspace_key': 'workspaceKey',
+            'description': 'description',
+            'tool_type': 'toolType',
+            'position_x': 'positionX',
+            'position_y': 'positionY',
+            'properties': 'properties',
+            'time_created': 'timeCreated',
+            'time_updated': 'timeUpdated',
+            'created_by': 'createdBy',
+            'updated_by': 'updatedBy',
+            'input_schema': 'inputSchema',
+            'tool_config': 'toolConfig'
+        }
+
+        self._key = None
+        self._display_name = None
+        self._workspace_key = None
+        self._description = None
+        self._tool_type = None
+        self._position_x = None
+        self._position_y = None
+        self._properties = None
+        self._time_created = None
+        self._time_updated = None
+        self._created_by = None
+        self._updated_by = None
+        self._input_schema = None
+        self._tool_config = None
+        self._tool_type = 'NL2SQL'
+
+    @property
+    def input_schema(self):
+        """
+        **[Required]** Gets the input_schema of this NlToSqlTool.
+
+        :return: The input_schema of this NlToSqlTool.
+        :rtype: oci.aidataplatform_dp.models.NlToSqlToolInputSchema
+        """
+        return self._input_schema
+
+    @input_schema.setter
+    def input_schema(self, input_schema):
+        """
+        Sets the input_schema of this NlToSqlTool.
+
+        :param input_schema: The input_schema of this NlToSqlTool.
+        :type: oci.aidataplatform_dp.models.NlToSqlToolInputSchema
+        """
+        self._input_schema = input_schema
+
+    @property
+    def tool_config(self):
+        """
+        **[Required]** Gets the tool_config of this NlToSqlTool.
+
+        :return: The tool_config of this NlToSqlTool.
+        :rtype: oci.aidataplatform_dp.models.NlToSqlToolConfiguration
+        """
+        return self._tool_config
+
+    @tool_config.setter
+    def tool_config(self, tool_config):
+        """
+        Sets the tool_config of this NlToSqlTool.
+
+        :param tool_config: The tool_config of this NlToSqlTool.
+        :type: oci.aidataplatform_dp.models.NlToSqlToolConfiguration
+        """
+        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