Merge branch 'release/14.4'

Author: kevanescence

HISTORY.txt

@@ -2,6 +2,21 @@ Changelog
 ==========
 
 
+14.4.2 (2026-03-06)
+-------------------
+
+* Initial release for DSS 14.4.2
+
+14.4.1 (2026-02-23)
+-------------------
+
+* Initial release for DSS 14.4.1
+
+14.4.0 (2026-02-09)
+-------------------
+
+* Initial release for DSS 14.4.0
+
 14.3.3 (2026-01-30)
 -------------------
 

dataikuapi/__init__.py

@@ -8,3 +8,14 @@
 from .dss.recipe import GroupingRecipeCreator, UpsertRecipeCreator, JoinRecipeCreator, StackRecipeCreator, WindowRecipeCreator, SyncRecipeCreator, SamplingRecipeCreator, SQLQueryRecipeCreator, CodeRecipeCreator, SplitRecipeCreator, SortRecipeCreator, TopNRecipeCreator, DistinctRecipeCreator, DownloadRecipeCreator, PredictionScoringRecipeCreator, ClusteringScoringRecipeCreator
 
 from .dss.admin import DSSUserImpersonationRule, DSSGroupImpersonationRule
+
+import sys
+import warnings
+
+if sys.version_info >= (3, 6):
+    from .launchpad_client import LaunchpadClient
+else:
+    warnings.warn(
+        "dataikuapi.LaunchpadClient is only available for Python 3.6 and above",
+        DeprecationWarning,
+    )

dataikuapi/dss/admin.py

@@ -2266,7 +2266,7 @@ def get_version(self, version_id=None):
                 if version_id == version.get("versionId"):
                     return DSSAutomationCodeEnvVersionSettings(self.codeenv, version)
             raise Exception("Version %s not found in : %s" % (version_id, ', '.join(version_ids)))
-        elif deployment_mode in ['PLUGIN_NON_MANAGED', 'PLUGIN_MANAGED', 'AUTOMATION_NON_MANAGED_PATH', 'EXTERNAL_CONDA_NAMED']:
+        elif deployment_mode in ['PLUGIN_NON_MANAGED', 'PLUGIN_MANAGED', 'BUSINESS_APP_NON_MANAGED', 'BUSINESS_APP_MANAGED', 'AUTOMATION_NON_MANAGED_PATH', 'EXTERNAL_CONDA_NAMED']:
             return DSSAutomationCodeEnvVersionSettings(self.codeenv, self.settings.get('noVersion', {}))
         else:
             raise Exception("Unexpected deployment mode %s for an automation node code env. Alter the settings directly with get_raw()", deployment_mode)

dataikuapi/dss/agent.py

@@ -80,6 +80,46 @@ def delete(self):
         """
         return self.client._perform_empty("DELETE", "/projects/%s/agents/%s" % (self.project_key, self.id))
 
+    def shutdown(self, version_id=None, force=False):
+        """
+        Shutdown all instances of the given version of this agent
+
+        :param version_id: If unspecified, uses the active version.
+        :type version_id: str | None
+        :param force: If True, cancel requests being processed and stop the instances. If False, let those active requests complete before stopping.
+        :type force: bool
+        """
+        return self.client._perform_empty(
+            "POST", "/projects/%s/agents/%s/actions/shutdown" % (self.project_key, self.id), body={
+                "versionId": version_id,
+                "force": force
+            })
+
+    def status(self, version_id=None):
+        """
+        Query status of instances of the given version of this agent
+
+        :param version_id: If unspecified, uses the active version.
+        :type version_id: str | None
+        :return: A dict holding the list of the status for each instance.
+        """
+        return self.client._perform_json(
+            "GET", "/projects/%s/agents/%s/status" % (self.project_key, self.id), body={
+                "versionId": version_id
+            })
+
+    def wake_up(self, version_id=None):
+        """
+        Start an instance of an agent if none is started
+
+        :param version_id: If unspecified, uses the active version.
+        :type version_id: str | None
+        """
+        return self.client._perform_empty(
+            "POST", "/projects/%s/agents/%s/actions/wakeup" % (self.project_key, self.id), body={
+                "versionId": version_id
+            })
+
 class DSSAgentSettings(DSSTaggableObjectSettings):
     """
     Settings for a agent
@@ -149,13 +189,13 @@ def llm_id(self):
         :rtype: :class:`str`
         """
         if not self._agent_settings.type == "TOOLS_USING_AGENT":
-            raise ValueError("Only valid for Visual Agents")
+            raise ValueError("Only valid for Simple Visual Agents")
         return self._version_settings["toolsUsingAgentSettings"]["llmId"]
 
     @llm_id.setter
     def llm_id(self, value):
         if not self._agent_settings.type == "TOOLS_USING_AGENT":
-            raise ValueError("Only valid for Visual Agents")
+            raise ValueError("Only valid for Simple Visual Agents")
         self._version_settings["toolsUsingAgentSettings"]["llmId"] = value
 
     @property
@@ -166,6 +206,8 @@ def tools(self):
         Each tool is a dict, containing at least "toolRef", which is the identifier of the tool.
         The dict may also contain "additionalDescription" which is added to the description of the tool
         """
+        if not self._agent_settings.type == "TOOLS_USING_AGENT":
+            raise ValueError("Only valid for Simple Visual Agents")
         return self._version_settings["toolsUsingAgentSettings"]["tools"]