feat: ai gateway disabled check

src/commands/dev/dev.ts

@@ -155,7 +155,7 @@ export const dev = async (options: OptionValues, command: BaseCommand) => {
     siteInfo,
   })
 
-  if (!options.offline && !options.offlineEnv) {
+  if (!options.offline && !options.offlineEnv && !capabilities.aiGatewayDisabled) {
     await setupAIGateway({ api, env, siteID: site.id, siteURL: siteUrl })
 
     // Parse AI Gateway context and inject provider API keys
@@ -169,6 +169,8 @@ export const dev = async (options: OptionValues, command: BaseCommand) => {
         }
       }
     }
+  } else if (capabilities.aiGatewayDisabled) {
+    log(`${NETLIFYDEVLOG} AI Gateway is disabled for this account`)
   }
 
   injectEnvVariables(env)

src/commands/functions/functions-serve.ts

@@ -6,6 +6,7 @@ import { parseAIGatewayContext, setupAIGateway } from '@netlify/ai/bootstrap'
 
 import { getBlobsContextWithEdgeAccess } from '../../lib/blobs/blobs.js'
 import { startFunctionsServer } from '../../lib/functions/server.js'
+import { NETLIFYDEVLOG, log } from '../../utils/command-helpers.js'
 import { printBanner } from '../../utils/dev-server-banner.js'
 import {
   UNLINKED_SITE_MOCK_ID,
@@ -38,8 +39,10 @@ export const functionsServe = async (options: OptionValues, command: BaseCommand
     siteInfo,
   })
 
-  if (!options.offline) {
+  if (!options.offline && !capabilities.aiGatewayDisabled) {
     await setupAIGateway({ api, env, siteID: site.id, siteURL: siteUrl })
+  } else if (capabilities.aiGatewayDisabled) {
+    log(`${NETLIFYDEVLOG} AI Gateway is disabled for this account`)
   }
 
   injectEnvVariables(env)

src/utils/dev.ts

@@ -58,6 +58,11 @@ type Capabilities = NonNullable<ApiAccount['capabilities']> & {
         included?: boolean | undefined
       }
     | undefined
+  ai_gateway_disabled?:
+    | {
+        included?: boolean | undefined
+      }
+    | undefined
 }
 export type Capability = keyof Capabilities
 export type Account = ApiAccount & {
@@ -138,6 +143,7 @@ export const getSiteInformation = async ({ api, offline, site, siteInfo }) => {
       accountId: account?.id,
       capabilities: {
         backgroundFunctions: supportsBackgroundFunctions(account),
+        aiGatewayDisabled: account?.capabilities?.ai_gateway_disabled?.included ?? false,
       },
       timeouts: {
         syncFunctions: siteInfo.functions_timeout ?? siteInfo.functions_config?.timeout ?? SYNCHRONOUS_FUNCTION_TIMEOUT,
@@ -150,7 +156,9 @@ export const getSiteInformation = async ({ api, offline, site, siteInfo }) => {
   return {
     addonsUrls: {},
     siteUrl: '',
-    capabilities: {},
+    capabilities: {
+      aiGatewayDisabled: false,
+    },
     timeouts: {
       syncFunctions: SYNCHRONOUS_FUNCTION_TIMEOUT,
       backgroundFunctions: BACKGROUND_FUNCTION_TIMEOUT,

tests/integration/commands/dev/ai-gateway.test.ts

@@ -6,6 +6,7 @@ import { withSiteBuilder } from '../../utils/site-builder.js'
 import {
   assertAIGatewayValue,
   createAIGatewayCheckFunction,
+  createAIGatewayDisabledTestData,
   createAIGatewayTestData,
   createMockApiFailureRoutes,
 } from '../../utils/ai-gateway-helpers.js'
@@ -191,6 +192,42 @@ describe.concurrent('AI Gateway Integration', () => {
     })
   })
 
+  test('should not setup AI Gateway when account has ai_gateway_disabled', async (t) => {
+    await withSiteBuilder(t, async (builder) => {
+      const { siteInfo, routes } = createAIGatewayDisabledTestData()
+      const checkFunction = createAIGatewayCheckFunction()
+
+      await builder
+        .withContentFile({
+          path: checkFunction.path,
+          content: checkFunction.content,
+        })
+        .build()
+
+      await withMockApi(routes, async ({ apiUrl }) => {
+        await withDevServer(
+          {
+            cwd: builder.directory,
+            offline: false,
+            env: {
+              NETLIFY_API_URL: apiUrl,
+              NETLIFY_SITE_ID: siteInfo.id,
+              NETLIFY_AUTH_TOKEN: 'fake-token',
+              AI_GATEWAY: undefined,
+            },
+          },
+          async (server) => {
+            const response = await fetch(`${server.url}${checkFunction.urlPath}`)
+            const result = (await response.json()) as { hasAIGateway: boolean }
+
+            t.expect(response.status).toBe(200)
+            t.expect(result.hasAIGateway).toBe(false)
+          },
+        )
+      })
+    })
+  })
+
   test('should work with V2 functions', async (t) => {
     await withSiteBuilder(t, async (builder) => {
       const { siteInfo, aiGatewayToken, routes } = createAIGatewayTestData()

tests/integration/utils/ai-gateway-helpers.ts

@@ -83,3 +83,24 @@ export const createMockApiFailureRoutes = (siteInfo: {
   { path: 'sites/test-site-id/ai-gateway/token', status: 404, response: { message: 'Not Found' } },
   { path: 'ai-gateway/providers', status: 404, response: { message: 'Not Found' } },
 ]
+
+export const createAIGatewayDisabledTestData = () => {
+  const siteInfo = {
+    account_slug: 'test-account',
+    id: 'test-site-id',
+    name: 'site-name',
+    ssl_url: 'https://test-site.netlify.app',
+  }
+
+  const routes = [
+    { path: 'sites/test-site-id', response: siteInfo },
+    { path: 'sites/test-site-id/service-instances', response: [] },
+    {
+      path: 'accounts',
+      response: [{ slug: siteInfo.account_slug, capabilities: { ai_gateway_disabled: { included: true } } }],
+    },
+    { path: 'accounts/test-account/env', response: [] },
+  ]
+
+  return { siteInfo, routes }
+}