dataiku-api-client 14.6.1__tar.gz → 14.7.0__tar.gz

{dataiku_api_client-14.6.1 → dataiku_api_client-14.7.0}/HISTORY.txt

@@ -2,6 +2,16 @@ Changelog
 ==========
 
 
+14.7.0 (2026-06-18)
+-------------------
+
+* Initial release for DSS 14.7.0
+
+14.6.2 (2026-06-15)
+-------------------
+
+* Initial release for DSS 14.6.2
+
 14.6.1 (2026-06-05)
 -------------------
 

{dataiku_api_client-14.6.1/dataiku_api_client.egg-info → dataiku_api_client-14.7.0}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: dataiku-api-client
-Version: 14.6.1
+Version: 14.7.0
 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.6.1
+Download-URL: https://github.com/dataiku/dataiku-api-client-python/tarball/14.7.0
 Author: Dataiku
 Author-email: support@dataiku.com
 License: Apache Software License
@@ -36,6 +36,16 @@ Changelog
 ==========
 
 
+14.7.0 (2026-06-18)
+-------------------
+
+* Initial release for DSS 14.7.0
+
+14.6.2 (2026-06-15)
+-------------------
+
+* Initial release for DSS 14.6.2
+
 14.6.1 (2026-06-05)
 -------------------
 

{dataiku_api_client-14.6.1 → dataiku_api_client-14.7.0/dataiku_api_client.egg-info}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: dataiku-api-client
-Version: 14.6.1
+Version: 14.7.0
 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.6.1
+Download-URL: https://github.com/dataiku/dataiku-api-client-python/tarball/14.7.0
 Author: Dataiku
 Author-email: support@dataiku.com
 License: Apache Software License
@@ -36,6 +36,16 @@ Changelog
 ==========
 
 
+14.7.0 (2026-06-18)
+-------------------
+
+* Initial release for DSS 14.7.0
+
+14.6.2 (2026-06-15)
+-------------------
+
+* Initial release for DSS 14.6.2
+
 14.6.1 (2026-06-05)
 -------------------
 

{dataiku_api_client-14.6.1 → dataiku_api_client-14.7.0}/dataiku_api_client.egg-info/SOURCES.txt

@@ -32,6 +32,7 @@ dataikuapi/dss/apideployer.py
 dataikuapi/dss/apiservice.py
 dataikuapi/dss/app.py
 dataikuapi/dss/businessapp.py
+dataikuapi/dss/cobuild.py
 dataikuapi/dss/codestudio.py
 dataikuapi/dss/continuousactivity.py
 dataikuapi/dss/dashboard.py

{dataiku_api_client-14.6.1 → dataiku_api_client-14.7.0}/dataikuapi/dss/admin.py

@@ -27,7 +27,7 @@ class DSSConnectionListItem(dict):
 
         :rtype: :class:`DSSConnection`
         """
-        return DSSConnection(self.client, self["name"])
+        return DSSConnection(self.client, self.name)
 
     @property
     def name(self):
@@ -36,6 +36,8 @@ class DSSConnectionListItem(dict):
 
         :rtype: string
         """
+        if "name" in self:
+            return self["name"]
         return self["id"]
    
     @property
