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

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

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

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

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

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

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

{dataiku_api_client-14.7.2 → dataiku_api_client-14.7.4}/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.2 → dataiku_api_client-14.7.4}/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.2 → dataiku_api_client-14.7.4}/dataikuapi/dss/cobuild.py

@@ -435,7 +435,7 @@ class DSSCobuildConversation(object):
         self._messages.append(response)
         return response
 
-    def answer_question(self, answers=None, rejected=False, used_custom_answer=False):
+    def answer_question(self, answers=None, rejected=False, used_custom_answer=False, selected_objects=None):
         """
         Answer a pending question request and wait for the assistant's next response.
 
@@ -448,6 +448,9 @@ class DSSCobuildConversation(object):
             ``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
+        :param selected_objects: object selection the assistant should focus on. It is reused for
+            subsequent messages and question answers, unless overwritten
+        :type selected_objects: list[:class:`.DSSDataset`, :class:`.DSSRecipe`, :class:`.DSSLabelingTask`, :class:`.DSSManagedFolder`, :class:`.DSSSavedModel`, :class:`.DSSKnowledgeBank`, :class:`.DSSModelEvaluationStore` or :class:`.DSSStreamingEndpoint`]
 
         :returns: the assistant's response after the question answer
         :rtype: :class:`CobuildAssistantResponse`
@@ -458,6 +461,8 @@ class DSSCobuildConversation(object):
             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.")
+        if selected_objects is not None:
+            self._selected_objects = _DSS_objects_to_selected(self.project_key, selected_objects)
 
         question_id = self._pending_question_id
         self._pending_question_id = None
@@ -472,6 +477,7 @@ class DSSCobuildConversation(object):
                 "rejected": rejected,
                 "answers": answers,
                 "usedCustomAnswer": used_custom_answer,
+                "selectedObjects": self._selected_objects or [],
             },
         )
         response = CobuildAssistantResponse(raw)

{dataiku_api_client-14.7.2 → dataiku_api_client-14.7.4}/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
 

{dataiku_api_client-14.7.2 → dataiku_api_client-14.7.4}/dataikuapi/dss/llm.py

@@ -199,6 +199,8 @@ class DSSLLMEmbeddingsQuery(object):
     def new_guardrail(self, type):
         """
         Start adding a guardrail to the request. You need to configure the returned object, and call add() to actually add it
+
+        :rtype: :class:`DSSLLMRequestGuardrailBuilder`
         """
         return DSSLLMRequestGuardrailBuilder(self, type)
 
@@ -420,15 +422,27 @@ class SettingsMixin(object):
 
 
 class DSSLLMRequestGuardrailBuilder(object):
+    """
+      .. important::
+        Do not create this class directly, use :meth:`dataikuapi.dss.llm.DSSLLMCompletionQuery.new_guardrail`,
+        :meth:`dataikuapi.dss.llm.DSSLLMCompletionsQuery.new_guardrail`, :meth:`dataikuapi.dss.llm.DSSLLMEmbeddingsQuery.new_guardrail` or
+        :meth:`dataikuapi.dss.llm.DSSLLMImageGenerationQuery.new_guardrail`.
+    """
+
     def __init__(self, request, type):
         self.request = request
-        self.guardrail = { "type" : type, "enabled": True, "params" : {}}
+        self.guardrail = {"type" : type, "enabled": True, "params" : {}}
 
     @property
     def params(self):
+        """
+        :return: The parameters of this guardrail
+        :rtype: dict
+        """
         return self.guardrail["params"]
 
     def add(self):
+        """Add this guardrail to the completion query"""
         if self.request._guardrails is None:
             self.request._guardrails = {"guardrails" : []}
         self.request._guardrails["guardrails"].append(self.guardrail)
@@ -537,6 +551,8 @@ class DSSLLMCompletionQuery(DSSLLMCompletionsQuerySingleQuery, SettingsMixin):
     def new_guardrail(self, type):
         """
         Start adding a guardrail to the request. You need to configure the returned object, and call add() to actually add it
+
+        :rtype: :class:`DSSLLMRequestGuardrailBuilder`
         """
         return DSSLLMRequestGuardrailBuilder(self, type)
 
@@ -638,6 +654,8 @@ class DSSLLMCompletionsQuery(SettingsMixin):
     def new_guardrail(self, type):
         """
         Start adding a guardrail to the request. You need to configure the returned object, and call add() to actually add it
+
+        :rtype: :class:`DSSLLMRequestGuardrailBuilder`
         """
         return DSSLLMRequestGuardrailBuilder(self, type)
 
@@ -664,7 +682,7 @@ class DSSLLMCompletionsQuery(SettingsMixin):
         return DSSLLMCompletionsResponse(ret["responses"], response_parser=self._response_parser)
 
 
-class DSSLLMCompletionQueryMultipartBuilder(object):
+class _DSSLLMCompletionQueryMultipartBuilder(object):
     def __init__(self):
         self.parts = []
 
@@ -681,6 +699,8 @@ class DSSLLMCompletionQueryMultipartBuilder(object):
     def with_text(self, text):
         """
         Add a text part to the multipart message
