@mastra/duckdb 1.16.0-alpha.1 → 1.16.0

package/dist/docs/SKILL.md

@@ -3,7 +3,7 @@ name: mastra-duckdb
 description: Documentation for @mastra/duckdb. Use when working with @mastra/duckdb APIs, configuration, or implementation.
 metadata:
   package: "@mastra/duckdb"
-  version: "1.16.0-alpha.1"
+  version: "1.16.0"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.16.0-alpha.1",
+  "version": "1.16.0",
   "package": "@mastra/duckdb",
   "exports": {},
   "modules": {}

package/dist/docs/references/reference-vectors-duckdb.md

@@ -245,11 +245,11 @@ const results = await store.query({
 
 ## Distance metrics
 
-| Metric       | Description       | Score Interpretation   | Best For                            |
-| ------------ | ----------------- | ---------------------- | ----------------------------------- |
-| `cosine`     | Cosine similarity | 0-1 (1 = most similar) | Text embeddings, normalized vectors |
-| `euclidean`  | L2 distance       | 0-∞ (0 = most similar) | Image embeddings, spatial data      |
-| `dotproduct` | Inner product     | Higher = more similar  | When vector magnitude matters       |
+| Metric       | Description       | Score Interpretation                                          | Best For                            |
+| ------------ | ----------------- | ------------------------------------------------------------- | ----------------------------------- |
+| `cosine`     | Cosine similarity | -1 to 1, computed as `1 - cosine distance` (1 = most similar) | Text embeddings, normalized vectors |
+| `euclidean`  | L2 distance       | 0-1, computed as `1 / (1 + distance)` (1 = most similar)      | Image embeddings, spatial data      |
+| `dotproduct` | Inner product     | Higher = more similar                                         | When vector magnitude matters       |
 
 ## Error handling
 

package/dist/index.cjs

@@ -564,7 +564,7 @@ var ObservabilityStorageDuckDB = class extends _mastra_core_storage.Observabilit
 	async loadDelegate() {
 		if (this.delegate) return this.delegate;
 		if (this.unavailableError) return null;
-		if (!this.loadPromise) this.loadPromise = Promise.resolve().then(() => require("./observability-Dlt85fvH.cjs")).then(({ ObservabilityStorageDuckDB }) => {
+		if (!this.loadPromise) this.loadPromise = Promise.resolve().then(() => require("./observability-BEdkwaeY.cjs")).then(({ ObservabilityStorageDuckDB }) => {
 			const delegate = new ObservabilityStorageDuckDB({ db: this.db });
 			this.delegate = delegate;
 			return delegate;

package/dist/index.js

@@ -563,7 +563,7 @@ var ObservabilityStorageDuckDB = class extends ObservabilityStorage {
 	async loadDelegate() {
 		if (this.delegate) return this.delegate;
 		if (this.unavailableError) return null;
-		if (!this.loadPromise) this.loadPromise = import("./observability-Chrw9Di2.js").then(({ ObservabilityStorageDuckDB }) => {
+		if (!this.loadPromise) this.loadPromise = import("./observability-B-Kt8vrw.js").then(({ ObservabilityStorageDuckDB }) => {
 			const delegate = new ObservabilityStorageDuckDB({ db: this.db });
 			this.delegate = delegate;
 			return delegate;

package/dist/{observability-Chrw9Di2.js → observability-B-Kt8vrw.js}

@@ -1,7 +1,7 @@
 import { a as timeToFirstChunkSupported, i as timeToFirstChunkFeatures, n as spanQueryFeatures, o as DuckDBConnection, s as DuckDBQueryTimeoutError, t as spanAggregateFeatures } from "./features-B8gl38Ue.js";
 import { ErrorCategory, ErrorDomain, MastraError } from "@mastra/core/error";
 import * as coreStorage from "@mastra/core/storage";
-import { BRANCH_SPAN_TYPES, METRIC_DISTINCT_COLUMNS, ObservabilityStorage, computeTraceStatus, createStorageErrorId, executeRetentionPrune, feedbackRecordSchema, listBranchesArgsSchema, listFeedbackArgsSchema, listLogsArgsSchema, listMetricsArgsSchema, listScoresArgsSchema, listTracesArgsSchema, resolveRetentionTargets, retentionCutoffMs, toTraceSpans } from "@mastra/core/storage";
+import { BRANCH_SPAN_TYPES, METRIC_DISTINCT_COLUMNS, ObservabilityStorage, computeTraceStatus, createStorageErrorId, executeRetentionPrune, feedbackRecordSchema, getStringAttribute, listBranchesArgsSchema, listFeedbackArgsSchema, listLogsArgsSchema, listMetricsArgsSchema, listScoresArgsSchema, listTracesArgsSchema, resolveRetentionTargets, retentionCutoffMs, toTraceSpans } from "@mastra/core/storage";
 import { coreFeatures } from "@mastra/core/features";
 import { EntityType } from "@mastra/core/observability";
 import { parseFieldKey } from "@mastra/core/utils";
@@ -106,6 +106,17 @@ CREATE TABLE IF NOT EXISTS span_events (
   endedAt TIMESTAMP,
   timeToFirstChunkMs DOUBLE,
 
+  -- Native span fields, stored as supplied (NULL when the producer set none)
+  externalParentSpanId VARCHAR,
+  resumedFromSpanId VARCHAR,
+  model VARCHAR,
+  provider VARCHAR,
+  outcome VARCHAR,
+  stopReason VARCHAR,
+  timeToFirstTokenMs DOUBLE,
+  errorCount INTEGER,
+  stopReasonContext JSON,
+
   -- JSON fields
   attributes JSON,
   metadata JSON,
@@ -377,6 +388,15 @@ const ALL_MIGRATIONS = [
 	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS hasOutputPayload BOOLEAN`,
 	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS hasErrorPayload BOOLEAN`,
 	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS timeToFirstChunkMs DOUBLE`,
+	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS externalParentSpanId VARCHAR`,
+	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS resumedFromSpanId VARCHAR`,
+	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS model VARCHAR`,
+	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS provider VARCHAR`,
+	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS outcome VARCHAR`,
+	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS stopReason VARCHAR`,
+	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS timeToFirstTokenMs DOUBLE`,
+	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS errorCount INTEGER`,
+	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS stopReasonContext JSON`,
 	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS parentEntityType VARCHAR`,
 	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS parentEntityId VARCHAR`,
 	`ALTER TABLE span_events ADD COLUMN IF NOT EXISTS parentEntityName VARCHAR`,
@@ -4342,7 +4362,7 @@ async function queryThreadRows(db, plan, strategy = "auto") {
 			return `${field} ${field === "organizationId" || field === "resourceId" ? "IS NOT DISTINCT FROM" : "="} ?`;
 		}).join(" AND ")})`;
 	}));
-	const payloads = await run(`SELECT organizationId, traceId, cursorId, spanType,
+	const payloads = await run(`SELECT organizationId, traceId, cursorId,
     ${contextFields.map((field) => field === "executionSource" ? "source AS executionSource" : field).join(", ")},
     CASE WHEN spanType = 'agent_run' THEN timeToFirstChunkMs END AS timeToFirstChunkMs,
     CASE WHEN error IS NULL THEN 'success' ELSE 'error' END AS status,
@@ -4358,34 +4378,10 @@ async function queryThreadRows(db, plan, strategy = "auto") {
 		String(version)
 	]);
 	const byRoot = new Map(payloads.map((row) => [key(row.organizationId, row.traceId, row.cursorId), row]));
-	const agents = page.filter((row) => byRoot.get(key(row.organizationId, row.firstTraceId, row.firstCursorId))?.spanType === "agent_run");
-	const models = /* @__PURE__ */ new Map();
-	if (agents.length) {
-		const modelValues = [];
-		const rows = await run(`WITH generations AS (
-        SELECT organizationId, traceId, spanId, endedAt, cursorId, attributes,
-          min(timestamp) FILTER (WHERE eventType = 'start') OVER (PARTITION BY organizationId, traceId, spanId) AS startedAt
-        FROM span_events WHERE spanType = 'model_generation' AND (${agents.map((row) => `(${Object.entries({
-			organizationId: row.organizationId,
-			resourceId: row.resourceId,
-			traceId: row.firstTraceId,
-			parentSpanId: row.firstSpanId
-		}).map(([field, value]) => {
-			modelValues.push(value);
-			return `${field} ${field === "organizationId" || field === "resourceId" ? "IS NOT DISTINCT FROM" : "="} ?`;
-		}).join(" AND ")})`).join(" OR ")})
-      ) SELECT organizationId, traceId,
-        CASE WHEN json_type(attributes, '$.model') = 'VARCHAR' THEN json_extract_string(attributes, '$.model') END AS model,
-        CASE WHEN json_type(attributes, '$.provider') = 'VARCHAR' THEN json_extract_string(attributes, '$.provider') END AS provider
-        FROM generations WHERE endedAt IS NOT NULL
-        QUALIFY row_number() OVER (PARTITION BY organizationId, traceId ORDER BY startedAt, spanId, cursorId DESC) = 1`, modelValues);
-		for (const row of rows) models.set(JSON.stringify([row.organizationId, row.traceId]), row);
-	}
 	const threads = page.flatMap((row) => {
 		const first = byRoot.get(key(row.organizationId, row.firstTraceId, row.firstCursorId));
 		const last = byRoot.get(key(row.organizationId, row.lastTraceId, row.lastCursorId));
 		if (!first || !last) return [];
-		const model = models.get(JSON.stringify([row.organizationId, row.firstTraceId]));
 		return [coreStorage.threadQueryRowSchema.parse({
 			organizationId: row.organizationId,
 			threadId: row.threadId,
@@ -4397,8 +4393,6 @@ async function queryThreadRows(db, plan, strategy = "auto") {
 			endedAt: row.endedAt.toISOString(),
 			traceCount: Number(row.traceCount),
 			timeToFirstChunkMs: first.timeToFirstChunkMs,
-			model: model?.model ?? null,
-			provider: model?.provider ?? null,
 			inputPreview: readPreview(first.inputPreview, first.input, true),
 			outputPreview: readPreview(last.outputPreview, last.output, true),
 			errorPreview: readPreview(last.errorPreview, last.error, true)
@@ -4808,6 +4802,15 @@ const COLUMNS_SQL = [
 	"isEvent",
 	"endedAt",
 	"timeToFirstChunkMs",
+	"externalParentSpanId",
+	"resumedFromSpanId",
+	"model",
+	"provider",
+	"outcome",
+	"stopReason",
+	"timeToFirstTokenMs",
+	"errorCount",
+	"stopReasonContext",
 	"experimentId",
 	"entityType",
 	"entityId",
@@ -4923,7 +4926,16 @@ const SPAN_RECONSTRUCT_SELECT = `
     ${argMaxNonNull("input")},
     ${argMaxNonNull("output")},
     ${argMaxNonNull("error")},
-    ${argMaxNonNull("requestContext")}
+    ${argMaxNonNull("requestContext")},
+    ${argMaxNonNull("externalParentSpanId")},
+    ${argMaxNonNull("resumedFromSpanId")},
+    ${argMaxNonNull("model")},
+    ${argMaxNonNull("provider")},
+    ${argMaxNonNull("outcome")},
+    ${argMaxNonNull("stopReason")},
+    ${argMaxNonNull("timeToFirstTokenMs")},
+    ${argMaxNonNull("errorCount")},
+    ${argMaxNonNull("stopReasonContext")}
   FROM span_events
 `;
 /** Lightweight variant — only timeline-relevant columns. */
@@ -5119,6 +5131,7 @@ function rowToLightSpanRecordWithPreview(row) {
 	};
 }
 function rowToSpanRecord(row) {
+	const attributes = parseJson(row.attributes);
 	return {
 		traceId: row.traceId,
 		spanId: row.spanId,
@@ -5128,6 +5141,15 @@ function rowToSpanRecord(row) {
 		isEvent: row.isEvent,
 		startedAt: toDate(row.startedAt),
 		endedAt: rowEndedAt(row),
+		externalParentSpanId: row.externalParentSpanId ?? null,
+		resumedFromSpanId: row.resumedFromSpanId ?? null,
+		model: row.model ?? getStringAttribute(attributes, "model"),
+		provider: row.provider ?? getStringAttribute(attributes, "provider"),
+		outcome: row.outcome ?? null,
+		stopReason: row.stopReason ?? null,
+		timeToFirstTokenMs: row.timeToFirstTokenMs ?? null,
+		errorCount: row.errorCount ?? null,
+		stopReasonContext: parseJson(row.stopReasonContext),
 		experimentId: row.experimentId ?? null,
 		entityType: row.entityType ?? null,
 		entityId: row.entityId ?? null,
@@ -5151,7 +5173,7 @@ function rowToSpanRecord(row) {
 		environment: row.environment ?? null,
 		source: row.source ?? null,
 		serviceName: row.serviceName ?? null,
-		attributes: parseJson(row.attributes),
+		attributes,
 		metadata: parseJson(row.metadata),
 		tags: parseJsonArray(row.tags),
 		scope: parseJson(row.scope),
@@ -5314,6 +5336,15 @@ function toValuesTuple(row) {
 		v(row.isEvent),
 		v(row.endedAt),
 		v(row.timeToFirstChunkMs),
+		v(row.externalParentSpanId),
+		v(row.resumedFromSpanId),
+		v(row.model),
+		v(row.provider),
+		v(row.outcome),
+		v(row.stopReason),
+		v(row.timeToFirstTokenMs),
+		v(row.errorCount),
+		jsonV(row.stopReasonContext),
 		v(row.experimentId),
 		v(row.entityType),
 		v(row.entityId),
@@ -5359,6 +5390,24 @@ async function insertSpanEvents(db, rows) {
 	const tuples = rows.map((row) => `(${toValuesTuple(row)})`).join(",\n");
 	await db.execute(`INSERT INTO span_events (${COLUMNS_SQL}) VALUES ${tuples}`);
 }
+/**
+* Columns copied as supplied onto both event rows (NULL when unset). Previews
+* stay out: the start row derives them from the payload it keeps.
+*/
+function copiedSpanColumns(s) {
+	return {
+		timeToFirstChunkMs: s.timeToFirstChunkMs ?? null,
+		externalParentSpanId: s.externalParentSpanId ?? null,
+		resumedFromSpanId: s.resumedFromSpanId ?? null,
+		model: s.model ?? null,
+		provider: s.provider ?? null,
+		outcome: s.outcome ?? null,
+		stopReason: s.stopReason ?? null,
+		timeToFirstTokenMs: s.timeToFirstTokenMs ?? null,
+		errorCount: s.errorCount ?? null,
+		stopReasonContext: s.stopReasonContext ?? null
+	};
+}
 function createStartSpanRow(s) {
 	const payloadOnEndRow = !!s.endedAt && !s.isEvent;
 	const input = previewOf(!payloadOnEndRow, s.input, s.inputPreview);
@@ -5372,7 +5421,7 @@ function createStartSpanRow(s) {
 		spanType: s.spanType,
 		isEvent: s.isEvent,
 		endedAt: s.isEvent ? s.startedAt : null,
-		timeToFirstChunkMs: s.timeToFirstChunkMs ?? null,
+		...copiedSpanColumns(s),
 		experimentId: s.experimentId ?? null,
 		entityType: s.entityType ?? null,
 		entityId: s.entityId ?? null,
@@ -5427,7 +5476,7 @@ function createEndSpanRow(s) {
 		spanType: s.spanType,
 		isEvent: s.isEvent,
 		endedAt: s.endedAt ?? null,
-		timeToFirstChunkMs: s.timeToFirstChunkMs ?? null,
+		...copiedSpanColumns(s),
 		experimentId: s.experimentId ?? null,
 		entityType: s.entityType ?? null,
 		entityId: s.entityId ?? null,
@@ -6482,4 +6531,4 @@ var ObservabilityStorageDuckDB = class ObservabilityStorageDuckDB extends Observ
 };
 export { ObservabilityStorageDuckDB };
 
-//# sourceMappingURL=observability-Chrw9Di2.js.map
+//# sourceMappingURL=observability-B-Kt8vrw.js.map