dataiku-api-client 14.7.1__tar.gz → 14.7.3__tar.gz

{dataiku_api_client-14.7.1 → dataiku_api_client-14.7.3}/HISTORY.txt

@@ -2,6 +2,16 @@ Changelog
 ==========
 
 
+14.7.3 (2026-08-03)
+-------------------
+
+* Initial release for DSS 14.7.3
+
+14.7.2 (2026-07-13)
+-------------------
+
+* Initial release for DSS 14.7.2
+
 14.7.1 (2026-07-13)
 -------------------
 

{dataiku_api_client-14.7.1/dataiku_api_client.egg-info → dataiku_api_client-14.7.3}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: dataiku-api-client
-Version: 14.7.1
+Version: 14.7.3
 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.1
+Download-URL: https://github.com/dataiku/dataiku-api-client-python/tarball/14.7.3
 Author: Dataiku
 Author-email: support@dataiku.com
 License: Apache Software License
@@ -36,6 +36,16 @@ Changelog
 ==========
 
 
+14.7.3 (2026-08-03)
+-------------------
+
+* Initial release for DSS 14.7.3
+
+14.7.2 (2026-07-13)
+-------------------
+
+* Initial release for DSS 14.7.2
+
 14.7.1 (2026-07-13)
 -------------------
 

{dataiku_api_client-14.7.1 → dataiku_api_client-14.7.3/dataiku_api_client.egg-info}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: dataiku-api-client
-Version: 14.7.1
+Version: 14.7.3
 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.1
+Download-URL: https://github.com/dataiku/dataiku-api-client-python/tarball/14.7.3
 Author: Dataiku
 Author-email: support@dataiku.com
 License: Apache Software License
@@ -36,6 +36,16 @@ Changelog
 ==========
 
 
+14.7.3 (2026-08-03)
+-------------------
+
+* Initial release for DSS 14.7.3
+
+14.7.2 (2026-07-13)
+-------------------
+
+* Initial release for DSS 14.7.2
+
 14.7.1 (2026-07-13)
 -------------------
 

{dataiku_api_client-14.7.1 → dataiku_api_client-14.7.3}/dataiku_api_client.egg-info/SOURCES.txt

@@ -61,6 +61,7 @@ dataikuapi/dss/managedfolder.py
 dataikuapi/dss/meaning.py
 dataikuapi/dss/messaging_channel.py
 dataikuapi/dss/metrics.py
+dataikuapi/dss/mira.py
 dataikuapi/dss/ml.py
 dataikuapi/dss/mlflow.py
 dataikuapi/dss/modelcomparison.py

{dataiku_api_client-14.7.1 → dataiku_api_client-14.7.3}/dataikuapi/dss/agent.py

@@ -354,6 +354,51 @@ class DSSAgent(object):
                 "versionId": version_id
             })
 
+    def get_metrics_series(self, from_timestamp_ms=None, to_timestamp_ms=None, aggregation="MINUTE", timezone=None):
+        """
+        Get the operational metrics series for this agent.
+
+        The returned payload may include the ongoing interval for the requested granularity.
+        As a consequence, the latest datapoint is temporarily inconsistent and may evolve
+        as more raw events are flushed and aggregated at read time.
+        The requested time window is aligned to the bucket boundaries of the selected
+        aggregation before being read.
+
+        :param int from_timestamp_ms: Beginning of the requested window, inclusive, as an
+            epoch timestamp in milliseconds. The effective lower bound is rounded down
+            to the start of its bucket. Optional, defaults to the oldest retained
+            timestamp available for the requested aggregation.
+        :param int to_timestamp_ms: End of the requested window, exclusive, as an epoch
+            timestamp in milliseconds. The effective upper bound is rounded up to the
+            next bucket boundary when it falls inside a bucket.
+            Optional, defaults to the current time when omitted.
+        :param str aggregation: Aggregation granularity. Supported values are ``MINUTE``,
+            ``FIVE_MINUTES``, ``HOUR``, ``DAY`` and ``MONTH``.
+        :param str timezone: Timezone used to align bucket boundaries. Optional,
+            defaults to ``UTC``. Can be a timezone name like ``Europe/Paris``.
+        :return: The list of datapoints. Each datapoint contains a ``timestampMs``
+                 expressed as the start timestamp of its bucket in epoch
+                 milliseconds. For example, with ``HOUR`` aggregation, a datapoint
+                 at ``18:00`` represents the interval ``[18:00, 19:00)``.
+        :rtype: list[dict]
+        """
+        if from_timestamp_ms is not None and not isinstance(from_timestamp_ms, int):
+            raise TypeError("Expected int for from_timestamp_ms, got %s" % type(from_timestamp_ms).__name__)
+        if to_timestamp_ms is not None and not isinstance(to_timestamp_ms, int):
+            raise TypeError("Expected int for to_timestamp_ms, got %s" % type(to_timestamp_ms).__name__)
+        if timezone is not None and not isinstance(timezone, str):
+            raise TypeError("Expected str for timezone, got %s" % type(timezone).__name__)
+
+        return self.client._perform_json(
+            "GET", "/projects/%s/agents/%s/operational-metrics/series" % (self.project_key, self.id),
+            params={
+                "fromTimestampMs": from_timestamp_ms,
+                "toTimestampMs": to_timestamp_ms,
+                "aggregation": aggregation,
+                "timezone": timezone
+            }
+        )
+
 
 class DSSAgentSettings(DSSTaggableObjectSettings):
     """

{dataiku_api_client-14.7.1 → dataiku_api_client-14.7.3}/dataikuapi/dss/agent_tool.py

@@ -64,7 +64,7 @@ class DSSAgentTool(object):
         """
         return self.tool_id
 
