@case-framework/survey-assistant 0.1.0 → 0.2.0

package/dist/server.mjs

@@ -1,6 +1,137 @@
 import { n as SURVEY_ASSISTANT_TASK_AGENT_ID, t as SURVEY_ASSISTANT_AGENT_ID } from "./constants-CFUWrRtl.mjs";
-import { n as parseSurveyAssistantSnapshot, r as requireSurveyAssistantTurnState, t as buildSurveyAssistantRequestContext } from "./request-context-jcmXT_Q2.mjs";
-import { a as SURVEY_ASSISTANT_MAX_REQUEST_BYTES, c as SurveyAssistantRequestError, i as SURVEY_ASSISTANT_MAX_MESSAGES, l as readSurveyAssistantRequestJson, n as SURVEY_ASSISTANT_MAX_ATTACHMENTS, o as SURVEY_ASSISTANT_MAX_TEXT_PART_CHARS, r as SURVEY_ASSISTANT_MAX_ATTACHMENT_BYTES, s as SURVEY_ASSISTANT_MAX_TOTAL_ATTACHMENT_BYTES, t as SURVEY_ASSISTANT_ACCEPTED_ATTACHMENT_TYPES, u as validateSurveyAssistantMessages } from "./request-guardrails-C59IWnH8.mjs";
-import { a as describeSurveyAssistantStreamError, c as loadSurveyAssistantDocuments, d as removeHistoricalSurveyAssistantAttachments, f as createSurveyAssistantTurnStream, i as createSurveyAssistantChatHandler, l as loadSurveyAssistantThreadAttachments, n as createSurveyAssistantThreadHandlers, o as formatSurveyAssistantStreamError, r as SurveyAssistantAuthorizationError, s as getSurveyAssistantReplayableDocuments, t as createMastraMemoryThreadRepository, u as registerSurveyAssistantThreadAttachments } from "./memory-thread-repository-CJkkVE2-.mjs";
-import { n as surveyAuthoringReferences, t as surveyAuthoringReferenceTopics } from "./authoring-references-X6uRaWuf.mjs";
-export { SURVEY_ASSISTANT_ACCEPTED_ATTACHMENT_TYPES, SURVEY_ASSISTANT_AGENT_ID, SURVEY_ASSISTANT_MAX_ATTACHMENTS, SURVEY_ASSISTANT_MAX_ATTACHMENT_BYTES, SURVEY_ASSISTANT_MAX_MESSAGES, SURVEY_ASSISTANT_MAX_REQUEST_BYTES, SURVEY_ASSISTANT_MAX_TEXT_PART_CHARS, SURVEY_ASSISTANT_MAX_TOTAL_ATTACHMENT_BYTES, SURVEY_ASSISTANT_TASK_AGENT_ID, SurveyAssistantAuthorizationError, SurveyAssistantRequestError, buildSurveyAssistantRequestContext, createMastraMemoryThreadRepository, createSurveyAssistantChatHandler, createSurveyAssistantThreadHandlers, createSurveyAssistantTurnStream, describeSurveyAssistantStreamError, formatSurveyAssistantStreamError, getSurveyAssistantReplayableDocuments, loadSurveyAssistantDocuments, loadSurveyAssistantThreadAttachments, parseSurveyAssistantSnapshot, readSurveyAssistantRequestJson, registerSurveyAssistantThreadAttachments, removeHistoricalSurveyAssistantAttachments, requireSurveyAssistantTurnState, surveyAuthoringReferenceTopics, surveyAuthoringReferences, validateSurveyAssistantMessages };
+import { a as requireSurveyAssistantTurnState, i as parseSurveyAssistantSnapshot, n as deriveSurveyAssistantDocumentTransition, r as parseSurveyAssistantPreviousSettledVersion, t as buildSurveyAssistantRequestContext } from "./request-context-BDBw4-Gw.mjs";
+import { a as SURVEY_ASSISTANT_MAX_ATTACHMENT_NAME_CHARS, c as SURVEY_ASSISTANT_MAX_TEXT_PART_CHARS, d as readSurveyAssistantRequestJson, f as validateSurveyAssistantMessages, i as SURVEY_ASSISTANT_MAX_ATTACHMENT_BYTES, l as SURVEY_ASSISTANT_MAX_TOTAL_ATTACHMENT_BYTES, n as SURVEY_ASSISTANT_DEFAULT_REQUEST_LIMITS, o as SURVEY_ASSISTANT_MAX_MESSAGES, r as SURVEY_ASSISTANT_MAX_ATTACHMENTS, s as SURVEY_ASSISTANT_MAX_REQUEST_BYTES, t as SURVEY_ASSISTANT_ACCEPTED_ATTACHMENT_TYPES, u as SurveyAssistantRequestError } from "./request-guardrails-h4t6u8mP.mjs";
+import { a as describeSurveyAssistantStreamError, c as loadSurveyAssistantDocuments, d as removeHistoricalSurveyAssistantAttachments, f as replaceCurrentSurveyAssistantAttachmentsWithReferences, i as createSurveyAssistantChatHandler, l as loadSurveyAssistantThreadAttachments, n as createSurveyAssistantThreadHandlers, o as formatSurveyAssistantStreamError, p as createSurveyAssistantTurnStream, r as SurveyAssistantAuthorizationError, s as getSurveyAssistantReplayableDocuments, t as createMastraMemoryThreadRepository, u as registerSurveyAssistantThreadAttachments } from "./memory-thread-repository-Cm1BJyRw.mjs";
+import { n as surveyAuthoringReferenceTopics, r as surveyAuthoringReferences } from "./authoring-references-DYuOAdVx.mjs";
+//#region src/server/survey-lifecycle.ts
+const SURVEY_ASSISTANT_SURVEY_ID_METADATA_KEY = "surveyId";
+const normalizeIdentifier = (value, label) => {
+	const normalized = value.trim();
+	if (!normalized || normalized.length > 240) throw new Error(`${label} must contain between 1 and 240 characters.`);
+	return normalized;
+};
+const normalizeScopeMetadata = (metadata) => {
+	const entries = Object.entries(metadata);
+	if (entries.length === 0) throw new Error("Survey assistant lifecycle scope metadata must not be empty.");
+	if ("surveyId" in metadata) throw new Error(`Survey assistant lifecycle scope metadata must not contain "${SURVEY_ASSISTANT_SURVEY_ID_METADATA_KEY}".`);
+	return Object.fromEntries(entries.map(([key, value]) => [normalizeIdentifier(key, "Survey assistant lifecycle metadata key"), normalizeIdentifier(value, `Survey assistant lifecycle metadata "${key}"`)]));
+};
+const hasMetadata = (threadMetadata, expected) => Object.entries(expected).every(([key, value]) => threadMetadata?.[key] === value);
+const createSurveyAssistantSurveyThreadMetadata = ({ scopeMetadata, surveyId }) => ({
+	...normalizeScopeMetadata(scopeMetadata),
+	[SURVEY_ASSISTANT_SURVEY_ID_METADATA_KEY]: normalizeIdentifier(surveyId, "Survey assistant survey id")
+});
+/**
+* Creates an idempotent, server-only lifecycle service for removing every
+* assistant thread owned by a survey. Hosts should call this after preventing
+* new assistant activity and before completing (or from a durable retry of)
+* survey deletion.
+*/
+const createSurveyAssistantSurveyLifecycle = ({ getMemory, getThreadApi, scopeMetadata, threadDocuments }) => {
+	const normalizedScopeMetadata = Object.freeze(normalizeScopeMetadata(scopeMetadata));
+	const repository = createMastraMemoryThreadRepository({
+		getMemory,
+		threadDocuments: threadDocuments || void 0
+	});
+	const deleteMatchingThreads = async ({ metadata, resourceId }) => {
+		const memory = await getMemory();
+		const matchedThreads = (await memory.listThreads({
+			filter: {
+				metadata,
+				...resourceId ? { resourceId } : {}
+			},
+			orderBy: {
+				field: "createdAt",
+				direction: "ASC"
+			},
+			perPage: false
+		})).threads.filter((thread) => (!resourceId || thread.resourceId === resourceId) && hasMetadata(thread.metadata, metadata));
+		const deletedThreadIds = [];
+		for (const listedThread of matchedThreads) {
+			const currentThread = await memory.getThreadById({ threadId: listedThread.id });
+			if (!currentThread || resourceId && currentThread.resourceId !== resourceId || !hasMetadata(currentThread.metadata, metadata)) continue;
+			await repository.delete(currentThread.id);
+			deletedThreadIds.push(currentThread.id);
+		}
+		return {
+			deletedThreadIds,
+			matchedThreadCount: matchedThreads.length
+		};
+	};
+	const deleteSurveyThreads = async ({ resourceId, surveyId }) => {
+		const normalizedResourceId = resourceId ? normalizeIdentifier(resourceId, "Survey assistant resource id") : void 0;
+		return deleteMatchingThreads({
+			metadata: createSurveyAssistantSurveyThreadMetadata({
+				scopeMetadata: normalizedScopeMetadata,
+				surveyId
+			}),
+			resourceId: normalizedResourceId
+		});
+	};
+	const deleteSurvey = (surveyId) => deleteSurveyThreads({ surveyId });
+	const deleteResource = (resourceId) => deleteMatchingThreads({
+		metadata: normalizedScopeMetadata,
+		resourceId: normalizeIdentifier(resourceId, "Survey assistant resource id")
+	});
+	const forSurvey = (surveyId, resourceId) => {
+		const normalizedSurveyId = normalizeIdentifier(surveyId, "Survey assistant survey id");
+		const normalizedResourceId = normalizeIdentifier(resourceId, "Survey assistant resource id");
+		const metadata = Object.freeze(createSurveyAssistantSurveyThreadMetadata({
+			scopeMetadata: normalizedScopeMetadata,
+			surveyId: normalizedSurveyId
+		}));
+		const threadApi = getThreadApi?.(normalizedSurveyId)?.trim();
+		if (getThreadApi && !threadApi) throw new Error("Survey assistant thread API must not be empty.");
+		const panelProps = Object.freeze({
+			resourceId: normalizedResourceId,
+			surveyId: normalizedSurveyId,
+			...threadApi ? { threadApi } : {}
+		});
+		const threadIdentity = Object.freeze({
+			metadata,
+			resourceId: normalizedResourceId
+		});
+		const ownsThread = (thread) => Boolean(thread && thread.resourceId === normalizedResourceId && hasMetadata(thread.metadata, metadata));
+		const authorizeChat = async ({ requestedResourceId, requestedSurveyId, requestedThreadId }) => {
+			if (requestedResourceId !== normalizedResourceId || requestedSurveyId !== normalizedSurveyId || !requestedThreadId) throw new SurveyAssistantAuthorizationError("Survey assistant chat not found.", 404);
+			const thread = await (await getMemory()).getThreadById({ threadId: requestedThreadId });
+			if (!ownsThread(thread)) throw new SurveyAssistantAuthorizationError("Survey assistant chat not found.", 404);
+			return {
+				authority: "authenticated-host",
+				resourceId: normalizedResourceId,
+				surveyId: normalizedSurveyId,
+				threadId: thread.id
+			};
+		};
+		const authorizeTask = ({ requestedResourceId, requestedSurveyId }) => {
+			if (requestedResourceId !== normalizedResourceId || requestedSurveyId !== normalizedSurveyId) throw new SurveyAssistantAuthorizationError("Survey assistant task not found.", 404);
+			return {
+				resourceId: normalizedResourceId,
+				surveyId: normalizedSurveyId
+			};
+		};
+		return Object.freeze({
+			authorizeChat,
+			authorizeTask,
+			deleteThreads: () => deleteSurveyThreads({
+				resourceId: normalizedResourceId,
+				surveyId: normalizedSurveyId
+			}),
+			ownsThread,
+			panelProps,
+			resourceId: normalizedResourceId,
+			surveyId: normalizedSurveyId,
+			threadIdentity
+		});
+	};
+	return Object.freeze({
+		deleteResource,
+		deleteSurvey,
+		deleteSurveyThreads,
+		forSurvey
+	});
+};
+//#endregion
+export { SURVEY_ASSISTANT_ACCEPTED_ATTACHMENT_TYPES, SURVEY_ASSISTANT_AGENT_ID, SURVEY_ASSISTANT_DEFAULT_REQUEST_LIMITS, SURVEY_ASSISTANT_MAX_ATTACHMENTS, SURVEY_ASSISTANT_MAX_ATTACHMENT_BYTES, SURVEY_ASSISTANT_MAX_ATTACHMENT_NAME_CHARS, SURVEY_ASSISTANT_MAX_MESSAGES, SURVEY_ASSISTANT_MAX_REQUEST_BYTES, SURVEY_ASSISTANT_MAX_TEXT_PART_CHARS, SURVEY_ASSISTANT_MAX_TOTAL_ATTACHMENT_BYTES, SURVEY_ASSISTANT_SURVEY_ID_METADATA_KEY, SURVEY_ASSISTANT_TASK_AGENT_ID, SurveyAssistantAuthorizationError, SurveyAssistantRequestError, buildSurveyAssistantRequestContext, createMastraMemoryThreadRepository, createSurveyAssistantChatHandler, createSurveyAssistantSurveyLifecycle, createSurveyAssistantSurveyThreadMetadata, createSurveyAssistantThreadHandlers, createSurveyAssistantTurnStream, deriveSurveyAssistantDocumentTransition, describeSurveyAssistantStreamError, formatSurveyAssistantStreamError, getSurveyAssistantReplayableDocuments, loadSurveyAssistantDocuments, loadSurveyAssistantThreadAttachments, parseSurveyAssistantPreviousSettledVersion, parseSurveyAssistantSnapshot, readSurveyAssistantRequestJson, registerSurveyAssistantThreadAttachments, removeHistoricalSurveyAssistantAttachments, replaceCurrentSurveyAssistantAttachmentsWithReferences, requireSurveyAssistantTurnState, surveyAuthoringReferenceTopics, surveyAuthoringReferences, validateSurveyAssistantMessages };
+
+//# sourceMappingURL=server.mjs.map

