@anvia/studio 0.5.14 → 0.6.0

package/dist/index.js

@@ -835,16 +835,29 @@ function transcriptFromMessages(messages) {
       continue;
     }
     if (message.role === "user") {
+      const attachments = attachmentsFromMessage(message);
+      let textEntryAdded = false;
       for (const content of message.content) {
         if (content.type === "text") {
           transcript.push({
             entryId: transcript.length,
             kind: "message",
             role: "user",
-            text: content.text
+            text: content.text,
+            ...attachments.length === 0 ? {} : { attachments }
           });
+          textEntryAdded = true;
         }
       }
+      if (!textEntryAdded && attachments.length > 0) {
+        transcript.push({
+          entryId: transcript.length,
+          kind: "message",
+          role: "user",
+          text: "",
+          attachments
+        });
+      }
       continue;
     }
     if (message.role === "tool") {
@@ -885,6 +898,32 @@ function transcriptFromMessages(messages) {
   }
   return transcript;
 }
+function attachmentsFromMessage(message) {
+  if (message.role !== "user" && message.role !== "assistant") {
+    return [];
+  }
+  return message.content.flatMap((content) => {
+    if (content.type === "image") {
+      return [
+        {
+          kind: "image",
+          ...content.source.type === "base64" ? { data: content.source.data, mediaType: content.source.mediaType } : { url: content.source.url }
+        }
+      ];
+    }
+    if (content.type === "document") {
+      return [
+        {
+          kind: "document",
+          ...content.source.filename === void 0 ? {} : { name: content.source.filename },
+          ...content.source.mediaType === void 0 ? {} : { mediaType: content.source.mediaType },
+          ...content.source.type === "base64" ? { data: content.source.data } : content.source.type === "url" ? { url: content.source.url } : {}
+        }
+      ];
+    }
+    return [];
+  });
+}
 function appendAssistantTranscriptText(transcript, text) {
   const last = transcript.at(-1);
   if (last?.kind === "message" && last.role === "assistant") {
@@ -940,6 +979,19 @@ var InMemoryStudioStore = class {
     const session = this.sessions.get(id);
     return session === void 0 ? void 0 : materializeSession(session);
   }
+  updateSessionMetadata(id, metadata) {
+    const session = this.sessions.get(id);
+    if (session === void 0) {
+      return void 0;
+    }
+    if (metadata === void 0) {
+      delete session.metadata;
+    } else {
+      session.metadata = metadata;
+    }
+    session.updatedAt = (/* @__PURE__ */ new Date()).toISOString();
+    return materializeSession(session);
+  }
   load(context) {
     return Promise.resolve(this.sessions.get(context.sessionId)?.messages ?? []);
   }
@@ -1145,6 +1197,342 @@ function isJsonValue(value) {
   return isJsonObject(value) && Object.values(value).every(isJsonValue);
 }
 
+// src/runtime/models.ts
+var STUDIO_MODEL_METADATA_KEY = "studioModel";
+function createStudioModelRegistry(config) {
+  if (config === void 0) {
+    return void 0;
+  }
+  const providers = /* @__PURE__ */ new Map();
+  for (const provider of config.providers) {
+    const id = normalizeProviderId(provider.id);
+    if (providers.has(id)) {
+      throw new Error(`Duplicate Studio model provider id: ${id}`);
+    }
+    const staticModels = /* @__PURE__ */ new Map();
+    for (const model of provider.models ?? []) {
+      const modelId = normalizeModelId(model.id);
+      if (staticModels.has(modelId)) {
+        throw new Error(`Duplicate Studio model id for provider ${id}: ${modelId}`);
+      }
+      staticModels.set(modelId, { ...model, id: modelId });
+    }
+    providers.set(id, {
+      ...provider,
+      id,
+      ...provider.defaultModel === void 0 ? {} : { defaultModel: normalizeModelId(provider.defaultModel) },
+      staticModels
+    });
+  }
+  return {
+    ...config.default === void 0 ? {} : { defaultModel: normalizeModelRef(config.default) },
+    providers,
+    agentPolicies: normalizeAgentPolicies(config.agents ?? {}),
+    modelCache: /* @__PURE__ */ new Map()
+  };
+}
+function studioModelsConfig(registry, agents) {
+  if (registry === void 0) {
+    return void 0;
+  }
+  const providers = [...registry.providers.values()].map((provider) => {
+    const models = [...provider.staticModels.values()].map(
+      (model) => modelSummary(provider, model.id, model)
+    );
+    return {
+      id: provider.id,
+      ...provider.name === void 0 ? {} : { name: provider.name },
+      ...provider.defaultModel === void 0 ? {} : { defaultModel: provider.defaultModel },
+      models,
+      ...provider.metadata === void 0 ? {} : { metadata: provider.metadata }
+    };
+  });
+  const agentIds = new Set(agents.map((agent) => agent.id));
+  const agentsConfig = Object.fromEntries(
+    [...registry.agentPolicies.entries()].filter(([agentId]) => agentIds.has(agentId)).map(([agentId, policy]) => [agentId, publicPolicy(policy)])
+  );
+  return {
+    providers,
+    ...registry.defaultModel === void 0 ? {} : { default: registry.defaultModel },
+    agents: agentsConfig
+  };
+}
+function registerModelRoutes(app, props) {
+  app.get("/models", async (c) => {
+    if (props.registry === void 0) {
+      return errorResponse(c, 404, "not_found", "Model registry not configured");
+    }
+    const providers = await Promise.all(
+      [...props.registry.providers.values()].map((provider) => providerCatalog(provider))
+    );
+    return c.json({
+      providers,
+      ...props.registry.defaultModel === void 0 ? {} : { defaultModel: props.registry.defaultModel }
+    });
+  });
+  app.get("/models/:providerId", async (c) => {
+    if (props.registry === void 0) {
+      return errorResponse(c, 404, "not_found", "Model registry not configured");
+    }
+    const provider = props.registry.providers.get(c.req.param("providerId"));
+    if (provider === void 0) {
+      return errorResponse(c, 404, "not_found", "Model provider not found");
+    }
+    return c.json(await providerCatalog(provider));
+  });
+  app.get("/agents/:agentId/models", async (c) => {
+    const agentId = c.req.param("agentId");
+    const agent = props.agentMap.get(agentId);
+    if (agent === void 0) {
+      return errorResponse(c, 404, "not_found", "Agent not found");
+    }
+    if (props.registry === void 0) {
+      return c.json({
+        agentId,
+        models: []
+      });
+    }
+    return c.json(await agentModelsCatalog(props.registry, agent));
+  });
+}
+function resolveStudioModel(registry, input) {
+  if (registry === void 0) {
+    return { warnings: [] };
+  }
+  const selectedRef = normalizeOptionalModelRef(input.request.model) ?? sessionModelRef(input.sessionMetadata) ?? registry.agentPolicies.get(input.agent.id)?.default ?? registry.defaultModel;
+  if (selectedRef === void 0) {
+    return { warnings: [] };
+  }
+  ensureModelAllowed(registry, input.agent.id, selectedRef);
+  const { providerId, modelId } = parseModelRef(selectedRef);
+  const provider = registry.providers.get(providerId);
+  if (provider === void 0) {
+    throw new ModelSelectionError(`Unknown model provider: ${providerId}`);
+  }
+  let model = registry.modelCache.get(selectedRef);
+  if (model === void 0) {
+    model = provider.createCompletionModel(modelId);
+    registry.modelCache.set(selectedRef, model);
+  }
+  const metadata = provider.staticModels.get(modelId);
+  return {
+    ref: selectedRef,
+    model,
+    warnings: modelWarnings(selectedRef, metadata, input.request)
+  };
+}
+function sessionModelRef(metadata) {
+  const value = metadata?.[STUDIO_MODEL_METADATA_KEY];
+  return typeof value === "string" && value.trim().length > 0 ? value.trim() : void 0;
+}
+function normalizeOptionalModelRef(ref) {
+  return ref === void 0 ? void 0 : normalizeModelRef(ref);
+}
+function normalizeModelRef(ref) {
+  if (typeof ref === "string") {
+    const trimmed = ref.trim();
+    const parsed = parseModelRef(trimmed);
+    return `${parsed.providerId}:${parsed.modelId}`;
+  }
+  return `${normalizeProviderId(ref.provider)}:${normalizeModelId(ref.model)}`;
+}
+function parseModelRef(ref) {
+  const index = ref.indexOf(":");
+  if (index <= 0 || index === ref.length - 1) {
+    throw new ModelSelectionError(`Model ref must use provider:model format: ${ref}`);
+  }
+  return {
+    providerId: normalizeProviderId(ref.slice(0, index)),
+    modelId: normalizeModelId(ref.slice(index + 1))
+  };
+}
+var ModelSelectionError = class extends Error {
+  constructor(message) {
+    super(message);
+    this.name = "ModelSelectionError";
+  }
+};
+async function agentModelsCatalog(registry, agent) {
+  const policy = registry.agentPolicies.get(agent.id);
+  const catalogs = await Promise.all([...registry.providers.values()].map(providerCatalog));
+  const warnings = catalogs.flatMap(
+    (catalog) => catalog.warning === void 0 ? [] : [{ providerId: catalog.id, warning: catalog.warning }]
+  );
+  const models = catalogs.flatMap((catalog) => catalog.models).filter((model) => {
+    if (policy?.allowed === void 0) {
+      return true;
+    }
+    return allowedByPolicy(policy.allowed, model.ref);
+  });
+  const exactPolicyModels = (policy?.allowed ?? []).filter((allowed) => !allowed.endsWith(":*")).filter((allowed) => models.every((model) => model.ref !== allowed)).flatMap((ref) => {
+    try {
+      const { providerId, modelId } = parseModelRef(ref);
+      const provider = registry.providers.get(providerId);
+      return provider === void 0 ? [] : [
+        modelSummary(provider, modelId, {
+          id: modelId
+        })
+      ];
+    } catch {
+      return [];
+    }
+  });
+  const defaultModel = policy?.default ?? registry.defaultModel;
+  return {
+    agentId: agent.id,
+    ...defaultModel === void 0 ? {} : { defaultModel },
+    models: [...models, ...exactPolicyModels],
+    ...warnings.length === 0 ? {} : { warnings }
+  };
+}
+async function providerCatalog(provider) {
+  const models = /* @__PURE__ */ new Map();
+  for (const model of provider.staticModels.values()) {
+    models.set(model.id, modelSummary(provider, model.id, model));
+  }
+  let warning;
+  if (provider.listModels !== void 0) {
+    try {
+      const listed = await provider.listModels();
+      for (const model of listed.data) {
+        const staticModel = provider.staticModels.get(model.id);
+        models.set(
+          model.id,
+          modelSummary(provider, model.id, {
+            id: model.id,
+            ...model.name === void 0 ? {} : { name: model.name },
+            ...model.description === void 0 ? {} : { description: model.description },
+            ...staticModel ?? {},
+            metadata: {
+              ...model.type === void 0 ? {} : { type: model.type },
+              ...model.createdAt === void 0 ? {} : { createdAt: model.createdAt },
+              ...model.ownedBy === void 0 ? {} : { ownedBy: model.ownedBy },
+              ...model.contextLength === void 0 ? {} : { contextLength: model.contextLength },
+              ...staticModel?.metadata ?? {}
+            }
+          })
+        );
+      }
+    } catch (error) {
+      const serialized = serializeError(error);
+      warning = typeof serialized === "object" && serialized !== null && "message" in serialized ? String(serialized.message) : String(error);
+    }
+  }
+  return {
+    id: provider.id,
+    ...provider.name === void 0 ? {} : { name: provider.name },
+    ...provider.defaultModel === void 0 ? {} : { defaultModel: provider.defaultModel },
+    models: [...models.values()].sort((left, right) => left.ref.localeCompare(right.ref)),
+    ...provider.metadata === void 0 ? {} : { metadata: provider.metadata },
+    ...warning === void 0 ? {} : { warning }
+  };
+}
+function modelSummary(provider, modelId, model) {
+  return {
+    ...model,
+    id: modelId,
+    ref: `${provider.id}:${modelId}`,
+    providerId: provider.id,
+    ...provider.name === void 0 ? {} : { providerName: provider.name }
+  };
+}
+function ensureModelAllowed(registry, agentId, ref) {
+  const { providerId } = parseModelRef(ref);
+  if (!registry.providers.has(providerId)) {
+    throw new ModelSelectionError(`Unknown model provider: ${providerId}`);
+  }
+  const policy = registry.agentPolicies.get(agentId);
+  if (policy?.allowed !== void 0 && !allowedByPolicy(policy.allowed, ref)) {
+    throw new ModelSelectionError(`Model ${ref} is not allowed for agent ${agentId}`);
+  }
+}
+function allowedByPolicy(allowed, ref) {
+  return allowed.some(
+    (entry) => entry.endsWith(":*") ? ref.startsWith(entry.slice(0, -1)) : entry === ref
+  );
+}
+function normalizeAgentPolicies(policies) {
+  return new Map(
+    Object.entries(policies).map(([agentId, policy]) => [
+      agentId.trim(),
+      {
+        ...policy.default === void 0 ? {} : { default: normalizeModelRef(policy.default) },
+        ...policy.allowed === void 0 ? {} : {
+          allowed: policy.allowed.map(
+            (entry) => typeof entry === "string" && entry.trim().endsWith(":*") ? `${normalizeProviderId(entry.trim().slice(0, -2))}:*` : normalizeModelRef(entry)
+          )
+        }
+      }
+    ])
+  );
+}
+function publicPolicy(policy) {
+  return {
+    ...policy.default === void 0 ? {} : { default: policy.default },
+    ...policy.allowed === void 0 ? {} : { allowed: policy.allowed }
+  };
+}
+function modelWarnings(ref, model, request) {
+  const warnings = [];
+  const modalities = requestModalities(request);
+  const missingModalities = [...modalities].filter(
+    (modality) => !model?.modalities?.input.includes(modality)
+  );
+  if (model?.modalities !== void 0 && missingModalities.length > 0) {
+    warnings.push({
+      model: ref,
+      kind: "modality",
+      message: `Model ${ref} does not declare input support for ${missingModalities.join(", ")}`,
+      missing: missingModalities
+    });
+  }
+  if (request.stream === true && model?.capabilities?.streaming === false) {
+    warnings.push({
+      model: ref,
+      kind: "capability",
+      message: `Model ${ref} is configured as non-streaming but the run requested streaming`,
+      capability: "streaming"
+    });
+  }
+  return warnings;
+}
+function requestModalities(request) {
+  const modalities = /* @__PURE__ */ new Set(["text"]);
+  for (const message of requestMessages(request)) {
+    if (typeof message === "string") {
+      continue;
+    }
+    if (message.role === "user" || message.role === "assistant") {
+      for (const content of message.content) {
+        if (content.type === "image") {
+          modalities.add("image");
+        }
+        if (content.type === "document") {
+          modalities.add("document");
+        }
+      }
+    }
+  }
+  return modalities;
+}
+function requestMessages(request) {
+  return [...request.history ?? [], request.message];
+}
+function normalizeProviderId(id) {
+  const trimmed = id.trim();
+  if (trimmed.length === 0 || trimmed.includes(":")) {
+    throw new ModelSelectionError("Model provider id must be non-empty and cannot contain ':'");
+  }
+  return trimmed;
+}
+function normalizeModelId(id) {
+  const trimmed = id.trim();
+  if (trimmed.length === 0) {
+    throw new ModelSelectionError("Model id cannot be empty");
+  }
+  return trimmed;
+}
+
 // src/runtime/tool-metadata.ts
 import { ToolSet } from "@anvia/core/tool";
 var MCP_TOOL_METADATA_KEY = /* @__PURE__ */ Symbol.for("anvia.mcp.tool.metadata");
@@ -1296,12 +1684,14 @@ function normalizePipelines(pipelines) {
   });
 }
 function buildConfig(options, agents, pipelines, stores) {
+  const models = options.models === void 0 ? void 0 : studioModelsConfig(createStudioModelRegistry(options.models), agents);
   return {
     id: runnerId(options),
     ...options.name === void 0 ? {} : { name: options.name },
     ...options.description === void 0 ? {} : { description: options.description },
     ...options.version === void 0 ? {} : { version: options.version },
     agents: agents.map(agentConfig),
+    ...models === void 0 ? {} : { models },
     pipelines: pipelines.map(pipelineConfig),
     evals: options.evals.map(evalConfig),
     chat: {
@@ -3097,7 +3487,8 @@ function messageToTranscriptEntry(message, entryId) {
     entryId,
     kind: "message",
     role,
-    text: extractMessageText(message)
+    text: extractMessageText(message),
+    ...role === "user" ? optionalTranscriptAttachments(message) : {}
   };
 }
 function appendTranscriptAssistantText(transcript, delta) {
@@ -3193,6 +3584,33 @@ function extractMessageText(message) {
     return [];
   }).join("\n");
 }
+function optionalTranscriptAttachments(message) {
+  if (typeof message === "string" || message.role !== "user") {
+    return {};
+  }
+  const attachments = message.content.flatMap((content) => {
+    if (content.type === "image") {
+      return [
+        {
+          kind: "image",
+          ...content.source.type === "base64" ? { data: content.source.data, mediaType: content.source.mediaType } : { url: content.source.url }
+        }
+      ];
+    }
+    if (content.type === "document") {
+      return [
+        {
+          kind: "document",
+          ...content.source.filename === void 0 ? {} : { name: content.source.filename },
+          ...content.source.mediaType === void 0 ? {} : { mediaType: content.source.mediaType },
+          ...content.source.type === "base64" ? { data: content.source.data } : content.source.type === "url" ? { url: content.source.url } : {}
+        }
+      ];
+    }
+    return [];
+  });
+  return attachments.length === 0 ? {} : { attachments };
+}
 function formatJson2(value) {
   try {
     return JSON.stringify(value, null, 2);
@@ -3257,6 +3675,25 @@ async function parseRunRequest(c) {
     }
     request.toolConcurrency = body.toolConcurrency;
   }
+  if ("model" in body) {
+    if (typeof body.model === "string") {
+      request.model = body.model;
+    } else if (isObject(body.model) && typeof body.model.provider === "string" && typeof body.model.model === "string") {
+      request.model = {
+        provider: body.model.provider,
+        model: body.model.model
+      };
+    } else {
+      return {
+        error: errorResponse(
+          c,
+          400,
+          "bad_request",
+          "model must be a provider:model string or { provider, model } object"
+        )
+      };
+    }
+  }
   if ("metadata" in body) {
     if (!isJsonObject2(body.metadata)) {
       return { error: errorResponse(c, 400, "bad_request", "metadata must be an object") };
@@ -4825,6 +5262,7 @@ function studioOptionsFromTargets(targets, options) {
     agents: inferStudioAgents(agents, options.quickPrompts ?? {}),
     pipelines: inferStudioPipelines(pipelines),
     evals: options.evals ?? [],
+    ...options.models === void 0 ? {} : { models: options.models },
     ...options.stores === void 0 ? {} : { stores: options.stores },
     ...options.ui === void 0 ? {} : { ui: options.ui }
   };
@@ -4882,6 +5320,7 @@ function createStudioApp(options) {
   const agents = normalizeAgents(options.agents).map(
     (agent) => withStudioTraceObserver(agent, stores.traces)
   );
+  const modelRegistry = createStudioModelRegistry(options.models);
   const pipelines = normalizePipelines(options.pipelines);
   const agentMap = new Map(agents.map((agent) => [agent.id, agent]));
   const pipelineMap = new Map(pipelines.map((pipeline) => [pipeline.id, pipeline]));
@@ -4924,6 +5363,7 @@ function createStudioApp(options) {
     }
     return c.json(agentRuntimeSummary(agent));
   });
+  registerModelRoutes(app, { registry: modelRegistry, agentMap });
   registerMcpRoutes(app, { agentMap });
   registerToolRoutes(app, { agentMap });
   registerApprovalRoutes(app, approvalRuntime);
@@ -4963,6 +5403,20 @@ function createStudioApp(options) {
     if (session !== void 0 && session.agentId !== agentId) {
       return errorResponse(c, 400, "bad_request", "Session belongs to another agent");
     }
+    let selectedModel;
+    try {
+      selectedModel = resolveStudioModel(modelRegistry, {
+        agent,
+        request: body,
+        sessionMetadata: session?.metadata
+      });
+    } catch (error) {
+      if (error instanceof ModelSelectionError) {
+        return errorResponse(c, 400, "bad_request", error.message);
+      }
+      throw error;
+    }
+    const runAgent = selectedModel.model === void 0 ? agent.agent : cloneAgent(agent.agent, { model: selectedModel.model });
     const runId = globalThis.crypto.randomUUID();
     const runStartedAt = Date.now();
     if (session !== void 0) {
@@ -4977,18 +5431,43 @@ function createStudioApp(options) {
           ...body.maxTurns === void 0 ? {} : { maxTurns: body.maxTurns },
           ...body.toolConcurrency === void 0 ? {} : { toolConcurrency: body.toolConcurrency },
           hasTrace: body.trace !== void 0,
-          ...body.metadata === void 0 ? {} : { metadata: body.metadata }
+          ...body.metadata === void 0 && selectedModel.ref === void 0 ? {} : {
+            metadata: {
+              ...body.metadata ?? {},
+              ...selectedModel.ref === void 0 ? {} : { [STUDIO_MODEL_METADATA_KEY]: selectedModel.ref }
+            }
+          }
         })
       );
     }
+    if (session !== void 0 && selectedModel.ref !== void 0) {
+      for (const warning of selectedModel.warnings) {
+        await appendSessionLog(stores.sessions, {
+          sessionId: session.id,
+          runId,
+          level: "warn",
+          category: "model",
+          event: "model.warning",
+          message: typeof warning.message === "string" ? warning.message : "Model warning",
+          metadata: warning
+        });
+      }
+      if (sessionModelRef(session.metadata) !== selectedModel.ref) {
+        await stores.sessions?.updateSessionMetadata?.(session.id, {
+          ...session.metadata ?? {},
+          [STUDIO_MODEL_METADATA_KEY]: selectedModel.ref
+        });
+      }
+    }
     const memoryMetadata = {
       agentId,
       ...body.metadata ?? {},
+      ...selectedModel.ref === void 0 ? {} : { [STUDIO_MODEL_METADATA_KEY]: selectedModel.ref },
       studioRunId: runId
     };
     const promptMessage = normalizePromptMessage(body.message);
     const sessionStore = stores.sessions;
-    const shouldPersistSessionMessages = session !== void 0 && sessionStore !== void 0 && !usesStoreAsAgentMemory(agent.agent, sessionStore);
+    const shouldPersistSessionMessages = session !== void 0 && sessionStore !== void 0 && !usesStoreAsAgentMemory(runAgent, sessionStore);
     if (shouldPersistSessionMessages) {
       await sessionStore.append({
         context: { sessionId: session.id, metadata: memoryMetadata },
@@ -4997,7 +5476,7 @@ function createStudioApp(options) {
         messages: [promptMessage]
       });
     }
-    const request = session !== void 0 ? agent.agent.memory === void 0 ? agent.agent.prompt([...session.messages, promptMessage]) : agent.agent.session(session.id, { metadata: memoryMetadata }).prompt(body.message) : agent.agent.prompt(
+    const request = session !== void 0 ? runAgent.memory === void 0 ? runAgent.prompt([...session.messages, promptMessage]) : runAgent.session(session.id, { metadata: memoryMetadata }).prompt(body.message) : runAgent.prompt(
       body.history !== void 0 ? [...body.history, promptMessage] : body.message
     );
     if (body.maxTurns !== void 0) {
@@ -5015,13 +5494,13 @@ function createStudioApp(options) {
       const runtimeEvents = new AsyncEventQueue();
       const effectiveHook = composeHooks(
         composeHooks(
-          agent.agent.hook,
+          runAgent.hook,
           approvalRuntime.createHook({
             runId,
             agentId,
             ...session?.id === void 0 ? {} : { sessionId: session.id },
             ...body.metadata === void 0 ? {} : { metadata: body.metadata },
-            getTool: (toolName) => agent.agent.getTool(toolName),
+            getTool: (toolName) => runAgent.getTool(toolName),
             emit: (event) => runtimeEvents.push(event)
           })
         ),
@@ -5060,13 +5539,13 @@ function createStudioApp(options) {
       }
       const effectiveHook = composeHooks(
         composeHooks(
-          agent.agent.hook,
+          runAgent.hook,
           approvalRuntime.createHook({
             runId,
             agentId,
             ...session?.id === void 0 ? {} : { sessionId: session.id },
             ...body.metadata === void 0 ? {} : { metadata: body.metadata },
-            getTool: (toolName) => agent.agent.getTool(toolName)
+            getTool: (toolName) => runAgent.getTool(toolName)
           })
         ),
         questionRuntime.createHook({
@@ -5322,6 +5801,24 @@ var SqliteSessionStore = class {
     const row = this.getSessionRow(id);
     return row === void 0 ? void 0 : toSession(row, this.listSessionMessages(id), this.listSessionRunRows(id));
   }
+  updateSessionMetadata(id, metadata) {
+    const db = this.database();
+    const now = (/* @__PURE__ */ new Date()).toISOString();
+    const result = db.prepare(
+      `UPDATE anvia_studio_sessions
+         SET metadata_json = $metadata,
+             updated_at = $now
+         WHERE id = $id`
+    ).run({
+      $id: id,
+      $metadata: metadata === void 0 ? null : JSON.stringify(metadata),
+      $now: now
+    });
+    if (result.changes === 0) {
+      return void 0;
+    }
+    return this.getSession(id);
+  }
   load(context) {
     const session = this.getSession(context.sessionId);
     return Promise.resolve(session?.messages ?? []);