-    def get_descriptor(self):
+    def get_descriptor(self, context=None):
         """
         Get the descriptor of the tool
 
@@ -73,7 +73,10 @@ class DSSAgentTool(object):
         """
 
         if self._descriptor is None:
-            self._descriptor = self.client._perform_json("GET", "/projects/%s/agents/tools/%s/descriptor" % (self.project_key, self.tool_id))
+            if context is None:
+                self._descriptor = self.client._perform_json("GET", "/projects/%s/agents/tools/%s/descriptor" % (self.project_key, self.tool_id))
+            else:
+                self._descriptor = self.client._perform_json("POST", "/projects/%s/agents/tools/%s/descriptor" % (self.project_key, self.tool_id), body={"context": context})
         return self._descriptor
 
     def get_settings(self):

{dataiku_api_client-14.7.1 → dataiku_api_client-14.7.3}/dataikuapi/dss/cobuild.py

@@ -28,7 +28,8 @@ class CobuildAssistantResponse(CobuildMessage):
 
     .. important::
         Do not create this class directly, it is returned by :meth:`DSSCobuildConversation.send_message`
-        and :meth:`DSSCobuildConversation.answer_confirmation`.
+        :meth:`DSSCobuildConversation.answer_confirmation`, and
+        :meth:`DSSCobuildConversation.answer_question`.
     """
 
     @property
@@ -44,12 +45,14 @@ class CobuildAssistantResponse(CobuildMessage):
     def type(self):
         """
         Type of the response: ``"assistant_message"``, ``"delete_confirmation_request"``,
-        or ``"error"``.
+        ``"ask_question_to_user_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.
+        - ``"ask_question_to_user_request"``: the assistant is requesting an explicit answer before it can
+          continue. Call :meth:`DSSCobuildConversation.answer_question` 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.
 
@@ -82,6 +85,16 @@ class CobuildAssistantResponse(CobuildMessage):
         """
         return self.type == "delete_confirmation_request"
 
+    @property
+    def is_question_request(self):
+        """
+        Whether the assistant is requesting an explicit answer before continuing.
+        If ``True``, call :meth:`DSSCobuildConversation.answer_question`.
+
+        :rtype: bool
+        """
+        return self.type == "ask_question_to_user_request"
+
     @property
     def is_error(self):
         """
@@ -116,6 +129,72 @@ class CobuildAssistantResponse(CobuildMessage):
         """
         return self._raw.get("deletionImpacts")
 
+    @property
+    def question_id(self):
+        """
+        For ``"ask_question_to_user_request"`` responses: the identifier of the pending question.
+
+        ``None`` for other response types.
+
+        :rtype: str or None
+        """
+        return self._raw.get("questionId")
+
+    @property
+    def title(self):
+        """
+        For ``"ask_question_to_user_request"`` responses: the short question shown to the user.
+
+        ``None`` for other response types.
+
+        :rtype: str or None
+        """
+        return self._raw.get("title")
+
+    @property
+    def predefined_answers(self):
+        """
+        For ``"ask_question_to_user_request"`` responses: the list of predefined answers proposed by Cobuild.
+
+        ``None`` for other response types.
+
+        :rtype: list[str] or None
+        """
+        return self._raw.get("predefinedAnswers")
+
+    @property
+    def allow_custom_answer(self):
+        """
+        For ``"ask_question_to_user_request"`` responses: whether a custom free-text answer is allowed.
+
+        ``None`` for other response types.
+
+        :rtype: bool or None
+        """
+        return self._raw.get("allowCustomAnswer")
+
+    @property
+    def allow_multiple_answers(self):
+        """
+        For ``"ask_question_to_user_request"`` responses: whether multiple answers can be selected.
+
+        ``None`` for other response types.
+
+        :rtype: bool or None
+        """
+        return self._raw.get("allowMultipleAnswers")
+
+    @property
+    def default_answer_set(self):
+        """
+        For ``"ask_question_to_user_request"`` responses: whether Cobuild suggested a default answer.
+
+        ``None`` for other response types.
+
+        :rtype: bool or None
+        """
+        return self._raw.get("selectFirstAnswerByDefault")
+
     def __repr__(self):
         return "CobuildAssistantResponse(type=%r, message=%r)" % (self.type, self.message)
 
@@ -127,7 +206,8 @@ class CobuildUserMessage(CobuildMessage):
     .. 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`.
+        :meth:`DSSCobuildConversation.answer_confirmation`, and
+        :meth:`DSSCobuildConversation.answer_question`.
     """
 
     @property
