@neon/tools 0.0.0 → 0.3.0

package/dist/index.d.ts

@@ -0,0 +1,56 @@
+import { type NeonConfig, NeonError } from "@neon/sdk";
+import { type NeonBearerCredential } from "./lib/auth.js";
+import { type InjectedNeonTool, type NeonToolCustomizeOptions, type NeonToolInjectOptions, type NeonToolNameOverrides } from "./lib/customize.js";
+import { type NeonOperationId, operationFactories, operationIds } from "./operations.gen.js";
+export type { NeonBearerCredential } from "./lib/auth.js";
+export type { InjectedNeonTool, NeonToolCustomizeOptions, NeonToolDescriptionOverrides, NeonToolDescriptionSource, NeonToolInjectOptions, NeonToolInjectValue, NeonToolNameOverrides, NeonToolNameSource, NeonToolOnExecute, NeonToolOnExecuteEvent, } from "./lib/customize.js";
+export type { JsonSafe, JsonSafeBlob, NeonTool, NeonToolAnnotations, NeonToolExecutionContext, NeonToolMetadata, NeonToolResult, } from "./lib/operation.js";
+export type { NeonOperationId };
+export { NeonError, operationIds };
+type OperationFactories = typeof operationFactories;
+export type NeonTools<Operations extends readonly NeonOperationId[]> = {
+    [Operation in Operations[number]]: ReturnType<OperationFactories[Operation]>;
+};
+export type InjectedNeonTools<Operations extends readonly NeonOperationId[], Inject> = {
+    [Operation in Operations[number]]: InjectedNeonTool<ReturnType<OperationFactories[Operation]>, Inject>;
+};
+type WithPublishedId<Tool> = Tool extends {
+    id: string;
+} ? Omit<Tool, "id"> & {
+    id: string;
+} : Tool;
+export interface NeonToolsClientOptions extends Pick<NeonConfig, "baseUrl" | "fetch">, Omit<NeonToolCustomizeOptions, "name" | "names"> {
+    apiKey?: NeonBearerCredential;
+}
+export interface CreateNeonToolsOptions<Operations extends readonly NeonOperationId[]> extends NeonToolsClientOptions {
+    operations: Operations;
+}
+type NamedNeonTools<Operations extends readonly NeonOperationId[], Inject> = {
+    [Operation in Operations[number]]: WithPublishedId<Inject extends NeonToolInjectOptions ? InjectedNeonTool<ReturnType<OperationFactories[Operation]>, Inject> : ReturnType<OperationFactories[Operation]>>;
+};
+type NamedNeonTool<Operation extends NeonOperationId, Inject> = WithPublishedId<Inject extends NeonToolInjectOptions ? InjectedNeonTool<ReturnType<OperationFactories[Operation]>, Inject> : ReturnType<OperationFactories[Operation]>>;
+export declare function createNeonTools<const Operations extends readonly NeonOperationId[], const Inject extends NeonToolInjectOptions | undefined = undefined>(options: CreateNeonToolsOptions<Operations> & {
+    inject?: Inject;
+} & ({
+    name: (id: string) => string;
+    names?: NeonToolNameOverrides;
+} | {
+    names: NeonToolNameOverrides;
+    name?: (id: string) => string;
+})): NamedNeonTools<Operations, Inject>;
+export declare function createNeonTools<const Operations extends readonly NeonOperationId[], const Inject extends NeonToolInjectOptions | undefined = undefined>(options: CreateNeonToolsOptions<Operations> & {
+    inject?: Inject;
+}): Inject extends NeonToolInjectOptions ? InjectedNeonTools<Operations, Inject> : NeonTools<Operations>;
+export declare function createNeonTool<const Operation extends NeonOperationId, const Inject extends NeonToolInjectOptions | undefined = undefined>(operationId: Operation, options: NeonToolsClientOptions & {
+    inject?: Inject;
+} & ({
+    name: (id: string) => string;
+    names?: NeonToolNameOverrides;
+} | {
+    names: NeonToolNameOverrides;
+    name?: (id: string) => string;
+})): NamedNeonTool<Operation, Inject>;
+export declare function createNeonTool<const Operation extends NeonOperationId, const Inject extends NeonToolInjectOptions | undefined = undefined>(operationId: Operation, options: NeonToolsClientOptions & {
+    inject?: Inject;
+}): Inject extends NeonToolInjectOptions ? InjectedNeonTool<ReturnType<OperationFactories[Operation]>, Inject> : ReturnType<OperationFactories[Operation]>;
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAoB,KAAK,UAAU,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACzE,OAAO,EAEN,KAAK,oBAAoB,EAEzB,MAAM,eAAe,CAAC;AACvB,OAAO,EAIN,KAAK,gBAAgB,EACrB,KAAK,wBAAwB,EAC7B,KAAK,qBAAqB,EAC1B,KAAK,qBAAqB,EAC1B,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACN,KAAK,eAAe,EACpB,kBAAkB,EAClB,YAAY,EACZ,MAAM,qBAAqB,CAAC;AAE7B,YAAY,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAC1D,YAAY,EACX,gBAAgB,EAChB,wBAAwB,EACxB,4BAA4B,EAC5B,yBAAyB,EACzB,qBAAqB,EACrB,mBAAmB,EACnB,qBAAqB,EACrB,kBAAkB,EAClB,iBAAiB,EACjB,sBAAsB,GACtB,MAAM,oBAAoB,CAAC;AAC5B,YAAY,EACX,QAAQ,EACR,YAAY,EACZ,QAAQ,EACR,mBAAmB,EACnB,wBAAwB,EACxB,gBAAgB,EAChB,cAAc,GACd,MAAM,oBAAoB,CAAC;AAC5B,YAAY,EAAE,eAAe,EAAE,CAAC;AAChC,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,CAAC;AAEnC,KAAK,kBAAkB,GAAG,OAAO,kBAAkB,CAAC;AAEpD,MAAM,MAAM,SAAS,CAAC,UAAU,SAAS,SAAS,eAAe,EAAE,IAAI;KACrE,SAAS,IAAI,UAAU,CAAC,MAAM,CAAC,GAAG,UAAU,CAC5C,kBAAkB,CAAC,SAAS,CAAC,CAC7B;CACD,CAAC;AAEF,MAAM,MAAM,iBAAiB,CAC5B,UAAU,SAAS,SAAS,eAAe,EAAE,EAC7C,MAAM,IACH;KACF,SAAS,IAAI,UAAU,CAAC,MAAM,CAAC,GAAG,gBAAgB,CAClD,UAAU,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,EACzC,MAAM,CACN;CACD,CAAC;AAEF,KAAK,eAAe,CAAC,IAAI,IAAI,IAAI,SAAS;IAAE,EAAE,EAAE,MAAM,CAAA;CAAE,GACrD,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG;IAAE,EAAE,EAAE,MAAM,CAAA;CAAE,GACjC,IAAI,CAAC;AAER,MAAM,WAAW,sBAChB,SAAQ,IAAI,CAAC,UAAU,EAAE,SAAS,GAAG,OAAO,CAAC,EAC5C,IAAI,CAAC,wBAAwB,EAAE,MAAM,GAAG,OAAO,CAAC;IACjD,MAAM,CAAC,EAAE,oBAAoB,CAAC;CAC9B;AAED,MAAM,WAAW,sBAAsB,CACtC,UAAU,SAAS,SAAS,eAAe,EAAE,CAC5C,SAAQ,sBAAsB;IAC/B,UAAU,EAAE,UAAU,CAAC;CACvB;AAqGD,KAAK,cAAc,CAAC,UAAU,SAAS,SAAS,eAAe,EAAE,EAAE,MAAM,IAAI;KAC3E,SAAS,IAAI,UAAU,CAAC,MAAM,CAAC,GAAG,eAAe,CACjD,MAAM,SAAS,qBAAqB,GACjC,gBAAgB,CAChB,UAAU,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,EACzC,MAAM,CACN,GACA,UAAU,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,CAC5C;CACD,CAAC;AAEF,KAAK,aAAa,CAAC,SAAS,SAAS,eAAe,EAAE,MAAM,IAAI,eAAe,CAC9E,MAAM,SAAS,qBAAqB,GACjC,gBAAgB,CAAC,UAAU,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,EAAE,MAAM,CAAC,GACnE,UAAU,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,CAC5C,CAAC;AAEF,wBAAgB,eAAe,CAC9B,KAAK,CAAC,UAAU,SAAS,SAAS,eAAe,EAAE,EACnD,KAAK,CAAC,MAAM,SAAS,qBAAqB,GAAG,SAAS,GAAG,SAAS,EAElE,OAAO,EAAE,sBAAsB,CAAC,UAAU,CAAC,GAAG;IAC7C,MAAM,CAAC,EAAE,MAAM,CAAC;CAChB,GAAG,CACA;IAAE,IAAI,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,MAAM,CAAC;IAAC,KAAK,CAAC,EAAE,qBAAqB,CAAA;CAAE,GAC/D;IAAE,KAAK,EAAE,qBAAqB,CAAC;IAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,MAAM,CAAA;CAAE,CACjE,GACA,cAAc,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;AACtC,wBAAgB,eAAe,CAC9B,KAAK,CAAC,UAAU,SAAS,SAAS,eAAe,EAAE,EACnD,KAAK,CAAC,MAAM,SAAS,qBAAqB,GAAG,SAAS,GAAG,SAAS,EAElE,OAAO,EAAE,sBAAsB,CAAC,UAAU,CAAC,GAAG;IAAE,MAAM,CAAC,EAAE,MAAM,CAAA;CAAE,GAC/D,MAAM,SAAS,qBAAqB,GACpC,iBAAiB,CAAC,UAAU,EAAE,MAAM,CAAC,GACrC,SAAS,CAAC,UAAU,CAAC,CAAC;AAgBzB,wBAAgB,cAAc,CAC7B,KAAK,CAAC,SAAS,SAAS,eAAe,EACvC,KAAK,CAAC,MAAM,SAAS,qBAAqB,GAAG,SAAS,GAAG,SAAS,EAElE,WAAW,EAAE,SAAS,EACtB,OAAO,EAAE,sBAAsB,GAAG;IACjC,MAAM,CAAC,EAAE,MAAM,CAAC;CAChB,GAAG,CACA;IAAE,IAAI,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,MAAM,CAAC;IAAC,KAAK,CAAC,EAAE,qBAAqB,CAAA;CAAE,GAC/D;IAAE,KAAK,EAAE,qBAAqB,CAAC;IAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,MAAM,CAAA;CAAE,CACjE,GACA,aAAa,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;AACpC,wBAAgB,cAAc,CAC7B,KAAK,CAAC,SAAS,SAAS,eAAe,EACvC,KAAK,CAAC,MAAM,SAAS,qBAAqB,GAAG,SAAS,GAAG,SAAS,EAElE,WAAW,EAAE,SAAS,EACtB,OAAO,EAAE,sBAAsB,GAAG;IAAE,MAAM,CAAC,EAAE,MAAM,CAAA;CAAE,GACnD,MAAM,SAAS,qBAAqB,GACpC,gBAAgB,CAAC,UAAU,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,EAAE,MAAM,CAAC,GACnE,UAAU,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,CAAC"}

