@better-agent/drizzle 0.2.0-beta.1

package/dist/cli.d.mts

@@ -0,0 +1 @@
+export { };

package/dist/cli.mjs

@@ -0,0 +1,143 @@
+#!/usr/bin/env node
+import { renderDrizzleSchema } from "@farming-labs/orm";
+import { resolveSchemaDomains, selectBetterAgentSchema } from "@better-agent/schema";
+import path from "node:path";
+import process$1 from "node:process";
+import { Command } from "commander";
+import { mkdir, readFile, writeFile } from "node:fs/promises";
+import { cancel, isCancel, multiselect } from "@clack/prompts";
+
+//#region src/generate.ts
+const DRIZZLE_START_MARKER = "// @better-agent/drizzle:start";
+const DRIZZLE_END_MARKER = "// @better-agent/drizzle:end";
+function upsertManagedDrizzleBlock(existing, schema) {
+	const block = `${DRIZZLE_START_MARKER}\n${schema.trim()}\n${DRIZZLE_END_MARKER}\n`;
+	const start = existing.indexOf(DRIZZLE_START_MARKER);
+	const end = existing.indexOf(DRIZZLE_END_MARKER);
+	if (start >= 0 && end > start) return `${existing.slice(0, start).replace(/\s*$/, "\n\n")}${block}${existing.slice(end + 28).replace(/^\s*/, "\n\n")}`.replace(/\n{3,}/g, "\n\n");
+	return existing.trim().length === 0 ? block : `${existing.trim()}\n\n${block}`;
+}
+async function readExistingSchema(out) {
+	try {
+		return await readFile(out, "utf8");
+	} catch {
+		return null;
+	}
+}
+async function generateDrizzleSchemaFile(options) {
+	const { out, dialect, mode, only } = options;
+	const schema = renderDrizzleSchema(selectBetterAgentSchema(only), { dialect });
+	const resultBase = {
+		out,
+		dialect,
+		mode,
+		...only ? { only } : {}
+	};
+	await mkdir(path.dirname(out), { recursive: true });
+	if (mode === "replace") {
+		await writeFile(out, `${schema.trim()}\n`, "utf8");
+		return {
+			...resultBase,
+			operation: "replaced"
+		};
+	}
+	const existing = await readExistingSchema(out);
+	await writeFile(out, upsertManagedDrizzleBlock(existing ?? "", schema), "utf8");
+	return {
+		...resultBase,
+		operation: existing === null ? "created" : "updated"
+	};
+}
+
+//#endregion
+//#region src/interactive.ts
+const drizzleDomains = [
+	"memory",
+	"runs",
+	"streams"
+];
+const domainOptions = {
+	memory: {
+		label: "Memory",
+		hint: "threads/messages"
+	},
+	runs: {
+		label: "Runs",
+		hint: "history"
+	},
+	streams: {
+		label: "Streams",
+		hint: "resume"
+	}
+};
+async function promptDrizzleDomains() {
+	const selected = await multiselect({
+		message: "Select schema domains",
+		required: false,
+		initialValues: [...drizzleDomains],
+		options: drizzleDomains.map((domain) => ({
+			value: domain,
+			label: domainOptions[domain].label,
+			hint: domainOptions[domain].hint
+		}))
+	});
+	if (isCancel(selected)) {
+		cancel("Generation cancelled.");
+		process.exit(0);
+	}
+	return selected.length === drizzleDomains.length ? void 0 : resolveSchemaDomains(selected);
+}
+
+//#endregion
+//#region src/cli.ts
+const program = new Command();
+const dialects = [
+	"pg",
+	"mysql",
+	"sqlite"
+];
+const modes = ["block", "replace"];
+const green = (value) => process$1.stdout.isTTY ? `\x1b[32m${value}\x1b[0m` : value;
+const dim = (value) => process$1.stdout.isTTY ? `\x1b[2m${value}\x1b[0m` : value;
+const successMark = green("✓");
+function printHeader() {
+	console.log(green("BETTER-AGENT"));
+	console.log(dim("Drizzle adapter"));
+	console.log("");
+}
+function parseDomains(input) {
+	if (!input) return;
+	const only = input.split(",").map((domain) => domain.trim()).filter(Boolean);
+	if (only.some((domain) => !drizzleDomains.includes(domain))) throw new Error(`Unsupported domain list '${input}'.`);
+	return resolveSchemaDomains(only);
+}
+function parseDialect(input) {
+	if (!dialects.includes(input)) throw new Error(`Unsupported dialect '${input}'.`);
+	return input;
+}
+function parseMode(input) {
+	if (!modes.includes(input)) throw new Error(`Unsupported mode '${input}'.`);
+	return input;
+}
+program.name("better-agent-drizzle").description("Better Agent Drizzle helpers");
+program.command("generate").description("Generate Better Agent Drizzle schema into your project").option("--out <path>", "Drizzle schema file path", "src/db/better-agent.schema.ts").option("--dialect <dialect>", "pg | mysql | sqlite", "pg").option("--mode <mode>", "block | replace", "block").option("--only <domains>", "memory | runs | streams", "").action(async (options) => {
+	const only = !options.only && process$1.stdin.isTTY ? await promptDrizzleDomains() : parseDomains(options.only);
+	const result = await generateDrizzleSchemaFile({
+		out: path.resolve(process$1.cwd(), options.out),
+		dialect: parseDialect(options.dialect),
+		mode: parseMode(options.mode),
+		only
+	});
+	console.log(`${successMark} updated Better Agent Drizzle schema: ${result.operation} ${result.out}`);
+	console.log("");
+});
+printHeader();
+program.parseAsync().catch((error) => {
+	const message = error instanceof Error ? error.message : String(error);
+	console.error(message);
+	process$1.exitCode = 1;
+});
+
+//#endregion
+export {  };
+//# sourceMappingURL=cli.mjs.map

