fix(lib): forward dreaming on document adds

better-fetch parses request bodies through the input schema, so the
missing dreaming field was stripped before requests left the client.
Add the field to single and batch add schemas and cover it with tests.

Refs #1851

packages/lib/api.ts

@@ -4,6 +4,7 @@ import {
 	AnalyticsChatResponseSchema,
 	AnalyticsMemoryResponseSchema,
 	AnalyticsUsageResponseSchema,
+	BatchAddDocumentsRequestSchema,
 	BulkDeleteMemoriesResponseSchema,
 	BulkDeleteMemoriesSchema,
 	ConnectionResponseSchema,
@@ -215,23 +216,7 @@ export const apiSchema = createSchema({
 		output: MemoryResponseSchema,
 	},
 	"@post/documents/batch": {
-		input: z.object({
-			documents: z
-				.array(
-					z.object({
-						content: z.string(),
-						containerTags: z.array(z.string()).optional(),
-						containerTag: z.string().optional(),
-						entityContext: z.string().max(1500).optional(),
-						metadata: z.record(z.unknown()).optional(),
-					}),
-				)
-				.min(1)
-				.max(600),
-			containerTag: z.string().optional(),
-			entityContext: z.string().max(1500).optional(),
-			metadata: z.record(z.unknown()).optional(),
-		}),
+		input: BatchAddDocumentsRequestSchema,
 		output: z.object({
 			results: z.array(
 				z.object({

packages/validation/api.test.ts

@@ -1,9 +1,11 @@
 import { describe, expect, it } from "bun:test"
 import { readFileSync } from "node:fs"
 import {
+	BatchAddDocumentsRequestSchema,
 	BulkDeleteMemoriesSchema,
 	DocumentsWithMemoriesQuerySchema,
 	ListMemoriesQuerySchema,
+	MemoryAddSchema,
 	SearchRequestSchema,
 	Searchv4RequestSchema,
 } from "./api"
@@ -175,3 +177,32 @@ describe("pagination query schemas", () => {
 		expect(BulkDeleteMemoriesSchema.safeParse(ok).success).toBe(true)
 	})
 })
+
+describe("document add input schemas", () => {
+	it("preserves dreaming on a single document add", () => {
+		const parsed = MemoryAddSchema.parse({
+			content: "probe",
+			dreaming: "instant",
+		})
+
+		expect(parsed).toEqual({ content: "probe", dreaming: "instant" })
+	})
+
+	it("preserves dreaming on a batch document add", () => {
+		const parsed = BatchAddDocumentsRequestSchema.parse({
+			documents: [{ content: "probe" }],
+			dreaming: "dynamic",
+		})
+
+		expect(parsed).toEqual({
+			documents: [{ content: "probe" }],
+			dreaming: "dynamic",
+		})
+	})
+
+	it("rejects an invalid dreaming mode", () => {
+		expect(() =>
+			MemoryAddSchema.parse({ content: "probe", dreaming: "later" }),
+		).toThrow()
+	})
+})

packages/validation/api.ts

@@ -170,7 +170,34 @@ export const MemoryUpdateSchema = z.object({
 	}),
 })
 
-export const MemoryAddSchema = MemoryUpdateSchema
+export const DreamingModeSchema = z.enum(["dynamic", "instant"]).openapi({
+	description:
+		'Processing mode. "dynamic" (default) groups related documents together so memories form from coherent units. "instant" processes each document on its own right away and bills one extra operation per document.',
+	example: "instant",
+})
+
+export const MemoryAddSchema = MemoryUpdateSchema.extend({
+	dreaming: DreamingModeSchema.optional(),
+})
+
+export const BatchAddDocumentsRequestSchema = z.object({
+	documents: z
+		.array(
+			z.object({
+				content: z.string(),
+				containerTags: z.array(z.string()).optional(),
+				containerTag: z.string().optional(),
+				entityContext: z.string().max(1500).optional(),
+				metadata: z.record(z.unknown()).optional(),
+			}),
+		)
+		.min(1)
+		.max(600),
+	containerTag: z.string().optional(),
+	entityContext: z.string().max(1500).optional(),
+	metadata: z.record(z.unknown()).optional(),
+	dreaming: DreamingModeSchema.optional(),
+})
 
 export const PaginationSchema = z
 	.object({