+
+        :param str text: The text to add
         """
         self.parts.append({"type": "TEXT", "text": text})
         return self
@@ -692,7 +712,7 @@ class DSSLLMCompletionQueryMultipartBuilder(object):
         :param Union[str, bytes] image: The image
         :param str mime_type: None for default
         """
-        img_b64 = DSSLLMCompletionQueryMultipartMessage._encode_image(image)
+        img_b64 = _DSSLLMCompletionQueryMultipartBuilder._encode_image(image)
 
         part = {
             "type": "IMAGE_INLINE",
@@ -713,7 +733,7 @@ class DSSLLMCompletionQueryMultipartBuilder(object):
         :param Union[str, bytes] image: The image
         :param str mime_type: None for default
         """
-        img_b64 = DSSLLMCompletionQueryMultipartMessage._encode_image(image)
+        img_b64 = _DSSLLMCompletionQueryMultipartBuilder._encode_image(image)
 
         image_part = {
             "type": "IMAGE_INLINE",
@@ -736,14 +756,13 @@ class DSSLLMCompletionQueryMultipartBuilder(object):
         """
         Add an image url part to the multipart message
 
-        :param image: str the image url
+        :param str image: the image url
         """
-
         self.parts.append({"type": "IMAGE_URI", "imageUrl": image})
         return self
 
 
-class DSSLLMCompletionQueryMultipartMessage(DSSLLMCompletionQueryMultipartBuilder):
+class DSSLLMCompletionQueryMultipartMessage(_DSSLLMCompletionQueryMultipartBuilder):
     """
       .. important::
         Do not create this class directly, use :meth:`dataikuapi.dss.llm.DSSLLMCompletionQuery.new_multipart_message` or
@@ -761,8 +780,43 @@ class DSSLLMCompletionQueryMultipartMessage(DSSLLMCompletionQueryMultipartBuilde
         self.q.cq["messages"].append(self.msg)
         return self.q
 
+    def with_text(self, text):
+        """
+        Add a text part to the multipart message
+
+        :param str text: The text to add
+        """
+        return super().with_text(text)
+
+    def with_inline_image(self, image, mime_type=None):
+        """
+        Add an image part to the multipart message
+
+        :param Union[str, bytes] image: The image
+        :param str mime_type: None for default
+        """
+        return super().with_inline_image(image, mime_type)
 
-class DSSLLMCompletionQueryMultipartToolOutput(DSSLLMCompletionQueryMultipartBuilder):
+    def with_captioned_image_inline(self, caption, image, mime_type=None):
+        """
+        Add a captioned image part to the multipart message
+
+        :param str caption: Image caption
+        :param Union[str, bytes] image: The image
+        :param str mime_type: None for default
+        """
+        return super().with_captioned_image_inline(caption, image, mime_type)
+
+    def with_image_url(self, image):
+        """
+        Add an image url part to the multipart message
+
+        :param str image: The image url
+        """
+        return super().with_image_url(image)
+
+
+class DSSLLMCompletionQueryMultipartToolOutput(_DSSLLMCompletionQueryMultipartBuilder):
     """
       .. important::
         Do not create this class directly, use :meth:`dataikuapi.dss.llm.DSSLLMCompletionQuery.new_multipart_tool_output` or
@@ -787,24 +841,75 @@ class DSSLLMCompletionQueryMultipartToolOutput(DSSLLMCompletionQueryMultipartBui
         self.q.cq["messages"].append(self.msg)
         return self.q
 
+    def with_text(self, text):
+        """
+        Add a text part to the multipart tool output
+
+        :param str text: The text to add
+        """
+        return super().with_text(text)
+
+    def with_inline_image(self, image, mime_type=None):
+        """
+        Add an image part to the multipart tool output
+
+        :param Union[str, bytes] image: The image
+        :param str mime_type: None for default
+        """
+        return super().with_inline_image(image, mime_type)
+
+    def with_captioned_image_inline(self, caption, image, mime_type=None):
+        """
+        Add a captioned image part to the multipart tool output
+
+        :param str caption: Image caption
+        :param Union[str, bytes] image: The image
+        :param str mime_type: None for default
+        """
+        return super().with_captioned_image_inline(caption, image, mime_type)
+
+    def with_image_url(self, image):
+        """
+        Add an image url part to the multipart tool output
+
+        :param str image: The image url
+        """
+        return super().with_image_url(image)
+
 
 class DSSLLMStreamedCompletionChunk(object):
+    """
+    A handle to interact with a streamed completion query chunk.
+
+    .. important::
+        Do not create this class directly, iterate over a :class:`dataikuapi.dss.llm.DSSLLMStreamedCompletionChunks` iterator instead to generate the chunks instead.
+    """
+
     def __init__(self, data):
         self.data = data
 
     @property
     def type(self):
-        """Type of this chunk, either "content" or "event" """
+        """
+        :return: Type of this chunk, either "content" or "event"
+        :rtype: Literal["content", "event"]
+        """
         return self.data.get("type", "content")
 
     @property
     def text(self):
-        """If this chunk is content and has text, the (partial) text"""
+        """
+        :return: If this chunk is content and has text, the (partial) text
+        :rtype: bool
+        """
         return self.data.get("text", None)
 
     @property
     def event_kind(self):
-        """If this chunk is an event, its kind"""
+        """
+        :return: If this chunk is an event, its kind
+        :rtype: str
+        """
         return self.data.get("eventKind", None)
 
     def __repr__(self):
@@ -812,16 +917,31 @@ class DSSLLMStreamedCompletionChunk(object):
 
 
 class DSSLLMStreamedCompletionFooter(object):
+    """
+    A handle to interact with a streamed completion query footer.
+
+    .. important::
+        Do not create this class directly, iterate over a :class:`dataikuapi.dss.llm.DSSLLMStreamedCompletionChunks` iterator instead to generate the chunks instead.
+    """
+
     def __init__(self, data):
         self.data = data
 
     # Compatibility for code that just checks for "type""
     @property
     def type(self):
+        """
+        :return: Type of this chunk, to distinguish it from :class:`dataikuapi.dss.llm.DSSLLMStreamedCompletionChunk` chunks. Can only be "footer"
+        :rtype: Literal["footer"]
+        """
         return "footer"
 
     @property
     def trace(self):
+        """
+        :return: The trace of the completion query if available, None otherwise.
+        :rtype: Union[dict, None]
+        """
         return self.data.get("trace", None)
 
     @property
@@ -846,14 +966,14 @@ class _SSEClient(object):
         """Reads the raw source and yields events. Reassembles events
         that may span multiple HTTP chunks"""
         #logging.debug("SSEClient._read")
-        data = b''
+        data = bytearray()
         for chunk in self.raw_source:
             #logging.info("SSEClient._read: got chunk (len=%s): %s" % (len(chunk), chunk))
             for line in chunk.splitlines(True):
-                data += line
+                data.extend(line)
                 if data.endswith(b'\r\r') or data.endswith(b'\n\n') or data.endswith(b'\r\n\r\n'):
                     yield data
-                    data = b''
+                    data = bytearray()
         #logging.info("SSEClient._read: no more chunk")
         if data:
             yield data
@@ -863,6 +983,7 @@ class _SSEClient(object):
             #logging.info("SSEClient._iterevents: got event")
             evt = _SSEEvent()
 
+            data_lines = []
             for line in event_chunk.splitlines():
                 line = line.decode("utf8")
 
@@ -879,10 +1000,13 @@ class _SSEClient(object):
                     value = ''
 
                 if field == 'data':
-                    evt.__dict__[field] += value + '\n'
+                    data_lines.append(value)
                 else:
                     evt.__dict__[field] = value
 
+            if data_lines:
+                evt.data = '\n'.join(data_lines) + '\n'
+
             if evt.event is not None:
                 #logging.info("Yielding event: %s" % evt.__dict__)
                 yield evt
@@ -890,7 +1014,11 @@ class _SSEClient(object):
 
 class DSSLLMCompletionResponse(object):
     """
-    Response to a completion
+    A handle to interact with a completion query result.
+
+    .. important::
+        Do not create this class directly, use :meth:`dataikuapi.dss.llm.DSSLLMCompletionQuery.execute` or
+        :attr:`dataikuapi.dss.llm.DSSLLMCompletionsResponse.responses` or :attr:`dataikuapi.dss.llm.DSSLLMStreamedCompletionChunks.response` instead.
     """
     def __init__(self, raw_resp=None, text=None, finish_reason=None, response_parser=None, trace=None, query=None):
         if raw_resp is not None:
@@ -990,6 +1118,10 @@ class DSSLLMCompletionResponse(object):
 
     @property
     def trace(self):
+        """
+        :return: The trace of the completion query if available, None otherwise.
+        :rtype: Union[dict, None]
+        """
         return self._raw.get("trace", None)
 
     @property
@@ -1157,6 +1289,8 @@ class DSSLLMImageGenerationQuery(object):
     def new_guardrail(self, type):
         """
         Start adding a guardrail to the request. You need to configure the returned object, and call add() to actually add it
+
+        :rtype: :class:`DSSLLMRequestGuardrailBuilder`
         """
         return DSSLLMRequestGuardrailBuilder(self, type)
 
@@ -1381,6 +1515,10 @@ class DSSLLMImageGenerationResponse(object):
 
     @property
     def trace(self):
+        """
+        :return: The trace of the image generation query if available, None otherwise.
+        :rtype: Union[dict, None]
+        """
         return self._raw.get("trace", None)
 
     @property