package/dist/cli.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli.mjs","names":["process"],"sources":["../src/generate.ts","../src/interactive.ts","../src/cli.ts"],"sourcesContent":["import { mkdir, readFile, writeFile } from \"node:fs/promises\";\nimport path from \"node:path\";\nimport { selectBetterAgentSchema } from \"@better-agent/schema\";\nimport type { SchemaDomain } from \"@better-agent/schema\";\nimport { renderDrizzleSchema } from \"@farming-labs/orm\";\n\nexport const DRIZZLE_START_MARKER = \"// @better-agent/drizzle:start\";\nexport const DRIZZLE_END_MARKER = \"// @better-agent/drizzle:end\";\n\nexport type DrizzleDialect = \"pg\" | \"mysql\" | \"sqlite\";\nexport type GenerateMode = \"block\" | \"replace\";\n\nexport interface GenerateDrizzleSchemaOptions {\n    out: string;\n    dialect: DrizzleDialect;\n    mode: GenerateMode;\n    only?: readonly SchemaDomain[];\n}\n\nexport interface GenerateDrizzleSchemaResult {\n    out: string;\n    dialect: DrizzleDialect;\n    mode: GenerateMode;\n    only?: readonly SchemaDomain[];\n    operation: \"created\" | \"updated\" | \"replaced\";\n}\n\nexport function upsertManagedDrizzleBlock(existing: string, schema: string): string {\n    const block = `${DRIZZLE_START_MARKER}\\n${schema.trim()}\\n${DRIZZLE_END_MARKER}\\n`;\n    const start = existing.indexOf(DRIZZLE_START_MARKER);\n    const end = existing.indexOf(DRIZZLE_END_MARKER);\n\n    if (start >= 0 && end > start) {\n        const before = existing.slice(0, start).replace(/\\s*$/, \"\\n\\n\");\n        const after = existing.slice(end + DRIZZLE_END_MARKER.length).replace(/^\\s*/, \"\\n\\n\");\n        return `${before}${block}${after}`.replace(/\\n{3,}/g, \"\\n\\n\");\n    }\n\n    return existing.trim().length === 0 ? block : `${existing.trim()}\\n\\n${block}`;\n}\n\nasync function readExistingSchema(out: string): Promise<string | null> {\n    try {\n        return await readFile(out, \"utf8\");\n    } catch {\n        return null;\n    }\n}\n\nexport async function generateDrizzleSchemaFile(\n    options: GenerateDrizzleSchemaOptions,\n): Promise<GenerateDrizzleSchemaResult> {\n    const { out, dialect, mode, only } = options;\n    const schema = renderDrizzleSchema(selectBetterAgentSchema(only), { dialect });\n    const resultBase = {\n        out,\n        dialect,\n        mode,\n        ...(only ? { only } : {}),\n    };\n    await mkdir(path.dirname(out), { recursive: true });\n\n    if (mode === \"replace\") {\n        await writeFile(out, `${schema.trim()}\\n`, \"utf8\");\n        return {\n            ...resultBase,\n            operation: \"replaced\",\n        };\n    }\n\n    const existing = await readExistingSchema(out);\n    const next = upsertManagedDrizzleBlock(existing ?? \"\", schema);\n    await writeFile(out, next, \"utf8\");\n\n    return {\n        ...resultBase,\n        operation: existing === null ? \"created\" : \"updated\",\n    };\n}\n","import { resolveSchemaDomains } from \"@better-agent/schema\";\nimport type { SchemaDomain } from \"@better-agent/schema\";\nimport { cancel, isCancel, multiselect } from \"@clack/prompts\";\n\nexport const drizzleDomains = [\n    \"memory\",\n    \"runs\",\n    \"streams\",\n] as const satisfies readonly SchemaDomain[];\nexport type DrizzleDomain = (typeof drizzleDomains)[number];\n\nconst domainOptions = {\n    memory: {\n        label: \"Memory\",\n        hint: \"threads/messages\",\n    },\n    runs: {\n        label: \"Runs\",\n        hint: \"history\",\n    },\n    streams: {\n        label: \"Streams\",\n        hint: \"resume\",\n    },\n} as const satisfies Record<DrizzleDomain, { label: string; hint: string }>;\n\nexport async function promptDrizzleDomains(): Promise<readonly DrizzleDomain[] | undefined> {\n    const selected = await multiselect<DrizzleDomain>({\n        message: \"Select schema domains\",\n        required: false,\n        initialValues: [...drizzleDomains],\n        options: drizzleDomains.map((domain) => ({\n            value: domain,\n            label: domainOptions[domain].label,\n            hint: domainOptions[domain].hint,\n        })),\n    });\n\n    if (isCancel(selected)) {\n        cancel(\"Generation cancelled.\");\n        process.exit(0);\n    }\n\n    return selected.length === drizzleDomains.length ? undefined : resolveSchemaDomains(selected);\n}\n","#!/usr/bin/env node\n\nimport path from \"node:path\";\nimport process from \"node:process\";\nimport { resolveSchemaDomains } from \"@better-agent/schema\";\nimport type { SchemaDomain } from \"@better-agent/schema\";\nimport { Command } from \"commander\";\nimport { type DrizzleDialect, type GenerateMode, generateDrizzleSchemaFile } from \"./generate\";\nimport { drizzleDomains, promptDrizzleDomains } from \"./interactive\";\n\nconst program = new Command();\nconst dialects = [\"pg\", \"mysql\", \"sqlite\"] as const;\nconst modes = [\"block\", \"replace\"] as const;\nconst green = (value: string) => (process.stdout.isTTY ? `\\x1b[32m${value}\\x1b[0m` : value);\nconst dim = (value: string) => (process.stdout.isTTY ? `\\x1b[2m${value}\\x1b[0m` : value);\nconst successMark = green(\"✓\");\n\nfunction printHeader() {\n    console.log(green(\"BETTER-AGENT\"));\n    console.log(dim(\"Drizzle adapter\"));\n    console.log(\"\");\n}\n\nfunction parseDomains(input?: string): readonly SchemaDomain[] | undefined {\n    if (!input) {\n        return undefined;\n    }\n\n    const only = input\n        .split(\",\")\n        .map((domain) => domain.trim())\n        .filter(Boolean);\n\n    if (only.some((domain) => !drizzleDomains.includes(domain as SchemaDomain))) {\n        throw new Error(`Unsupported domain list '${input}'.`);\n    }\n\n    return resolveSchemaDomains(only as SchemaDomain[]);\n}\n\nfunction parseDialect(input: string): DrizzleDialect {\n    if (!dialects.includes(input as DrizzleDialect)) {\n        throw new Error(`Unsupported dialect '${input}'.`);\n    }\n\n    return input as DrizzleDialect;\n}\n\nfunction parseMode(input: string): GenerateMode {\n    if (!modes.includes(input as GenerateMode)) {\n        throw new Error(`Unsupported mode '${input}'.`);\n    }\n\n    return input as GenerateMode;\n}\n\nprogram.name(\"better-agent-drizzle\").description(\"Better Agent Drizzle helpers\");\n\nprogram\n    .command(\"generate\")\n    .description(\"Generate Better Agent Drizzle schema into your project\")\n    .option(\"--out <path>\", \"Drizzle schema file path\", \"src/db/better-agent.schema.ts\")\n    .option(\"--dialect <dialect>\", \"pg | mysql | sqlite\", \"pg\")\n    .option(\"--mode <mode>\", \"block | replace\", \"block\")\n    .option(\"--only <domains>\", \"memory | runs | streams\", \"\")\n    .action(\n        async (options: {\n            out: string;\n            dialect: DrizzleDialect;\n            mode: GenerateMode;\n            only?: string;\n        }) => {\n            const only =\n                !options.only && process.stdin.isTTY\n                    ? await promptDrizzleDomains()\n                    : parseDomains(options.only);\n            const result = await generateDrizzleSchemaFile({\n                out: path.resolve(process.cwd(), options.out),\n                dialect: parseDialect(options.dialect),\n                mode: parseMode(options.mode),\n                only,\n            });\n\n            console.log(\n                `${successMark} updated Better Agent Drizzle schema: ${result.operation} ${result.out}`,\n            );\n            console.log(\"\");\n        },\n    );\n\nprintHeader();\n\nprogram.parseAsync().catch((error) => {\n    const message = error instanceof Error ? error.message : String(error);\n    console.error(message);\n    process.exitCode = 1;\n});\n"],"mappings":";;;;;;;;;;AAMA,MAAa,uBAAuB;AACpC,MAAa,qBAAqB;AAoBlC,SAAgB,0BAA0B,UAAkB,QAAwB;CAChF,MAAM,QAAQ,GAAG,qBAAqB,IAAI,OAAO,MAAM,CAAC,IAAI,mBAAmB;CAC/E,MAAM,QAAQ,SAAS,QAAQ,qBAAqB;CACpD,MAAM,MAAM,SAAS,QAAQ,mBAAmB;AAEhD,KAAI,SAAS,KAAK,MAAM,MAGpB,QAAO,GAFQ,SAAS,MAAM,GAAG,MAAM,CAAC,QAAQ,QAAQ,OAAO,GAE5C,QADL,SAAS,MAAM,MAAM,GAA0B,CAAC,QAAQ,QAAQ,OAAO,GAClD,QAAQ,WAAW,OAAO;AAGjE,QAAO,SAAS,MAAM,CAAC,WAAW,IAAI,QAAQ,GAAG,SAAS,MAAM,CAAC,MAAM;;AAG3E,eAAe,mBAAmB,KAAqC;AACnE,KAAI;AACA,SAAO,MAAM,SAAS,KAAK,OAAO;SAC9B;AACJ,SAAO;;;AAIf,eAAsB,0BAClB,SACoC;CACpC,MAAM,EAAE,KAAK,SAAS,MAAM,SAAS;CACrC,MAAM,SAAS,oBAAoB,wBAAwB,KAAK,EAAE,EAAE,SAAS,CAAC;CAC9E,MAAM,aAAa;EACf;EACA;EACA;EACA,GAAI,OAAO,EAAE,MAAM,GAAG,EAAE;EAC3B;AACD,OAAM,MAAM,KAAK,QAAQ,IAAI,EAAE,EAAE,WAAW,MAAM,CAAC;AAEnD,KAAI,SAAS,WAAW;AACpB,QAAM,UAAU,KAAK,GAAG,OAAO,MAAM,CAAC,KAAK,OAAO;AAClD,SAAO;GACH,GAAG;GACH,WAAW;GACd;;CAGL,MAAM,WAAW,MAAM,mBAAmB,IAAI;AAE9C,OAAM,UAAU,KADH,0BAA0B,YAAY,IAAI,OAAO,EACnC,OAAO;AAElC,QAAO;EACH,GAAG;EACH,WAAW,aAAa,OAAO,YAAY;EAC9C;;;;;ACzEL,MAAa,iBAAiB;CAC1B;CACA;CACA;CACH;AAGD,MAAM,gBAAgB;CAClB,QAAQ;EACJ,OAAO;EACP,MAAM;EACT;CACD,MAAM;EACF,OAAO;EACP,MAAM;EACT;CACD,SAAS;EACL,OAAO;EACP,MAAM;EACT;CACJ;AAED,eAAsB,uBAAsE;CACxF,MAAM,WAAW,MAAM,YAA2B;EAC9C,SAAS;EACT,UAAU;EACV,eAAe,CAAC,GAAG,eAAe;EAClC,SAAS,eAAe,KAAK,YAAY;GACrC,OAAO;GACP,OAAO,cAAc,QAAQ;GAC7B,MAAM,cAAc,QAAQ;GAC/B,EAAE;EACN,CAAC;AAEF,KAAI,SAAS,SAAS,EAAE;AACpB,SAAO,wBAAwB;AAC/B,UAAQ,KAAK,EAAE;;AAGnB,QAAO,SAAS,WAAW,eAAe,SAAS,SAAY,qBAAqB,SAAS;;;;;ACjCjG,MAAM,UAAU,IAAI,SAAS;AAC7B,MAAM,WAAW;CAAC;CAAM;CAAS;CAAS;AAC1C,MAAM,QAAQ,CAAC,SAAS,UAAU;AAClC,MAAM,SAAS,UAAmBA,UAAQ,OAAO,QAAQ,WAAW,MAAM,WAAW;AACrF,MAAM,OAAO,UAAmBA,UAAQ,OAAO,QAAQ,UAAU,MAAM,WAAW;AAClF,MAAM,cAAc,MAAM,IAAI;AAE9B,SAAS,cAAc;AACnB,SAAQ,IAAI,MAAM,eAAe,CAAC;AAClC,SAAQ,IAAI,IAAI,kBAAkB,CAAC;AACnC,SAAQ,IAAI,GAAG;;AAGnB,SAAS,aAAa,OAAqD;AACvE,KAAI,CAAC,MACD;CAGJ,MAAM,OAAO,MACR,MAAM,IAAI,CACV,KAAK,WAAW,OAAO,MAAM,CAAC,CAC9B,OAAO,QAAQ;AAEpB,KAAI,KAAK,MAAM,WAAW,CAAC,eAAe,SAAS,OAAuB,CAAC,CACvE,OAAM,IAAI,MAAM,4BAA4B,MAAM,IAAI;AAG1D,QAAO,qBAAqB,KAAuB;;AAGvD,SAAS,aAAa,OAA+B;AACjD,KAAI,CAAC,SAAS,SAAS,MAAwB,CAC3C,OAAM,IAAI,MAAM,wBAAwB,MAAM,IAAI;AAGtD,QAAO;;AAGX,SAAS,UAAU,OAA6B;AAC5C,KAAI,CAAC,MAAM,SAAS,MAAsB,CACtC,OAAM,IAAI,MAAM,qBAAqB,MAAM,IAAI;AAGnD,QAAO;;AAGX,QAAQ,KAAK,uBAAuB,CAAC,YAAY,+BAA+B;AAEhF,QACK,QAAQ,WAAW,CACnB,YAAY,yDAAyD,CACrE,OAAO,gBAAgB,4BAA4B,gCAAgC,CACnF,OAAO,uBAAuB,uBAAuB,KAAK,CAC1D,OAAO,iBAAiB,mBAAmB,QAAQ,CACnD,OAAO,oBAAoB,2BAA2B,GAAG,CACzD,OACG,OAAO,YAKD;CACF,MAAM,OACF,CAAC,QAAQ,QAAQA,UAAQ,MAAM,QACzB,MAAM,sBAAsB,GAC5B,aAAa,QAAQ,KAAK;CACpC,MAAM,SAAS,MAAM,0BAA0B;EAC3C,KAAK,KAAK,QAAQA,UAAQ,KAAK,EAAE,QAAQ,IAAI;EAC7C,SAAS,aAAa,QAAQ,QAAQ;EACtC,MAAM,UAAU,QAAQ,KAAK;EAC7B;EACH,CAAC;AAEF,SAAQ,IACJ,GAAG,YAAY,wCAAwC,OAAO,UAAU,GAAG,OAAO,MACrF;AACD,SAAQ,IAAI,GAAG;EAEtB;AAEL,aAAa;AAEb,QAAQ,YAAY,CAAC,OAAO,UAAU;CAClC,MAAM,UAAU,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM;AACtE,SAAQ,MAAM,QAAQ;AACtB,WAAQ,WAAW;EACrB"}

