@neon/tools 0.3.0 → 0.4.0

package/README.md

@@ -1,6 +1,6 @@
 # `@neon/tools`
 
-Generated agent tools for every operation in the Neon Management API. Select only the operations an agent needs, then use the canonical descriptors directly or adapt them for MCP, Eve, or Mastra.
+Agent tools for the Neon Management API and for the `@neon/sdk` workflows that attach compute and return a connection string. Select only the operations and workflows an agent needs, then use the canonical descriptors directly or adapt them for MCP, Eve, and Mastra.
 
 ```bash
 npm install @neon/tools
@@ -48,9 +48,53 @@ await tools.listProjects.execute(
 );
 ```
 
-The returned record is keyed by OpenAPI operation ID. Each tool includes its generated Zod 4 `inputSchema`, snake-case `id`, title, description, safety annotations, stability metadata, and an `execute()` function. Inputs are flat: path, query, header, and body fields sit on one object. A body that is a single object wrapper is lifted, so `create_project` takes `{ name, region_id, org_id, ... }` rather than `{ project: { name } }`. A body with several properties keeps those properties, so `create_project_branch` takes `{ project_id, branch, endpoints, annotation_value }` and `branch` is still an object. Two email-provider updates keep a `body` field because the API request is a discriminated union.
+The returned record is keyed by OpenAPI operation ID or workflow id. Each tool includes its Zod 4 `inputSchema`, snake-case `id`, title, description, safety annotations, stability metadata, and an `execute()` function. Inputs are flat: path, query, header, and body fields sit on one object. A body that is a single object wrapper is lifted, so `create_project` takes `{ name, region_id, org_id, ... }` rather than `{ project: { name } }`. A body with several properties keeps those properties, so `create_project_branch` takes `{ project_id, branch, endpoints, annotation_value }` and `branch` is still an object. Two email-provider updates keep a `body` field because the API request is a discriminated union.
 
-`operationIds` exports every valid selector. `execute()` strictly validates the input, rejects unknown fields instead of dropping them, and returns typed, JSON-safe `{ data }`. Neon SDK errors remain typed and are thrown to the caller.
+`operationIds` exports every valid operation selector. `execute()` strictly validates the input, rejects unknown fields instead of dropping them, and returns typed, JSON-safe `{ data }`. Neon SDK errors remain typed and are thrown to the caller.
+
+## Workflows
+
+`workflows` selects methods from the `@neon/sdk` ergonomic client. These are not OpenAPI operations: they attach a default compute, wait until the resource is ready, and return a connection string. `operations` is the generated Management API. At least one of the two arrays is required.
+
+Both workflow tools poll until the resource is ready. The default deadline is five minutes (`wait.timeoutMs`). Pass `wait: { timeoutMs: 30_000 }` on `createNeonTools` or `createNeonTool` to bound that. Set it below the host's own tool-call timeout, otherwise the host gives up first. An abort `signal` on `execute` or a wait timeout stops the poll, not the create: the branch or project may already exist, and the error does not include its id. List before retrying. `metadata.method` and `metadata.path` name the first request; extra readiness GETs are not listed there. `inject.projectId` still works on `createBranchWithCompute` because its path is `/projects/{project_id}/branches`.
+
+```ts
+const tools = createNeonTools({
+	apiKey,
+	operations: ["listProjects"],
+	workflows: ["createBranchWithCompute", "createProjectAndConnect"],
+});
+
+const { data } = await tools.createBranchWithCompute.execute({
+	project_id: "project-id",
+	name: "feature-x",
+});
+
+await tools.createProjectAndConnect.execute({
+	name: "agent-project",
+	region_id: "aws-us-east-1",
+});
+```
+
+`createNeonTool` accepts the same workflow ids:
+
+```ts
+const createBranch = createNeonTool("createBranchWithCompute", { apiKey });
+```
+
+`workflowIds` lists the selectors. Record keys name the entity (`createBranchWithCompute`, `createProjectAndConnect`) and call `neon.branches.createWithCompute` and `neon.projects.createAndConnect`. Published ids are `create_branch_with_compute` and `create_project_and_connect`. Input fields are snake_case, same as generated tools. `names` can rename a workflow the same way it renames an operation:
+
+```ts
+const tools = createNeonTools({
+	apiKey,
+	workflows: ["createBranchWithCompute"],
+	names: { createBranchWithCompute: "create_branch" },
+});
+
+tools.createBranchWithCompute.id; // "create_branch"
+```
+
+`create_project_branch` still creates a branch with no compute when `endpoints` is omitted. It can attach compute if you pass `endpoints`; it does not wait or return a connection string. Use `createBranchWithCompute` when the next step needs to connect.
 
 ## Optional host add-ons
 
@@ -116,7 +160,7 @@ The record is still keyed by operation ID (`tools.createProjectBranch`). MCP and
 
 ### Project and branch injection
 