@@ -46,6 +48,8 @@ class DSSConnectionListItem(dict):
         :return: a DSS connection type, like PostgreSQL, EC2, Azure, ...
         :rtype: string
         """
+        if "type" in self:
+            return self["type"]
         return self["label"]
 
 class DSSConnectionInfo(dict):
@@ -702,6 +706,11 @@ class DSSUser(object):
                                          extra_headers={"X-DKU-ProxyUser":  self.login}, client_certificate=self.client._session.cert)
             client_as._session.verify = self.client._session.verify
             return client_as
+        elif self.client.jwt_bearer_token is not None:
+            client_as = DSSClient(self.client.host, jwt_bearer_token = self.client.jwt_bearer_token,
+                                         extra_headers={"X-DKU-ProxyUser":  self.login}, client_certificate=self.client._session.cert)
+            client_as._session.verify = self.client._session.verify
+            return client_as
         else:
             raise ValueError("Don't know how to proxy this client")
 

dataiku_api_client-14.7.0/dataikuapi/dss/cobuild.py

@@ -0,0 +1,317 @@
+from dataikuapi.dss.dataset import DSSDataset
+from dataikuapi.dss.evaluationstore import DSSEvaluationStore
+from dataikuapi.dss.knowledgebank import DSSKnowledgeBank
+from dataikuapi.dss.labeling_task import DSSLabelingTask
+from dataikuapi.dss.managedfolder import DSSManagedFolder
+from dataikuapi.dss.recipe import DSSRecipe
+from dataikuapi.dss.savedmodel import DSSSavedModel
+from dataikuapi.dss.streaming_endpoint import DSSStreamingEndpoint
+
+
+class CobuildMessage(object):
+    def __init__(self, raw):
+        self._raw = raw
+
+    @property
+    def message(self):
+        """
+        The text content of the response.
+
+        :rtype: str or None
+        """
+        return self._raw.get("message")
+
+
+class CobuildAssistantResponse(CobuildMessage):
+    """
+    A response from the Cobuild AI assistant.
+
+    .. important::
+        Do not create this class directly, it is returned by :meth:`DSSCobuildConversation.send_message`
+        and :meth:`DSSCobuildConversation.answer_confirmation`.
+    """
+
+    @property
+    def role(self):
+        """
+        Role of the message sender ("user" or "assistant")
+
+        :rtype: str
+        """
+        return "assistant"
+
+    @property
+    def type(self):
+        """
+        Type of the response: ``"assistant_message"``, ``"delete_confirmation_request"``,
+        or ``"error"``.
+
+        - ``"assistant_message"``: the assistant has completed its turn; the conversation is idle.
+        - ``"delete_confirmation_request"``: the assistant is requesting confirmation before
+          deleting objects. Call :meth:`DSSCobuildConversation.answer_confirmation`
+          with a choice of ``"APPROVE"`` or ``"CANCEL"`` to continue.
+        - ``"error"``: an error occurred while processing the Cobuild turn. In the public API,
+          missing tool permissions are reported through this error response too.
+
+        :rtype: str
+        """
+        return self._raw.get("type")
+
+    @property
+    def is_question(self):
+        """
+        DEPRECATED ... DEPRECATED ... DEPRECATED ... DEPRECATED
+        Whether the assistant is asking the user a question. If ``True``, call
+        :meth:`DSSCobuildConversation.send_message` again with the answer to continue.
+
+        :rtype: bool
+        """
+        return False
+
+    @property
+    def is_confirmation_request(self):
+        """
+        Whether the assistant is requesting confirmation before deleting objects
+        or doing other operations requiring confirmation.
+        If ``True``, call :meth:`DSSCobuildConversation.answer_confirmation` with a choice of
+        ``"APPROVE"`` or ``"CANCEL"`` to continue.
+        If the confirmation is about a deletion, inspect :attr:`objects_to_delete`
+        and :attr:`deletion_impacts` for details.
+
+        :rtype: bool
+        """
+        return self.type == "delete_confirmation_request"
+
+    @property
+    def is_error(self):
+        """
+        Whether the response represents an error.
+
+        :rtype: bool
+        """
+        return self._raw.get("error", False)
+
+    @property
+    def objects_to_delete(self):
+        """
+        For ``"delete_confirmation_request"`` responses: the list of objects that the assistant
+        is requesting permission to delete (or unshare). Each entry is a dict with fields
+        ``projectKey``, ``type``, ``id``, and ``displayName``.
+
+        ``None`` for other response types.
+
+        :rtype: list of dict or None
+        """
+        return self._raw.get("objectsToDelete")
+
+    @property
+    def deletion_impacts(self):
+        """
+        For ``"delete_confirmation_request"`` responses: a dict describing the cascading effects
+        of the requested deletion (recipes that would be deleted, datasets left unchanged, etc.).
+
+        ``None`` for other response types.
+
+        :rtype: dict or None
+        """
+        return self._raw.get("deletionImpacts")
+
+    def __repr__(self):
+        return "CobuildAssistantResponse(type=%r, message=%r)" % (self.type, self.message)
+
+
+class CobuildUserMessage(CobuildMessage):
+    """
+    A message sent to Cobuild AI assistant by the user.
+
+    .. important::
+        Do not create this class directly, it is automatically created when using :meth:`dataikuapi.dss.project.DSSProject.new_cobuild_conversation`,
+        :meth:`DSSCobuildConversation.send_message`, and
+        :meth:`DSSCobuildConversation.answer_confirmation`.
+    """
+
+    @property
+    def role(self):
+        """
+        Role of the message sender ("user" or "assistant")
+
+        :rtype: str
+        """
+        return "user"
+
+    @property
+    def type(self):
+        """
+        Type of the message: ``"request"`` or ``"delete_confirmation_response"``.
+
+        - ``"request"``: a normal message sent to cobuild.
+        - ``"delete_confirmation_response"``: user response when cobuild requests confirmation before deleting objects
+
+        :rtype: str
+        """
+        return self._raw.get("type", "request")
+
+    @property
+    def is_confirmation_response(self):
+        """
+        Whether this message is the answer for assistant requesting confirmation before deleting objects
+
+        :rtype: bool
+        """
+        return self.type == "delete_confirmation_response"
+
+    def __repr__(self):
+        return "CobuildUserMessage(type=%r, message=%r)" % (self.type, self.message)
+
+def _DSS_objects_to_selected(project_key, objects):
+    return [_DSS_object_to_selected(project_key, object) for object in objects]
+
+def _DSS_object_to_selected(project_key, object):
+    prefix = "" if project_key == object.project_key else object.project_key + "."
+
+    if isinstance(object, DSSDataset):
+        return { "type": "DATASET", "id": prefix + object.id}
+    if isinstance(object, DSSRecipe):
+        return { "type": "RECIPE", "id": prefix + object.id}
+    if isinstance(object, DSSManagedFolder):
+        return { "type": "MANAGED_FOLDER", "id": prefix + object.id}
+    if isinstance(object, DSSSavedModel):
+        return { "type": "SAVED_MODEL", "id": prefix + object.id}
+    if isinstance(object, DSSStreamingEndpoint):
+        return { "type": "STREAMING_ENDPOINT", "id": prefix + object.id}
+    if isinstance(object, DSSLabelingTask):
+        return { "type": "LABELING_TASK", "id": prefix + object.id}
+    if isinstance(object, DSSKnowledgeBank):
+        return { "type": "RETRIEVABLE_KNOWLEDGE", "id": prefix + object.id}
+    if isinstance(object, DSSEvaluationStore):
+        return { "type": "MODEL_EVALUATION_STORE", "id": prefix + object.id}
+    else:
+        raise ValueError("Unsupported object type")
+
+
+class DSSCobuildConversation(object):
+    """
+    A handle to a conversation with Cobuild AI assistant.
+
+    .. important::
+        Do not create this class directly, instead use
+        :meth:`dataikuapi.dss.project.DSSProject.new_cobuild_conversation`.
+
+    The :attr:`messages` property accumulates all exchanges made through this handle (user inputs
+    and assistant responses).
+
+    A typical interaction::
+
+        conv = project.new_cobuild_conversation()
+        response = conv.send_message("List the datasets in this project")
+        print(response.message)
+
+        conv.send_message("Now filter the Orders dataset to keep only orders with an amount > 1000")
+        print(conv.messages[-1].message)
+
+    When the assistant needs to delete objects, it first asks for confirmation::
+
+        response = conv.send_message("Delete the Orders dataset")
+        if response.is_confirmation_request:
+            print("Objects to delete:", response.objects_to_delete)
+            response = conv.answer_confirmation("APPROVE")
+            print(response.message)
+
+    If a tool requires edit permission, pass ``allow_edit_project=True`` to
+    :meth:`send_message` to allow Cobuild to create and edit objects for that message.
+    """
+
+    def __init__(self, client, project_key, conversation_id, selected_objects=None):
+        self.client = client
+        self.project_key = project_key
+        self.conversation_id = conversation_id
+        self._messages = []
+        self._pending_confirmation_id = None
+        self._selected_objects = selected_objects
+
+    @property
+    def messages(self):
+        """
+        All messages exchanged during this conversation, in order.
+
+        User message entries are :class:`CobuildUserMessage`.
+
+        Assistant message entries are :class:`CobuildAssistantResponse`.
+
+        :rtype: list of :class:`CobuildAssistantResponse` or :class:`CobuildUserMessage`
+        """
+        return list(self._messages)
+
+    def send_message(self, message, selected_objects=None, allow_edit_project=False):
+        """
+        Send a message to the assistant and wait for its response.
+
+        The message and assistant response are appended to :attr:`messages`.
+
+        :param str message: the message to send
+        :param selected_objects: object selection the assistant should focus on. It is reused for any subsequent message, unless overwritten
+        :type selected_objects: list[:class:`.DSSDataset`, :class:`.DSSRecipe`, :class:`.DSSLabelingTask`, :class:`.DSSManagedFolder`, :class:`.DSSSavedModel`, :class:`.DSSKnowledgeBank`, :class:`.DSSModelEvaluationStore` or :class:`.DSSStreamingEndpoint`]
+        :param bool allow_edit_project: whether to allow Cobuild to create and edit everything
+            needed in this project to follow this message. This permission applies only to this
+            message.
+
+        :returns: the assistant's response
+        :rtype: :class:`CobuildAssistantResponse`
+        """
+        if selected_objects is not None:
+            self._selected_objects = _DSS_objects_to_selected(self.project_key, selected_objects)
+
+        self._messages.append(CobuildUserMessage({"type": "request", "message": message, "selected_objects": self._selected_objects}))
+        raw = self.client._perform_json(
+            "POST",
+            "/projects/%s/cobuild/conversations/%s/messages" % (self.project_key, self.conversation_id),
+            body={
+                "message": message,
+                "selectedObjects": self._selected_objects or [],
+                "allowEditProject": allow_edit_project,
+            },
+        )
+        response = CobuildAssistantResponse(raw)
+        self._pending_confirmation_id = raw["confirmationId"] if response.is_confirmation_request else None
+        self._messages.append(response)
+        return response
+
+    def answer_confirmation(self, choice, options=None):
+        """
+        Answer a pending confirmation request and wait for the assistant's next response.
+
+        Call this after receiving a response with :attr:`~CobuildAssistantResponse.is_confirmation_request`
+        set to ``True``.
+
+        The choice and assistant response are appended to :attr:`messages`.
+
+        :param str choice: ``"APPROVE"`` to proceed with the operation, or ``"CANCEL"`` to abort it
+        :param list options: optional list of per-object deletion options (advanced use, for delete
+            confirmations only). Each entry is a dict with ``projectKey``, ``type``, ``id``, and an
+            ``options`` dict containing ``dropData``, ``dropMetastoreTable``, and
+            ``deleteOrphanInsights`` booleans. When omitted, all options default to ``False``
+            (data is not dropped, metastore tables are not dropped, orphan insights are not deleted).
+
+        :returns: the assistant's response after the confirmation
+        :rtype: :class:`CobuildAssistantResponse`
+        """
+        if choice not in ("APPROVE", "CANCEL"):
+            raise ValueError("choice must be 'APPROVE' or 'CANCEL', got %r" % choice)
+        if self._pending_confirmation_id is None:
+            raise ValueError("No pending confirmation request. Call send_message first and check is_confirmation_request.")
+
+        confirmation_id = self._pending_confirmation_id
+        self._pending_confirmation_id = None
+        self._messages.append(CobuildUserMessage({"type": "delete_confirmation_response", "message": choice}))
+        body = {"choice": choice}
+        if options is not None:
+            body["options"] = options
+        raw = self.client._perform_json(
+            "POST",
+            "/projects/%s/cobuild/conversations/%s/confirmation/%s" % (self.project_key, self.conversation_id, confirmation_id),
+            body=body,
+        )
+        response = CobuildAssistantResponse(raw)
+        self._pending_confirmation_id = raw["confirmationId"] if response.is_confirmation_request else None
+        self._messages.append(response)
+        return response

{dataiku_api_client-14.6.1 → dataiku_api_client-14.7.0}/dataikuapi/dss/llm_tracing/__init__.py

@@ -97,6 +97,7 @@ class SpanBuilder:
     @staticmethod
     def create_event(name):
         sb = SpanBuilder(name)
+        sb.span["type"] = "event"
         ts = int(time.time() * 1000)
         sb.span["timestamp"] = datetime.datetime.fromtimestamp(ts / 1000, datetime.timezone.utc).strftime("%Y-%m-%dT%H:%M:%S.%fZ")
         return sb

{dataiku_api_client-14.6.1 → dataiku_api_client-14.7.0}/dataikuapi/dss/messaging_channel.py

@@ -285,9 +285,9 @@ class DSSMailMessagingChannel(DSSMessagingChannel):
 
             channel = client.get_messaging_channel("other-mail-channel-id")
             for file in paths:
-            with open(file) as f:
-                # Optionally include file type ("text/csv")
-                attachments.append(file, f.read(), "text/csv")
+                with open(file) as f:
+                    # Optionally include file type ("text/csv")
+                    attachments.append([file, f.read(), "text/csv"])
             channel.send("PROJECT_KEY", ["joe@dataiku.com"], "Subject", "Body in plain text",  attachments=attachments, False)
 
         :param project_key: project issuing the email. The user must have "Write content" permission on the specified project.
@@ -299,7 +299,7 @@ class DSSMailMessagingChannel(DSSMessagingChannel):
         :param body: email body (in plain text or HTML format)
         :type body: str
         :param attachments: files to be attached to the mail, defaults to None
-        :type attachments: list[BufferedReader]
+        :type attachments: list[list[str, bytes, str]]
         :param plain_text: True to send email as plain text, False to send it as HTML. Defaults to False.
         :type plain_text: bool
         :param sender: sender email address. Use None to use the sender defined at the channel level.

{dataiku_api_client-14.6.1 → dataiku_api_client-14.7.0}/dataikuapi/dss/ml.py

@@ -3433,6 +3433,7 @@ class DSSMLAssertionCondition(object):
 class DSSMLAssertionsMetrics(object):
     """
     Object that represents the assertions metrics for all assertions on a trained model
