viagen 0.0.46 → 0.0.48

package/dist/index.js

@@ -110,7 +110,7 @@ function registerHealthRoutes(server, env, errorRef) {
       );
     }
   });
-  const currentVersion = true ? "0.0.46" : "0.0.0";
+  const currentVersion = true ? "0.0.48" : "0.0.0";
   debug("health", `version resolved: ${currentVersion}`);
   let versionCache = null;
   server.middlewares.use("/via/version", (_req, res) => {
@@ -2455,16 +2455,14 @@ function buildUiHtml(opts) {
   <div class="header">
     <div style="display:flex;align-items:center;gap:0;">
       <h1><span class="status-dot" id="status-dot"></span> via <span class="session-timer" id="session-timer"></span></h1>
-      <div style="display:flex;gap:4px;align-items:center;">
-        <button class="btn icon-btn" id="view-preview" title="Preview app" style="display:none;">
-          <svg xmlns="http://www.w3.org/2000/svg" width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="1.5" stroke-linecap="round" stroke-linejoin="round"><path d="M18 13v6a2 2 0 0 1-2 2H5a2 2 0 0 1-2-2V8a2 2 0 0 1 2-2h6"></path><polyline points="15 3 21 3 21 9"></polyline><line x1="10" y1="14" x2="21" y2="3"></line></svg>
-        </button>
-        <button class="btn icon-btn" id="view-toggle" title="Toggle view">
-          <svg id="view-toggle-icon" xmlns="http://www.w3.org/2000/svg" width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="1.5" stroke-linecap="round" stroke-linejoin="round"></svg>
-        </button>
-      </div>
     </div>
     <div style="display:flex;gap:4px;align-items:center;">
+      <button class="btn icon-btn" id="view-preview" title="Preview app" style="display:none;">
+        <svg xmlns="http://www.w3.org/2000/svg" width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="1.5" stroke-linecap="round" stroke-linejoin="round"><path d="M18 13v6a2 2 0 0 1-2 2H5a2 2 0 0 1-2-2V8a2 2 0 0 1 2-2h6"></path><polyline points="15 3 21 3 21 9"></polyline><line x1="10" y1="14" x2="21" y2="3"></line></svg>
+      </button>
+      <button class="btn icon-btn" id="view-toggle" title="Toggle view">
+        <svg id="view-toggle-icon" xmlns="http://www.w3.org/2000/svg" width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="1.5" stroke-linecap="round" stroke-linejoin="round"></svg>
+      </button>
       <button class="btn icon-btn" id="sound-btn" title="Toggle completion sound">
         <svg id="sound-icon" xmlns="http://www.w3.org/2000/svg" width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="1.5" stroke-linecap="round" stroke-linejoin="round"></svg>
       </button>
@@ -18411,39 +18409,116 @@ import {
   createSdkMcpServer,
   tool
 } from "@anthropic-ai/claude-agent-sdk";
-import { updateTask } from "viagen-sdk/sandbox";
-function createViagenTools() {
-  return createSdkMcpServer({
-    name: "viagen",
-    tools: [
+import {
+  updateTask,
+  listTasks,
+  getTask,
+  createTask
+} from "viagen-sdk/sandbox";
+function createViagenTools(config2) {
+  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.",
+      {
+        status: external_exports.enum(["review", "completed"]).describe(
+          "'review' = PR created, ready for review. 'completed' = task fully done."
+        ),
+        prUrl: external_exports.string().optional().describe("Full URL of the pull request, if one was created."),
+        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.")
+      },
+      async (args) => {
+        await updateTask({
+          status: args.status,
+          prUrl: args.prUrl,
+          result: args.result,
+          inputTokens: args.inputTokens,
+          outputTokens: args.outputTokens,
+          costUsd: args.costUsd
+        });
+        return {
+          content: [
+            {
+              type: "text",
+              text: `Task status updated to '${args.status}'.`
+            }
+          ]
+        };
+      }
+    )
+  ];
+  if (config2) {
+    tools.push(
+      tool(
+        "viagen_list_tasks",
+        "List tasks in the current project. Optionally filter by status.",
+        {
+          status: external_exports.enum(["ready", "running", "validating", "completed", "timed_out"]).optional().describe("Filter tasks by status.")
+        },
+        async (args) => {
+          const tasks = await listTasks({ status: args.status });
+          return {
+            content: [
+              {
+                type: "text",
+                text: JSON.stringify(tasks, null, 2)
+              }
+            ]
+          };
+        }
+      )
+    );
+    tools.push(
+      tool(
+        "viagen_get_task",
+        "Get full details of a specific task by ID.",
+        {
+          taskId: external_exports.string().describe("The task ID to retrieve.")
+        },
+        async (args) => {
+          const task = await getTask(args.taskId);
+          return {
+            content: [
+              {
+                type: "text",
+                text: JSON.stringify(task, null, 2)
+              }
+            ]
+          };
+        }
+      )
+    );
+    tools.push(
       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.",
+        "viagen_create_task",
+        "Create a new task in the current project. Use this to create follow-up work.",
         {
-          status: external_exports.enum(["review", "completed"]).describe(
-            "'review' = PR created, ready for review. 'completed' = task fully done."
-          ),
-          prUrl: external_exports.string().optional().describe("Full URL of the pull request, if one was created."),
-          result: external_exports.string().describe("Brief one-line summary of what was done.")
+          prompt: external_exports.string().describe("The task prompt / instructions."),
+          branch: external_exports.string().optional().describe("Git branch name for the task."),
+          type: external_exports.enum(["task", "plan"]).optional().describe("Task type: 'task' for code changes, 'plan' for implementation plans.")
         },
         async (args) => {
-          await updateTask({
-            status: args.status,
-            prUrl: args.prUrl,
-            result: args.result
+          const task = await createTask({
+            prompt: args.prompt,
+            branch: args.branch,
+            type: args.type
           });
           return {
             content: [
               {
                 type: "text",
-                text: `Task status updated to '${args.status}'.`
+                text: JSON.stringify(task, null, 2)
               }
             ]
           };
         }
       )
-    ]
-  });
+    );
+  }
+  return createSdkMcpServer({ name: "viagen", tools });
 }
 var PLAN_MODE_DISALLOWED_TOOLS = ["Edit", "NotebookEdit"];
 var planModeCanUseTool = async (toolName, input) => {
@@ -18472,6 +18547,15 @@ Constraints:
 - Only create new files inside the plans/ directory.
 - Your plan should include: context, proposed changes (with file paths and descriptions), implementation order, and potential risks.
 `;
+var TASK_TOOLS_PROMPT = `
+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')
+
+Use these to understand project context and create follow-up work when appropriate.
+`;
 
 // src/sandbox.ts
 import { randomUUID } from "crypto";
@@ -18865,9 +18949,12 @@ ${payload.err.frame || ""}`
       const resolvedModel = env["VIAGEN_MODEL"] || opts.model;
       debug("server", `creating ChatSession (model: ${resolvedModel})`);
       let mcpServers;
-      if (env["VIAGEN_CALLBACK_URL"] && env["VIAGEN_AUTH_TOKEN"] && env["VIAGEN_TASK_ID"]) {
+      const hasSandboxContext = !!(env["VIAGEN_CALLBACK_URL"] && env["VIAGEN_AUTH_TOKEN"] && env["VIAGEN_TASK_ID"]);
+      if (hasSandboxContext) {
         debug("server", "creating viagen MCP tools (sandbox mode)");
-        const viagenMcp = createViagenTools();
+        const viagenMcp = createViagenTools(
+          env["VIAGEN_PROJECT_ID"] ? { projectId: env["VIAGEN_PROJECT_ID"] } : void 0
+        );
         mcpServers = { [viagenMcp.name]: viagenMcp };
       }
       const isPlanMode = env["VIAGEN_TASK_TYPE"] === "plan";
@@ -18875,6 +18962,8 @@ ${payload.err.frame || ""}`
       if (isPlanMode) {
         debug("server", "plan mode active \u2014 restricting tools");
         systemPrompt = PLAN_SYSTEM_PROMPT;
+      } else if (hasSandboxContext && env["VIAGEN_PROJECT_ID"]) {
+        systemPrompt = (systemPrompt || "") + TASK_TOOLS_PROMPT;
       }
       const chatSession = new ChatSession({
         env,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "viagen",
-  "version": "0.0.46",
+  "version": "0.0.48",
   "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.8"
+    "viagen-sdk": "^0.0.10"
   },
   "license": "MIT",
   "repository": {