@@ -142,10 +222,12 @@ class CobuildUserMessage(CobuildMessage):
     @property
     def type(self):
         """
-        Type of the message: ``"request"`` or ``"delete_confirmation_response"``.
+        Type of the message: ``"request"``, ``"delete_confirmation_response"``, or
+        ``"ask_question_to_user_response"``.
 
         - ``"request"``: a normal message sent to cobuild.
         - ``"delete_confirmation_response"``: user response when cobuild requests confirmation before deleting objects
+        - ``"ask_question_to_user_response"``: user response when cobuild requests an explicit answer
 
         :rtype: str
         """
@@ -160,6 +242,15 @@ class CobuildUserMessage(CobuildMessage):
         """
         return self.type == "delete_confirmation_response"
 
+    @property
+    def is_question_response(self):
+        """
+        Whether this message is the answer for assistant requesting an explicit answer.
+
+        :rtype: bool
+        """
+        return self.type == "ask_question_to_user_response"
+
     def __repr__(self):
         return "CobuildUserMessage(type=%r, message=%r)" % (self.type, self.message)
 
@@ -217,6 +308,31 @@ class DSSCobuildConversation(object):
             response = conv.answer_confirmation("APPROVE")
             print(response.message)
 
+    When the assistant needs an explicit answer, it asks a question::
+
+        response = conv.send_message("Use the best date column for sorting")
+        if response.is_question_request:
+            print("Question:", response.title)
+            print("Choices:", response.predefined_answers)
+            print("Allow custom answer:", response.allow_custom_answer)
+            response = conv.answer_question(
+                answers=["OrderDate"],
+                rejected=False,
+                used_custom_answer=False
+            )
+            print(response.message)
+
+    You can also decline answering a question::
+
+        response = conv.send_message("Pick the dataset to build")
+        if response.is_question_request:
+            response = conv.answer_question(
+                answers=[],
+                rejected=True,
+                used_custom_answer=False
+            )
+            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.
     """
@@ -227,6 +343,7 @@ class DSSCobuildConversation(object):
         self.conversation_id = conversation_id
         self._messages = []
         self._pending_confirmation_id = None
+        self._pending_question_id = None
         self._selected_objects = selected_objects
 
     @property
@@ -273,6 +390,7 @@ class DSSCobuildConversation(object):
         )
         response = CobuildAssistantResponse(raw)
         self._pending_confirmation_id = raw["confirmationId"] if response.is_confirmation_request else None
+        self._pending_question_id = raw["questionId"] if response.is_question_request else None
         self._messages.append(response)
         return response
 
@@ -313,5 +431,51 @@ class DSSCobuildConversation(object):
         )
         response = CobuildAssistantResponse(raw)
         self._pending_confirmation_id = raw["confirmationId"] if response.is_confirmation_request else None
