@saadnvd1/agent-os 0.3.833 → 0.3.839

package/app/api/fresh-start/route.test.ts

@@ -0,0 +1,33 @@
+import { describe, expect, it } from "vitest";
+import { NextRequest } from "next/server";
+import { refreshSettings } from "@/lib/chat/refresh";
+import { SETTING_DEFAULTS } from "@/lib/settings/defaults";
+import { GET, PATCH } from "./route";
+
+const patch = (body: unknown) =>
+  PATCH(
+    new NextRequest("http://localhost/api/fresh-start", {
+      method: "PATCH",
+      body: JSON.stringify(body),
+    })
+  );
+
+describe("PATCH /api/fresh-start", () => {
+  it("changes only the field it's sent", async () => {
+    const res = await patch({ compactTokens: 120_000 });
+    expect(res.status).toBe(200);
+    expect(refreshSettings()).toEqual({
+      ...SETTING_DEFAULTS.fresh,
+      compactTokens: 120_000,
+    });
+    expect(await (await GET()).json()).toMatchObject({
+      interactiveAutoCompactTokens: 300_000,
+    });
+  });
+
+  it("rejects an out-of-range value and saves nothing", async () => {
+    const res = await patch({ taskTokens: 10_000 });
+    expect(res.status).toBe(400);
+    expect(refreshSettings().taskTokens).toBe(0);
+  });
+});

package/app/api/fresh-start/route.ts

@@ -10,6 +10,9 @@ export async function GET() {
   return NextResponse.json(refreshSettings());
 }
 
