@k2b/cloud 0.6.0 → 0.7.0

package/src/ai/chat/composer-adapter.ts

@@ -1,7 +1,7 @@
-import type { ChatAttachment, ChatModelOption, ChatSubmitInput } from "@k2b/ui";
+import type { ChatAttachment, ChatMention, ChatModelOption, ChatSubmitInput } from "@k2b/ui";
 import { AI_TURN_IMAGE_MAX_TOTAL_BYTES } from "../limits";
 import type { AiResourceMarker } from "../resource-markers";
-import type { AiPublicModelProfile, AiUserContentPart } from "../types";
+import type { AiDraftContentPart, AiPublicModelProfile, AiUserContentPart } from "../types";
 import {
   type AiComposerAttachment,
   FILE_INPUT_ACCEPT,
@@ -21,6 +21,7 @@ const AI_COMPOSER_TEXT_MAX_BYTES = AI_COMPOSER_TEXT_MAX_CHARS * 4;
 const AI_PASTED_TEXT_FILE_NAME = /^pasted-(?:text|[a-f0-9]{8})(?:-\d+)?\.txt$/i;
 
 export type AiComposerSendInput = {
+  draftContent?: AiDraftContentPart[];
   message?: string;
   content?: AiUserContentPart[];
   files?: File[];
@@ -96,7 +97,7 @@ const isAiComposerAttachment = (value: unknown): value is AiComposerAttachment =
   if (!value || typeof value !== "object") return false;
   const candidate = value as Partial<AiComposerAttachment>;
   return (
-    ["image", "file", "stored-file", "resource"].includes(candidate.kind ?? "") &&
+    ["image", "file", "stored-file", "resource", "project-file"].includes(candidate.kind ?? "") &&
     typeof candidate.id === "string" &&
     typeof candidate.name === "string"
   );
@@ -107,6 +108,24 @@ export const aiComposerAttachmentRecords = (attachments: readonly ChatAttachment
 
 export const aiComposerSendInput = (input: ChatSubmitInput): AiComposerSendInput => {
   const attachments = aiComposerAttachmentRecords(input.attachments);
+  if (input.mentions?.length || attachments.some(attachment => attachment.kind === "project-file")) {
+    const draftContent: AiDraftContentPart[] = [];
+    let offset = 0;
+    for (const mention of [...(input.mentions ?? [])].sort((a,b) => a.start - b.start)) {
+      if (mention.start < offset || mention.end > input.text.length || mention.start >= mention.end) continue;
+      const attachment = aiComposerAttachmentRecords([mention.attachment])[0];
+      if (!attachment || attachment.kind === "file" || attachment.kind === "image") continue;
+      if (mention.start > offset) draftContent.push({ type: "text", text: input.text.slice(offset, mention.start) });
+      draftContent.push(composerAttachmentPart(attachment, true));
+      offset = mention.end;
+    }
+    if (offset < input.text.length) draftContent.push({ type: "text", text: input.text.slice(offset) });
+    for (const attachment of attachments) {
+      if (attachment.kind !== "file" && attachment.kind !== "image") draftContent.push(composerAttachmentPart(attachment));
+    }
+    const files = attachments.flatMap(attachment => attachment.kind === "file" || attachment.kind === "image" ? [attachment.file] : []);
+    return { draftContent, files: files.length ? files : undefined };
+  }
   const files = attachments.filter(
     (attachment): attachment is PendingAiImage | PendingAiVfsFile => attachment.kind === "image" || attachment.kind === "file",
   );
@@ -167,3 +186,29 @@ export const readAiComposerFiles = async (
     discarded: Math.max(0, files.length - candidates.length),
   };
 };
+
+const composerAttachmentPart = (attachment: Exclude<AiComposerAttachment, {kind: "image" | "file"}>, inline = false): AiDraftContentPart => {
+  if (attachment.kind === "resource") return { type: "resource", ref: attachment.ref, title: attachment.name, icon: attachment.icon, href: attachment.href, inline };
+  if (attachment.kind === "project-file") return { type: "project-file", path: attachment.path, inline };
+  return { type: "file", path: attachment.path, version: attachment.version, mediaType: attachment.mediaType, size: attachment.size, inline };
+};
+
+/** Hydrate the same ordered content used by the server draft and queued messages. */
+export function aiComposerDraft(content: readonly AiDraftContentPart[]): { text: string; mentions: ChatMention[]; attachments: AiComposerAttachment[] } {
+  let text = "";
+  const mentions: ChatMention[] = [];
+  const attachments: AiComposerAttachment[] = [];
+  for (const part of content) {
+    if (part.type === "text") { text += part.text; continue; }
+    const attachment: AiComposerAttachment = part.type === "resource"
+      ? { kind: "resource", id: `resource:${part.ref.type}:${part.ref.id}`, ref: part.ref, name: part.title ?? part.ref.id, icon: part.icon ?? "ti ti-link", href: part.href }
+      : part.type === "project-file"
+        ? { kind: "project-file", id: part.path, path: part.path, name: part.path, icon: "ti ti-file" }
+        : { kind: "stored-file", id: `file:${part.path}:${part.version}`, path: part.path, name: part.path.split("/").at(-1) || part.path, version: part.version, mediaType: part.mediaType, size: part.size, icon: "ti ti-file" };
+    if (part.inline) {
+      mentions.push({ start: text.length, end: text.length + attachment.name.length, attachment: aiChatAttachments([attachment])[0]! });
+      text += attachment.name;
+    } else attachments.push(attachment);
+  }
+  return { text, mentions, attachments };
+}

package/src/ai/chat/message-actions.tsx

@@ -28,6 +28,8 @@ type RetryMessageHandler = (entry: AiStoredMessage, input?: AiRetryMessageInput)
 type RetrySteerHandler = (block: Extract<AiTurnBlock, { kind: "steer_message" }>) => void | Promise<void>;
 
 export type AiChatActions = {
+  /** Host-owned visualizations, persisted independently of the execution session. */
+  renderCodePresentation?: (result: unknown) => JSX.Element;
   /** Prevents turn-continuation actions while the current turn is stopping. */
   actionDisabled?: () => boolean;
   onApproval?: ApprovalHandler;
@@ -36,8 +38,12 @@ export type AiChatActions = {
   onRetryMessage?: RetryMessageHandler;
   onRetrySteer?: RetrySteerHandler;
   onMessageFeedback?: (entry: AiStoredMessage, feedback: Omit<AiMessageFeedback, "updatedAt"> | null) => void | Promise<void>;
+  /** Open the originating background run for a delivered scheduled result. */
+  onOpenScheduledTaskRun?: (taskId: string, occurrenceId: string) => void;
   /** Open a conversation VFS file in the host application's artifact surface. */
   onOpenFile?: (path: string) => void;
+  /** Resolve a Markdown link only when it identifies a known conversation file. */
+  resolveFileLink?: (href: string) => { path: string; href: string } | null;
   /** Download URL for a conversation VFS file (present blocks, attachment chips). */
   fileUrl?: (path: string) => string | null;
 };
@@ -53,8 +59,8 @@ export function AiChatActionsProvider(props: { actions?: AiChatActions; children
   return <AiChatActionContext.Provider value={props.actions ?? {}}>{props.children}</AiChatActionContext.Provider>;
 }
 
-const usageValue = (usage: Usage | null | undefined, key: "input" | "output" | "total" | "creditsUsed") => {
-  const value = (usage as Partial<Record<"input" | "output" | "total" | "creditsUsed", unknown>> | null | undefined)?.[key];
+const usageValue = (usage: Usage | null | undefined, key: "input" | "output" | "total") => {
+  const value = (usage as Partial<Record<"input" | "output" | "total", unknown>> | null | undefined)?.[key];
   return typeof value === "number" && Number.isFinite(value) ? value : null;
 };
 
@@ -70,7 +76,6 @@ const assistantResponseInfo = (entries: AiStoredMessage[], locale: string) => {
   const toolCalls = blocks.filter((block) => block.type === "tool_call");
   const aggregate = entry.loopAggregate;
   const usage = aggregate?.usage ?? entry.usage;
-  const credits = usageValue(usage, "creditsUsed");
   const timing = aggregate?.timing;
 
   const meta = [
@@ -100,7 +105,6 @@ const assistantResponseInfo = (entries: AiStoredMessage[], locale: string) => {
       input: usageValue(usage, "input"),
       output: usageValue(usage, "output"),
       total: usageValue(usage, "total"),
-      credits,
     },
     turns: aggregate?.assistantMessageCount ?? 1,
     toolCallCount: aggregate?.toolCallCount ?? toolCalls.length,
@@ -184,9 +188,6 @@ const openAssistantResponseInfo = (entries: AiStoredMessage[], locale: string) =
                 accent={{ tone: "red", icon: "ti ti-alert-triangle" }}
               />
             </Show>
-            <Show when={info.usage.credits !== null && info.usage.credits > 0}>
-              <StatCell label="Credits" value={info.usage.credits?.toLocaleString(locale, { maximumFractionDigits: 6 }) ?? "–"} />
-            </Show>
           </StatGrid>
 
           <Show when={info.tools.length > 0}>

package/src/ai/chat/message-utils.ts

@@ -1,3 +1,4 @@
+import { AI_AUDIO_EXTENSIONS } from "../audio-format";
 import type { Message, Usage } from "@k2b/nessi";
 import { fileIcons } from "@k2b/stdlib";
 import { formatBytes as sharedFormatBytes } from "../../shared/format";
@@ -20,6 +21,7 @@ export type AiForkMessageInput = {
 };
 
 export type AiComposerAttachment =
+  | { kind: "project-file"; id: string; name: string; path: string; icon: string }
   | {
       kind: "image";
       id: string;
@@ -111,6 +113,8 @@ export const TEXT_ATTACHMENT_MEDIA_TYPES = new Set([
 ]);
 export const FILE_INPUT_ACCEPT = [
   ...AI_IMAGE_MEDIA_TYPES,
+  "audio/*",
+  ...AI_AUDIO_EXTENSIONS.map((extension) => `.${extension}`),
   "text/*",
   ...Array.from(TEXT_ATTACHMENT_MEDIA_TYPES),
   ...Array.from(new Set([...TEXT_ATTACHMENT_EXTENSIONS, ...DOCUMENT_ATTACHMENT_EXTENSIONS]), (extension) => `.${extension}`),
@@ -228,14 +232,20 @@ export type AiLatestUsageSnapshot = {
 export const latestUsageSnapshot = (messages: AiStoredMessage[]): AiLatestUsageSnapshot | null => {
   for (let i = messages.length - 1; i >= 0; i--) {
     const entry = messages[i];
+    if (entry?.message.role !== "assistant") continue;
     const request = entry?.loopAggregate?.turns.findLast((turn) => Boolean(turn.usage))?.usage ?? entry?.usage;
     if (request) {
       return {
         request,
-        loop: entry?.loopAggregate?.usage ?? entry?.usage ?? request,
+        loop: entry?.loopAggregate?.usage ?? (entry?.loopId ? messages.filter(message => message.loopId === entry.loopId && message.message.role === "assistant").reduce<Usage>((sum, message) => ({
+          input: sum.input + (message.usage?.input ?? 0), output: sum.output + (message.usage?.output ?? 0), total: sum.total + (message.usage?.total ?? 0),
+          ...((sum.cacheRead !== undefined || message.usage?.cacheRead !== undefined) ? { cacheRead: (sum.cacheRead ?? 0) + (message.usage?.cacheRead ?? 0) } : {}),
+        }), { input: 0, output: 0, total: 0 }) : request),
         modelProfileId: entry?.modelProfileId ?? null,
       };
     }
+    // A provider without usage must not inherit a previous request's numbers.
+    return null;
   }
   return null;
 };
@@ -294,6 +304,7 @@ const BUILT_IN_TOOL_ICONS = new Map<string, string>([
   ["search_tools", "ti ti-ai-gateway"],
   ["load_tools", "ti ti-ai-gateway"],
   ["load_skill", "ti ti-sparkles"],
+  ["todo_write", "ti ti-list-check"],
   ["list_apps", "ti ti-apps"],
   ["read_cloud_resource", "ti ti-ai-gateway"],
 ]);

package/src/ai/chat/messages.ts

@@ -4,6 +4,7 @@ const messages = i18n.define({
   baseLocale: "en",
   messages: {
     en: {
+      backgroundRun: "Background run",
       tableLinks: "Links",
       tableOpen: "Open",
       tableEmpty: "No matching rows.",
@@ -20,6 +21,7 @@ const messages = i18n.define({
       byteRange: ({ start, end }: { start: string; end: string }) => `Bytes ${start}–${end}`,
     },
     de: {
+      backgroundRun: "Hintergrundlauf",
       tableLinks: "Links",
       tableOpen: "Öffnen",
       tableEmpty: "Keine passenden Zeilen.",

package/src/ai/chat/presentation.tsx

@@ -1,4 +1,4 @@
-import { Chat, type ChatTimelineItem, useLocale } from "@k2b/ui";
+import { Button, Chat, type ChatTimelineItem, useLocale } from "@k2b/ui";
 import { type Accessor, createEffect, createMemo, createSignal, type JSX, onCleanup, Show } from "solid-js";
 import type { AiActiveTurn } from "../client/projection";
 import { type AiActiveTurnSegment, isRenderableTurnBlock, splitActiveTurnBlocks } from "../protocol";
@@ -6,7 +6,7 @@ import { type AiAssistantTimelineItem, buildAiMessageTimeline, copyTextFromAssis
 import type { AiConversationTimelineEntry, AiStoredMessage } from "../types";
 import { AiTurnBlockList } from "./blocks";
 import { type AiChatActions, AiChatActionsProvider, createAssistantMessageActions, useAiChatActions } from "./message-actions";
-import { formatWorkedDuration, isCardToolName, isRecord, isSurveyToolName, textFromMessage } from "./message-utils";
+import { formatWorkedDuration, isRecord, isSurveyToolName, textFromMessage } from "./message-utils";
 import { type AiToolDisclosureState, createAiToolDisclosureState } from "./tool-disclosure";
 import { aiChatMessages } from "./messages";
 import { CloudSurveyResultBlock } from "./visual-tools";
@@ -63,53 +63,17 @@ const surveyItem = (block: SurveyResultBlock, turnId: string): ChatTimelineItem
   content: <CloudSurveyResultBlock args={block.args} result={block.result} />,
 });
 
-const isDirectCompletedResult = (block: AssistantBlock): boolean =>
-  block.kind === "tool" && block.status === "completed" && !block.isError && (isCardToolName(block.name) || block.name === "present");
-
-const isFailedBlock = (block: AssistantBlock): boolean =>
-  (block.kind === "tool" && (block.isError || block.status === "failed" || block.status === "rejected")) ||
-  (block.kind === "compaction" && block.status === "failed");
-
-export const partitionCompletedAssistantBlocks = (blocks: AssistantBlock[]) => {
-  const renderable = blocks.filter(isRenderableTurnBlock);
-  const finalTextIndex = renderable.findLastIndex((block) => block.kind === "text");
-  return {
-    worked: renderable.filter((block, index) => index !== finalTextIndex && !isDirectCompletedResult(block)),
-    visible: renderable.filter((block, index) => index === finalTextIndex || isDirectCompletedResult(block)),
-  };
-};
-
 export function AiAssistantContent(props: {
   item: AiAssistantTimelineItem;
   disclosureState?: AiToolDisclosureState;
   segmentId?: string;
 }): JSX.Element {
   const locale = useLocale();
-  const blocks = createMemo(() => partitionCompletedAssistantBlocks(props.item.blocks));
   const turnId = () => props.item.loopId ?? props.item.id;
-  const workedDisclosureId = () => `worked:${props.segmentId ?? turnId()}`;
-  const workedOpen = () => props.disclosureState?.get(workedDisclosureId());
-  const setWorkedOpen = (open: boolean) => props.disclosureState?.set(workedDisclosureId(), open);
-  const workedFailed = () => blocks().worked.some(isFailedBlock);
-
-  return (
-    <div class="flex flex-col gap-2">
-      <Show when={blocks().worked.length > 0}>
-        <Chat.Activity
-          icon="ti ti-route"
-          label={props.segmentId ? aiChatMessages(locale()).workSteps : `Worked for ${formatWorkedDuration(props.item.workedMs)}`}
-          tone={workedFailed() ? "danger" : undefined}
-          bodyInset={false}
-          defaultOpen={workedFailed()}
-          open={workedOpen()}
-          onOpenChange={setWorkedOpen}
-        >
-          <AiTurnBlockList blocks={blocks().worked} turnId={turnId()} compact disclosureState={props.disclosureState} />
-        </Chat.Activity>
-      </Show>
-      <AiTurnBlockList blocks={blocks().visible} turnId={turnId()} disclosureState={props.disclosureState} />
-    </div>
-  );
+  return <div class="flex flex-col gap-2">
+    <Chat.Activity icon="ti ti-route" label={`${locale().startsWith("de") ? "Gearbeitet für" : "Worked for"} ${formatWorkedDuration(props.item.workedMs)}`} />
+    <AiTurnBlockList blocks={props.item.blocks} turnId={turnId()} disclosureState={props.disclosureState} />
+  </div>;
 }
 
 const storedItems = (
@@ -206,11 +170,18 @@ const storedItems = (
         createdAt: actionEntry?.createdAt ?? item.entries.at(-1)?.createdAt,
         class: segment.blocks.some(isWideBlock) ? "ai-chat-message-wide" : undefined,
         content: (
-          <AiAssistantContent
-            item={{ ...item, blocks: segment.blocks }}
-            segmentId={segments.length > 1 ? `${item.id}:${index}` : undefined}
-            disclosureState={disclosureState}
-          />
+          <>
+            <Show when={index === 0 && item.entries.find(entry => entry.meta?.scheduledTask)?.meta?.scheduledTask}>
+              {task => <Button variant="ghost" size="xs" class="mb-2" onClick={() => actions.onOpenScheduledTaskRun?.(task().taskId, task().occurrenceId)} disabled={!actions.onOpenScheduledTaskRun}>
+                <i class="ti ti-calendar-time" aria-hidden="true" /> {aiChatMessages(locale).backgroundRun}
+              </Button>}
+            </Show>
+            <AiAssistantContent
+              item={{ ...item, blocks: segment.blocks }}
+              segmentId={segments.length > 1 ? `${item.id}:${index}` : undefined}
+              disclosureState={disclosureState}
+            />
+          </>
         ),
         actions:
           actionEntry && index === lastAssistant
@@ -268,7 +239,7 @@ const activeItems = (turn: AiActiveTurn | null, actions: AiChatActions, disclosu
           blocks={blocks}
           turnId={turn.turnId}
           streaming={turn.status === "running" && index === segments.length - 1}
-          active
+          active={index === segments.length - 1}
           disclosureState={disclosureState}
         />
       ),

package/src/ai/chat/primitives.tsx

@@ -1,3 +1,45 @@
+import { createEffect, onCleanup, onMount } from "solid-js";
+import { useAiChatActions } from "./message-actions";
+
 export function AssistantMarkdownBlock(props: { html: string }) {
-  return <div class="assistant-markdown-block" innerHTML={props.html} />;
+  const actions = useAiChatActions();
+  let element!: HTMLDivElement;
+  const originals = new WeakMap<HTMLAnchorElement, { href: string; target: string | null }>();
+  const files = new WeakMap<HTMLAnchorElement, string>();
+  createEffect(() => {
+    props.html;
+    for (const link of Array.from(element.querySelectorAll<HTMLAnchorElement>("a[href]"))) {
+      const original = originals.get(link) ?? { href: link.getAttribute("href")!, target: link.getAttribute("target") };
+      originals.set(link, original);
+      const file = actions.resolveFileLink?.(original.href);
+      link.setAttribute("href", file?.href ?? original.href);
+      const target = file ? null : original.target;
+      if (target) link.setAttribute("target", target);
+      else link.removeAttribute("target");
+      if (file) files.set(link, file.path);
+      else files.delete(link);
+    }
+  });
+  onMount(() => {
+    const handleClick = (event: MouseEvent) => {
+      if (
+        event.defaultPrevented ||
+        event.button !== 0 ||
+        event.metaKey ||
+        event.ctrlKey ||
+        event.shiftKey ||
+        event.altKey ||
+        !actions.onOpenFile
+      )
+        return;
+      const link = event.target instanceof Element ? event.target.closest("a") : null;
+      const path = link ? files.get(link) : undefined;
+      if (!path) return;
+      event.preventDefault();
+      actions.onOpenFile(path);
+    };
+    element.addEventListener("click", handleClick);
+    onCleanup(() => element.removeEventListener("click", handleClick));
+  });
+  return <div ref={element} class="assistant-markdown-block" innerHTML={props.html} />;
 }

package/src/ai/chat/tool-disclosure.tsx

@@ -8,9 +8,24 @@ export type AiToolDisclosureState = {
 
 export const createAiToolDisclosureState = (): AiToolDisclosureState => {
   const openByBlockId = new Map<string, boolean>();
+  const storageKey = (blockId: string) => `cloud.ai.tool-disclosure:${blockId}`;
   return {
-    get: (blockId) => openByBlockId.get(blockId),
-    set: (blockId, open) => openByBlockId.set(blockId, open),
+    get: (blockId) => {
+      if (!openByBlockId.has(blockId)) {
+        try {
+          const saved = window.sessionStorage.getItem(storageKey(blockId));
+          if (saved === "open") openByBlockId.set(blockId, true);
+        } catch { /* SSR and disabled storage keep the in-memory state. */ }
+      }
+      return openByBlockId.get(blockId);
+    },
+    set: (blockId, open) => {
+      openByBlockId.set(blockId, open);
+      try {
+        if (open) window.sessionStorage.setItem(storageKey(blockId), "open");
+        else window.sessionStorage.removeItem(storageKey(blockId));
+      } catch { /* Storage limits must not prevent expanding a tool. */ }
+    },
   };
 };
 

package/src/ai/chat/tool-groups.ts

@@ -0,0 +1,52 @@
+import { hasCapabilityTable } from "./capability-result";
+import type { AiTurnBlock } from "../protocol";
+import { isCardToolName, isSurveyToolName, isTextEditorToolName } from "./message-utils";
+
+type Tool = Extract<AiTurnBlock, { kind: "tool" }>;
+type Group = { kind: "tools"; blocks: Tool[] } | { kind: "block"; block: AiTurnBlock };
+
+export function isFailedTool(tool: Tool): boolean {
+  if (tool.isError || tool.status === "failed" || tool.status === "rejected") return true;
+  const result = tool.result;
+  return ["code_run","code_inspect","code_interact"].includes(tool.name) && result !== null && typeof result === "object" && "status" in result && result.status === "error";
+}
+
+/** Visible content and interactive decisions are boundaries; model request boundaries are not. */
+export function groupToolBlocks(blocks: readonly AiTurnBlock[]): Group[] {
+  const groups: Group[] = [];
+  for (const block of blocks) {
+    const ordinary =
+      block.kind === "tool" &&
+      !block.approval &&
+      block.status !== "awaiting_approval" &&
+      !["present", "code_present", "view_image", "code_secret"].includes(block.name) &&
+      !isCardToolName(block.name) &&
+      !isSurveyToolName(block.name) &&
+      !isTextEditorToolName(block.name) &&
+      !hasCapabilityTable(block);
+    if (ordinary) {
+      const last = groups.at(-1);
+      if (last?.kind === "tools") last.blocks.push(block);
+      else groups.push({ kind: "tools", blocks: [block] });
+    } else groups.push({ kind: "block", block });
+  }
+  return groups;
+}
+
+export function summarizeToolGroup(tools: readonly Tool[], locale: string): string {
+  const de = locale.startsWith("de");
+  const categories = tools.map((tool) => {
+    if (tool.name === "todo_write") return de ? "Arbeitsplan aktualisiert" : "Updated working plan";
+    if (tool.name.startsWith("code_")) return de ? "Mit Code gearbeitet" : "Worked with code";
+    if (tool.presentation?.kind === "capability" || tool.name === "read_cloud_resource")
+      return de ? "Cloud-Integration verwendet" : "Used Cloud integration";
+    if (["read_file", "list_files", "view_image"].includes(tool.name)) return de ? "Dateien gelesen" : "Read files";
+    if (tool.name === "write_file") return de ? "Dateien geschrieben" : "Wrote files";
+    if (["load_skill", "load_tools", "search_tools"].includes(tool.name))
+      return de ? "Werkzeuge und Wissen geladen" : "Loaded tools and guidance";
+    if (tool.name.startsWith("web_") || tool.name === "fetch_file") return de ? "Im Web recherchiert" : "Searched the web";
+    return de ? "Werkzeuge verwendet" : "Used tools";
+  });
+  const errors = tools.filter(isFailedTool).length;
+  return [...new Set(categories)].join(", ") + (errors ? ` · ${errors} ${de ? "fehlgeschlagen" : "failed"}` : "");
+}

package/src/ai/chat-quotas.ts

@@ -0,0 +1,25 @@
+import { hasBillableAiPricing } from "../shared/ai-costs";
+import { readAiSettingsState } from "./settings";
+import type { AccessSubject } from "../server/services/access";
+import type { AiChatQuotaSnapshot } from "../shared/ai-quotas";
+import { listAssistantAiModels } from "./assistant-models";
+import { aiQuotas } from "./quotas";
+
+/** Own allowances only. Grant identities and unavailable model profiles stay private. */
+export async function getAiChatQuotas(subject: AccessSubject): Promise<AiChatQuotaSnapshot> {
+  const config = await aiQuotas.config();
+  if (!config.enabled) return { enabled: false, balances: [] };
+  const models = await listAssistantAiModels(subject);
+  const snapshot = await aiQuotas.snapshot(subject);
+  const profiles = (await readAiSettingsState()).profiles;
+  return {
+    enabled: snapshot.enabled,
+    unit: config.unit ?? "EUR",
+    unlimitedModels: models
+      .filter((model) => !hasBillableAiPricing(profiles.find((profile) => profile.id === model.id)?.pricing))
+      .map((model) => model.id),
+    balances: snapshot.balances
+      .filter((b) => b.scope === "*" || models.some((m) => m.id === b.scope))
+      .map(({ sources: _sources, sourceDetails: _sourceDetails, ...balance }) => balance),
+  };
+}

package/src/ai/chat-task-contracts.ts

@@ -1,3 +1,5 @@
+import { CapabilityGrantsSchema, type CapabilityGrant } from "../services/mandates";
+export { CapabilityGrantsSchema as ChatTaskGrantsSchema };
 import { dates } from "@k2b/stdlib";
 import {
   type AiChatTask,
@@ -16,9 +18,11 @@ export const ChatTaskOccurrenceIdSchema = z.string().regex(AI_SHORT_ID_PATTERN).
 export const AiConversationIdSchema = z.string().regex(AI_SHORT_ID_PATTERN).describe("Readable six-character AI conversation ID.");
 
 export type AiChatTaskView = {
+  nextRunAt?: string | null;
   id: string;
   chatId: string;
   chatTitle: string;
+  grants: CapabilityGrant[];
   prompt: string;
   schedule: AiChatTaskSchedule;
   timezone: string;
@@ -35,6 +39,7 @@ export type AiChatTaskOccurrenceView = {
   trigger: "scheduled" | "manual";
   state: AiChatTaskOccurrenceState;
   error: string | null;
+  resultText: string | null;
   createdAt: string;
   startedAt: string | null;
   completedAt: string | null;
@@ -44,6 +49,7 @@ export const toAiChatTaskView = (task: AiChatTask): AiChatTaskView => ({
   id: task.shortId,
   chatId: task.chatId,
   chatTitle: task.chatTitle,
+  grants: task.grants,
   prompt: task.prompt,
   schedule: task.schedule,
   timezone: task.timezone,
@@ -60,6 +66,7 @@ export const toAiChatTaskOccurrenceView = (occurrence: AiChatTaskOccurrence, tas
   trigger: occurrence.trigger,
   state: occurrence.state,
   error: occurrence.error,
+  resultText: occurrence.resultText,
   createdAt: occurrence.createdAt,
   startedAt: occurrence.startedAt,
   completedAt: occurrence.completedAt,
@@ -92,9 +99,10 @@ export const chatTaskCreateFingerprint = (input: {
   prompt: string;
   schedule: z.infer<typeof ChatTaskScheduleInputSchema>;
   timezone?: string;
+  grants?: CapabilityGrant[];
 }): string =>
   new Bun.CryptoHasher("sha256")
-    .update(JSON.stringify([input.chatId, input.prompt.trim(), input.schedule, input.timezone ?? null]))
+    .update(JSON.stringify([input.chatId, input.prompt.trim(), input.schedule, input.timezone ?? null, input.grants ?? []]))
     .digest("hex");
 
 export const normalizeChatTaskSchedule = async (