staklink 0.4.20 → 0.4.22

package/dist/proxy-server.cjs

@@ -60927,7 +60927,7 @@ var SSEManager = class {
 var sseManager = new SSEManager();
 
 // src/proxy/version.ts
-var VERSION = "0.4.20";
+var VERSION = "0.4.22";
 
 // node_modules/uuid/dist/esm/stringify.js
 var byteToHex = [];
@@ -137626,6 +137626,9 @@ function sanitizeShellArg2(arg) {
 }
 
 // src/agent/goose.ts
+function isStreamDecodeError(output) {
+  return output.includes("Stream decode error: error decoding response body.") && output.includes("Please retry if you think this is a transient or recoverable error.");
+}
 var MODELS2 = {
   sonnet: GooseModels["claude-sonnet-4-6"],
   opus: GooseModels["claude-opus-4-6"],
@@ -137645,6 +137648,27 @@ function goose_env(apiKey, model = "sonnet") {
     PATH: `${path7.join(os.homedir(), ".local", "bin")}:${process.env.PATH}`
   };
 }
+async function extractSessionUsage(session) {
+  try {
+    const safeName = session.replaceAll(" ", "_");
+    const res = await executeCommand(
+      `goose session export --name ${safeName} --format json`,
+      process.cwd(),
+      {}
+    );
+    const data = JSON.parse(res.stdout);
+    if (typeof data.input_tokens === "number" && typeof data.output_tokens === "number") {
+      return {
+        inputTokens: data.input_tokens,
+        outputTokens: data.output_tokens,
+        totalTokens: data.total_tokens ?? data.input_tokens + data.output_tokens
+      };
+    }
+    return void 0;
+  } catch {
+    return void 0;
+  }
+}
 async function runAgent({
   prompt,
   apiKey,
@@ -137679,14 +137703,38 @@ ${prompt}` : prompt;
     } catch {
     }
   }
-  const res = await executeCommand(cmd, cwd, env2, (l) => console.log(l));
+  let res = await executeCommand(cmd, cwd, env2, (l) => console.log(l));
+  if (session && isStreamDecodeError(res.stdout)) {
+    const maxRetries = 2;
+    for (let attempt = 1; attempt <= maxRetries; attempt++) {
+      console.log(`Stream decode error detected, retrying (${attempt}/${maxRetries})...`);
+      await new Promise((r) => setTimeout(r, 2e3 * attempt));
+      const retryCmd = `goose run --with-builtin developer -t "continue" --system "${system}" --name ${session} --resume`;
+      res = await executeCommand(retryCmd, cwd, env2, (l) => console.log(l));
+      if (!isStreamDecodeError(res.stdout)) break;
+    }
+  }
   const output = res.stdout;
+  const resolvedModel = getGooseModel(model || "sonnet");
+  const provider = resolvedModel.includes("/") ? resolvedModel.split("/")[0] : "unknown";
+  const usage = session ? await extractSessionUsage(session) : void 0;
   if (summarize && session) {
     const sessionData = exportSession(session, "assistant");
     const summary = await summarizeSession(sessionData, summaryApiKey || apiKey);
-    return { output, summary };
+    return {
+      output,
+      summary,
+      model: resolvedModel,
+      provider,
+      ...usage && { usage }
+    };
   }
-  return output;
+  return {
+    output,
+    model: resolvedModel,
+    provider,
+    ...usage && { usage }
+  };
 }
 function createGooseConfig(exaApiKey, mcpServers) {
   const configDir = path7.join(os.homedir(), ".config", "goose");
@@ -137804,7 +137852,18 @@ async function runAgent2({
       cwd,
       executablePath: findClaudePath()
     });
-    return res.text;
+    return {
+      output: res.text,
+      model: "claude_code",
+      provider: "anthropic",
+      ...res.usage && {
+        usage: {
+          inputTokens: res.usage.inputTokens,
+          outputTokens: res.usage.outputTokens,
+          totalTokens: res.usage.totalTokens
+        }
+      }
+    };
   } catch (error88) {
     console.log("Model error:", error88);
     throw error88;
@@ -139311,15 +139370,20 @@ async function runAgent3({
     };
   }
   const model = codexCli(MODEL, settings);
-  const { text: text3 } = await generateText({
+  const { text: text3, usage } = await generateText({
     model,
     prompt,
     system
   });
+  const agentUsage = {
+    inputTokens: usage.inputTokens ?? 0,
+    outputTokens: usage.outputTokens ?? 0,
+    totalTokens: usage.totalTokens ?? 0
+  };
   if (summarize) {
-    return text3;
+    return { output: text3, model: MODEL, provider: "openai", usage: agentUsage };
   }
-  return text3;
+  return { output: text3, model: MODEL, provider: "openai", usage: agentUsage };
 }
 
 // src/agent/repair.ts
@@ -139985,7 +140049,11 @@ var createAsyncAgentHandler = (getParams, transformResult) => {
         const recordings = await collectRecordings(startTime);
         finishReq(request_id, {
           success: true,
-          result: finalResult,
+          result: finalResult.output ?? finalResult,
+          ...finalResult.summary && { summary: finalResult.summary },
+          ...finalResult.usage && { usage: finalResult.usage },
+          ...finalResult.model && { model: finalResult.model },
+          ...finalResult.provider && { provider: finalResult.provider },
           ...screenshots.length > 0 && { screenshots },
           ...recordings.length > 0 && { recordings }
         });
@@ -142691,7 +142759,7 @@ async function startProxyServer() {
           system: repair_exports.SYSTEM
         };
       },
-      (result) => repair_exports.parseXML(result)
+      (result) => repair_exports.parseXML(result.output)
     )
   );
   app.post(

package/dist/staklink-cli.cjs

@@ -11004,7 +11004,7 @@ var glob = Object.assign(glob_, {
 glob.glob = glob;
 
 // src/proxy/version.ts
-var VERSION = "0.4.20";
+var VERSION = "0.4.22";
 
 // src/deps.ts
 var import_child_process = require("child_process");

package/package.json

@@ -2,7 +2,7 @@
   "name": "staklink",
   "displayName": "staklink",
   "description": "staklink process manager",
-  "version": "0.4.20",
+  "version": "0.4.22",
   "type": "module",
   "publisher": "stakwork",
   "engines": {