+// One field at a time, or several: only those named change.
+export const PATCH = PUT;
+
 export async function PUT(request: NextRequest) {
   try {
     const body = await request.json();

package/app/api/sessions/route.ts

@@ -10,6 +10,7 @@ import { ModelUnavailableError } from "@/lib/models/agent-binding";
 import { namesLine, waitBadges } from "@/lib/bus/waits/store";
 import { claimsSnapshot } from "@/lib/claims";
 import { isProjectRef } from "@/lib/tasks/project-ref";
+import { worktreeRemoved } from "@/lib/chat/worker/cwd";
 import { omitNulls } from "@/lib/sessions/omit-null";
 
 // The sidebar's Done shelf lists this many finished tasks; ?done=all lists
@@ -34,7 +35,12 @@ export async function GET(request?: NextRequest) {
     const sessions = rows.map((s) => {
       const b = badges.get(s.id);
       const overlap = claims.bySession.get(s.id);
-      const row = overlap ? { ...s, overlap } : s;
+      const gone = worktreeRemoved(s);
+      const row = {
+        ...s,
+        ...(overlap && { overlap }),
+        ...(gone && { worktree_removed: true }),
+      };
       return omitNulls(
         b
           ? {

package/app/globals.css

@@ -334,3 +334,17 @@ body {
 ::highlight(conv-search) {
   background-color: hsl(var(--find-match));
 }
+
+/* Toast close button inside the toast's right edge, not half outside the left. */
+[data-sonner-toaster]
+  [data-sonner-toast][data-styled="true"]
+  [data-close-button] {
+  left: auto;
+  right: 6px;
+  top: 50%;
+  margin-top: -10px;
+  transform: none;
+  background: transparent;
+  border-color: transparent;
+  color: hsl(var(--muted-foreground));
+}

package/components/Chat/ChatPanel.tsx

@@ -8,17 +8,24 @@ import {
   useRef,
   useState,
 } from "react";
-import { ArrowDown, Loader2 } from "lucide-react";
+import { ArrowDown, GitFork, Loader2 } from "lucide-react";
 import type { ClipboardEvent } from "react";
 import { useChat } from "@/data/chat/useChat";
 import { PausedBanner } from "./PausedBanner";
+import { WorktreeRemovedNotice } from "./WorktreeRemovedNotice";
 import { groupTimeline } from "@/lib/chat/group";
 import type { ChatItem } from "@/lib/chat/events";
 import { isFreshCommand, withFresh } from "@/lib/chat/commands";
 import { useExperiment } from "@/data/experiments";
 import { useStartForks } from "@/data/fork";
 import { toast } from "sonner";
-import { ForksStrip, forkLabel } from "./Forks";
+import {
+  leaveForkMode,
+  placeholderFor,
+  suggestionFor,
+  type ForkDraft,
+} from "./fork-draft";
+import { ForksStrip, forkLabel, startedToast } from "./Forks";
 import { ReplyingTo } from "./ReplyingTo";
 import { compactsFirst } from "@/lib/chat/pending";
 import { useFreshStart } from "@/data/sessions";
@@ -106,12 +113,11 @@ export function ChatPanel({
   const forksOn = useExperiment("forks");
   const startFork = useStartForks(sessionId);
   // Set while a selection is being forked; per session, like asIs below.
-  const [forkDraft, setForkDraft] = useState<{
-    sessionId: string;
-    quote: string;
-    own: boolean;
-  } | null>(null);
-  const fork = forksOn && forkDraft?.sessionId === sessionId ? forkDraft : null;
+  const [forkDraft, setForkDraft] = useState<ForkDraft | null>(null);
+  const draftHere = forkDraft?.sessionId === sessionId ? forkDraft : null;
+  const fork = forksOn && draftHere?.mode ? draftHere : null;
+  const quoteChip =
+    draftHere && !draftHere.mode && draftHere.quote ? draftHere : null;
   const {
     items,
     hasMore,
@@ -181,7 +187,7 @@ export function ChatPanel({
 
   // The suggestion the reader set aside (typed over it, or Esc) stays hidden.
   const [dismissed, setDismissed] = useState<string | null>(null);
-  const shownSuggestion = suggestion === dismissed ? null : suggestion;
+  const shownSuggestion = suggestionFor(suggestion, dismissed, !!fork);
   const dismissSuggestion = useCallback((s: string) => setDismissed(s), []);
   const history = useMemo(() => sentPrompts(items), [items]);
   const [insert, setInsert] = useState<{
@@ -190,15 +196,13 @@ export function ChatPanel({
     append?: boolean;
   }>();
 
-  // Leaving Fork mode keeps the selection as a normal quote in the draft.
+  // Leaving Fork mode keeps the selection as a quote chip and the typed text.
   const leaveFork = () => {
-    if (fork?.quote)
-      setInsert({
-        text: quoteMarkdown(fork.quote),
-        at: Date.now(),
-        append: true,
-      });
-    setForkDraft(null);
+    if (!fork) return;
+    setForkDraft(leaveForkMode(fork));
+    toast("Fork mode off · your text is kept", {
+      icon: <GitFork aria-hidden className="size-4" />,
+    });
   };
 
   // Esc sets the composer's ghost text aside, or else stops the turn
@@ -269,7 +273,7 @@ export function ChatPanel({
       focusComposer
     );
   const forkQuote = (text: string) => {
-    setForkDraft({ sessionId, quote: text, own: false });
+    setForkDraft({ sessionId, quote: text, own: false, mode: true });
     focusComposer();
   };
   const togglePlan = plan === null ? undefined : () => setPlan(!plan);
@@ -688,6 +692,7 @@ export function ChatPanel({
           onFork={forksOn ? forkQuote : undefined}
         />
         <PausedBanner sessionId={sessionId} />
+        <WorktreeRemovedNotice sessionId={sessionId} />
         <ForksStrip sessionId={sessionId} />
         <Queue
           queue={queue}
@@ -725,13 +730,7 @@ export function ChatPanel({
           insert={insert}
           running={running}
           disabled={!connected}
-          placeholder={
-            !connected
-              ? "Reconnecting…"
-              : fork
-                ? "What should the fork do?"
-                : `Message ${sessionName}`
-          }
+          placeholder={placeholderFor(connected, !!fork, sessionName)}
           fork={
             fork
               ? {
@@ -743,6 +742,14 @@ export function ChatPanel({
                 }
               : undefined
           }
+          quoteChip={
+            quoteChip
+              ? {
+                  text: quoteChip.quote,
+                  onDrop: () => setForkDraft(null),
+                }
+              : undefined
+          }
           onSend={(text, images, files) => {
             if (freshStart && isFreshCommand(text)) {
               startFresh.mutate(sessionId);
@@ -758,8 +765,8 @@ export function ChatPanel({
                   labels: [forkLabel(text, fork.quote)],
                 },
                 {
-                  onSuccess: () => {
-                    toast.success("Started fork");
+                  onSuccess: (res) => {
+                    toast.success(startedToast(res.children.map((c) => c.n)));
                     setForkDraft(null);
                   },
                   onError: (e) => {
@@ -771,14 +778,19 @@ export function ChatPanel({
               return;
             }
             toEnd();
-            const compact = composerCompacts(text, context, {
+            // A quote kept from Fork mode goes out ahead of what was typed.
+            const body = quoteChip
+              ? (quoteMarkdown(quoteChip.quote) + text).trim()
+              : text;
+            if (quoteChip) setForkDraft(null);
+            const compact = composerCompacts(body, context, {
               state,
               thresholdTokens: compactTokens,
               asIs,
             });
             setAsIs(false);
             send(
-              text,
+              body,
               images.length ? images : undefined,
               compact,
               files.length ? files : undefined

package/components/Chat/Forks.tsx

@@ -10,6 +10,7 @@ import {
   useStartForks,
   type ChildView,
 } from "@/data/fork";
+import { formatElapsed } from "@/lib/chat/elapsed";
 import { Button } from "@/components/ui/button";
 import { StatusPill } from "@/components/ui/status-pill";
 import { SessionAnchor } from "./SessionAnchor";
@@ -34,6 +35,9 @@ export const forkLabel = (typed: string, quote: string): string => {
   return line.length > 60 ? `${line.slice(0, 59)}…` : line;
 };
 
+export const startedToast = (ns: number[]): string =>
+  ns.length === 1 ? `Started fork ${ns[0]}` : `Started forks ${ns.join(", ")}`;
+
 // What a fork that ended with nothing to read says in place of a result.
 export const emptyNote = (c: ChildView) =>
   c.summary && c.summary !== "ended without a result"
@@ -49,13 +53,23 @@ export function useRunAgain(sessionId: string) {
       start.mutate(
         { instructions: [c.instruction], worktree: c.worktree },
         {
-          onSuccess: () => toast.success("Started fork"),
+          onSuccess: (res) =>
+            toast.success(startedToast(res.children.map((x) => x.n))),
           onError: (e) => toast.error(e.message),
         }
       ),
   };
 }
 
+// "2m 14s · reading sync/worker.ts": how long it has gone and its latest step.
+export const runningNote = (c: ChildView, now = Date.now()): string | null => {
+  const parts = [
+    c.startedAt ? formatElapsed(now - c.startedAt) : null,
+    c.activity,
+  ].filter(Boolean);
+  return parts.length ? parts.join(" · ") : null;
+};
+
 function Row({
   c,
   onStop,
@@ -72,6 +86,7 @@ function Row({
   const empty = c.outcome === "empty";
   const note =
     (empty ? emptyNote(c) : c.summary) ??
+    (c.state === "running" ? runningNote(c) : null) ??
     (c.asks
       ? `Asks: ${c.asks}`
       : c.state === "queued"

package/components/Chat/Messages.tsx

@@ -9,6 +9,7 @@ import {
   Check,
   ChevronRight,
   Download,
+  GitFork,
   Circle,
   CircleDot,
   Quote,
@@ -41,6 +42,7 @@ import {
 } from "@/components/Bookmarks/BookmarkButton";
 import { useOwnerName } from "@/data/client-config/queries";
 import { ownerVerb } from "@/lib/owner-label";
+import { FORK_NOTE_PREFIX } from "@/lib/fork/note";
 
 type Of<K extends ChatItem["kind"]> = Extract<ChatItem, { kind: K }>;
 
@@ -481,12 +483,24 @@ const NOTE_LABEL = {
   away: "Away",
 } as const;
 
+// Where forks were started: one quiet line, not a boxed note.
+function ForkMarker({ item }: { item: Of<"note"> }) {
+  return (
+    <div className="text-muted-foreground flex min-w-0 items-center gap-2 text-xs">
+      <GitFork aria-hidden className="size-3.5 shrink-0" />
+      <span className="truncate">{item.text}</span>
+      <i className="bg-border h-px w-3 shrink-0" />
+    </div>
+  );
+}
+
 export function NoteLine({ item }: { item: Of<"note"> }) {
   const name = useOwnerName();
   const label =
     item.tone === "escalation"
       ? ownerVerb(name, "decides", "decide")
       : NOTE_LABEL[item.tone];
+  if (item.id.startsWith(FORK_NOTE_PREFIX)) return <ForkMarker item={item} />;
   return (
     <div
       className={cn(

package/components/Chat/WorktreeRemovedNotice.tsx

@@ -0,0 +1,17 @@
+"use client";
+
+import { useSessionsQuery } from "@/data/sessions";
+import { Notice } from "@/components/ui/notice";
+
+// Above the composer once a finished task's worktree is removed: its chat
+// still answers from the project's checkout, but can't change anything.
+export function WorktreeRemovedNotice({ sessionId }: { sessionId: string }) {
+  const { data } = useSessionsQuery();
+  const session = data?.sessions.find((s) => s.id === sessionId);
+  if (!session?.worktree_removed) return null;
+  return (
+    <Notice className="mb-2">
+      Merged · worktree removed. Questions work; changes need a new task.
+    </Notice>
+  );
+}

package/components/Chat/composer/ComposerBody.tsx

@@ -32,7 +32,7 @@ import { offeredSuggestion } from "@/lib/chat/suggestion";
 import { historyStep } from "@/lib/chat/history";
 import { markdownParagraphs } from "@/lib/chat/quote";
 import { insertMarkdown } from "./insert";
-import { ForkHeader, type ForkComposer } from "./ForkHeader";
+import { ForkHeader, QuoteChip, type ForkComposer } from "./ForkHeader";
 import { useCoarsePointer } from "./useCoarsePointer";
 import { useMentions } from "./useMentions";
 import { Attachments } from "./Attachments";
@@ -56,6 +56,8 @@ export interface ComposerProps {
   // The send button's label, when Enter does something else than send.
   // Fork mode: the quote chip, pill and own-checkout choice.
   fork?: ForkComposer;
+  // After Fork mode: the selection stays as a chip on a normal reply.
+  quoteChip?: { text: string; onDrop: () => void };
   onSend: (text: string, images: ChatImage[], files: ChatFile[]) => void;
   // Where attached files are kept: the session's id, or "draft" before one exists.
   attachScope: string;
@@ -99,6 +101,7 @@ export function ComposerBody({
   disabled,
   placeholder,
   fork,
+  quoteChip,
   onSend,
   onStop,
   commands = [],
@@ -276,6 +279,7 @@ export function ComposerBody({
 
   const empty =
     !fork?.quote &&
+    !quoteChip &&
     !text.trim() &&
     !images.length &&
     !files.length &&
@@ -371,6 +375,11 @@ export function ComposerBody({
         )}
       >
         {fork && <ForkHeader fork={fork} />}
+        {quoteChip && (
+          <div className="mb-1.5">
+            <QuoteChip text={quoteChip.text} onDrop={quoteChip.onDrop} />
+          </div>
+        )}
         <Attachments
           images={images}
           files={files}

package/components/Chat/composer/ForkHeader.tsx

@@ -35,21 +35,32 @@ export function ForkHeader({ fork }: { fork: ForkComposer }) {
           ↵ start · Esc cancel
         </span>
       </div>
-      {fork.quote && (
-        <div className="bg-muted/60 text-muted-foreground flex items-start gap-2 rounded-lg py-1.5 pr-1 pl-2.5 text-sm">
-          <Quote aria-hidden className="mt-1 size-3.5 shrink-0" />
-          <p className="line-clamp-2 min-w-0 flex-1">{fork.quote}</p>
-          <button
-            type="button"
-            onMouseDown={(e) => e.preventDefault()}
-            onClick={fork.onDropQuote}
-            aria-label="Remove quote"
-            className="hover:text-foreground flex size-8 shrink-0 items-center justify-center"
-          >
-            <X aria-hidden className="size-4" />
-          </button>
-        </div>
-      )}
+      {fork.quote && <QuoteChip text={fork.quote} onDrop={fork.onDropQuote} />}
+    </div>
+  );
+}
+
+// The selection as a chip above the box; it goes out ahead of what's typed.
+export function QuoteChip({
+  text,
+  onDrop,
+}: {
+  text: string;
+  onDrop: () => void;
+}) {
+  return (
+    <div className="bg-muted/60 text-muted-foreground flex items-start gap-2 rounded-lg py-1.5 pr-1 pl-2.5 text-sm">
+      <Quote aria-hidden className="mt-1 size-3.5 shrink-0" />
+      <p className="line-clamp-2 min-w-0 flex-1">{text}</p>
+      <button
+        type="button"
+        onMouseDown={(e) => e.preventDefault()}
+        onClick={onDrop}
+        aria-label="Remove quote"
+        className="hover:text-foreground flex size-8 shrink-0 items-center justify-center"
+      >
+        <X aria-hidden className="size-4" />
+      </button>
     </div>
   );
 }

package/components/Chat/fork-draft.test.ts

@@ -0,0 +1,57 @@
+import { describe, expect, it } from "vitest";
+import {
+  FORK_PLACEHOLDER,
+  leaveForkMode,
+  placeholderFor,
+  suggestionFor,
+  type ForkDraft,
+} from "./fork-draft";
+import { forkLabel, runningNote, startedToast } from "./Forks";
+import { startedText } from "@/lib/fork/note";
+import type { ChildView } from "@/lib/fork/types";
+
+const draft: ForkDraft = {
+  sessionId: "s",
+  quote: "the text",
+  own: true,
+  mode: true,
+};
+
+describe("Fork mode composer", () => {
+  it("hides the parent's suggestion and uses the fork placeholder", () => {
+    expect(
+      suggestionFor("what does --rollback --to do", null, true)
+    ).toBeNull();
+    expect(suggestionFor("next", null, false)).toBe("next");
+    expect(suggestionFor("next", "next", false)).toBeNull();
+    expect(placeholderFor(true, true, "Work")).toBe(FORK_PLACEHOLDER);
+    expect(placeholderFor(true, false, "Work")).toBe("Message Work");
+    expect(placeholderFor(false, true, "Work")).toBe("Reconnecting…");
+  });
+
+  it("keeps the quote and drops only the Fork mode on leaving", () => {
+    expect(leaveForkMode(draft)).toEqual({ ...draft, mode: false });
+    expect(leaveForkMode({ ...draft, quote: "" })).toBeNull();
+  });
+});
+
+describe("fork copy", () => {
+  it("words the started marker and toast", () => {
+    expect(startedText([1], ["Check it"])).toBe("Fork 1 started · Check it");
+    expect(startedText([2, 3], ["a", "b"])).toBe("Forks 2, 3 started · a · b");
+    expect(startedToast([4])).toBe("Started fork 4");
+    expect(forkLabel("", "> q")).toBe("> q");
+  });
+
+  it("says how long a running fork has gone and what it's doing", () => {
+    const c = {
+      startedAt: 1_000,
+      activity: "Read sync/worker.ts",
+    } as ChildView;
+    expect(runningNote(c, 1_000 + 134_000)).toBe(
+      "2m 14s · Read sync/worker.ts"
+    );
+    expect(runningNote({ ...c, activity: null }, 6_000)).toBe("5s");
+    expect(runningNote({ ...c, startedAt: null, activity: null })).toBeNull();
+  });
+});

package/components/Chat/fork-draft.ts

@@ -0,0 +1,34 @@
+// A selection being forked, and what's left of it after leaving Fork mode.
+export interface ForkDraft {
+  sessionId: string;
+  quote: string;
+  own: boolean;
+  // Off after leaving Fork mode: the quote stays as a chip on a normal reply.
+  mode: boolean;
+}
+
+export const FORK_PLACEHOLDER = "What should the fork do?";
+
+// Esc or the pill's x: only the Fork pill goes; the quote chip stays when
+// there is one, and what was typed stays in the composer.
+export const leaveForkMode = (draft: ForkDraft): ForkDraft | null =>
+  draft.quote ? { ...draft, mode: false } : null;
+
+// The parent's guess has no place in Fork mode, which asks its own question.
+export const suggestionFor = (
+  suggestion: string | null,
+  dismissed: string | null,
+  inForkMode: boolean
+): string | null =>
+  inForkMode || suggestion === dismissed ? null : suggestion;
+
+export const placeholderFor = (
+  connected: boolean,
+  inForkMode: boolean,
+  sessionName: string
+): string =>
+  !connected
+    ? "Reconnecting…"
+    : inForkMode
+      ? FORK_PLACEHOLDER
+      : `Message ${sessionName}`;