package/dist/index.js

@@ -0,0 +1,62 @@
+import { missingBearerCredential, requireBearerCredential } from "./lib/auth.js";
+import { applyToolCustomization, assertToolCustomizeOptions, hasToolCustomization } from "./lib/customize.js";
+import { operationFactories, operationIds } from "./operations.gen.js";
+import { NeonError, createNeonClient } from "@neon/sdk";
+//#region src/index.ts
+const createRawClient = (options) => createNeonClient({
+	apiKey: options.apiKey === void 0 ? missingBearerCredential : requireBearerCredential(options.apiKey),
+	...options.baseUrl === void 0 ? {} : { baseUrl: options.baseUrl },
+	...options.fetch === void 0 ? {} : { fetch: options.fetch }
+}).client;
+const operationFactoryFor = (operationId) => {
+	const knownOperationId = operationIds.find((candidate) => candidate === operationId);
+	if (knownOperationId === void 0) throw new TypeError(`Unknown Neon operation "${operationId}".`);
+	return operationFactories[knownOperationId];
+};
+const assertKnownNameKeys = (names, tools) => {
+	if (names === void 0 || typeof names === "function") return;
+	const known = new Set(tools.flatMap((tool) => [tool.operationId, tool.id]));
+	const unknown = Object.keys(names).filter((key) => !known.has(key));
+	if (unknown.length > 0) throw new TypeError(`Unknown Neon tool name override: ${unknown.join(", ")}`);
+};
+function assertNeonTools(value, operations) {
+	if (typeof value !== "object" || value === null) throw new TypeError("Expected generated Neon tools to be an object.");
+	for (const operationId of operations) if (!(operationId in value)) throw new TypeError(`Missing generated Neon tool "${operationId}".`);
+}
+const bindTools = (options) => {
+	assertToolCustomizeOptions(options);
+	const selected = /* @__PURE__ */ new Set();
+	const selectedFactories = options.operations.map((operationId) => {
+		if (selected.has(operationId)) throw new Error(`Duplicate Neon operation "${operationId}"`);
+		selected.add(operationId);
+		return [operationId, operationFactoryFor(operationId)];
+	});
+	const client = createRawClient(options);
+	const rawEntries = selectedFactories.map(([operationId, factory]) => {
+		return [operationId, factory(client)];
+	});
+	assertKnownNameKeys(options.names, rawEntries.map(([, tool]) => tool));
+	const entries = rawEntries.map(([operationId, tool]) => {
+		return [operationId, hasToolCustomization(options) ? applyToolCustomization(tool, options) : tool];
+	});
+	const publishedIds = /* @__PURE__ */ new Map();
+	for (const [operationId, tool] of entries) {
+		const previous = publishedIds.get(tool.id);
+		if (previous !== void 0) throw new Error(`Duplicate Neon tool id "${tool.id}" for ${previous}, ${operationId}`);
+		publishedIds.set(tool.id, operationId);
+	}
+	const tools = Object.fromEntries(entries);
+	assertNeonTools(tools, options.operations);
+	return tools;
+};
+function createNeonTools(options) {
+	return bindTools(options);
+}
+function createNeonTool(operationId, options) {
+	assertToolCustomizeOptions(options);
+	const tool = operationFactoryFor(operationId)(createRawClient(options));
+	assertKnownNameKeys(options.names, [tool]);
+	return hasToolCustomization(options) ? applyToolCustomization(tool, options) : tool;
+}
+//#endregion
+export { NeonError, createNeonTool, createNeonTools, operationIds };

