viagen 0.0.55 → 0.0.57

package/README.md

@@ -86,10 +86,29 @@ viagen({
   sandboxFiles: [...],       // copy files manually into sandbox
   systemPrompt: '...',       // custom system prompt (see below)
   editable: ['src','conf'],  // files/dirs editable in the UI
+  mcpServers: { ... },       // additional MCP servers for Claude
 })
 ```
 
 
+### Custom MCP Servers
+
+Pass additional [MCP server](https://modelcontextprotocol.io) configurations to give Claude access to custom tools:
+
+```ts
+viagen({
+  mcpServers: {
+    'my-db': {
+      command: 'npx',
+      args: ['-y', '@my-org/db-mcp-server'],
+      env: { DATABASE_URL: process.env.DATABASE_URL },
+    },
+  },
+})
+```
+
+These are merged with viagen's built-in platform tools (when connected). User-provided servers take precedence if names collide.
+
 ### Editable Files
 
 Add a file editor panel to the chat UI:

package/dist/cli.js

@@ -1144,7 +1144,7 @@ async function sandbox(args, options) {
   }
 }
 var SYNC_DENY_KEYS = /* @__PURE__ */ new Set([
-  "VIAGEN_PROJECT_ID",
+  "VIAGEN_ENVIRONMENT_ID",
   "VIAGEN_PLATFORM_URL",
   "INFISICAL_CLIENT_ID",
   "INFISICAL_CLIENT_SECRET",
@@ -1165,45 +1165,45 @@ async function sync() {
   const env = loadDotenv(cwd);
   console.log("viagen sync");
   console.log("");
-  let projectId = env["VIAGEN_PROJECT_ID"];
-  let projectName;
-  if (projectId) {
+  let environmentId = env["VIAGEN_ENVIRONMENT_ID"];
+  let environmentName;
+  if (environmentId) {
     try {
-      const existing = await client.projects.get(projectId);
-      projectName = existing.name;
-      console.log(`Project: ${projectName}`);
+      const existing = await client.environments.get(environmentId);
+      environmentName = existing.name;
+      console.log(`Environment: ${environmentName}`);
     } catch {
-      console.log("Previously synced project not found. Choose a project:");
+      console.log("Previously synced environment not found. Choose an environment:");
       console.log("");
-      projectId = void 0;
+      environmentId = void 0;
     }
   }
-  if (!projectId) {
-    const projects2 = await client.projects.list();
-    if (projects2.length > 0) {
-      console.log("Your projects:");
+  if (!environmentId) {
+    const environments2 = await client.environments.list();
+    if (environments2.length > 0) {
+      console.log("Your environments:");
       console.log("");
-      for (let i = 0; i < projects2.length; i++) {
-        const repo = projects2[i].githubRepo ? ` (${projects2[i].githubRepo})` : "";
-        console.log(`  ${i + 1}) ${projects2[i].name}${repo}`);
+      for (let i = 0; i < environments2.length; i++) {
+        const repo = environments2[i].githubRepo ? ` (${environments2[i].githubRepo})` : "";
+        console.log(`  ${i + 1}) ${environments2[i].name}${repo}`);
       }
-      console.log(`  ${projects2.length + 1}) Create new project`);
+      console.log(`  ${environments2.length + 1}) Create new environment`);
       console.log("");
       const choice = await promptUser("Choose: ");
       const idx = parseInt(choice, 10) - 1;
-      if (idx >= 0 && idx < projects2.length) {
-        projectId = projects2[idx].id;
-        projectName = projects2[idx].name;
+      if (idx >= 0 && idx < environments2.length) {
+        environmentId = environments2[idx].id;
+        environmentName = environments2[idx].name;
       } else {
-        projectName = await promptUser("Project name: ");
-        if (!projectName) {
+        environmentName = await promptUser("Environment name: ");
+        if (!environmentName) {
           console.log("Cancelled.");
           return;
         }
       }
     } else {
-      projectName = await promptUser("Project name: ");
-      if (!projectName) {
+      environmentName = await promptUser("Environment name: ");
+      if (!environmentName) {
         console.log("Cancelled.");
         return;
       }
@@ -1249,23 +1249,21 @@ async function sync() {
   const gitInfo = getGitInfo(cwd);
   const gitRemote = env["GIT_REMOTE_URL"] || gitInfo?.remoteUrl;
   const githubRepo = gitRemote ? repoNwo(gitRemote) : void 0;
-  const gitBranch = env["GIT_BRANCH"] || gitInfo?.branch;
   console.log("");
   console.log("Syncing...");
   const vercelProjectId = env["VERCEL_PROJECT_ID"];
   const vercelOrgId = env["VERCEL_ORG_ID"];
   const syncInput = {
-    ...projectId ? { id: projectId } : {},
-    name: projectName,
+    ...environmentId ? { id: environmentId } : {},
+    name: environmentName,
     ...githubRepo ? { githubRepo } : {},
-    ...gitBranch ? { gitBranch } : {},
     ...vercelProjectId ? { vercelProjectId } : {},
     ...vercelOrgId ? { vercelOrgId } : {},
     secrets
   };
   let result;
   try {
-    result = await client.projects.sync(syncInput);
+    result = await client.environments.sync(syncInput);
   } catch (err) {
     const logDir = join2(cwd, ".viagen");
     const logFile = join2(logDir, "sync-error.log");
@@ -1288,11 +1286,11 @@ async function sync() {
     console.error(`  ${logFile}`);
     process.exit(1);
   }
-  if (!env["VIAGEN_PROJECT_ID"]) {
-    writeEnvVars(cwd, { VIAGEN_PROJECT_ID: result.project.id });
+  if (!env["VIAGEN_ENVIRONMENT_ID"]) {
+    writeEnvVars(cwd, { VIAGEN_ENVIRONMENT_ID: result.app.id });
   }
   console.log(
-    `Synced "${result.project.name}" \u2014 ${result.secrets.stored} secrets stored.`
+    `Synced "${result.app.name}" \u2014 ${result.secrets.stored} secrets stored.`
   );
   if (result.secrets.failed && result.secrets.failed.length > 0) {
     console.log(`  Failed (${result.secrets.failed.length}): ${result.secrets.failed.join(", ")}`);
@@ -1389,42 +1387,42 @@ async function requireClient() {
   }
   return createViagen({ baseUrl: creds.baseUrl, token: creds.token, orgId: creds.orgId });
 }
-async function requireProjectId(client) {
+async function requireEnvironmentId(client) {
   const cwd = process.cwd();
   const env = loadDotenv(cwd);
-  let projectId = env["VIAGEN_PROJECT_ID"];
-  if (projectId) {
+  let environmentId = env["VIAGEN_ENVIRONMENT_ID"];
+  if (environmentId) {
     try {
-      await client.projects.get(projectId);
-      return projectId;
+      await client.environments.get(environmentId);
+      return environmentId;
     } catch {
-      console.log("Project from .env not found. Choose a project:");
+      console.log("Environment from .env not found. Choose an environment:");
       console.log("");
-      projectId = void 0;
+      environmentId = void 0;
     }
   }
-  const projects2 = await client.projects.list();
-  if (projects2.length === 0) {
-    console.error("No projects. Create one with `viagen projects create <name>` or `viagen sync`.");
+  const environments2 = await client.environments.list();
+  if (environments2.length === 0) {
+    console.error("No environments found. Create one with `viagen environments create <name>` or `viagen sync`.");
     process.exit(1);
   }
-  if (projects2.length === 1) {
-    return projects2[0].id;
+  if (environments2.length === 1) {
+    return environments2[0].id;
   }
-  console.log("Select a project:");
+  console.log("Select an environment:");
   console.log("");
-  for (let i = 0; i < projects2.length; i++) {
-    const repo = projects2[i].githubRepo ? ` (${projects2[i].githubRepo})` : "";
-    console.log(`  ${i + 1}) ${projects2[i].name}${repo}`);
+  for (let i = 0; i < environments2.length; i++) {
+    const repo = environments2[i].githubRepo ? ` (${environments2[i].githubRepo})` : "";
+    console.log(`  ${i + 1}) ${environments2[i].name}${repo}`);
   }
   console.log("");
   const choice = await promptUser("Choose: ");
   const idx = parseInt(choice, 10) - 1;
-  if (idx < 0 || idx >= projects2.length) {
+  if (idx < 0 || idx >= environments2.length) {
     console.error("Invalid selection.");
     process.exit(1);
   }
-  return projects2[idx].id;
+  return environments2[idx].id;
 }
 async function teams() {
   const client = await requireClient();
@@ -1490,56 +1488,56 @@ async function orgs(args) {
     console.log(`  ${m.name}${role}`);
   }
 }
-async function projects(args) {
+async function environments(args) {
   const sub = args[0];
   if (sub === "create") {
     const name = args.slice(1).join(" ");
     if (!name) {
-      console.error("Usage: viagen projects create <name>");
+      console.error("Usage: viagen environments create <name>");
       process.exit(1);
     }
     const client2 = await requireClient();
-    const project = await client2.projects.create({ name });
-    console.log(`Created project "${project.name}" (${project.id})`);
+    const environment = await client2.environments.create({ name });
+    console.log(`Created environment "${environment.name}" (${environment.id})`);
     return;
   }
   if (sub === "get") {
     const id = args[1];
     if (!id) {
-      console.error("Usage: viagen projects get <id>");
+      console.error("Usage: viagen environments get <id>");
       process.exit(1);
     }
     const client2 = await requireClient();
-    const project = await client2.projects.get(id);
-    console.log(`  Name:     ${project.name}`);
-    console.log(`  ID:       ${project.id}`);
-    if (project.githubRepo) console.log(`  GitHub:   ${project.githubRepo}`);
-    if (project.templateId) console.log(`  Template: ${project.templateId}`);
-    console.log(`  Created:  ${project.createdAt}`);
+    const environment = await client2.environments.get(id);
+    console.log(`  Name:     ${environment.name}`);
+    console.log(`  ID:       ${environment.id}`);
+    if (environment.githubRepo) console.log(`  GitHub:   ${environment.githubRepo}`);
+    if (environment.templateId) console.log(`  Template: ${environment.templateId}`);
+    console.log(`  Created:  ${environment.createdAt}`);
     return;
   }
   if (sub === "delete") {
     const id = args[1];
     if (!id) {
-      console.error("Usage: viagen projects delete <id>");
+      console.error("Usage: viagen environments delete <id>");
       process.exit(1);
     }
-    const answer = await promptUser(`Delete project ${id}? [y/n]: `);
+    const answer = await promptUser(`Delete environment ${id}? [y/n]: `);
     if (answer !== "y" && answer !== "yes") return;
     const client2 = await requireClient();
-    await client2.projects.delete(id);
-    console.log("Project deleted.");
+    await client2.environments.delete(id);
+    console.log("Environment deleted.");
     return;
   }
   const client = await requireClient();
-  const list = await client.projects.list() ?? [];
+  const list = await client.environments.list() ?? [];
   if (list.length === 0) {
-    console.log("No projects. Create one with `viagen projects create <name>`.");
+    console.log("No environments. Create one with `viagen environments create <name>`.");
     return;
   }
-  for (const p of list) {
-    const repo = p.githubRepo ? ` (${p.githubRepo})` : "";
-    console.log(`  ${p.name}${repo}  ${p.id}`);
+  for (const e of list) {
+    const repo = e.githubRepo ? ` (${e.githubRepo})` : "";
+    console.log(`  ${e.name}${repo}  ${e.id}`);
   }
 }
 function formatTaskStatus(status) {
@@ -1554,9 +1552,9 @@ function formatTaskStatus(status) {
 }
 async function tasksList(args) {
   const client = await requireClient();
-  const projectId = await requireProjectId(client);
+  const environmentId = await requireEnvironmentId(client);
   const status = parseFlag(args, "--status") || parseFlag(args, "-s");
-  const list = await client.tasks.list(projectId, status);
+  const list = await client.tasks.list(environmentId, status);
   if (list.length === 0) {
     console.log(
       status ? `No tasks with status "${status}".` : "No tasks. Create one with `viagen tasks create <prompt>`."
@@ -1602,11 +1600,11 @@ async function tasksCreate(args) {
     }
   }
   const client = await requireClient();
-  const projectId = await requireProjectId(client);
+  const environmentId = await requireEnvironmentId(client);
   const input = { prompt };
   if (branch) input.branch = branch;
   if (type) input.type = type;
-  const task = await client.tasks.create(projectId, input);
+  const task = await client.tasks.create(environmentId, input);
   console.log(`Task created: ${task.id}`);
   console.log(`  Prompt: ${task.prompt}`);
   console.log(`  Type:   ${task.type}`);
@@ -1622,8 +1620,8 @@ async function tasksGet(args) {
     process.exit(1);
   }
   const client = await requireClient();
-  const projectId = await requireProjectId(client);
-  const task = await client.tasks.get(projectId, taskId);
+  const environmentId = await requireEnvironmentId(client);
+  const task = await client.tasks.get(environmentId, taskId);
   console.log(`  ID:         ${task.id}`);
   console.log(`  Type:       ${task.type}`);
   console.log(`  Status:     ${formatTaskStatus(task.status)}`);
@@ -1654,13 +1652,13 @@ async function tasksRun(args) {
     process.exit(1);
   }
   const client = await requireClient();
-  const projectId = await requireProjectId(client);
-  const task = await client.tasks.get(projectId, taskId);
+  const environmentId = await requireEnvironmentId(client);
+  const task = await client.tasks.get(environmentId, taskId);
   console.log(`Running task: ${task.id}`);
   console.log(`  Prompt: ${task.prompt}`);
   console.log(`  Branch: ${task.branch}`);
   console.log("");
-  await client.tasks.update(projectId, taskId, { status: "running" });
+  await client.tasks.update(environmentId, taskId, { status: "running" });
   const sandboxArgs = [];
   if (task.branch) {
     sandboxArgs.push("--branch", task.branch);
@@ -1674,12 +1672,12 @@ async function tasksRun(args) {
       promptOverride: task.prompt,
       extraEnvVars: {
         VIAGEN_TASK_ID: task.id,
-        VIAGEN_PROJECT_ID: projectId
+        VIAGEN_ENVIRONMENT_ID: environmentId
       }
     });
   } catch (err) {
     console.error(`Sandbox deploy failed: ${err instanceof Error ? err.message : String(err)}`);
-    await client.tasks.update(projectId, taskId, { status: "ready" }).catch(() => {
+    await client.tasks.update(environmentId, taskId, { status: "ready" }).catch(() => {
     });
     process.exit(1);
   }
@@ -1714,11 +1712,11 @@ function help() {
   console.log("  orgs                           List your organizations");
   console.log("  orgs create <name>             Create a new organization");
   console.log("  orgs invite <email>            Invite a member to the org");
-  console.log("  projects                       List projects in current org");
-  console.log("  projects create <name>         Create a new project");
-  console.log("  projects get <id>              Show project details");
-  console.log("  projects delete <id>           Delete a project");
-  console.log("  tasks                          List tasks for current project");
+  console.log("  environments                   List environments in current org");
+  console.log("  environments create <name>     Create a new environment");
+  console.log("  environments get <id>          Show environment details");
+  console.log("  environments delete <id>       Delete an environment");
+  console.log("  tasks                          List tasks for current environment");
   console.log("  tasks create <prompt>          Create a new task");
   console.log("  tasks get <id>                 Show task details");
   console.log("  tasks run <id>                 Run a task in a sandbox");
@@ -1797,8 +1795,8 @@ async function main() {
     await teams();
   } else if (command === "orgs") {
     await orgs(args.slice(1));
-  } else if (command === "projects") {
-    await projects(args.slice(1));
+  } else if (command === "environments") {
+    await environments(args.slice(1));
   } else if (command === "tasks") {
     await tasks(args.slice(1));
   } else if (command === "dev") {

package/dist/index.d.ts

@@ -1,4 +1,5 @@
 import { Plugin } from 'vite';
+import { McpServerConfig } from '@anthropic-ai/claude-agent-sdk';
 
 declare const DEFAULT_SYSTEM_PROMPT = "\n  You are embedded in a Vite dev server as the \"viagen\" plugin.\n  Your job is to help build and modify the app. Files you edit will trigger Vite HMR automatically.\n  You can read .viagen/server.log to check recent Vite dev server output (compile errors, HMR updates, warnings).\n  Be concise.\n\n  You have agent-browser available for interacting with the running app in a real browser.\n  Use it to visually verify your changes, test interactions, and catch UI issues.\n  Key commands:\n    agent-browser open <url>          \u2014 open a page\n    agent-browser snapshot            \u2014 get accessibility tree with element refs (@e1, @e2, etc.)\n    agent-browser click @e<N>         \u2014 click an element by ref\n    agent-browser fill @e<N> \"value\"  \u2014 fill an input\n    agent-browser screenshot <file>   \u2014 capture a screenshot (you can view the image)\n    agent-browser diff snapshot --baseline <file> \u2014 compare page state before/after\n    agent-browser close               \u2014 close the browser\n  After making changes, consider using agent-browser to verify the result visually.\n";
 
@@ -97,6 +98,8 @@ interface ViagenOptions {
      * @example ['src/components', '.env', 'vite.config.ts']
      */
     editable?: string[];
+    /** Additional MCP servers to make available to Claude. Merged with built-in viagen tools. */
+    mcpServers?: Record<string, McpServerConfig>;
     /** Enable verbose debug logging. Also enabled by VIAGEN_DEBUG=1 in .env. */
     debug?: boolean;
 }

package/dist/index.js

@@ -100,17 +100,17 @@ function registerHealthRoutes(server, env, errorRef) {
     } : null;
     const prompt = env["VIAGEN_PROMPT"] || null;
     const taskId = env["VIAGEN_TASK_ID"] || null;
-    const projectId = env["VIAGEN_PROJECT_ID"] || null;
+    const environmentId = env["VIAGEN_ENVIRONMENT_ID"] || null;
     res.setHeader("Content-Type", "application/json");
     if (configured) {
-      res.end(JSON.stringify({ status: "ok", configured: true, git, vercel, branch, session, prompt, taskId, projectId, missing }));
+      res.end(JSON.stringify({ status: "ok", configured: true, git, vercel, branch, session, prompt, taskId, environmentId, missing }));
     } else {
       res.end(
-        JSON.stringify({ status: "error", configured: false, git, vercel, branch, session, prompt, taskId, projectId, missing })
+        JSON.stringify({ status: "error", configured: false, git, vercel, branch, session, prompt, taskId, environmentId, missing })
       );
     }
   });
-  const currentVersion = true ? "0.0.55" : "0.0.0";
+  const currentVersion = true ? "0.0.57" : "0.0.0";
   debug("health", `version resolved: ${currentVersion}`);
   let versionCache = null;
   server.middlewares.use("/via/version", (_req, res) => {
@@ -770,10 +770,10 @@ data: ${JSON.stringify(doneData)}
 `);
           res.end();
         }
-        if (opts.viagenClient && opts.projectId) {
+        if (opts.viagenClient && opts.environmentId) {
           const taskId = opts.env["VIAGEN_TASK_ID"];
           if (taskId && (event.inputTokens || event.outputTokens)) {
-            opts.viagenClient.tasks.update(opts.projectId, taskId, {
+            opts.viagenClient.tasks.update(opts.environmentId, taskId, {
               ...event.inputTokens != null && { inputTokens: event.inputTokens },
               ...event.outputTokens != null && { outputTokens: event.outputTokens }
             }).catch((err) => {
@@ -3798,7 +3798,7 @@ function buildUiHtml(opts) {
 
         // Store task context for history rendering
         healthTaskId = data.taskId || null;
-        healthProjectId = data.projectId || null;
+        healthProjectId = data.environmentId || null;
 
         // Load chat history from server (source of truth)
         await loadHistory();
@@ -3892,7 +3892,7 @@ function buildUiHtml(opts) {
         if (data.prompt && data.configured && chatLog.length === 0) {
           if (data.taskId) {
             // Task mode: show link instead of raw prompt
-            var taskUrl = 'https://app.viagen.dev/projects/' + data.projectId + '/tasks/' + data.taskId;
+            var taskUrl = 'https://app.viagen.dev/environments/' + data.environmentId + '/tasks/' + data.taskId;
             var div = document.createElement('div');
             div.className = 'msg msg-user';
             div.innerHTML = '<span class="label">Task</span><span class="text">Working on <a href="' + escapeHtml(taskUrl) + '" target="_blank" style="color:#2563eb;text-decoration:underline;">Via Task</a></span>';
@@ -18657,7 +18657,7 @@ import {
   tool
 } from "@anthropic-ai/claude-agent-sdk";
 function createViagenTools(config2) {
-  const { client, projectId } = config2;
+  const { client, environmentId } = config2;
   const taskId = process.env["VIAGEN_TASK_ID"];
   const tools = [
     tool(
@@ -18682,7 +18682,7 @@ function createViagenTools(config2) {
         }
         const internalStatus = args.status === "review" ? "validating" : args.status === "completed" ? "completed" : args.status;
         try {
-          await client.tasks.update(projectId, id, {
+          await client.tasks.update(environmentId, id, {
             ...internalStatus && { status: internalStatus },
             ...args.prUrl && { prUrl: args.prUrl },
             result: args.result,
@@ -18703,12 +18703,12 @@ function createViagenTools(config2) {
     ),
     tool(
       "viagen_list_tasks",
-      "List tasks in the current project. Optionally filter by status.",
+      "List tasks in the current environment. 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 client.tasks.list(projectId, args.status);
+        const tasks = await client.tasks.list(environmentId, args.status);
         return {
           content: [
             {
@@ -18726,7 +18726,7 @@ function createViagenTools(config2) {
         taskId: external_exports.string().describe("The task ID to retrieve.")
       },
       async (args) => {
-        const task = await client.tasks.get(projectId, args.taskId);
+        const task = await client.tasks.get(environmentId, args.taskId);
         return {
           content: [
             {
@@ -18739,14 +18739,14 @@ function createViagenTools(config2) {
     ),
     tool(
       "viagen_create_task",
-      "Create a new task in the current project. Use this to create follow-up work.",
+      "Create a new task in the current environment. Use this to create follow-up work.",
       {
         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) => {
-        const task = await client.tasks.create(projectId, {
+        const task = await client.tasks.create(environmentId, {
           prompt: args.prompt,
           branch: args.branch,
           type: args.type
@@ -18793,12 +18793,12 @@ Constraints:
 `;
 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_list_tasks: List tasks in this environment (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: 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.
+Use these to understand environment context and create follow-up work when appropriate.
 `;
 
 // src/index.ts
@@ -19182,7 +19182,7 @@ ${payload.err.frame || ""}`
       }
       const platformToken = env["VIAGEN_USER_TOKEN"] || env["VIAGEN_AUTH_TOKEN"];
       const platformUrl = env["VIAGEN_PLATFORM_URL"] || "https://app.viagen.dev";
-      const projectId = env["VIAGEN_PROJECT_ID"];
+      const environmentId = env["VIAGEN_ENVIRONMENT_ID"];
       let viagenClient = null;
       if (platformToken) {
         viagenClient = createViagen({ token: platformToken, baseUrl: platformUrl });
@@ -19222,9 +19222,9 @@ ${payload.err.frame || ""}`
             res.end(JSON.stringify({ error: "Method not allowed" }));
             return;
           }
-          if (!viagenClient || !projectId) {
+          if (!viagenClient || !environmentId) {
             res.writeHead(503, { "Content-Type": "application/json" });
-            res.end(JSON.stringify({ error: "Task creation not configured: missing platform credentials or VIAGEN_PROJECT_ID" }));
+            res.end(JSON.stringify({ error: "Task creation not configured: missing platform credentials or VIAGEN_ENVIRONMENT_ID" }));
             return;
           }
           let body = "";
@@ -19246,7 +19246,7 @@ Page URL: ${pageUrl}`);
               if (screenshot) parts.push("\n[Screenshot attached]");
               parts.push("\n\n[Submitted via viagen preview feedback]");
               const fullPrompt = parts.join("");
-              const task = await viagenClient.tasks.create(projectId, { prompt: fullPrompt, type: "task" });
+              const task = await viagenClient.tasks.create(environmentId, { prompt: fullPrompt, type: "task" });
               if (screenshot && task.id) {
                 try {
                   const [header, b64] = screenshot.split(",");
@@ -19254,7 +19254,7 @@ Page URL: ${pageUrl}`);
                   const binary = Buffer.from(b64, "base64");
                   const blob = new Blob([binary], { type: mime });
                   const ext = mime === "image/png" ? "png" : "jpeg";
-                  await viagenClient.tasks.addAttachment(projectId, task.id, blob, `screenshot.${ext}`);
+                  await viagenClient.tasks.addAttachment(environmentId, task.id, blob, `screenshot.${ext}`);
                   debug("preview", `screenshot attached to task ${task.id}`);
                 } catch (attachErr) {
                   debug("preview", `screenshot attachment failed (non-fatal): ${attachErr}`);
@@ -19277,15 +19277,19 @@ Page URL: ${pageUrl}`);
       const resolvedModel = env["VIAGEN_MODEL"] || opts.model;
       debug("server", `creating ChatSession (model: ${resolvedModel})`);
       let mcpServers;
-      const hasPlatformContext = !!(viagenClient && projectId);
+      const hasPlatformContext = !!(viagenClient && environmentId);
       if (hasPlatformContext) {
         debug("server", "creating viagen MCP tools (platform connected)");
         const viagenMcp = createViagenTools({
           client: viagenClient,
-          projectId
+          environmentId
         });
         mcpServers = { [viagenMcp.name]: viagenMcp };
       }
+      if (options?.mcpServers) {
+        mcpServers = { ...mcpServers, ...options.mcpServers };
+        debug("server", `merged ${Object.keys(options.mcpServers).length} user MCP server(s)`);
+      }
       const isPlanMode = env["VIAGEN_TASK_TYPE"] === "plan";
       let systemPrompt = options?.systemPrompt;
       if (isPlanMode) {
@@ -19310,7 +19314,7 @@ Page URL: ${pageUrl}`);
       registerChatRoutes(server, chatSession, {
         env,
         viagenClient: viagenClient ?? void 0,
-        projectId
+        environmentId
       });
       if (hasEditor) {
         registerFileRoutes(server, {
@@ -19330,8 +19334,8 @@ Page URL: ${pageUrl}`);
             debug("server", "auto-prompt completed");
             logBuffer.push("info", `[viagen] Prompt completed`);
             const currentTaskId = env["VIAGEN_TASK_ID"];
-            if (viagenClient && projectId && currentTaskId && (event.inputTokens || event.outputTokens)) {
-              viagenClient.tasks.update(projectId, currentTaskId, {
+            if (viagenClient && environmentId && currentTaskId && (event.inputTokens || event.outputTokens)) {
+              viagenClient.tasks.update(environmentId, currentTaskId, {
                 ...event.inputTokens != null && { inputTokens: event.inputTokens },
                 ...event.outputTokens != null && { outputTokens: event.outputTokens }
               }).catch((err) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "viagen",
-  "version": "0.0.55",
+  "version": "0.0.57",
   "description": "Vite dev server plugin that exposes endpoints for chatting with Claude Code SDK",
   "type": "module",
   "main": "./dist/index.js",
@@ -45,7 +45,7 @@
     "agent-browser": "^0.20.12",
     "lucide-react": "^0.564.0",
     "simple-git": "^3.31.1",
-    "viagen-sdk": "^0.0.12"
+    "viagen-sdk": "^0.1.0"
   },
   "license": "MIT",
   "repository": {