feat: support specifying account id to wrangler

Author: RihanArfan

src/module.ts

@@ -30,6 +30,7 @@ export default defineNuxtModule<ModuleOptions>({
     remoteArg = (remoteArg === '' ? 'true' : remoteArg)
     const runtimeConfig = nuxt.options.runtimeConfig
     const hub = defu(runtimeConfig.hub || {}, options, {
+      accountId: process.env.CLOUDFLARE_ACCOUNT_ID || '',
       // Self-hosted project
       projectUrl: process.env.NUXT_HUB_PROJECT_URL || '',
       projectSecretKey: process.env.NUXT_HUB_PROJECT_SECRET_KEY || '',

src/runtime/ai/server/utils/ai.ts

@@ -23,7 +23,7 @@ export function hubAi(): Ai {
     return _ai
   }
   // @ts-expect-error globalThis.__env__ is not defined
-  const binding = process.env.AI || globalThis.__env__?.AI || globalThis.AI;
+  const binding = process.env.AI || globalThis.__env__?.AI || globalThis.AI
   if (binding) {
     _ai = binding as Ai
     return _ai

src/types/module.ts

@@ -2,6 +2,9 @@ export interface ModuleOptions {
   /**
    * Set `true` to enable AI for the project.
    *
+   * Requires running `npx wrangler login` for local development one time system wide.
+   * Set `accountId` if logged-in user has access to several accounts.
+   *
    * @default false
    * @see https://developers.cloudflare.com/workers-ai/configuration/bindings/#methods
    */
@@ -77,4 +80,10 @@ export interface ModuleOptions {
    * @default '.data/hub'
    */
   dir?: string
+  /**
+   * The account ID of the Cloudflare account to use for local development.
+   * Only necessary if using AI and logged-in user has access to multiple accounts.
+   * @default process.env.CLOUDFLARE_ACCOUNT_ID
+   */
+  accountId?: string
 }

src/utils/wrangler.ts

@@ -1,8 +1,12 @@
 import { stringifyTOML } from 'confbox'
 
-export function generateWrangler(hub: { kv: boolean, database: boolean, blob: boolean, analytics: boolean, ai: boolean }) {
+export function generateWrangler(hub: { kv: boolean, database: boolean, blob: boolean, analytics: boolean, ai: boolean, accountId?: string }) {
   const wrangler: { [key: string]: any } = {}
 
+  if (hub.accountId) {
+    wrangler['account_id'] = hub.accountId
+  }
+
   if (hub.ai) {
     wrangler['ai'] = {
       binding: 'AI'