pi-web 0.9.0 → 0.10.0

package/build/server/server.js

@@ -299,6 +299,37 @@ function registerDiscoveredSessionKey(managed, event) {
     const key = buildSessionKey(managed.cwd, basename(sessionPath));
     registerManagedSessionKey(managed, key);
 }
+function deriveModelSupportsImages(model) {
+    if (!model || typeof model !== 'object')
+        return null;
+    const input = model.input;
+    if (!Array.isArray(input))
+        return null;
+    return input.includes('image');
+}
+function updateSessionModelCapability(managed, event) {
+    if (!event || typeof event !== 'object')
+        return;
+    if (event.type === 'model_changed') {
+        const supports = deriveModelSupportsImages(event.model);
+        if (supports != null)
+            managed.currentModelSupportsImages = supports;
+        return;
+    }
+    if (event.type === 'response') {
+        if (event.command === 'get_state') {
+            const supports = deriveModelSupportsImages(event.data?.model);
+            if (supports != null)
+                managed.currentModelSupportsImages = supports;
+            return;
+        }
+        if (event.command === 'set_model' && event.success) {
+            const supports = deriveModelSupportsImages(event.data);
+            if (supports != null)
+                managed.currentModelSupportsImages = supports;
+        }
+    }
+}
 function createManagedSession(cwd, sessionFile) {
     const sessionPath = sessionFile ? getSessionFilePath(cwd, sessionFile, AGENT) : undefined;
     let managed = null;
@@ -317,6 +348,7 @@ function createManagedSession(cwd, sessionFile) {
                 managed.isAgentRunning = false;
                 cleanupIfIdle(managed);
             }
+            updateSessionModelCapability(managed, event);
             registerDiscoveredSessionKey(managed, event);
             broadcast(managed, { type: 'rpc_event', event });
         },
@@ -346,6 +378,7 @@ function createManagedSession(cwd, sessionFile) {
         rpc,
         clients: new Set(),
         isAgentRunning: false,
+        currentModelSupportsImages: null,
         keys: new Set(),
         idleCleanupTimer: null,
         isClosing: false,
@@ -394,7 +427,19 @@ wss.on('connection', (ws) => {
                 sendToSocket(ws, { type: 'error', message: 'no active session' });
                 return;
             }
-            managed.rpc.send(msg.command);
+            const command = msg.command;
+            const isPromptLikeCommand = command?.type === 'prompt' || command?.type === 'steer' || command?.type === 'follow_up';
+            const hasImages = Array.isArray(command?.images) && command.images.length > 0;
+            if (isPromptLikeCommand &&
+                hasImages &&
+                managed.currentModelSupportsImages === false) {
+                sendToSocket(ws, {
+                    type: 'error',
+                    message: 'selected model does not support file attachments',
+                });
+                return;
+            }
+            managed.rpc.send(command);
             return;
         }
         if (msg.type === 'ping') {