-Generated tools take path parameters as `project_id` and `branch_id`. A host that already knows those values can inject them. Without `omitFromSchema`, the published field becomes optional and a caller-supplied value wins. With `omitFromSchema: true`, the field is removed from the published schema and the injector is the only source:
+Tools take path parameters as `project_id` and `branch_id`. A host that already knows those values can inject them, including on `createBranchWithCompute`. Without `omitFromSchema`, the published field becomes optional and a caller-supplied value wins. With `omitFromSchema: true`, the field is removed from the published schema and the injector is the only source:
 
 ```ts
 const tools = createNeonTools({
@@ -152,7 +196,7 @@ await grant.run({ projectId: "project-id" }, () => tools.getProject.execute({}))
 
 Injectors only apply to tools that have that path key. `listProjects` is unchanged. Empty inject values fail closed. Invalid ids still fail the original path schema before fetch.
 
-These add-ons do not add host-only behavior such as returning a connection string from `createProject`. Grant filtering, read-only filtering, and access-control notices stay in the host.
+`onExecute` and `inject` apply to workflow tools the same way they apply to generated tools. Grant filtering, read-only filtering, and access-control notices stay in the host.
 
 Injection reads the URL template, so it fills path `project_id` and `branch_id` only. Query and body fields with those names stay caller-supplied, including `getConnectionURI.branch_id`, `createOrgApiKey.project_id`, `createNeonAuthIntegration.project_id` / `branch_id`, `createNeonAuthNewUser.project_id`, `createNeonAuthProviderSDKKeys.project_id`, `transferNeonAuthProviderProject.project_id`, `addProjectJWKS.branch_id`, `createProjectEndpoint.branch_id`, `updateProjectEndpoint.branch_id`, and `getProjectAdvisorSecurityIssues.branch_id`. `omitFromSchema: true` does not hide those fields.
 

package/dist/index.d.ts

@@ -1,35 +1,68 @@
 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 NeonWorkflowId, type WorkflowFactories, workflowIds } from "./lib/workflows.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 };
+export type { NeonOperationId, NeonWorkflowId };
+export { NeonError, operationIds, workflowIds };
 type OperationFactories = typeof operationFactories;
-export type NeonTools<Operations extends readonly NeonOperationId[]> = {
+export type NeonTools<Operations extends readonly NeonOperationId[] = [], Workflows extends readonly NeonWorkflowId[] = []> = {
     [Operation in Operations[number]]: ReturnType<OperationFactories[Operation]>;
+} & {
+    [Workflow in Workflows[number]]: ReturnType<WorkflowFactories[Workflow]>;
 };
-export type InjectedNeonTools<Operations extends readonly NeonOperationId[], Inject> = {
+export type InjectedNeonTools<Operations extends readonly NeonOperationId[], Inject, Workflows extends readonly NeonWorkflowId[] = []> = {
     [Operation in Operations[number]]: InjectedNeonTool<ReturnType<OperationFactories[Operation]>, Inject>;
+} & {
+    [Workflow in Workflows[number]]: InjectedNeonTool<ReturnType<WorkflowFactories[Workflow]>, 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"> {
+export interface NeonToolsClientOptions extends Pick<NeonConfig, "baseUrl" | "fetch" | "wait">, Omit<NeonToolCustomizeOptions, "name" | "names"> {
     apiKey?: NeonBearerCredential;
 }
-export interface CreateNeonToolsOptions<Operations extends readonly NeonOperationId[]> extends NeonToolsClientOptions {
-    operations: Operations;
+interface OperationSelection {
+    operations: readonly NeonOperationId[];
+    workflows?: readonly NeonWorkflowId[];
+}
+interface WorkflowSelection {
+    workflows: readonly NeonWorkflowId[];
+    operations?: readonly NeonOperationId[];
 }
-type NamedNeonTools<Operations extends readonly NeonOperationId[], Inject> = {
+type CreateNeonToolsInput = NeonToolsClientOptions & (OperationSelection | WorkflowSelection);
+export type CreateNeonToolsOptions<Operations extends readonly NeonOperationId[] = readonly NeonOperationId[], Workflows extends readonly NeonWorkflowId[] = readonly []> = NeonToolsClientOptions & (Workflows extends readonly [] ? {
+    operations: Operations;
+    workflows?: Workflows;
+} : Operations extends readonly [] ? {
+    workflows: Workflows;
+    operations?: Operations;
+} : {
+    operations: Operations;
+    workflows: Workflows;
+});
+type SelectedOperations<T> = T extends {
+    operations: infer O extends readonly NeonOperationId[];
+} ? O : [];
+type SelectedWorkflows<T> = T extends {
+    workflows: infer W extends readonly NeonWorkflowId[];
+} ? W : [];
+type NeonToolsFor<T, Inject> = T extends CreateNeonToolsInput ? Inject extends NeonToolInjectOptions ? InjectedNeonTools<SelectedOperations<T>, Inject, SelectedWorkflows<T>> : NeonTools<SelectedOperations<T>, SelectedWorkflows<T>> : never;
+type NeonToolId = NeonOperationId | NeonWorkflowId;
+type ToolForId<Id extends NeonToolId> = Id extends NeonWorkflowId ? ReturnType<WorkflowFactories[Id]> : Id extends NeonOperationId ? ReturnType<OperationFactories[Id]> : never;
+type NamedNeonTools<Operations extends readonly NeonOperationId[], Workflows extends readonly NeonWorkflowId[], Inject> = {
     [Operation in Operations[number]]: WithPublishedId<Inject extends NeonToolInjectOptions ? InjectedNeonTool<ReturnType<OperationFactories[Operation]>, Inject> : ReturnType<OperationFactories[Operation]>>;
+} & {
+    [Workflow in Workflows[number]]: WithPublishedId<Inject extends NeonToolInjectOptions ? InjectedNeonTool<ReturnType<WorkflowFactories[Workflow]>, Inject> : ReturnType<WorkflowFactories[Workflow]>>;
 };
-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> & {
+type NamedNeonTool<Id extends NeonToolId, Inject> = WithPublishedId<Inject extends NeonToolInjectOptions ? InjectedNeonTool<ToolForId<Id>, Inject> : ToolForId<Id>>;
+type NamedNeonToolsFor<T, Inject> = T extends CreateNeonToolsInput ? NamedNeonTools<SelectedOperations<T>, SelectedWorkflows<T>, Inject> : never;
+export declare function createNeonTools<const T extends CreateNeonToolsInput, const Inject extends NeonToolInjectOptions | undefined = undefined>(options: T & {
     inject?: Inject;
 } & ({
     name: (id: string) => string;
@@ -37,11 +70,11 @@ export declare function createNeonTools<const Operations extends readonly NeonOp
 } | {
     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> & {
+})): NamedNeonToolsFor<T, Inject>;
+export declare function createNeonTools<const T extends CreateNeonToolsInput, const Inject extends NeonToolInjectOptions | undefined = undefined>(options: T & {
     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 & {
+}): NeonToolsFor<T, Inject>;
+export declare function createNeonTool<const Id extends NeonToolId, const Inject extends NeonToolInjectOptions | undefined = undefined>(id: Id, options: NeonToolsClientOptions & {
     inject?: Inject;
 } & ({
     name: (id: string) => string;
@@ -49,8 +82,8 @@ export declare function createNeonTool<const Operation extends NeonOperationId,
 } | {
     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 & {
+})): NamedNeonTool<Id, Inject>;
+export declare function createNeonTool<const Id extends NeonToolId, const Inject extends NeonToolInjectOptions | undefined = undefined>(id: Id, options: NeonToolsClientOptions & {
     inject?: Inject;
-}): Inject extends NeonToolInjectOptions ? InjectedNeonTool<ReturnType<OperationFactories[Operation]>, Inject> : ReturnType<OperationFactories[Operation]>;
+}): Inject extends NeonToolInjectOptions ? InjectedNeonTool<ToolForId<Id>, Inject> : ToolForId<Id>;
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +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"}
+{"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;AAE5B,OAAO,EAEN,KAAK,cAAc,EACnB,KAAK,iBAAiB,EAEtB,WAAW,EACX,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,EAAE,CAAC;AAChD,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC;AAEhD,KAAK,kBAAkB,GAAG,OAAO,kBAAkB,CAAC;AAEpD,MAAM,MAAM,SAAS,CACpB,UAAU,SAAS,SAAS,eAAe,EAAE,GAAG,EAAE,EAClD,SAAS,SAAS,SAAS,cAAc,EAAE,GAAG,EAAE,IAC7C;KACF,SAAS,IAAI,UAAU,CAAC,MAAM,CAAC,GAAG,UAAU,CAC5C,kBAAkB,CAAC,SAAS,CAAC,CAC7B;CACD,GAAG;KACF,QAAQ,IAAI,SAAS,CAAC,MAAM,CAAC,GAAG,UAAU,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;CACxE,CAAC;AAEF,MAAM,MAAM,iBAAiB,CAC5B,UAAU,SAAS,SAAS,eAAe,EAAE,EAC7C,MAAM,EACN,SAAS,SAAS,SAAS,cAAc,EAAE,GAAG,EAAE,IAC7C;KACF,SAAS,IAAI,UAAU,CAAC,MAAM,CAAC,GAAG,gBAAgB,CAClD,UAAU,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,EACzC,MAAM,CACN;CACD,GAAG;KACF,QAAQ,IAAI,SAAS,CAAC,MAAM,CAAC,GAAG,gBAAgB,CAChD,UAAU,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,EACvC,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,GAAG,MAAM,CAAC,EACrD,IAAI,CAAC,wBAAwB,EAAE,MAAM,GAAG,OAAO,CAAC;IACjD,MAAM,CAAC,EAAE,oBAAoB,CAAC;CAC9B;AAED,UAAU,kBAAkB;IAC3B,UAAU,EAAE,SAAS,eAAe,EAAE,CAAC;IACvC,SAAS,CAAC,EAAE,SAAS,cAAc,EAAE,CAAC;CACtC;AAED,UAAU,iBAAiB;IAC1B,SAAS,EAAE,SAAS,cAAc,EAAE,CAAC;IACrC,UAAU,CAAC,EAAE,SAAS,eAAe,EAAE,CAAC;CACxC;AAED,KAAK,oBAAoB,GAAG,sBAAsB,GACjD,CAAC,kBAAkB,GAAG,iBAAiB,CAAC,CAAC;AAE1C,MAAM,MAAM,sBAAsB,CACjC,UAAU,SAAS,SAAS,eAAe,EAAE,GAAG,SAAS,eAAe,EAAE,EAC1E,SAAS,SAAS,SAAS,cAAc,EAAE,GAAG,SAAS,EAAE,IACtD,sBAAsB,GACzB,CAAC,SAAS,SAAS,SAAS,EAAE,GAC3B;IAAE,UAAU,EAAE,UAAU,CAAC;IAAC,SAAS,CAAC,EAAE,SAAS,CAAA;CAAE,GACjD,UAAU,SAAS,SAAS,EAAE,GAC7B;IAAE,SAAS,EAAE,SAAS,CAAC;IAAC,UAAU,CAAC,EAAE,UAAU,CAAA;CAAE,GACjD;IAAE,UAAU,EAAE,UAAU,CAAC;IAAC,SAAS,EAAE,SAAS,CAAA;CAAE,CAAC,CAAC;AAEvD,KAAK,kBAAkB,CAAC,CAAC,IAAI,CAAC,SAAS;IACtC,UAAU,EAAE,MAAM,CAAC,SAAS,SAAS,eAAe,EAAE,CAAC;CACvD,GACE,CAAC,GACD,EAAE,CAAC;AAEN,KAAK,iBAAiB,CAAC,CAAC,IAAI,CAAC,SAAS;IACrC,SAAS,EAAE,MAAM,CAAC,SAAS,SAAS,cAAc,EAAE,CAAC;CACrD,GACE,CAAC,GACD,EAAE,CAAC;AAEN,KAAK,YAAY,CAAC,CAAC,EAAE,MAAM,IAAI,CAAC,SAAS,oBAAoB,GAC1D,MAAM,SAAS,qBAAqB,GACnC,iBAAiB,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,iBAAiB,CAAC,CAAC,CAAC,CAAC,GACtE,SAAS,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,iBAAiB,CAAC,CAAC,CAAC,CAAC,GACvD,KAAK,CAAC;AAET,KAAK,UAAU,GAAG,eAAe,GAAG,cAAc,CAAC;AAEnD,KAAK,SAAS,CAAC,EAAE,SAAS,UAAU,IAAI,EAAE,SAAS,cAAc,GAC9D,UAAU,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC,GACjC,EAAE,SAAS,eAAe,GACzB,UAAU,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC,GAClC,KAAK,CAAC;AAkIV,KAAK,cAAc,CAClB,UAAU,SAAS,SAAS,eAAe,EAAE,EAC7C,SAAS,SAAS,SAAS,cAAc,EAAE,EAC3C,MAAM,IACH;KACF,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,GAAG;KACF,QAAQ,IAAI,SAAS,CAAC,MAAM,CAAC,GAAG,eAAe,CAC/C,MAAM,SAAS,qBAAqB,GACjC,gBAAgB,CAAC,UAAU,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,EAAE,MAAM,CAAC,GACjE,UAAU,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAC1C;CACD,CAAC;AAEF,KAAK,aAAa,CAAC,EAAE,SAAS,UAAU,EAAE,MAAM,IAAI,eAAe,CAClE,MAAM,SAAS,qBAAqB,GACjC,gBAAgB,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,MAAM,CAAC,GACvC,SAAS,CAAC,EAAE,CAAC,CAChB,CAAC;AAEF,KAAK,iBAAiB,CAAC,CAAC,EAAE,MAAM,IAAI,CAAC,SAAS,oBAAoB,GAC/D,cAAc,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,iBAAiB,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,GACnE,KAAK,CAAC;AAET,wBAAgB,eAAe,CAC9B,KAAK,CAAC,CAAC,SAAS,oBAAoB,EACpC,KAAK,CAAC,MAAM,SAAS,qBAAqB,GAAG,SAAS,GAAG,SAAS,EAElE,OAAO,EAAE,CAAC,GAAG;IAAE,MAAM,CAAC,EAAE,MAAM,CAAA;CAAE,GAAG,CAC/B;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,iBAAiB,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;AAChC,wBAAgB,eAAe,CAC9B,KAAK,CAAC,CAAC,SAAS,oBAAoB,EACpC,KAAK,CAAC,MAAM,SAAS,qBAAqB,GAAG,SAAS,GAAG,SAAS,EACjE,OAAO,EAAE,CAAC,GAAG;IAAE,MAAM,CAAC,EAAE,MAAM,CAAA;CAAE,GAAG,YAAY,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;AAY7D,wBAAgB,cAAc,CAC7B,KAAK,CAAC,EAAE,SAAS,UAAU,EAC3B,KAAK,CAAC,MAAM,SAAS,qBAAqB,GAAG,SAAS,GAAG,SAAS,EAElE,EAAE,EAAE,EAAE,EACN,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,EAAE,EAAE,MAAM,CAAC,CAAC;AAC7B,wBAAgB,cAAc,CAC7B,KAAK,CAAC,EAAE,SAAS,UAAU,EAC3B,KAAK,CAAC,MAAM,SAAS,qBAAqB,GAAG,SAAS,GAAG,SAAS,EAElE,EAAE,EAAE,EAAE,EACN,OAAO,EAAE,sBAAsB,GAAG;IAAE,MAAM,CAAC,EAAE,MAAM,CAAA;CAAE,GACnD,MAAM,SAAS,qBAAqB,GACpC,gBAAgB,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,MAAM,CAAC,GACvC,SAAS,CAAC,EAAE,CAAC,CAAC"}

package/dist/index.js

@@ -1,12 +1,14 @@
 import { missingBearerCredential, requireBearerCredential } from "./lib/auth.js";
 import { applyToolCustomization, assertToolCustomizeOptions, hasToolCustomization } from "./lib/customize.js";
+import { isNeonWorkflowId, workflowFactoryFor, workflowIds } from "./lib/workflows.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 }
+	...options.fetch === void 0 ? {} : { fetch: options.fetch },
+	...options.wait === void 0 ? {} : { wait: options.wait }
 }).client;
 const operationFactoryFor = (operationId) => {
 	const knownOperationId = operationIds.find((candidate) => candidate === operationId);
@@ -19,44 +21,57 @@ const assertKnownNameKeys = (names, tools) => {
 	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) {
+function assertNeonTools(value, operations, workflows) {
 	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}".`);
