@neondatabase/ai-sdk-provider 0.0.0 → 0.1.1

package/dist/lib/neon-responses-language-model.d.ts

@@ -0,0 +1,26 @@
+import { OpenAIResponsesLanguageModel } from "@ai-sdk/openai/internal";
+import { LanguageModelV4, LanguageModelV4CallOptions, LanguageModelV4GenerateResult, LanguageModelV4StreamResult } from "@ai-sdk/provider";
+
+//#region src/lib/neon-responses-language-model.d.ts
+
+/**
+ * Language model for OpenAI models served via the Neon AI Gateway's native
+ * Responses API (`/ai-gateway/openai/v1/responses`). This route is required for
+ * models that are only served natively (e.g. Codex) and unlocks native
+ * reasoning and the image-generation tool.
+ *
+ * The shared OpenAI Responses model shapes a request based on whether the model
+ * is a reasoning model, but it detects that from the bare model id (`gpt-5`),
+ * which the gateway's required `databricks-` prefix defeats. For the GPT-5
+ * reasoning family we set the model's own `forceReasoning` provider option so it
+ * applies the correct reasoning behavior. Users can still override it.
+ */
+declare class NeonResponsesLanguageModel extends OpenAIResponsesLanguageModel implements LanguageModelV4 {
+  private get isReasoningFamily();
+  private withForcedReasoning;
+  doGenerate(options: LanguageModelV4CallOptions): Promise<LanguageModelV4GenerateResult>;
+  doStream(options: LanguageModelV4CallOptions): Promise<LanguageModelV4StreamResult>;
+}
+//#endregion
+export { NeonResponsesLanguageModel };
+//# sourceMappingURL=neon-responses-language-model.d.ts.map

package/dist/lib/neon-responses-language-model.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"neon-responses-language-model.d.ts","names":[],"sources":["../../src/lib/neon-responses-language-model.ts"],"mappings":";;;;;;;AAoBA;;;;;;;;;;AAE2B,cAFd,0BAAA,SACJ,4BAAA,YACG,eAAe,CAAA;;;sBA2BhB,6BACP,QAAQ;oBAKD,6BACP,QAAQ"}

package/dist/lib/neon-responses-language-model.js

@@ -0,0 +1,44 @@
+import { OpenAIResponsesLanguageModel } from "@ai-sdk/openai/internal";
+//#region src/lib/neon-responses-language-model.ts
+/**
+* Language model for OpenAI models served via the Neon AI Gateway's native
+* Responses API (`/ai-gateway/openai/v1/responses`). This route is required for
+* models that are only served natively (e.g. Codex) and unlocks native
+* reasoning and the image-generation tool.
+*
+* The shared OpenAI Responses model shapes a request based on whether the model
+* is a reasoning model, but it detects that from the bare model id (`gpt-5`),
+* which the gateway's required `databricks-` prefix defeats. For the GPT-5
+* reasoning family we set the model's own `forceReasoning` provider option so it
+* applies the correct reasoning behavior. Users can still override it.
+*/
+var NeonResponsesLanguageModel = class extends OpenAIResponsesLanguageModel {
+	get isReasoningFamily() {
+		return /gpt-5/.test(this.modelId.toLowerCase());
+	}
+	withForcedReasoning(options) {
+		if (!this.isReasoningFamily) return options;
+		const openai = options.providerOptions?.openai;
+		if (openai != null && "forceReasoning" in openai) return options;
+		return {
+			...options,
+			providerOptions: {
+				...options.providerOptions,
+				openai: {
+					...openai,
+					forceReasoning: true
+				}
+			}
+		};
+	}
+	doGenerate(options) {
+		return super.doGenerate(this.withForcedReasoning(options));
+	}
+	doStream(options) {
+		return super.doStream(this.withForcedReasoning(options));
+	}
+};
+//#endregion
+export { NeonResponsesLanguageModel };
+
+//# sourceMappingURL=neon-responses-language-model.js.map