package/dist/index.d.mts

@@ -0,0 +1,21 @@
+import { AnyModelDefinition, OrmClient, SchemaDefinition } from "@farming-labs/orm";
+import { DrizzleDriverConfig } from "@farming-labs/orm-drizzle";
+import { BetterAgentStorage } from "@better-agent/core";
+
+//#region src/types.d.ts
+type DrizzleStorage = BetterAgentStorage;
+type BetterAgentDrizzleDriverConfig = DrizzleDriverConfig<SchemaDefinition<any>>;
+type DrizzleDialect = BetterAgentDrizzleDriverConfig["dialect"];
+interface DrizzleOptions {
+  db: NonNullable<BetterAgentDrizzleDriverConfig["db"]>;
+  dialect: BetterAgentDrizzleDriverConfig["dialect"];
+}
+//#endregion
+//#region src/orm.d.ts
+declare function createDrizzleOrm<TSchema extends SchemaDefinition<Record<string, AnyModelDefinition>>>(schema: TSchema, options: DrizzleOptions): OrmClient<TSchema>;
+//#endregion
+//#region src/storage/adapter.d.ts
+declare function drizzleStorage(options: DrizzleOptions): BetterAgentStorage;
+//#endregion
+export { DrizzleDialect, DrizzleOptions, DrizzleStorage, createDrizzleOrm, drizzleStorage };
+//# sourceMappingURL=index.d.mts.map