package/dist/server.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.mjs","names":[],"sources":["../src/server/survey-lifecycle.ts"],"sourcesContent":["import type { MastraMemory } from \"@mastra/core/memory\";\nimport {\n  SurveyAssistantAuthorizationError,\n  type SurveyAssistantAuthorizationInput,\n  type SurveyAssistantAuthorizedIdentity,\n  type SurveyAssistantAuthorize,\n} from \"./handler\";\nimport { createMastraMemoryThreadRepository } from \"./memory-thread-repository\";\nimport type { SurveyAssistantThreadDocumentRepository } from \"./thread-documents\";\nimport type { SurveyAssistantStoredThread, SurveyAssistantThreadIdentity } from \"./thread-handlers\";\nimport type {\n  SurveyAssistantTaskAuthorizationInput,\n  SurveyAssistantTaskAuthorizedIdentity,\n} from \"./tasks\";\n\nexport const SURVEY_ASSISTANT_SURVEY_ID_METADATA_KEY = \"surveyId\";\n\ntype SurveyLifecycleMemory = Pick<\n  MastraMemory,\n  \"deleteThread\" | \"getThreadById\" | \"listThreads\" | \"recall\" | \"createThread\" | \"updateThread\"\n>;\n\nexport interface CreateSurveyAssistantSurveyLifecycleOptions {\n  getMemory(): SurveyLifecycleMemory | Promise<SurveyLifecycleMemory>;\n  /** Optionally derives the survey-scoped thread endpoint exposed in `panelProps`. */\n  getThreadApi?(surveyId: string): string;\n  /**\n   * Stable host metadata that separates this assistant deployment from any\n   * other application using the same Mastra storage.\n   */\n  scopeMetadata: Record<string, string>;\n  /**\n   * Pass the same repository used by the chat handler, or explicitly use\n   * `false` when cross-turn attachment persistence/replay is disabled.\n   */\n  threadDocuments: SurveyAssistantThreadDocumentRepository | false;\n}\n\nexport interface DeleteSurveyAssistantSurveyThreadsInput {\n  surveyId: string;\n  /** Restrict cleanup to one user. Omit this when the survey itself is deleted. */\n  resourceId?: string;\n}\n\nexport interface SurveyAssistantThreadCleanupResult {\n  deletedThreadIds: string[];\n  matchedThreadCount: number;\n}\n\nexport type DeleteSurveyAssistantSurveyThreadsResult = SurveyAssistantThreadCleanupResult;\n\nexport interface SurveyAssistantSurveyPanelProps {\n  resourceId: string;\n  surveyId: string;\n  threadApi?: string;\n}\n\nexport interface SurveyAssistantScopedSurvey {\n  /** Validates the requested ids and returns a trusted identity for the chat handler. */\n  authorizeChat: SurveyAssistantAuthorize;\n  /** Validates the requested survey/resource ids for the bounded task handler. */\n  authorizeTask(\n    input: SurveyAssistantTaskAuthorizationInput,\n  ): SurveyAssistantTaskAuthorizedIdentity;\n  /** Deletes this user's threads for the survey. */\n  deleteThreads(): Promise<DeleteSurveyAssistantSurveyThreadsResult>;\n  /** Returns true only for a thread in this exact deployment, survey, and user scope. */\n  ownsThread(\n    thread: SurveyAssistantStoredThread | null | undefined,\n  ): thread is SurveyAssistantStoredThread;\n  /** Props that can be passed directly to ConnectedSurveyAssistantPanel. */\n  panelProps: SurveyAssistantSurveyPanelProps;\n  resourceId: string;\n  surveyId: string;\n  /** Identity returned by the thread handler's authorize callback. */\n  threadIdentity: SurveyAssistantThreadIdentity;\n}\n\nexport interface SurveyAssistantSurveyLifecycle {\n  /** Deletes this resource's threads across every survey in this deployment scope. */\n  deleteResource(resourceId: string): Promise<SurveyAssistantThreadCleanupResult>;\n  /** Deletes every user's threads for a survey. */\n  deleteSurvey(surveyId: string): Promise<SurveyAssistantThreadCleanupResult>;\n  deleteSurveyThreads(\n    input: DeleteSurveyAssistantSurveyThreadsInput,\n  ): Promise<SurveyAssistantThreadCleanupResult>;\n  forSurvey(surveyId: string, resourceId: string): SurveyAssistantScopedSurvey;\n}\n\nconst normalizeIdentifier = (value: string, label: string) => {\n  const normalized = value.trim();\n  if (!normalized || normalized.length > 240) {\n    throw new Error(`${label} must contain between 1 and 240 characters.`);\n  }\n  return normalized;\n};\n\nconst normalizeScopeMetadata = (metadata: Record<string, string>) => {\n  const entries = Object.entries(metadata);\n  if (entries.length === 0) {\n    throw new Error(\"Survey assistant lifecycle scope metadata must not be empty.\");\n  }\n  if (SURVEY_ASSISTANT_SURVEY_ID_METADATA_KEY in metadata) {\n    throw new Error(\n      `Survey assistant lifecycle scope metadata must not contain \"${SURVEY_ASSISTANT_SURVEY_ID_METADATA_KEY}\".`,\n    );\n  }\n\n  return Object.fromEntries(\n    entries.map(([key, value]) => [\n      normalizeIdentifier(key, \"Survey assistant lifecycle metadata key\"),\n      normalizeIdentifier(value, `Survey assistant lifecycle metadata \"${key}\"`),\n    ]),\n  );\n};\n\nconst hasMetadata = (\n  threadMetadata: Record<string, unknown> | undefined,\n  expected: Record<string, string>,\n) => Object.entries(expected).every(([key, value]) => threadMetadata?.[key] === value);\n\nexport const createSurveyAssistantSurveyThreadMetadata = ({\n  scopeMetadata,\n  surveyId,\n}: {\n  scopeMetadata: Record<string, string>;\n  surveyId: string;\n}): Record<string, string> => ({\n  ...normalizeScopeMetadata(scopeMetadata),\n  [SURVEY_ASSISTANT_SURVEY_ID_METADATA_KEY]: normalizeIdentifier(\n    surveyId,\n    \"Survey assistant survey id\",\n  ),\n});\n\n/**\n * Creates an idempotent, server-only lifecycle service for removing every\n * assistant thread owned by a survey. Hosts should call this after preventing\n * new assistant activity and before completing (or from a durable retry of)\n * survey deletion.\n */\nexport const createSurveyAssistantSurveyLifecycle = ({\n  getMemory,\n  getThreadApi,\n  scopeMetadata,\n  threadDocuments,\n}: CreateSurveyAssistantSurveyLifecycleOptions): SurveyAssistantSurveyLifecycle => {\n  const normalizedScopeMetadata = Object.freeze(normalizeScopeMetadata(scopeMetadata));\n  const repository = createMastraMemoryThreadRepository({\n    getMemory,\n    threadDocuments: threadDocuments || undefined,\n  });\n\n  const deleteMatchingThreads = async ({\n    metadata,\n    resourceId,\n  }: {\n    metadata: Record<string, string>;\n    resourceId?: string;\n  }): Promise<DeleteSurveyAssistantSurveyThreadsResult> => {\n    const memory = await getMemory();\n    const result = await memory.listThreads({\n      filter: {\n        metadata,\n        ...(resourceId ? { resourceId } : {}),\n      },\n      orderBy: { field: \"createdAt\", direction: \"ASC\" },\n      perPage: false,\n    });\n    const matchedThreads = result.threads.filter(\n      (thread) =>\n        (!resourceId || thread.resourceId === resourceId) && hasMetadata(thread.metadata, metadata),\n    );\n    const deletedThreadIds: string[] = [];\n\n    for (const listedThread of matchedThreads) {\n      // Re-check the current record so a concurrently reassigned thread is not\n      // deleted from a stale list result.\n      const currentThread = await memory.getThreadById({ threadId: listedThread.id });\n      if (\n        !currentThread ||\n        (resourceId && currentThread.resourceId !== resourceId) ||\n        !hasMetadata(currentThread.metadata, metadata)\n      ) {\n        continue;\n      }\n      await repository.delete(currentThread.id);\n      deletedThreadIds.push(currentThread.id);\n    }\n\n    return {\n      deletedThreadIds,\n      matchedThreadCount: matchedThreads.length,\n    };\n  };\n\n  const deleteSurveyThreads = async ({\n    resourceId,\n    surveyId,\n  }: DeleteSurveyAssistantSurveyThreadsInput): Promise<DeleteSurveyAssistantSurveyThreadsResult> => {\n    const normalizedResourceId = resourceId\n      ? normalizeIdentifier(resourceId, \"Survey assistant resource id\")\n      : undefined;\n    return deleteMatchingThreads({\n      metadata: createSurveyAssistantSurveyThreadMetadata({\n        scopeMetadata: normalizedScopeMetadata,\n        surveyId,\n      }),\n      resourceId: normalizedResourceId,\n    });\n  };\n\n  const deleteSurvey = (surveyId: string) => deleteSurveyThreads({ surveyId });\n  const deleteResource = (resourceId: string) =>\n    deleteMatchingThreads({\n      metadata: normalizedScopeMetadata,\n      resourceId: normalizeIdentifier(resourceId, \"Survey assistant resource id\"),\n    });\n\n  const forSurvey = (surveyId: string, resourceId: string): SurveyAssistantScopedSurvey => {\n    const normalizedSurveyId = normalizeIdentifier(surveyId, \"Survey assistant survey id\");\n    const normalizedResourceId = normalizeIdentifier(resourceId, \"Survey assistant resource id\");\n    const metadata = Object.freeze(\n      createSurveyAssistantSurveyThreadMetadata({\n        scopeMetadata: normalizedScopeMetadata,\n        surveyId: normalizedSurveyId,\n      }),\n    );\n    const threadApi = getThreadApi?.(normalizedSurveyId)?.trim();\n    if (getThreadApi && !threadApi) {\n      throw new Error(\"Survey assistant thread API must not be empty.\");\n    }\n    const panelProps: SurveyAssistantSurveyPanelProps = Object.freeze({\n      resourceId: normalizedResourceId,\n      surveyId: normalizedSurveyId,\n      ...(threadApi ? { threadApi } : {}),\n    });\n    const threadIdentity: SurveyAssistantThreadIdentity = Object.freeze({\n      metadata,\n      resourceId: normalizedResourceId,\n    });\n    const ownsThread = (\n      thread: SurveyAssistantStoredThread | null | undefined,\n    ): thread is SurveyAssistantStoredThread =>\n      Boolean(\n        thread &&\n        thread.resourceId === normalizedResourceId &&\n        hasMetadata(thread.metadata, metadata),\n      );\n    const authorizeChat = async ({\n      requestedResourceId,\n      requestedSurveyId,\n      requestedThreadId,\n    }: SurveyAssistantAuthorizationInput): Promise<SurveyAssistantAuthorizedIdentity> => {\n      if (\n        requestedResourceId !== normalizedResourceId ||\n        requestedSurveyId !== normalizedSurveyId ||\n        !requestedThreadId\n      ) {\n        throw new SurveyAssistantAuthorizationError(\"Survey assistant chat not found.\", 404);\n      }\n      const memory = await getMemory();\n      const thread = await memory.getThreadById({ threadId: requestedThreadId });\n      if (!ownsThread(thread)) {\n        throw new SurveyAssistantAuthorizationError(\"Survey assistant chat not found.\", 404);\n      }\n      return {\n        authority: \"authenticated-host\",\n        resourceId: normalizedResourceId,\n        surveyId: normalizedSurveyId,\n        threadId: thread.id,\n      };\n    };\n    const authorizeTask = ({\n      requestedResourceId,\n      requestedSurveyId,\n    }: SurveyAssistantTaskAuthorizationInput): SurveyAssistantTaskAuthorizedIdentity => {\n      if (\n        requestedResourceId !== normalizedResourceId ||\n        requestedSurveyId !== normalizedSurveyId\n      ) {\n        throw new SurveyAssistantAuthorizationError(\"Survey assistant task not found.\", 404);\n      }\n      return {\n        resourceId: normalizedResourceId,\n        surveyId: normalizedSurveyId,\n      };\n    };\n\n    return Object.freeze({\n      authorizeChat,\n      authorizeTask,\n      deleteThreads: () =>\n        deleteSurveyThreads({\n          resourceId: normalizedResourceId,\n          surveyId: normalizedSurveyId,\n        }),\n      ownsThread,\n      panelProps,\n      resourceId: normalizedResourceId,\n      surveyId: normalizedSurveyId,\n      threadIdentity,\n    });\n  };\n\n  return Object.freeze({ deleteResource, deleteSurvey, deleteSurveyThreads, forSurvey });\n};\n"],"mappings":";;;;;;AAeA,MAAa,0CAA0C;AA0EvD,MAAM,uBAAuB,OAAe,UAAkB;CAC5D,MAAM,aAAa,MAAM,MAAM;AAC/B,KAAI,CAAC,cAAc,WAAW,SAAS,IACrC,OAAM,IAAI,MAAM,GAAG,MAAM,6CAA6C;AAExE,QAAO;;AAGT,MAAM,0BAA0B,aAAqC;CACnE,MAAM,UAAU,OAAO,QAAQ,SAAS;AACxC,KAAI,QAAQ,WAAW,EACrB,OAAM,IAAI,MAAM,+DAA+D;AAEjF,KAAA,cAA+C,SAC7C,OAAM,IAAI,MACR,+DAA+D,wCAAwC,IACxG;AAGH,QAAO,OAAO,YACZ,QAAQ,KAAK,CAAC,KAAK,WAAW,CAC5B,oBAAoB,KAAK,0CAA0C,EACnE,oBAAoB,OAAO,wCAAwC,IAAI,GAAG,CAC3E,CAAC,CACH;;AAGH,MAAM,eACJ,gBACA,aACG,OAAO,QAAQ,SAAS,CAAC,OAAO,CAAC,KAAK,WAAW,iBAAiB,SAAS,MAAM;AAEtF,MAAa,6CAA6C,EACxD,eACA,gBAI6B;CAC7B,GAAG,uBAAuB,cAAc;EACvC,0CAA0C,oBACzC,UACA,6BACD;CACF;;;;;;;AAQD,MAAa,wCAAwC,EACnD,WACA,cACA,eACA,sBACiF;CACjF,MAAM,0BAA0B,OAAO,OAAO,uBAAuB,cAAc,CAAC;CACpF,MAAM,aAAa,mCAAmC;EACpD;EACA,iBAAiB,mBAAmB,KAAA;EACrC,CAAC;CAEF,MAAM,wBAAwB,OAAO,EACnC,UACA,iBAIuD;EACvD,MAAM,SAAS,MAAM,WAAW;EAShC,MAAM,kBAAiB,MARF,OAAO,YAAY;GACtC,QAAQ;IACN;IACA,GAAI,aAAa,EAAE,YAAY,GAAG,EAAE;IACrC;GACD,SAAS;IAAE,OAAO;IAAa,WAAW;IAAO;GACjD,SAAS;GACV,CAAC,EAC4B,QAAQ,QACnC,YACE,CAAC,cAAc,OAAO,eAAe,eAAe,YAAY,OAAO,UAAU,SAAS,CAC9F;EACD,MAAM,mBAA6B,EAAE;AAErC,OAAK,MAAM,gBAAgB,gBAAgB;GAGzC,MAAM,gBAAgB,MAAM,OAAO,cAAc,EAAE,UAAU,aAAa,IAAI,CAAC;AAC/E,OACE,CAAC,iBACA,cAAc,cAAc,eAAe,cAC5C,CAAC,YAAY,cAAc,UAAU,SAAS,CAE9C;AAEF,SAAM,WAAW,OAAO,cAAc,GAAG;AACzC,oBAAiB,KAAK,cAAc,GAAG;;AAGzC,SAAO;GACL;GACA,oBAAoB,eAAe;GACpC;;CAGH,MAAM,sBAAsB,OAAO,EACjC,YACA,eACgG;EAChG,MAAM,uBAAuB,aACzB,oBAAoB,YAAY,+BAA+B,GAC/D,KAAA;AACJ,SAAO,sBAAsB;GAC3B,UAAU,0CAA0C;IAClD,eAAe;IACf;IACD,CAAC;GACF,YAAY;GACb,CAAC;;CAGJ,MAAM,gBAAgB,aAAqB,oBAAoB,EAAE,UAAU,CAAC;CAC5E,MAAM,kBAAkB,eACtB,sBAAsB;EACpB,UAAU;EACV,YAAY,oBAAoB,YAAY,+BAA+B;EAC5E,CAAC;CAEJ,MAAM,aAAa,UAAkB,eAAoD;EACvF,MAAM,qBAAqB,oBAAoB,UAAU,6BAA6B;EACtF,MAAM,uBAAuB,oBAAoB,YAAY,+BAA+B;EAC5F,MAAM,WAAW,OAAO,OACtB,0CAA0C;GACxC,eAAe;GACf,UAAU;GACX,CAAC,CACH;EACD,MAAM,YAAY,eAAe,mBAAmB,EAAE,MAAM;AAC5D,MAAI,gBAAgB,CAAC,UACnB,OAAM,IAAI,MAAM,iDAAiD;EAEnE,MAAM,aAA8C,OAAO,OAAO;GAChE,YAAY;GACZ,UAAU;GACV,GAAI,YAAY,EAAE,WAAW,GAAG,EAAE;GACnC,CAAC;EACF,MAAM,iBAAgD,OAAO,OAAO;GAClE;GACA,YAAY;GACb,CAAC;EACF,MAAM,cACJ,WAEA,QACE,UACA,OAAO,eAAe,wBACtB,YAAY,OAAO,UAAU,SAAS,CACvC;EACH,MAAM,gBAAgB,OAAO,EAC3B,qBACA,mBACA,wBACmF;AACnF,OACE,wBAAwB,wBACxB,sBAAsB,sBACtB,CAAC,kBAED,OAAM,IAAI,kCAAkC,oCAAoC,IAAI;GAGtF,MAAM,SAAS,OAAM,MADA,WAAW,EACJ,cAAc,EAAE,UAAU,mBAAmB,CAAC;AAC1E,OAAI,CAAC,WAAW,OAAO,CACrB,OAAM,IAAI,kCAAkC,oCAAoC,IAAI;AAEtF,UAAO;IACL,WAAW;IACX,YAAY;IACZ,UAAU;IACV,UAAU,OAAO;IAClB;;EAEH,MAAM,iBAAiB,EACrB,qBACA,wBACkF;AAClF,OACE,wBAAwB,wBACxB,sBAAsB,mBAEtB,OAAM,IAAI,kCAAkC,oCAAoC,IAAI;AAEtF,UAAO;IACL,YAAY;IACZ,UAAU;IACX;;AAGH,SAAO,OAAO,OAAO;GACnB;GACA;GACA,qBACE,oBAAoB;IAClB,YAAY;IACZ,UAAU;IACX,CAAC;GACJ;GACA;GACA,YAAY;GACZ,UAAU;GACV;GACD,CAAC;;AAGJ,QAAO,OAAO,OAAO;EAAE;EAAgB;EAAc;EAAqB;EAAW,CAAC"}