package/dist/lib/neon-responses-language-model.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"neon-responses-language-model.js","names":[],"sources":["../../src/lib/neon-responses-language-model.ts"],"sourcesContent":["import { OpenAIResponsesLanguageModel } from \"@ai-sdk/openai/internal\";\nimport type {\n\tLanguageModelV4,\n\tLanguageModelV4CallOptions,\n\tLanguageModelV4GenerateResult,\n\tLanguageModelV4StreamResult,\n} from \"@ai-sdk/provider\";\n\n/**\n * Language model for OpenAI models served via the Neon AI Gateway's native\n * Responses API (`/ai-gateway/openai/v1/responses`). This route is required for\n * models that are only served natively (e.g. Codex) and unlocks native\n * reasoning and the image-generation tool.\n *\n * The shared OpenAI Responses model shapes a request based on whether the model\n * is a reasoning model, but it detects that from the bare model id (`gpt-5`),\n * which the gateway's required `databricks-` prefix defeats. For the GPT-5\n * reasoning family we set the model's own `forceReasoning` provider option so it\n * applies the correct reasoning behavior. Users can still override it.\n */\nexport class NeonResponsesLanguageModel\n\textends OpenAIResponsesLanguageModel\n\timplements LanguageModelV4\n{\n\tprivate get isReasoningFamily(): boolean {\n\t\treturn /gpt-5/.test(this.modelId.toLowerCase());\n\t}\n\n\tprivate withForcedReasoning(\n\t\toptions: LanguageModelV4CallOptions,\n\t): LanguageModelV4CallOptions {\n\t\tif (!this.isReasoningFamily) {\n\t\t\treturn options;\n\t\t}\n\t\tconst openai = options.providerOptions?.openai;\n\t\t// Respect an explicit user setting.\n\t\tif (openai != null && \"forceReasoning\" in openai) {\n\t\t\treturn options;\n\t\t}\n\t\treturn {\n\t\t\t...options,\n\t\t\tproviderOptions: {\n\t\t\t\t...options.providerOptions,\n\t\t\t\topenai: { ...openai, forceReasoning: true },\n\t\t\t},\n\t\t};\n\t}\n\n\toverride doGenerate(\n\t\toptions: LanguageModelV4CallOptions,\n\t): Promise<LanguageModelV4GenerateResult> {\n\t\treturn super.doGenerate(this.withForcedReasoning(options));\n\t}\n\n\toverride doStream(\n\t\toptions: LanguageModelV4CallOptions,\n\t): Promise<LanguageModelV4StreamResult> {\n\t\treturn super.doStream(this.withForcedReasoning(options));\n\t}\n}\n"],"mappings":";;;;;;;;;;;;;;AAoBA,IAAa,6BAAb,cACS,6BAET;CACC,IAAY,oBAA6B;EACxC,OAAO,QAAQ,KAAK,KAAK,QAAQ,YAAY,CAAC;CAC/C;CAEA,oBACC,SAC6B;EAC7B,IAAI,CAAC,KAAK,mBACT,OAAO;EAER,MAAM,SAAS,QAAQ,iBAAiB;EAExC,IAAI,UAAU,QAAQ,oBAAoB,QACzC,OAAO;EAER,OAAO;GACN,GAAG;GACH,iBAAiB;IAChB,GAAG,QAAQ;IACX,QAAQ;KAAE,GAAG;KAAQ,gBAAgB;IAAK;GAC3C;EACD;CACD;CAEA,WACC,SACyC;EACzC,OAAO,MAAM,WAAW,KAAK,oBAAoB,OAAO,CAAC;CAC1D;CAEA,SACC,SACuC;EACvC,OAAO,MAAM,SAAS,KAAK,oBAAoB,OAAO,CAAC;CACxD;AACD"}

package/dist/lib/provider.d.ts

@@ -0,0 +1,48 @@
+import { NeonChatModelId } from "./neon-chat-options.js";
+import { FetchFunction } from "@ai-sdk/provider-utils";
+import { LanguageModelV4, ProviderV4 } from "@ai-sdk/provider";
+import { z } from "zod/v4";
+
+//#region src/lib/provider.d.ts
+declare const neonErrorSchema: z.ZodObject<{
+  error: z.ZodObject<{
+    message: z.ZodString;
+    type: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+    param: z.ZodOptional<z.ZodNullable<z.ZodUnknown>>;
+    code: z.ZodOptional<z.ZodNullable<z.ZodUnknown>>;
+  }, z.core.$strip>;
+}, z.core.$strip>;
+type NeonErrorData = z.infer<typeof neonErrorSchema>;
+interface NeonProviderSettings {
+  /**
+   * Neon AI Gateway base URL — the branch-scoped host root, e.g.
+   * `https://<branch-id>-api.ai.<region>.aws.neon.tech`. The gateway paths are
+   * appended internally. Falls back to the `NEON_AI_GATEWAY_BASE_URL` env var.
+   */
+  baseURL?: string;
+  /**
+   * Neon AI Gateway platform token (the `nt_live_...` value). Falls back to the
+   * `NEON_AI_GATEWAY_TOKEN` env var.
+   */
+  apiKey?: string;
+  /** Custom headers to include in the requests. */
+  headers?: Record<string, string>;
+  /** Custom fetch implementation (e.g. for testing or middleware). */
+  fetch?: FetchFunction;
+}
+interface NeonProvider extends ProviderV4 {
+  /** Creates a Neon AI Gateway model for text generation. */
+  (modelId: NeonChatModelId): LanguageModelV4;
+  /** Creates a Neon AI Gateway model for text generation. */
+  languageModel(modelId: NeonChatModelId): LanguageModelV4;
+  /** Creates a Neon AI Gateway chat model for text generation. */
+  chat(modelId: NeonChatModelId): LanguageModelV4;
+  /** @deprecated Use `embeddingModel` instead. */
+  textEmbeddingModel(modelId: string): never;
+}
+declare function createNeon(options?: NeonProviderSettings): NeonProvider;
+/** Default Neon provider instance (reads `NEON_AI_GATEWAY_*` env vars). */
+declare const neon: NeonProvider;
+//#endregion
+export { NeonErrorData, NeonProvider, NeonProviderSettings, createNeon, neon };
+//# sourceMappingURL=provider.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"provider.d.ts","names":[],"sources":["../../src/lib/provider.ts"],"mappings":";;;;;;cAiCM,iBAAe,CAAA,CAAA;;IAAf,OAAA,aAOJ;IAAA,IAAA,eAAA,cAAA,YAAA,CAAA,CAAA;;;;;KAEU,aAAA,GAAgB,CAAA,CAAE,aAAa;UA4C1B,oBAAA;;;;;;;;;AArDI;AASrB;EAAyB,MAAA,CAAA,EAAA,MAAA;;SAAK,CAAA,EA2DnB,MA3DmB,CAAA,MAAA,EAAA,MAAA,CAAA;EAAK;EA4ClB,KAAA,CAAA,EAkBR,aAlBQ;;AAeN,UAMM,YAAA,SAAqB,UAN3B,CAAA;;EAGW,CAAA,OAAA,EAKX,eALW,CAAA,EAKO,eALP;EAGL;EAAa,aAAA,CAAA,OAAA,EAKN,eALM,CAAA,EAKY,eALZ;;MAED,CAAA,OAAA,EAMd,eANc,CAAA,EAMI,eANJ;;oBAGa,CAAA,OAAA,EAAA,MAAA,CAAA,EAAA,KAAA;;AAGT,iBAMjB,UAAA,CANiB,OAAA,CAAA,EAMG,oBANH,CAAA,EAM+B,YAN/B;;AARe,cAuGnC,IAvGmC,EAuG/B,YAvG+B"}