package/dist/lib/auth.d.ts

@@ -0,0 +1,5 @@
+import type { NeonConfig } from "@neon/sdk";
+export type NeonBearerCredential = NeonConfig["apiKey"];
+export declare const missingBearerCredential: () => never;
+export declare const requireBearerCredential: (value: NeonBearerCredential) => NeonBearerCredential;
+//# sourceMappingURL=auth.d.ts.map

package/dist/lib/auth.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth.d.ts","sourceRoot":"","sources":["../../src/lib/auth.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AAE5C,MAAM,MAAM,oBAAoB,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC;AAKxD,eAAO,MAAM,uBAAuB,QAAO,KAE1C,CAAC;AASF,eAAO,MAAM,uBAAuB,GACnC,OAAO,oBAAoB,KACzB,oBAKF,CAAC"}

package/dist/lib/auth.js

@@ -0,0 +1,15 @@
+//#region src/lib/auth.ts
+const missingCredentialMessage = "A Neon API key or OAuth access token is required";
+const missingBearerCredential = () => {
+	throw new TypeError(missingCredentialMessage);
+};
+const requireNonEmptyCredential = (value) => {
+	if (typeof value !== "string" || value.length === 0) throw new TypeError(missingCredentialMessage);
+	return value;
+};
+const requireBearerCredential = (value) => {
+	if (typeof value === "function") return async () => requireNonEmptyCredential(await value());
+	return requireNonEmptyCredential(value);
+};
+//#endregion
+export { missingBearerCredential, requireBearerCredential };

package/dist/lib/binary.d.ts

@@ -0,0 +1,7 @@
+export declare const decodeBase64: (value: string) => Blob;
+export declare const encodeBlob: (value: Blob) => Promise<{
+    base64: string;
+    contentType: string;
+    size: number;
+}>;
+//# sourceMappingURL=binary.d.ts.map

package/dist/lib/binary.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"binary.d.ts","sourceRoot":"","sources":["../../src/lib/binary.ts"],"names":[],"mappings":"AAQA,eAAO,MAAM,YAAY,GAAI,OAAO,MAAM,KAAG,IAO5C,CAAC;AAEF,eAAO,MAAM,UAAU,GACtB,OAAO,IAAI,KACT,OAAO,CAAC;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,WAAW,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,CAI9D,CAAC"}

package/dist/lib/binary.js

@@ -0,0 +1,19 @@
+//#region src/lib/binary.ts
+const bytesToBase64 = (bytes) => {
+	let binary = "";
+	for (const byte of bytes) binary += String.fromCharCode(byte);
+	return btoa(binary);
+};
+const decodeBase64 = (value) => {
+	const binary = atob(value);
+	const bytes = new Uint8Array(binary.length);
+	for (let index = 0; index < binary.length; index += 1) bytes[index] = binary.charCodeAt(index);
+	return new Blob([bytes]);
+};
+const encodeBlob = async (value) => ({
+	base64: bytesToBase64(new Uint8Array(await value.arrayBuffer())),
+	contentType: value.type,
+	size: value.size
+});
+//#endregion
+export { decodeBase64, encodeBlob };

package/dist/lib/customize.d.ts

