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

aidp_python_client/aidataplatform_dp/models/sql_result.py

@@ -0,0 +1,70 @@
+# 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 SqlResult(object):
+    """
+    Rows for SQL result.
+    """
+
+    def __init__(self, **kwargs):
+        """
+        Initializes a new SqlResult object with values from keyword arguments.
+        The following keyword arguments are supported (corresponding to the getters/setters of this class):
+
+        :param rows:
+            The value to assign to the rows property of this SqlResult.
+        :type rows: list[dict(str, str)]
+
+        """
+        self.swagger_types = {
+            'rows': 'list[dict(str, str)]'
+        }
+
+        self.attribute_map = {
+            'rows': 'rows'
+        }
+
+        self._rows = None
+
+    @property
+    def rows(self):
+        """
+        Gets the rows of this SqlResult.
+        Array of result rows with dynamic columns.
+
+
+        :return: The rows of this SqlResult.
+        :rtype: list[dict(str, str)]
+        """
+        return self._rows
+
+    @rows.setter
+    def rows(self, rows):
+        """
+        Sets the rows of this SqlResult.
+        Array of result rows with dynamic columns.
+
+
+        :param rows: The rows of this SqlResult.
+        :type: list[dict(str, str)]
+        """
+        self._rows = rows
+
+    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/sql_test_tool_result.py

@@ -0,0 +1,76 @@
+# coding: utf-8
+# Copyright (c) 2026, Oracle and/or its affiliates.  All rights reserved.
+
+
+from .test_tool_result import TestToolResult
+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 SqlTestToolResult(TestToolResult):
+    """
+    Response when toolType = SQL.
+    """
+
+    def __init__(self, **kwargs):
+        """
+        Initializes a new SqlTestToolResult object with values from keyword arguments. The default value of the :py:attr:`~oci.aidataplatform_dp.models.SqlTestToolResult.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 tool_type:
+            The value to assign to the tool_type property of this SqlTestToolResult.
+            Allowed values for this property are: "SQL", "PROMPT", "RAG", "MCP", "HTTP", "CUSTOM"
+        :type tool_type: str
+
+        :param result:
+            The value to assign to the result property of this SqlTestToolResult.
+        :type result: oci.aidataplatform_dp.models.SqlResult
+
+        """
+        self.swagger_types = {
+            'tool_type': 'str',
+            'result': 'SqlResult'
+        }
+
+        self.attribute_map = {
+            'tool_type': 'toolType',
+            'result': 'result'
+        }
+
+        self._tool_type = None
+        self._result = None
+        self._tool_type = 'SQL'
+
+    @property
+    def result(self):
+        """
+        **[Required]** Gets the result of this SqlTestToolResult.
+
+        :return: The result of this SqlTestToolResult.
+        :rtype: oci.aidataplatform_dp.models.SqlResult
+        """
+        return self._result
+
+    @result.setter
+    def result(self, result):
+        """
+        Sets the result of this SqlTestToolResult.
+
+        :param result: The result of this SqlTestToolResult.
+        :type: oci.aidataplatform_dp.models.SqlResult
+        """
+        self._result = result
+
+    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/sql_tool.py