package/dist/lib/provider.js

@@ -0,0 +1,112 @@
+import { NeonAnthropicLanguageModel } from "./neon-anthropic-language-model.js";
+import { getNeonModelRoute } from "./neon-model-capabilities.js";
+import { NeonChatLanguageModel } from "./neon-chat-language-model.js";
+import { NeonResponsesLanguageModel } from "./neon-responses-language-model.js";
+import { VERSION } from "./version.js";
+import { generateId, loadApiKey, loadSetting, withUserAgentSuffix, withoutTrailingSlash } from "@ai-sdk/provider-utils";
+import { NoSuchModelError } from "@ai-sdk/provider";
+import { z } from "zod/v4";
+//#region src/lib/provider.ts
+const neonErrorStructure = {
+	errorSchema: z.object({ error: z.object({
+		message: z.string(),
+		type: z.string().nullish(),
+		param: z.unknown().nullish(),
+		code: z.unknown().nullish()
+	}) }),
+	errorToMessage: (data) => data.error.message
+};
+/**
+* Recursively remove the JSON Schema `$schema` marker, which some gateway
+* backends (notably Gemini) reject in tool/structured-output schemas. Other
+* backends ignore its absence, so stripping it everywhere is safe.
+*/
+function stripJsonSchemaMarker(value) {
+	if (Array.isArray(value)) return value.map(stripJsonSchemaMarker);
+	if (value !== null && typeof value === "object") {
+		const result = {};
+		for (const [key, entry] of Object.entries(value)) {
+			if (key === "$schema") continue;
+			result[key] = stripJsonSchemaMarker(entry);
+		}
+		return result;
+	}
+	return value;
+}
+function transformNeonRequestBody(args) {
+	const transformed = { ...args };
+	if (transformed.tools != null) transformed.tools = stripJsonSchemaMarker(transformed.tools);
+	if (transformed.response_format != null) transformed.response_format = stripJsonSchemaMarker(transformed.response_format);
+	return transformed;
+}
+function createNeon(options = {}) {
+	const getHost = () => withoutTrailingSlash(loadSetting({
+		settingValue: options.baseURL,
+		environmentVariableName: "NEON_AI_GATEWAY_BASE_URL",
+		settingName: "baseURL",
+		description: "Neon AI Gateway base URL"
+	}));
+	const getHeaders = (extra) => withUserAgentSuffix({
+		Authorization: `Bearer ${loadApiKey({
+			apiKey: options.apiKey,
+			environmentVariableName: "NEON_AI_GATEWAY_TOKEN",
+			description: "Neon AI Gateway token"
+		})}`,
+		...extra,
+		...options.headers
+	}, `neondatabase/ai-sdk-provider/${VERSION}`);
+	const createAnthropicModel = (modelId) => new NeonAnthropicLanguageModel(modelId, {
+		provider: "neon.anthropic",
+		baseURL: `${getHost()}/ai-gateway/anthropic/v1`,
+		headers: () => getHeaders({ "anthropic-version": "2023-06-01" }),
+		fetch: options.fetch,
+		generateId
+	});
+	const createOpenAIModel = (modelId) => new NeonResponsesLanguageModel(modelId, {
+		provider: "neon.openai.responses",
+		url: ({ path }) => `${getHost()}/ai-gateway/openai/v1${path}`,
+		headers: getHeaders,
+		fetch: options.fetch,
+		fileIdPrefixes: ["file-"]
+	});
+	const createChatModel = (modelId) => new NeonChatLanguageModel(modelId, {
+		provider: "neon.chat",
+		url: ({ path }) => `${getHost()}/ai-gateway/mlflow/v1${path}`,
+		headers: getHeaders,
+		fetch: options.fetch,
+		errorStructure: neonErrorStructure,
+		transformRequestBody: transformNeonRequestBody,
+		supportsStructuredOutputs: true
+	});
+	const createLanguageModel = (modelId) => {
+		switch (getNeonModelRoute(modelId)) {
+			case "anthropic": return createAnthropicModel(modelId);
+			case "openai": return createOpenAIModel(modelId);
+			default: return createChatModel(modelId);
+		}
+	};
+	const provider = (modelId) => createLanguageModel(modelId);
+	provider.specificationVersion = "v4";
+	provider.languageModel = createLanguageModel;
+	provider.chat = createLanguageModel;
+	provider.embeddingModel = (modelId) => {
+		throw new NoSuchModelError({
+			modelId,
+			modelType: "embeddingModel"
+		});
+	};
+	provider.textEmbeddingModel = provider.embeddingModel;
+	provider.imageModel = (modelId) => {
+		throw new NoSuchModelError({
+			modelId,
+			modelType: "imageModel"
+		});
+	};
+	return provider;
+}
+/** Default Neon provider instance (reads `NEON_AI_GATEWAY_*` env vars). */
+const neon = createNeon();
+//#endregion
+export { createNeon, neon };
+
+//# sourceMappingURL=provider.js.map

