@saadnvd1/agent-os 0.3.671 → 0.3.673

package/components/Chat/composer/Toolbar.tsx

@@ -3,7 +3,7 @@
 import { useRef, useState } from "react";
 import {
   ArrowUp,
-  ImagePlus,
+  Paperclip,
   ListChecks,
   MonitorUp,
   Square,
@@ -23,7 +23,8 @@ import { ModelPicker } from "../ModelPicker";
 export function Toolbar({
   plain,
   onTogglePlain,
-  onAddImages,
+  onAddFiles,
+  accept,
   plan,
   onTogglePlan,
   accessory,
@@ -42,7 +43,9 @@ export function Toolbar({
 }: {
   plain: boolean;
   onTogglePlain: () => void;
-  onAddImages: (files: Iterable<File>) => void;
+  onAddFiles: (files: Iterable<File>) => void;
+  // The types the server takes, for the picker.
+  accept: string;
   // Plan mode, when the session can use it.
   plan?: boolean | null;
   onTogglePlan?: () => void;
@@ -68,7 +71,7 @@ export function Toolbar({
   const capture = async () => {
     try {
       const file = await captureScreenFrame();
-      if (file) onAddImages([file]);
+      if (file) onAddFiles([file]);
     } catch (error) {
       toast.error(
         `Couldn't capture the screen: ${error instanceof Error ? error.message : String(error)}`
@@ -81,20 +84,20 @@ export function Toolbar({
         type="button"
         variant="ghost"
         size="icon"
-        aria-label="Attach image"
+        aria-label="Attach files"
         className="h-10 w-10 shrink-0"
         onClick={() => fileRef.current?.click()}
       >
-        <ImagePlus className="h-4 w-4" />
+        <Paperclip className="h-4 w-4" />
       </Button>
       <input
         ref={fileRef}
         type="file"
-        accept="image/*"
+        accept={accept}
         multiple
         hidden
         onChange={(e) => {
-          if (e.target.files) onAddImages(e.target.files);
+          if (e.target.files) onAddFiles(e.target.files);
           e.target.value = "";
         }}
       />

package/components/Chat/composer/editor.test.ts

@@ -168,7 +168,7 @@ function paste(
   return handlePaste(editor.view, event, {
     plain: false,
     asText: false,
-    onImages: () => {},
+    onFiles: () => {},
     onLongPaste: () => {},
     ...h,
   });

package/components/Chat/composer/paste.ts

@@ -8,7 +8,8 @@ export interface PasteHandlers {
   plain: boolean;
   // Cmd/Ctrl+Shift+V was pressed just before this paste.
   asText: boolean;
-  onImages: (files: File[]) => void;
+  // Files pasted: images, documents, anything the composer may attach.
+  onFiles: (files: File[]) => void;
   onLongPaste: (text: string, language?: string) => void;
 }
 
@@ -58,9 +59,9 @@ export function handlePaste(
 ): boolean {
   const data = event.clipboardData;
   if (!data) return false;
-  const images = [...data.files].filter(isImage);
-  if (images.length) {
-    h.onImages(images);
+  const files = [...data.files];
+  if (files.length) {
+    h.onFiles(files);
     return true;
   }
   const text = data.getData("text/plain").replace(/\r\n?/g, "\n");

package/components/Chat/composer/useAttachFiles.ts

@@ -0,0 +1,142 @@
+"use client";
+
+import { useRef, useState } from "react";
+import { useQueryClient } from "@tanstack/react-query";
+import { toast } from "sonner";
+import {
+  clientConfigQuery,
+  useClientConfig,
+} from "@/data/client-config/queries";
+import {
+  acceptAttr,
+  tooBigMessage,
+  typeFor,
+  unsupportedMessage,
+} from "@/lib/attachments/policy";
+import type { ChatFile, ChatImage } from "@/lib/chat/events";
+import { uploadFile } from "@/lib/file-upload";
+import { isImage } from "./paste";
+
+export interface Upload {
+  key: string;
+  file: ChatFile;
+  uploading?: boolean;
+}
+
+function readImage(file: File): Promise<ChatImage> {
+  return new Promise((resolve, reject) => {
+    const reader = new FileReader();
+    reader.onload = () => {
+      const [meta, data] = String(reader.result).split(",");
+      resolve({ mediaType: meta.slice(5, meta.indexOf(";")), data });
+    };
+    reader.onerror = reject;
+    reader.readAsDataURL(file);
+  });
+}
+
+const carriesFiles = (e: React.DragEvent) =>
+  [...e.dataTransfer.types].includes("Files");
+
+let seq = 0;
+
+// Files picked, pasted or dropped onto the composer, and the outline shown
+// while something is dragged over it. Images are read in and sent with the
+// message; any other file is uploaded to the session's attachments folder
+// at once, so a wrong type or a too-big file fails here, not at send. The
+// types and sizes come from the server (/api/client-config).
+export function useAttachFiles({
+  scope,
+  imageCount,
+  onImages,
+}: {
+  // The session's id, or "draft" for a chat that hasn't started.
+  scope: string;
+  imageCount: number;
+  onImages: (images: ChatImage[]) => void;
+}) {
+  const queryClient = useQueryClient();
+  const types = useClientConfig().data?.attachments.types;
+  const [dragging, setDragging] = useState(false);
+  const [uploads, setUploads] = useState<Upload[]>([]);
+  const held = useRef(0);
+  held.current = imageCount + uploads.length;
+
+  const addFiles = async (picked: Iterable<File>) => {
+    // Copied first: a file input's live list is emptied right after the pick.
+    const list = [...picked];
+    let limits;
+    try {
+      limits = (await queryClient.fetchQuery(clientConfigQuery)).attachments;
+    } catch (e) {
+      toast.error(e instanceof Error ? e.message : "Couldn't attach");
+      return;
+    }
+    const images: File[] = [];
+    const others: File[] = [];
+    for (const f of list) {
+      let error: string | null = null;
+      if (held.current + images.length + others.length >= limits.max)
+        error = `You can attach up to ${limits.max} files to one message.`;
+      else if (isImage(f)) {
+        if (f.size > limits.maxImageBytes)
+          error = tooBigMessage(f.name, f.size, limits.maxImageBytes);
+        else images.push(f);
+      } else if (!typeFor(f.name, limits.types))
+        error = unsupportedMessage(f.name);
+      else if (f.size > limits.maxBytes)
+        error = tooBigMessage(f.name, f.size, limits.maxBytes);
+      else others.push(f);
+      if (error) toast.error(error);
+    }
+    if (images.length) onImages(await Promise.all(images.map(readImage)));
+    for (const f of others) {
+      const key = `up-${++seq}`;
+      const file = { path: "", name: f.name, size: f.size, mediaType: f.type };
+      setUploads((prev) => [...prev, { key, file, uploading: true }]);
+      const done = await uploadFile(f, scope);
+      if (done.ok)
+        setUploads((prev) =>
+          prev.map((u) => (u.key === key ? { key, file: done.file } : u))
+        );
+      else {
+        setUploads((prev) => prev.filter((u) => u.key !== key));
+        toast.error(done.error);
+      }
+    }
+  };
+
+  const dropProps = {
+    onDragOver: (e: React.DragEvent) => {
+      if (!carriesFiles(e)) return;
+      e.preventDefault();
+      e.dataTransfer.dropEffect = "copy";
+      setDragging(true);
+    },
+    onDragLeave: (e: React.DragEvent) => {
+      if (!e.currentTarget.contains(e.relatedTarget as Node | null))
+        setDragging(false);
+    },
+    onDrop: (e: React.DragEvent) => {
+      if (!carriesFiles(e)) return;
+      e.preventDefault();
+      setDragging(false);
+      void addFiles(e.dataTransfer.files);
+    },
+  };
+
+  return {
+    dragging,
+    dropProps,
+    addFiles,
+    uploads,
+    // Until the limits are read, the picker offers images only.
+    accept: types ? `image/*,${acceptAttr(types)}` : "image/*",
+    uploading: uploads.some((u) => u.uploading),
+    remove: (key: string) =>
+      setUploads((prev) => prev.filter((u) => u.key !== key)),
+    clear: () => setUploads([]),
+    restore: (files: ChatFile[]) =>
+      setUploads(files.map((file) => ({ key: `up-${++seq}`, file }))),
+  };
+}

package/components/Chat/composer/useComposerEditor.ts

@@ -13,7 +13,7 @@ import { docToMarkdown } from "@/lib/chat/markdown/serialize";
 import { plainExtensions, richExtensions } from "./extensions";
 import { loadMarkdown } from "./load";
 import { handleKeyDown, type KeyHandlers } from "./keys";
-import { handlePaste, isImage, type PasteHandlers } from "./paste";
+import { handlePaste, type PasteHandlers } from "./paste";
 import { usePlainMode } from "./usePlainMode";
 
 export type ComposerHandlers = Omit<
@@ -120,9 +120,9 @@ export function useComposerEditor({
             asText,
           });
         },
-        // Dropped images are the composer's to take, not the document's.
+        // Dropped files are the composer's to take, not the document's.
         handleDrop: (view, event) =>
-          [...(event.dataTransfer?.files ?? [])].some(isImage),
+          (event.dataTransfer?.files.length ?? 0) > 0,
       },
       // Only edits change the text: not loading a draft, not setEditable.
       onUpdate: ({ editor: ed, transaction }) => {

package/components/Chat/useDraft.ts

@@ -1,7 +1,7 @@
 "use client";
 
 import { useEffect, useRef } from "react";
-import type { ChatImage } from "@/lib/chat/events";
+import type { ChatFile, ChatImage } from "@/lib/chat/events";
 import type { TextAttachment } from "@/lib/chat/paste";
 
 // `text` is the composer's markdown, so a reload restores its formatting.
@@ -9,6 +9,8 @@ interface Draft {
   text: string;
   images: ChatImage[];
   files?: TextAttachment[];
+  // Attached files, saved on the server: references only.
+  uploads?: ChatFile[];
 }
 
 const keyFor = (id: string) => `agentos:draft:${id}`;
@@ -39,7 +41,12 @@ export function useSaveDraft(id: string | undefined, draft: Draft): void {
     clearTimeout(timer.current);
     timer.current = setTimeout(() => {
       const key = keyFor(id);
-      if (!draft.text.trim() && !draft.images.length && !draft.files?.length) {
+      if (
+        !draft.text.trim() &&
+        !draft.images.length &&
+        !draft.files?.length &&
+        !draft.uploads?.length
+      ) {
         localStorage.removeItem(key);
         return;
       }

package/components/Draft/DraftPanel.tsx

@@ -5,7 +5,7 @@ import { useSnapshot } from "valtio";
 import { toast } from "sonner";
 import { Composer } from "@/components/Chat/Composer";
 import { Button } from "@/components/ui/button";
-import type { ChatImage } from "@/lib/chat/events";
+import type { ChatFile, ChatImage } from "@/lib/chat/events";
 import type { Project } from "@/lib/db";
 import { draftComposerKey, type Draft } from "@/lib/drafts";
 import { getModelOptions } from "@/lib/model-catalog";
@@ -57,11 +57,12 @@ export function DraftPanel({
       </div>
     );
 
-  const send = async (text: string, images: ChatImage[]) => {
+  const send = async (text: string, images: ChatImage[], files: ChatFile[]) => {
     try {
       if (draft.openPr) {
         if (!project) throw new Error("Pick a project for a task");
-        if (images.length) throw new Error("A task takes text only");
+        if (images.length || files.length)
+          throw new Error("A task takes text only");
         const { session, queued } = await createTask.mutateAsync({
           projectId: project.id,
           prompt: text,
@@ -96,6 +97,7 @@ export function DraftPanel({
         baseBranch: draft.baseBranch,
         prompt: text,
         images: images.length ? images : undefined,
+        files: files.length ? files : undefined,
       });
       // A resend whose first answer was lost: the terminal still needs its
       // first prompt, and this is the text that was sent.
@@ -134,6 +136,7 @@ export function DraftPanel({
         />
         <Composer
           draftKey={draftComposerKey(draft.id)}
+          attachScope="draft"
           running={false}
           disabled={pending}
           placeholder={
@@ -143,7 +146,7 @@ export function DraftPanel({
                 ? "Describe the task"
                 : `Message ${project?.name ?? "a new chat"}`
           }
-          onSend={(text, images) => void send(text, images)}
+          onSend={(text, images, files) => void send(text, images, files)}
           onStop={() => {}}
           models={getModelOptions(draft.agentType)}
           model={draft.model}

package/components/Projects/ProjectSettingsDialog.tsx

@@ -12,6 +12,8 @@ import {
 import { Button } from "@/components/ui/button";
 import { Input } from "@/components/ui/input";
 import { Textarea } from "@/components/ui/textarea";
+import { ProjectTierField } from "./ProjectTierField";
+import type { ProjectTier } from "@/lib/project-tier/types";
 import {
   Select,
   SelectContent,
@@ -95,6 +97,7 @@ export function ProjectSettingsDialog({
     getDefaultModelForAgent("claude")
   );
   const [initialPrompt, setInitialPrompt] = useState("");
+  const [tier, setTier] = useState<ProjectTier>("fast");
   const [devServers, setDevServers] = useState<DevServerConfig[]>([]);
   const [repositories, setRepositories] = useState<RepositoryConfig[]>([]);
   const [isLoading, setIsLoading] = useState(false);
@@ -118,6 +121,7 @@ export function ProjectSettingsDialog({
       setName(project.name);
       setWorkingDirectory(project.working_directory);
       setAgentType(project.agent_type);
+      setTier(project.tier);
       setDefaultModel(
         isSupportedModelForAgent(project.agent_type, project.default_model)
           ? project.default_model
@@ -336,6 +340,7 @@ export function ProjectSettingsDialog({
         agentType,
         defaultModel,
         initialPrompt: initialPrompt.trim() || null,
+        tier,
       });
 
       // Handle dev server changes
@@ -526,6 +531,8 @@ export function ProjectSettingsDialog({
               </Select>
             </div>
 
+            <ProjectTierField value={tier} onChange={setTier} />
+
             {/* Initial Prompt */}
             <div className="space-y-2">
               <label className="text-sm font-medium">Initial Prompt</label>

package/components/Projects/ProjectTierField.tsx

@@ -0,0 +1,39 @@
+"use client";
+
+import {
+  Select,
+  SelectContent,
+  SelectItem,
+  SelectTrigger,
+  SelectValue,
+} from "@/components/ui/select";
+import type { ProjectTier } from "@/lib/project-tier/types";
+
+const DETAIL: Record<ProjectTier, string> = {
+  production: "CI, the independent review and scope must pass before a merge.",
+  fast: "A task's PR merges as soon as it opens: no CI wait, no review.",
+};
+
+export function ProjectTierField({
+  value,
+  onChange,
+}: {
+  value: ProjectTier;
+  onChange: (tier: ProjectTier) => void;
+}) {
+  return (
+    <div className="space-y-2">
+      <label className="text-sm font-medium">Tier</label>
+      <Select value={value} onValueChange={(v) => onChange(v as ProjectTier)}>
+        <SelectTrigger>
+          <SelectValue />
+        </SelectTrigger>
+        <SelectContent>
+          <SelectItem value="fast">Fast</SelectItem>
+          <SelectItem value="production">Production</SelectItem>
+        </SelectContent>
+      </Select>
+      <p className="text-muted-foreground text-xs">{DETAIL[value]}</p>
+    </div>
+  );
+}

package/components/Terminal/index.tsx

@@ -19,7 +19,8 @@ import { useTerminalConnection, useTerminalSearch } from "./hooks";
 import type { TerminalScrollState } from "./hooks";
 import { useViewport } from "@/hooks/useViewport";
 import { useFileDrop } from "@/hooks/useFileDrop";
-import { uploadFileToTemp } from "@/lib/file-upload";
+import { toast } from "sonner";
+import { uploadFile } from "@/lib/file-upload";
 import { FilePicker } from "@/components/FilePicker";
 import type { AttachSpec } from "@/lib/hosts/attach";
 
@@ -121,11 +122,11 @@ export const Terminal = forwardRef<TerminalHandle, TerminalProps>(
       async (file: File) => {
         setIsUploading(true);
         try {
-          const path = await uploadFileToTemp(file);
-          if (path) {
-            sendInput(path);
+          const done = await uploadFile(file);
+          if (done.ok) {
+            sendInput(done.file.path);
             focus();
-          }
+          } else toast.error(done.error);
         } catch (err) {
           console.error("Failed to upload file:", err);
         } finally {

package/components/Workspaces/ProjectTiersForm.tsx

@@ -0,0 +1,33 @@
+"use client";
+
+import { SettingToggle } from "@/components/Settings/SettingToggle";
+import { useProjectsQuery, useUpdateProject } from "@/data/projects";
+
+// A workspace's projects, each switched between production (every merge gate)
+// and fast (a task's PR merges as soon as it opens).
+export function ProjectTiersForm({ workspaceId }: { workspaceId: string }) {
+  const { data: projects } = useProjectsQuery();
+  const update = useUpdateProject();
+  const mine = (projects ?? []).filter(
+    (p) => p.workspace_id === workspaceId && !p.is_uncategorized
+  );
+  if (!mine.length) return null;
+  return (
+    <div className="space-y-2">
+      {mine.map((p) => (
+        <SettingToggle
+          key={p.id}
+          title={`${p.name}: production`}
+          detail="On: CI, the independent review and scope gate its merges. Off (fast): a task's PR merges as soon as it opens."
+          checked={p.tier === "production"}
+          onChange={(on) =>
+            update.mutate({
+              projectId: p.id,
+              tier: on ? "production" : "fast",
+            })
+          }
+        />
+      ))}
+    </div>
+  );
+}

package/components/Workspaces/WorkspaceSettings.tsx

@@ -7,6 +7,7 @@ import { RulesForm } from "./RulesForm";
 import { FreshStartForm } from "./FreshStartForm";
 import { McpServersForm } from "./McpServersForm";
 import { HqFolderForm } from "./HqFolderForm";
+import { ProjectTiersForm } from "./ProjectTiersForm";
 import { StallForm } from "./StallForm";
 import { WorkspaceModelsForm } from "./WorkspaceModelsForm";
 import { AwayWindowForm } from "@/components/Away/AwayWindowForm";
@@ -49,6 +50,7 @@ export function WorkspaceSettings() {
           <WorkspaceModelsForm workspaceId={w.id} />
           <McpServersForm workspace={w} />
           <HqFolderForm workspaceId={w.id} />
+          <ProjectTiersForm workspaceId={w.id} />
           <SettingToggle
             title="Wake tasks on CI failures and reviews"
             detail="A task's agent is messaged once when a check fails on its PR's head, with the log's tail, and when someone reviews or comments on it."

package/data/chat/useChat.ts

@@ -9,6 +9,7 @@ import type {
   ChatAccess,
   ChatCommand,
   ChatContext,
+  ChatFile,
   ChatImage,
   ChatItem,
   ChatModel,
@@ -34,12 +35,13 @@ import { usePending } from "./usePending";
 
 // A pending send as the socket's `send`, under its own id.
 const sendFrame = (
-  p: Pick<PendingSend, "id" | "text" | "images" | "compact">
+  p: Pick<PendingSend, "id" | "text" | "images" | "files" | "compact">
 ) =>
   JSON.stringify({
     type: "send",
     text: p.text,
     images: p.images,
+    files: p.files,
     clientId: p.id,
     ...(p.compact && { compact: true }),
   });
@@ -400,8 +402,13 @@ export function useChat(
   const send = useCallback(
     // `compact`: the reader let a stale chat compact first (lib/chat/stale).
     // It shows as a pending bubble at once, until it's in the chat.
-    (text: string, images?: ChatImage[], compact?: boolean) => {
-      const p = { id: clientId(), text, images, compact };
+    (
+      text: string,
+      images?: ChatImage[],
+      compact?: boolean,
+      files?: ChatFile[]
+    ) => {
+      const p = { id: clientId(), text, images, files, compact };
       pend({ type: "add", send: p, now: Date.now() });
       const ws = wsRef.current;
       if (ws?.readyState !== WebSocket.OPEN) {

package/data/client-config/queries.ts

@@ -0,0 +1,18 @@
+import { queryOptions, useQuery } from "@tanstack/react-query";
+import type { ClientConfig } from "@/lib/config/client-config-types";
+
+export const clientConfigKeys = { all: ["client-config"] as const };
+
+// What the server says clients render with and are limited by; it only
+// changes with a server release, so one read per page load.
+export const clientConfigQuery = queryOptions({
+  queryKey: clientConfigKeys.all,
+  queryFn: async (): Promise<ClientConfig> => {
+    const res = await fetch("/api/client-config");
+    if (!res.ok) throw new Error("Couldn't read the server's limits");
+    return res.json();
+  },
+  staleTime: Infinity,
+});
+
+export const useClientConfig = () => useQuery(clientConfigQuery);

package/data/projects/queries.ts

@@ -3,6 +3,7 @@ import type {
   ProjectWithRepositories,
   DetectedDevServer,
 } from "@/lib/projects";
+import type { ProjectTier } from "@/lib/db/types";
 import { projectKeys } from "./keys";
 import { sessionKeys } from "../sessions/keys";
 
@@ -131,6 +132,7 @@ export function useUpdateProject() {
       agentType,
       defaultModel,
       initialPrompt,
+      tier,
     }: {
       projectId: string;
       name?: string;
@@ -138,6 +140,7 @@ export function useUpdateProject() {
       agentType?: string;
       defaultModel?: string;
       initialPrompt?: string | null;
+      tier?: ProjectTier;
     }) => {
       const res = await fetch(`/api/projects/${projectId}`, {
         method: "PATCH",
@@ -148,6 +151,7 @@ export function useUpdateProject() {
           agentType,
           defaultModel,
           initialPrompt,
+          tier,
         }),
       });
       if (!res.ok) throw new Error("Failed to update project");

package/data/sessions/queries.ts

@@ -1,7 +1,7 @@
 import { useQuery, useMutation, useQueryClient } from "@tanstack/react-query";
 import type { Session, Group } from "@/lib/db";
 import type { AgentType } from "@/lib/providers";
-import type { ChatAccess, ChatImage } from "@/lib/chat/events";
+import type { ChatAccess, ChatFile, ChatImage } from "@/lib/chat/events";
 import { sessionKeys } from "./keys";
 import { usePollWhenOffline, usePushConnected } from "../push/connection";
 
@@ -192,6 +192,7 @@ export interface LaunchSessionInput {
   baseBranch: string | null;
   prompt: string;
   images?: ChatImage[];
+  files?: ChatFile[];
 }
 
 interface LaunchSessionResponse {

package/lib/attachments/check.ts

@@ -0,0 +1,50 @@
+import { isText, sniff } from "../intake/sniff";
+import {
+  ATTACHMENT_LIMITS,
+  tooBigMessage,
+  typeFor,
+  unsupportedMessage,
+  type AttachmentType,
+} from "./policy";
+
+export type UploadCheck =
+  | { ok: true; type: AttachmentType }
+  | { ok: false; status: 400 | 413 | 415; error: string };
+
+const ZIP = [0x50, 0x4b, 0x03, 0x04];
+const OLE = [0xd0, 0xcf, 0x11, 0xe0];
+const OOXML_ZIP = new Set(["xlsx", "docx", "pptx", "ods", "odt"]);
+const OLE_EXTS = new Set(["xls", "doc"]);
+
+const starts = (b: Buffer, sig: number[]) => sig.every((v, i) => b[i] === v);
+
+const bad = (name: string): UploadCheck => ({
+  ok: false,
+  status: 400,
+  error: `${name} doesn't look like a real .${name.split(".").pop()?.toLowerCase()} file.`,
+});
+
+/** Whether a file's name, size and bytes are something a chat accepts. */
+export function checkUpload(name: string, data: Buffer): UploadCheck {
+  const type = typeFor(name);
+  if (!type) return { ok: false, status: 415, error: unsupportedMessage(name) };
+  const limit =
+    type.kind === "image"
+      ? ATTACHMENT_LIMITS.maxImageBytes
+      : ATTACHMENT_LIMITS.maxBytes;
+  if (data.length > limit)
+    return {
+      ok: false,
+      status: 413,
+      error: tooBigMessage(name, data.length, limit),
+    };
+  if (!data.length)
+    return { ok: false, status: 400, error: `${name} is empty.` };
+  let real: boolean;
+  if (type.kind === "image" || type.kind === "pdf")
+    real = sniff(data, name)?.mediaType === type.mediaType;
+  else if (OOXML_ZIP.has(type.ext)) real = starts(data, ZIP);
+  else if (OLE_EXTS.has(type.ext)) real = starts(data, OLE);
+  else real = isText(data);
+  return real ? { ok: true, type } : bad(name);
+}