dataiku-api-client 14.7.4__tar.gz → 15.0.1__tar.gz

{dataiku_api_client-14.7.4 → dataiku_api_client-15.0.1}/HISTORY.txt

@@ -1,6 +1,15 @@
 Changelog
 ==========
 
+15.0.1 (2026-09-15)
+-------------------
+
+* Initial release for DSS 15.0.1
+
+15.0.0 (2026-08-14)
+-------------------
+
+* Initial release for DSS 15.0.0
 
 14.7.4 (2026-09-14)
 -------------------

{dataiku_api_client-14.7.4/dataiku_api_client.egg-info → dataiku_api_client-15.0.1}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: dataiku-api-client
-Version: 14.7.4
+Version: 15.0.1
 Summary: Python API client for Dataiku APIs
 Home-page: https://www.dataiku.com
-Download-URL: https://github.com/dataiku/dataiku-api-client-python/tarball/14.7.4
+Download-URL: https://github.com/dataiku/dataiku-api-client-python/tarball/15.0.1
 Author: Dataiku
 Author-email: support@dataiku.com
 License: Apache Software License
@@ -35,6 +35,15 @@ https://doc.dataiku.com/dss/latest/python-api/rest-api-client/index.html
 Changelog
 ==========
 
+15.0.1 (2026-09-15)
+-------------------
+
+* Initial release for DSS 15.0.1
+
+15.0.0 (2026-08-14)
+-------------------
+
+* Initial release for DSS 15.0.0
 
 14.7.4 (2026-09-14)
 -------------------

{dataiku_api_client-14.7.4 → dataiku_api_client-15.0.1/dataiku_api_client.egg-info}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: dataiku-api-client
-Version: 14.7.4
+Version: 15.0.1
 Summary: Python API client for Dataiku APIs
 Home-page: https://www.dataiku.com
-Download-URL: https://github.com/dataiku/dataiku-api-client-python/tarball/14.7.4
+Download-URL: https://github.com/dataiku/dataiku-api-client-python/tarball/15.0.1
 Author: Dataiku
 Author-email: support@dataiku.com
 License: Apache Software License
@@ -35,6 +35,15 @@ https://doc.dataiku.com/dss/latest/python-api/rest-api-client/index.html
 Changelog
 ==========
 
+15.0.1 (2026-09-15)
+-------------------
+
+* Initial release for DSS 15.0.1
+
+15.0.0 (2026-08-14)
+-------------------
+
+* Initial release for DSS 15.0.0
 
 14.7.4 (2026-09-14)
 -------------------

{dataiku_api_client-14.7.4 → dataiku_api_client-15.0.1}/dataiku_api_client.egg-info/SOURCES.txt

@@ -26,6 +26,7 @@ dataikuapi/dss/__init__.py
 dataikuapi/dss/admin.py
 dataikuapi/dss/agent.py
 dataikuapi/dss/agent_review.py
+dataikuapi/dss/agent_skill.py
 dataikuapi/dss/agent_tool.py
 dataikuapi/dss/analysis.py
 dataikuapi/dss/apideployer.py

{dataiku_api_client-14.7.4 → dataiku_api_client-15.0.1}/dataikuapi/dss/admin.py

@@ -4,6 +4,8 @@ import json
 import warnings
 import logging
 from datetime import datetime
+
+from .utils import DSSInfoMessages
 from ..utils import _timestamp_ms_to_zoned_datetime
 
 logger = logging.getLogger("dataikuapi.dss.admin")
@@ -2208,6 +2210,7 @@ class DSSCodeEnvContainerConfsBearer(object):
         * SYSTEM_LEVEL_CUDA_122_CUDNN_897
         * CUDA_SUPPORT_FOR_TORCH2_WITH_PYPI_NVIDIA_PACKAGES
         * BASIC_GPU_ENABLING
+        * HUGGING_FACE_LOCAL_CPU
         * PYTHON36_SUPPORT
         * PYTHON37_SUPPORT
         * PYTHON38_SUPPORT
@@ -3305,6 +3308,58 @@ class DSSCodeStudioTemplate(object):
             params={"withNoCache": disable_docker_cache})
         return DSSFuture(self.client, future_response.get('jobId', None), future_response)
 
