@yeaft/webchat-agent 1.0.422 → 1.0.424

package/local-runtime/server/handlers/agent-file-terminal.js

@@ -137,6 +137,17 @@ async function handleTerminalResponse(agentId, msg, routeKey) {
   }
 }
 
+async function handleAgentDirectoryPickerResponse(agentId, msg) {
+  const pending = consumeWorkbenchRequest({
+    agentId,
+    requestId: msg._workbenchRequestId,
+    responseType: msg.type,
+    routeKey: `agent-directory-picker:${agentId}`,
+  });
+  if (!pending || pending.requestType !== 'agent_directory_picker') return;
+  await sendToPendingClient(agentId, msg, pending);
+}
+
 async function handleOneShotResponse(agentId, msg, routeKey) {
   const pending = consumeWorkbenchRequest({
     agentId,
@@ -186,6 +197,13 @@ export async function handleAgentFileTerminal(agentId, agent, rawMsg) {
     return true;
   }
 
+  if (msg.type === 'directory_listing'
+      && msg.conversationId === '_workdir_picker'
+      && msg._workbenchRequestId) {
+    await handleAgentDirectoryPickerResponse(agentId, msg);
+    return true;
+  }
+
   // `_workbench:` is reserved for Server-authored route conversations. An
   // invalid or cross-Agent value is not a legacy conversation.
   if (typeof msg.conversationId === 'string' && msg.conversationId.startsWith('_workbench:')) {

package/local-runtime/server/handlers/client-workbench.js

@@ -1,4 +1,5 @@
 import { CONFIG } from '../config.js';
+import { agents } from '../context.js';
 import {
   sendToWebClient, forwardToAgent,
   verifyConversationOwnership, getCachedDir
@@ -37,6 +38,8 @@ async function denyWorkbenchRoute(client, msg) {
   await sendToWebClient(client, { type: 'error', message: 'Invalid Workbench Session route' });
 }
 
+const AGENT_DIRECTORY_PICKER_CONVERSATION = '_workdir_picker';
+
 const RESPONSE_TYPES = Object.freeze({
   terminal_create: ['terminal_created', 'terminal_error'],
   read_file: ['file_content'],
@@ -248,6 +251,47 @@ export async function handleClientWorkbench(clientId, client, msg, checkAgentAcc
       if (!dirAgentId) return;
       if (!await checkAgentAccess(dirAgentId)) return;
 
+      const isAgentDirectoryPicker = msg.directoryPickerScope === 'agent'
+        && msg.conversationId === AGENT_DIRECTORY_PICKER_CONVERSATION
+        && !msg.workbenchRoute
+        && typeof msg.requestId === 'string'
+        && msg.requestId.length > 0;
+      if (isAgentDirectoryPicker) {
+        const agent = agents.get(dirAgentId);
+        const defaultWorkDir = typeof agent?.workDir === 'string' ? agent.workDir : '';
+        const correlationKey = `agent-directory-picker:${dirAgentId}`;
+        const internalRequestId = registerWorkbenchRequest({
+          agentId: dirAgentId,
+          clientId,
+          userId: client.userId,
+          routeKey: correlationKey,
+          conversationId: AGENT_DIRECTORY_PICKER_CONVERSATION,
+          workspaceGeneration: correlationKey,
+          route: null,
+          role: client.role,
+          requestType: 'agent_directory_picker',
+          expectedResponseTypes: ['directory_listing'],
+          publicRequestId: msg.requestId,
+        });
+        if (!internalRequestId) return;
+        const canonical = {
+          type: 'list_directory',
+          agentId: dirAgentId,
+          conversationId: AGENT_DIRECTORY_PICKER_CONVERSATION,
+          directoryPickerScope: 'agent',
+          dirPath: typeof msg.dirPath === 'string' ? msg.dirPath : defaultWorkDir,
+          workDir: defaultWorkDir,
+          _workbenchRequestId: internalRequestId,
+        };
+        try {
+          await forwardToAgent(dirAgentId, canonical);
+        } catch (error) {
+          deleteWorkbenchRequest({ agentId: dirAgentId, requestId: internalRequestId });
+          throw error;
+        }
+        return;
+      }
+
       const resolved = resolveWorkbenchRequest(client, msg, dirAgentId);
       if (!resolved) {
         await denyWorkbenchRoute(client, msg);

package/local-runtime/version.json

@@ -1 +1 @@
-{"version":"1.0.422"}
+{"version":"1.0.424"}