@papercraneai/sandbox-agent 0.1.10 → 0.1.12

package/dist/index.js

@@ -373,6 +373,8 @@ const app = express();
 app.use(express.json());
 const PORT = cliArgs.agentPort;
 let PROJECT_DIR = getProjectDir(); // Will be updated after template setup
+// Per-request UI context for GetContext tool (keyed by requestId, cleaned up after each turn)
+const sessionContext = {};
 // Registration state
 let environmentId = null;
 let connectionToken = null;
@@ -390,10 +392,17 @@ const showPreviewTool = tool("ShowPreview", "Shows the preview iframe for a spec
 });
 // Factory to create a fresh MCP server per chat session
 // (each Protocol instance can only be connected to one transport at a time)
-function createClientToolsServer() {
+// contextKey links to the per-request entry in sessionContext for GetContext
+function createClientToolsServer(contextKey) {
+    const getContextTool = tool("GetContext", "Returns the current UI context — which dashboard the user is viewing in the preview panel. Use this when the user's request is ambiguous about which dashboard they're referring to.", {}, async () => ({
+        content: [{
+                type: "text",
+                text: JSON.stringify(sessionContext[contextKey] || { selectedDashboard: null })
+            }]
+    }));
     return createSdkMcpServer({
         name: "client-tools",
-        tools: [showPreviewTool]
+        tools: [showPreviewTool, getContextTool]
     });
 }
 // Recursively build file tree
@@ -1125,7 +1134,7 @@ const buildHooks = (res, verbose, requestId, requestStartTime) => {
 app.post("/chat", async (req, res) => {
     const requestStartTime = Date.now();
     const requestId = Math.random().toString(36).substring(7);
-    const { message, sessionId, systemPrompt, verbose = false, subdir, 
+    const { message, sessionId, systemPrompt, verbose = false, subdir, selectedDashboard, 
     // Configurable agent options
     maxTurns = 40, allowedTools, disallowedTools, model, maxBudgetUsd } = req.body;
     const ctx = { requestId, sessionId };
@@ -1187,9 +1196,12 @@ app.post("/chat", async (req, res) => {
         "WebSearch",
         "AgentOutput",
         "KillShell",
-        "mcp__client-tools__ShowPreview"
+        "mcp__client-tools__ShowPreview",
+        "mcp__client-tools__GetContext"
     ];
-    const clientTools = createClientToolsServer();
+    // Store UI context for GetContext tool
+    sessionContext[requestId] = { selectedDashboard: selectedDashboard || null };
+    const clientTools = createClientToolsServer(requestId);
     const options = {
         maxTurns,
         cwd,
@@ -1238,7 +1250,21 @@ app.post("/chat", async (req, res) => {
                 const resultWithModel = { ...msg, model: usedModel };
                 res.write(`data: ${JSON.stringify(resultWithModel)}\n\n`);
                 gotResult = true;
-                log.info({ ...ctx, elapsed: Date.now() - requestStartTime }, "Query complete");
+                const baseLogCtx = {
+                    ...ctx,
+                    elapsed: Date.now() - requestStartTime,
+                    subtype: msg.subtype,
+                    num_turns: msg.num_turns,
+                    is_error: msg.is_error,
+                    cost_usd: msg.total_cost_usd
+                };
+                if (msg.is_error) {
+                    const errMsg = msg;
+                    log.warn({ ...baseLogCtx, errors: errMsg.errors }, `Query ended with error: ${msg.subtype}`);
+                }
+                else {
+                    log.info(baseLogCtx, "Query complete");
+                }
                 res.write("data: [DONE]\n\n");
                 res.end();
                 return;
@@ -1261,6 +1287,7 @@ app.post("/chat", async (req, res) => {
         const isAbortError = error instanceof Error &&
             (error.name === "AbortError" || error.constructor.name === "AbortError");
         if (isAbortError || isAborted) {
+            log.info(ctx, "Chat request aborted by client");
             if (!res.writableEnded) {
                 res.write(`data: ${JSON.stringify({ type: "aborted" })}\n\n`);
                 res.write("data: [DONE]\n\n");
@@ -1268,17 +1295,23 @@ app.post("/chat", async (req, res) => {
             }
             return;
         }
-        log.error({ ...ctx, error: error instanceof Error ? error.message : String(error) }, "Agent error");
+        log.error({
+            ...ctx,
+            err: error instanceof Error ? error.message : String(error),
+            stack: error instanceof Error ? error.stack : undefined
+        }, "Unhandled error in chat stream — stream terminated");
         if (!res.writableEnded) {
             res.write(`data: ${JSON.stringify({
                 type: "error",
                 error: error instanceof Error ? error.message : "Unknown error"
             })}\n\n`);
+            res.write("data: [DONE]\n\n");
             res.end();
         }
     }
     finally {
-        // Close the per-session MCP server to free resources
+        // Clean up per-request context and close MCP server
+        delete sessionContext[requestId];
         await clientTools.instance?.close().catch(() => { });
     }
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@papercraneai/sandbox-agent",
-  "version": "0.1.10",
+  "version": "0.1.12",
   "description": "Claude Agent SDK server for sandbox environments",
   "license": "MIT",
   "type": "module",