dataiku-api-client 14.5.2__tar.gz → 14.6.0__tar.gz

{dataiku_api_client-14.5.2 → dataiku_api_client-14.6.0}/HISTORY.txt

@@ -2,6 +2,10 @@ Changelog
 ==========
 
 
+14.6.0 (2026-05-27)
+-------------------
+
+* Initial release for DSS 14.6.0
 
 14.5.2 (2026-05-25)
 -------------------

{dataiku_api_client-14.5.2/dataiku_api_client.egg-info → dataiku_api_client-14.6.0}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: dataiku-api-client
-Version: 14.5.2
+Version: 14.6.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.5.2
+Download-URL: https://github.com/dataiku/dataiku-api-client-python/tarball/14.6.0
 Author: Dataiku
 Author-email: support@dataiku.com
 License: Apache Software License
@@ -36,6 +36,10 @@ Changelog
 ==========
 
 
+14.6.0 (2026-05-27)
+-------------------
+
+* Initial release for DSS 14.6.0
 
 14.5.2 (2026-05-25)
 -------------------

{dataiku_api_client-14.5.2 → dataiku_api_client-14.6.0/dataiku_api_client.egg-info}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: dataiku-api-client
-Version: 14.5.2
+Version: 14.6.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.5.2
+Download-URL: https://github.com/dataiku/dataiku-api-client-python/tarball/14.6.0
 Author: Dataiku
 Author-email: support@dataiku.com
 License: Apache Software License
@@ -36,6 +36,10 @@ Changelog
 ==========
 
 
+14.6.0 (2026-05-27)
+-------------------
+
+* Initial release for DSS 14.6.0
 
 14.5.2 (2026-05-25)
 -------------------

{dataiku_api_client-14.5.2 → dataiku_api_client-14.6.0}/dataikuapi/dss/admin.py

@@ -2202,6 +2202,7 @@ class DSSCodeEnvContainerConfsBearer(object):
         * PYTHON36_SUPPORT
         * PYTHON37_SUPPORT
         * PYTHON38_SUPPORT
+        * HUGGING_FACE_LOCAL_GPU
 
         :param dict container_runtime_addition: a dict with the container runtime addition definition
         """

{dataiku_api_client-14.5.2 → dataiku_api_client-14.6.0}/dataikuapi/dss/agent.py

@@ -226,6 +226,11 @@ class DSSAgentInteractionLoggingSelection(object):
         self.mode = self.MODE_NONE
 
 
+# Neutral aliases for shared LLM interaction logging concepts.
+DSSLLMInteractionLoggingSettings = DSSAgentInteractionLoggingSettings
+DSSLLMInteractionLoggingSelection = DSSAgentInteractionLoggingSelection
+
+
 class DSSAgentListItem(DSSTaggableObjectListItem):
     """
     An item in a list of agents
@@ -521,7 +526,7 @@ class DSSAgentVersionSettings(object):
         .. code-block:: python
 
             project = client.get_project("MYPROJECT")