+        self._pending_question_id = raw["questionId"] if response.is_question_request else None
+        self._messages.append(response)
+        return response
+
+    def answer_question(self, answers=None, rejected=False, used_custom_answer=False):
+        """
+        Answer a pending question request and wait for the assistant's next response.
+
+        Call this after receiving a response with :attr:`~CobuildAssistantResponse.is_question_request`
+        set to ``True``.
+
+        The answer and assistant response are appended to :attr:`messages`.
+
+        :param list[str] answers: answers selected or entered by the user. Use an empty list when
+            ``rejected=True``. Defaults to ``[]``.
+        :param bool rejected: whether to decline answering the question
+        :param bool used_custom_answer: whether one of the answers came from the custom free-text input
+
+        :returns: the assistant's response after the question answer
+        :rtype: :class:`CobuildAssistantResponse`
+        """
+        if answers is None:
+            answers = []
+        if not isinstance(answers, list):
+            raise ValueError("answers must be a list of strings")
+        if self._pending_question_id is None:
+            raise ValueError("No pending question request. Call send_message first and check is_question_request.")
+
+        question_id = self._pending_question_id
+        self._pending_question_id = None
+        self._messages.append(CobuildUserMessage({
+            "type": "ask_question_to_user_response",
+            "message": None if rejected else ", ".join(answers),
+        }))
+        raw = self.client._perform_json(
+            "POST",
+            "/projects/%s/cobuild/conversations/%s/question/%s" % (self.project_key, self.conversation_id, question_id),
+            body={
+                "rejected": rejected,
+                "answers": answers,
+                "usedCustomAnswer": used_custom_answer,
+            },
+        )
+        response = CobuildAssistantResponse(raw)
+        self._pending_confirmation_id = raw["confirmationId"] if response.is_confirmation_request else None
+        self._pending_question_id = raw["questionId"] if response.is_question_request else None
         self._messages.append(response)
         return response

{dataiku_api_client-14.7.1 → dataiku_api_client-14.7.3}/dataikuapi/dss/knowledgebank.py

@@ -115,6 +115,16 @@ class DSSKnowledgeBank(object):
         """
         return self.client._perform_empty("DELETE", "/projects/%s/knowledge-banks/%s" % (self.project_key, self.id))
 
+    def clear(self):
+        """
+        Clear data in this knowledge bank.
+
+        :returns: a dict containing the method call status.
+        :rtype: dict
+        """
+        return self.client._perform_json(
+            "POST", "/projects/%s/knowledge-banks/%s/clear" % (self.project_key, self.id))
+
     def build(self, job_type="NON_RECURSIVE_FORCED_BUILD", wait=True):
         """
         Start a new job to build this knowledge bank and wait for it to complete.
@@ -449,18 +459,17 @@ class DSSKnowledgeBankSearchResultDocument(object):
 
         folder_smart_id = source_file_info.get("folder_ref")
         folder_full_id = source_file_info.get("folder_full_id") # deprecated but to support existing KBs
-        folder_loc = None
 
         if folder_smart_id is not None:
             try:
-                folder_loc = AnyLoc.from_ref("donotmatter", folder_smart_id) # unused, just to check format
+                _ = AnyLoc.from_ref("donotmatter", folder_smart_id) # unused, just to check format
                 return ManagedFolderDocumentRef(path, folder_smart_id)
             except ValueError as e:
                 logger.error("Invalid folder_ref in DKU_DOCUMENT_INFO: {}, {}".format(e, document_info))
                 return None
         elif folder_full_id is not None:
             try:
-                folder_loc = AnyLoc.from_full(folder_full_id) # unused, just to check format
+                _ = AnyLoc.from_full(folder_full_id) # unused, just to check format
                 return ManagedFolderDocumentRef(path, folder_full_id)
             except ValueError as e:
                 logger.error("Invalid folder_full_id in DKU_DOCUMENT_INFO: {}, {}".format(e, document_info))

{dataiku_api_client-14.7.1 → dataiku_api_client-14.7.3}/dataikuapi/dss/langchain/embeddings.py

@@ -2,16 +2,25 @@
 import asyncio
 import concurrent
 import logging
-import threading 
+import threading
+import itertools
 
 from typing import Callable, List, Any, Union
 
 import pydantic
+
+_thread_pool_executor_counter = itertools.count().__next__
+
+def next_thread_pool_executor_prefix(prefix):
+    return "{}-{}".format(prefix, _thread_pool_executor_counter())
+
 try:
     from langchain_core.embeddings.embeddings import Embeddings
 except ModuleNotFoundError:
     from langchain.embeddings.base import Embeddings
 from langchain_core.callbacks import BaseCallbackHandler, LLMManagerMixin
+
+
 from dataikuapi.dss.llm_tracing import new_trace, SpanBuilder
 
 from dataikuapi.dss.langchain.utils import must_use_deprecated_pydantic_config
@@ -121,7 +130,7 @@ class DKUEmbeddings(LockedDownBaseModel, Embeddings):
 
     async def aembed_documents(self, texts: List[str]) -> List[List[float]]:
         loop = asyncio.get_event_loop()
-        with concurrent.futures.ThreadPoolExecutor() as executor:
+        with concurrent.futures.ThreadPoolExecutor(thread_name_prefix=next_thread_pool_executor_prefix("DKUEmbeddingsAsyncExecutor")) as executor:
             result = await loop.run_in_executor(executor, self.embed_documents, texts)
         return result