@@ -0,0 +1,52 @@
+import * as z from "zod";
+import type { NeonTool, NeonToolExecutionContext, NeonToolResult } from "./operation.js";
+export type NeonToolInjectValue = string | (() => string | undefined | Promise<string | undefined>);
+export interface NeonToolInjectOptions {
+    projectId?: NeonToolInjectValue;
+    branchId?: NeonToolInjectValue;
+    omitFromSchema?: boolean;
+}
+export interface NeonToolDescriptionSource {
+    operationId: string;
+    id: string;
+    title: string;
+    description: string;
+}
+export type NeonToolDescriptionOverrides = Readonly<Record<string, string>> | ((tool: NeonToolDescriptionSource) => string);
+export type NeonToolNameSource = Pick<NeonToolDescriptionSource, "operationId" | "id">;
+export type NeonToolNameOverrides = Readonly<Record<string, string>> | ((tool: NeonToolNameSource) => string);
+export interface NeonToolOnExecuteEvent<Output = unknown> {
+    operationId: string;
+    id: string;
+    input: unknown;
+    execute: () => Promise<NeonToolResult<Output>>;
+}
+export type NeonToolOnExecute = <Output>(event: NeonToolOnExecuteEvent<Output>) => Promise<NeonToolResult<Output>>;
+export interface NeonToolCustomizeOptions {
+    descriptions?: NeonToolDescriptionOverrides;
+    names?: NeonToolNameOverrides;
+    name?: (id: string) => string;
+    onExecute?: NeonToolOnExecute;
+    inject?: NeonToolInjectOptions;
+}
+type InjectedPathKey<Inject> = ("projectId" extends keyof Inject ? Inject["projectId"] extends undefined ? never : "project_id" : never) | ("branchId" extends keyof Inject ? Inject["branchId"] extends undefined ? never : "branch_id" : never);
+type RequiredKeys<Input> = {
+    [Key in keyof Input]-?: object extends Pick<Input, Key> ? never : Key;
+}[keyof Input];
+type InjectedRequiredKey<Input, Inject> = Extract<InjectedPathKey<Inject>, RequiredKeys<Input>>;
+type ApplyInjectToInput<Input, Inject> = [
+    InjectedRequiredKey<Input, Inject>
+] extends [never] ? Input : Inject extends {
+    omitFromSchema: true;
+} ? Omit<Input, InjectedRequiredKey<Input, Inject>> : Omit<Input, InjectedRequiredKey<Input, Inject>> & Partial<Pick<Input, Extract<InjectedRequiredKey<Input, Inject>, keyof Input>>>;
+export type InjectedNeonTool<Tool, Inject> = Tool extends NeonTool<infer Schema, string, infer Output> ? Omit<Tool, "execute"> & {
+    execute(input: ApplyInjectToInput<z.input<Schema>, Inject>, context?: NeonToolExecutionContext): Promise<NeonToolResult<Output>>;
+} : Tool;
+export declare const pathTemplateParams: (path: string) => Set<string>;
+export declare const assertToolCustomizeOptions: (options: NeonToolCustomizeOptions) => void;
+export declare const hasToolCustomization: (options: NeonToolCustomizeOptions) => boolean;
+export declare const applyToolCustomization: <Tool extends NeonTool>(tool: Tool, options: NeonToolCustomizeOptions) => Omit<Tool, "id"> & {
+    id: string;
+};
+export {};
+//# sourceMappingURL=customize.d.ts.map

package/dist/lib/customize.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"customize.d.ts","sourceRoot":"","sources":["../../src/lib/customize.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,CAAC,MAAM,KAAK,CAAC;AACzB,OAAO,KAAK,EACX,QAAQ,EACR,wBAAwB,EACxB,cAAc,EACd,MAAM,gBAAgB,CAAC;AAExB,MAAM,MAAM,mBAAmB,GAC5B,MAAM,GACN,CAAC,MAAM,MAAM,GAAG,SAAS,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC;AAE5D,MAAM,WAAW,qBAAqB;IACrC,SAAS,CAAC,EAAE,mBAAmB,CAAC;IAChC,QAAQ,CAAC,EAAE,mBAAmB,CAAC;IAC/B,cAAc,CAAC,EAAE,OAAO,CAAC;CACzB;AAED,MAAM,WAAW,yBAAyB;IACzC,WAAW,EAAE,MAAM,CAAC;IACpB,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,MAAM,4BAA4B,GACrC,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,GAChC,CAAC,CAAC,IAAI,EAAE,yBAAyB,KAAK,MAAM,CAAC,CAAC;AAEjD,MAAM,MAAM,kBAAkB,GAAG,IAAI,CACpC,yBAAyB,EACzB,aAAa,GAAG,IAAI,CACpB,CAAC;AAEF,MAAM,MAAM,qBAAqB,GAC9B,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,GAChC,CAAC,CAAC,IAAI,EAAE,kBAAkB,KAAK,MAAM,CAAC,CAAC;AAE1C,MAAM,WAAW,sBAAsB,CAAC,MAAM,GAAG,OAAO;IACvD,WAAW,EAAE,MAAM,CAAC;IACpB,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,OAAO,CAAC;IACf,OAAO,EAAE,MAAM,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC;CAC/C;AAED,MAAM,MAAM,iBAAiB,GAAG,CAAC,MAAM,EACtC,KAAK,EAAE,sBAAsB,CAAC,MAAM,CAAC,KACjC,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC;AAErC,MAAM,WAAW,wBAAwB;IACxC,YAAY,CAAC,EAAE,4BAA4B,CAAC;IAC5C,KAAK,CAAC,EAAE,qBAAqB,CAAC;IAC9B,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,MAAM,CAAC;IAC9B,SAAS,CAAC,EAAE,iBAAiB,CAAC;IAC9B,MAAM,CAAC,EAAE,qBAAqB,CAAC;CAC/B;AAED,KAAK,eAAe,CAAC,MAAM,IACxB,CAAC,WAAW,SAAS,MAAM,MAAM,GAC/B,MAAM,CAAC,WAAW,CAAC,SAAS,SAAS,GACpC,KAAK,GACL,YAAY,GACb,KAAK,CAAC,GACR,CAAC,UAAU,SAAS,MAAM,MAAM,GAC9B,MAAM,CAAC,UAAU,CAAC,SAAS,SAAS,GACnC,KAAK,GACL,WAAW,GACZ,KAAK,CAAC,CAAC;AAEZ,KAAK,YAAY,CAAC,KAAK,IAAI;KACzB,GAAG,IAAI,MAAM,KAAK,CAAC,CAAC,GAAG,MAAM,SAAS,IAAI,CAAC,KAAK,EAAE,GAAG,CAAC,GAAG,KAAK,GAAG,GAAG;CACrE,CAAC,MAAM,KAAK,CAAC,CAAC;AAEf,KAAK,mBAAmB,CAAC,KAAK,EAAE,MAAM,IAAI,OAAO,CAChD,eAAe,CAAC,MAAM,CAAC,EACvB,YAAY,CAAC,KAAK,CAAC,CACnB,CAAC;AAEF,KAAK,kBAAkB,CAAC,KAAK,EAAE,MAAM,IAAI;IACxC,mBAAmB,CAAC,KAAK,EAAE,MAAM,CAAC;CAClC,SAAS,CAAC,KAAK,CAAC,GACd,KAAK,GACL,MAAM,SAAS;IAAE,cAAc,EAAE,IAAI,CAAA;CAAE,GACtC,IAAI,CAAC,KAAK,EAAE,mBAAmB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,GAC/C,IAAI,CAAC,KAAK,EAAE,mBAAmB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,GAC/C,OAAO,CACN,IAAI,CACH,KAAK,EACL,OAAO,CAAC,mBAAmB,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,MAAM,KAAK,CAAC,CACxD,CACD,CAAC;AAEN,MAAM,MAAM,gBAAgB,CAAC,IAAI,EAAE,MAAM,IACxC,IAAI,SAAS,QAAQ,CAAC,MAAM,MAAM,EAAE,MAAM,EAAE,MAAM,MAAM,CAAC,GACtD,IAAI,CAAC,IAAI,EAAE,SAAS,CAAC,GAAG;IACxB,OAAO,CACN,KAAK,EAAE,kBAAkB,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC,EAClD,OAAO,CAAC,EAAE,wBAAwB,GAChC,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC;CACnC,GACA,IAAI,CAAC;AAsBT,eAAO,MAAM,kBAAkB,GAAI,MAAM,MAAM,KAAG,GAAG,CAAC,MAAM,CACU,CAAC;AAEvE,eAAO,MAAM,0BAA0B,GACtC,SAAS,wBAAwB,KAC/B,IAgBF,CAAC;AAEF,eAAO,MAAM,oBAAoB,GAChC,SAAS,wBAAwB,KAC/B,OAKkC,CAAC;AAEtC,eAAO,MAAM,sBAAsB,GAAI,IAAI,SAAS,QAAQ,EAC3D,MAAM,IAAI,EACV,SAAS,wBAAwB,KAC/B,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG;IAAE,EAAE,EAAE,MAAM,CAAA;CAoBjC,CAAC"}