package/dist/lib/provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.js","names":[],"sources":["../../src/lib/provider.ts"],"sourcesContent":["/**\n * Community Vercel AI SDK provider for the Neon AI Gateway.\n *\n * `createNeon()` returns a provider that routes each model to the best gateway\n * endpoint based on its id (Anthropic → native Messages, OpenAI → native\n * Responses incl. Codex, everything else → unified MLflow), so a single\n * `neon('databricks-...')` call (same base URL + token) reaches the whole\n * catalog. Configure with the branch-scoped `NEON_AI_GATEWAY_BASE_URL` +\n * `NEON_AI_GATEWAY_TOKEN` emitted by `neonctl env pull` / `neon dev`, or pass\n * `baseURL` / `apiKey` explicitly.\n */\nimport type { ProviderErrorStructure } from \"@ai-sdk/openai-compatible\";\nimport {\n\ttype LanguageModelV4,\n\tNoSuchModelError,\n\ttype ProviderV4,\n} from \"@ai-sdk/provider\";\nimport {\n\ttype FetchFunction,\n\tgenerateId,\n\tloadApiKey,\n\tloadSetting,\n\twithoutTrailingSlash,\n\twithUserAgentSuffix,\n} from \"@ai-sdk/provider-utils\";\nimport { z } from \"zod/v4\";\nimport { NeonAnthropicLanguageModel } from \"./neon-anthropic-language-model.js\";\nimport { NeonChatLanguageModel } from \"./neon-chat-language-model.js\";\nimport type { NeonChatModelId } from \"./neon-chat-options.js\";\nimport { getNeonModelRoute } from \"./neon-model-capabilities.js\";\nimport { NeonResponsesLanguageModel } from \"./neon-responses-language-model.js\";\nimport { VERSION } from \"./version.js\";\n\nconst neonErrorSchema = z.object({\n\terror: z.object({\n\t\tmessage: z.string(),\n\t\ttype: z.string().nullish(),\n\t\tparam: z.unknown().nullish(),\n\t\tcode: z.unknown().nullish(),\n\t}),\n});\n\nexport type NeonErrorData = z.infer<typeof neonErrorSchema>;\n\nconst neonErrorStructure: ProviderErrorStructure<NeonErrorData> = {\n\terrorSchema: neonErrorSchema,\n\terrorToMessage: (data) => data.error.message,\n};\n\n/**\n * Recursively remove the JSON Schema `$schema` marker, which some gateway\n * backends (notably Gemini) reject in tool/structured-output schemas. Other\n * backends ignore its absence, so stripping it everywhere is safe.\n */\nfunction stripJsonSchemaMarker(value: unknown): unknown {\n\tif (Array.isArray(value)) {\n\t\treturn value.map(stripJsonSchemaMarker);\n\t}\n\tif (value !== null && typeof value === \"object\") {\n\t\tconst result: Record<string, unknown> = {};\n\t\tfor (const [key, entry] of Object.entries(value)) {\n\t\t\tif (key === \"$schema\") {\n\t\t\t\tcontinue;\n\t\t\t}\n\t\t\tresult[key] = stripJsonSchemaMarker(entry);\n\t\t}\n\t\treturn result;\n\t}\n\treturn value;\n}\n\nfunction transformNeonRequestBody(\n\targs: Record<string, unknown>,\n): Record<string, unknown> {\n\tconst transformed = { ...args };\n\tif (transformed.tools != null) {\n\t\ttransformed.tools = stripJsonSchemaMarker(transformed.tools);\n\t}\n\tif (transformed.response_format != null) {\n\t\ttransformed.response_format = stripJsonSchemaMarker(\n\t\t\ttransformed.response_format,\n\t\t);\n\t}\n\treturn transformed;\n}\n\nexport interface NeonProviderSettings {\n\t/**\n\t * Neon AI Gateway base URL — the branch-scoped host root, e.g.\n\t * `https://<branch-id>-api.ai.<region>.aws.neon.tech`. The gateway paths are\n\t * appended internally. Falls back to the `NEON_AI_GATEWAY_BASE_URL` env var.\n\t */\n\tbaseURL?: string;\n\n\t/**\n\t * Neon AI Gateway platform token (the `nt_live_...` value). Falls back to the\n\t * `NEON_AI_GATEWAY_TOKEN` env var.\n\t */\n\tapiKey?: string;\n\n\t/** Custom headers to include in the requests. */\n\theaders?: Record<string, string>;\n\n\t/** Custom fetch implementation (e.g. for testing or middleware). */\n\tfetch?: FetchFunction;\n}\n\nexport interface NeonProvider extends ProviderV4 {\n\t/** Creates a Neon AI Gateway model for text generation. */\n\t(modelId: NeonChatModelId): LanguageModelV4;\n\n\t/** Creates a Neon AI Gateway model for text generation. */\n\tlanguageModel(modelId: NeonChatModelId): LanguageModelV4;\n\n\t/** Creates a Neon AI Gateway chat model for text generation. */\n\tchat(modelId: NeonChatModelId): LanguageModelV4;\n\n\t/** @deprecated Use `embeddingModel` instead. */\n\ttextEmbeddingModel(modelId: string): never;\n}\n\nexport function createNeon(options: NeonProviderSettings = {}): NeonProvider {\n\t// Resolved lazily so that `createNeon()` and the default `neon` instance do\n\t// not throw at import time when configuration comes from the environment.\n\tconst getHost = () =>\n\t\twithoutTrailingSlash(\n\t\t\tloadSetting({\n\t\t\t\tsettingValue: options.baseURL,\n\t\t\t\tenvironmentVariableName: \"NEON_AI_GATEWAY_BASE_URL\",\n\t\t\t\tsettingName: \"baseURL\",\n\t\t\t\tdescription: \"Neon AI Gateway base URL\",\n\t\t\t}),\n\t\t);\n\n\tconst getHeaders = (extra?: Record<string, string>) =>\n\t\twithUserAgentSuffix(\n\t\t\t{\n\t\t\t\tAuthorization: `Bearer ${loadApiKey({\n\t\t\t\t\tapiKey: options.apiKey,\n\t\t\t\t\tenvironmentVariableName: \"NEON_AI_GATEWAY_TOKEN\",\n\t\t\t\t\tdescription: \"Neon AI Gateway token\",\n\t\t\t\t})}`,\n\t\t\t\t...extra,\n\t\t\t\t...options.headers,\n\t\t\t},\n\t\t\t`neondatabase/ai-sdk-provider/${VERSION}`,\n\t\t);\n\n\t// Anthropic models -> native Messages API.\n\tconst createAnthropicModel = (modelId: NeonChatModelId) =>\n\t\tnew NeonAnthropicLanguageModel(modelId, {\n\t\t\tprovider: \"neon.anthropic\",\n\t\t\tbaseURL: `${getHost()}/ai-gateway/anthropic/v1`,\n\t\t\theaders: () => getHeaders({ \"anthropic-version\": \"2023-06-01\" }),\n\t\t\tfetch: options.fetch,\n\t\t\tgenerateId,\n\t\t});\n\n\t// OpenAI models (incl. Codex, only served natively) -> Responses API.\n\tconst createOpenAIModel = (modelId: NeonChatModelId) =>\n\t\tnew NeonResponsesLanguageModel(modelId, {\n\t\t\tprovider: \"neon.openai.responses\",\n\t\t\turl: ({ path }) => `${getHost()}/ai-gateway/openai/v1${path}`,\n\t\t\theaders: getHeaders,\n\t\t\tfetch: options.fetch,\n\t\t\tfileIdPrefixes: [\"file-\"],\n\t\t});\n\n\t// Everything else (Gemini, Llama, Qwen, gpt-oss, ...) -> unified MLflow\n\t// endpoint. Gemini is here because its native endpoint can't stream.\n\tconst createChatModel = (modelId: NeonChatModelId) =>\n\t\tnew NeonChatLanguageModel(modelId, {\n\t\t\tprovider: \"neon.chat\",\n\t\t\turl: ({ path }) => `${getHost()}/ai-gateway/mlflow/v1${path}`,\n\t\t\theaders: getHeaders,\n\t\t\tfetch: options.fetch,\n\t\t\terrorStructure: neonErrorStructure,\n\t\t\ttransformRequestBody: transformNeonRequestBody,\n\t\t\tsupportsStructuredOutputs: true,\n\t\t});\n\n\tconst createLanguageModel = (modelId: NeonChatModelId): LanguageModelV4 => {\n\t\tswitch (getNeonModelRoute(modelId)) {\n\t\t\tcase \"anthropic\":\n\t\t\t\treturn createAnthropicModel(modelId);\n\t\t\tcase \"openai\":\n\t\t\t\treturn createOpenAIModel(modelId);\n\t\t\tdefault:\n\t\t\t\treturn createChatModel(modelId);\n\t\t}\n\t};\n\n\tconst provider = (modelId: NeonChatModelId) => createLanguageModel(modelId);\n\n\tprovider.specificationVersion = \"v4\" as const;\n\tprovider.languageModel = createLanguageModel;\n\tprovider.chat = createLanguageModel;\n\n\tprovider.embeddingModel = (modelId: string) => {\n\t\tthrow new NoSuchModelError({ modelId, modelType: \"embeddingModel\" });\n\t};\n\tprovider.textEmbeddingModel = provider.embeddingModel;\n\tprovider.imageModel = (modelId: string) => {\n\t\tthrow new NoSuchModelError({ modelId, modelType: \"imageModel\" });\n\t};\n\n\treturn provider;\n}\n\n/** Default Neon provider instance (reads `NEON_AI_GATEWAY_*` env vars). */\nexport const neon = createNeon();\n"],"mappings":";;;;;;;;;AA4CA,MAAM,qBAA4D;CACjE,aAZuB,EAAE,OAAO,EAChC,OAAO,EAAE,OAAO;EACf,SAAS,EAAE,OAAO;EAClB,MAAM,EAAE,OAAO,CAAC,CAAC,QAAQ;EACzB,OAAO,EAAE,QAAQ,CAAC,CAAC,QAAQ;EAC3B,MAAM,EAAE,QAAQ,CAAC,CAAC,QAAQ;CAC3B,CAAC,EACF,CAK4B;CAC3B,iBAAiB,SAAS,KAAK,MAAM;AACtC;;;;;;AAOA,SAAS,sBAAsB,OAAyB;CACvD,IAAI,MAAM,QAAQ,KAAK,GACtB,OAAO,MAAM,IAAI,qBAAqB;CAEvC,IAAI,UAAU,QAAQ,OAAO,UAAU,UAAU;EAChD,MAAM,SAAkC,CAAC;EACzC,KAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,KAAK,GAAG;GACjD,IAAI,QAAQ,WACX;GAED,OAAO,OAAO,sBAAsB,KAAK;EAC1C;EACA,OAAO;CACR;CACA,OAAO;AACR;AAEA,SAAS,yBACR,MAC0B;CAC1B,MAAM,cAAc,EAAE,GAAG,KAAK;CAC9B,IAAI,YAAY,SAAS,MACxB,YAAY,QAAQ,sBAAsB,YAAY,KAAK;CAE5D,IAAI,YAAY,mBAAmB,MAClC,YAAY,kBAAkB,sBAC7B,YAAY,eACb;CAED,OAAO;AACR;AAqCA,SAAgB,WAAW,UAAgC,CAAC,GAAiB;CAG5E,MAAM,gBACL,qBACC,YAAY;EACX,cAAc,QAAQ;EACtB,yBAAyB;EACzB,aAAa;EACb,aAAa;CACd,CAAC,CACF;CAED,MAAM,cAAc,UACnB,oBACC;EACC,eAAe,UAAU,WAAW;GACnC,QAAQ,QAAQ;GAChB,yBAAyB;GACzB,aAAa;EACd,CAAC;EACD,GAAG;EACH,GAAG,QAAQ;CACZ,GACA,gCAAgC,SACjC;CAGD,MAAM,wBAAwB,YAC7B,IAAI,2BAA2B,SAAS;EACvC,UAAU;EACV,SAAS,GAAG,QAAQ,EAAE;EACtB,eAAe,WAAW,EAAE,qBAAqB,aAAa,CAAC;EAC/D,OAAO,QAAQ;EACf;CACD,CAAC;CAGF,MAAM,qBAAqB,YAC1B,IAAI,2BAA2B,SAAS;EACvC,UAAU;EACV,MAAM,EAAE,WAAW,GAAG,QAAQ,EAAE,uBAAuB;EACvD,SAAS;EACT,OAAO,QAAQ;EACf,gBAAgB,CAAC,OAAO;CACzB,CAAC;CAIF,MAAM,mBAAmB,YACxB,IAAI,sBAAsB,SAAS;EAClC,UAAU;EACV,MAAM,EAAE,WAAW,GAAG,QAAQ,EAAE,uBAAuB;EACvD,SAAS;EACT,OAAO,QAAQ;EACf,gBAAgB;EAChB,sBAAsB;EACtB,2BAA2B;CAC5B,CAAC;CAEF,MAAM,uBAAuB,YAA8C;EAC1E,QAAQ,kBAAkB,OAAO,GAAjC;GACC,KAAK,aACJ,OAAO,qBAAqB,OAAO;GACpC,KAAK,UACJ,OAAO,kBAAkB,OAAO;GACjC,SACC,OAAO,gBAAgB,OAAO;EAChC;CACD;CAEA,MAAM,YAAY,YAA6B,oBAAoB,OAAO;CAE1E,SAAS,uBAAuB;CAChC,SAAS,gBAAgB;CACzB,SAAS,OAAO;CAEhB,SAAS,kBAAkB,YAAoB;EAC9C,MAAM,IAAI,iBAAiB;GAAE;GAAS,WAAW;EAAiB,CAAC;CACpE;CACA,SAAS,qBAAqB,SAAS;CACvC,SAAS,cAAc,YAAoB;EAC1C,MAAM,IAAI,iBAAiB;GAAE;GAAS,WAAW;EAAa,CAAC;CAChE;CAEA,OAAO;AACR;;AAGA,MAAa,OAAO,WAAW"}

