pi-orchestrator-config 4.2.0 → 4.3.2

package/extensions/coms/probe-socket.ts

@@ -0,0 +1,42 @@
+/**
+ * Socket probe for coms stale peer detection.
+ * Extracted for testability — the main coms-p2p module imports this.
+ */
+import * as net from "node:net";
+import { createLogger } from "../shared/logger.js";
+import { getSetting } from "../orchestrator/project-settings.js";
+
+const log = createLogger("coms");
+
+/**
+ * Probe a Unix socket endpoint to determine if it's in use or stale.
+ * Returns "in_use" if a connection succeeds, "stale" if refused/missing/timeout.
+ */
+export function probeStaleSocket(endpoint: string, name?: string, cwd = process.cwd()): Promise<"in_use" | "stale"> {
+	return new Promise((resolve) => {
+		const sock = net.createConnection({ path: endpoint });
+		let settled = false;
+		const finish = (verdict: "in_use" | "stale") => {
+			if (settled) return;
+			settled = true;
+			try { sock.destroy(); } catch { /* ignore */ }
+			resolve(verdict);
+		};
+		const timeoutMs = getSetting(cwd, "coms_probe_timeout_ms");
+		const timer = setTimeout(() => finish("stale"), timeoutMs);
+		log.debug("probe_stale", name ?? endpoint, "timeout_ms", timeoutMs);
+		sock.once("connect", () => {
+			clearTimeout(timer);
+			finish("in_use");
+		});
+		sock.once("error", (err: any) => {
+			clearTimeout(timer);
+			if (err && (err.code === "ECONNREFUSED" || err.code === "ENOENT")) {
+				finish("stale");
+			} else {
+				// Transient errors (EMFILE, EACCES, etc.) — treat as live to avoid false pruning
+				finish("in_use");
+			}
+		});
+	});
+}

package/extensions/coms/themeMap.ts

@@ -26,7 +26,6 @@ export const THEME_MAP: Record<string, string> = {
 	"agent-chain":        "midnight-ocean",   // deep sequential pipeline
 	"agent-team":         "dracula",          // rich orchestration palette
 	"coms":               "ocean-breeze",     // peer-to-peer messaging, cross-boundary
-	"coms-net":           "ocean-breeze",     // peer-to-peer messaging, cross-boundary
 	"cross-agent":        "ocean-breeze",     // cross-boundary, connecting
 	"damage-control":     "gruvbox",          // grounded, earthy safety
 	"minimal":            "synthwave",        // synthwave by default now!

package/extensions/image-gen/image-gen.ts

@@ -9,6 +9,8 @@
 import { Type } from "typebox";
 import type { ToolDefinition } from "@earendil-works/pi-coding-agent";
 import { getSetting } from "../orchestrator/project-settings.js";
+import { isImageGenModelId } from "../orchestrator/settings-tui-helpers.js";
+import { createLogger } from "../shared/logger.js";
 import * as fs from "node:fs";
 import * as path from "node:path";
 import { execFileSync, spawn } from "node:child_process";
@@ -16,6 +18,7 @@ import { fileURLToPath } from "node:url";
 
 const ASPECT_RATIOS = ["1:1", "3:4", "4:3", "9:16", "16:9"] as const;
 const IMAGE_TIMEOUT = 3 * 60 * 1000; // 3 minutes
+const log = createLogger("image_gen");
 
 const imageGenSchema = Type.Object({
     subject: Type.String({ description: "Main subject of the image" }),
@@ -51,7 +54,13 @@ function getApiKey(): string | null {
 
 function getModel(cwd: string): string | null {
     const model = getSetting(cwd, "image_model");
-    return model || null;
+    const id = typeof model === "string" ? model.trim() : "";
+    if (!id) {
+        log.debug("getModel", "image_model unset");
+        return null;
+    }
+    log.debug("getModel", id);
+    return id;
 }
 
 function assemblePrompt(params: {
@@ -166,6 +175,15 @@ export function createImageGenTool(): ToolDefinition {
                     }],
                 };
             }
+            if (!isImageGenModelId(model)) {
+                log.warn("generate_image rejected non-image model", model);
+                return {
+                    content: [{
+                        type: "text" as const,
+                        text: `Error: image_model '${model}' is not an image-generation model. Choose a Gemini image model (id contains image or imagen).`,
+                    }],
+                };
+            }
 
             const apiKey = getApiKey();
             if (!apiKey) {