@neon/tools 0.5.0 → 0.6.0

package/dist/lib/ergonomic/bind.js

@@ -0,0 +1,121 @@
+import { missingBearerCredential, requireBearerCredential } from "../auth.js";
+import { toToolResult } from "../result.js";
+import { operationFactories } from "../../operations.gen.js";
+import { NeonError, createNeonClient } from "@neon/sdk";
+import * as z from "zod";
+//#region src/lib/ergonomic/bind.ts
+const ALL_PAGES = " This tool returns every page. Pass limit to cap how many items come back. Do not pass a cursor.";
+const inertClient = createNeonClient({ apiKey: "unused" }).client;
+const publishedId = (id) => id.split(".").map((segment) => segment.replace(/[A-Z]/g, (character) => `_${character.toLowerCase()}`)).join("_");
+const resolveApiKey = (options, context) => {
+	if (context !== void 0 && "apiKey" in context) {
+		if (context.apiKey === void 0) throw new TypeError("A Neon API key or OAuth access token is required");
+		return requireBearerCredential(context.apiKey);
+	}
+	return options.apiKey === void 0 ? missingBearerCredential : requireBearerCredential(options.apiKey);
+};
+const toolClient = (options, context) => createNeonClient({
+	apiKey: resolveApiKey(options, context),
+	throwOnError: true,
+	waitForReadiness: true,
+	...options.baseUrl === void 0 ? {} : { baseUrl: options.baseUrl },
+	...options.fetch === void 0 ? {} : { fetch: options.fetch },
+	...options.wait === void 0 ? {} : { wait: options.wait }
+});
+const objectListPage = (value) => {
+	if ("objects" in value && "folders" in value) return value;
+	if ("error" in value && value.error !== void 0) throw value.error;
+	if ("data" in value && value.data !== void 0) return value.data;
+	throw new NeonError("Object list returned no data.", "client");
+};
+const collectObjectList = async (neon, input, signal) => {
+	const folders = [];
+	const objects = [];
+	let prefix = "";
+	let cursor;
+	for (;;) {
+		const page = objectListPage(await neon.storage.objects.list(input.project_id, input.branch_id, input.bucket_name, {
+			...input.prefix === void 0 ? {} : { prefix: input.prefix },
+			...input.delimiter === void 0 ? {} : { delimiter: input.delimiter },
+			...input.limit === void 0 ? {} : { limit: input.limit },
+			...cursor === void 0 ? {} : { cursor }
+		}, { signal }));
+		for (const folder of page.folders) {
+			if (input.limit !== void 0 && folders.length + objects.length >= input.limit) break;
+			folders.push(folder);
+		}
+		for (const object of page.objects) {
+			if (input.limit !== void 0 && folders.length + objects.length >= input.limit) break;
+			objects.push(object);
+		}
+		prefix = page.prefix;
+		if (input.limit !== void 0 && folders.length + objects.length >= input.limit) break;
+		if (page.next_cursor) {
+			cursor = page.next_cursor;
+			continue;
+		}
+		if (page.is_truncated) throw new NeonError("Object list was truncated without a next cursor.", "client");
+		break;
+	}
+	return {
+		folders,
+		objects,
+		prefix
+	};
+};
+const collectPages = async (list, maxItems) => {
+	if (maxItems === void 0) {
+		const result = await list.all();
+		if (result.error) throw result.error;
+		if (result.data === void 0) throw new NeonError("List returned no data.", "client");
+		return result.data;
+	}
+	const items = [];
+	let cursor;
+	for (;;) {
+		const result = await list.page(cursor);
+		if (result.error) throw result.error;
+		if (result.data === void 0) throw new NeonError("List returned no data.", "client");
+		items.push(...result.data.items);
+		if (items.length >= maxItems) return items.slice(0, maxItems);
+		if (!result.data.cursor || result.data.items.length === 0) break;
+		cursor = result.data.cursor;
+	}
+	return items;
+};
+const bindTool = (options, tool, run) => ({
+	...tool,
+	async execute(input, context) {
+		const parsed = await tool.inputSchema.parseAsync(input);
+		const result = await run(toolClient(options, context), parsed, context?.signal);
+		return toToolResult(result);
+	}
+});
+function fromGenerated(options, spec) {
+	const generated = operationFactories[spec.generated](inertClient);
+	const schema = generated.inputSchema;
+	const inputSchema = spec.omit === void 0 || spec.omit.length === 0 ? schema : omitObjectKeys(schema, spec.omit);
+	return {
+		operationId: spec.id,
+		id: publishedId(spec.id),
+		title: generated.title,
+		description: spec.list ? `${generated.description.trimEnd()}${ALL_PAGES}` : generated.description,
+		inputSchema,
+		annotations: spec.annotations ?? generated.annotations,
+		requiresApproval: spec.requiresApproval ?? generated.requiresApproval,
+		metadata: generated.metadata,
+		async execute(input, context) {
+			const parsed = await inputSchema.parseAsync(input);
+			const result = await spec.run(toolClient(options, context), parsed, context?.signal);
+			return toToolResult(result);
+		}
+	};
+}
+const omitObjectKeys = (schema, keys) => {
+	if (!(schema instanceof z.ZodObject)) throw new TypeError("Generated tool input schemas must be objects");
+	const mask = {};
+	for (const key of keys) mask[key] = true;
+	return schema.omit(mask);
+};
+//#endregion
+export { bindTool, collectObjectList, collectPages, fromGenerated, publishedId, toolClient };