Skip to content

Commit 5d91518

Browse files
committed
Merge branch 'release/14.6'
2 parents cee8f41 + 26b0e34 commit 5d91518

29 files changed

Lines changed: 2339 additions & 174 deletions

HISTORY.txt

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,39 @@ Changelog
22
==========
33

44

5+
14.6.0 (2026-05-27)
6+
-------------------
7+
8+
* Initial release for DSS 14.6.0
9+
10+
14.5.2 (2026-05-25)
11+
-------------------
12+
13+
* Initial release for DSS 14.5.2
14+
15+
14.5.1 (2026-04-27)
16+
-------------------
17+
18+
* Initial release for DSS 14.5.1
19+
20+
21+
14.5.0 (2026-04-17)
22+
-------------------
23+
24+
* Initial release for DSS 14.5.0
25+
26+
14.4.4 (2026-04-15)
27+
-------------------
28+
29+
* Initial release for DSS 14.4.4
30+
31+
32+
14.4.3 (2026-03-19)
33+
-------------------
34+
35+
* Initial release for DSS 14.4.3
36+
37+
538
14.4.2 (2026-03-06)
639
-------------------
740

dataikuapi/dss/admin.py

Lines changed: 31 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
from .future import DSSFuture
2+
from .local_model import DSSLocalModel
23
import json
34
import warnings
45
import logging
@@ -267,7 +268,35 @@ def set_definition(self, definition):
267268
return self.client._perform_json(
268269
"PUT", "/admin/connections/%s" % self.name,
269270
body = definition)
270-
271+
272+
273+
def list_local_models(self):
274+
"""
275+
:returns: List local models defined in this connection.
276+
:rtype: list[:class:`~dataikuapi.dss.local_model.DSSLocalModel`]
277+
:raises Exception: If this connection is not of type HuggingFaceLocal.
278+
"""
279+
definition = self.get_definition()
280+
if definition.get("type") != "HuggingFaceLocal":
281+
raise Exception("Connection %s is not a HuggingFaceLocal connection" % self.name)
282+
params = definition.get("params") or {}
283+
models = params.get("models") or []
284+
return [DSSLocalModel(self.client, self.name, model.get("id")) for model in models]
285+
286+
def get_local_model(self, model_id):
287+
"""
288+
Get a handle on this local model.
289+
290+
:param str model_id: Identifier of the model.
291+
:rtype: :class:`~dataikuapi.dss.local_model.DSSLocalModel`
292+
"""
293+
if not model_id.strip():
294+
raise ValueError("model_id must be a non-empty string")
295+
if not self.name.strip():
296+
raise ValueError("connection_name must be a non-empty string")
297+
298+
return DSSLocalModel(self.client, self.name, model_id)
299+
271300
########################################################
272301
# Security
273302
########################################################
@@ -2173,6 +2202,7 @@ def add_container_runtime_addition(self, container_runtime_addition):
21732202
* PYTHON36_SUPPORT
21742203
* PYTHON37_SUPPORT
21752204
* PYTHON38_SUPPORT
2205+
* HUGGING_FACE_LOCAL_GPU
21762206
21772207
:param dict container_runtime_addition: a dict with the container runtime addition definition
21782208
"""
@@ -3361,4 +3391,3 @@ def get_counter(self, id):
33613391
:rtype: dict
33623392
"""
33633393
return next((counter for counter in self.counters if counter["id"] == id), None)
3364-

0 commit comments

Comments
 (0)