staklink 0.4.21 → 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.21";
+var VERSION = "0.4.22";
 
 // node_modules/uuid/dist/esm/stringify.js
 var byteToHex = [];
@@ -137648,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,
@@ -137694,12 +137715,26 @@ ${prompt}` : prompt;
     }
   }
   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");
@@ -137817,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;
@@ -139324,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
@@ -139998,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 }
         });
@@ -142704,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.21";
+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.21",
+  "version": "0.4.22",
   "type": "module",
   "publisher": "stakwork",
   "engines": {