+    def get_export_stream(self):
+        """
+        Export the template as a zip archive.
+
+        :returns: the exported archive as a stream
+        :rtype: file-like object
+        """
+        return self.client._perform_raw(
+            "GET", "/admin/code-studios/%s/export" % (self.template_id)
+        ).raw
+
+    def export_to_file(self, path):
+        """
+        Export the template to a file.
+
+        This produces a zip file with the template's definition and its resources, if any. The zip can be imported with
+        :meth:`dataikuapi.DSSClient.import_code_studio_template`
+
+        :param str path: the destination file path
+        """
+        with open(path, 'wb') as f:
+            export_stream = self.client._perform_raw(
+                "GET", "/admin/code-studios/%s/export" % (self.template_id)
+            )
+            for chunk in export_stream.iter_content(chunk_size=32768):
+                if chunk:
+                    f.write(chunk)
+            f.flush()
+
+    ########################################################
+    # Template deletion
+    ########################################################
+
+    def delete(self, delete_runtime=False, delete_images=False):
+        """
+        Delete the template.
+
+        :param bool delete_runtime: if True, also delete the Code Studio runtimes created from this template
+        :param bool delete_images: if True, also delete the Docker images and build folders associated with this template
+
+        :return: backend info/warning/error messages about the deletion
+        :rtype: dict
+        """
+        resp = self.client._perform_json(
+            "DELETE",
+            "/admin/code-studios/%s" % (self.template_id),
+            params={
+                "deleteRuntime": delete_runtime,
+                "deleteImages": delete_images
+            })
+        return DSSInfoMessages(resp)
+
 class DSSCodeStudioTemplateSettings(object):
     """
     The settings of a code studio template

dataiku_api_client-15.0.1/dataikuapi/dss/agent_skill.py

@@ -0,0 +1,237 @@
+from requests import utils
+
+from .utils import DSSTaggableObjectListItem, DSSTaggableObjectSettings
+
+
+class DSSAgentSkillListItem(DSSTaggableObjectListItem):
+    """
+    .. important::
+        Do not instantiate this class directly, instead use :meth:`dataikuapi.dss.project.DSSProject.list_agent_skills`.
+    """
+
+    def __init__(self, client, project_key, data):
+        super(DSSAgentSkillListItem, self).__init__(data)
+        self.client = client
+        self.project_key = data.get("projectKey", project_key)
+
+    def to_agent_skill(self):
+        """
+        Convert the current item.
+
+        :rtype: :class:`dataikuapi.dss.agent_skill.DSSAgentSkill`
+        """
+        return DSSAgentSkill(self.client, self.project_key, self._data["id"])
+
+    @property
+    def id(self):
+        """
+        :returns: The id of the skill.
+        :rtype: string
+        """
+        return self._data["id"]
+
+    @property
+    def name(self):
+        """
+        :returns: The name of the skill.
+        :rtype: string
+        """
+        return self._data.get("name")
+
+
+class DSSAgentSkill(object):
+    """
+    .. important::
+        Do not instantiate this class directly, instead use :meth:`dataikuapi.dss.project.DSSProject.get_agent_skill`.
+    """
+
+    def __init__(self, client, project_key, skill_id):
+        self.client = client
+        self.project_key = project_key
+        self.skill_id = skill_id
+
+    @property
+    def id(self):
+        """
+        :returns: The id of the skill.
+        :rtype: string
+        """
+        return self.skill_id
+
+    def get_settings(self):
+        """
+        Get the DSS metadata settings of the agent skill.
+
+        The parsed ``SKILL.md`` fields are available through
+        :meth:`get_skill_content` and the raw file through :meth:`get_file`.
+
+        :return: a handle on the skill settings
+        :rtype: :class:`dataikuapi.dss.agent_skill.DSSAgentSkillSettings`
+        """
+        settings = self.client._perform_json(
+            "GET",
+            "/projects/%s/agents/skills/%s" % (self.project_key, self.id),
+        )
+        return DSSAgentSkillSettings(self, settings)
+
+    def get_skill_content(self):
+        """
+        Get the parsed contents of ``SKILL.md``.
+
+        :returns: A dictionary containing ``name``, ``description``,
+            ``metadata``, and ``instructions``.
+        :rtype: dict
+        """
+        return self.client._perform_json(
+            "GET",
+            "/projects/%s/agents/skills/%s/content"
+            % (self.project_key, self.id),
+        )
+
+    def delete(self):
+        """
+        Delete the agent skill.
+        """
+        return self.client._perform_empty("DELETE", "/projects/%s/agents/skills/%s" % (self.project_key, self.id))
+
+    def list_resources(self):
+        """
+        List the files and folders attached to this skill as a recursive tree.
+
+        :rtype: list[dict]
+        """
+        return self.client._perform_json(
+            "GET",
+            "/projects/%s/agents/skills/%s/resources/contents" % (self.project_key, self.id),
+        )
+
+    def get_file(self, path):
+        """
+        Get a file's contents.
+
+        :param str path: Root-relative path of the file to download
+        :rtype: :class:`requests.models.Response`
+        """
+        return self.client._perform_raw(
+            "GET",
+            "/projects/%s/agents/skills/%s/resources/contents/%s"
+            % (self.project_key, self.id, utils.quote(path)),
+        )
+
+    def get_file_details(self, path):
+        """
+        Get a file's metadata without its content.
+
+        :param str path: Root-relative path of the file
+        :rtype: dict
+        """
+        return self.client._perform_json(
+            "GET",
+            "/projects/%s/agents/skills/%s/resources/details/%s"
+            % (self.project_key, self.id, utils.quote(path)),
+        )
+
+    def put_file(self, path, data):
+        """
+        Create or overwrite a file.
+
+        Strings are encoded as UTF-8 and bytes are stored unchanged. Parent
+        folders must already exist. Replacing the root ``SKILL.md`` validates
+        the supplied content and rejects an invalid skill file.
+
+        :param str path: Root-relative path of the file
+        :param data: String, bytes, or file-like content
+        :rtype: dict
+        """
+        if isinstance(data, str):
+            data = data.encode("utf-8")
+        elif not isinstance(data, bytes) and not hasattr(data, "read"):
+            raise TypeError("data must be a string, bytes, or file-like object")
+        return self.client._perform_json(
+            "PUT",
+            "/projects/%s/agents/skills/%s/resources/contents/%s"
+            % (self.project_key, self.id, utils.quote(path)),
+            files={"file": (path.rsplit("/", 1)[-1], data)},
+        )
+
+    def rename_resource(self, path, new_name):
+        """
+        Rename a resource.
+
+        :param str path: Root-relative path of the existing resource
+        :param str new_name: New file name, without a folder path
+        :rtype: str
+        """
+        return self.client._perform_raw(
+            "POST",
+            "/projects/%s/agents/skills/%s/resources/contents-actions/rename"
+            % (self.project_key, self.id),
+            body={"oldPath": path, "newName": new_name},
+        ).text
+
+    def move_resource(self, path, new_path):
+        """
+        Move a resource to a destination folder.
+
+        :param str path: Root-relative path of the existing resource
+        :param str new_path: Root-relative path of the destination folder, or an empty string for the skill root
+        :rtype: str
+        """
+        return self.client._perform_raw(
+            "POST",
+            "/projects/%s/agents/skills/%s/resources/contents-actions/move"
+            % (self.project_key, self.id),
+            body={"oldPath": path, "newPath": new_path},
+        ).text
+
+    def create_folder(self, path):
+        """
+        Create a resource folder in the skill.
+
+        Missing parent folders are created as needed.
+
+        :param str path: Root-relative path of the folder to create
+        """
+        return self.client._perform_empty(
+            "POST",
+            "/projects/%s/agents/skills/%s/resources/folders/%s"
+            % (self.project_key, self.id, utils.quote(path)),
+        )
+
+    def delete_resource(self, path):
+        """
+        Delete a resource from the skill.
+
+        :param str path: Root-relative path of the resource to delete
+        """
+        return self.client._perform_empty(
+            "DELETE",
+            "/projects/%s/agents/skills/%s/resources/contents/%s"
+            % (self.project_key, self.id, utils.quote(path)),
+        )
+
+class DSSAgentSkillSettings(DSSTaggableObjectSettings):
+    def __init__(self, agent_skill, settings):
+        super(DSSAgentSkillSettings, self).__init__(settings)
+        self.agent_skill = agent_skill
+
+    def get_raw(self):
+        """
+        Get the raw settings of the skill.
+
+        :rtype: dict
+        """
+        return self._tod
+
+    def save(self):
+        """
+        Saves the DSS metadata settings of the agent skill.
+
+        This does not modify ``SKILL.md``. Use :meth:`DSSAgentSkill.put_file`
+        to replace the skill file.
+        """
+        self.agent_skill.client._perform_empty(
+            "PUT",
+            "/projects/%s/agents/skills/%s" % (self.agent_skill.project_key, self.agent_skill.id),
+            body=self._tod,
+        )

{dataiku_api_client-14.7.4 → dataiku_api_client-15.0.1}/dataikuapi/dss/agent_tool.py

@@ -1,7 +1,8 @@
+import hashlib
+import json
 from .utils import DSSTaggableObjectListItem, DSSTaggableObjectSettings, AnyLoc
 from .knowledgebank import DSSKnowledgeBank, DSSKnowledgeBankListItem
 from .llm_tracing import prepare_query_for_nested_llm_mesh_call
-import json
 
 class DSSAgentToolListItem(DSSTaggableObjectListItem):
     """