package/dist/lib/customize.js

@@ -0,0 +1,158 @@
+import * as z from "zod";
+//#region src/lib/customize.ts
+const PATH_INJECT = [["projectId", "project_id"], ["branchId", "branch_id"]];
+const PUBLISHED_ID = /^[a-z][a-z0-9_]*$/;
+const missingInjectMessage = (name) => `A ${name} inject value is required`;
+const isPlainObject = (value) => typeof value === "object" && value !== null && !Array.isArray(value);
+const pathTemplateParams = (path) => new Set([...path.matchAll(/\{([^{}]+)\}/g)].map((match) => match[1]));
+const assertToolCustomizeOptions = (options) => {
+	resolveInject(options.inject);
+	if (options.name !== void 0 && typeof options.name !== "function") throw new TypeError("Neon tool name transforms must be functions");
+	if (options.names !== void 0 && typeof options.names !== "function" && (typeof options.names !== "object" || options.names === null || Array.isArray(options.names))) throw new TypeError("Neon tool name overrides must be a map or a function");
+};
+const hasToolCustomization = (options) => options.descriptions !== void 0 || options.names !== void 0 || options.name !== void 0 || options.onExecute !== void 0 || resolveInject(options.inject).hasAny;
+const applyToolCustomization = (tool, options) => {
+	const inject = resolveInject(options.inject);
+	const id = publishedId(tool, options);
+	const description = applyDescription(tool, options.descriptions);
+	const inputSchema = publishInputSchema(tool.inputSchema, inject, pathTemplateParams(tool.metadata.path));
+	const wrapExecute = options.onExecute !== void 0 || inject.hasAny;
+	return {
+		...tool,
+		id,
+		description,
+		inputSchema,
+		execute: wrapExecute ? wrapToolExecute({
+			...tool,
+			id
+		}, options.onExecute, inject) : tool.execute
+	};
+};
+function publishedId(tool, options) {
+	let id = tool.id;
+	if (options.names !== void 0) {
+		if (typeof options.names === "function") {
+			id = options.names({
+				operationId: tool.operationId,
+				id: tool.id
+			});
+			if (typeof id !== "string") throw new TypeError(`Neon tool name overrides must return a string for ${tool.operationId}`);
+		} else {
+			const override = options.names[tool.operationId] ?? options.names[tool.id];
+			if (override !== void 0) id = override;
+		}
+	}
+	if (options.name !== void 0) id = options.name(id);
+	if (typeof id !== "string" || !PUBLISHED_ID.test(id)) throw new TypeError(`Neon tool id must match ${PUBLISHED_ID} for ${tool.operationId}: received ${JSON.stringify(id)}`);
+	return id;
+}
+function resolveInject(inject) {
+	if (inject === void 0) return {
+		omitFromSchema: false,
+		hasAny: false
+	};
+	const projectId = requireInjectValue("projectId", inject.projectId);
+	const branchId = requireInjectValue("branchId", inject.branchId);
+	const hasAny = projectId !== void 0 || branchId !== void 0;
+	const omitFromSchema = inject.omitFromSchema === true;
+	if (omitFromSchema && !hasAny) throw new TypeError("omitFromSchema requires inject.projectId or inject.branchId");
+	return {
+		projectId,
+		branchId,
+		omitFromSchema,
+		hasAny
+	};
+}
+function requireInjectValue(name, value) {
+	if (value === void 0) return;
+	if (typeof value === "function") return value;
+	if (typeof value === "string") {
+		if (value.length === 0) throw new TypeError(missingInjectMessage(name));
+		return value;
+	}
+	throw new TypeError(missingInjectMessage(name));
+}
+function applyDescription(tool, descriptions) {
+	if (descriptions === void 0) return tool.description;
+	if (typeof descriptions === "function") {
+		const description = descriptions({
+			operationId: tool.operationId,
+			id: tool.id,
+			title: tool.title,
+			description: tool.description
+		});
+		if (typeof description !== "string") throw new TypeError("Neon tool description overrides must return a string");
+		return description;
+	}
+	const description = descriptions[tool.operationId] ?? descriptions[tool.id];
+	if (description === void 0) return tool.description;
+	if (typeof description !== "string") throw new TypeError("Neon tool description overrides must be strings");
+	return description;
+}
+function wrapToolExecute(tool, onExecute, inject) {
+	const originalExecute = tool.execute.bind(tool);
+	const availableKeys = pathTemplateParams(tool.metadata.path);
+	const mode = inject.omitFromSchema ? "override" : "fill";
+	return async (input, context) => {
+		const run = async () => originalExecute(mergeFields(input, await resolveInjectedFields(input, inject, availableKeys, mode), mode), context);
+		if (onExecute === void 0) return run();
+		return onExecute({
+			operationId: tool.operationId,
+			id: tool.id,
+			input: cloneInput(input),
+			execute: run
+		});
+	};
+}
+async function resolveInjectedFields(input, inject, availableKeys, mode) {
+	const present = isPlainObject(input) ? input : void 0;
+	const patch = {};
+	for (const [name, pathKey] of PATH_INJECT) {
+		if (!availableKeys.has(pathKey)) continue;
+		const injector = inject[name];
+		if (injector === void 0) continue;
+		if (mode === "fill" && present?.[pathKey] !== void 0) continue;
+		const value = typeof injector === "function" ? await injector() : injector;
+		if (value === void 0) {
+			if (mode === "override") throw new TypeError(missingInjectMessage(name));
+			continue;
+		}
+		if (typeof value !== "string" || value.length === 0) throw new TypeError(missingInjectMessage(name));
+		patch[pathKey] = value;
+	}
+	return patch;
+}
+function mergeFields(input, patch, mode) {
+	if (Object.keys(patch).length === 0) return input === void 0 ? {} : input;
+	if (input === void 0) return patch;
+	if (!isPlainObject(input)) return input;
+	const next = { ...input };
+	for (const [key, value] of Object.entries(patch)) if (mode === "override" || next[key] === void 0) next[key] = value;
+	return next;
+}
+function cloneInput(input) {
+	if (!isPlainObject(input)) return input;
+	return { ...input };
+}
+function objectShape(schema) {
+	if (schema instanceof z.ZodOptional) return objectShape(schema.unwrap());
+	if (schema instanceof z.ZodObject) return schema.shape;
+}
+function publishInputSchema(schema, inject, pathKeys) {
+	if (!inject.hasAny) return schema;
+	const shape = objectShape(schema);
+	if (shape === void 0) return schema;
+	const nextShape = { ...shape };
+	let changed = false;
+	for (const [name, pathKey] of PATH_INJECT) {
+		if (inject[name] === void 0 || !pathKeys.has(pathKey)) continue;
+		if (nextShape[pathKey] === void 0) continue;
+		changed = true;
+		if (inject.omitFromSchema) delete nextShape[pathKey];
+		else nextShape[pathKey] = nextShape[pathKey].optional();
+	}
+	if (!changed) return schema;
+	return Object.keys(nextShape).length === 0 ? z.strictObject({}) : z.strictObject(nextShape);
+}
+//#endregion
+export { applyToolCustomization, assertToolCustomizeOptions, hasToolCustomization, pathTemplateParams };