package/dist/index.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.mts","names":[],"sources":["../src/types.ts","../src/orm.ts","../src/storage/adapter.ts"],"mappings":";;;;;KAIY,cAAA,GAAiB,kBAAA;AAAA,KAGxB,8BAAA,GAAiC,mBAAA,CAAoB,gBAAA;AAAA,KAE9C,cAAA,GAAiB,8BAAA;AAAA,UAEZ,cAAA;EACb,EAAA,EAAI,WAAA,CAAY,8BAAA;EAChB,OAAA,EAAS,8BAAA;AAAA;;;iBCRG,gBAAA,iBACI,gBAAA,CAAiB,MAAA,SAAe,kBAAA,GAAA,CAClD,MAAA,EAAQ,OAAA,EAAS,OAAA,EAAS,cAAA,GAAiB,SAAA,CAAU,OAAA;;;iBCwGvC,cAAA,CAAe,OAAA,EAAS,cAAA,GAAiB,kBAAA"}

package/dist/index.mjs

@@ -0,0 +1,231 @@
+import { createOrm } from "@farming-labs/orm";
+import { createDrizzleDriver } from "@farming-labs/orm-drizzle";
+import { storageTables, throwUnsupportedStorageTable } from "@better-agent/core";
+import { selectBetterAgentSchema, toBetterAgentOrmListQuery } from "@better-agent/schema";
+
+//#region src/orm.ts
+function createDrizzleOrm(schema, options) {
+	return createOrm({
+		schema,
+		driver: createDrizzleDriver({
+			db: options.db,
+			dialect: options.dialect
+		})
+	});
+}
+
+//#endregion
+//#region src/storage/mappers.ts
+function fromRunRow(row) {
+	return {
+		runId: row.runId,
+		agentName: row.agentName,
+		...row.threadId ? { threadId: row.threadId } : {},
+		...row.scope ? { scope: row.scope } : {},
+		status: row.status,
+		...row.abortRequestedAt ? { abortRequestedAt: row.abortRequestedAt.getTime() } : {},
+		...row.finalEvent ? { finalEvent: row.finalEvent } : {},
+		config: row.config ?? void 0,
+		startedAt: row.startedAt.getTime(),
+		updatedAt: row.updatedAt.getTime(),
+		...row.finishedAt ? { finishedAt: row.finishedAt.getTime() } : {}
+	};
+}
+function toRunRow(id, record) {
+	return {
+		runId: record.runId || id,
+		agentName: record.agentName,
+		threadId: record.threadId ?? null,
+		scope: record.scope ?? null,
+		status: record.status,
+		abortRequestedAt: record.abortRequestedAt === void 0 ? null : new Date(record.abortRequestedAt),
+		finalEvent: record.finalEvent ?? null,
+		config: record.config ?? null,
+		startedAt: new Date(record.startedAt),
+		updatedAt: new Date(record.updatedAt),
+		finishedAt: record.finishedAt === void 0 ? null : new Date(record.finishedAt)
+	};
+}
+function fromThreadRow(row) {
+	return {
+		id: row.id,
+		agentName: row.agentName ?? "",
+		...row.scope ? { scope: row.scope } : {},
+		...row.title ? { title: row.title } : {},
+		...row.state !== null ? { state: row.state } : {},
+		...row.metadata && typeof row.metadata === "object" && !Array.isArray(row.metadata) ? { metadata: row.metadata } : {},
+		createdAt: row.createdAt.getTime(),
+		updatedAt: row.updatedAt.getTime()
+	};
+}
+function toThreadRow(id, record) {
+	return {
+		id: record.id || id,
+		agentName: record.agentName,
+		scope: record.scope ?? null,
+		title: record.title ?? null,
+		state: record.state ?? null,
+		metadata: record.metadata ?? null,
+		createdAt: new Date(record.createdAt),
+		updatedAt: new Date(record.updatedAt)
+	};
+}
+function fromMessageRow(row) {
+	return {
+		...row.message,
+		threadId: row.threadId,
+		...row.runId ? { runId: row.runId } : {},
+		createdAt: row.createdAt.getTime()
+	};
+}
+function toMessageRow(id, record) {
+	const { threadId, runId, createdAt, ...message } = record;
+	return {
+		id,
+		threadId,
+		runId: runId ?? null,
+		message,
+		createdAt: new Date(createdAt)
+	};
+}
+function fromStreamRow(row) {
+	return {
+		runId: row.runId,
+		status: row.status,
+		createdAt: row.createdAt.getTime(),
+		updatedAt: row.updatedAt.getTime(),
+		...row.closedAt ? { closedAt: row.closedAt.getTime() } : {}
+	};
+}
+function toStreamRow(id, record) {
+	return {
+		id,
+		runId: record.runId,
+		status: record.status,
+		createdAt: new Date(record.createdAt),
+		updatedAt: new Date(record.updatedAt),
+		closedAt: record.closedAt === void 0 ? null : new Date(record.closedAt)
+	};
+}
+function fromStreamEventRow(row) {
+	return {
+		id: row.id,
+		runId: row.runId,
+		seq: row.seq,
+		timestamp: row.timestamp.getTime(),
+		event: row.event
+	};
+}
+function toStreamEventRow(id, record) {
+	return {
+		id: record.id || id,
+		runId: record.runId,
+		seq: record.seq,
+		timestamp: new Date(record.timestamp),
+		event: record.event
+	};
+}
+
+//#endregion
+//#region src/storage/adapter.ts
+const drizzleSchema = selectBetterAgentSchema();
+async function upsertRow(input) {
+	const where = { [input.keyField]: input.id };
+	if (!await input.model.findUnique({ where })) {
+		await input.model.upsert({
+			where,
+			create: input.row,
+			update: input.row
+		});
+		return;
+	}
+	await input.model.update({
+		where,
+		data: input.row
+	});
+}
+function createTableAdapter(input) {
+	return {
+		async get(id) {
+			const row = await input.model.findUnique({ where: { [input.keyField]: id } });
+			return row ? input.fromRow(row) : void 0;
+		},
+		async set(id, value) {
+			await upsertRow({
+				model: input.model,
+				keyField: input.keyField,
+				id,
+				row: input.toRow(id, value)
+			});
+		},
+		async delete(id) {
+			await input.model.deleteMany({ where: { [input.keyField]: id } });
+		},
+		async list(query) {
+			return { items: (await input.model.findMany(toBetterAgentOrmListQuery(query))).map(input.fromRow) };
+		}
+	};
+}
+function resolveAdapter(adapters, table) {
+	return adapters.get(table) ?? throwUnsupportedStorageTable(table, "drizzle storage");
+}
+function requireModel(orm, modelName) {
+	return orm[modelName];
+}
+function drizzleStorage(options) {
+	const orm = createDrizzleOrm(drizzleSchema, options);
+	const adapters = /* @__PURE__ */ new Map();
+	const runModel = requireModel(orm, "run");
+	if (runModel) adapters.set(storageTables.runs, createTableAdapter({
+		model: runModel,
+		keyField: "runId",
+		fromRow: fromRunRow,
+		toRow: toRunRow
+	}));
+	const threadModel = requireModel(orm, "thread");
+	if (threadModel) adapters.set(storageTables.memoryThreads, createTableAdapter({
+		model: threadModel,
+		keyField: "id",
+		fromRow: fromThreadRow,
+		toRow: toThreadRow
+	}));
+	const messageModel = requireModel(orm, "message");
+	if (messageModel) adapters.set(storageTables.memoryMessages, createTableAdapter({
+		model: messageModel,
+		keyField: "id",
+		fromRow: fromMessageRow,
+		toRow: toMessageRow
+	}));
+	const streamModel = requireModel(orm, "stream");
+	if (streamModel) adapters.set(storageTables.streams, createTableAdapter({
+		model: streamModel,
+		keyField: "id",
+		fromRow: fromStreamRow,
+		toRow: toStreamRow
+	}));
+	const streamEventModel = requireModel(orm, "streamEvent");
+	if (streamEventModel) adapters.set(storageTables.streamEvents, createTableAdapter({
+		model: streamEventModel,
+		keyField: "id",
+		fromRow: fromStreamEventRow,
+		toRow: toStreamEventRow
+	}));
+	return {
+		get(table, id) {
+			return resolveAdapter(adapters, table).get(id);
+		},
+		set(table, id, value) {
+			return resolveAdapter(adapters, table).set(id, value);
+		},
+		delete(table, id) {
+			return resolveAdapter(adapters, table).delete(id);
+		},
+		list(table, query) {
+			return resolveAdapter(adapters, table).list(query);
+		}
+	};
+}
+
+//#endregion
+export { createDrizzleOrm, drizzleStorage };
+//# sourceMappingURL=index.mjs.map