+
     .. important::
         Do not create this object directly, use :meth:`DSSTrainedPredictionModelDetails.get_assertions_metrics` instead
     """
@@ -4129,7 +4130,7 @@ class DSSTrainedPredictionModelDetails(DSSTrainedModelDetails):
 
 class DSSTrainedTimeseriesForecastingModelDetails(DSSTrainedPredictionModelDetails):
     """
-    Object to read details of a timeseries forecasting model, for instance the per time series metrics
+    Object to read details of a time series forecasting model, for instance the per time series metrics
 
     .. important::
         Do not create this object directly, use :meth:`DSSMLTask.get_trained_model_details()` instead
@@ -4139,10 +4140,10 @@ class DSSTrainedTimeseriesForecastingModelDetails(DSSTrainedPredictionModelDetai
     
     def compute_residuals(self, wait=True):
         """
-        Launch computation of residuals for this trained timeseries model.
+        Launch computation of residuals for this trained time series model.
 
         :param wait: a flag to wait for the operation to complete (defaults to **True**)
-        :returns: if wait is True, a dictionary containing the residuals per-timeseries, else a future to wait on the result
+        :returns: if wait is True, a dictionary containing the residuals per time series, else a future to wait on the result
         :rtype: Union[:class:`dict`, :class:`dataikuapi.dss.future.DSSFuture`]
         """
         if self.mltask is not None:
@@ -4166,7 +4167,7 @@ class DSSTrainedTimeseriesForecastingModelDetails(DSSTrainedPredictionModelDetai
         """
         Retrieve a list of residuals for this trained time-series models
 
-        :returns: A dictionary, which contains a residuals object per-timeseries
+        :returns: A dictionary, which contains a residuals object per time series
         :rtype: dict
         """
         if self.mltask is not None:
@@ -4184,7 +4185,7 @@ class DSSTrainedTimeseriesForecastingModelDetails(DSSTrainedPredictionModelDetai
 
     def compute_permutation_importance(self, wait=True, n_iterations=None, per_identifier=None):
         """
-        Launch computation of permutation importance for this trained timeseries model.
+        Launch feature importance computation (measured using permutation importance) for this trained time series model.
 
         :param wait: a flag to wait for the operation to complete (defaults to **True**)
         :returns: if wait is True, a dictionary, else a future to wait on the result
@@ -4238,7 +4239,7 @@ class DSSTrainedTimeseriesForecastingModelDetails(DSSTrainedPredictionModelDetai
 
     def get_per_timeseries_metrics(self):
             """
-            Returns per timeseries performance metrics for this model.
+            Returns per time series performance metrics for this model.
 
             :returns: a dict of performance metrics values
             :rtype: dict
@@ -4257,9 +4258,9 @@ class DSSTrainedTimeseriesForecastingModelDetails(DSSTrainedPredictionModelDetai
 
     def get_per_timeseries_evaluation_forecasts(self):
         """
-        Returns per timeseries evaluation forecasts for this model.
+        Returns per time series evaluation forecasts for this model.
 
-        :returns: a dict of evaluation forecasts per timeseries identifier
+        :returns: a dict of evaluation forecasts per time series identifier
         :rtype: dict
         """
         if self.mltask is not None: