nolo-cli 0.1.21 → 0.1.23

package/client/agentRun.ts

@@ -1,6 +1,13 @@
 import { existsSync } from "node:fs";
 import { join } from "node:path";
 import { runLocalAgentTurn } from "../agentRuntimeLocal";
+import {
+  FRONTEND_IMPLEMENTER_AGENT_KEY,
+  NOLO_FULLSTACK_AGENT_KEY,
+  NOLO_PROJECT_MANAGER_AGENT_KEY,
+  NOLO_REVIEW_AGENT_KEY,
+  NOLO_SENIOR_FULLSTACK_AGENT_KEY,
+} from "../agentAliases";
 import type { LocalAgentToolEvent } from "../agent-runtime/localLoop";
 import type { AgentRuntimeHostAdapter, AgentRuntimeRequestedMode } from "../agentRuntimeLocal";
 import { createCliLocalRuntimeAdapter } from "./localRuntimeAdapter";
@@ -52,7 +59,7 @@ type RunAgentTurnOptions = {
   taskRunContext?: TaskRunPromptContext;
   taskRunRecorder?: (args: {
     options: RunAgentTurnOptions;
-    status: "completed" | "failed";
+    status: "running" | "completed" | "failed";
     dialogId?: string;
     resultSummary?: string;
     errorCode?: string;
@@ -62,10 +69,11 @@ type RunAgentTurnOptions = {
   fetchImpl?: typeof fetch;
 };
 
-export type RunAgentTurnResult = {
-  exitCode: number;
-  dialogId?: string;
-};
+export type RunAgentTurnResult = {
+  exitCode: number;
+  dialogId?: string;
+  streamInterrupted?: boolean;
+};
 
 type ScriptBridgeDecision = {
   hasAuthToken: boolean;
@@ -84,6 +92,38 @@ const SERVER_PLATFORM_TOOL_NAMES = new Set([
   "updateTableRows",
 ]);
 
+const KNOWN_SERVER_PLATFORM_AGENT_KEYS = new Set([
+  FRONTEND_IMPLEMENTER_AGENT_KEY,
+  NOLO_FULLSTACK_AGENT_KEY,
+  NOLO_PROJECT_MANAGER_AGENT_KEY,
+  NOLO_REVIEW_AGENT_KEY,
+  NOLO_SENIOR_FULLSTACK_AGENT_KEY,
+]);
+
+const KNOWN_SERVER_PLATFORM_AGENT_ALIASES = new Set([
+  "code-review",
+  "frontend",
+  "frontend-agent",
+  "frontend-implementer",
+  "full-stack",
+  "fullstack",
+  "nolo code review",
+  "nolo frontend",
+  "nolo fullstack",
+  "nolo project manager",
+  "nolo reviewer",
+  "nolo-code-review",
+  "nolo-frontend",
+  "nolo-fullstack",
+  "nolo-pm",
+  "nolo-project-manager",
+  "nolo-reviewer",
+  "pm",
+  "project-manager",
+  "review",
+  "reviewer",
+]);
+
 export function shouldUseScriptBridge(decision: ScriptBridgeDecision) {
   return !decision.hasAuthToken && decision.scriptPathExists;
 }
@@ -92,6 +132,25 @@ export function findServerPlatformTools(toolNames?: string[]) {
   if (!Array.isArray(toolNames)) return [];
   return toolNames.filter((toolName) => SERVER_PLATFORM_TOOL_NAMES.has(toolName));
 }
+
+function resolveServerPlatformToolNames(agentConfig: any) {
+  return findServerPlatformTools([
+    ...(Array.isArray(agentConfig?.toolNames) ? agentConfig.toolNames : []),
+    ...(Array.isArray(agentConfig?.runtimeToolPolicy?.agentTools)
+      ? agentConfig.runtimeToolPolicy.agentTools
+      : []),
+  ]);
+}
+
+function normalizeAgentRef(ref?: string) {
+  return ref?.trim().toLowerCase().replace(/\s+/g, " ");
+}
+
+function isKnownServerPlatformAgent(options: RunAgentTurnOptions) {
+  if (KNOWN_SERVER_PLATFORM_AGENT_KEYS.has(options.agentKey)) return true;
+  const normalizedKey = normalizeAgentRef(options.agentKey);
+  return Boolean(normalizedKey && KNOWN_SERVER_PLATFORM_AGENT_ALIASES.has(normalizedKey));
+}
 
 function resolveAuthToken(env: EnvLike) {
   return env.AUTH_TOKEN || env.AUTH || env.BENCHMARK_AUTH_TOKEN || "";
@@ -126,6 +185,13 @@ function resolveLocalRuntimeAdapter(options: RunAgentTurnOptions) {
 }
 
 async function shouldSkipAutoLocalForServerPlatformTools(options: RunAgentTurnOptions) {
+  if (isKnownServerPlatformAgent(options)) {
+    options.output.write(
+      `[nolo] auto runtime: skipping local runtime because ${options.agentKey} is a known platform agent. ` +
+        "Use --local explicitly to force local workspace tools.\n"
+    );
+    return true;
+  }
   const adapter = resolveLocalRuntimeAdapter(options);
   if (!adapter) return false;
   let agentConfig;
@@ -134,7 +200,7 @@ async function shouldSkipAutoLocalForServerPlatformTools(options: RunAgentTurnOp
   } catch {
     return false;
   }
-  const serverTools = findServerPlatformTools(agentConfig?.toolNames);
+  const serverTools = resolveServerPlatformToolNames(agentConfig);
   if (serverTools.length === 0) return false;
   options.output.write(
     `[nolo] auto runtime: skipping local runtime because ${options.agentKey} declares server platform tools ` +
@@ -292,7 +358,7 @@ async function runTaskRunCli(args: {
 
 async function defaultTaskRunRecorder(args: {
   options: RunAgentTurnOptions;
-  status: "completed" | "failed";
+  status: "running" | "completed" | "failed";
   dialogId?: string;
   resultSummary?: string;
   errorCode?: string;
@@ -301,7 +367,7 @@ async function defaultTaskRunRecorder(args: {
   const context = args.options.taskRunContext;
   if (!context?.rowDbKey) return;
   const role = inferTaskRunAgentRole(args.options.agentName || args.options.agentKey);
-  if (role === "reviewer") {
+  if (role === "reviewer" && args.status !== "running") {
     await runTaskRunCli({
       options: args.options,
       cliArgs: [
@@ -359,7 +425,7 @@ async function defaultTaskRunRecorder(args: {
       ],
     });
   }
-  if (role === "reviewer") {
+  if (role === "reviewer" && args.status !== "running") {
     const reviewStatus =
       args.status === "failed" ? "blocked" : classifyTaskRunReviewStatus(args.resultSummary);
     if (reviewStatus) {
@@ -385,7 +451,7 @@ async function defaultTaskRunRecorder(args: {
 
 async function recordTaskRunOutcome(args: {
   options: RunAgentTurnOptions;
-  status: "completed" | "failed";
+  status: "running" | "completed" | "failed";
   dialogId?: string;
   resultSummary?: string;
   errorCode?: string;
@@ -510,12 +576,28 @@ async function runHttpAgentTurn(options: RunAgentTurnOptions, authToken: string)
           options.imageUrls
         ),
         runtimeContext: {
-          surface: "cli",
-          host: "terminal",
-          runtime: "bun",
-          entrypoint: "nolo-cli",
-          capabilities: ["text-io", "streaming", "slash-commands"],
-        },
+          surface: "cli",
+          host: "terminal",
+          runtime: "bun",
+          entrypoint: "nolo-cli",
+          capabilities: ["text-io", "streaming", "slash-commands"],
+          ...(options.taskRunContext
+            ? {
+                taskRun: {
+                  rowDbKey: options.taskRunContext.rowDbKey,
+                  ...(options.taskRunContext.taskRunId
+                    ? { taskRunId: options.taskRunContext.taskRunId }
+                    : {}),
+                  ...(options.taskRunContext.workItemId
+                    ? { workItemId: options.taskRunContext.workItemId }
+                    : {}),
+                  ...(options.taskRunContext.artifactIds?.length
+                    ? { artifactIds: options.taskRunContext.artifactIds }
+                    : {}),
+                },
+              }
+            : {}),
+        },
         ...(options.continueDialogId
           ? { continueDialogId: options.continueDialogId }
           : {}),
@@ -541,7 +623,11 @@ async function runHttpAgentTurn(options: RunAgentTurnOptions, authToken: string)
     const result = await readStreamingAgentRun(options, res);
     await recordTaskRunOutcome({
       options,
-      status: result.exitCode === 0 ? "completed" : "failed",
+      status: result.streamInterrupted
+        ? "running"
+        : result.exitCode === 0
+          ? "completed"
+          : "failed",
       dialogId: result.dialogId,
       ...(result.exitCode !== 0 ? { errorCode: "STREAM_FAILED" } : {}),
     });
@@ -586,9 +672,9 @@ async function runHttpAgentTurn(options: RunAgentTurnOptions, authToken: string)
   };
   await recordTaskRunOutcome({
     options,
-    status: "completed",
+    status: options.background ? "running" : "completed",
     dialogId: result.dialogId,
-    resultSummary: content || undefined,
+    ...(!options.background && content ? { resultSummary: content } : {}),
   });
   return result;
 }
@@ -691,9 +777,9 @@ async function runLocalAgentTurnForCli(
     }
     await recordTaskRunOutcome({
       options,
-      status: "completed",
+      status: options.background ? "running" : "completed",
       dialogId: result.dialogId,
-      resultSummary: content || undefined,
+      ...(!options.background && content ? { resultSummary: content } : {}),
       toolEvidence: summarizeLocalToolEvidence(toolEvents),
     });
     return { exitCode: 0, dialogId: result.dialogId };
@@ -735,15 +821,20 @@ async function readStreamingAgentRun(
     hasPrintedLabel = true;
   };
 
-  const handlePayload = (payload: any) => {
-    if (payload?.error || payload?.type === "error") {
-      throw new Error(String(payload.error || payload.message || "Agent stream failed"));
-    }
-    if (payload?.type === "done") {
-      if (typeof payload.dialogId === "string") dialogId = payload.dialogId;
-      usage = payload.usage;
-      return;
-    }
+  const handlePayload = (payload: any) => {
+    if (typeof payload?.dialogId === "string" && payload.dialogId.trim()) {
+      dialogId = payload.dialogId;
+    }
+    if (payload?.error || payload?.type === "error") {
+      throw new Error(String(payload.error || payload.message || "Agent stream failed"));
+    }
+    if (payload?.type === "done") {
+      usage = payload.usage;
+      return;
+    }
+    if (payload?.type === "dialog" || payload?.type === "status") {
+      return;
+    }
 
     const chunk =
       payload?.type === "text"
@@ -786,10 +877,18 @@ async function readStreamingAgentRun(
         .trim();
       if (raw) handlePayload(JSON.parse(raw));
     }
-  } catch (error) {
-    options.output.write(`\n[nolo] Agent stream failed: ${error instanceof Error ? error.message : String(error)}\n`);
-    return { exitCode: 1 };
-  } finally {
+  } catch (error) {
+    const message = error instanceof Error ? error.message : String(error);
+    if (dialogId) {
+      options.output.write(
+        `\n[nolo] Agent stream transport interrupted after dialog ${dialogId} was created: ${message}\n`
+      );
+      options.output.write("[nolo] The agent run may still finish on the server; read the dialog before retrying.\n");
+      return { exitCode: 0, dialogId, streamInterrupted: true };
+    }
+    options.output.write(`\n[nolo] Agent stream failed: ${message}\n`);
+    return { exitCode: 1 };
+  } finally {
     writer.flushAll();
   }
 

package/client/localRuntimeAdapter.test.ts

@@ -353,6 +353,7 @@ describe("CLI local runtime adapter", () => {
         tool_choice: "auto",
         url: "https://api.fireworks.ai/inference/v1/chat/completions",
         provider: "fireworks",
+        agentKey: "agent-user-1-frontend",
       },
     });
     expect(toolNamesFromRequest(requests[0])).toEqual(DEFAULT_LOCAL_CODING_TOOL_NAMES);
@@ -419,6 +420,7 @@ describe("CLI local runtime adapter", () => {
       body: {
         provider: "fireworks",
         apiSource: "platform",
+        agentKey: "agent-user-1-frontend",
       },
     });
   });

package/client/localRuntimeAdapter.ts

@@ -58,6 +58,7 @@ export type CliLocalRuntimeAdapterDeps = {
   createId?: () => string;
   fetchImpl?: typeof fetch;
   cwd?: string;
+  useCwdAsTaskWorkspaceBase?: boolean;
   output?: { write(chunk: string): unknown };
   prepareTaskWorktree?: typeof prepareTaskWorktree;
   localToolExecutors?: Record<string, (call: any) => Promise<{ content: string; metadata?: Record<string, unknown> }>>;
@@ -254,7 +255,11 @@ export function createCliLocalRuntimeAdapter(
   const localToolBudgets = parseLocalToolBudgets(deps.env);
   const localToolUsage = new Map<string, number>();
   let activeAgentToolNames: string[] = [];
-  let workspaceSession: WorkspaceSession = createWorkspaceSession({ cwd: deps.cwd });
+  let workspaceSession: WorkspaceSession = createWorkspaceSession(
+    deps.useCwdAsTaskWorkspaceBase
+      ? { defaultCwd: deps.cwd }
+      : { cwd: deps.cwd }
+  );
   let localToolExecutors = buildLocalToolExecutors({
     workspaceRoot: workspaceSession.workspaceRoot,
     localToolExecutors: deps.localToolExecutors,
@@ -378,12 +383,20 @@ export function createCliLocalRuntimeAdapter(
         budgets: localToolBudgets,
         usage: localToolUsage,
       });
-      return executeLocalToolWithPolicy({
+      const result = await executeLocalToolWithPolicy({
         env: deps.env,
         agentToolNames: activeAgentToolNames,
         call,
         executors: localToolExecutors,
       });
+      return {
+        ...result,
+        metadata: {
+          ...(result.metadata ?? {}),
+          workspaceRoot: workspaceSession.workspaceRoot,
+          workspaceKind: workspaceSession.kind,
+        },
+      };
     },
   };
 }

package/database/actions/common.ts

@@ -2,7 +2,8 @@
 
 import pino from "pino";
 import { getIsDesktopApp } from "app/utils/env";
-import { API_ENDPOINTS, NOLO_CLUSTER_SERVERS } from "../config";
+import { fetchWithTransientReadRetry } from "app/utils/retryFetch";
+import { API_ENDPOINTS, NOLO_CLUSTER_SERVERS, normalizeKnownServerOrigin } from "../config";
 
 // RN 下 pino 的 browser 写法可能有兼容性问题
 // 使用简单的 console 封装作为 fallback
@@ -23,7 +24,7 @@ export const logger = isRN ? {
   // },
 });
 const normalizeServer = (server: string): string =>
-  server.trim().replace(/\/+$/, "");
+  normalizeKnownServerOrigin(server) ?? server.trim().replace(/\/+$/, "");
 const isNoloClusterServer = (server: string): boolean =>
   /^https?:\/\/(?:us\.)?nolo\.chat$/i.test(normalizeServer(server));
 
@@ -202,7 +203,7 @@ export const fetchFromServer = async (
   signal?.addEventListener("abort", onExternalAbort);
 
   try {
-    const res = await fetch(
+    const res = await fetchWithTransientReadRetry(
       `${server}${buildReadUrl(dbKey)}`,
       {
         signal: controller.signal as any,

package/database/config.ts

@@ -8,6 +8,25 @@ export const SERVERS = {
 } as const;
 export const NOLO_CLUSTER_SERVERS = Object.values(SERVERS);
 
+const LEGACY_SERVER_ORIGIN_MAP: Record<string, string> = {
+  "https://nolotus.com": SERVERS.MAIN,
+  "https://www.nolotus.com": SERVERS.MAIN,
+  "https://us.nolotus.com": SERVERS.US,
+  "https://www.us.nolotus.com": SERVERS.US,
+};
+
+export const normalizeKnownServerOrigin = (server: unknown): string | null => {
+  if (typeof server !== "string" || server.trim().length === 0) return null;
+  const trimmed = server.trim();
+  let origin: string;
+  try {
+    origin = new URL(trimmed).origin;
+  } catch {
+    origin = trimmed.replace(/\/+$/, "");
+  }
+  return LEGACY_SERVER_ORIGIN_MAP[origin.toLowerCase()] ?? origin;
+};
+
 export const API_ENDPOINTS = {
   DATABASE: `${API_VERSION}/db`,
   SHARE: `${API_VERSION}/share`,