package/dist/lib/envelope.d.ts

@@ -0,0 +1,5 @@
+export declare function optionalGroup<Value extends Record<string, unknown>>(value: Value, required: true): Value;
+export declare function optionalGroup<Value extends Record<string, unknown>>(value: Value, required: false): Value | undefined;
+export declare function decodeBinaryFields(body: Record<string, unknown>, fields: readonly string[], decode: (value: string) => Blob): Record<string, unknown>;
+export declare function decodeBinaryFields(body: Record<string, unknown> | undefined, fields: readonly string[], decode: (value: string) => Blob): Record<string, unknown> | undefined;
+//# sourceMappingURL=envelope.d.ts.map

package/dist/lib/envelope.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"envelope.d.ts","sourceRoot":"","sources":["../../src/lib/envelope.ts"],"names":[],"mappings":"AAYA,wBAAgB,aAAa,CAAC,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAClE,KAAK,EAAE,KAAK,EACZ,QAAQ,EAAE,IAAI,GACZ,KAAK,CAAC;AACT,wBAAgB,aAAa,CAAC,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAClE,KAAK,EAAE,KAAK,EACZ,QAAQ,EAAE,KAAK,GACb,KAAK,GAAG,SAAS,CAAC;AAYrB,wBAAgB,kBAAkB,CACjC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,MAAM,EAAE,SAAS,MAAM,EAAE,EACzB,MAAM,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,GAC7B,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AAC3B,wBAAgB,kBAAkB,CACjC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,EACzC,MAAM,EAAE,SAAS,MAAM,EAAE,EACzB,MAAM,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,GAC7B,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAC"}

package/dist/lib/envelope.js

@@ -0,0 +1,22 @@
+//#region src/lib/envelope.ts
+const defined = (value) => {
+	const next = {};
+	for (const [key, child] of Object.entries(value)) if (child !== void 0) next[key] = child;
+	return next;
+};
+function optionalGroup(value, required) {
+	const next = defined(value);
+	if (!required && Object.keys(next).length === 0) return;
+	return next;
+}
+function decodeBinaryFields(body, fields, decode) {
+	if (body === void 0) return;
+	const next = { ...body };
+	for (const field of fields) {
+		const value = next[field];
+		if (typeof value === "string") next[field] = decode(value);
+	}
+	return next;
+}
+//#endregion
+export { decodeBinaryFields, optionalGroup };

package/dist/lib/operation.d.ts

@@ -0,0 +1,56 @@
+import { type Client } from "@neon/sdk/raw";
+import type * as z from "zod";
+import { type NeonBearerCredential } from "./auth.js";
+export interface NeonToolAnnotations {
+    readOnlyHint: boolean;
+    destructiveHint?: boolean;
+    idempotentHint?: boolean;
+    openWorldHint: boolean;
+}
+export interface NeonToolExecutionContext {
+    signal?: AbortSignal;
+    apiKey?: NeonBearerCredential;
+}
+export interface NeonToolMetadata {
+    method: string;
+    path: string;
+    stability: string;
+    deprecated: boolean;
+    tags: readonly string[];
+}
+export interface JsonSafeBlob {
+    base64: string;
+    contentType: string;
+    size: number;
+}
+export type JsonSafe<Value> = unknown extends Value ? unknown : Value extends null ? null : Value extends void ? null : Value extends bigint | Date ? string : Value extends Blob ? JsonSafeBlob : Value extends string | number | boolean ? Value : Value extends readonly (infer Item)[] ? JsonSafe<Item>[] : Value extends object ? {
+    [Key in keyof Value]: JsonSafe<Value[Key]>;
+} : unknown;
+export interface NeonToolResult<Data = unknown> {
+    data: Data;
+}
+export interface NeonTool<InputSchema extends z.ZodType = z.ZodType, Id extends string = string, Output = unknown> {
+    operationId: string;
+    id: Id;
+    title: string;
+    description: string;
+    inputSchema: InputSchema;
+    annotations: NeonToolAnnotations;
+    requiresApproval: boolean;
+    metadata: NeonToolMetadata;
+    execute(input: z.input<InputSchema>, context?: NeonToolExecutionContext): Promise<NeonToolResult<Output>>;
+}
+export interface NeonOperation<InputSchema extends z.ZodType, Id extends string = string, Output = unknown> {
+    operationId: string;
+    id: Id;
+    title: string;
+    description: string;
+    inputSchema: InputSchema;
+    annotations: NeonToolAnnotations;
+    requiresApproval: boolean;
+    metadata: NeonToolMetadata;
+    invoke(client: Client, input: z.output<InputSchema>, signal?: AbortSignal): Promise<Output>;
+}
+export declare const defineOperation: <const InputSchema extends z.ZodType, const Id extends string, Output>(operation: NeonOperation<InputSchema, Id, Output>) => NeonOperation<InputSchema, Id, Output>;
+export declare const bindOperation: <const InputSchema extends z.ZodType, const Id extends string, Output>(operation: NeonOperation<InputSchema, Id, Output>, client: Client) => NeonTool<InputSchema, Id, JsonSafe<Awaited<Output>>>;
+//# sourceMappingURL=operation.d.ts.map