package/dist/lib/version.d.ts

@@ -0,0 +1,5 @@
+//#region src/lib/version.d.ts
+declare const VERSION: string;
+//#endregion
+export { VERSION };
+//# sourceMappingURL=version.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"version.d.ts","names":[],"sources":["../../src/lib/version.ts"],"mappings":";cAIa"}

package/dist/lib/version.js

@@ -0,0 +1,7 @@
+import { version } from "../package.js";
+//#region src/lib/version.ts
+const VERSION = version;
+//#endregion
+export { VERSION };
+
+//# sourceMappingURL=version.js.map

package/dist/lib/version.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"version.js","names":["pkg.version"],"sources":["../../src/lib/version.ts"],"sourcesContent":["import pkg from \"../../package.json\" with { type: \"json\" };\n\n// Package version, used in the User-Agent suffix sent to the gateway. Derived\n// from package.json so it stays in sync with the changeset version bump flow.\nexport const VERSION: string = pkg.version;\n"],"mappings":";;AAIA,MAAa,UAAkBA"}

package/dist/package.js

@@ -0,0 +1,6 @@
+//#region package.json
+var version = "0.1.1";
+//#endregion
+export { version };
+
+//# sourceMappingURL=package.js.map

package/dist/package.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"package.js","names":[],"sources":["../package.json"],"sourcesContent":[""],"mappings":""}