+	for (const workflowId of workflows) if (!(workflowId in value)) throw new TypeError(`Missing Neon workflow tool "${workflowId}".`);
 }
 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 operations = options.operations ?? [];
+	const workflows = options.workflows ?? [];
+	if (operations.length === 0 && workflows.length === 0) throw new TypeError("createNeonTools requires at least one operation or workflow");
+	const selectedOperations = /* @__PURE__ */ new Set();
+	const operationEntries = operations.map((operationId) => {
+		if (selectedOperations.has(operationId)) throw new Error(`Duplicate Neon operation "${operationId}"`);
+		selectedOperations.add(operationId);
+		return [operationId, operationFactoryFor(operationId)(createRawClient(options))];
 	});
-	const client = createRawClient(options);
-	const rawEntries = selectedFactories.map(([operationId, factory]) => {
-		return [operationId, factory(client)];
+	const selectedWorkflows = /* @__PURE__ */ new Set();
+	const workflowEntries = workflows.map((workflowId) => {
+		if (selectedWorkflows.has(workflowId)) throw new Error(`Duplicate Neon workflow "${workflowId}"`);
+		selectedWorkflows.add(workflowId);
+		return [workflowId, workflowFactoryFor(workflowId)(options)];
 	});
+	const rawEntries = [...operationEntries, ...workflowEntries];
 	assertKnownNameKeys(options.names, rawEntries.map(([, tool]) => tool));
-	const entries = rawEntries.map(([operationId, tool]) => {
-		return [operationId, hasToolCustomization(options) ? applyToolCustomization(tool, options) : tool];
+	const entries = rawEntries.map(([id, tool]) => {
+		return [id, hasToolCustomization(options) ? applyToolCustomization(tool, options) : tool];
 	});
 	const publishedIds = /* @__PURE__ */ new Map();
-	for (const [operationId, tool] of entries) {
+	for (const [id, 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);
+		if (previous !== void 0) throw new Error(`Duplicate Neon tool id "${tool.id}" for ${previous}, ${id}`);
+		publishedIds.set(tool.id, id);
 	}
 	const tools = Object.fromEntries(entries);
-	assertNeonTools(tools, options.operations);
+	assertNeonTools(tools, operations, workflows);
 	return tools;
 };
 function createNeonTools(options) {
 	return bindTools(options);
 }
-function createNeonTool(operationId, options) {
+function createNeonTool(id, options) {
 	assertToolCustomizeOptions(options);
-	const tool = operationFactoryFor(operationId)(createRawClient(options));
+	let tool;
+	if (isNeonWorkflowId(id)) tool = workflowFactoryFor(id)(options);
+	else {
+		const knownOperationId = operationIds.find((candidate) => candidate === id);
+		if (knownOperationId === void 0) throw new TypeError(`Unknown Neon operation or workflow "${id}".`);
+		tool = operationFactories[knownOperationId](createRawClient(options));
+	}
 	assertKnownNameKeys(options.names, [tool]);
 	return hasToolCustomization(options) ? applyToolCustomization(tool, options) : tool;
 }
 //#endregion
-export { NeonError, createNeonTool, createNeonTools, operationIds };
+export { NeonError, createNeonTool, createNeonTools, operationIds, workflowIds };

package/dist/lib/workflows.d.ts

@@ -0,0 +1,315 @@
+import { type NeonConfig } from "@neon/sdk";
+import * as z from "zod";
+import { type NeonBearerCredential } from "./auth.js";
+import type { NeonTool } from "./operation.js";
+export interface WorkflowClientOptions {
+    apiKey?: NeonBearerCredential;
+    baseUrl?: string;
+    fetch?: typeof fetch;
+    wait?: NeonConfig["wait"];
+}
+export declare const createBranchWithComputeInputSchema: z.ZodObject<{
+    project_id: z.ZodString;
+    name: z.ZodOptional<z.ZodString>;
+    parent_id: z.ZodOptional<z.ZodString>;
+    compute: z.ZodOptional<z.ZodObject<{
+        min_cu: z.ZodOptional<z.ZodNumber>;
+        max_cu: z.ZodOptional<z.ZodNumber>;
+        suspend_timeout_seconds: z.ZodOptional<z.ZodInt>;
+    }, z.core.$strict>>;
+    pooled: z.ZodOptional<z.ZodBoolean>;
+}, z.core.$strict>;
+export declare const createProjectAndConnectInputSchema: z.ZodObject<{
+    pooled: z.ZodOptional<z.ZodBoolean>;
+    settings: z.ZodOptional<z.ZodObject<{
+        quota: z.ZodOptional<z.ZodObject<{
+            active_time_seconds: z.ZodOptional<z.ZodInt>;
+            compute_time_seconds: z.ZodOptional<z.ZodInt>;
+            written_data_bytes: z.ZodOptional<z.ZodInt>;
+            data_transfer_bytes: z.ZodOptional<z.ZodInt>;
+            logical_size_bytes: z.ZodOptional<z.ZodInt>;
+        }, z.core.$strict>>;
+        allowed_ips: z.ZodOptional<z.ZodObject<{
+            ips: z.ZodOptional<z.ZodArray<z.ZodString>>;
+            protected_branches_only: z.ZodOptional<z.ZodBoolean>;
+        }, z.core.$strict>>;
+        enable_logical_replication: z.ZodOptional<z.ZodBoolean>;
+        maintenance_window: z.ZodOptional<z.ZodObject<{
+            weekdays: z.ZodArray<z.ZodInt>;
+            start_time: z.ZodString;
+            end_time: z.ZodString;
+        }, z.core.$strict>>;
+        block_public_connections: z.ZodOptional<z.ZodBoolean>;
+        block_vpc_connections: z.ZodOptional<z.ZodBoolean>;
+        audit_log_level: z.ZodOptional<z.ZodEnum<{
+            base: "base";
+            extended: "extended";
+            full: "full";
+        }>>;
+        hipaa: z.ZodOptional<z.ZodBoolean>;
+        preload_libraries: z.ZodOptional<z.ZodObject<{
+            use_defaults: z.ZodOptional<z.ZodBoolean>;
+            enabled_libraries: z.ZodOptional<z.ZodArray<z.ZodString>>;
+        }, z.core.$strict>>;
+    }, z.core.$strict>>;
+    name: z.ZodOptional<z.ZodString>;
+    branch: z.ZodOptional<z.ZodObject<{
+        name: z.ZodOptional<z.ZodString>;
+        role_name: z.ZodOptional<z.ZodString>;
+        database_name: z.ZodOptional<z.ZodString>;
+        annotations: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
+    }, z.core.$strict>>;
+    autoscaling_limit_min_cu: z.ZodOptional<z.ZodNumber>;
+    autoscaling_limit_max_cu: z.ZodOptional<z.ZodNumber>;
+    provisioner: z.ZodOptional<z.ZodString>;
+    region_id: z.ZodOptional<z.ZodString>;
+    default_endpoint_settings: z.ZodOptional<z.ZodObject<{
+        pg_settings: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
+        pgbouncer_settings: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
+        autoscaling_limit_min_cu: z.ZodOptional<z.ZodNumber>;
+        autoscaling_limit_max_cu: z.ZodOptional<z.ZodNumber>;
+        suspend_timeout_seconds: z.ZodOptional<z.ZodInt>;
+    }, z.core.$strict>>;
+    pg_version: z.ZodOptional<z.ZodInt>;
+    store_passwords: z.ZodOptional<z.ZodBoolean>;
+    history_retention_seconds: z.ZodOptional<z.ZodInt>;
+    org_id: z.ZodOptional<z.ZodString>;
+}, z.core.$strict>;
+export declare const workflowIds: readonly ["createBranchWithCompute", "createProjectAndConnect"];
+export type NeonWorkflowId = (typeof workflowIds)[number];
+export declare const workflowFactories: {
+    createBranchWithCompute: (options: WorkflowClientOptions) => NeonTool<z.ZodObject<{
+        project_id: z.ZodString;
+        name: z.ZodOptional<z.ZodString>;
+        parent_id: z.ZodOptional<z.ZodString>;
+        compute: z.ZodOptional<z.ZodObject<{
+            min_cu: z.ZodOptional<z.ZodNumber>;
+            max_cu: z.ZodOptional<z.ZodNumber>;
+            suspend_timeout_seconds: z.ZodOptional<z.ZodInt>;
+        }, z.core.$strict>>;
+        pooled: z.ZodOptional<z.ZodBoolean>;
+    }, z.core.$strict>, "create_branch_with_compute", {
+        branch: {
+            id: string;
+            project_id: string;
+            parent_id?: string | null | undefined;
+            parent_lsn?: string | null | undefined;
+            parent_timestamp?: string | null | undefined;
+            name: string;
+            current_state: import("@neon/sdk").BranchState;
+            pending_state?: string | null | undefined;
+            state_changed_at: string;
+            logical_size?: number | null | undefined;
+            creation_source: string;
+            primary?: boolean | null | undefined;
+            default: boolean;
+            protected: boolean;
+            cpu_used_sec: number;
+            compute_time_seconds: number;
+            active_time_seconds: number;
+            written_data_bytes: number;
+            data_transfer_bytes: number;
+            created_at: string;
+            updated_at: string;
+            ttl_interval_seconds?: number | null | undefined;
+            expires_at?: string | null | undefined;
+            last_reset_at?: string | null | undefined;
+            created_by?: {
+                name?: string | null | undefined;
+                image?: string | null | undefined;
+            } | null | undefined;
+            init_source?: string | null | undefined;
+            restore_status?: string | null | undefined;
+            restored_from?: string | null | undefined;
+            restored_as?: string | null | undefined;
+            restricted_actions?: {
+                name: string;
+                reason: string;
+            }[] | null | undefined;
+            recovery?: {
+                deleted_at: string;
+                recoverable_until: string;
+                deletion_method: "user" | "ttl";
+            } | null | undefined;
+        };
+        endpoint: {
+            host: string;
+            id: string;
+            name?: string | null | undefined;
+            project_id: string;
+            branch_id: string;
+            autoscaling_limit_min_cu: import("@neon/sdk").ComputeUnit;
+            autoscaling_limit_max_cu: import("@neon/sdk").ComputeUnit;
+            region_id: string;
+            type: import("@neon/sdk").EndpointType;
+            current_state: import("@neon/sdk").EndpointState;
+            pending_state?: import("@neon/sdk").EndpointState | null | undefined;
+            settings: {
+                pg_settings?: {
+                    [x: string]: string;
+                } | null | undefined;
+                pgbouncer_settings?: {
+                    [x: string]: string;
+                } | null | undefined;
+                preload_libraries?: {
+                    use_defaults?: boolean | null | undefined;
+                    enabled_libraries?: string[] | null | undefined;
+                } | null | undefined;
+            };
+            pooler_enabled: boolean;
+            pooler_mode: import("@neon/sdk").EndpointPoolerMode;
+            disabled: boolean;
+            passwordless_access: boolean;
+            last_active?: string | null | undefined;
+            creation_source: string;
+            created_at: string;
+            updated_at: string;
+            started_at?: string | null | undefined;
+            suspended_at?: string | null | undefined;
+            proxy_host: string;
+            suspend_timeout_seconds: import("@neon/sdk").SuspendTimeoutSeconds;
+            provisioner: import("@neon/sdk").Provisioner;
+            compute_release_version?: string | null | undefined;
+        };
+        connectionString: string;
+    }>;
+    createProjectAndConnect: (options: WorkflowClientOptions) => NeonTool<z.ZodObject<{
+        pooled: z.ZodOptional<z.ZodBoolean>;
+        settings: z.ZodOptional<z.ZodObject<{
+            quota: z.ZodOptional<z.ZodObject<{
+                active_time_seconds: z.ZodOptional<z.ZodInt>;
+                compute_time_seconds: z.ZodOptional<z.ZodInt>;
+                written_data_bytes: z.ZodOptional<z.ZodInt>;
+                data_transfer_bytes: z.ZodOptional<z.ZodInt>;
+                logical_size_bytes: z.ZodOptional<z.ZodInt>;
+            }, z.core.$strict>>;
+            allowed_ips: z.ZodOptional<z.ZodObject<{
+                ips: z.ZodOptional<z.ZodArray<z.ZodString>>;
+                protected_branches_only: z.ZodOptional<z.ZodBoolean>;
+            }, z.core.$strict>>;
+            enable_logical_replication: z.ZodOptional<z.ZodBoolean>;
+            maintenance_window: z.ZodOptional<z.ZodObject<{
+                weekdays: z.ZodArray<z.ZodInt>;
+                start_time: z.ZodString;
+                end_time: z.ZodString;
+            }, z.core.$strict>>;
+            block_public_connections: z.ZodOptional<z.ZodBoolean>;
+            block_vpc_connections: z.ZodOptional<z.ZodBoolean>;
+            audit_log_level: z.ZodOptional<z.ZodEnum<{
+                base: "base";
+                extended: "extended";
+                full: "full";
+            }>>;
+            hipaa: z.ZodOptional<z.ZodBoolean>;
+            preload_libraries: z.ZodOptional<z.ZodObject<{
+                use_defaults: z.ZodOptional<z.ZodBoolean>;
+                enabled_libraries: z.ZodOptional<z.ZodArray<z.ZodString>>;
+            }, z.core.$strict>>;
+        }, z.core.$strict>>;
+        name: z.ZodOptional<z.ZodString>;
+        branch: z.ZodOptional<z.ZodObject<{
+            name: z.ZodOptional<z.ZodString>;
+            role_name: z.ZodOptional<z.ZodString>;
+            database_name: z.ZodOptional<z.ZodString>;
+            annotations: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
+        }, z.core.$strict>>;
+        autoscaling_limit_min_cu: z.ZodOptional<z.ZodNumber>;
+        autoscaling_limit_max_cu: z.ZodOptional<z.ZodNumber>;
+        provisioner: z.ZodOptional<z.ZodString>;
+        region_id: z.ZodOptional<z.ZodString>;
+        default_endpoint_settings: z.ZodOptional<z.ZodObject<{
+            pg_settings: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
+            pgbouncer_settings: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
+            autoscaling_limit_min_cu: z.ZodOptional<z.ZodNumber>;
+            autoscaling_limit_max_cu: z.ZodOptional<z.ZodNumber>;
+            suspend_timeout_seconds: z.ZodOptional<z.ZodInt>;
+        }, z.core.$strict>>;
+        pg_version: z.ZodOptional<z.ZodInt>;
+        store_passwords: z.ZodOptional<z.ZodBoolean>;
+        history_retention_seconds: z.ZodOptional<z.ZodInt>;
+        org_id: z.ZodOptional<z.ZodString>;
+    }, z.core.$strict>, "create_project_and_connect", {
+        project: {
+            data_storage_bytes_hour: number;
+            data_transfer_bytes: number;
+            written_data_bytes: number;
+            compute_time_seconds: number;
+            active_time_seconds: number;
+            cpu_used_sec: number;
+            id: string;
+            platform_id: string;
+            region_id: string;
+            name: string;
+            provisioner: import("@neon/sdk").Provisioner;
+            default_endpoint_settings?: {
+                pg_settings?: {
+                    [x: string]: string;
+                } | null | undefined;
+                pgbouncer_settings?: {
+                    [x: string]: string;
+                } | null | undefined;
+                autoscaling_limit_min_cu?: number | null | undefined;
+                autoscaling_limit_max_cu?: number | null | undefined;
+                suspend_timeout_seconds?: number | null | undefined;
+            } | null | undefined;
+            settings?: {
+                quota?: {
+                    active_time_seconds?: number | null | undefined;
+                    compute_time_seconds?: number | null | undefined;
+                    written_data_bytes?: number | null | undefined;
+                    data_transfer_bytes?: number | null | undefined;
+                    logical_size_bytes?: number | null | undefined;
+                } | null | undefined;
+                allowed_ips?: {
+                    ips?: string[] | null | undefined;
+                    protected_branches_only?: boolean | null | undefined;
+                } | null | undefined;
+                enable_logical_replication?: boolean | null | undefined;
+                maintenance_window?: {
+                    weekdays: number[];
+                    start_time: string;
+                    end_time: string;
+                } | null | undefined;
+                block_public_connections?: boolean | null | undefined;
+                block_vpc_connections?: boolean | null | undefined;
+                audit_log_level?: import("@neon/sdk").ProjectAuditLogLevel | null | undefined;
+                hipaa?: boolean | null | undefined;
+                preload_libraries?: {
+                    use_defaults?: boolean | null | undefined;
+                    enabled_libraries?: string[] | null | undefined;
+                } | null | undefined;
+            } | null | undefined;
+            pg_version: import("@neon/sdk").PgVersion;
+            proxy_host: string;
+            branch_logical_size_limit: number;
+            branch_logical_size_limit_bytes: number;
+            store_passwords: boolean;
+            maintenance_starts_at?: string | null | undefined;
+            creation_source: string;
+            history_retention_seconds: number;
+            created_at: string;
+            updated_at: string;
+            synthetic_storage_size?: number | null | undefined;
+            consumption_period_start: string;
+            consumption_period_end: string;
+            quota_reset_at?: string | null | undefined;
+            owner_id: string;
+            owner?: {
+                email: string;
+                name: string;
+                branches_limit: number;
+                subscription_type: import("@neon/sdk").BillingSubscriptionType;
+            } | null | undefined;
+            compute_last_active_at?: string | null | undefined;
+            org_id?: string | null | undefined;
+            maintenance_scheduled_for?: string | null | undefined;
+            hipaa_enabled_at?: string | null | undefined;
+            effective_project_permission?: (import("@neon/sdk").ProjectPermissionLevel | null) | undefined;
+        };
+        connectionString: string;
+    }>;
+};
+export type WorkflowFactories = typeof workflowFactories;
+export declare const isNeonWorkflowId: (id: string) => id is NeonWorkflowId;
+export declare const workflowFactoryFor: (workflowId: NeonWorkflowId) => WorkflowFactories[NeonWorkflowId];
+//# sourceMappingURL=workflows.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"workflows.d.ts","sourceRoot":"","sources":["../../src/lib/workflows.ts"],"names":[],"mappings":"AAAA,OAAO,EAAqC,KAAK,UAAU,EAAE,MAAM,WAAW,CAAC;AAC/E,OAAO,KAAK,CAAC,MAAM,KAAK,CAAC;AAEzB,OAAO,EAEN,KAAK,oBAAoB,EAEzB,MAAM,WAAW,CAAC;AACnB,OAAO,KAAK,EAEX,QAAQ,EAER,MAAM,gBAAgB,CAAC;AAGxB,MAAM,WAAW,qBAAqB;IACrC,MAAM,CAAC,EAAE,oBAAoB,CAAC;IAC9B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,OAAO,KAAK,CAAC;IACrB,IAAI,CAAC,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC;CAC1B;AAkBD,eAAO,MAAM,kCAAkC;;;;;;;;;;kBAY7C,CAAC;AAEH,eAAO,MAAM,kCAAkC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;kBAG7C,CAAC;AAEH,eAAO,MAAM,WAAW,iEAGd,CAAC;AAEX,MAAM,MAAM,cAAc,GAAG,CAAC,OAAO,WAAW,CAAC,CAAC,MAAM,CAAC,CAAC;AAkI1D,eAAO,MAAM,iBAAiB;uCA1EgB,qBAAqB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;uCA6CrB,qBAAqB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAgClE,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG,OAAO,iBAAiB,CAAC;AAEzD,eAAO,MAAM,gBAAgB,GAAI,IAAI,MAAM,KAAG,EAAE,IAAI,cACJ,CAAC;AAEjD,eAAO,MAAM,kBAAkB,GAC9B,YAAY,cAAc,KACxB,iBAAiB,CAAC,cAAc,CAQlC,CAAC"}

package/dist/lib/workflows.js

@@ -0,0 +1,113 @@
+import { missingBearerCredential, requireBearerCredential } from "./auth.js";
+import { zBranchCreateRequest, zComputeUnit, zCreateProjectBody, zCreateProjectBranchPath, zSuspendTimeoutSeconds } from "../generated/zod.gen.js";
+import { toToolResult } from "./result.js";
+import { createNeonClient } from "@neon/sdk";
+import * as z from "zod";
+//#region src/lib/workflows.ts
+const branchCreateFields = zBranchCreateRequest.shape.branch.unwrap().shape;
+const projectCreateFields = zCreateProjectBody.shape.project.shape;
+const writeAnnotations = {
+	readOnlyHint: false,
+	destructiveHint: true,
+	openWorldHint: true
+};
+const pooledField = z.boolean().describe("Return a pooled connection string. Default true. Set false for a direct connection.").optional();
+const createBranchWithComputeInputSchema = z.strictObject({
+	project_id: zCreateProjectBranchPath.shape.project_id,
+	name: branchCreateFields.name,
+	parent_id: branchCreateFields.parent_id,
+	compute: z.strictObject({
+		min_cu: zComputeUnit.optional(),
+		max_cu: zComputeUnit.optional(),
+		suspend_timeout_seconds: zSuspendTimeoutSeconds.optional()
+	}).optional(),
+	pooled: pooledField
+});
+const createProjectAndConnectInputSchema = z.strictObject({
+	...projectCreateFields,
+	pooled: pooledField
+});
+const workflowIds = ["createBranchWithCompute", "createProjectAndConnect"];
+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 workflowClient = (options, context) => createNeonClient({
+	apiKey: resolveApiKey(options, context),
+	throwOnError: true,
+	...options.baseUrl === void 0 ? {} : { baseUrl: options.baseUrl },
+	...options.fetch === void 0 ? {} : { fetch: options.fetch },
+	...options.wait === void 0 ? {} : { wait: options.wait }
+});
+const bindWorkflow = (options, tool, run) => ({
+	...tool,
+	async execute(input, context) {
+		const parsed = await tool.inputSchema.parseAsync(input);
+		const result = await run(workflowClient(options, context), parsed, context?.signal);
+		return toToolResult(result);
+	}
+});
+const createBranchWithComputeTool = (options) => bindWorkflow(options, {
+	operationId: "createBranchWithCompute",
+	id: "create_branch_with_compute",
+	title: "Create branch with compute",
+	description: "Create a branch with a read-write endpoint and return its connection string. Use this instead of create_project_branch when the next step needs to connect: create_project_branch omits compute unless you pass endpoints, and it does not wait or return a connection string. The call waits until the compute is ready, up to five minutes by default.",
+	inputSchema: createBranchWithComputeInputSchema,
+	annotations: writeAnnotations,
+	requiresApproval: true,
+	metadata: {
+		method: "POST",
+		path: "/projects/{project_id}/branches",
+		stability: "stable",
+		deprecated: false,
+		tags: ["Branch"]
+	}
+}, (neon, input, signal) => neon.branches.createWithCompute(input.project_id, {
+	name: input.name,
+	parentId: input.parent_id,
+	compute: input.compute === void 0 ? void 0 : {
+		minCu: input.compute.min_cu,
+		maxCu: input.compute.max_cu,
+		suspendTimeoutSeconds: input.compute.suspend_timeout_seconds
+	}
+}, {
+	signal,
+	...input.pooled === void 0 ? {} : { pooled: input.pooled }
+}));
+const createProjectAndConnectTool = (options) => bindWorkflow(options, {
+	operationId: "createProjectAndConnect",
+	id: "create_project_and_connect",
+	title: "Create project and connect",
+	description: "Create a project and return a connection string to its default branch. Use this instead of create_project when the next step needs to connect. The call waits until the default compute is ready, up to five minutes by default.",
+	inputSchema: createProjectAndConnectInputSchema,
+	annotations: writeAnnotations,
+	requiresApproval: true,
+	metadata: {
+		method: "POST",
+		path: "/projects",
+		stability: "stable",
+		deprecated: false,
+		tags: ["Project"]
+	}
+}, (neon, input, signal) => {
+	const { pooled, ...project } = input;
+	return neon.projects.createAndConnect(project, {
+		signal,
+		...pooled === void 0 ? {} : { pooled }
+	});
+});
+const workflowFactories = {
+	createBranchWithCompute: createBranchWithComputeTool,
+	createProjectAndConnect: createProjectAndConnectTool
+};
+const isNeonWorkflowId = (id) => workflowIds.includes(id);
+const workflowFactoryFor = (workflowId) => {
+	const knownWorkflowId = workflowIds.find((candidate) => candidate === workflowId);
+	if (knownWorkflowId === void 0) throw new TypeError(`Unknown Neon workflow "${workflowId}".`);
+	return workflowFactories[knownWorkflowId];
+};
+//#endregion
+export { createBranchWithComputeInputSchema, createProjectAndConnectInputSchema, isNeonWorkflowId, workflowFactories, workflowFactoryFor, workflowIds };

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@neon/tools",
-  "version": "0.3.0",
-  "description": "Generated agent tools for every Neon Management API operation, compatible with MCP, Eve, and Mastra.",
+  "version": "0.4.0",
+  "description": "Agent tools for the Neon Management API and SDK workflows, compatible with MCP, Eve, and Mastra.",
   "keywords": [
     "neon",
     "database",