@qvac/harness 0.8.5 → 0.8.7

package/dist-lib/lib/tools/office/ops.js

@@ -7,7 +7,7 @@ export const PPTX_OPS = {
     setText: { slots: [['slide', 'shape'], 'text'] },
     setCellText: { slots: [['slide', 'shape', 'row', 'column'], 'text'] },
     setStyle: { slots: [['slide', 'shape'], 'style', ['paragraph']] },
-    addSlide: { slots: [['like', 'layout', 'at', 'title', 'body']] },
+    addSlide: { slots: [['like', 'layout', 'at', 'title', 'body', 'image']] },
     addPicture: { slots: ['slide', 'image', [...BOX, 'name', 'description']], inches: true },
     addTextBox: { slots: ['slide', [...BOX, 'text', 'style', 'fill', 'name']], inches: true },
     addTable: { slots: ['slide', [...BOX, 'rows', 'header', 'name']], inches: true },

package/dist-lib/lib/tools/office.js

@@ -1,12 +1,13 @@
 import b4a from 'b4a';
 import { errorMessage, isRecord } from '../guards.js';
-import { docx, pptx, units, xlsx } from '../office/engine/index.js';
+import { DocsError, docx, pptx, units, xlsx } from '../office/engine/index.js';
 import { stageInputs } from './exec/adapters/python/attachments.js';
 import { checkOutputBytes, RUN_OUTPUT_BUDGET_BYTES } from './exec/adapters/python/output-policy.js';
 import { DOCX_OPS, opArguments, opKeys, PPTX_OPS, withAliases, XLSX_OPS } from './office/ops.js';
 const READ_CELL_LIMIT = 10_000;
 const attachedThisTurn = new WeakMap();
 const FIRST_SHEET = /^sheet ?1$/i;
+const NEW_DECK_HINT = '; a new deck has only the slides its addSlide ops make, slides count from 0, and an op on a slide goes after the addSlide that makes it';
 const SHAPE_ROLES = new Set(['title', 'subtitle', 'body']);
 const HOISTED = ['attachmentId', 'output'];
 const WHERE_IDS_ARE = 'copy the id from the [Attached file "…" — attachmentId: …] line or an earlier tool result';
@@ -25,6 +26,9 @@ export function shapeFor(deck, op) {
     const found = deck.read()[slide]?.shapes.find((entry) => entry.name.toLowerCase() === wanted);
     return found && found.id !== null ? { ...op, shape: found.id } : op;
 }
+function newDeckHint(err, created) {
+    return created && err instanceof DocsError && err.code === 'SLIDE_NOT_FOUND' ? NEW_DECK_HINT : '';
+}
 export function sheetFor(book, op, created) {
     if (typeof op.sheet !== 'string')
         return op;
@@ -70,7 +74,8 @@ const PPTX_FORMAT = {
     create: (options) => pptx.create(options),
     read: (deck) => deck.read(),
     count: (deck) => ({ slides: deck.slideCount }),
-    prepare: (deck, op) => shapeFor(deck, op)
+    prepare: (deck, op) => shapeFor(deck, op),
+    hint: newDeckHint
 };
 const DOCX_FORMAT = {
     name: 'docx',
@@ -141,7 +146,11 @@ function parseOp(format, op, i) {
         return `ops[${i}] sheets lists the sheets and takes no arguments; to add a sheet use { op: "addSheet", name }`;
     }
     const takes = keys.length ? `it takes ${keys.join(', ')}` : 'it takes no arguments';
-    return `ops[${i}] ${op.op} does not take ${unknown.join(', ')}; ${takes}`;
+    const nested = unknown.filter((key) => format.ops[key]);
+    const apart = nested.length
+        ? `; ${nested.map((name) => `{ "op": "${name}", ... }`).join(' and ')} goes in ops as its own entry, next to this one, not inside it`
+        : '';
+    return `ops[${i}] ${op.op} does not take ${unknown.join(', ')}; ${takes}${apart}`;
 }
 function parseOps(format, ops) {
     if (!Array.isArray(ops)) {
@@ -252,8 +261,11 @@ async function imageInput(op, i, ctx) {
     }
     return { attachmentId: id ? await imageId(ctx, id) : undefined, path: `image-${i}.png` };
 }
+function takesImage(op) {
+    return op.spec.slots.includes('image') || op.args.image !== undefined;
+}
 async function stagedImages(request, ctx) {
-    const wanted = request.ops.flatMap((op, i) => (opKeys(op.spec).includes('image') ? [i] : []));
+    const wanted = request.ops.flatMap((op, i) => (takesImage(op) ? [i] : []));
     const images = new Map();
     if (!wanted.length) {
         return images;
@@ -358,7 +370,8 @@ function runOps({ format, document, request, images }) {
             results.push(runOp(document, op, input));
         }
         catch (err) {
-            return { error: `ops[${i}] ${op.name}: ${errorMessage(err)}` };
+            const hint = format.hint?.(err, created) ?? '';
+            return { error: `ops[${i}] ${op.name}: ${errorMessage(err)}${hint}` };
         }
     }
     return results;
@@ -452,6 +465,7 @@ async function execute(format, args, ctx) {
     const renamed = output !== request.output
         ? `Saved as ${output}, so the attached ${fileName} keeps its own name. `
         : '';
+    const unplaced = !!ctx.turnImages?.length && !images.size;
     return {
         results,
         attachments: [attachment],
@@ -459,9 +473,18 @@ async function execute(format, args, ctx) {
         reply: `${attachment.fileName} — ${countLine(counts)}`,
         next: renamed +
             `Done: ${attachment.fileName} ${repeated ? 'was already' : 'is'} attached to the chat and shown to the user. ` +
-            `Do not call ${format.name} again for it. Your answer is the reply value alone, word for word.`
+            afterSave(format, attachment.attachmentId, unplaced)
     };
 }
+function afterSave(format, attachmentId, unplaced) {
+    const reply = 'Your answer is the reply value alone, word for word.';
+    if (!unplaced) {
+        return `Do not call ${format.name} again for it. ${reply}`;
+    }
+    return (`The image the user attached to their message is not in it. If they asked for that image in the ${format.noun}, ` +
+        `call ${format.name} once more with attachmentId "${attachmentId}" and one op that adds it with "image": {}. ` +
+        `Otherwise: ${reply}`);
+}
 function createOfficeTool(format) {
     return {
         name: format.name,

package/dist-lib/lib/tools/skills/catalog.d.ts

@@ -20,6 +20,7 @@ export interface SkillCatalogEntry {
     requires?: SkillRequires;
     install?: InstallStep[];
     setup?: SkillSetup;
+    env?: Readonly<Record<string, string>>;
     cliValidator?: CliValidator;
     stub?: boolean;
     minCapabilityProfile?: string;

package/dist-lib/lib/tools/skills/catalog.js

@@ -82,6 +82,7 @@ function mergeOverlay(base, overlay) {
         credentials: base.credentials,
         allowList: base.allowList,
         platform: base.platform,
+        env: base.env,
         operations: base.operations,
         mcpReads: base.mcpReads
     };
@@ -166,6 +167,7 @@ async function loadEntry(skillsDir, dir, origin) {
             requires: meta.requires,
             install: meta.install,
             setup: meta.setup,
+            env: origin === 'overlay' ? undefined : meta.env,
             cliValidator: cli ? buildCliValidator(cli) : undefined,
             stub: meta.stub,
             minCapabilityProfile: meta.minCapabilityProfile || undefined,

package/dist-lib/lib/tools/skills/cli-schema-check.d.ts

@@ -1,3 +1,4 @@
 import type { JsonSchema } from './cli-schema-types.ts';
 export declare function buildSchemaAnnotation(schema: JsonSchema): (argv: readonly string[]) => 'read' | 'write';
+export declare function buildSchemaOutputCap(schema: JsonSchema): (argv: readonly string[]) => number | undefined;
 export declare function buildSchemaCheck(schema: JsonSchema, resourcePaths: readonly string[] | undefined): (argv: readonly string[]) => string | null;

package/dist-lib/lib/tools/skills/cli-schema-check.js

@@ -13,6 +13,9 @@ export function buildSchemaAnnotation(schema) {
         return walkCommands(argv, schema).node['x-effect'] === 'read' ? 'read' : 'write';
     };
 }
+export function buildSchemaOutputCap(schema) {
+    return (argv) => walkCommands(argv, schema).node['x-max-output-chars'];
+}
 function basename(p) {
     return p.slice(Math.max(p.lastIndexOf('/'), p.lastIndexOf('\\')) + 1);
 }

package/dist-lib/lib/tools/skills/cli-schema-load.js

@@ -28,6 +28,9 @@ export async function resolveCliSchema(skillDir, bin, readFile = (p) => fs.promi
         if (!resourceEffectsValid(schema)) {
             return { bin, schemaFile: CLI_SCHEMA_FILENAME };
         }
+        if (!outputCapsValid(schema)) {
+            return { bin, schemaFile: CLI_SCHEMA_FILENAME };
+        }
         const resourcePaths = resolveResourcePaths(skillDir, schema);
         return {
             bin,
@@ -63,6 +66,13 @@ function resourceEffectsValid(schema) {
     const files = new Set(schema['x-resource-files'] ?? []);
     return Object.entries(effects).every(([file, effect]) => effect === 'read' && files.has(file));
 }
+function outputCapsValid(schema) {
+    const cap = schema['x-max-output-chars'];
+    if (cap !== undefined && !(Number.isInteger(cap) && cap > 0)) {
+        return false;
+    }
+    return Object.values(schema.properties ?? {}).every(outputCapsValid);
+}
 function resolveResourcePaths(skillDir, schema) {
     const resources = schema['x-resource-files'] ?? [];
     return resources

package/dist-lib/lib/tools/skills/cli-schema-types.d.ts

@@ -13,6 +13,7 @@ export interface JsonSchema {
     readonly 'x-resource-files'?: readonly string[];
     readonly 'x-resource-effects'?: Readonly<Record<string, string>>;
     readonly 'x-effect'?: string;
+    readonly 'x-max-output-chars'?: number;
     readonly [key: string]: unknown;
 }
 export interface SkillCli {
@@ -27,4 +28,5 @@ export interface CliValidator {
     readonly check: (argv: readonly string[]) => string | null;
     readonly annotation: (argv: readonly string[]) => 'read' | 'write';
     readonly effect: (argv: readonly string[]) => 'read' | 'write';
+    readonly outputCap: (argv: readonly string[]) => number | undefined;
 }

package/dist-lib/lib/tools/skills/cli-schema.js

@@ -1,7 +1,7 @@
 export { CLI_SCHEMA_FILENAME } from './cli-schema-types.js';
 export { argvToObject } from './cli-argv.js';
 export { loadSkillCli, resolveCliSchema, execBinFromTools } from './cli-schema-load.js';
-import { buildSchemaAnnotation, buildSchemaCheck } from './cli-schema-check.js';
+import { buildSchemaAnnotation, buildSchemaCheck, buildSchemaOutputCap } from './cli-schema-check.js';
 export function buildCliValidator(def) {
     const bin = def.bin.toLowerCase();
     if (!def.schema) {
@@ -9,16 +9,19 @@ export function buildCliValidator(def) {
             bin,
             check: () => 'skill cli schema is missing',
             annotation: () => 'write',
-            effect: () => 'write'
+            effect: () => 'write',
+            outputCap: () => undefined
         };
     }
     const check = buildSchemaCheck(def.schema, def.resourcePaths);
     const annotation = buildSchemaAnnotation(def.schema);
+    const outputCap = buildSchemaOutputCap(def.schema);
     return {
         bin,
         check,
         annotation,
-        effect: (argv) => (check(argv) === null ? annotation(argv) : 'write')
+        effect: (argv) => (check(argv) === null ? annotation(argv) : 'write'),
+        outputCap: (argv) => (check(argv) === null ? outputCap(argv) : undefined)
     };
 }
 export function checkCliOps(argv, compiled) {

package/dist-lib/lib/tools/skills/frontmatter-parser.d.ts

@@ -12,6 +12,7 @@ export interface SkillFrontMatter {
     requires?: SkillRequires;
     install?: InstallStep[];
     setup?: SkillSetup;
+    env?: Readonly<Record<string, string>>;
     stub?: boolean;
     minCapabilityProfile?: string;
     emoji?: string;

package/dist-lib/lib/tools/skills/manifest-parse.d.ts

@@ -3,6 +3,7 @@ export interface ParsedManifest {
     requires?: SkillRequires;
     install?: InstallStep[];
     setup?: SkillSetup;
+    env?: Readonly<Record<string, string>>;
     stub?: boolean;
     minCapabilityProfile?: string;
     emoji?: string;

package/dist-lib/lib/tools/skills/manifest-parse.js

@@ -12,11 +12,23 @@ export function parseManifestBlock(block, key = 'metadata') {
         requires,
         install,
         setup,
+        env: readEnv(oc.env),
         stub: typeof oc.stub === 'boolean' ? oc.stub : undefined,
         minCapabilityProfile: typeof oc.minCapabilityProfile === 'string' ? oc.minCapabilityProfile : undefined,
         emoji: typeof oc.emoji === 'string' ? oc.emoji : undefined
     };
 }
+function readEnv(v) {
+    if (!isObj(v))
+        return undefined;
+    const env = {};
+    for (const [name, value] of Object.entries(v)) {
+        if (typeof value === 'string' && /^[A-Z_][A-Z0-9_]*$/.test(name)) {
+            env[name] = value;
+        }
+    }
+    return Object.keys(env).length > 0 ? env : undefined;
+}
 function extractJsonBlock(block, key) {
     const at = block.indexOf(`${key}:`);
     if (at === -1) {

package/dist-lib/lib/tools/skills/mention.d.ts

@@ -3,4 +3,3 @@ import { skillMentionName } from './mention-name.ts';
 export { skillMentionName };
 export declare function resolveSkillMention(text: string, catalog: readonly SkillCatalogEntry[]): SkillCatalogEntry | null;
 export declare function resolveImplicitSkill(text: string, activeSkills: readonly SkillCatalogEntry[]): SkillCatalogEntry | null;
-export declare function skillPreloadMessage(content: string, name: string, body: string): string;

package/dist-lib/lib/tools/skills/mention.js

@@ -38,6 +38,3 @@ export function resolveImplicitSkill(text, activeSkills) {
     }
     return best;
 }
-export function skillPreloadMessage(content, name, body) {
-    return `[Skill invoked by user: ${name}]\n${body}\n\nThe user's message:\n${content}`;
-}

package/dist-lib/lib/tools/translate.d.ts

@@ -3,6 +3,7 @@ import type { ToolDefinition } from '../capabilities/completion.ts';
 import type { ModelSrc } from '../models/context.ts';
 import { type NmtPlan } from '../models/nmt-catalog.ts';
 import { type LoadTracker, type TrackedLoad } from '../models/model-pool.ts';
+import type { HoldFn } from '../models/download-gate.ts';
 import { type ModelRegistrySdk } from '../models/model-source.ts';
 interface BergamotConfig {
     engine: 'Bergamot';
@@ -42,6 +43,7 @@ export interface TranslationRuntimeOptions {
     loadSdk: () => Promise<TranslationSdk>;
     loads?: LoadTracker;
     onProgress?: (model: string, progress: Record<string, unknown>) => void;
+    hold?: HoldFn;
     stallMs?: number;
 }
 export interface TranslationRuntime {
@@ -53,7 +55,7 @@ export interface TranslationRuntime {
     enter: (signal?: BareAbortSignal) => Promise<void>;
     exit: () => void;
 }
-export declare function createTranslationRuntime({ loadSdk, loads, onProgress, stallMs }: TranslationRuntimeOptions): TranslationRuntime;
+export declare function createTranslationRuntime({ loadSdk, loads, onProgress, hold, stallMs }: TranslationRuntimeOptions): TranslationRuntime;
 export type SlmTranslateFn = (model: string, params: {
     text: string;
     from: string;

package/dist-lib/lib/tools/translate.js

@@ -26,7 +26,7 @@ function bergamotConfig(plan) {
     }
     return config;
 }
-export function createTranslationRuntime({ loadSdk, loads = createLoadTracker(), onProgress, stallMs = TRANSLATION_LOAD_STALL_MS }) {
+export function createTranslationRuntime({ loadSdk, loads = createLoadTracker(), onProgress, hold, stallMs = TRANSLATION_LOAD_STALL_MS }) {
     let loading = null;
     let residentFingerprint = null;
     const lock = new ScopeLock();
@@ -95,6 +95,11 @@ export function createTranslationRuntime({ loadSdk, loads = createLoadTracker(),
         loadSdk,
         ensureLoaded: async (sdk, plan, signal) => {
             const fingerprint = fingerprintOf(plan);
+            if (hold && !(loading && residentFingerprint === fingerprint)) {
+                for (const model of plan.pivot ? [plan.model, plan.pivot] : [plan.model]) {
+                    await hold(model, 'nmt', signal);
+                }
+            }
             await lock.lock();
             try {
                 if (loading && residentFingerprint !== fingerprint) {

package/dist-lib/schema/harness.js

@@ -113,7 +113,8 @@ export function registerHarnessTypes(ns) {
             { name: 'captureInferenceCalls', type: 'bool' },
             { name: 'capability', type: 'string' },
             { name: 'arguments', type: 'json' },
-            { name: 'enhance', type: 'string' }
+            { name: 'enhance', type: 'string' },
+            { name: 'session', type: 'string' }
         ]
     });
     ns.register({
@@ -175,7 +176,16 @@ export function registerHarnessTypes(ns) {
     });
     ns.register({
         name: 'model-event-status',
-        enum: ['loading', 'downloading', 'ready', 'error', 'unloading', 'unloaded', 'downloaded'],
+        enum: [
+            'loading',
+            'downloading',
+            'ready',
+            'error',
+            'unloading',
+            'unloaded',
+            'downloaded',
+            'held'
+        ],
         strings: true
     });
     ns.register({
@@ -195,7 +205,8 @@ export function registerHarnessTypes(ns) {
             { name: 'loaded', type: '@harness/loaded-model', array: true },
             { name: 'catalogChanged', type: 'bool' },
             { name: 'downloaded', type: 'uint' },
-            { name: 'total', type: 'uint' }
+            { name: 'total', type: 'uint' },
+            { name: 'addon', type: 'string' }
         ]
     });
     ns.register({
@@ -340,7 +351,8 @@ export function registerHarnessTypes(ns) {
             { name: 'expectedSize', type: 'uint', required: true },
             { name: 'actualSize', type: 'uint' },
             { name: 'label', type: 'string' },
-            { name: 'isLoading', type: 'bool' }
+            { name: 'isLoading', type: 'bool' },
+            { name: 'memoryBytes', type: 'uint' }
         ]
     });
     ns.register({
@@ -738,6 +750,18 @@ export function registerHarnessTypes(ns) {
         name: 'unload-model-response',
         fields: [{ name: 'unloaded', type: 'bool', required: true }]
     });
+    ns.register({
+        name: 'set-model-memory-cap-request',
+        fields: [{ name: 'percent', type: 'uint', required: true }]
+    });
+    ns.register({
+        name: 'memory-usage',
+        fields: [{ name: 'usedBytes', type: 'uint', required: true }]
+    });
+    ns.register({
+        name: 'set-downloads-allowed-request',
+        fields: [{ name: 'allowed', type: 'bool', required: true }]
+    });
 }
 export function registerHarnessApi(api) {
     api.register({
@@ -905,4 +929,19 @@ export function registerHarnessApi(api) {
         request: { name: '@harness/unload-model-request', stream: false },
         response: { name: '@harness/unload-model-response', stream: false }
     });
+    api.register({
+        name: 'set-model-memory-cap',
+        request: { name: '@harness/set-model-memory-cap-request', stream: false },
+        response: { name: '@harness/ok-response', stream: false }
+    });
+    api.register({
+        name: 'memory-usage',
+        request: { name: '@harness/empty', stream: false },
+        response: { name: '@harness/memory-usage', stream: false }
+    });
+    api.register({
+        name: 'set-downloads-allowed',
+        request: { name: '@harness/set-downloads-allowed-request', stream: false },
+        response: { name: '@harness/ok-response', stream: false }
+    });
 }

package/dist-lib/spec/bind.js

@@ -139,4 +139,13 @@ export function bindApi(rpc, api) {
   if (api.unloadModel) {
     rpc.onUnloadModel((req) => api.unloadModel(req));
   }
+  if (api.setModelMemoryCap) {
+    rpc.onSetModelMemoryCap((req) => api.setModelMemoryCap(req));
+  }
+  if (api.memoryUsage) {
+    rpc.onMemoryUsage((req) => api.memoryUsage(req));
+  }
+  if (api.setDownloadsAllowed) {
+    rpc.onSetDownloadsAllowed((req) => api.setDownloadsAllowed(req));
+  }
 }

package/dist-lib/spec/calls.js

@@ -43,5 +43,8 @@ export function harnessClient(rpc) {
     setModelIdleTimeout: (input) => rpc.setModelIdleTimeout(input),
     loadModel: (input) => rpc.loadModel(input),
     unloadModel: (input) => rpc.unloadModel(input),
+    setModelMemoryCap: (input) => rpc.setModelMemoryCap(input),
+    memoryUsage: (input) => rpc.memoryUsage(input),
+    setDownloadsAllowed: (input) => rpc.setDownloadsAllowed(input),
   };
 }

package/dist-lib/spec/hrpc/hrpc.json

@@ -429,6 +429,45 @@
         "stream": false
       },
       "version": 1
+    },
+    {
+      "id": 33,
+      "name": "@harness/set-model-memory-cap",
+      "request": {
+        "name": "@harness/set-model-memory-cap-request",
+        "stream": false
+      },
+      "response": {
+        "name": "@harness/ok-response",
+        "stream": false
+      },
+      "version": 1
+    },
+    {
+      "id": 34,
+      "name": "@harness/memory-usage",
+      "request": {
+        "name": "@harness/empty",
+        "stream": false
+      },
+      "response": {
+        "name": "@harness/memory-usage",
+        "stream": false
+      },
+      "version": 1
+    },
+    {
+      "id": 35,
+      "name": "@harness/set-downloads-allowed",
+      "request": {
+        "name": "@harness/set-downloads-allowed-request",
+        "stream": false
+      },
+      "response": {
+        "name": "@harness/ok-response",
+        "stream": false
+      },
+      "version": 1
     }
   ]
 }

package/dist-lib/spec/hrpc/index.js

@@ -71,7 +71,13 @@ const methods = new Map([
   ['@harness/load-model', 31],
   [31, '@harness/load-model'],
   ['@harness/unload-model', 32],
-  [32, '@harness/unload-model']
+  [32, '@harness/unload-model'],
+  ['@harness/set-model-memory-cap', 33],
+  [33, '@harness/set-model-memory-cap'],
+  ['@harness/memory-usage', 34],
+  [34, '@harness/memory-usage'],
+  ['@harness/set-downloads-allowed', 35],
+  [35, '@harness/set-downloads-allowed']
 ])
 
 class HRPC {
@@ -111,7 +117,10 @@ class HRPC {
       ['@harness/preview-model-source', getEncoding('@harness/preview-model-source-request')],
       ['@harness/set-model-idle-timeout', getEncoding('@harness/set-model-idle-timeout-request')],
       ['@harness/load-model', getEncoding('@harness/warm-model-request')],
-      ['@harness/unload-model', getEncoding('@harness/unload-model-request')]
+      ['@harness/unload-model', getEncoding('@harness/unload-model-request')],
+      ['@harness/set-model-memory-cap', getEncoding('@harness/set-model-memory-cap-request')],
+      ['@harness/memory-usage', getEncoding('@harness/empty')],
+      ['@harness/set-downloads-allowed', getEncoding('@harness/set-downloads-allowed-request')]
     ])
     this._responseEncodings = new Map([
       ['@harness/completion', getEncoding('@harness/completion-event')],
@@ -146,7 +155,10 @@ class HRPC {
       ['@harness/preview-model-source', getEncoding('@harness/preview-model-source-response')],
       ['@harness/set-model-idle-timeout', getEncoding('@harness/ok-response')],
       ['@harness/load-model', getEncoding('@harness/load-model-response')],
-      ['@harness/unload-model', getEncoding('@harness/unload-model-response')]
+      ['@harness/unload-model', getEncoding('@harness/unload-model-response')],
+      ['@harness/set-model-memory-cap', getEncoding('@harness/ok-response')],
+      ['@harness/memory-usage', getEncoding('@harness/memory-usage')],
+      ['@harness/set-downloads-allowed', getEncoding('@harness/ok-response')]
     ])
     this._rpc = new RPC(stream, async (req) => {
       const command = methods.get(req.command)
@@ -376,6 +388,18 @@ class HRPC {
     return this._call('@harness/unload-model', args)
   }
 
+  async setModelMemoryCap(args) {
+    return this._call('@harness/set-model-memory-cap', args)
+  }
+
+  async memoryUsage(args) {
+    return this._call('@harness/memory-usage', args)
+  }
+
+  async setDownloadsAllowed(args) {
+    return this._call('@harness/set-downloads-allowed', args)
+  }
+
   onCompletion(responseFn) {
     this._handlers['@harness/completion'] = responseFn
   }
@@ -508,6 +532,18 @@ class HRPC {
     this._handlers['@harness/unload-model'] = responseFn
   }
 
+  onSetModelMemoryCap(responseFn) {
+    this._handlers['@harness/set-model-memory-cap'] = responseFn
+  }
+
+  onMemoryUsage(responseFn) {
+    this._handlers['@harness/memory-usage'] = responseFn
+  }
+
+  onSetDownloadsAllowed(responseFn) {
+    this._handlers['@harness/set-downloads-allowed'] = responseFn
+  }
+
   _requestIsStream(command) {
     return [
       '@harness/completion',