package/dist/v1.d.ts

@@ -0,0 +1,7 @@
+import { NeonAnthropicLanguageModel } from "./lib/neon-anthropic-language-model.js";
+import { NeonChatLanguageModel } from "./lib/neon-chat-language-model.js";
+import { NeonChatModelId } from "./lib/neon-chat-options.js";
+import { NeonModelCapabilities, NeonModelFamily, NeonModelRoute, getNeonModelCapabilities, getNeonModelRoute } from "./lib/neon-model-capabilities.js";
+import { NeonResponsesLanguageModel } from "./lib/neon-responses-language-model.js";
+import { NeonErrorData, NeonProvider, NeonProviderSettings, createNeon, neon } from "./lib/provider.js";
+export { NeonAnthropicLanguageModel, NeonChatLanguageModel, type NeonChatModelId, type NeonErrorData, type NeonModelCapabilities, type NeonModelFamily, type NeonModelRoute, type NeonProvider, type NeonProviderSettings, NeonResponsesLanguageModel, createNeon, getNeonModelCapabilities, getNeonModelRoute, neon };

package/dist/v1.js

@@ -0,0 +1,6 @@
+import { NeonAnthropicLanguageModel } from "./lib/neon-anthropic-language-model.js";
+import { getNeonModelCapabilities, getNeonModelRoute } from "./lib/neon-model-capabilities.js";
+import { NeonChatLanguageModel } from "./lib/neon-chat-language-model.js";
+import { NeonResponsesLanguageModel } from "./lib/neon-responses-language-model.js";
+import { createNeon, neon } from "./lib/provider.js";
+export { NeonAnthropicLanguageModel, NeonChatLanguageModel, NeonResponsesLanguageModel, createNeon, getNeonModelCapabilities, getNeonModelRoute, neon };

