@cfbender/cesium 0.7.1 → 0.7.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cfbender/cesium",
-  "version": "0.7.1",
+  "version": "0.7.2",
   "description": "Beautiful self-contained HTML artifacts from your opencode agent.",
   "keywords": [
     "agent",

package/src/index.ts

@@ -12,6 +12,7 @@ import { createStyleguideTool } from "./tools/styleguide.ts";
 import { createCritiqueTool } from "./tools/critique.ts";
 import { createStopTool } from "./tools/stop.ts";
 import { generateBlockFieldReference } from "./prompt/field-reference.ts";
+import { recordSessionModel } from "./session-model.ts";
 
 const rawFragment = await readFile(
   join(dirname(fileURLToPath(import.meta.url)), "prompt/system-fragment.md"),
@@ -37,6 +38,11 @@ export const CesiumPlugin: Plugin = async (ctx): Promise<Hooks> => {
     "experimental.chat.system.transform": async (_input, output) => {
       output.system.push(PROMPT_FRAGMENT);
     },
+    "chat.params": async (input, _output) => {
+      // Record the model resolved for this session so tool handlers can stamp
+      // it onto generated artifacts. Latest call wins (model can change mid-session).
+      recordSessionModel(input.sessionID, input.model.providerID, input.model.id);
+    },
   };
 };
 

package/src/session-model.ts

@@ -0,0 +1,23 @@
+// Tracks the model in use per opencode session.
+//
+// opencode's `chat.params` hook fires before each LLM request with the resolved
+// Model. We record `providerID/id` keyed by sessionID so that tool handlers
+// (publish/ask/annotate) can stamp the generating model onto every artifact's
+// metadata footer. The map is process-local and resets on plugin reload.
+
+const sessionModels = new Map<string, string>();
+
+export function recordSessionModel(sessionID: string, providerID: string, modelID: string): void {
+  if (!sessionID || !providerID || !modelID) return;
+  sessionModels.set(sessionID, `${providerID}/${modelID}`);
+}
+
+export function getSessionModel(sessionID: string | undefined | null): string | null {
+  if (!sessionID) return null;
+  return sessionModels.get(sessionID) ?? null;
+}
+
+/** Test-only: clear all recorded session models. */
+export function resetSessionModels(): void {
+  sessionModels.clear();
+}

package/src/tools/annotate.ts

@@ -26,6 +26,7 @@ import {
   type LifecycleConfig,
 } from "../server/lifecycle.ts";
 import { buildTerminalSummary, resolveDisplayHost } from "./publish.ts";
+import { getSessionModel } from "../session-model.ts";
 
 export interface AnnotateToolOverrides {
   loadConfig?: () => CesiumConfig;
@@ -84,7 +85,7 @@ export function createAnnotateTool(
       tags: tool.schema.array(tool.schema.string()).optional(),
       expiresAt: tool.schema.string().optional(),
     },
-    async execute(args, _context) {
+    async execute(args, context) {
       // 1. Validate top-level input shape
       const validation = validateAnnotateInput(args);
       if (!validation.ok) {
@@ -181,8 +182,8 @@ export function createAnnotateTool(
         summary: input.summary ?? null,
         tags: input.tags ?? [],
         createdAt: createdAt.toISOString(),
-        model: null,
-        sessionId: null,
+        model: getSessionModel(context.sessionID),
+        sessionId: context.sessionID ?? null,
         projectSlug: identity.slug,
         projectName: identity.name,
         cwd: ctx.directory,

package/src/tools/ask.ts

@@ -25,6 +25,7 @@ import {
   type LifecycleConfig,
 } from "../server/lifecycle.ts";
 import { buildTerminalSummary, resolveDisplayHost } from "./publish.ts";
+import { getSessionModel } from "../session-model.ts";
 
 export interface AskToolOverrides {
   loadConfig?: () => CesiumConfig;
@@ -106,7 +107,7 @@ export function createAskTool(
       expiresAt: tool.schema.string().optional(),
       requireAll: tool.schema.boolean().optional(),
     },
-    async execute(args, _context) {
+    async execute(args, context) {
       // 1. Validate input
       const validation = validateAskInput(args);
       if (!validation.ok) {
@@ -193,8 +194,8 @@ export function createAskTool(
         summary: input.summary ?? null,
         tags: input.tags ?? [],
         createdAt: createdAt.toISOString(),
-        model: null,
-        sessionId: null,
+        model: getSessionModel(context.sessionID),
+        sessionId: context.sessionID ?? null,
         projectSlug: identity.slug,
         projectName: identity.name,
         cwd: ctx.directory,

package/src/tools/publish.ts

@@ -31,6 +31,7 @@ import {
   type RunningInfo,
   type LifecycleConfig,
 } from "../server/lifecycle.ts";
+import { getSessionModel } from "../session-model.ts";
 
 export interface PublishToolOverrides {
   loadConfig?: () => CesiumConfig;
@@ -107,7 +108,7 @@ export function createPublishTool(
       tags: tool.schema.array(tool.schema.string()).optional(),
       supersedes: tool.schema.string().optional(),
     },
-    async execute(args, _context) {
+    async execute(args, context) {
       // 1. Validate input
       const validation = validatePublishInput(args);
       if (!validation.ok) {
@@ -190,8 +191,8 @@ export function createPublishTool(
         summary: input.summary ?? null,
         tags: input.tags ?? [],
         createdAt: createdAt.toISOString(),
-        model: null,
-        sessionId: null,
+        model: getSessionModel(context.sessionID),
+        sessionId: context.sessionID ?? null,
         projectSlug: identity.slug,
         projectName: identity.name,
         cwd: ctx.directory,