@neta-art/cohub-cli 1.9.1 → 1.10.0

package/dist/commands/generations.js

@@ -1,5 +1,6 @@
 import { mkdir, readFile, stat, writeFile } from "node:fs/promises";
 import { basename, dirname, extname, join } from "node:path";
+import { GenerationPolicyError, assertGenerationRequestAllowedByPolicy, parseGenerationPolicyFromEnv, } from "@neta-art/cohub";
 import { createClient } from "../client.js";
 import { resolveSpace } from "../space.js";
 import { json as outJson, jsonRequested, ok, error, handleHttp, spinner } from "../output.js";
@@ -153,6 +154,45 @@ function parseTimeoutMs(value) {
         return error("Invalid timeout", "--timeout-ms must be a positive integer");
     return timeoutMs;
 }
+function envValue(name) {
+    const value = process.env[name]?.trim();
+    return value || undefined;
+}
+function parseMetadata(value) {
+    if (!value)
+        return undefined;
+    let parsed;
+    try {
+        parsed = JSON.parse(value);
+    }
+    catch {
+        return error("Invalid metadata", "--metadata must be a JSON object");
+    }
+    if (!parsed || typeof parsed !== "object" || Array.isArray(parsed)) {
+        return error("Invalid metadata", "--metadata must be a JSON object");
+    }
+    return parsed;
+}
+function mergeCohubMetadata(metadata) {
+    const sessionId = envValue("COHUB_SESSION_ID");
+    const turnId = envValue("COHUB_TURN_ID");
+    const toolCallId = envValue("COHUB_TOOL_CALL_ID");
+    if (!sessionId && !turnId && !toolCallId)
+        return metadata;
+    const existingCohub = metadata?.cohub;
+    const cohub = existingCohub && typeof existingCohub === "object" && !Array.isArray(existingCohub)
+        ? existingCohub
+        : {};
+    return {
+        ...(metadata ?? {}),
+        cohub: {
+            ...cohub,
+            ...(sessionId ? { sessionId } : {}),
+            ...(turnId ? { turnId } : {}),
+            ...(toolCallId ? { toolCallId } : {}),
+        },
+    };
+}
 export function registerGenerations(program) {
     program
         .command("generate")
@@ -184,13 +224,29 @@ Examples:
             content.push(...await Promise.all(opts.image.map((value) => contentFromPathOrUrl("image", value))));
             content.push(...await Promise.all(opts.video.map((value) => contentFromPathOrUrl("video", value))));
             content.push(...await Promise.all(opts.audio.map((value) => contentFromPathOrUrl("audio", value))));
+            const parameters = parseParams(opts.param, opts.parameters);
+            try {
+                assertGenerationRequestAllowedByPolicy({
+                    policy: parseGenerationPolicyFromEnv(process.env),
+                    model: opts.model,
+                    parameters,
+                });
+            }
+            catch (policyError) {
+                if (policyError instanceof GenerationPolicyError)
+                    return error("Generation settings", policyError.message);
+                throw policyError;
+            }
+            const metadata = mergeCohubMetadata(parseMetadata(opts.metadata));
             const client = createClient();
             const created = await client.generations.create({
                 spaceId,
+                sessionId: envValue("COHUB_SESSION_ID"),
+                turnId: envValue("COHUB_TURN_ID"),
                 model: opts.model,
                 content,
-                parameters: parseParams(opts.param, opts.parameters),
-                metadata: opts.metadata ? JSON.parse(opts.metadata) : undefined,
+                parameters,
+                metadata,
             });
             if (opts.async) {
                 if (jsonRequested(opts))

package/dist/commands/models.js

@@ -1,3 +1,4 @@
+import { filterGenerationDeclarationsByPolicy, parseGenerationPolicyFromEnv, } from "@neta-art/cohub";
 import { createClient } from "../client.js";
 import { table, json as outJson, jsonRequested, error, handleHttp } from "../output.js";
 function toMultimodalModelSummary(model) {
@@ -88,7 +89,8 @@ Examples:
         try {
             if (opts.modelType === "multimodal") {
                 const response = await client.models.listMultimodal();
-                const models = response.models.map(toMultimodalModelSummary);
+                const filtered = filterGenerationDeclarationsByPolicy(response.models, parseGenerationPolicyFromEnv(process.env));
+                const models = filtered.map(toMultimodalModelSummary);
                 if (jsonRequested(opts))
                     return outJson({ models });
                 table(models, [
@@ -127,7 +129,8 @@ Examples:
         const client = createClient();
         try {
             const response = await client.models.listMultimodal();
-            const model = response.models.find((item) => item.model === modelId);
+            const models = filterGenerationDeclarationsByPolicy(response.models, parseGenerationPolicyFromEnv(process.env));
+            const model = models.find((item) => item.model === modelId);
             if (!model) {
                 return error("Model not found", `No multimodal model named ${modelId}`);
             }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@neta-art/cohub-cli",
-  "version": "1.9.1",
+  "version": "1.10.0",
   "description": "CLI for Cohub — spaces, sessions, and agent collaboration.",
   "type": "module",
   "license": "UNLICENSED",
@@ -16,7 +16,7 @@
     "@neta-art/generation": "^0.1.2",
     "commander": "^13.1.0",
     "sharp": "^0.34.5",
-    "@neta-art/cohub": "1.17.0"
+    "@neta-art/cohub": "1.18.0"
   },
   "publishConfig": {
     "access": "public"