package/dist/lib/operation.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"operation.d.ts","sourceRoot":"","sources":["../../src/lib/operation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,MAAM,EAAgB,MAAM,eAAe,CAAC;AAC1D,OAAO,KAAK,KAAK,CAAC,MAAM,KAAK,CAAC;AAC9B,OAAO,EAAE,KAAK,oBAAoB,EAA2B,MAAM,WAAW,CAAC;AAG/E,MAAM,WAAW,mBAAmB;IACnC,YAAY,EAAE,OAAO,CAAC;IACtB,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,aAAa,EAAE,OAAO,CAAC;CACvB;AAED,MAAM,WAAW,wBAAwB;IACxC,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,MAAM,CAAC,EAAE,oBAAoB,CAAC;CAC9B;AAED,MAAM,WAAW,gBAAgB;IAChC,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,OAAO,CAAC;IACpB,IAAI,EAAE,SAAS,MAAM,EAAE,CAAC;CACxB;AAED,MAAM,WAAW,YAAY;IAC5B,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;IACpB,IAAI,EAAE,MAAM,CAAC;CACb;AAED,MAAM,MAAM,QAAQ,CAAC,KAAK,IAAI,OAAO,SAAS,KAAK,GAChD,OAAO,GACP,KAAK,SAAS,IAAI,GACjB,IAAI,GACJ,KAAK,SAAS,IAAI,GACjB,IAAI,GACJ,KAAK,SAAS,MAAM,GAAG,IAAI,GAC1B,MAAM,GACN,KAAK,SAAS,IAAI,GACjB,YAAY,GACZ,KAAK,SAAS,MAAM,GAAG,MAAM,GAAG,OAAO,GACtC,KAAK,GACL,KAAK,SAAS,SAAS,CAAC,MAAM,IAAI,CAAC,EAAE,GACpC,QAAQ,CAAC,IAAI,CAAC,EAAE,GAChB,KAAK,SAAS,MAAM,GACnB;KAAG,GAAG,IAAI,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;CAAE,GAC9C,OAAO,CAAC;AAElB,MAAM,WAAW,cAAc,CAAC,IAAI,GAAG,OAAO;IAC7C,IAAI,EAAE,IAAI,CAAC;CACX;AAED,MAAM,WAAW,QAAQ,CACxB,WAAW,SAAS,CAAC,CAAC,OAAO,GAAG,CAAC,CAAC,OAAO,EACzC,EAAE,SAAS,MAAM,GAAG,MAAM,EAC1B,MAAM,GAAG,OAAO;IAEhB,WAAW,EAAE,MAAM,CAAC;IACpB,EAAE,EAAE,EAAE,CAAC;IACP,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,WAAW,CAAC;IACzB,WAAW,EAAE,mBAAmB,CAAC;IACjC,gBAAgB,EAAE,OAAO,CAAC;IAC1B,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,OAAO,CACN,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,WAAW,CAAC,EAC3B,OAAO,CAAC,EAAE,wBAAwB,GAChC,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC;CACnC;AAED,MAAM,WAAW,aAAa,CAC7B,WAAW,SAAS,CAAC,CAAC,OAAO,EAC7B,EAAE,SAAS,MAAM,GAAG,MAAM,EAC1B,MAAM,GAAG,OAAO;IAEhB,WAAW,EAAE,MAAM,CAAC;IACpB,EAAE,EAAE,EAAE,CAAC;IACP,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,WAAW,CAAC;IACzB,WAAW,EAAE,mBAAmB,CAAC;IACjC,gBAAgB,EAAE,OAAO,CAAC;IAC1B,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,MAAM,CACL,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,EAC5B,MAAM,CAAC,EAAE,WAAW,GAClB,OAAO,CAAC,MAAM,CAAC,CAAC;CACnB;AAED,eAAO,MAAM,eAAe,GAC3B,KAAK,CAAC,WAAW,SAAS,CAAC,CAAC,OAAO,EACnC,KAAK,CAAC,EAAE,SAAS,MAAM,EACvB,MAAM,EAEN,WAAW,aAAa,CAAC,WAAW,EAAE,EAAE,EAAE,MAAM,CAAC,2CACpC,CAAC;AAEf,eAAO,MAAM,aAAa,GACzB,KAAK,CAAC,WAAW,SAAS,CAAC,CAAC,OAAO,EACnC,KAAK,CAAC,EAAE,SAAS,MAAM,EACvB,MAAM,EAEN,WAAW,aAAa,CAAC,WAAW,EAAE,EAAE,EAAE,MAAM,CAAC,EACjD,QAAQ,MAAM,KACZ,QAAQ,CAAC,WAAW,EAAE,EAAE,EAAE,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAsBpD,CAAC"}

package/dist/lib/operation.js

@@ -0,0 +1,30 @@
+import { requireBearerCredential } from "./auth.js";
+import { toToolResult } from "./result.js";
+import { createClient } from "@neon/sdk/raw";
+//#region src/lib/operation.ts
+const defineOperation = (operation) => operation;
+const bindOperation = (operation, client) => ({
+	operationId: operation.operationId,
+	id: operation.id,
+	title: operation.title,
+	description: operation.description,
+	inputSchema: operation.inputSchema,
+	annotations: operation.annotations,
+	requiresApproval: operation.requiresApproval,
+	metadata: operation.metadata,
+	async execute(input, context) {
+		const parsed = await operation.inputSchema.parseAsync(input);
+		const activeClient = context !== void 0 && "apiKey" in context ? clientWithCredential(client, context.apiKey) : client;
+		const result = await operation.invoke(activeClient, parsed, context?.signal);
+		return toToolResult(result);
+	}
+});
+const clientWithCredential = (client, apiKey) => {
+	if (apiKey === void 0) throw new TypeError("A Neon API key or OAuth access token is required");
+	return createClient({
+		...client.getConfig(),
+		auth: requireBearerCredential(apiKey)
+	});
+};
+//#endregion
+export { bindOperation, defineOperation };

package/dist/lib/result.d.ts

@@ -0,0 +1,3 @@
+import type { JsonSafe, NeonToolResult } from "./operation.js";
+export declare const toToolResult: <Value>(value: Value) => Promise<NeonToolResult<JsonSafe<Value>>>;
+//# sourceMappingURL=result.d.ts.map