-            project.create_agent_interaction_logging_dataset(
+            project.create_llm_interaction_logging_dataset(
                 "agent_logs",
                 connection_id="filesystem_managed",
                 time_partitioning="DAY",

{dataiku_api_client-14.5.2 → dataiku_api_client-14.6.0}/dataikuapi/dss/agent_review.py

@@ -214,6 +214,32 @@ class DSSAgentReview(object):
     def nb_executions(self, value):
         self.data["nbExecutions"] = value
 
+    @property
+    def hitl_validation_policy(self):
+        """
+        Review-wide default HITL validation policy used when creating a test.
+        Possible values are ``ACCEPT`` or ``REJECT``.
+        :rtype: str
+        """
+        return self.data.get("hitlValidationPolicy")
+
+    @hitl_validation_policy.setter
+    def hitl_validation_policy(self, value):
+        self.data["hitlValidationPolicy"] = value
+
+    @property
+    def hitl_max_validation_turns(self):
+        """
+        Maximum number of validation turns auto-answered during a test run before stopping.
+        Possible values are integers greater than or equal to ``1``.
+        :rtype: int
+        """
+        return self.data.get("hitlMaxValidationTurns")
+
+    @hitl_max_validation_turns.setter
+    def hitl_max_validation_turns(self, value):
+        self.data["hitlMaxValidationTurns"] = value
+
     def get_trait(self, trait_id):
         """
         Get a specific trait by its ID.
@@ -278,13 +304,14 @@ class DSSAgentReview(object):
         )
         return DSSAgentReviewTest(self.dss_client, self.project_key, test)
 
-    def create_test(self, query=None, reference_answer=None, expectations=None):
+    def create_test(self, query=None, reference_answer=None, expectations=None, hitl_validation_policy=None):
         """
         Create a new test for this agent review.
 
         :param str query: Query to test the agent. Optional.
         :param str reference_answer: Reference answer. Optional.
         :param str expectations: Expectations on the agent answer. Optional.
+        :param str hitl_validation_policy: Per-test HITL validation policy. Optional.
         :returns: The created test object.
         :rtype: :class:`DSSAgentReviewTest`
         """
@@ -294,13 +321,14 @@ class DSSAgentReview(object):
             "query": query,
             "referenceAnswer": reference_answer,
             "expectations": expectations,
+            "hitlValidationPolicy": hitl_validation_policy,
         }
         test = self.dss_client._perform_json(
             "POST", "/projects/%s/agent-reviews/tests" % self.project_key, body=body
         )
         return DSSAgentReviewTest(self.dss_client, self.project_key, test)
 
-    def create_tests_from_dataset(self, full_dataset_name, query_column, reference_answer_column=None, expectations_column=None, top_n=None, partitions=None, latest_partitions_n=None):
+    def create_tests_from_dataset(self, full_dataset_name, query_column, reference_answer_column=None, expectations_column=None, top_n=None, partitions=None, latest_partitions_n=None, hitl_validation_policy_column=None):
         """
         Create new tests for this agent review by importing them from a dataset.
 
@@ -311,6 +339,7 @@ class DSSAgentReview(object):
         :param int top_n: Only take the first n rows of the dataset. Optional.
         :param list[str] partitions: For partitioned datasets, only consider the given partitions. Optional.
         :param int latest_partitions_n: For partitioned datasets and if partitions is not set, only consider the latest n partitions. Optional.
+        :param str hitl_validation_policy_column: Name of the column containing per-test HITL validation policy. Optional.
         :returns: A dictionary with keys:
             - "createdTestIds": list of ids of the created tests
             - "error": The error message if any occurred
@@ -324,6 +353,7 @@ class DSSAgentReview(object):
             "queryColumn": query_column,
             "referenceAnswerColumn": reference_answer_column,
             "expectationsColumn": expectations_column,
+            "hitlValidationPolicyColumn": hitl_validation_policy_column,
         }
         if top_n is not None:
             body["samplingMethod"] = "HEAD_SEQUENTIAL"
@@ -587,6 +617,19 @@ class DSSAgentReviewTest(object):
     def expectations(self, value):
         self.data["expectations"] = value
 
+    @property
+    def hitl_validation_policy(self):
+        """
+        Per-test HITL validation policy.
+        Possible values are ``ACCEPT`` or ``REJECT``.
+        :rtype: str
+        """
+        return self.data.get("hitlValidationPolicy")
+
+    @hitl_validation_policy.setter
+    def hitl_validation_policy(self, value):
+        self.data["hitlValidationPolicy"] = value
+
     @property
     def creation_timestamp(self):
         """
@@ -735,6 +778,15 @@ class DSSAgentReviewTestListItem(dict):
         """
         return self.get("expectations")
 
+    @property
+    def hitl_validation_policy(self):
+        """
+        Per-test HITL validation policy.
+        Possible values are ``ACCEPT`` or ``REJECT``.
+        :rtype: str
+        """
+        return self.get("hitlValidationPolicy")
+
 
 class DSSAgentReviewRun(object):
     """

{dataiku_api_client-14.5.2 → dataiku_api_client-14.6.0}/dataikuapi/dss/knowledgebank.py

@@ -245,7 +245,7 @@ class DSSKnowledgeBankSettings(DSSTaggableObjectSettings):
     @property
     def vector_store_type(self):
         """
-        Returns the type of storage backing the vector store (could be CHROMA, PINECONE, ELASTICSEARCH, AZURE_AI_SEARCH, VERTEX_AI_GCS_BASED, FAISS, QDRANT_LOCAL)
+        Returns the type of storage backing the vector store (could be CHROMA, PINECONE, ELASTICSEARCH, AZURE_AI_SEARCH, SNOWFLAKE_CORTEX_SEARCH, VERTEX_AI_GCS_BASED, FAISS, QDRANT_LOCAL)
 
         :rtype: str
         """

{dataiku_api_client-14.5.2 → dataiku_api_client-14.6.0}/dataikuapi/dss/ml.py

@@ -1493,11 +1493,9 @@ class PredictionAlgorithmSettings(dict):
                              "To update the search strategy, use <HyperparameterSearchSettings object>.strategy = ..., "
                              "obtained with <DSSPredictionMLTaskSettings object>.get_hyperparameter_search_settings()")
 
-
-class RandomForestSettings(PredictionAlgorithmSettings):
-
+class _RandomForestSettingsBase(PredictionAlgorithmSettings):
     def __init__(self, raw_settings, hyperparameter_search_params):
-        super(RandomForestSettings, self).__init__(raw_settings, hyperparameter_search_params)
+        super(_RandomForestSettingsBase, self).__init__(raw_settings, hyperparameter_search_params)
         self.n_estimators = self._register_numerical_hyperparameter("n_estimators")
         self.min_samples_leaf = self._register_numerical_hyperparameter("min_samples_leaf")
         self.max_tree_depth = self._register_numerical_hyperparameter("max_tree_depth")
@@ -1507,10 +1505,19 @@ class RandomForestSettings(PredictionAlgorithmSettings):
         self.selection_mode = self._register_single_category_hyperparameter("selection_mode", accepted_values=["sqrt", "log2", "number", "prop"])
 
 
-class LightGBMSettings(PredictionAlgorithmSettings):
+class TimeseriesRandomForestSettings(_RandomForestSettingsBase):
+    pass
+
 
+class RandomForestSettings(_RandomForestSettingsBase):
     def __init__(self, raw_settings, hyperparameter_search_params):
-        super(LightGBMSettings, self).__init__(raw_settings, hyperparameter_search_params)
+        super(RandomForestSettings, self).__init__(raw_settings, hyperparameter_search_params)
+        self.allow_sparse_matrices = self._register_single_value_hyperparameter("allow_sparse_matrices", accepted_types=[bool])
+
+
+class _LightGBMSettingsBase(PredictionAlgorithmSettings):
+    def __init__(self, raw_settings, hyperparameter_search_params):
+        super(_LightGBMSettingsBase, self).__init__(raw_settings, hyperparameter_search_params)
         self.boosting_type = self._register_categorical_hyperparameter("boosting_type")
         self.num_leaves = self._register_numerical_hyperparameter("num_leaves")
         self.learning_rate = self._register_numerical_hyperparameter("learning_rate")
@@ -1521,9 +1528,6 @@ class LightGBMSettings(PredictionAlgorithmSettings):
         self.colsample_bytree = self._register_numerical_hyperparameter("colsample_bytree")
         self.reg_alpha = self._register_numerical_hyperparameter("reg_alpha")
         self.reg_lambda = self._register_numerical_hyperparameter("reg_lambda")
-
-        self.early_stopping = self._register_single_value_hyperparameter("early_stopping", accepted_types=[bool])
-        self.early_stopping_rounds = self._register_single_value_hyperparameter("early_stopping_rounds", accepted_types=[int])
         self.random_state = self._register_single_value_hyperparameter("random_state", accepted_types=[int])
         self.n_jobs = self._register_single_value_hyperparameter("n_jobs", accepted_types=[int])
         self.max_depth = self._register_single_value_hyperparameter("max_depth", accepted_types=[int])
@@ -1532,6 +1536,18 @@ class LightGBMSettings(PredictionAlgorithmSettings):
         self.use_bagging = self._register_single_value_hyperparameter("use_bagging", accepted_types=[bool])
 
 
+class TimeseriesLightGBMSettings(_LightGBMSettingsBase):
+    pass
+
+
+class LightGBMSettings(_LightGBMSettingsBase):
+    def __init__(self, raw_settings, hyperparameter_search_params):
+        super(LightGBMSettings, self).__init__(raw_settings, hyperparameter_search_params)
+        self.early_stopping = self._register_single_value_hyperparameter("early_stopping", accepted_types=[bool])
+        self.early_stopping_rounds = self._register_single_value_hyperparameter("early_stopping_rounds", accepted_types=[int])
+        self.allow_sparse_matrices = self._register_single_value_hyperparameter("allow_sparse_matrices", accepted_types=[bool])
+
+
 class TabICLSettings(PredictionAlgorithmSettings):
 
     def __init__(self, raw_settings, hyperparameter_search_params):
@@ -1551,10 +1567,10 @@ class TabICLSettings(PredictionAlgorithmSettings):
         self.outlier_threshold = self._register_single_value_hyperparameter("outlier_threshold")
 
 
-class XGBoostSettings(PredictionAlgorithmSettings):
+class _XGBoostSettingsBase(PredictionAlgorithmSettings):
 
     def __init__(self, raw_settings, hyperparameter_search_params):
-        super(XGBoostSettings, self).__init__(raw_settings, hyperparameter_search_params)
+        super(_XGBoostSettingsBase, self).__init__(raw_settings, hyperparameter_search_params)
         self.max_depth = self._register_numerical_hyperparameter("max_depth")
         self.learning_rate = self._register_numerical_hyperparameter("learning_rate")
         self.gamma = self._register_numerical_hyperparameter("gamma")
@@ -1575,8 +1591,6 @@ class XGBoostSettings(PredictionAlgorithmSettings):
         self.missing = self._register_single_value_hyperparameter("missing", accepted_types=[int, float])
         self.tree_method = self._register_single_category_hyperparameter("tree_method", accepted_values=["auto", "exact", "approx", "hist"])
         self.seed = self._register_single_value_hyperparameter("seed", accepted_types=[int])
-        self.enable_early_stopping = self._register_single_value_hyperparameter("enable_early_stopping", accepted_types=[bool])
-        self.early_stopping_rounds = self._register_single_value_hyperparameter("early_stopping_rounds", accepted_types=[int])
         self.tweedie_variance_power = self._register_single_value_hyperparameter("tweedie_variance_power", accepted_types=[int, float])
 
     @property
@@ -1614,6 +1628,18 @@ class XGBoostSettings(PredictionAlgorithmSettings):
             self.tree_method.set_value(value)
 
 
+class TimeseriesXGBoostSettings(_XGBoostSettingsBase):
+    pass
+
+
+class XGBoostSettings(_XGBoostSettingsBase):
+    def __init__(self, raw_settings, hyperparameter_search_params):
+        super(XGBoostSettings, self).__init__(raw_settings, hyperparameter_search_params)
+        self.enable_early_stopping = self._register_single_value_hyperparameter("enable_early_stopping", accepted_types=[bool])
+        self.early_stopping_rounds = self._register_single_value_hyperparameter("early_stopping_rounds", accepted_types=[int])
+        self.allow_sparse_matrices = self._register_single_value_hyperparameter("allow_sparse_matrices", accepted_types=[bool])
+
+
 class GradientBoostedTreesSettings(PredictionAlgorithmSettings):
 
     def __init__(self, raw_settings, hyperparameter_search_params):
@@ -1866,6 +1892,71 @@ class AutoArimaSettings(PredictionAlgorithmSettings):
         self.maxiter = self._register_single_value_hyperparameter("maxiter", accepted_types=[int])
 
 
+class ArimaSettings(PredictionAlgorithmSettings):
+    def __init__(self, raw_settings, hyperparameter_search_params):
+        super(ArimaSettings, self).__init__(raw_settings, hyperparameter_search_params)
+        self.p = self._register_single_value_hyperparameter("p", accepted_types=[int])
+        self.d = self._register_single_value_hyperparameter("d", accepted_types=[int])
+        self.q = self._register_single_value_hyperparameter("q", accepted_types=[int])
+        self.P = self._register_single_value_hyperparameter("P", accepted_types=[int])
+        self.D = self._register_single_value_hyperparameter("D", accepted_types=[int])
+        self.Q = self._register_single_value_hyperparameter("Q", accepted_types=[int])
+        self.s = self._register_single_value_hyperparameter("s", accepted_types=[int])
+        self.trend = self._register_single_category_hyperparameter("trend", accepted_values=["n", "c", "t", "ct"])
+        self.trend_offset = self._register_single_value_hyperparameter("trend_offset", accepted_types=[int])
+        self.enforce_stationarity = self._register_single_value_hyperparameter("enforce_stationarity", accepted_types=[bool])
+        self.enforce_invertibility = self._register_single_value_hyperparameter("enforce_invertibility", accepted_types=[bool])
+        self.concentrate_scale = self._register_single_value_hyperparameter("concentrate_scale", accepted_types=[bool])
+
+
+class CrostonSettings(PredictionAlgorithmSettings):
+    def __init__(self, raw_settings, hyperparameter_search_params):
+        super(CrostonSettings, self).__init__(raw_settings, hyperparameter_search_params)
+        self.variant = self._register_categorical_hyperparameter("variant")
+        self.alpha_d = self._register_numerical_hyperparameter("alpha_d")
+        self.alpha_p = self._register_numerical_hyperparameter("alpha_p")
+
+
+class ETSSettings(PredictionAlgorithmSettings):
+
+    def __init__(self, raw_settings, hyperparameter_search_params):
+        super(ETSSettings, self).__init__(raw_settings, hyperparameter_search_params)
+        self.trend = self._register_categorical_hyperparameter("trend")
+        self.damped_trend = self._register_categorical_hyperparameter("damped_trend")
+        self.seasonal = self._register_categorical_hyperparameter("seasonal")
+        self.error = self._register_categorical_hyperparameter("error")
+        self.seasonal_periods = self._register_single_value_hyperparameter("seasonal_periods", accepted_types=[int])
+        self.include_unstable = self._register_single_value_hyperparameter("include_unstable", accepted_types=[bool])
+        self.seed = self._register_single_value_hyperparameter("seed", accepted_types=[int])
+
+
+class NHITSSettings(PredictionAlgorithmSettings):
+    def __init__(self, raw_settings, hyperparameter_search_params):
+        super(NHITSSettings, self).__init__(raw_settings, hyperparameter_search_params)
+        self.context_length = self._register_numerical_hyperparameter("context_length")
+        self.learning_rate = self._register_numerical_hyperparameter("learning_rate")
+        self.batch_size = self._register_single_value_hyperparameter("batch_size", accepted_types=[int])
+        self.max_steps = self._register_single_value_hyperparameter("max_steps", accepted_types=[int])
+        self.patience = self._register_single_value_hyperparameter("patience", accepted_types=[int])
+        self.random_state = self._register_single_value_hyperparameter("random_state", accepted_types=[int])
+
+
+class TFTSettings(PredictionAlgorithmSettings):
+    def __init__(self, raw_settings, hyperparameter_search_params):
+        super(TFTSettings, self).__init__(raw_settings, hyperparameter_search_params)
+        self.context_length = self._register_numerical_hyperparameter("context_length")
+        self.hidden_size_factor = self._register_numerical_hyperparameter("hidden_size_factor")
+        self.learning_rate = self._register_numerical_hyperparameter("learning_rate")
+        self.n_rnn_layers = self._register_numerical_hyperparameter("n_rnn_layers")
+        self.n_head = self._register_numerical_hyperparameter("n_head")
+        self.batch_size = self._register_single_value_hyperparameter("batch_size", accepted_types=[int])
+        self.max_steps = self._register_single_value_hyperparameter("max_steps", accepted_types=[int])
+        self.patience = self._register_single_value_hyperparameter("patience", accepted_types=[int])
+        self.random_state = self._register_single_value_hyperparameter("random_state", accepted_types=[int])
+        self.max_hidden_size = self._register_single_value_hyperparameter("max_hidden_size", accepted_types=[int])
+        self.limit_hidden_size = self._register_single_value_hyperparameter("limit_hidden_size", accepted_types=[bool])
+
+
 class SeasonalLoessSettings(PredictionAlgorithmSettings):
 
     def __init__(self, raw_settings, hyperparameter_search_params):
@@ -2377,6 +2468,15 @@ class DSSClusteringMLTaskSettings(DSSMLTaskSettings):
 class DSSTimeseriesForecastingMLTaskSettings(AbstractTabularPredictionMLTaskSettings):
     __doc__ = []
     _algorithm_remap = {
+        "ARIMA": PredictionAlgorithmMeta("arima_timeseries", ArimaSettings),
+        "CROSTON": PredictionAlgorithmMeta("croston_timeseries", CrostonSettings),
+        "ETS": PredictionAlgorithmMeta("ets_timeseries", ETSSettings),
+        "NHITS": PredictionAlgorithmMeta("nhits_timeseries", NHITSSettings),
+        "TFT": PredictionAlgorithmMeta("tft_timeseries", TFTSettings),
+        "RANDOM_FOREST_REGRESSION": PredictionAlgorithmMeta("random_forest_regression", TimeseriesRandomForestSettings),
+        "XGBOOST_REGRESSION": PredictionAlgorithmMeta("xgboost", TimeseriesXGBoostSettings),
+        "RIDGE_REGRESSION": PredictionAlgorithmMeta("ridge_regression", RidgeRegressionSettings),
+        "LIGHTGBM_REGRESSION": PredictionAlgorithmMeta("lightgbm_regression", TimeseriesLightGBMSettings),
         "TRIVIAL_IDENTITY_TIMESERIES": PredictionAlgorithmMeta("trivial_identity_timeseries"),
         "SEASONAL_NAIVE": PredictionAlgorithmMeta("seasonal_naive_timeseries", SeasonalNaiveSettings),
         "AUTO_ARIMA": PredictionAlgorithmMeta("autoarima_timeseries", AutoArimaSettings),

{dataiku_api_client-14.5.2 → dataiku_api_client-14.6.0}/dataikuapi/dss/project.py

@@ -2744,6 +2744,7 @@ class DSSProject(object):
             * PINECONE
             * ELASTICSEARCH
             * AZURE_AI_SEARCH
+            * SNOWFLAKE_CORTEX_SEARCH
             * VERTEX_AI_GCS_BASED
             * FAISS *(not recommended)*
             * QDRANT_LOCAL *(not recommended)*
@@ -2918,6 +2919,27 @@ class DSSProject(object):
         """
         Create a new agent interaction logging dataset.
 
+        .. caution:: Deprecated. Use :meth:`create_llm_interaction_logging_dataset`
+
+        :param dataset_name: Identifier for the dataset to create.
+        :type dataset_name: str
+        :param connection_id: Identifier of the connection to use (SQL or filesystem-like).
+        :type connection_id: str
+        :param time_partitioning: Time period of the partitioning. Can be one of "HOUR", "DAY", "MONTH". Set to None for no partitioning.
+        :type time_partitioning: str | None
+        """
+        return self.create_llm_interaction_logging_dataset(
+            dataset_name=dataset_name,
+            connection_id=connection_id,
+            time_partitioning=time_partitioning,
+        )
+
+    def create_llm_interaction_logging_dataset(self, dataset_name, connection_id, time_partitioning=None):
+        """
+        Create a new LLM interaction logging dataset.
+
+        This dataset can be used by both agents and retrieval-augmented LLMs.
+
         :param dataset_name: Identifier for the dataset to create.
         :type dataset_name: str
         :param connection_id: Identifier of the connection to use (SQL or filesystem-like).
@@ -2931,7 +2953,7 @@ class DSSProject(object):
             "timePartitioning": time_partitioning
         }
         return self.client._perform_empty(
-            "POST", "/projects/%s/agents/interaction-logging-dataset" % self.project_key, params=params
+            "POST", "/projects/%s/llm-interaction-logging-datasets" % self.project_key, params=params
         )
 
     ########################################################

