@platforma-sdk/workflow-tengo 5.12.1 → 5.13.0

package/dist/tengo/lib/exec.delayed-compute-allocation.lib.tengo

@@ -21,7 +21,8 @@ create := func(workdir, req) {
 	validation.assertType(req, {
 		"queue": "string",
 		"cpu,?": "number",
-		"ram,?": ["or", "number", "string"]
+		"ram,?": ["or", "number", "string"],
+		"gpu,?": "number"
 	}, "request must be a valid request")
 
 	reqMap := { queue: req.queue }
@@ -31,6 +32,9 @@ create := func(workdir, req) {
 	if !is_undefined(req.ram) {
 		reqMap.ram = req.ram
 	}
+	if !is_undefined(req.gpu) && req.gpu > 0 {
+		reqMap.gpu = req.gpu
+	}
 
 	delayedComputeAllocation := render.createEphemeral(tpl, {
 		request: smart.createJsonResource(reqMap),

package/dist/tengo/lib/exec.lib.tengo

@@ -51,6 +51,7 @@ builder := func() {
 	queue := execConstants.HEAVY_QUEUE
 	cpuRequest := undefined
 	ramRequest := undefined
+	gpuCount := 0
 
 	commandName := ""
 	cache := 0
@@ -441,6 +442,20 @@ builder := func() {
 
 
 
+
+		gpu: func(count) {
+			validation.assertType(count, "number", "exec.builder.gpu: count must be a number")
+			ll.assert(count > 0, "exec.builder.gpu: count must be greater than 0")
+			gpuCount = count
+			return self
+		},
+
+
+
+
+
+
+
 		cache: func(time) {
 			ll.assert(
 				is_int(time),
@@ -1019,12 +1034,16 @@ builder := func() {
 			if (is_int(ramRequest)) {
 				ramRequest = string(ramRequest) + "b"
 			}
+			quotaData := {
+				queue: queue,
+				cpu: cpuRequest,
+				ram: ramRequest
+			}
+			if gpuCount > 0 {
+				quotaData.gpu = gpuCount
+			}
 			metaExecInputs := {
-				quota: smart.createJsonResource({
-					queue: queue,
-					cpu: cpuRequest,
-					ram: ramRequest
-				})
+				quota: smart.createJsonResource(quotaData)
 			}
 
 

package/dist/tengo/lib/exec.limits.lib.tengo

@@ -51,7 +51,7 @@ quotaBuilder := func() {
 	queue_name := undefined
 	cpu_request := undefined
 	ram_request := undefined
-
+	gpu_request := undefined
 
 	self = ll.toStrict({
 
@@ -104,6 +104,20 @@ quotaBuilder := func() {
 
 
 
+
+		gpu: func(amount) {
+			validation.assertType(amount, ["or", "number", "null"], "exec.quota.builder.gpu: GPU amount should be a number")
+			ll.assert(is_undefined(amount) || amount > 0, "exec.quota.builder.gpu: amount should be greater than 0")
+			gpu_request = amount
+
+			return self
+		},
+
+
+
+
+
+
 		build: func() {
 			ll.assert(queue_name != "", "exec.quota.builder: queue name is mandatory and could not be empty string")
 
@@ -116,6 +130,9 @@ quotaBuilder := func() {
 			if !is_undefined(ram_request) {
 				data.memory = ram_request
 			}
+			if !is_undefined(gpu_request) && gpu_request > 0 {
+				data.gpu = gpu_request
+			}
 
 			return smart.ephemeralBuilder(_RTYPE_QUOTA, json.encode(data)).lockAndBuild()
 		}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@platforma-sdk/workflow-tengo",
-  "version": "5.12.1",
+  "version": "5.13.0",
   "description": "Tengo SDK library",
   "type": "module",
   "types": "./dist/index.d.ts",