package/dist/lib/result.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"result.d.ts","sourceRoot":"","sources":["../../src/lib/result.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAsC/D,eAAO,MAAM,YAAY,GAAU,KAAK,EACvC,OAAO,KAAK,KACV,OAAO,CAAC,cAAc,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAExC,CAAC"}

package/dist/lib/result.js

@@ -0,0 +1,18 @@
+import { encodeBlob } from "./binary.js";
+//#region src/lib/result.ts
+async function toJsonSafe(value) {
+	if (value === void 0 || value === null) return null;
+	if (typeof value === "bigint") return value.toString();
+	if (typeof value === "string" || typeof value === "boolean" || typeof value === "number") return value;
+	if (value instanceof Blob) return encodeBlob(value);
+	if (value instanceof Date) return value.toISOString();
+	if (Array.isArray(value)) return Promise.all(value.map(toJsonSafe));
+	if (typeof value === "object") {
+		const entries = await Promise.all(Object.entries(value).map(async ([key, child]) => [key, await toJsonSafe(child)]));
+		return Object.fromEntries(entries);
+	}
+	throw new TypeError(`Unsupported tool result value: ${typeof value}`);
+}
+const toToolResult = async (value) => ({ data: await toJsonSafe(value) });
+//#endregion
+export { toToolResult };

package/dist/mastra.d.ts

@@ -0,0 +1,32 @@
+import type * as z from "zod";
+import type { NeonTool, NeonToolExecutionContext, NeonToolResult } from "./lib/operation.js";
+export interface MastraToolContext {
+    abortSignal?: AbortSignal;
+}
+export type MastraToolConfig<Tool extends NeonTool> = {
+    id: Tool["id"];
+    description: string;
+    inputSchema: Tool["inputSchema"];
+    requireApproval: boolean;
+    execute(input: Parameters<Tool["execute"]>[0], context: MastraToolContext): ReturnType<Tool["execute"]>;
+};
+export type MastraTools<Tools extends Readonly<Record<string, NeonTool>>> = {
+    [Tool in Tools[keyof Tools] as Tool["id"]]: MastraToolConfig<Tool>;
+};
+type MastraToolSource = {
+    id: string;
+    description: string;
+    inputSchema: z.ZodType;
+    requiresApproval: boolean;
+    execute: (input: never, context?: NeonToolExecutionContext) => Promise<NeonToolResult<unknown>>;
+};
+export declare const toMastraTool: <const Tool extends MastraToolSource>(tool: Tool) => {
+    id: string;
+    description: string;
+    inputSchema: z.ZodType<unknown, unknown, z.core.$ZodTypeInternals<unknown, unknown>>;
+    requireApproval: boolean;
+    execute: (input: Parameters<Tool["execute"]>[0], context: MastraToolContext) => ReturnType<Tool["execute"]>;
+};
+export declare const toMastraTools: <const Tools extends Readonly<Record<string, NeonTool>>>(tools: Tools) => MastraTools<Tools>;
+export {};
+//# sourceMappingURL=mastra.d.ts.map

package/dist/mastra.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mastra.d.ts","sourceRoot":"","sources":["../src/mastra.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,KAAK,CAAC,MAAM,KAAK,CAAC;AAC9B,OAAO,KAAK,EACX,QAAQ,EACR,wBAAwB,EACxB,cAAc,EACd,MAAM,oBAAoB,CAAC;AAE5B,MAAM,WAAW,iBAAiB;IACjC,WAAW,CAAC,EAAE,WAAW,CAAC;CAC1B;AAED,MAAM,MAAM,gBAAgB,CAAC,IAAI,SAAS,QAAQ,IAAI;IACrD,EAAE,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;IACjC,eAAe,EAAE,OAAO,CAAC;IACzB,OAAO,CACN,KAAK,EAAE,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,EACrC,OAAO,EAAE,iBAAiB,GACxB,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;CAC/B,CAAC;AAEF,MAAM,MAAM,WAAW,CAAC,KAAK,SAAS,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC,IAAI;KAC1E,IAAI,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,gBAAgB,CAAC,IAAI,CAAC;CAClE,CAAC;AAEF,KAAK,gBAAgB,GAAG;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,CAAC,CAAC,OAAO,CAAC;IACvB,gBAAgB,EAAE,OAAO,CAAC;IAC1B,OAAO,EAAE,CACR,KAAK,EAAE,KAAK,EACZ,OAAO,CAAC,EAAE,wBAAwB,KAC9B,OAAO,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC;CACtC,CAAC;AAEF,eAAO,MAAM,YAAY,GAAI,KAAK,CAAC,IAAI,SAAS,gBAAgB,EAC/D,MAAM,IAAI;;;;;qBAOF,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,WAC5B,iBAAiB,KACxB,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;CAI7B,CAAC;AAgBH,eAAO,MAAM,aAAa,GACzB,KAAK,CAAC,KAAK,SAAS,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC,EAEtD,OAAO,KAAK,KACV,WAAW,CAAC,KAAK,CAMnB,CAAC"}

package/dist/mastra.js

@@ -0,0 +1,19 @@
+//#region src/mastra.ts
+const toMastraTool = (tool) => ({
+	id: tool.id,
+	description: tool.description,
+	inputSchema: tool.inputSchema,
+	requireApproval: tool.requiresApproval,
+	execute: (input, context) => tool.execute(input, { signal: context.abortSignal })
+});
+function assertMastraTools(value, tools) {
+	if (typeof value !== "object" || value === null) throw new TypeError("Expected Mastra tools to be an object.");
+	for (const tool of Object.values(tools)) if (!(tool.id in value)) throw new TypeError(`Missing Mastra tool "${tool.id}".`);
+}
+const toMastraTools = (tools) => {
+	const adapted = Object.fromEntries(Object.values(tools).map((tool) => [tool.id, toMastraTool(tool)]));
+	assertMastraTools(adapted, tools);
+	return adapted;
+};
+//#endregion
+export { toMastraTool, toMastraTools };

package/dist/mcp-v1.d.ts

@@ -0,0 +1,3 @@
+export type { McpToolResult, McpToolServer } from "./mcp.js";
+export { registerNeonTools } from "./mcp.js";
+//# sourceMappingURL=mcp-v1.d.ts.map

package/dist/mcp-v1.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-v1.d.ts","sourceRoot":"","sources":["../src/mcp-v1.ts"],"names":[],"mappings":"AAAA,YAAY,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAC7D,OAAO,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC"}

package/dist/mcp-v1.js

@@ -0,0 +1,2 @@
+import { registerNeonTools } from "./mcp.js";
+export { registerNeonTools };