package/dist/index.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.mjs","names":[],"sources":["../src/orm.ts","../src/storage/mappers.ts","../src/storage/adapter.ts"],"sourcesContent":["import { createOrm } from \"@farming-labs/orm\";\nimport type { AnyModelDefinition, OrmClient, SchemaDefinition } from \"@farming-labs/orm\";\nimport { createDrizzleDriver } from \"@farming-labs/orm-drizzle\";\nimport type { DrizzleOptions } from \"./types\";\n\nexport function createDrizzleOrm<\n    TSchema extends SchemaDefinition<Record<string, AnyModelDefinition>>,\n>(schema: TSchema, options: DrizzleOptions): OrmClient<TSchema> {\n    return createOrm({\n        schema,\n        driver: createDrizzleDriver({\n            db: options.db,\n            dialect: options.dialect,\n        }),\n    });\n}\n","import type {\n    MemoryMessage,\n    MemoryThread,\n    RunRecord,\n    StreamEventRecord,\n    StreamRecord,\n} from \"@better-agent/core\";\nimport type { memorySchema, runsSchema, streamsSchema } from \"@better-agent/schema\";\nimport type { JsonValue, ScalarRecord } from \"@farming-labs/orm\";\n\nexport type RunRow = ScalarRecord<typeof runsSchema, \"run\">;\nexport type MemoryThreadRow = ScalarRecord<typeof memorySchema, \"thread\">;\nexport type MemoryMessageRow = ScalarRecord<typeof memorySchema, \"message\">;\nexport type StreamRow = ScalarRecord<typeof streamsSchema, \"stream\">;\nexport type StreamEventRow = ScalarRecord<typeof streamsSchema, \"streamEvent\">;\n\nexport function fromRunRow(row: RunRow): RunRecord {\n    return {\n        runId: row.runId,\n        agentName: row.agentName,\n        ...(row.threadId ? { threadId: row.threadId } : {}),\n        ...(row.scope ? { scope: row.scope } : {}),\n        status: row.status,\n        ...(row.abortRequestedAt ? { abortRequestedAt: row.abortRequestedAt.getTime() } : {}),\n        ...(row.finalEvent\n            ? { finalEvent: row.finalEvent as unknown as RunRecord[\"finalEvent\"] }\n            : {}),\n        config: row.config ?? undefined,\n        startedAt: row.startedAt.getTime(),\n        updatedAt: row.updatedAt.getTime(),\n        ...(row.finishedAt ? { finishedAt: row.finishedAt.getTime() } : {}),\n    };\n}\n\nexport function toRunRow(id: string, record: RunRecord): RunRow {\n    return {\n        runId: record.runId || id,\n        agentName: record.agentName,\n        threadId: record.threadId ?? null,\n        scope: record.scope ?? null,\n        status: record.status,\n        abortRequestedAt:\n            record.abortRequestedAt === undefined ? null : new Date(record.abortRequestedAt),\n        finalEvent: (record.finalEvent ?? null) as JsonValue | null,\n        config: (record.config ?? null) as JsonValue | null,\n        startedAt: new Date(record.startedAt),\n        updatedAt: new Date(record.updatedAt),\n        finishedAt: record.finishedAt === undefined ? null : new Date(record.finishedAt),\n    };\n}\n\nexport function fromThreadRow(row: MemoryThreadRow): MemoryThread {\n    return {\n        id: row.id,\n        agentName: row.agentName ?? \"\",\n        ...(row.scope ? { scope: row.scope } : {}),\n        ...(row.title ? { title: row.title } : {}),\n        ...(row.state !== null ? { state: row.state } : {}),\n        ...(row.metadata && typeof row.metadata === \"object\" && !Array.isArray(row.metadata)\n            ? { metadata: row.metadata as Record<string, unknown> }\n            : {}),\n        createdAt: row.createdAt.getTime(),\n        updatedAt: row.updatedAt.getTime(),\n    };\n}\n\nexport function toThreadRow(id: string, record: MemoryThread): MemoryThreadRow {\n    return {\n        id: record.id || id,\n        agentName: record.agentName,\n        scope: record.scope ?? null,\n        title: record.title ?? null,\n        state: (record.state ?? null) as JsonValue | null,\n        metadata: (record.metadata ?? null) as JsonValue | null,\n        createdAt: new Date(record.createdAt),\n        updatedAt: new Date(record.updatedAt),\n    };\n}\n\nexport function fromMessageRow(row: MemoryMessageRow): MemoryMessage {\n    const message = {\n        ...(row.message as unknown as Omit<MemoryMessage, \"threadId\" | \"runId\" | \"createdAt\">),\n        threadId: row.threadId,\n        ...(row.runId ? { runId: row.runId } : {}),\n        createdAt: row.createdAt.getTime(),\n    };\n\n    return message as MemoryMessage;\n}\n\nexport function toMessageRow(id: string, record: MemoryMessage): MemoryMessageRow {\n    const { threadId, runId, createdAt, ...message } = record;\n\n    return {\n        id,\n        threadId,\n        runId: runId ?? null,\n        message: message as JsonValue,\n        createdAt: new Date(createdAt),\n    };\n}\n\nexport function fromStreamRow(row: StreamRow): StreamRecord {\n    return {\n        runId: row.runId,\n        status: row.status,\n        createdAt: row.createdAt.getTime(),\n        updatedAt: row.updatedAt.getTime(),\n        ...(row.closedAt ? { closedAt: row.closedAt.getTime() } : {}),\n    };\n}\n\nexport function toStreamRow(id: string, record: StreamRecord): StreamRow {\n    return {\n        id,\n        runId: record.runId,\n        status: record.status,\n        createdAt: new Date(record.createdAt),\n        updatedAt: new Date(record.updatedAt),\n        closedAt: record.closedAt === undefined ? null : new Date(record.closedAt),\n    };\n}\n\nexport function fromStreamEventRow(row: StreamEventRow): StreamEventRecord {\n    return {\n        id: row.id,\n        runId: row.runId,\n        seq: row.seq,\n        timestamp: row.timestamp.getTime(),\n        event: row.event as unknown as StreamEventRecord[\"event\"],\n    };\n}\n\nexport function toStreamEventRow(id: string, record: StreamEventRecord): StreamEventRow {\n    return {\n        id: record.id || id,\n        runId: record.runId,\n        seq: record.seq,\n        timestamp: new Date(record.timestamp),\n        event: record.event as unknown as JsonValue,\n    };\n}\n","import type { BetterAgentStorage, StoreListQuery, StoreListResult } from \"@better-agent/core\";\nimport { storageTables, throwUnsupportedStorageTable } from \"@better-agent/core\";\nimport { selectBetterAgentSchema, toBetterAgentOrmListQuery } from \"@better-agent/schema\";\nimport type { ModelClient, ModelName, OrmClient } from \"@farming-labs/orm\";\nimport { createDrizzleOrm } from \"../orm\";\nimport type { DrizzleOptions } from \"../types\";\nimport {\n    fromMessageRow,\n    fromRunRow,\n    fromStreamEventRow,\n    fromStreamRow,\n    fromThreadRow,\n    toMessageRow,\n    toRunRow,\n    toStreamEventRow,\n    toStreamRow,\n    toThreadRow,\n} from \"./mappers\";\n\nconst drizzleSchema = selectBetterAgentSchema();\n\ntype DrizzleSchema = typeof drizzleSchema;\ntype DrizzleModelName = ModelName<DrizzleSchema>;\ntype DrizzleOrm = OrmClient<DrizzleSchema>;\n\ntype TableAdapter = {\n    get(id: string): Promise<unknown | undefined>;\n    set(id: string, value: unknown): Promise<void>;\n    delete(id: string): Promise<void>;\n    list(query?: StoreListQuery): Promise<StoreListResult<unknown>>;\n};\n\nasync function upsertRow<TModelName extends DrizzleModelName, TRow>(input: {\n    model: ModelClient<DrizzleSchema, TModelName>;\n    keyField: string;\n    id: string;\n    row: TRow;\n}): Promise<void> {\n    type FindUniqueArgs = Parameters<typeof input.model.findUnique>[0];\n    type UpdateArgs = Parameters<typeof input.model.update>[0];\n    type UpsertArgs = Parameters<typeof input.model.upsert>[0];\n\n    const where = { [input.keyField]: input.id };\n    const existing = await input.model.findUnique({ where } as FindUniqueArgs);\n\n    if (!existing) {\n        await input.model.upsert({\n            where,\n            create: input.row,\n            update: input.row,\n        } as UpsertArgs);\n        return;\n    }\n\n    await input.model.update({\n        where,\n        data: input.row,\n    } as UpdateArgs);\n}\n\nfunction createTableAdapter<TModelName extends DrizzleModelName, TRecord, TRow>(input: {\n    model: ModelClient<DrizzleSchema, TModelName>;\n    keyField: string;\n    fromRow(row: TRow): TRecord;\n    toRow(id: string, record: TRecord): TRow;\n}): TableAdapter {\n    return {\n        async get(id) {\n            type FindUniqueArgs = Parameters<typeof input.model.findUnique>[0];\n\n            const row = (await input.model.findUnique({\n                where: { [input.keyField]: id },\n            } as FindUniqueArgs)) as TRow | null;\n\n            return row ? input.fromRow(row) : undefined;\n        },\n        async set(id, value) {\n            await upsertRow({\n                model: input.model,\n                keyField: input.keyField,\n                id,\n                row: input.toRow(id, value as TRecord),\n            });\n        },\n        async delete(id) {\n            type DeleteManyArgs = Parameters<typeof input.model.deleteMany>[0];\n\n            await input.model.deleteMany({\n                where: { [input.keyField]: id },\n            } as DeleteManyArgs);\n        },\n        async list(query) {\n            const rows = (await input.model.findMany(toBetterAgentOrmListQuery(query))) as TRow[];\n            return {\n                items: rows.map(input.fromRow),\n            };\n        },\n    };\n}\n\nfunction resolveAdapter(adapters: ReadonlyMap<string, TableAdapter>, table: string): TableAdapter {\n    return adapters.get(table) ?? throwUnsupportedStorageTable(table, \"drizzle storage\");\n}\n\nfunction requireModel<TModelName extends DrizzleModelName>(\n    orm: DrizzleOrm,\n    modelName: TModelName,\n): ModelClient<DrizzleSchema, TModelName> | undefined {\n    return orm[modelName];\n}\n\nexport function drizzleStorage(options: DrizzleOptions): BetterAgentStorage {\n    const orm = createDrizzleOrm(drizzleSchema, options) as DrizzleOrm;\n    const adapters = new Map<string, TableAdapter>();\n\n    const runModel = requireModel(orm, \"run\");\n    if (runModel) {\n        adapters.set(\n            storageTables.runs,\n            createTableAdapter({\n                model: runModel,\n                keyField: \"runId\",\n                fromRow: fromRunRow,\n                toRow: toRunRow,\n            }),\n        );\n    }\n\n    const threadModel = requireModel(orm, \"thread\");\n    if (threadModel) {\n        adapters.set(\n            storageTables.memoryThreads,\n            createTableAdapter({\n                model: threadModel,\n                keyField: \"id\",\n                fromRow: fromThreadRow,\n                toRow: toThreadRow,\n            }),\n        );\n    }\n\n    const messageModel = requireModel(orm, \"message\");\n    if (messageModel) {\n        adapters.set(\n            storageTables.memoryMessages,\n            createTableAdapter({\n                model: messageModel,\n                keyField: \"id\",\n                fromRow: fromMessageRow,\n                toRow: toMessageRow,\n            }),\n        );\n    }\n\n    const streamModel = requireModel(orm, \"stream\");\n    if (streamModel) {\n        adapters.set(\n            storageTables.streams,\n            createTableAdapter({\n                model: streamModel,\n                keyField: \"id\",\n                fromRow: fromStreamRow,\n                toRow: toStreamRow,\n            }),\n        );\n    }\n\n    const streamEventModel = requireModel(orm, \"streamEvent\");\n    if (streamEventModel) {\n        adapters.set(\n            storageTables.streamEvents,\n            createTableAdapter({\n                model: streamEventModel,\n                keyField: \"id\",\n                fromRow: fromStreamEventRow,\n                toRow: toStreamEventRow,\n            }),\n        );\n    }\n\n    return {\n        get<T = unknown>(table: string, id: string) {\n            return resolveAdapter(adapters, table).get(id) as Promise<T | undefined>;\n        },\n        set<T = unknown>(table: string, id: string, value: T) {\n            return resolveAdapter(adapters, table).set(id, value);\n        },\n        delete(table: string, id: string) {\n            return resolveAdapter(adapters, table).delete(id);\n        },\n        list<T = unknown>(table: string, query?: StoreListQuery) {\n            return resolveAdapter(adapters, table).list(query) as Promise<StoreListResult<T>>;\n        },\n    };\n}\n"],"mappings":";;;;;;AAKA,SAAgB,iBAEd,QAAiB,SAA6C;AAC5D,QAAO,UAAU;EACb;EACA,QAAQ,oBAAoB;GACxB,IAAI,QAAQ;GACZ,SAAS,QAAQ;GACpB,CAAC;EACL,CAAC;;;;;ACEN,SAAgB,WAAW,KAAwB;AAC/C,QAAO;EACH,OAAO,IAAI;EACX,WAAW,IAAI;EACf,GAAI,IAAI,WAAW,EAAE,UAAU,IAAI,UAAU,GAAG,EAAE;EAClD,GAAI,IAAI,QAAQ,EAAE,OAAO,IAAI,OAAO,GAAG,EAAE;EACzC,QAAQ,IAAI;EACZ,GAAI,IAAI,mBAAmB,EAAE,kBAAkB,IAAI,iBAAiB,SAAS,EAAE,GAAG,EAAE;EACpF,GAAI,IAAI,aACF,EAAE,YAAY,IAAI,YAAkD,GACpE,EAAE;EACR,QAAQ,IAAI,UAAU;EACtB,WAAW,IAAI,UAAU,SAAS;EAClC,WAAW,IAAI,UAAU,SAAS;EAClC,GAAI,IAAI,aAAa,EAAE,YAAY,IAAI,WAAW,SAAS,EAAE,GAAG,EAAE;EACrE;;AAGL,SAAgB,SAAS,IAAY,QAA2B;AAC5D,QAAO;EACH,OAAO,OAAO,SAAS;EACvB,WAAW,OAAO;EAClB,UAAU,OAAO,YAAY;EAC7B,OAAO,OAAO,SAAS;EACvB,QAAQ,OAAO;EACf,kBACI,OAAO,qBAAqB,SAAY,OAAO,IAAI,KAAK,OAAO,iBAAiB;EACpF,YAAa,OAAO,cAAc;EAClC,QAAS,OAAO,UAAU;EAC1B,WAAW,IAAI,KAAK,OAAO,UAAU;EACrC,WAAW,IAAI,KAAK,OAAO,UAAU;EACrC,YAAY,OAAO,eAAe,SAAY,OAAO,IAAI,KAAK,OAAO,WAAW;EACnF;;AAGL,SAAgB,cAAc,KAAoC;AAC9D,QAAO;EACH,IAAI,IAAI;EACR,WAAW,IAAI,aAAa;EAC5B,GAAI,IAAI,QAAQ,EAAE,OAAO,IAAI,OAAO,GAAG,EAAE;EACzC,GAAI,IAAI,QAAQ,EAAE,OAAO,IAAI,OAAO,GAAG,EAAE;EACzC,GAAI,IAAI,UAAU,OAAO,EAAE,OAAO,IAAI,OAAO,GAAG,EAAE;EAClD,GAAI,IAAI,YAAY,OAAO,IAAI,aAAa,YAAY,CAAC,MAAM,QAAQ,IAAI,SAAS,GAC9E,EAAE,UAAU,IAAI,UAAqC,GACrD,EAAE;EACR,WAAW,IAAI,UAAU,SAAS;EAClC,WAAW,IAAI,UAAU,SAAS;EACrC;;AAGL,SAAgB,YAAY,IAAY,QAAuC;AAC3E,QAAO;EACH,IAAI,OAAO,MAAM;EACjB,WAAW,OAAO;EAClB,OAAO,OAAO,SAAS;EACvB,OAAO,OAAO,SAAS;EACvB,OAAQ,OAAO,SAAS;EACxB,UAAW,OAAO,YAAY;EAC9B,WAAW,IAAI,KAAK,OAAO,UAAU;EACrC,WAAW,IAAI,KAAK,OAAO,UAAU;EACxC;;AAGL,SAAgB,eAAe,KAAsC;AAQjE,QAPgB;EACZ,GAAI,IAAI;EACR,UAAU,IAAI;EACd,GAAI,IAAI,QAAQ,EAAE,OAAO,IAAI,OAAO,GAAG,EAAE;EACzC,WAAW,IAAI,UAAU,SAAS;EACrC;;AAKL,SAAgB,aAAa,IAAY,QAAyC;CAC9E,MAAM,EAAE,UAAU,OAAO,WAAW,GAAG,YAAY;AAEnD,QAAO;EACH;EACA;EACA,OAAO,SAAS;EACP;EACT,WAAW,IAAI,KAAK,UAAU;EACjC;;AAGL,SAAgB,cAAc,KAA8B;AACxD,QAAO;EACH,OAAO,IAAI;EACX,QAAQ,IAAI;EACZ,WAAW,IAAI,UAAU,SAAS;EAClC,WAAW,IAAI,UAAU,SAAS;EAClC,GAAI,IAAI,WAAW,EAAE,UAAU,IAAI,SAAS,SAAS,EAAE,GAAG,EAAE;EAC/D;;AAGL,SAAgB,YAAY,IAAY,QAAiC;AACrE,QAAO;EACH;EACA,OAAO,OAAO;EACd,QAAQ,OAAO;EACf,WAAW,IAAI,KAAK,OAAO,UAAU;EACrC,WAAW,IAAI,KAAK,OAAO,UAAU;EACrC,UAAU,OAAO,aAAa,SAAY,OAAO,IAAI,KAAK,OAAO,SAAS;EAC7E;;AAGL,SAAgB,mBAAmB,KAAwC;AACvE,QAAO;EACH,IAAI,IAAI;EACR,OAAO,IAAI;EACX,KAAK,IAAI;EACT,WAAW,IAAI,UAAU,SAAS;EAClC,OAAO,IAAI;EACd;;AAGL,SAAgB,iBAAiB,IAAY,QAA2C;AACpF,QAAO;EACH,IAAI,OAAO,MAAM;EACjB,OAAO,OAAO;EACd,KAAK,OAAO;EACZ,WAAW,IAAI,KAAK,OAAO,UAAU;EACrC,OAAO,OAAO;EACjB;;;;;ACzHL,MAAM,gBAAgB,yBAAyB;AAa/C,eAAe,UAAqD,OAKlD;CAKd,MAAM,QAAQ,GAAG,MAAM,WAAW,MAAM,IAAI;AAG5C,KAAI,CAFa,MAAM,MAAM,MAAM,WAAW,EAAE,OAAO,CAAmB,EAE3D;AACX,QAAM,MAAM,MAAM,OAAO;GACrB;GACA,QAAQ,MAAM;GACd,QAAQ,MAAM;GACjB,CAAe;AAChB;;AAGJ,OAAM,MAAM,MAAM,OAAO;EACrB;EACA,MAAM,MAAM;EACf,CAAe;;AAGpB,SAAS,mBAAuE,OAK/D;AACb,QAAO;EACH,MAAM,IAAI,IAAI;GAGV,MAAM,MAAO,MAAM,MAAM,MAAM,WAAW,EACtC,OAAO,GAAG,MAAM,WAAW,IAAI,EAClC,CAAmB;AAEpB,UAAO,MAAM,MAAM,QAAQ,IAAI,GAAG;;EAEtC,MAAM,IAAI,IAAI,OAAO;AACjB,SAAM,UAAU;IACZ,OAAO,MAAM;IACb,UAAU,MAAM;IAChB;IACA,KAAK,MAAM,MAAM,IAAI,MAAiB;IACzC,CAAC;;EAEN,MAAM,OAAO,IAAI;AAGb,SAAM,MAAM,MAAM,WAAW,EACzB,OAAO,GAAG,MAAM,WAAW,IAAI,EAClC,CAAmB;;EAExB,MAAM,KAAK,OAAO;AAEd,UAAO,EACH,QAFU,MAAM,MAAM,MAAM,SAAS,0BAA0B,MAAM,CAAC,EAE1D,IAAI,MAAM,QAAQ,EACjC;;EAER;;AAGL,SAAS,eAAe,UAA6C,OAA6B;AAC9F,QAAO,SAAS,IAAI,MAAM,IAAI,6BAA6B,OAAO,kBAAkB;;AAGxF,SAAS,aACL,KACA,WACkD;AAClD,QAAO,IAAI;;AAGf,SAAgB,eAAe,SAA6C;CACxE,MAAM,MAAM,iBAAiB,eAAe,QAAQ;CACpD,MAAM,2BAAW,IAAI,KAA2B;CAEhD,MAAM,WAAW,aAAa,KAAK,MAAM;AACzC,KAAI,SACA,UAAS,IACL,cAAc,MACd,mBAAmB;EACf,OAAO;EACP,UAAU;EACV,SAAS;EACT,OAAO;EACV,CAAC,CACL;CAGL,MAAM,cAAc,aAAa,KAAK,SAAS;AAC/C,KAAI,YACA,UAAS,IACL,cAAc,eACd,mBAAmB;EACf,OAAO;EACP,UAAU;EACV,SAAS;EACT,OAAO;EACV,CAAC,CACL;CAGL,MAAM,eAAe,aAAa,KAAK,UAAU;AACjD,KAAI,aACA,UAAS,IACL,cAAc,gBACd,mBAAmB;EACf,OAAO;EACP,UAAU;EACV,SAAS;EACT,OAAO;EACV,CAAC,CACL;CAGL,MAAM,cAAc,aAAa,KAAK,SAAS;AAC/C,KAAI,YACA,UAAS,IACL,cAAc,SACd,mBAAmB;EACf,OAAO;EACP,UAAU;EACV,SAAS;EACT,OAAO;EACV,CAAC,CACL;CAGL,MAAM,mBAAmB,aAAa,KAAK,cAAc;AACzD,KAAI,iBACA,UAAS,IACL,cAAc,cACd,mBAAmB;EACf,OAAO;EACP,UAAU;EACV,SAAS;EACT,OAAO;EACV,CAAC,CACL;AAGL,QAAO;EACH,IAAiB,OAAe,IAAY;AACxC,UAAO,eAAe,UAAU,MAAM,CAAC,IAAI,GAAG;;EAElD,IAAiB,OAAe,IAAY,OAAU;AAClD,UAAO,eAAe,UAAU,MAAM,CAAC,IAAI,IAAI,MAAM;;EAEzD,OAAO,OAAe,IAAY;AAC9B,UAAO,eAAe,UAAU,MAAM,CAAC,OAAO,GAAG;;EAErD,KAAkB,OAAe,OAAwB;AACrD,UAAO,eAAe,UAAU,MAAM,CAAC,KAAK,MAAM;;EAEzD"}