@@ -55,6 +56,7 @@ class DSSAgentTool(object):
         self.project_key = project_key
         self.tool_id = tool_id
         self._descriptor = descriptor
+        self._subtool_mapping = {}
 
     @property
     def id(self):
@@ -108,11 +110,106 @@ class DSSAgentTool(object):
         from dataikuapi.dss.langchain.tool import convert_to_langchain_structured_tool
         return convert_to_langchain_structured_tool(self, context)
 
+    @staticmethod
+    def _compute_spec_hash(tool_ref, additional_description, subtool_name=None):
+        additional_description = (additional_description or "").strip()
+        subtool_name = (subtool_name or "").strip()
+        fields = [tool_ref, additional_description, subtool_name]
+        serialized_fields = json.dumps(fields, sort_keys=True, ensure_ascii=True)
+        return hashlib.sha256(serialized_fields.encode("utf-8")).hexdigest()[:6]
+
+    @staticmethod
+    def _get_function_tool_descriptor_parts(
+        tool_descriptor,
+        spec_hash,
+        additional_description,
+        subtool_descriptor=None,
+    ):
+
+        name = "{}__{}".format(tool_descriptor['name'], spec_hash)
+
+        if subtool_descriptor:
+            description = subtool_descriptor.get("description", "")
+            parameters = subtool_descriptor["inputSchema"]
+            name += "__" + subtool_descriptor["name"]
+        else:
+            description = tool_descriptor.get("description", "")
+            parameters = tool_descriptor["inputSchema"]
+
+        if additional_description:
+            description = "{}\n\n{}".format(description, additional_description).strip()
+
+        return (
+            name,
+            description,
+            parameters,
+        )
+
+    @staticmethod
+    def _build_function_tool_descriptor(name, description, parameters):
+        return {
+            "type": "function",
+            "function": {
+                "name": name,
+                "description": description,
+                "parameters": parameters,
+            }
+        }
+
+    def _as_llm_mesh_tool(self):
+        tool_descriptor = self.get_descriptor()
+        if type(self._descriptor) is not dict:
+            raise Exception("Invalid descriptor type: expected `dict`, got %s" % type(self._descriptor))
+        settings = self.get_settings().get_raw()
+        additional_description = settings.get("additionalDescriptionForLLM") or ""
+        result = []
+        if tool_descriptor.get("multiple"):
+            for subtool in tool_descriptor.get("subtools", []):
+                if not subtool.get("enabled"):
+                    continue
+                subtool_spec_hash = self._compute_spec_hash(
+                    self.tool_id,
+                    additional_description,
+                    subtool.get("name"),
+                )
+                name, description, parameters = self._get_function_tool_descriptor_parts(
+                    tool_descriptor,
+                    subtool_spec_hash,
+                    additional_description,
+                    subtool_descriptor=subtool,
+                )
+                self._subtool_mapping[name] = subtool.get("name")
+                result.append(self._build_function_tool_descriptor(
+                    name=name,
+                    description=description,
+                    parameters=parameters,
+                ))
+        else:
+            computed_spec_hash = self._compute_spec_hash(
+                self.tool_id,
+                additional_description,
+            )
+            name, description, parameters = self._get_function_tool_descriptor_parts(
+                tool_descriptor,
+                computed_spec_hash,
+                additional_description,
+            )
+            result.append(self._build_function_tool_descriptor(
+                name=name,
+                description=description,
+                parameters=parameters,
+            ))
+
+        return result
+
+    def _get_subtool_name(self, tool_name):
+        return self._subtool_mapping.get(tool_name)
+
     def run(self, input, context=None, subtool_name=None, memory_fragment=None, tool_validation_responses=None, tool_validation_requests=None):
         """
         Execute a tool call
 
-        :param str input: Text input
+        :param dict input: Input for the tool
         :param dict context: Additional request context
         :param str subtool_name: Name of the sub-tool, if applicable (e.g., for a MCP tool)
         :rtype: dict

{dataiku_api_client-14.7.4 → dataiku_api_client-15.0.1}/dataikuapi/dss/analysis.py

@@ -258,6 +258,39 @@ class DSSAnalysis(object):
             mltask.wait_guess_complete()
         return mltask
 
+    def create_multi_target_prediction_ml_task(self,
+                                               target_variables,
+                                               guess_policy="MULTI_TARGET_DEFAULT",
+                                               wait_guess_complete=True):
+        """Creates a new multi-target regression task in this visual analysis lab for a dataset.
+
+        :param list target_variables: List of target variables to predict
+        :param string guess_policy: Policy to use for setting the default parameters.
+                                    Valid values are: MULTI_TARGET_DEFAULT
+        :param boolean wait_guess_complete: If False, the returned ML task will be in 'guessing' state, i.e. analyzing the input dataset to determine feature handling and algorithms.
+                                            You should wait for the guessing to be completed by calling
+                                            ``wait_guess_complete`` on the returned object before doing anything
+                                            else (in particular calling ``train`` or ``get_settings``)
+        :return :class dataiku.dss.ml.DSSMLTask
+        """
+        obj = {
+            "taskType": "PREDICTION",
+            "targetVariables": target_variables,
+            "backendType": "PY_MEMORY",
+            "guessPolicy":  guess_policy,
+            "predictionType": "MULTI_TARGET_REGRESSION"
+        }
+        ref = self.client._perform_json(
+            "POST",
+            "/projects/{project_key}/lab/{analysis_id}/models/".format(project_key=self.project_key, analysis_id=self.analysis_id),
+            body=obj
+        )
+        mltask = DSSMLTask(self.client, self.project_key, self.analysis_id, ref["mlTaskId"])
+
+        if wait_guess_complete:
+            mltask.wait_guess_complete()
+        return mltask
+
     def create_causal_prediction_ml_task(self,
                                          outcome_variable,
                                          treatment_variable,

{dataiku_api_client-14.7.4 → dataiku_api_client-15.0.1}/dataikuapi/dss/apiservice.py

@@ -116,7 +116,7 @@ class DSSAPIServiceSettings(object):
         :return: ist of endpoints, each one a dict. Endpoint have different fields depending on their type, but always have at least:
 
                         * **id** : identifier of the endpoint
-                        * **type** : type of endpoint. Possible values: STD_PREDICTION, STD_CLUSTERING, STD_FORECAST, STD_CAUSAL_PREDICTION, CUSTOM_PREDICTION, CUSTOM_R_PREDICTION, R_FUNCTION, PY_FUNCTION, DATASETS_LOOKUP, SQL_QUERY
+                        * **type** : type of endpoint. Possible values: STD_PREDICTION, STD_CLUSTERING, STD_FORECAST, STD_CAUSAL_PREDICTION, STD_MULTITARGET_PREDICTION, CUSTOM_PREDICTION, CUSTOM_R_PREDICTION, R_FUNCTION, PY_FUNCTION, DATASETS_LOOKUP, SQL_QUERY
 
         :rtype: list[dict]
         """
@@ -135,6 +135,19 @@ class DSSAPIServiceSettings(object):
             "modelRef": saved_model_id
         })
 
+    def add_multitarget_prediction_endpoint(self, endpoint_id, saved_model_id):
+        """
+        Add a new "visual multi-target prediction" endpoint to this API service.
+
+        :param string endpoint_id: identifier of the new endpoint to create
+        :param string saved_model_id: identifier of the saved model (that is currently deployed to the Flow) to use
+        """
+        self.settings["endpoints"].append({
+            "id" : endpoint_id,
+            "type" : "STD_MULTITARGET_PREDICTION",
+            "modelRef": saved_model_id
+        })
+
     def add_clustering_endpoint(self, endpoint_id, saved_model_id):
         """
         Add a new "visual clustering" endpoint to this API service.