@@ -0,0 +1,184 @@
+# 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 SqlTool(Tool):
+    """
+    Derived model a SQL Tool
+    """
+
+    def __init__(self, **kwargs):
+        """
+        Initializes a new SqlTool object with values from keyword arguments. The default value of the :py:attr:`~oci.aidataplatform_dp.models.SqlTool.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 key:
+            The value to assign to the key property of this SqlTool.
+        :type key: str
+
+        :param display_name:
+            The value to assign to the display_name property of this SqlTool.
+        :type display_name: str
+
+        :param workspace_key:
+            The value to assign to the workspace_key property of this SqlTool.
+        :type workspace_key: str
+
+        :param description:
+            The value to assign to the description property of this SqlTool.
+        :type description: str
+
+        :param tool_type:
+            The value to assign to the tool_type property of this SqlTool.
+            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 SqlTool.
+        :type position_x: float
+
+        :param position_y:
+            The value to assign to the position_y property of this SqlTool.
+        :type position_y: float
+
+        :param properties:
+            The value to assign to the properties property of this SqlTool.
+        :type properties: dict(str, object)
+
+        :param time_created:
+            The value to assign to the time_created property of this SqlTool.
+        :type time_created: datetime
+
+        :param time_updated:
+            The value to assign to the time_updated property of this SqlTool.
+        :type time_updated: datetime
+
+        :param created_by:
+            The value to assign to the created_by property of this SqlTool.
+        :type created_by: str
+
+        :param updated_by:
+            The value to assign to the updated_by property of this SqlTool.
+        :type updated_by: str
+
+        :param input_schema:
+            The value to assign to the input_schema property of this SqlTool.
+        :type input_schema: dict(str, object)
+
+        :param tool_config:
+            The value to assign to the tool_config property of this SqlTool.
+        :type tool_config: oci.aidataplatform_dp.models.SqlToolConfiguration
+
+        """
+        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': 'dict(str, object)',
+            'tool_config': 'SqlToolConfiguration'
+        }
+
+        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 = 'SQL'
+
+    @property
+    def input_schema(self):
+        """
+        Gets the input_schema of this SqlTool.
+        The list of properties in the inputSchema, along with the default value and description of each property
+
+
+        :return: The input_schema of this SqlTool.
+        :rtype: dict(str, object)
+        """
+        return self._input_schema
+
+    @input_schema.setter
+    def input_schema(self, input_schema):
+        """
+        Sets the input_schema of this SqlTool.
+        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 SqlTool.
+        :type: dict(str, object)
+        """
+        self._input_schema = input_schema
+
+    @property
+    def tool_config(self):
+        """
+        Gets the tool_config of this SqlTool.
+
+        :return: The tool_config of this SqlTool.
+        :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 SqlTool.
+
+        :param tool_config: The tool_config of this SqlTool.
+        :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/sql_tool_configuration.py

@@ -0,0 +1,194 @@
+# 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 SqlToolConfiguration(object):
+    """
+    Tool configurations are set by the agent developer when they create the tool. | The agent does not see those configurations and can NOT modify their values
+    """
+
+    def __init__(self, **kwargs):
+        """
+        Initializes a new SqlToolConfiguration object with values from keyword arguments.
+        The following keyword arguments are supported (corresponding to the getters/setters of this class):
+
+        :param catalog_key:
+            The value to assign to the catalog_key property of this SqlToolConfiguration.
+        :type catalog_key: str
+
+        :param schema_key:
+            The value to assign to the schema_key property of this SqlToolConfiguration.
+        :type schema_key: str
+
+        :param query:
+            The value to assign to the query property of this SqlToolConfiguration.
+        :type query: str
+
+        :param is_row_limit_enabled:
+            The value to assign to the is_row_limit_enabled property of this SqlToolConfiguration.
+        :type is_row_limit_enabled: bool
+
+        :param max_rows:
+            The value to assign to the max_rows property of this SqlToolConfiguration.
+        :type max_rows: int
+
+        """
+        self.swagger_types = {
+            'catalog_key': 'str',
+            'schema_key': 'str',
+            'query': 'str',
+            'is_row_limit_enabled': 'bool',
+            'max_rows': 'int'
+        }
+
+        self.attribute_map = {
+            'catalog_key': 'catalogKey',
+            'schema_key': 'schemaKey',
+            'query': 'query',
+            'is_row_limit_enabled': 'isRowLimitEnabled',
+            'max_rows': 'maxRows'
+        }
+
+        self._catalog_key = None
+        self._schema_key = None
+        self._query = None
+        self._is_row_limit_enabled = None
+        self._max_rows = None
+
+    @property
+    def catalog_key(self):
+        """
+        Gets the catalog_key of this SqlToolConfiguration.
+        The Catalog to use for SQL query execution
+
+
+        :return: The catalog_key of this SqlToolConfiguration.
+        :rtype: str
+        """
+        return self._catalog_key
+
+    @catalog_key.setter
+    def catalog_key(self, catalog_key):
+        """
+        Sets the catalog_key of this SqlToolConfiguration.
+        The Catalog to use for SQL query execution
+
+
+        :param catalog_key: The catalog_key of this SqlToolConfiguration.
+        :type: str
+        """
+        self._catalog_key = catalog_key
+
+    @property
+    def schema_key(self):
+        """
+        Gets the schema_key of this SqlToolConfiguration.
+        The Schema to use for SQL query execution
+
+
+        :return: The schema_key of this SqlToolConfiguration.
+        :rtype: str
+        """
+        return self._schema_key
+
+    @schema_key.setter
+    def schema_key(self, schema_key):
+        """
+        Sets the schema_key of this SqlToolConfiguration.
+        The Schema to use for SQL query execution
+
+
+        :param schema_key: The schema_key of this SqlToolConfiguration.
+        :type: str
+        """
+        self._schema_key = schema_key
+
+    @property
+    def query(self):
+        """
+        Gets the query of this SqlToolConfiguration.
+        The SQL query to execute
+
+
+        :return: The query of this SqlToolConfiguration.
+        :rtype: str
+        """
+        return self._query
+
+    @query.setter
+    def query(self, query):
+        """
+        Sets the query of this SqlToolConfiguration.
+        The SQL query to execute
+
+
+        :param query: The query of this SqlToolConfiguration.
+        :type: str
+        """
+        self._query = query
+
+    @property
+    def is_row_limit_enabled(self):
+        """
+        Gets the is_row_limit_enabled of this SqlToolConfiguration.
+        Whether to enforce the row limit when running the SQL tool
+
+
+        :return: The is_row_limit_enabled of this SqlToolConfiguration.
+        :rtype: bool
+        """
+        return self._is_row_limit_enabled
+
+    @is_row_limit_enabled.setter
+    def is_row_limit_enabled(self, is_row_limit_enabled):
+        """
+        Sets the is_row_limit_enabled of this SqlToolConfiguration.
+        Whether to enforce the row limit when running the SQL tool
+
+
+        :param is_row_limit_enabled: The is_row_limit_enabled of this SqlToolConfiguration.
+        :type: bool
+        """
+        self._is_row_limit_enabled = is_row_limit_enabled
+
+    @property
+    def max_rows(self):
+        """
+        Gets the max_rows of this SqlToolConfiguration.
+        Maximum number of result rows returned per SQL tool execution to prevent oversized result sets
+
+
+        :return: The max_rows of this SqlToolConfiguration.
+        :rtype: int
+        """
+        return self._max_rows
+
+    @max_rows.setter
+    def max_rows(self, max_rows):
+        """
+        Sets the max_rows of this SqlToolConfiguration.
+        Maximum number of result rows returned per SQL tool execution to prevent oversized result sets
+
+
+        :param max_rows: The max_rows of this SqlToolConfiguration.
+        :type: int
+        """
+        self._max_rows = max_rows
+
+    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