package/package.json

@@ -0,0 +1,64 @@
+{
+    "name": "@better-agent/drizzle",
+    "version": "0.2.0-beta.1",
+    "description": "Better Agent Drizzle adapter",
+    "license": "MIT",
+    "repository": {
+        "type": "git",
+        "url": "git+https://github.com/better-agent/better-agent.git",
+        "directory": "packages/drizzle"
+    },
+    "homepage": "https://github.com/better-agent/better-agent/tree/main/packages/drizzle",
+    "bugs": {
+        "url": "https://github.com/better-agent/better-agent/issues"
+    },
+    "engines": {
+        "bun": ">=1.3.3",
+        "node": ">=20"
+    },
+    "type": "module",
+    "types": "./dist/index.d.mts",
+    "sideEffects": false,
+    "exports": {
+        ".": {
+            "types": "./dist/index.d.mts",
+            "import": "./dist/index.mjs",
+            "default": "./dist/index.mjs"
+        }
+    },
+    "bin": {
+        "better-agent-drizzle": "./dist/cli.mjs"
+    },
+    "scripts": {
+        "build": "bunx tsdown",
+        "dev": "bunx tsdown --watch --no-clean",
+        "lint:package": "bunx publint run --strict",
+        "lint:types": "npm_config_cache=../../.npm-cache bunx attw --profile esm-only --pack .",
+        "lint": "bunx biome check .",
+        "typecheck": "bunx tsc -p tsconfig.json --noEmit",
+        "format": "bunx biome format --write .",
+        "clean": "bunx rimraf dist"
+    },
+    "files": ["dist"],
+    "publishConfig": {
+        "access": "public",
+        "executableFiles": ["./dist/cli.mjs"]
+    },
+    "dependencies": {
+        "@better-agent/schema": "0.2.0-beta.1",
+        "@clack/prompts": "^1.2.0",
+        "@farming-labs/orm": "*",
+        "@farming-labs/orm-drizzle": "^0.0.61",
+        "commander": "^14.0.3"
+    },
+    "devDependencies": {
+        "@better-agent/core": "0.2.0-beta.1",
+        "@types/bun": "^1.3.12",
+        "@types/node": "^22.19.17",
+        "drizzle-orm": "*"
+    },
+    "peerDependencies": {
+        "@better-agent/core": "0.2.0-beta.1",
+        "drizzle-orm": "*"
+    }
+}