package/package.json

@@ -1,24 +1,75 @@
 {
-	"name": "@neondatabase/ai-sdk-provider",
-	"version": "0.0.0",
-	"description": "Community Vercel AI SDK provider for the Neon AI Gateway.",
-	"keywords": [
-		"neon",
-		"ai",
-		"ai-sdk",
-		"vercel-ai-sdk",
-		"provider",
-		"ai-gateway",
-		"llm",
-		"platform"
-	],
-	"repository": {
-		"type": "git",
-		"url": "git+https://github.com/neondatabase/neon-pkgs.git"
-	},
-	"license": "Apache-2.0",
-	"author": {
-		"name": "Neon",
-		"url": "https://neon.com"
-	}
-}
+  "name": "@neondatabase/ai-sdk-provider",
+  "version": "0.1.1",
+  "description": "Community Vercel AI SDK provider for the Neon AI Gateway.",
+  "keywords": [
+    "neon",
+    "ai",
+    "ai-sdk",
+    "vercel-ai-sdk",
+    "provider",
+    "ai-gateway",
+    "llm",
+    "platform"
+  ],
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/neondatabase/neon-pkgs.git"
+  },
+  "license": "Apache-2.0",
+  "author": {
+    "name": "Neon",
+    "url": "https://neon.com"
+  },
+  "type": "module",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js",
+      "default": "./dist/index.js"
+    },
+    "./v1": {
+      "types": "./dist/v1.d.ts",
+      "import": "./dist/v1.js",
+      "default": "./dist/v1.js"
+    }
+  },
+  "files": [
+    "README.md",
+    "dist/",
+    "package.json"
+  ],
+  "devDependencies": {
+    "@types/node": "^20.19.0",
+    "@vitest/coverage-v8": "3.0.9",
+    "console-fail-test": "0.5.0",
+    "tsdown": "^0.14.1",
+    "typescript": "^5.8.2",
+    "vitest": "^3.0.9",
+    "zod": "^4.4.3"
+  },
+  "engines": {
+    "node": ">=22"
+  },
+  "publishConfig": {
+    "provenance": false
+  },
+  "dependencies": {
+    "@ai-sdk/anthropic": "4.0.0-beta.42",
+    "@ai-sdk/openai": "4.0.0-beta.44",
+    "@ai-sdk/openai-compatible": "3.0.0-beta.36",
+    "@ai-sdk/provider": "4.0.0-beta.14",
+    "@ai-sdk/provider-utils": "5.0.0-beta.30"
+  },
+  "peerDependencies": {
+    "zod": "^3.25.76 || ^4.1.8"
+  },
+  "scripts": {
+    "build": "tsc --noEmit && tsdown",
+    "test": "vitest --passWithNoTests",
+    "test:ci": "vitest run --passWithNoTests",
+    "tsc": "tsc"
+  }
+}

package/src/index.ts

@@ -1,5 +0,0 @@
-/**
- * The default entry point re-exports the latest stable version. New consumers should import
- * directly from `@neondatabase/ai-sdk-provider/v1` to opt in to a specific major.
- */
-export * from "./v1.js";

package/src/lib/neon-anthropic-language-model.ts

@@ -1,51 +0,0 @@
-import { AnthropicLanguageModel } from "@ai-sdk/anthropic/internal";
-import type {
-	LanguageModelV4,
-	LanguageModelV4CallOptions,
-	LanguageModelV4GenerateResult,
-	LanguageModelV4StreamResult,
-} from "@ai-sdk/provider";
-
-/**
- * Language model for Anthropic models served via the Neon AI Gateway's native
- * Messages API (`/ai-gateway/anthropic/v1/messages`). Unlocks streaming
- * structured output and native reasoning for Claude.
- *
- * The shared Anthropic model defaults to fine-grained tool-input streaming
- * (`eager_input_streaming: true`) on streaming tool calls, which the gateway
- * rejects (`Extra inputs are not permitted`). We disable it via the model's own
- * `toolStreaming` option so streaming tool calls work. Users can still override.
- */
-export class NeonAnthropicLanguageModel
-	extends AnthropicLanguageModel
-	implements LanguageModelV4
-{
-	private withGatewayCompat(
-		options: LanguageModelV4CallOptions,
-	): LanguageModelV4CallOptions {
-		const anthropic = options.providerOptions?.anthropic;
-		// Respect an explicit user setting.
-		if (anthropic != null && "toolStreaming" in anthropic) {
-			return options;
-		}
-		return {
-			...options,
-			providerOptions: {
-				...options.providerOptions,
-				anthropic: { ...anthropic, toolStreaming: false },
-			},
-		};
-	}
-
-	override doGenerate(
-		options: LanguageModelV4CallOptions,
-	): Promise<LanguageModelV4GenerateResult> {
-		return super.doGenerate(this.withGatewayCompat(options));
-	}
-
-	override doStream(
-		options: LanguageModelV4CallOptions,
-	): Promise<LanguageModelV4StreamResult> {
-		return super.doStream(this.withGatewayCompat(options));
-	}
-}