{dataiku_api_client-14.5.2 → dataiku_api_client-14.6.0}/dataikuapi/dss/retrieval_augmented_llm.py

@@ -1,3 +1,4 @@
+from .agent import DSSLLMInteractionLoggingSelection
 from .utils import DSSTaggableObjectListItem, DSSTaggableObjectSettings
 
 class DSSRetrievalAugmentedLLMListItem(DSSTaggableObjectListItem):
@@ -137,6 +138,17 @@ class DSSRetrievalAugmentedLLMVersionSettings(object):
     def get_raw(self):
         return self._version_settings
 
+    def _get_or_create_interaction_logging_selection(self):
+        rag_settings = self._version_settings["ragllmSettings"]
+        interaction_logging_selection = rag_settings.get("interactionLoggingSelection")
+        if interaction_logging_selection is None:
+            interaction_logging_selection = {
+                "mode": DSSLLMInteractionLoggingSelection.MODE_INHERIT,
+                "explicitSettings": {},
+            }
+            rag_settings["interactionLoggingSelection"] = interaction_logging_selection
+        return interaction_logging_selection
+
     @property
     def llm_id(self):
         """
@@ -148,4 +160,77 @@ class DSSRetrievalAugmentedLLMVersionSettings(object):
 
     @llm_id.setter
     def llm_id(self, value):
-        self._version_settings["ragllmSettings"]["llmId"] = value
+        self._version_settings["ragllmSettings"]["llmId"] = value
+
+    @property
+    def interaction_logging_selection(self):
+        """
+        Get the interaction logging selection for this version.
+
+        Before configuring interaction logging on a retrieval-augmented LLM version,
+        create the target dataset on the project:
+
+        .. code-block:: python
+
+            project = client.get_project("MYPROJECT")
+            project.create_llm_interaction_logging_dataset(
+                "llm_logs",
+                connection_id="filesystem_managed",
+                time_partitioning="DAY",
+            )
+
+        Example using inherited settings:
+
+        .. code-block:: python
+
+            rag = project.get_retrieval_augmented_llm("my_rag")
+            rag_settings = rag.get_settings()
+            version_settings = rag_settings.get_version_settings("v1")
+
+            logging_selection = version_settings.interaction_logging_selection
+            logging_selection.inherit()
+
+            rag_settings.save()
+
+        Example using explicit settings:
+
+        .. code-block:: python
+
+            rag = project.get_retrieval_augmented_llm("my_rag")
+            rag_settings = rag.get_settings()
+            version_settings = rag_settings.get_version_settings("v1")
+
+            logging_selection = version_settings.interaction_logging_selection
+            logging_selection.enable(
+                "llm_logs",
+                settings={
+                    "flushEveryS": 60,
+                    "flushEveryBytes": 1_000_000,
+                    "contentMode": "FULL",
+                },
+            )
+
+            rag_settings.save()
+
+        Example disabling interaction logging:
+
+        .. code-block:: python
+
+            rag = project.get_retrieval_augmented_llm("my_rag")
+            rag_settings = rag.get_settings()
+            version_settings = rag_settings.get_version_settings("v1")
+
+            logging_selection = version_settings.interaction_logging_selection
+            logging_selection.disable()
+
+            rag_settings.save()
+
+        :rtype: :class:`dataikuapi.dss.agent.DSSLLMInteractionLoggingSelection`
+        """
+        return DSSLLMInteractionLoggingSelection(self._get_or_create_interaction_logging_selection())
+
+    @interaction_logging_selection.setter
+    def interaction_logging_selection(self, value):
+        if isinstance(value, DSSLLMInteractionLoggingSelection):
+            value = value.get_raw()
+        self._version_settings["ragllmSettings"]["interactionLoggingSelection"] = value

{dataiku_api_client-14.5.2 → dataiku_api_client-14.6.0}/dataikuapi/dss/semantic_model.py

@@ -276,6 +276,7 @@ class DSSSemanticModelVersionSettings(object):
             "entities": [],
             "relationships": [],
             "goldenQueries": [],
+            "crossEntityMetrics": [],
             "glossaryTerms": [],
             "glossaryBindings": [],
             "indexingSettings": {

{dataiku_api_client-14.5.2 → dataiku_api_client-14.6.0}/dataikuapi/dssclient.py

@@ -960,7 +960,7 @@ class DSSClient(object):
 
             env_handle = client.create_internal_code_env(internal_env_type="RAG_CODE_ENV", python_interpreter="PYTHON310")
 
-        :param str internal_env_type: the internal env type, can be `DEEP_HUB_IMAGE_CLASSIFICATION_CODE_ENV`, `DEEP_HUB_IMAGE_OBJECT_DETECTION_CODE_ENV`, `PROXY_MODELS_CODE_ENV`, `DATABRICKS_UTILS_CODE_ENV`, `PII_DETECTION_CODE_ENV`, `HUGGINGFACE_LOCAL_CODE_ENV`, `RAG_CODE_ENV`, `DOCUMENT_EXTRACTION_CODE_ENV`, `DOCUMENT_TEMPLATING_CODE_ENV`.
+        :param str internal_env_type: the internal env type, can be `DEEP_HUB_IMAGE_CLASSIFICATION_CODE_ENV`, `DEEP_HUB_IMAGE_OBJECT_DETECTION_CODE_ENV`, `PROXY_MODELS_CODE_ENV`, `DATABRICKS_UTILS_CODE_ENV`, `PII_DETECTION_CODE_ENV`, `HUGGINGFACE_LOCAL_CODE_ENV`, `RAG_CODE_ENV`, `DOCUMENT_EXTRACTION_CODE_ENV`, `DOCUMENT_TEMPLATING_CODE_ENV`, `HUGGINGFACE_LOCAL_CPU_CODE_ENV`.
         :param str python_interpreter: Python interpreter version, can be `PYTHON39`, `PYTHON310`, `PYTHON311`, `PYTHON312` or `PYTHON313`. If None, DSS will try to select a supported & available interpreter.
         :param str code_env_version: Version of the code env. Reserved for future use.
         :param bool wait: wait for the code env to be created or return a future
@@ -1828,6 +1828,9 @@ class DSSClient(object):
         """
         Returns a dictionary with information about licensing status of this DSS instance
 
