viagen 0.0.47 → 0.0.49

package/dist/index.js

@@ -110,7 +110,7 @@ function registerHealthRoutes(server, env, errorRef) {
       );
     }
   });
-  const currentVersion = true ? "0.0.47" : "0.0.0";
+  const currentVersion = true ? "0.0.49" : "0.0.0";
   debug("health", `version resolved: ${currentVersion}`);
   let versionCache = null;
   server.middlewares.use("/via/version", (_req, res) => {
@@ -746,9 +746,38 @@ function registerChatRoutes(server, session, opts) {
         done = true;
         debug("chat", "SSE stream done, closing response");
         if (!res.writableEnded) {
-          res.write("event: done\ndata: {}\n\n");
+          const doneData = {
+            ...event.costUsd != null && { costUsd: event.costUsd },
+            ...event.inputTokens != null && { inputTokens: event.inputTokens },
+            ...event.outputTokens != null && { outputTokens: event.outputTokens },
+            ...event.durationMs != null && { durationMs: event.durationMs }
+          };
+          res.write(`event: done
+data: ${JSON.stringify(doneData)}
+
+`);
           res.end();
         }
+        if (opts.env["VIAGEN_CALLBACK_URL"] && opts.env["VIAGEN_AUTH_TOKEN"]) {
+          const taskId = opts.env["VIAGEN_TASK_ID"];
+          if (taskId && (event.inputTokens || event.outputTokens || event.costUsd)) {
+            fetch(opts.env["VIAGEN_CALLBACK_URL"], {
+              method: "POST",
+              headers: {
+                "Content-Type": "application/json",
+                Authorization: `Bearer ${opts.env["VIAGEN_AUTH_TOKEN"]}`
+              },
+              body: JSON.stringify({
+                taskId,
+                ...event.inputTokens != null && { inputTokens: event.inputTokens },
+                ...event.outputTokens != null && { outputTokens: event.outputTokens },
+                ...event.costUsd != null && { costUsd: event.costUsd }
+              })
+            }).catch((err) => {
+              debug("chat", `usage callback failed: ${err}`);
+            });
+          }
+        }
         return;
       }
       if (event.type === "error") {
@@ -18409,23 +18438,18 @@ import {
   createSdkMcpServer,
   tool
 } from "@anthropic-ai/claude-agent-sdk";
-import { updateTask } from "viagen-sdk/sandbox";
-import { createViagen } from "viagen-sdk";
+import {
+  listTasks,
+  getTask,
+  createTask
+} from "viagen-sdk/sandbox";
 function createViagenTools(config2) {
-  let client;
-  let projectId;
-  if (config2) {
-    client = createViagen({
-      baseUrl: config2.platformUrl,
-      token: config2.authToken
-    });
-    projectId = config2.projectId;
-  }
   const tools = [
     tool(
       "viagen_update_task",
-      "Report task status back to the viagen platform. Use status 'review' after creating a PR (ready for human review) or 'completed' when the task is fully done.",
+      "Update a task's status on the viagen platform. Use status 'review' after creating a PR (ready for human review) or 'completed' when the task is fully done. If no taskId is provided, updates the current task (requires VIAGEN_TASK_ID env var).",
       {
+        taskId: external_exports.string().optional().describe("Task ID to update. Defaults to the current task from VIAGEN_TASK_ID env var."),
         status: external_exports.enum(["review", "completed"]).describe(
           "'review' = PR created, ready for review. 'completed' = task fully done."
         ),
@@ -18433,31 +18457,49 @@ function createViagenTools(config2) {
         result: external_exports.string().describe("Brief one-line summary of what was done."),
         inputTokens: external_exports.number().optional().describe("Total input tokens used."),
         outputTokens: external_exports.number().optional().describe("Total output tokens used."),
-        costUsd: external_exports.number().optional().describe("Total cost in USD.")
+        costUsd: external_exports.number().optional().describe("Total cost in USD."),
+        prReviewStatus: external_exports.enum(["approved", "changes_requested", "commented"]).optional().describe("PR review outcome, if applicable.")
       },
       async (args) => {
-        await updateTask({
-          status: args.status,
-          prUrl: args.prUrl,
-          result: args.result,
-          inputTokens: args.inputTokens,
-          outputTokens: args.outputTokens,
-          costUsd: args.costUsd
+        const taskId = args.taskId || process.env["VIAGEN_TASK_ID"];
+        if (!taskId) {
+          return {
+            content: [{ type: "text", text: "Error: No taskId provided and VIAGEN_TASK_ID is not set." }]
+          };
+        }
+        const callbackUrl = process.env["VIAGEN_CALLBACK_URL"];
+        const authToken = process.env["VIAGEN_AUTH_TOKEN"];
+        const internalStatus = args.status === "review" ? "validating" : "completed";
+        const res = await fetch(callbackUrl, {
+          method: "POST",
+          headers: {
+            "Content-Type": "application/json",
+            Authorization: `Bearer ${authToken}`
+          },
+          body: JSON.stringify({
+            taskId,
+            status: internalStatus,
+            ...args.prUrl && { prUrl: args.prUrl },
+            result: args.result,
+            ...args.inputTokens != null && { inputTokens: args.inputTokens },
+            ...args.outputTokens != null && { outputTokens: args.outputTokens },
+            ...args.costUsd != null && { costUsd: args.costUsd },
+            ...args.prReviewStatus && { prReviewStatus: args.prReviewStatus }
+          })
         });
+        if (!res.ok) {
+          const text = await res.text().catch(() => "");
+          return {
+            content: [{ type: "text", text: `Error updating task (${res.status}): ${text}` }]
+          };
+        }
         return {
-          content: [
-            {
-              type: "text",
-              text: `Task status updated to '${args.status}'.`
-            }
-          ]
+          content: [{ type: "text", text: `Task ${taskId} status updated to '${args.status}'.` }]
         };
       }
     )
   ];
-  if (client && projectId) {
-    const c = client;
-    const pid = projectId;
+  if (config2?.projectId) {
     tools.push(
       tool(
         "viagen_list_tasks",
@@ -18466,7 +18508,7 @@ function createViagenTools(config2) {
           status: external_exports.enum(["ready", "running", "validating", "completed", "timed_out"]).optional().describe("Filter tasks by status.")
         },
         async (args) => {
-          const tasks = await c.tasks.list(pid, args.status);
+          const tasks = await listTasks({ status: args.status });
           return {
             content: [
               {
@@ -18486,7 +18528,7 @@ function createViagenTools(config2) {
           taskId: external_exports.string().describe("The task ID to retrieve.")
         },
         async (args) => {
-          const task = await c.tasks.get(pid, args.taskId);
+          const task = await getTask(args.taskId);
           return {
             content: [
               {
@@ -18508,7 +18550,7 @@ function createViagenTools(config2) {
           type: external_exports.enum(["task", "plan"]).optional().describe("Task type: 'task' for code changes, 'plan' for implementation plans.")
         },
         async (args) => {
-          const task = await c.tasks.create(pid, {
+          const task = await createTask({
             prompt: args.prompt,
             branch: args.branch,
             type: args.type
@@ -18559,7 +18601,7 @@ You have access to viagen platform tools for task management:
 - viagen_list_tasks: List tasks in this project (optionally filter by status)
 - viagen_get_task: Get full details of a specific task
 - viagen_create_task: Create follow-up tasks for work you identify
-- viagen_update_task: Report your current task status ('review' or 'completed')
+- viagen_update_task: Update a task's status ('review' or 'completed'). Accepts an optional taskId \u2014 defaults to the current task if one is set.
 
 Use these to understand project context and create follow-up work when appropriate.
 `;
@@ -18956,15 +18998,11 @@ ${payload.err.frame || ""}`
       const resolvedModel = env["VIAGEN_MODEL"] || opts.model;
       debug("server", `creating ChatSession (model: ${resolvedModel})`);
       let mcpServers;
-      const hasSandboxContext = !!(env["VIAGEN_CALLBACK_URL"] && env["VIAGEN_AUTH_TOKEN"] && env["VIAGEN_TASK_ID"]);
+      const hasSandboxContext = !!(env["VIAGEN_CALLBACK_URL"] && env["VIAGEN_AUTH_TOKEN"]);
       if (hasSandboxContext) {
         debug("server", "creating viagen MCP tools (sandbox mode)");
         const viagenMcp = createViagenTools(
-          env["VIAGEN_PROJECT_ID"] ? {
-            authToken: env["VIAGEN_AUTH_TOKEN"],
-            platformUrl: env["VIAGEN_PLATFORM_URL"] || "https://app.viagen.dev",
-            projectId: env["VIAGEN_PROJECT_ID"]
-          } : void 0
+          env["VIAGEN_PROJECT_ID"] ? { projectId: env["VIAGEN_PROJECT_ID"] } : void 0
         );
         mcpServers = { [viagenMcp.name]: viagenMcp };
       }
@@ -19007,6 +19045,24 @@ ${payload.err.frame || ""}`
           if (event.type === "done") {
             debug("server", "auto-prompt completed");
             logBuffer.push("info", `[viagen] Prompt completed`);
+            const taskId = env["VIAGEN_TASK_ID"];
+            if (hasSandboxContext && taskId && (event.inputTokens || event.outputTokens || event.costUsd)) {
+              fetch(env["VIAGEN_CALLBACK_URL"], {
+                method: "POST",
+                headers: {
+                  "Content-Type": "application/json",
+                  Authorization: `Bearer ${env["VIAGEN_AUTH_TOKEN"]}`
+                },
+                body: JSON.stringify({
+                  taskId,
+                  ...event.inputTokens != null && { inputTokens: event.inputTokens },
+                  ...event.outputTokens != null && { outputTokens: event.outputTokens },
+                  ...event.costUsd != null && { costUsd: event.costUsd }
+                })
+              }).catch((err) => {
+                debug("server", `usage callback failed: ${err}`);
+              });
+            }
           }
         });
       }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "viagen",
-  "version": "0.0.47",
+  "version": "0.0.49",
   "description": "Vite dev server plugin that exposes endpoints for chatting with Claude Code SDK",
   "type": "module",
   "main": "./dist/index.js",
@@ -44,7 +44,7 @@
     "@vercel/sandbox": "^1",
     "lucide-react": "^0.564.0",
     "simple-git": "^3.31.1",
-    "viagen-sdk": "^0.0.9"
+    "viagen-sdk": "^0.0.11"
   },
   "license": "MIT",
   "repository": {