@milaboratories/pl-drivers 1.10.3 → 1.10.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@milaboratories/pl-drivers",
-  "version": "1.10.3",
+  "version": "1.10.4",
   "engines": {
     "node": ">=20"
   },
@@ -34,10 +34,10 @@
     "upath": "^2.0.1",
     "zod": "~3.23.8",
     "@milaboratories/ts-helpers": "1.4.5",
+    "@milaboratories/helpers": "1.6.22",
     "@milaboratories/pl-client": "2.11.11",
-    "@milaboratories/pl-tree": "1.7.7",
     "@milaboratories/pl-model-common": "1.19.13",
-    "@milaboratories/helpers": "1.6.22",
+    "@milaboratories/pl-tree": "1.7.7",
     "@milaboratories/computable": "2.6.5"
   },
   "devDependencies": {
@@ -48,10 +48,10 @@
     "eslint": "^9.25.1",
     "typescript": "~5.6.3",
     "vitest": "^2.1.9",
-    "@milaboratories/ts-builder": "1.0.5",
+    "@milaboratories/eslint-config": "1.0.4",
     "@milaboratories/build-configs": "1.0.8",
-    "@milaboratories/ts-configs": "1.0.6",
-    "@milaboratories/eslint-config": "1.0.4"
+    "@milaboratories/ts-builder": "1.0.5",
+    "@milaboratories/ts-configs": "1.0.6"
   },
   "scripts": {
     "type-check": "ts-builder types --target node",

package/src/drivers/logs.test.ts

@@ -223,6 +223,20 @@ function createWd(tx: PlTransaction): FieldRef {
   return { resourceId: wd, fieldName: 'workdir' };
 }
 
+function requestComputeAllocation(tx: PlTransaction, queue: string, cpu: number, memory: string): FieldRef {
+  const quotaID = tx.createEphemeral({ name: 'Quota', version: '1' }, JSON.stringify({
+    queue: queue,
+    cpuCores: cpu,
+    memory: memory,
+  }));
+
+  const reqID = tx.createEphemeral({ name: 'ComputeRequest/executor', version: '1' });
+
+  tx.setField({ resourceId: reqID, fieldName: 'request' }, quotaID);
+
+  return { resourceId: reqID, fieldName: 'allocation' };
+}
+
 function createRunCommand(
   tx: PlTransaction,
   wdFId: FieldRef,
@@ -230,6 +244,7 @@ function createRunCommand(
   args: string[],
 ): FieldRef {
   const refsId = tx.createStruct({ name: 'RunCommandRefs', version: '1' });
+  const allocation = requestComputeAllocation(tx, 'heavy', 1, '1');
   tx.lock(refsId);
   const cmdData = {
     type: 'string',
@@ -242,14 +257,13 @@ function createRunCommand(
     };
   });
   const optsData = {
-    queueName: 'heavy',
     errorLines: 200,
     redirectStdout: 'logs.txt',
     redirectStderr: 'logs.txt',
     envs: [],
   };
 
-  const runCmdId = tx.createEphemeral({ name: 'RunCommand/executor', version: '1' });
+  const runCmdId = tx.createEphemeral({ name: 'RunCommand/executor', version: '2' });
 
   const setInputValue = (fName: string, rType: ResourceType, data: unknown) => {
     const valResId = tx.createValue(rType, jsonToData(data));
@@ -258,6 +272,7 @@ function createRunCommand(
 
   tx.setField({ resourceId: runCmdId, fieldName: 'workdirIn' }, wdFId);
   tx.setField({ resourceId: runCmdId, fieldName: 'refs' }, refsId);
+  tx.setField({ resourceId: runCmdId, fieldName: 'allocation' }, allocation);
   setInputValue('cmd', { name: 'RunCommandCmd', version: '1' }, cmdData);
   setInputValue('args', { name: 'RunCommandArgs', version: '1' }, argsData);
   setInputValue('options', { name: 'run-command/options', version: '1' }, optsData);