package/dist/storage-postgres.d.mts

@@ -1,7 +1,9 @@
-import { a as SurveyAssistantThreadDocumentRepository } from "./thread-documents-DamIq9Od.mjs";
+import { a as SurveyAssistantThreadDocumentRepository } from "./thread-documents-BlxNjc7S.mjs";
 import { PostgresStore, PostgresStoreConfig } from "@mastra/pg";
 
 //#region src/storage/postgres.d.ts
+declare const SURVEY_ASSISTANT_DEFAULT_MAX_STORED_ATTACHMENTS_PER_THREAD = 100;
+declare const SURVEY_ASSISTANT_DEFAULT_MAX_STORED_ATTACHMENT_BYTES_PER_THREAD: number;
 interface InitializeSurveyAssistantPostgresStorageOptions {
   ensureObservationalMemory?: boolean;
   schemaName?: string;
@@ -12,19 +14,27 @@ declare const initializeSurveyAssistantPostgresStorage: (storage: PostgresStore,
 }?: InitializeSurveyAssistantPostgresStorageOptions) => Promise<PostgresStore>;
 interface CreateSurveyAssistantPostgresThreadDocumentRepositoryOptions {
   getStorage(): PostgresStore | Promise<PostgresStore>;
+  maxStoredAttachmentsPerThread?: number;
+  maxStoredBytesPerThread?: number;
   schemaName?: string;
 }
 declare const createSurveyAssistantPostgresThreadDocumentRepository: ({
   getStorage,
+  maxStoredAttachmentsPerThread,
+  maxStoredBytesPerThread,
   schemaName
 }: CreateSurveyAssistantPostgresThreadDocumentRepositoryOptions) => SurveyAssistantThreadDocumentRepository;
 interface CreateSurveyAssistantPostgresRuntimeOptions {
   ensureObservationalMemory?: boolean;
+  maxStoredAttachmentsPerThread?: number;
+  maxStoredBytesPerThread?: number;
   schemaName?: string;
   store: PostgresStore | PostgresStoreConfig;
 }
 declare const createSurveyAssistantPostgresRuntime: ({
   ensureObservationalMemory,
+  maxStoredAttachmentsPerThread,
+  maxStoredBytesPerThread,
   schemaName,
   store: storeOrConfig
 }: CreateSurveyAssistantPostgresRuntimeOptions) => {
@@ -33,5 +43,5 @@ declare const createSurveyAssistantPostgresRuntime: ({
   threadDocuments: SurveyAssistantThreadDocumentRepository;
 };
 //#endregion
-export { CreateSurveyAssistantPostgresRuntimeOptions, CreateSurveyAssistantPostgresThreadDocumentRepositoryOptions, InitializeSurveyAssistantPostgresStorageOptions, createSurveyAssistantPostgresRuntime, createSurveyAssistantPostgresThreadDocumentRepository, initializeSurveyAssistantPostgresStorage };
+export { CreateSurveyAssistantPostgresRuntimeOptions, CreateSurveyAssistantPostgresThreadDocumentRepositoryOptions, InitializeSurveyAssistantPostgresStorageOptions, SURVEY_ASSISTANT_DEFAULT_MAX_STORED_ATTACHMENTS_PER_THREAD, SURVEY_ASSISTANT_DEFAULT_MAX_STORED_ATTACHMENT_BYTES_PER_THREAD, createSurveyAssistantPostgresRuntime, createSurveyAssistantPostgresThreadDocumentRepository, initializeSurveyAssistantPostgresStorage };
 //# sourceMappingURL=storage-postgres.d.mts.map

package/dist/storage-postgres.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"storage-postgres.d.mts","names":[],"sources":["../src/storage/postgres.ts"],"mappings":";;;;UAwHiB,+CAAA;EACf,yBAAA;EACA,UAAA;AAAA;AAAA,cAGW,wCAAA,GACX,OAAA,EAAS,aAAA;EACT,yBAAA;EAAA;AAAA,IAGG,+CAAA,KAAoD,OAAA,CAAA,aAAA;AAAA,UAwBxC,4DAAA;EACf,UAAA,IAAc,aAAA,GAAgB,OAAA,CAAQ,aAAA;EACtC,UAAA;AAAA;AAAA,cAGW,qDAAA;EAAyD,UAAA;EAAA;AAAA,GAGnE,4DAAA,KAA+D,uCAAA;AAAA,UAmFjD,2CAAA;EACf,yBAAA;EACA,UAAA;EACA,KAAA,EAAO,aAAA,GAAgB,mBAAA;AAAA;AAAA,cAGZ,oCAAA;EAAwC,yBAAA;EAAA,UAAA;EAAA,KAAA,EAAA;AAAA,GAIlD,2CAAA"}
+{"version":3,"file":"storage-postgres.d.mts","names":[],"sources":["../src/storage/postgres.ts"],"mappings":";;;;cAWa,0DAAA;AAAA,cACA,+DAAA;AAAA,UA0GI,+CAAA;EACf,yBAAA;EACA,UAAA;AAAA;AAAA,cAGW,wCAAA,GACX,OAAA,EAAS,aAAA;EACT,yBAAA;EAAA;AAAA,IAGG,+CAAA,KAAoD,OAAA,CAAA,aAAA;AAAA,UAwBxC,4DAAA;EACf,UAAA,IAAc,aAAA,GAAgB,OAAA,CAAQ,aAAA;EACtC,6BAAA;EACA,uBAAA;EACA,UAAA;AAAA;AAAA,cAGW,qDAAA;EAAyD,UAAA;EAAA,6BAAA;EAAA,uBAAA;EAAA;AAAA,GAKnE,4DAAA,KAA+D,uCAAA;AAAA,UA2FjD,2CAAA;EACf,yBAAA;EACA,6BAAA;EACA,uBAAA;EACA,UAAA;EACA,KAAA,EAAO,aAAA,GAAgB,mBAAA;AAAA;AAAA,cAGZ,oCAAA;EAAwC,yBAAA;EAAA,6BAAA;EAAA,uBAAA;EAAA,UAAA;EAAA,KAAA,EAAA;AAAA,GAMlD,2CAAA"}

package/dist/storage-postgres.mjs

@@ -1,10 +1,12 @@
-import { c as SurveyAssistantRequestError } from "./request-guardrails-C59IWnH8.mjs";
+import { u as SurveyAssistantRequestError } from "./request-guardrails-h4t6u8mP.mjs";
 import { createHash, randomUUID } from "node:crypto";
 import { OBSERVATIONAL_MEMORY_TABLE_SCHEMA } from "@mastra/core/storage";
 import { PostgresStore } from "@mastra/pg";
 //#region src/storage/postgres.ts
 const OBSERVATIONAL_MEMORY_TABLE = "mastra_observational_memory";
 const THREAD_DOCUMENTS_TABLE = "survey_assistant_thread_documents";
+const SURVEY_ASSISTANT_DEFAULT_MAX_STORED_ATTACHMENTS_PER_THREAD = 100;
+const SURVEY_ASSISTANT_DEFAULT_MAX_STORED_ATTACHMENT_BYTES_PER_THREAD = 200 * 1024 * 1024;
 const quoteIdentifier = (value) => `"${value.replaceAll("\"", "\"\"")}"`;
 const postgresType = (column) => {
 	switch (column.type) {
@@ -82,7 +84,8 @@ const toThreadDocument = (row) => ({
 	contentHash: row.content_hash,
 	createdAt: new Date(row.created_at).toISOString()
 });
-const createSurveyAssistantPostgresThreadDocumentRepository = ({ getStorage, schemaName = "public" }) => {
+const createSurveyAssistantPostgresThreadDocumentRepository = ({ getStorage, maxStoredAttachmentsPerThread = 100, maxStoredBytesPerThread = SURVEY_ASSISTANT_DEFAULT_MAX_STORED_ATTACHMENT_BYTES_PER_THREAD, schemaName = "public" }) => {
+	if (!Number.isSafeInteger(maxStoredAttachmentsPerThread) || maxStoredAttachmentsPerThread <= 0 || !Number.isSafeInteger(maxStoredBytesPerThread) || maxStoredBytesPerThread <= 0) throw new Error("Survey assistant stored attachment limits must be positive integers.");
 	const tableName = `${quoteIdentifier(schemaName)}.${quoteIdentifier(THREAD_DOCUMENTS_TABLE)}`;
 	return {
 		async put({ bytes, data, mediaType, name, resourceId, threadId }) {
@@ -99,8 +102,8 @@ const createSurveyAssistantPostgresThreadDocumentRepository = ({ getStorage, sch
 				if (existing) return existing;
 				const usage = await transaction.one(`SELECT COUNT(*) AS count, COALESCE(SUM(byte_size), 0) AS bytes FROM ${tableName}
            WHERE resource_id = $1 AND thread_id = $2`, [resourceId, threadId]);
-				if (Number(usage.count) >= 4) throw new SurveyAssistantRequestError(`A chat can contain up to 4 attachments.`);
-				if (Number(usage.bytes) + bytes > 12582912) throw new SurveyAssistantRequestError("The attachments stored for this chat are too large in total.");
+				if (Number(usage.count) >= maxStoredAttachmentsPerThread) throw new SurveyAssistantRequestError(`A chat can contain up to ${maxStoredAttachmentsPerThread} stored attachments.`);
+				if (Number(usage.bytes) + bytes > maxStoredBytesPerThread) throw new SurveyAssistantRequestError("The attachments stored for this chat are too large in total.");
 				return transaction.one(`INSERT INTO ${tableName}
             (id, resource_id, thread_id, name, media_type, byte_size, content_hash, data)
            VALUES
@@ -142,7 +145,7 @@ const createSurveyAssistantPostgresThreadDocumentRepository = ({ getStorage, sch
 		}
 	};
 };
-const createSurveyAssistantPostgresRuntime = ({ ensureObservationalMemory, schemaName = "public", store: storeOrConfig }) => {
+const createSurveyAssistantPostgresRuntime = ({ ensureObservationalMemory, maxStoredAttachmentsPerThread, maxStoredBytesPerThread, schemaName = "public", store: storeOrConfig }) => {
 	const storage = storeOrConfig instanceof PostgresStore ? storeOrConfig : new PostgresStore(storeOrConfig);
 	let initialization;
 	const getStorage = () => {
@@ -157,11 +160,13 @@ const createSurveyAssistantPostgresRuntime = ({ ensureObservationalMemory, schem
 		getStorage,
 		threadDocuments: createSurveyAssistantPostgresThreadDocumentRepository({
 			getStorage,
+			maxStoredAttachmentsPerThread,
+			maxStoredBytesPerThread,
 			schemaName
 		})
 	};
 };
 //#endregion
-export { createSurveyAssistantPostgresRuntime, createSurveyAssistantPostgresThreadDocumentRepository, initializeSurveyAssistantPostgresStorage };
+export { SURVEY_ASSISTANT_DEFAULT_MAX_STORED_ATTACHMENTS_PER_THREAD, SURVEY_ASSISTANT_DEFAULT_MAX_STORED_ATTACHMENT_BYTES_PER_THREAD, createSurveyAssistantPostgresRuntime, createSurveyAssistantPostgresThreadDocumentRepository, initializeSurveyAssistantPostgresStorage };
 
 //# sourceMappingURL=storage-postgres.mjs.map

package/dist/storage-postgres.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"storage-postgres.mjs","names":[],"sources":["../src/storage/postgres.ts"],"sourcesContent":["import { OBSERVATIONAL_MEMORY_TABLE_SCHEMA, type StorageColumn } from \"@mastra/core/storage\";\nimport { PostgresStore, type PostgresStoreConfig } from \"@mastra/pg\";\nimport { createHash, randomUUID } from \"node:crypto\";\nimport {\n  SURVEY_ASSISTANT_MAX_ATTACHMENTS,\n  SURVEY_ASSISTANT_MAX_TOTAL_ATTACHMENT_BYTES,\n  SurveyAssistantRequestError,\n} from \"../server/request-guardrails\";\nimport type {\n  SurveyAssistantThreadDocument,\n  SurveyAssistantThreadDocumentRepository,\n} from \"../server/thread-documents\";\n\nconst OBSERVATIONAL_MEMORY_TABLE = \"mastra_observational_memory\";\nconst THREAD_DOCUMENTS_TABLE = \"survey_assistant_thread_documents\";\n\nconst quoteIdentifier = (value: string) => `\"${value.replaceAll('\"', '\"\"')}\"`;\n\nconst postgresType = (column: StorageColumn) => {\n  switch (column.type) {\n    case \"text\":\n      return \"TEXT\";\n    case \"timestamp\":\n      return \"TIMESTAMP\";\n    case \"jsonb\":\n      return \"JSONB\";\n    case \"integer\":\n      return \"INTEGER\";\n    case \"float\":\n      return \"DOUBLE PRECISION\";\n    case \"bigint\":\n      return \"BIGINT\";\n    case \"boolean\":\n      return \"BOOLEAN\";\n    case \"uuid\":\n      return \"UUID\";\n  }\n};\n\nconst migrationDefault = (column: StorageColumn) => {\n  switch (column.type) {\n    case \"boolean\":\n      return \"FALSE\";\n    case \"integer\":\n    case \"bigint\":\n    case \"float\":\n      return \"0\";\n    case \"jsonb\":\n      return \"'{}'::jsonb\";\n    case \"timestamp\":\n      return \"NOW()\";\n    case \"text\":\n      return \"''\";\n    case \"uuid\":\n      return \"gen_random_uuid()\";\n  }\n};\n\nconst columnSql = (name: string, column: StorageColumn, includeMigrationDefault = false) => {\n  const constraints = [\n    column.primaryKey ? \"PRIMARY KEY\" : undefined,\n    column.nullable === false ? \"NOT NULL\" : undefined,\n    includeMigrationDefault && column.nullable === false\n      ? `DEFAULT ${migrationDefault(column)}`\n      : undefined,\n  ]\n    .filter(Boolean)\n    .join(\" \");\n  return `${quoteIdentifier(name)} ${postgresType(column)}${constraints ? ` ${constraints}` : \"\"}`;\n};\n\nconst ensureObservationalMemorySchema = async (storage: PostgresStore, schemaName: string) => {\n  const tableSchema = OBSERVATIONAL_MEMORY_TABLE_SCHEMA[OBSERVATIONAL_MEMORY_TABLE];\n  if (!tableSchema) throw new Error(\"Mastra observational memory schema is unavailable.\");\n\n  const tableName = `${quoteIdentifier(schemaName)}.${quoteIdentifier(OBSERVATIONAL_MEMORY_TABLE)}`;\n  const columns = Object.entries(tableSchema).flatMap(([name, column]) => [\n    columnSql(name, column),\n    ...(column.type === \"timestamp\"\n      ? [`${quoteIdentifier(`${name}Z`)} TIMESTAMPTZ DEFAULT NOW()`]\n      : []),\n  ]);\n\n  await storage.db.none(`CREATE TABLE IF NOT EXISTS ${tableName} (${columns.join(\", \")})`);\n  for (const [name, column] of Object.entries(tableSchema)) {\n    await storage.db.none(\n      `ALTER TABLE ${tableName} ADD COLUMN IF NOT EXISTS ${columnSql(name, column, true)}`,\n    );\n    if (column.type === \"timestamp\") {\n      await storage.db.none(\n        `ALTER TABLE ${tableName} ADD COLUMN IF NOT EXISTS ${quoteIdentifier(`${name}Z`)} TIMESTAMPTZ DEFAULT NOW()`,\n      );\n    }\n  }\n  await storage.db.none(\n    `CREATE INDEX IF NOT EXISTS idx_om_lookup_key ON ${tableName} (${quoteIdentifier(\"lookupKey\")})`,\n  );\n};\n\nconst ensureThreadDocumentsSchema = async (storage: PostgresStore, schemaName: string) => {\n  const tableName = `${quoteIdentifier(schemaName)}.${quoteIdentifier(THREAD_DOCUMENTS_TABLE)}`;\n  await storage.db.none(`\n    CREATE TABLE IF NOT EXISTS ${tableName} (\n      id TEXT PRIMARY KEY,\n      resource_id TEXT NOT NULL,\n      thread_id TEXT NOT NULL,\n      name TEXT NOT NULL,\n      media_type TEXT NOT NULL,\n      byte_size INTEGER NOT NULL,\n      content_hash TEXT NOT NULL,\n      data BYTEA NOT NULL,\n      created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),\n      UNIQUE (resource_id, thread_id, content_hash)\n    )\n  `);\n  await storage.db.none(\n    `CREATE INDEX IF NOT EXISTS idx_survey_assistant_thread_documents_thread ON ${tableName} (resource_id, thread_id, created_at)`,\n  );\n};\n\nexport interface InitializeSurveyAssistantPostgresStorageOptions {\n  ensureObservationalMemory?: boolean;\n  schemaName?: string;\n}\n\nexport const initializeSurveyAssistantPostgresStorage = async (\n  storage: PostgresStore,\n  {\n    ensureObservationalMemory = true,\n    schemaName = \"public\",\n  }: InitializeSurveyAssistantPostgresStorageOptions = {},\n) => {\n  await storage.init();\n  if (ensureObservationalMemory) await ensureObservationalMemorySchema(storage, schemaName);\n  await ensureThreadDocumentsSchema(storage, schemaName);\n  return storage;\n};\n\nconst toThreadDocument = (row: {\n  id: string;\n  name: string;\n  media_type: string;\n  byte_size: number;\n  content_hash: string;\n  created_at: Date | string;\n}): SurveyAssistantThreadDocument => ({\n  id: row.id,\n  name: row.name,\n  mediaType: row.media_type,\n  bytes: row.byte_size,\n  contentHash: row.content_hash,\n  createdAt: new Date(row.created_at).toISOString(),\n});\n\nexport interface CreateSurveyAssistantPostgresThreadDocumentRepositoryOptions {\n  getStorage(): PostgresStore | Promise<PostgresStore>;\n  schemaName?: string;\n}\n\nexport const createSurveyAssistantPostgresThreadDocumentRepository = ({\n  getStorage,\n  schemaName = \"public\",\n}: CreateSurveyAssistantPostgresThreadDocumentRepositoryOptions): SurveyAssistantThreadDocumentRepository => {\n  const tableName = `${quoteIdentifier(schemaName)}.${quoteIdentifier(THREAD_DOCUMENTS_TABLE)}`;\n  return {\n    async put({ bytes, data, mediaType, name, resourceId, threadId }) {\n      const storage = await getStorage();\n      const contentHash = createHash(\"sha256\").update(data).digest(\"hex\");\n      const row = await storage.db.tx(async (transaction) => {\n        await transaction.none(\"SELECT pg_advisory_xact_lock(hashtext($1), hashtext($2))\", [\n          resourceId,\n          threadId,\n        ]);\n        const existing = await transaction.oneOrNone(\n          `SELECT id, name, media_type, byte_size, content_hash, created_at FROM ${tableName}\n           WHERE resource_id = $1 AND thread_id = $2 AND content_hash = $3`,\n          [resourceId, threadId, contentHash],\n        );\n        if (existing) return existing;\n\n        const usage = await transaction.one<{ count: string; bytes: string }>(\n          `SELECT COUNT(*) AS count, COALESCE(SUM(byte_size), 0) AS bytes FROM ${tableName}\n           WHERE resource_id = $1 AND thread_id = $2`,\n          [resourceId, threadId],\n        );\n        if (Number(usage.count) >= SURVEY_ASSISTANT_MAX_ATTACHMENTS) {\n          throw new SurveyAssistantRequestError(\n            `A chat can contain up to ${SURVEY_ASSISTANT_MAX_ATTACHMENTS} attachments.`,\n          );\n        }\n        if (Number(usage.bytes) + bytes > SURVEY_ASSISTANT_MAX_TOTAL_ATTACHMENT_BYTES) {\n          throw new SurveyAssistantRequestError(\n            \"The attachments stored for this chat are too large in total.\",\n          );\n        }\n\n        return transaction.one(\n          `INSERT INTO ${tableName}\n            (id, resource_id, thread_id, name, media_type, byte_size, content_hash, data)\n           VALUES\n            ($1, $2, $3, $4, $5, $6, $7, decode($8, 'base64'))\n           RETURNING id, name, media_type, byte_size, content_hash, created_at`,\n          [randomUUID(), resourceId, threadId, name, mediaType, bytes, contentHash, data],\n        );\n      });\n      return toThreadDocument(row);\n    },\n    async list({ resourceId, threadId }) {\n      const storage = await getStorage();\n      const rows = await storage.db.manyOrNone(\n        `SELECT id, name, media_type, byte_size, content_hash, created_at\n         FROM ${tableName}\n         WHERE resource_id = $1 AND thread_id = $2\n         ORDER BY created_at ASC`,\n        [resourceId, threadId],\n      );\n      return rows.map(toThreadDocument);\n    },\n    async read({ documentId, resourceId, threadId }) {\n      const storage = await getStorage();\n      const row = await storage.db.oneOrNone(\n        `SELECT id, name, media_type, byte_size, content_hash, created_at,\n                encode(data, 'base64') AS data\n         FROM ${tableName}\n         WHERE id = $1 AND resource_id = $2 AND thread_id = $3`,\n        [documentId, resourceId, threadId],\n      );\n      return row\n        ? {\n            ...toThreadDocument(row),\n            // PostgreSQL wraps encoded BYTEA output at 76 characters.\n            data: String(row.data).replace(/\\s+/g, \"\"),\n          }\n        : null;\n    },\n    async deleteThread({ resourceId, threadId }) {\n      const storage = await getStorage();\n      await storage.db.none(`DELETE FROM ${tableName} WHERE resource_id = $1 AND thread_id = $2`, [\n        resourceId,\n        threadId,\n      ]);\n    },\n  };\n};\n\nexport interface CreateSurveyAssistantPostgresRuntimeOptions {\n  ensureObservationalMemory?: boolean;\n  schemaName?: string;\n  store: PostgresStore | PostgresStoreConfig;\n}\n\nexport const createSurveyAssistantPostgresRuntime = ({\n  ensureObservationalMemory,\n  schemaName = \"public\",\n  store: storeOrConfig,\n}: CreateSurveyAssistantPostgresRuntimeOptions) => {\n  const storage =\n    storeOrConfig instanceof PostgresStore ? storeOrConfig : new PostgresStore(storeOrConfig);\n  let initialization: Promise<PostgresStore> | undefined;\n  const getStorage = () => {\n    initialization ??= initializeSurveyAssistantPostgresStorage(storage, {\n      ensureObservationalMemory,\n      schemaName,\n    });\n    return initialization;\n  };\n\n  return {\n    storage,\n    getStorage,\n    threadDocuments: createSurveyAssistantPostgresThreadDocumentRepository({\n      getStorage,\n      schemaName,\n    }),\n  };\n};\n"],"mappings":";;;;;AAaA,MAAM,6BAA6B;AACnC,MAAM,yBAAyB;AAE/B,MAAM,mBAAmB,UAAkB,IAAI,MAAM,WAAW,MAAK,OAAK,CAAC;AAE3E,MAAM,gBAAgB,WAA0B;AAC9C,SAAQ,OAAO,MAAf;EACE,KAAK,OACH,QAAO;EACT,KAAK,YACH,QAAO;EACT,KAAK,QACH,QAAO;EACT,KAAK,UACH,QAAO;EACT,KAAK,QACH,QAAO;EACT,KAAK,SACH,QAAO;EACT,KAAK,UACH,QAAO;EACT,KAAK,OACH,QAAO;;;AAIb,MAAM,oBAAoB,WAA0B;AAClD,SAAQ,OAAO,MAAf;EACE,KAAK,UACH,QAAO;EACT,KAAK;EACL,KAAK;EACL,KAAK,QACH,QAAO;EACT,KAAK,QACH,QAAO;EACT,KAAK,YACH,QAAO;EACT,KAAK,OACH,QAAO;EACT,KAAK,OACH,QAAO;;;AAIb,MAAM,aAAa,MAAc,QAAuB,0BAA0B,UAAU;CAC1F,MAAM,cAAc;EAClB,OAAO,aAAa,gBAAgB,KAAA;EACpC,OAAO,aAAa,QAAQ,aAAa,KAAA;EACzC,2BAA2B,OAAO,aAAa,QAC3C,WAAW,iBAAiB,OAAO,KACnC,KAAA;EACL,CACE,OAAO,QAAQ,CACf,KAAK,IAAI;AACZ,QAAO,GAAG,gBAAgB,KAAK,CAAC,GAAG,aAAa,OAAO,GAAG,cAAc,IAAI,gBAAgB;;AAG9F,MAAM,kCAAkC,OAAO,SAAwB,eAAuB;CAC5F,MAAM,cAAc,kCAAkC;AACtD,KAAI,CAAC,YAAa,OAAM,IAAI,MAAM,qDAAqD;CAEvF,MAAM,YAAY,GAAG,gBAAgB,WAAW,CAAC,GAAG,gBAAgB,2BAA2B;CAC/F,MAAM,UAAU,OAAO,QAAQ,YAAY,CAAC,SAAS,CAAC,MAAM,YAAY,CACtE,UAAU,MAAM,OAAO,EACvB,GAAI,OAAO,SAAS,cAChB,CAAC,GAAG,gBAAgB,GAAG,KAAK,GAAG,CAAC,4BAA4B,GAC5D,EAAE,CACP,CAAC;AAEF,OAAM,QAAQ,GAAG,KAAK,8BAA8B,UAAU,IAAI,QAAQ,KAAK,KAAK,CAAC,GAAG;AACxF,MAAK,MAAM,CAAC,MAAM,WAAW,OAAO,QAAQ,YAAY,EAAE;AACxD,QAAM,QAAQ,GAAG,KACf,eAAe,UAAU,4BAA4B,UAAU,MAAM,QAAQ,KAAK,GACnF;AACD,MAAI,OAAO,SAAS,YAClB,OAAM,QAAQ,GAAG,KACf,eAAe,UAAU,4BAA4B,gBAAgB,GAAG,KAAK,GAAG,CAAC,4BAClF;;AAGL,OAAM,QAAQ,GAAG,KACf,mDAAmD,UAAU,IAAI,gBAAgB,YAAY,CAAC,GAC/F;;AAGH,MAAM,8BAA8B,OAAO,SAAwB,eAAuB;CACxF,MAAM,YAAY,GAAG,gBAAgB,WAAW,CAAC,GAAG,gBAAgB,uBAAuB;AAC3F,OAAM,QAAQ,GAAG,KAAK;iCACS,UAAU;;;;;;;;;;;;IAYvC;AACF,OAAM,QAAQ,GAAG,KACf,8EAA8E,UAAU,uCACzF;;AAQH,MAAa,2CAA2C,OACtD,SACA,EACE,4BAA4B,MAC5B,aAAa,aACsC,EAAE,KACpD;AACH,OAAM,QAAQ,MAAM;AACpB,KAAI,0BAA2B,OAAM,gCAAgC,SAAS,WAAW;AACzF,OAAM,4BAA4B,SAAS,WAAW;AACtD,QAAO;;AAGT,MAAM,oBAAoB,SAOY;CACpC,IAAI,IAAI;CACR,MAAM,IAAI;CACV,WAAW,IAAI;CACf,OAAO,IAAI;CACX,aAAa,IAAI;CACjB,WAAW,IAAI,KAAK,IAAI,WAAW,CAAC,aAAa;CAClD;AAOD,MAAa,yDAAyD,EACpE,YACA,aAAa,eAC8F;CAC3G,MAAM,YAAY,GAAG,gBAAgB,WAAW,CAAC,GAAG,gBAAgB,uBAAuB;AAC3F,QAAO;EACL,MAAM,IAAI,EAAE,OAAO,MAAM,WAAW,MAAM,YAAY,YAAY;GAChE,MAAM,UAAU,MAAM,YAAY;GAClC,MAAM,cAAc,WAAW,SAAS,CAAC,OAAO,KAAK,CAAC,OAAO,MAAM;AAsCnE,UAAO,iBAAiB,MArCN,QAAQ,GAAG,GAAG,OAAO,gBAAgB;AACrD,UAAM,YAAY,KAAK,4DAA4D,CACjF,YACA,SACD,CAAC;IACF,MAAM,WAAW,MAAM,YAAY,UACjC,yEAAyE,UAAU;6EAEnF;KAAC;KAAY;KAAU;KAAY,CACpC;AACD,QAAI,SAAU,QAAO;IAErB,MAAM,QAAQ,MAAM,YAAY,IAC9B,uEAAuE,UAAU;uDAEjF,CAAC,YAAY,SAAS,CACvB;AACD,QAAI,OAAO,MAAM,MAAM,IAAA,EACrB,OAAM,IAAI,4BACR,0CACD;AAEH,QAAI,OAAO,MAAM,MAAM,GAAG,QAAA,SACxB,OAAM,IAAI,4BACR,+DACD;AAGH,WAAO,YAAY,IACjB,eAAe,UAAU;;;;iFAKzB;KAAC,YAAY;KAAE;KAAY;KAAU;KAAM;KAAW;KAAO;KAAa;KAAK,CAChF;KACD,CAC0B;;EAE9B,MAAM,KAAK,EAAE,YAAY,YAAY;AASnC,WAAO,OAPY,MADG,YAAY,EACP,GAAG,WAC5B;gBACQ,UAAU;;mCAGlB,CAAC,YAAY,SAAS,CACvB,EACW,IAAI,iBAAiB;;EAEnC,MAAM,KAAK,EAAE,YAAY,YAAY,YAAY;GAE/C,MAAM,MAAM,OAAM,MADI,YAAY,EACR,GAAG,UAC3B;;gBAEQ,UAAU;iEAElB;IAAC;IAAY;IAAY;IAAS,CACnC;AACD,UAAO,MACH;IACE,GAAG,iBAAiB,IAAI;IAExB,MAAM,OAAO,IAAI,KAAK,CAAC,QAAQ,QAAQ,GAAG;IAC3C,GACD;;EAEN,MAAM,aAAa,EAAE,YAAY,YAAY;AAE3C,UAAM,MADgB,YAAY,EACpB,GAAG,KAAK,eAAe,UAAU,6CAA6C,CAC1F,YACA,SACD,CAAC;;EAEL;;AASH,MAAa,wCAAwC,EACnD,2BACA,aAAa,UACb,OAAO,oBAC0C;CACjD,MAAM,UACJ,yBAAyB,gBAAgB,gBAAgB,IAAI,cAAc,cAAc;CAC3F,IAAI;CACJ,MAAM,mBAAmB;AACvB,qBAAmB,yCAAyC,SAAS;GACnE;GACA;GACD,CAAC;AACF,SAAO;;AAGT,QAAO;EACL;EACA;EACA,iBAAiB,sDAAsD;GACrE;GACA;GACD,CAAC;EACH"}
+{"version":3,"file":"storage-postgres.mjs","names":[],"sources":["../src/storage/postgres.ts"],"sourcesContent":["import { OBSERVATIONAL_MEMORY_TABLE_SCHEMA, type StorageColumn } from \"@mastra/core/storage\";\nimport { PostgresStore, type PostgresStoreConfig } from \"@mastra/pg\";\nimport { createHash, randomUUID } from \"node:crypto\";\nimport { SurveyAssistantRequestError } from \"../server/request-guardrails\";\nimport type {\n  SurveyAssistantThreadDocument,\n  SurveyAssistantThreadDocumentRepository,\n} from \"../server/thread-documents\";\n\nconst OBSERVATIONAL_MEMORY_TABLE = \"mastra_observational_memory\";\nconst THREAD_DOCUMENTS_TABLE = \"survey_assistant_thread_documents\";\nexport const SURVEY_ASSISTANT_DEFAULT_MAX_STORED_ATTACHMENTS_PER_THREAD = 100;\nexport const SURVEY_ASSISTANT_DEFAULT_MAX_STORED_ATTACHMENT_BYTES_PER_THREAD = 200 * 1024 * 1024;\n\nconst quoteIdentifier = (value: string) => `\"${value.replaceAll('\"', '\"\"')}\"`;\n\nconst postgresType = (column: StorageColumn) => {\n  switch (column.type) {\n    case \"text\":\n      return \"TEXT\";\n    case \"timestamp\":\n      return \"TIMESTAMP\";\n    case \"jsonb\":\n      return \"JSONB\";\n    case \"integer\":\n      return \"INTEGER\";\n    case \"float\":\n      return \"DOUBLE PRECISION\";\n    case \"bigint\":\n      return \"BIGINT\";\n    case \"boolean\":\n      return \"BOOLEAN\";\n    case \"uuid\":\n      return \"UUID\";\n  }\n};\n\nconst migrationDefault = (column: StorageColumn) => {\n  switch (column.type) {\n    case \"boolean\":\n      return \"FALSE\";\n    case \"integer\":\n    case \"bigint\":\n    case \"float\":\n      return \"0\";\n    case \"jsonb\":\n      return \"'{}'::jsonb\";\n    case \"timestamp\":\n      return \"NOW()\";\n    case \"text\":\n      return \"''\";\n    case \"uuid\":\n      return \"gen_random_uuid()\";\n  }\n};\n\nconst columnSql = (name: string, column: StorageColumn, includeMigrationDefault = false) => {\n  const constraints = [\n    column.primaryKey ? \"PRIMARY KEY\" : undefined,\n    column.nullable === false ? \"NOT NULL\" : undefined,\n    includeMigrationDefault && column.nullable === false\n      ? `DEFAULT ${migrationDefault(column)}`\n      : undefined,\n  ]\n    .filter(Boolean)\n    .join(\" \");\n  return `${quoteIdentifier(name)} ${postgresType(column)}${constraints ? ` ${constraints}` : \"\"}`;\n};\n\nconst ensureObservationalMemorySchema = async (storage: PostgresStore, schemaName: string) => {\n  const tableSchema = OBSERVATIONAL_MEMORY_TABLE_SCHEMA[OBSERVATIONAL_MEMORY_TABLE];\n  if (!tableSchema) throw new Error(\"Mastra observational memory schema is unavailable.\");\n\n  const tableName = `${quoteIdentifier(schemaName)}.${quoteIdentifier(OBSERVATIONAL_MEMORY_TABLE)}`;\n  const columns = Object.entries(tableSchema).flatMap(([name, column]) => [\n    columnSql(name, column),\n    ...(column.type === \"timestamp\"\n      ? [`${quoteIdentifier(`${name}Z`)} TIMESTAMPTZ DEFAULT NOW()`]\n      : []),\n  ]);\n\n  await storage.db.none(`CREATE TABLE IF NOT EXISTS ${tableName} (${columns.join(\", \")})`);\n  for (const [name, column] of Object.entries(tableSchema)) {\n    await storage.db.none(\n      `ALTER TABLE ${tableName} ADD COLUMN IF NOT EXISTS ${columnSql(name, column, true)}`,\n    );\n    if (column.type === \"timestamp\") {\n      await storage.db.none(\n        `ALTER TABLE ${tableName} ADD COLUMN IF NOT EXISTS ${quoteIdentifier(`${name}Z`)} TIMESTAMPTZ DEFAULT NOW()`,\n      );\n    }\n  }\n  await storage.db.none(\n    `CREATE INDEX IF NOT EXISTS idx_om_lookup_key ON ${tableName} (${quoteIdentifier(\"lookupKey\")})`,\n  );\n};\n\nconst ensureThreadDocumentsSchema = async (storage: PostgresStore, schemaName: string) => {\n  const tableName = `${quoteIdentifier(schemaName)}.${quoteIdentifier(THREAD_DOCUMENTS_TABLE)}`;\n  await storage.db.none(`\n    CREATE TABLE IF NOT EXISTS ${tableName} (\n      id TEXT PRIMARY KEY,\n      resource_id TEXT NOT NULL,\n      thread_id TEXT NOT NULL,\n      name TEXT NOT NULL,\n      media_type TEXT NOT NULL,\n      byte_size INTEGER NOT NULL,\n      content_hash TEXT NOT NULL,\n      data BYTEA NOT NULL,\n      created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),\n      UNIQUE (resource_id, thread_id, content_hash)\n    )\n  `);\n  await storage.db.none(\n    `CREATE INDEX IF NOT EXISTS idx_survey_assistant_thread_documents_thread ON ${tableName} (resource_id, thread_id, created_at)`,\n  );\n};\n\nexport interface InitializeSurveyAssistantPostgresStorageOptions {\n  ensureObservationalMemory?: boolean;\n  schemaName?: string;\n}\n\nexport const initializeSurveyAssistantPostgresStorage = async (\n  storage: PostgresStore,\n  {\n    ensureObservationalMemory = true,\n    schemaName = \"public\",\n  }: InitializeSurveyAssistantPostgresStorageOptions = {},\n) => {\n  await storage.init();\n  if (ensureObservationalMemory) await ensureObservationalMemorySchema(storage, schemaName);\n  await ensureThreadDocumentsSchema(storage, schemaName);\n  return storage;\n};\n\nconst toThreadDocument = (row: {\n  id: string;\n  name: string;\n  media_type: string;\n  byte_size: number;\n  content_hash: string;\n  created_at: Date | string;\n}): SurveyAssistantThreadDocument => ({\n  id: row.id,\n  name: row.name,\n  mediaType: row.media_type,\n  bytes: row.byte_size,\n  contentHash: row.content_hash,\n  createdAt: new Date(row.created_at).toISOString(),\n});\n\nexport interface CreateSurveyAssistantPostgresThreadDocumentRepositoryOptions {\n  getStorage(): PostgresStore | Promise<PostgresStore>;\n  maxStoredAttachmentsPerThread?: number;\n  maxStoredBytesPerThread?: number;\n  schemaName?: string;\n}\n\nexport const createSurveyAssistantPostgresThreadDocumentRepository = ({\n  getStorage,\n  maxStoredAttachmentsPerThread = SURVEY_ASSISTANT_DEFAULT_MAX_STORED_ATTACHMENTS_PER_THREAD,\n  maxStoredBytesPerThread = SURVEY_ASSISTANT_DEFAULT_MAX_STORED_ATTACHMENT_BYTES_PER_THREAD,\n  schemaName = \"public\",\n}: CreateSurveyAssistantPostgresThreadDocumentRepositoryOptions): SurveyAssistantThreadDocumentRepository => {\n  if (\n    !Number.isSafeInteger(maxStoredAttachmentsPerThread) ||\n    maxStoredAttachmentsPerThread <= 0 ||\n    !Number.isSafeInteger(maxStoredBytesPerThread) ||\n    maxStoredBytesPerThread <= 0\n  ) {\n    throw new Error(\"Survey assistant stored attachment limits must be positive integers.\");\n  }\n  const tableName = `${quoteIdentifier(schemaName)}.${quoteIdentifier(THREAD_DOCUMENTS_TABLE)}`;\n  return {\n    async put({ bytes, data, mediaType, name, resourceId, threadId }) {\n      const storage = await getStorage();\n      const contentHash = createHash(\"sha256\").update(data).digest(\"hex\");\n      const row = await storage.db.tx(async (transaction) => {\n        await transaction.none(\"SELECT pg_advisory_xact_lock(hashtext($1), hashtext($2))\", [\n          resourceId,\n          threadId,\n        ]);\n        const existing = await transaction.oneOrNone(\n          `SELECT id, name, media_type, byte_size, content_hash, created_at FROM ${tableName}\n           WHERE resource_id = $1 AND thread_id = $2 AND content_hash = $3`,\n          [resourceId, threadId, contentHash],\n        );\n        if (existing) return existing;\n\n        const usage = await transaction.one<{ count: string; bytes: string }>(\n          `SELECT COUNT(*) AS count, COALESCE(SUM(byte_size), 0) AS bytes FROM ${tableName}\n           WHERE resource_id = $1 AND thread_id = $2`,\n          [resourceId, threadId],\n        );\n        if (Number(usage.count) >= maxStoredAttachmentsPerThread) {\n          throw new SurveyAssistantRequestError(\n            `A chat can contain up to ${maxStoredAttachmentsPerThread} stored attachments.`,\n          );\n        }\n        if (Number(usage.bytes) + bytes > maxStoredBytesPerThread) {\n          throw new SurveyAssistantRequestError(\n            \"The attachments stored for this chat are too large in total.\",\n          );\n        }\n\n        return transaction.one(\n          `INSERT INTO ${tableName}\n            (id, resource_id, thread_id, name, media_type, byte_size, content_hash, data)\n           VALUES\n            ($1, $2, $3, $4, $5, $6, $7, decode($8, 'base64'))\n           RETURNING id, name, media_type, byte_size, content_hash, created_at`,\n          [randomUUID(), resourceId, threadId, name, mediaType, bytes, contentHash, data],\n        );\n      });\n      return toThreadDocument(row);\n    },\n    async list({ resourceId, threadId }) {\n      const storage = await getStorage();\n      const rows = await storage.db.manyOrNone(\n        `SELECT id, name, media_type, byte_size, content_hash, created_at\n         FROM ${tableName}\n         WHERE resource_id = $1 AND thread_id = $2\n         ORDER BY created_at ASC`,\n        [resourceId, threadId],\n      );\n      return rows.map(toThreadDocument);\n    },\n    async read({ documentId, resourceId, threadId }) {\n      const storage = await getStorage();\n      const row = await storage.db.oneOrNone(\n        `SELECT id, name, media_type, byte_size, content_hash, created_at,\n                encode(data, 'base64') AS data\n         FROM ${tableName}\n         WHERE id = $1 AND resource_id = $2 AND thread_id = $3`,\n        [documentId, resourceId, threadId],\n      );\n      return row\n        ? {\n            ...toThreadDocument(row),\n            // PostgreSQL wraps encoded BYTEA output at 76 characters.\n            data: String(row.data).replace(/\\s+/g, \"\"),\n          }\n        : null;\n    },\n    async deleteThread({ resourceId, threadId }) {\n      const storage = await getStorage();\n      await storage.db.none(`DELETE FROM ${tableName} WHERE resource_id = $1 AND thread_id = $2`, [\n        resourceId,\n        threadId,\n      ]);\n    },\n  };\n};\n\nexport interface CreateSurveyAssistantPostgresRuntimeOptions {\n  ensureObservationalMemory?: boolean;\n  maxStoredAttachmentsPerThread?: number;\n  maxStoredBytesPerThread?: number;\n  schemaName?: string;\n  store: PostgresStore | PostgresStoreConfig;\n}\n\nexport const createSurveyAssistantPostgresRuntime = ({\n  ensureObservationalMemory,\n  maxStoredAttachmentsPerThread,\n  maxStoredBytesPerThread,\n  schemaName = \"public\",\n  store: storeOrConfig,\n}: CreateSurveyAssistantPostgresRuntimeOptions) => {\n  const storage =\n    storeOrConfig instanceof PostgresStore ? storeOrConfig : new PostgresStore(storeOrConfig);\n  let initialization: Promise<PostgresStore> | undefined;\n  const getStorage = () => {\n    initialization ??= initializeSurveyAssistantPostgresStorage(storage, {\n      ensureObservationalMemory,\n      schemaName,\n    });\n    return initialization;\n  };\n\n  return {\n    storage,\n    getStorage,\n    threadDocuments: createSurveyAssistantPostgresThreadDocumentRepository({\n      getStorage,\n      maxStoredAttachmentsPerThread,\n      maxStoredBytesPerThread,\n      schemaName,\n    }),\n  };\n};\n"],"mappings":";;;;;AASA,MAAM,6BAA6B;AACnC,MAAM,yBAAyB;AAC/B,MAAa,6DAA6D;AAC1E,MAAa,kEAAkE,MAAM,OAAO;AAE5F,MAAM,mBAAmB,UAAkB,IAAI,MAAM,WAAW,MAAK,OAAK,CAAC;AAE3E,MAAM,gBAAgB,WAA0B;AAC9C,SAAQ,OAAO,MAAf;EACE,KAAK,OACH,QAAO;EACT,KAAK,YACH,QAAO;EACT,KAAK,QACH,QAAO;EACT,KAAK,UACH,QAAO;EACT,KAAK,QACH,QAAO;EACT,KAAK,SACH,QAAO;EACT,KAAK,UACH,QAAO;EACT,KAAK,OACH,QAAO;;;AAIb,MAAM,oBAAoB,WAA0B;AAClD,SAAQ,OAAO,MAAf;EACE,KAAK,UACH,QAAO;EACT,KAAK;EACL,KAAK;EACL,KAAK,QACH,QAAO;EACT,KAAK,QACH,QAAO;EACT,KAAK,YACH,QAAO;EACT,KAAK,OACH,QAAO;EACT,KAAK,OACH,QAAO;;;AAIb,MAAM,aAAa,MAAc,QAAuB,0BAA0B,UAAU;CAC1F,MAAM,cAAc;EAClB,OAAO,aAAa,gBAAgB,KAAA;EACpC,OAAO,aAAa,QAAQ,aAAa,KAAA;EACzC,2BAA2B,OAAO,aAAa,QAC3C,WAAW,iBAAiB,OAAO,KACnC,KAAA;EACL,CACE,OAAO,QAAQ,CACf,KAAK,IAAI;AACZ,QAAO,GAAG,gBAAgB,KAAK,CAAC,GAAG,aAAa,OAAO,GAAG,cAAc,IAAI,gBAAgB;;AAG9F,MAAM,kCAAkC,OAAO,SAAwB,eAAuB;CAC5F,MAAM,cAAc,kCAAkC;AACtD,KAAI,CAAC,YAAa,OAAM,IAAI,MAAM,qDAAqD;CAEvF,MAAM,YAAY,GAAG,gBAAgB,WAAW,CAAC,GAAG,gBAAgB,2BAA2B;CAC/F,MAAM,UAAU,OAAO,QAAQ,YAAY,CAAC,SAAS,CAAC,MAAM,YAAY,CACtE,UAAU,MAAM,OAAO,EACvB,GAAI,OAAO,SAAS,cAChB,CAAC,GAAG,gBAAgB,GAAG,KAAK,GAAG,CAAC,4BAA4B,GAC5D,EAAE,CACP,CAAC;AAEF,OAAM,QAAQ,GAAG,KAAK,8BAA8B,UAAU,IAAI,QAAQ,KAAK,KAAK,CAAC,GAAG;AACxF,MAAK,MAAM,CAAC,MAAM,WAAW,OAAO,QAAQ,YAAY,EAAE;AACxD,QAAM,QAAQ,GAAG,KACf,eAAe,UAAU,4BAA4B,UAAU,MAAM,QAAQ,KAAK,GACnF;AACD,MAAI,OAAO,SAAS,YAClB,OAAM,QAAQ,GAAG,KACf,eAAe,UAAU,4BAA4B,gBAAgB,GAAG,KAAK,GAAG,CAAC,4BAClF;;AAGL,OAAM,QAAQ,GAAG,KACf,mDAAmD,UAAU,IAAI,gBAAgB,YAAY,CAAC,GAC/F;;AAGH,MAAM,8BAA8B,OAAO,SAAwB,eAAuB;CACxF,MAAM,YAAY,GAAG,gBAAgB,WAAW,CAAC,GAAG,gBAAgB,uBAAuB;AAC3F,OAAM,QAAQ,GAAG,KAAK;iCACS,UAAU;;;;;;;;;;;;IAYvC;AACF,OAAM,QAAQ,GAAG,KACf,8EAA8E,UAAU,uCACzF;;AAQH,MAAa,2CAA2C,OACtD,SACA,EACE,4BAA4B,MAC5B,aAAa,aACsC,EAAE,KACpD;AACH,OAAM,QAAQ,MAAM;AACpB,KAAI,0BAA2B,OAAM,gCAAgC,SAAS,WAAW;AACzF,OAAM,4BAA4B,SAAS,WAAW;AACtD,QAAO;;AAGT,MAAM,oBAAoB,SAOY;CACpC,IAAI,IAAI;CACR,MAAM,IAAI;CACV,WAAW,IAAI;CACf,OAAO,IAAI;CACX,aAAa,IAAI;CACjB,WAAW,IAAI,KAAK,IAAI,WAAW,CAAC,aAAa;CAClD;AASD,MAAa,yDAAyD,EACpE,YACA,gCAAA,KACA,0BAA0B,iEAC1B,aAAa,eAC8F;AAC3G,KACE,CAAC,OAAO,cAAc,8BAA8B,IACpD,iCAAiC,KACjC,CAAC,OAAO,cAAc,wBAAwB,IAC9C,2BAA2B,EAE3B,OAAM,IAAI,MAAM,uEAAuE;CAEzF,MAAM,YAAY,GAAG,gBAAgB,WAAW,CAAC,GAAG,gBAAgB,uBAAuB;AAC3F,QAAO;EACL,MAAM,IAAI,EAAE,OAAO,MAAM,WAAW,MAAM,YAAY,YAAY;GAChE,MAAM,UAAU,MAAM,YAAY;GAClC,MAAM,cAAc,WAAW,SAAS,CAAC,OAAO,KAAK,CAAC,OAAO,MAAM;AAsCnE,UAAO,iBAAiB,MArCN,QAAQ,GAAG,GAAG,OAAO,gBAAgB;AACrD,UAAM,YAAY,KAAK,4DAA4D,CACjF,YACA,SACD,CAAC;IACF,MAAM,WAAW,MAAM,YAAY,UACjC,yEAAyE,UAAU;6EAEnF;KAAC;KAAY;KAAU;KAAY,CACpC;AACD,QAAI,SAAU,QAAO;IAErB,MAAM,QAAQ,MAAM,YAAY,IAC9B,uEAAuE,UAAU;uDAEjF,CAAC,YAAY,SAAS,CACvB;AACD,QAAI,OAAO,MAAM,MAAM,IAAI,8BACzB,OAAM,IAAI,4BACR,4BAA4B,8BAA8B,sBAC3D;AAEH,QAAI,OAAO,MAAM,MAAM,GAAG,QAAQ,wBAChC,OAAM,IAAI,4BACR,+DACD;AAGH,WAAO,YAAY,IACjB,eAAe,UAAU;;;;iFAKzB;KAAC,YAAY;KAAE;KAAY;KAAU;KAAM;KAAW;KAAO;KAAa;KAAK,CAChF;KACD,CAC0B;;EAE9B,MAAM,KAAK,EAAE,YAAY,YAAY;AASnC,WAAO,OAPY,MADG,YAAY,EACP,GAAG,WAC5B;gBACQ,UAAU;;mCAGlB,CAAC,YAAY,SAAS,CACvB,EACW,IAAI,iBAAiB;;EAEnC,MAAM,KAAK,EAAE,YAAY,YAAY,YAAY;GAE/C,MAAM,MAAM,OAAM,MADI,YAAY,EACR,GAAG,UAC3B;;gBAEQ,UAAU;iEAElB;IAAC;IAAY;IAAY;IAAS,CACnC;AACD,UAAO,MACH;IACE,GAAG,iBAAiB,IAAI;IAExB,MAAM,OAAO,IAAI,KAAK,CAAC,QAAQ,QAAQ,GAAG;IAC3C,GACD;;EAEN,MAAM,aAAa,EAAE,YAAY,YAAY;AAE3C,UAAM,MADgB,YAAY,EACpB,GAAG,KAAK,eAAe,UAAU,6CAA6C,CAC1F,YACA,SACD,CAAC;;EAEL;;AAWH,MAAa,wCAAwC,EACnD,2BACA,+BACA,yBACA,aAAa,UACb,OAAO,oBAC0C;CACjD,MAAM,UACJ,yBAAyB,gBAAgB,gBAAgB,IAAI,cAAc,cAAc;CAC3F,IAAI;CACJ,MAAM,mBAAmB;AACvB,qBAAmB,yCAAyC,SAAS;GACnE;GACA;GACD,CAAC;AACF,SAAO;;AAGT,QAAO;EACL;EACA;EACA,iBAAiB,sDAAsD;GACrE;GACA;GACA;GACA;GACD,CAAC;EACH"}

package/dist/tasks-COQnhSeT.d.mts

@@ -0,0 +1,140 @@
+import { A as SurveyAssistantSuggestItemKeyResult, B as SurveyAssistantTranslateContentResult, F as SurveyAssistantTaskInvocationMode, M as SurveyAssistantSuggestItemLabelResult, P as SurveyAssistantTaskContext, j as SurveyAssistantSuggestItemLabelInput, k as SurveyAssistantSuggestItemKeyInput, z as SurveyAssistantTranslateContentInput } from "./index-DVxVyhAu.mjs";
+import { a as SurveyAssistantModelSelection, n as SURVEY_ASSISTANT_TASK_AGENT_ID, o as SurveyAssistantRequestContext } from "./constants-56jf_4ls.mjs";
+import { z } from "zod";
+import { RequestContext } from "@mastra/core/request-context";
+import * as _$_mastra_core_agent0 from "@mastra/core/agent";
+import { Agent, AgentConfig, AgentExecutionOptions } from "@mastra/core/agent";
+
+//#region src/server/authoring-references.d.ts
+declare const surveyAuthoringReferenceTopics: readonly ["core-rules", "survey-data-model", "source-material-surveys", "localization", "rich-text-content", "item-types", "embedded-forms", "expressions", "assistant-operations", "survey-quality"];
+type SurveyAuthoringReferenceTopic = (typeof surveyAuthoringReferenceTopics)[number];
+declare const surveyAuthoringReferences: {
+  readonly "core-rules": {
+    readonly title: "Core authoring rules";
+    readonly content: string;
+  };
+  readonly "survey-data-model": {
+    readonly title: "Survey data model";
+    readonly content: string;
+  };
+  readonly "source-material-surveys": {
+    readonly title: "Source-material survey implementation";
+    readonly content: string;
+  };
+  readonly localization: {
+    readonly title: "Survey localization";
+    readonly content: string;
+  };
+  readonly "rich-text-content": {
+    readonly title: "Rich text content";
+    readonly content: string;
+  };
+  readonly "item-types": {
+    readonly title: "Item types";
+    readonly content: string;
+  };
+  readonly "embedded-forms": {
+    readonly title: "Choice option embedded forms";
+    readonly content: string;
+  };
+  readonly expressions: {
+    readonly title: "Expressions";
+    readonly content: string;
+  };
+  readonly "assistant-operations": {
+    readonly title: "Assistant operations";
+    readonly content: string;
+  };
+  readonly "survey-quality": {
+    readonly title: "Survey quality review";
+    readonly content: string;
+  };
+};
+//#endregion
+//#region src/server/tasks.d.ts
+type SurveyAssistantTaskAgentConfig = AgentConfig<typeof SURVEY_ASSISTANT_TASK_AGENT_ID, Record<string, never>, undefined, SurveyAssistantRequestContext>;
+interface CreateSurveyAssistantTaskAgentOptions {
+  model: SurveyAssistantTaskAgentConfig["model"];
+}
+declare const createSurveyAssistantTaskAgent: ({
+  model
+}: CreateSurveyAssistantTaskAgentOptions) => Agent<"survey-assistant-task-agent", Record<string, never>, undefined, SurveyAssistantRequestContext, _$_mastra_core_agent0.AgentEditorConfig | undefined>;
+interface RunSurveyAssistantTaskOptions<TSchema extends z.ZodType> {
+  agent: Pick<Agent, "generate">;
+  taskId: string;
+  input: unknown;
+  context: SurveyAssistantTaskContext;
+  schema: TSchema;
+  instructions: readonly string[];
+  referenceTopics?: readonly SurveyAuthoringReferenceTopic[];
+  requestContext?: RequestContext<SurveyAssistantRequestContext>;
+  providerOptions?: AgentExecutionOptions["providerOptions"];
+  abortSignal?: AbortSignal;
+}
+declare const runSurveyAssistantTask: <TSchema extends z.ZodType>({
+  agent,
+  taskId,
+  input,
+  context,
+  schema,
+  instructions,
+  referenceTopics,
+  requestContext,
+  providerOptions,
+  abortSignal
+}: RunSurveyAssistantTaskOptions<TSchema>) => Promise<z.output<TSchema>>;
+interface RunSurveyAssistantBuiltInTaskOptions {
+  agent: Pick<Agent, "generate">;
+  context: SurveyAssistantTaskContext;
+  mode?: SurveyAssistantTaskInvocationMode;
+  requestContext?: RequestContext<SurveyAssistantRequestContext>;
+  providerOptions?: AgentExecutionOptions["providerOptions"];
+  abortSignal?: AbortSignal;
+}
+declare const runSurveyAssistantSuggestItemKeyTask: (input: SurveyAssistantSuggestItemKeyInput, options: RunSurveyAssistantBuiltInTaskOptions) => Promise<SurveyAssistantSuggestItemKeyResult>;
+declare const runSurveyAssistantSuggestItemLabelTask: (input: SurveyAssistantSuggestItemLabelInput, options: RunSurveyAssistantBuiltInTaskOptions) => Promise<SurveyAssistantSuggestItemLabelResult>;
+declare const runSurveyAssistantTranslateContentTask: (input: SurveyAssistantTranslateContentInput, options: RunSurveyAssistantBuiltInTaskOptions) => Promise<SurveyAssistantTranslateContentResult>;
+interface SurveyAssistantTaskAuthorizedIdentity {
+  resourceId: string;
+  surveyId?: string;
+}
+interface SurveyAssistantTaskAuthorizationInput {
+  request: Request;
+  requestedResourceId?: string;
+  requestedSurveyId?: string;
+}
+interface SurveyAssistantTaskModelResolutionInput {
+  identity: SurveyAssistantTaskAuthorizedIdentity;
+  request: Request;
+  requestedModelId?: string;
+}
+interface CreateSurveyAssistantTaskHandlerOptions {
+  agent: Pick<Agent, "generate"> | (() => Pick<Agent, "generate"> | Promise<Pick<Agent, "generate">>);
+  authorize(input: SurveyAssistantTaskAuthorizationInput): SurveyAssistantTaskAuthorizedIdentity | Promise<SurveyAssistantTaskAuthorizedIdentity>;
+  getReadinessError?(request: Request): string | null | Promise<string | null>;
+  isEnabled?(request: Request): boolean | Promise<boolean>;
+  maxConcurrentRequests?: number;
+  maxDurationMs?: number;
+  onTaskMetrics?(input: {
+    durationMs: number;
+    identity: SurveyAssistantTaskAuthorizedIdentity;
+    status: "completed" | "failed";
+    taskId: string;
+  }): void;
+  providerOptions?: AgentExecutionOptions["providerOptions"];
+  resolveModel?(input: SurveyAssistantTaskModelResolutionInput): SurveyAssistantModelSelection | null | undefined | Promise<SurveyAssistantModelSelection | null | undefined>;
+}
+declare const createSurveyAssistantTaskHandler: ({
+  agent,
+  authorize,
+  getReadinessError,
+  isEnabled,
+  maxConcurrentRequests,
+  maxDurationMs,
+  onTaskMetrics,
+  providerOptions,
+  resolveModel
+}: CreateSurveyAssistantTaskHandlerOptions) => (request: Request) => Promise<Response>;
+//#endregion
+export { SurveyAssistantTaskAuthorizationInput as a, createSurveyAssistantTaskAgent as c, runSurveyAssistantSuggestItemLabelTask as d, runSurveyAssistantTask as f, surveyAuthoringReferences as g, surveyAuthoringReferenceTopics as h, RunSurveyAssistantTaskOptions as i, createSurveyAssistantTaskHandler as l, SurveyAuthoringReferenceTopic as m, CreateSurveyAssistantTaskHandlerOptions as n, SurveyAssistantTaskAuthorizedIdentity as o, runSurveyAssistantTranslateContentTask as p, RunSurveyAssistantBuiltInTaskOptions as r, SurveyAssistantTaskModelResolutionInput as s, CreateSurveyAssistantTaskAgentOptions as t, runSurveyAssistantSuggestItemKeyTask as u };
+//# sourceMappingURL=tasks-COQnhSeT.d.mts.map

package/dist/tasks-COQnhSeT.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tasks-COQnhSeT.d.mts","names":[],"sources":["../src/server/authoring-references.ts","../src/server/tasks.ts"],"mappings":";;;;;;;;cAMa,8BAAA;AAAA,KAaD,6BAAA,WAAwC,8BAAA;AAAA,cAyBvC,yBAAA;EAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KClBR,8BAAA,GAAiC,WAAA,QAC7B,8BAAA,EACP,MAAA,4BAEA,6BAAA;AAAA,UAGe,qCAAA;EACf,KAAA,EAAO,8BAAA;AAAA;AAAA,cAGI,8BAAA;EAAkC;AAAA,GAAW,qCAAA,KAAqC,KAAA,gCAAA,MAAA,4BAAA,6BAAA,EAAA,qBAAA,CAAA,iBAAA;AAAA,UAc9E,6BAAA,iBAA8C,CAAA,CAAE,OAAA;EAC/D,KAAA,EAAO,IAAA,CAAK,KAAA;EACZ,MAAA;EACA,KAAA;EACA,OAAA,EAAS,0BAAA;EACT,MAAA,EAAQ,OAAA;EACR,YAAA;EACA,eAAA,YAA2B,6BAAA;EAC3B,cAAA,GAAiB,cAAA,CAAe,6BAAA;EAChC,eAAA,GAAkB,qBAAA;EAClB,WAAA,GAAc,WAAA;AAAA;AAAA,cAUH,sBAAA,mBAAgD,CAAA,CAAE,OAAA;EAAS,KAAA;EAAA,MAAA;EAAA,KAAA;EAAA,OAAA;EAAA,MAAA;EAAA,YAAA;EAAA,eAAA;EAAA,cAAA;EAAA,eAAA;EAAA;AAAA,GAWrE,6BAAA,CAA8B,OAAA,MAAW,OAAA,CAAQ,CAAA,CAAE,MAAA,CAAO,OAAA;AAAA,UAiF5C,oCAAA;EACf,KAAA,EAAO,IAAA,CAAK,KAAA;EACZ,OAAA,EAAS,0BAAA;EACT,IAAA,GAAO,iCAAA;EACP,cAAA,GAAiB,cAAA,CAAe,6BAAA;EAChC,eAAA,GAAkB,qBAAA;EAClB,WAAA,GAAc,WAAA;AAAA;AAAA,cA8CH,oCAAA,GACX,KAAA,EAAO,kCAAA,EACP,OAAA,EAAS,oCAAA,KACR,OAAA,CAAQ,mCAAA;AAAA,cA6DE,sCAAA,GACX,KAAA,EAAO,oCAAA,EACP,OAAA,EAAS,oCAAA,KACR,OAAA,CAAQ,qCAAA;AAAA,cA6CE,sCAAA,GACX,KAAA,EAAO,oCAAA,EACP,OAAA,EAAS,oCAAA,KACR,OAAA,CAAQ,qCAAA;AAAA,UA8CM,qCAAA;EACf,UAAA;EACA,QAAA;AAAA;AAAA,UAGe,qCAAA;EACf,OAAA,EAAS,OAAA;EACT,mBAAA;EACA,iBAAA;AAAA;AAAA,UAGe,uCAAA;EACf,QAAA,EAAU,qCAAA;EACV,OAAA,EAAS,OAAA;EACT,gBAAA;AAAA;AAAA,UAGe,uCAAA;EACf,KAAA,EACI,IAAA,CAAK,KAAA,uBACE,IAAA,CAAK,KAAA,gBAAqB,OAAA,CAAQ,IAAA,CAAK,KAAA;EAClD,SAAA,CACE,KAAA,EAAO,qCAAA,GACN,qCAAA,GAAwC,OAAA,CAAQ,qCAAA;EACnD,iBAAA,EAAmB,OAAA,EAAS,OAAA,mBAA0B,OAAA;EACtD,SAAA,EAAW,OAAA,EAAS,OAAA,aAAoB,OAAA;EACxC,qBAAA;EACA,aAAA;EACA,aAAA,EAAe,KAAA;IACb,UAAA;IACA,QAAA,EAAU,qCAAA;IACV,MAAA;IACA,MAAA;EAAA;EAEF,eAAA,GAAkB,qBAAA;EAClB,YAAA,EACE,KAAA,EAAO,uCAAA,GAEL,6BAAA,sBAGA,OAAA,CAAQ,6BAAA;AAAA;AAAA,cAmCD,gCAAA;EAAoC,KAAA;EAAA,SAAA;EAAA,iBAAA;EAAA,SAAA;EAAA,qBAAA;EAAA,aAAA;EAAA,aAAA;EAAA,eAAA;EAAA;AAAA,GAU9C,uCAAA,MAWa,OAAA,EAAS,OAAA,KAAU,OAAA,CAAQ,QAAA"}

package/dist/{thread-documents-DamIq9Od.d.mts → thread-documents-BlxNjc7S.d.mts}

@@ -49,6 +49,18 @@ interface SurveyAssistantThreadDocumentRepository {
     threadId: string;
   }): Promise<void>;
 }
+/**
+ * Replaces current-turn file data with small repository references before the
+ * message is handed to Mastra for persistence. The provider-bound attachment
+ * processor re-injects the authorized repository copy transiently.
+ */
+declare const replaceCurrentSurveyAssistantAttachmentsWithReferences: ({
+  documents,
+  messages
+}: {
+  documents: SurveyAssistantThreadDocument[];
+  messages: UIMessage[];
+}) => UIMessage[];
 declare const registerSurveyAssistantThreadAttachments: ({
   messages,
   repository,
@@ -61,16 +73,20 @@ declare const registerSurveyAssistantThreadAttachments: ({
   threadId: string;
 }) => Promise<SurveyAssistantThreadDocument[]>;
 declare const removeHistoricalSurveyAssistantAttachments: (messages: UIMessage[]) => UIMessage<unknown, _$ai.UIDataTypes, _$ai.UITools>[];
-declare const getSurveyAssistantReplayableDocuments: (documents: SurveyAssistantThreadDocument[], messages: UIMessage[]) => SurveyAssistantThreadDocument[];
+/**
+ * Excludes the exact repository documents already loaded for the current turn.
+ * Message parts are references by this stage and are not a deduplication source.
+ */
+declare const getSurveyAssistantReplayableDocuments: (documents: SurveyAssistantThreadDocument[], currentDocumentIds: ReadonlySet<string>) => SurveyAssistantThreadDocument[];
 declare const loadSurveyAssistantThreadAttachments: ({
+  currentDocumentIds,
   documents,
-  messages,
   repository,
   resourceId,
   threadId
 }: {
+  currentDocumentIds?: ReadonlySet<string>;
   documents: SurveyAssistantThreadDocument[];
-  messages: UIMessage[];
   repository: SurveyAssistantThreadDocumentRepository;
   resourceId: string;
   threadId: string;
@@ -87,5 +103,5 @@ declare const loadSurveyAssistantDocuments: ({
   threadId: string;
 }) => Promise<SurveyAssistantModelAttachment[]>;
 //#endregion
-export { SurveyAssistantThreadDocumentRepository as a, loadSurveyAssistantThreadAttachments as c, SurveyAssistantThreadDocumentContent as i, registerSurveyAssistantThreadAttachments as l, SurveyAssistantModelAttachment as n, getSurveyAssistantReplayableDocuments as o, SurveyAssistantThreadDocument as r, loadSurveyAssistantDocuments as s, SurveyAssistantAttachmentAccess as t, removeHistoricalSurveyAssistantAttachments as u };
-//# sourceMappingURL=thread-documents-DamIq9Od.d.mts.map
+export { SurveyAssistantThreadDocumentRepository as a, loadSurveyAssistantThreadAttachments as c, replaceCurrentSurveyAssistantAttachmentsWithReferences as d, SurveyAssistantThreadDocumentContent as i, registerSurveyAssistantThreadAttachments as l, SurveyAssistantModelAttachment as n, getSurveyAssistantReplayableDocuments as o, SurveyAssistantThreadDocument as r, loadSurveyAssistantDocuments as s, SurveyAssistantAttachmentAccess as t, removeHistoricalSurveyAssistantAttachments as u };
+//# sourceMappingURL=thread-documents-BlxNjc7S.d.mts.map

package/dist/thread-documents-BlxNjc7S.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"thread-documents-BlxNjc7S.d.mts","names":[],"sources":["../src/server/thread-documents.ts"],"mappings":";;;;UAOiB,6BAAA;EACf,EAAA;EACA,IAAA;EACA,SAAA;EACA,KAAA;EACA,WAAA;EACA,SAAA;AAAA;AAAA,UAGe,oCAAA,SAA6C,6BAAA;EAC5D,IAAA;AAAA;AAAA,UAGe,8BAAA;EACf,QAAA;EACA,SAAA;EACA,IAAA;EACA,WAAA;AAAA;AAAA,UAGe,+BAAA;EACf,SAAA,EAAW,6BAAA;EAXP;EAaJ,kBAAA;EACA,mBAAA,EAAqB,8BAAA;EACrB,IAAA,CAAK,WAAA,aAAwB,OAAA,CAAQ,8BAAA;AAAA;AAAA,UAGtB,uCAAA;EACf,GAAA,CAAI,KAAA;IACF,UAAA;IACA,QAAA;IACA,IAAA;IACA,SAAA;IACA,IAAA;IACA,KAAA;EAAA,IACE,OAAA,CAAQ,6BAAA;EACZ,IAAA,CAAK,KAAA;IAAS,UAAA;IAAoB,QAAA;EAAA,IAAqB,OAAA,CAAQ,6BAAA;EAC/D,IAAA,CAAK,KAAA;IACH,UAAA;IACA,QAAA;IACA,UAAA;EAAA,IACE,OAAA,CAAQ,oCAAA;EACZ,YAAA,CAAa,KAAA;IAAS,UAAA;IAAoB,QAAA;EAAA,IAAqB,OAAA;AAAA;;;;AAfjE;;cAgEa,sDAAA;EAA0D,SAAA;EAAA;AAAA;EAIrE,SAAA,EAAW,6BAAA;EACX,QAAA,EAAU,SAAA;AAAA,MACR,SAAA;AAAA,cA2BS,wCAAA;EAAkD,QAAA;EAAA,UAAA;EAAA,UAAA;EAAA;AAAA;EAM7D,QAAA,EAAU,SAAA;EACV,UAAA,EAAY,uCAAA;EACZ,UAAA;EACA,QAAA;AAAA,MACD,OAAA,CAAA,6BAAA;AAAA,cAkCY,0CAAA,GAA8C,QAAA,EAAU,SAAA,OAAW,SAAA,UAAF,IAAA,CAAE,WAAA,EAAA,IAAA,CAAA,OAAA;;;;;cAanE,qCAAA,GACX,SAAA,EAAW,6BAAA,IACX,kBAAA,EAAoB,WAAA,aAAmB,6BAAA;AAAA,cAG5B,oCAAA;EAA8C,kBAAA;EAAA,SAAA;EAAA,UAAA;EAAA,UAAA;EAAA;AAAA;EAOzD,kBAAA,GAAqB,WAAA;EACrB,SAAA,EAAW,6BAAA;EACX,UAAA,EAAY,uCAAA;EACZ,UAAA;EACA,QAAA;AAAA,MACE,OAAA,CAAQ,8BAAA;AAAA,cAcC,4BAAA;EAAsC,SAAA;EAAA,UAAA;EAAA,UAAA;EAAA;AAAA;EAMjD,SAAA,EAAW,6BAAA;EACX,UAAA,EAAY,uCAAA;EACZ,UAAA;EACA,QAAA;AAAA,MACE,OAAA,CAAQ,8BAAA"}