+        Note:
+            The API is not available on Cloud. Use the Launchpad or Launchpad API.
+
         :rtype: dict
         """
         return self._perform_json("GET", "/admin/licensing/status")
@@ -1836,6 +1839,9 @@ class DSSClient(object):
         """
         Sets a new licence for DSS
 
+        Note:
+            The API is not available on Cloud. Use the Launchpad or Launchpad API.
+
         :param license: license (content of license file)
         :return: None
         """

{dataiku_api_client-14.5.2 → dataiku_api_client-14.6.0}/dataikuapi/fm/instances.py

@@ -75,6 +75,7 @@ class FMInstanceCreator(object):
         data_volume_IOPS=None,
         data_volume_encryption=None,
         data_volume_encryption_key=None,
+        data_volume_throughput=None,
     ):
         """
         Set the options of the data volume to use with the DSS Instance
@@ -86,6 +87,7 @@ class FMInstanceCreator(object):
         :param data_volume_encryption: Optional, encryption mode of the data volume
         :type data_volume_encryption: :class:`dataikuapi.fm.instances.FMInstanceEncryptionMode`
         :param str data_volume_encryption_key: Optional, the encryption key to use when data_volume_encryption_key is FMInstanceEncryptionMode.CUSTOM
+        :param int data_volume_throughput: Optional, data volume throughput
         :rtype: :class:`dataikuapi.fm.instances.FMInstanceCreator`
         """
         if not data_volume_encryption:
@@ -99,6 +101,7 @@ class FMInstanceCreator(object):
         self.data["dataVolumeSizeGB"] = data_volume_size
         self.data["dataVolumeSizeMaxGB"] = data_volume_size_max
         self.data["dataVolumeIOPS"] = data_volume_IOPS
+        self.data["dataVolumeThroughput"] = data_volume_throughput
         self.data["volumesEncryption"] = data_volume_encryption.value
         self.data["volumesEncryptionKey"] = data_volume_encryption_key
         return self

{dataiku_api_client-14.5.2 → dataiku_api_client-14.6.0}/dataikuapi/govern_client.py

@@ -673,6 +673,9 @@ class GovernClient(object):
         """
         Returns a dictionary with information about licensing status of this Dataiku Govern instance
 
+        Note:
+            The API is not available on Cloud. Use the Launchpad or Launchpad API.
+
         :rtype: dict
         """
         return self._perform_json("GET", "/admin/licensing/status")
@@ -681,6 +684,9 @@ class GovernClient(object):
         """
         Sets a new licence for Dataiku Govern
 
+        Note:
+            The API is not available on Cloud. Use the Launchpad or Launchpad API.
+
         :param license: license (content of license file)
         :return: None
         """

{dataiku_api_client-14.5.2 → dataiku_api_client-14.6.0}/dataikuapi/launchpad/group.py

@@ -30,7 +30,7 @@ class LaunchpadGroup:
         group = client.build_group("Designers", "Designer group", ["user@example.com"])
         group.launchpad_permissions = {"mayTurnOnSpace": True}
         group.update_permissions({"mayCreateProjects": True}, node_type="dataiku")
-        group.update_permissions({"mayManageGovern": True}, node_type="govern")
+        group.update_permissions({"isGovernArchitect": True}, node_type="govern")
         group.save()
 
         # Get a group