@anvia/studio 0.7.28 → 0.7.29

package/dist/index.js

@@ -951,20 +951,17 @@ function serializeError(error) {
 }
 
 // src/runtime/type-guards.ts
+import {
+  isJsonValue as isCoreJsonValue
+} from "@anvia/core/completion";
 function isObject(value) {
   return typeof value === "object" && value !== null && !Array.isArray(value);
 }
 function isJsonObject(value) {
-  return isObject(value) && Object.values(value).every(isJsonValue);
+  return isObject(value) && isCoreJsonValue(value);
 }
 function isJsonValue(value) {
-  if (value === null || typeof value === "string" || typeof value === "number" || typeof value === "boolean") {
-    return true;
-  }
-  if (Array.isArray(value)) {
-    return value.every(isJsonValue);
-  }
-  return isJsonObject(value);
+  return isCoreJsonValue(value);
 }
 function isMessageInput(value) {
   return typeof value === "string" || isMessage(value);
@@ -4662,7 +4659,7 @@ async function parsePipelineRunRequest(c) {
   if (!isObject(body)) {
     return { error: errorResponse(c, 400, "bad_request", "Request body must be an object") };
   }
-  if (!("input" in body) || !isJsonValue2(body.input)) {
+  if (!("input" in body) || !isJsonValue(body.input)) {
     return { error: errorResponse(c, 400, "bad_request", "input must be JSON-compatible") };
   }
   const request = {
@@ -4734,18 +4731,6 @@ function parsePipelineLogAfter(value) {
   }
   return after;
 }
-function isJsonValue2(value) {
-  if (value === null || typeof value === "string" || typeof value === "number" || typeof value === "boolean") {
-    return Number.isFinite(value) || typeof value !== "number";
-  }
-  if (Array.isArray(value)) {
-    return value.every(isJsonValue2);
-  }
-  if (isObject(value)) {
-    return Object.values(value).every((item) => item === void 0 || isJsonValue2(item));
-  }
-  return false;
-}
 
 // src/runtime/questions.ts
 import { createHook as createHook2 } from "@anvia/core/hooks";
@@ -5362,7 +5347,7 @@ function traceSummary2(trace) {
 }
 function traceAgentId(trace) {
   const nestedMetadata = trace.metadata?.metadata;
-  return isJsonObject2(nestedMetadata) && typeof nestedMetadata.agentId === "string" ? nestedMetadata.agentId : void 0;
+  return isJsonObject(nestedMetadata) && typeof nestedMetadata.agentId === "string" ? nestedMetadata.agentId : void 0;
 }
 function studioRunId(context) {
   const value = context.metadata?.studioRunId;
@@ -5375,19 +5360,7 @@ function serializeJsonError(error) {
       message: error.message
     };
   }
-  return isJsonValue3(error) ? error : String(error);
-}
-function isJsonObject2(value) {
-  return typeof value === "object" && value !== null && !Array.isArray(value);
-}
-function isJsonValue3(value) {
-  if (value === null || typeof value === "string" || typeof value === "number" || typeof value === "boolean") {
-    return true;
-  }
-  if (Array.isArray(value)) {
-    return value.every(isJsonValue3);
-  }
-  return isJsonObject2(value) && Object.values(value).every(isJsonValue3);
+  return isJsonValue(error) ? error : String(error);
 }
 
 // src/runtime/shared.ts
@@ -5951,6 +5924,7 @@ function hasStudioTraceObserver(agent) {
 import { mkdirSync } from "fs";
 import { createRequire } from "module";
 import { dirname, resolve } from "path";
+import { isJsonValue as isJsonValue2 } from "@anvia/core/completion";
 var DatabaseSync;
 function createSqliteSessionStore(options = {}) {
   return new SqliteSessionStore(options.path ?? ":memory:");
@@ -6607,6 +6581,7 @@ var SqliteSessionStore = class {
         message_index INTEGER NOT NULL,
         role TEXT NOT NULL,
         message_id TEXT,
+        metadata_json TEXT,
         created_at TEXT NOT NULL,
         PRIMARY KEY(session_id, message_index),
         FOREIGN KEY(session_id) REFERENCES anvia_studio_sessions(id) ON DELETE CASCADE
@@ -6701,6 +6676,7 @@ var SqliteSessionStore = class {
       CREATE INDEX IF NOT EXISTS anvia_studio_traces_session_started_idx
         ON anvia_studio_traces(session_id, started_at DESC);
     `);
+    ensureMessageMetadataColumn(db);
     this.db = db;
     return db;
   }
@@ -6745,7 +6721,7 @@ var SqliteSessionStore = class {
   listSessionMessages(sessionId) {
     const db = this.database();
     const messageRows = db.prepare(
-      `SELECT session_id, message_index, role, message_id, created_at
+      `SELECT session_id, message_index, role, message_id, metadata_json, created_at
          FROM anvia_studio_session_messages
          WHERE session_id = $sessionId
          ORDER BY message_index ASC`
@@ -6785,8 +6761,9 @@ var SqliteSessionStore = class {
         message_index,
         role,
         message_id,
+        metadata_json,
         created_at
-      ) VALUES ($sessionId, $messageIndex, $role, $messageId, $createdAt)`
+      ) VALUES ($sessionId, $messageIndex, $role, $messageId, $metadata, $createdAt)`
     );
     const insertPart = db.prepare(
       `INSERT INTO anvia_studio_session_message_parts (
@@ -6799,11 +6776,15 @@ var SqliteSessionStore = class {
     );
     messages.forEach((message, messageOffset) => {
       const messageIndex = startIndex + messageOffset;
+      if (message.metadata !== void 0 && !isJsonValue2(message.metadata)) {
+        throw new TypeError("Studio message metadata must be a strict JSON value.");
+      }
       insertMessage.run({
         $sessionId: sessionId,
         $messageIndex: messageIndex,
         $role: message.role,
         $messageId: message.role === "assistant" ? message.id ?? null : null,
+        $metadata: message.metadata === void 0 ? null : JSON.stringify(message.metadata),
         $createdAt: createdAt
       });
       messageParts(message).forEach((part, partIndex) => {
@@ -6899,26 +6880,34 @@ function messageParts(message) {
 }
 function messageFromRows(row, partRows) {
   const parts = partRows.map((partRow) => JSON.parse(partRow.part_json));
+  const metadata = parseJsonValue(row.metadata_json);
+  if (metadata !== void 0 && !isJsonValue2(metadata)) {
+    throw new TypeError("Stored Studio message metadata is not a strict JSON value.");
+  }
+  const metadataProperties = compact({ metadata });
   if (row.role === "system") {
-    return { role: "system", content: systemContentFromParts(parts) };
+    return { role: "system", content: systemContentFromParts(parts), ...metadataProperties };
   }
   if (row.role === "user") {
     return {
       role: "user",
-      content: parts
+      content: parts,
+      ...metadataProperties
     };
   }
   if (row.role === "assistant") {
     return {
       role: "assistant",
       ...compact({ id: row.message_id ?? void 0 }),
-      content: parts
+      content: parts,
+      ...metadataProperties
     };
   }
   if (row.role === "tool") {
     return {
       role: "tool",
-      content: parts
+      content: parts,
+      ...metadataProperties
     };
   }
   throw new Error(`Unsupported stored message role: ${row.role}`);
@@ -6938,6 +6927,12 @@ function guardAgainstLegacySessionSchema(db) {
     );
   }
 }
+function ensureMessageMetadataColumn(db) {
+  const columns = db.prepare("PRAGMA table_info('anvia_studio_session_messages')").all();
+  if (!columns.some((column) => column.name === "metadata_json")) {
+    db.exec("ALTER TABLE anvia_studio_session_messages ADD COLUMN metadata_json TEXT");
+  }
+}
 function loadDatabaseSync() {
   if (DatabaseSync !== void 0) {
     return DatabaseSync;