package/src/lib/neon-capabilities.ts

@@ -1,133 +0,0 @@
-import type {
-	LanguageModelV4CallOptions,
-	LanguageModelV4StreamPart,
-	SharedV4ProviderOptions,
-	SharedV4Warning,
-} from "@ai-sdk/provider";
-import { isCustomReasoning } from "@ai-sdk/provider-utils";
-import { getNeonModelCapabilities } from "./neon-model-capabilities.js";
-
-/**
- * Drop call options the resolved model's upstream backend does not accept and
- * collect a warning for each, so callers get a clear signal instead of a hard
- * `400` from the gateway.
- */
-export function applyNeonCapabilities(
-	modelId: string,
-	options: LanguageModelV4CallOptions,
-): { options: LanguageModelV4CallOptions; warnings: SharedV4Warning[] } {
-	const caps = getNeonModelCapabilities(modelId);
-	const warnings: SharedV4Warning[] = [];
-	const patch: Partial<LanguageModelV4CallOptions> = {};
-
-	const dropUnsupported = (feature: string) => {
-		warnings.push({
-			type: "unsupported",
-			feature,
-			details: `${feature} is not supported by the Neon AI Gateway for ${caps.family} model "${modelId}" and was dropped.`,
-		});
-	};
-
-	if (options.temperature != null && !caps.supportsTemperature) {
-		patch.temperature = undefined;
-		dropUnsupported("temperature");
-	}
-	if (options.topP != null && !caps.supportsTopP) {
-		patch.topP = undefined;
-		dropUnsupported("topP");
-	}
-
-	// Anthropic-style models accept only one of temperature / topP.
-	const effectiveTemperature =
-		"temperature" in patch ? patch.temperature : options.temperature;
-	const effectiveTopP = "topP" in patch ? patch.topP : options.topP;
-	if (
-		caps.temperatureTopPMutuallyExclusive &&
-		effectiveTemperature != null &&
-		effectiveTopP != null
-	) {
-		patch.topP = undefined;
-		warnings.push({
-			type: "compatibility",
-			feature: "topP",
-			details: `${caps.family} models accept only one of temperature or topP; topP was dropped.`,
-		});
-	}
-
-	if (options.frequencyPenalty != null && !caps.supportsPenalties) {
-		patch.frequencyPenalty = undefined;
-		dropUnsupported("frequencyPenalty");
-	}
-	if (options.presencePenalty != null && !caps.supportsPenalties) {
-		patch.presencePenalty = undefined;
-		dropUnsupported("presencePenalty");
-	}
-	if (options.seed != null && !caps.supportsSeed) {
-		patch.seed = undefined;
-		dropUnsupported("seed");
-	}
-	if (options.stopSequences != null && !caps.supportsStopSequences) {
-		patch.stopSequences = undefined;
-		dropUnsupported("stopSequences");
-	}
-
-	if (!caps.supportsReasoningEffort) {
-		const { providerOptions } = options;
-		const hasProviderEffort =
-			providerOptions != null &&
-			Object.values(providerOptions).some(
-				(group) =>
-					"reasoningEffort" in group && group.reasoningEffort != null,
-			);
-		const hasReasoningOption =
-			isCustomReasoning(options.reasoning) &&
-			options.reasoning !== "none";
-
-		if (hasProviderEffort || hasReasoningOption) {
-			dropUnsupported("reasoningEffort");
-			if (hasProviderEffort && providerOptions != null) {
-				const cleaned: SharedV4ProviderOptions = {};
-				for (const [key, group] of Object.entries(providerOptions)) {
-					if ("reasoningEffort" in group) {
-						const { reasoningEffort: _removed, ...rest } = group;
-						cleaned[key] = rest;
-					} else {
-						cleaned[key] = group;
-					}
-				}
-				patch.providerOptions = cleaned;
-			}
-			if (hasReasoningOption) {
-				patch.reasoning = undefined;
-			}
-		}
-	}
-
-	if (Object.keys(patch).length === 0) {
-		return { options, warnings };
-	}
-	return { options: { ...options, ...patch }, warnings };
-}
-
-/**
- * Merge additional warnings into the `stream-start` part of a model stream.
- */
-export function mergeStreamStartWarnings(extra: SharedV4Warning[]) {
-	let merged = false;
-	return new TransformStream<
-		LanguageModelV4StreamPart,
-		LanguageModelV4StreamPart
-	>({
-		transform(part, controller) {
-			if (!merged && part.type === "stream-start") {
-				merged = true;
-				controller.enqueue({
-					...part,
-					warnings: [...extra, ...part.warnings],
-				});
-			} else {
-				controller.enqueue(part);
-			}
-		},
-	});
-}