diff --git a/.changeset/add-create-google-alias.md b/.changeset/add-create-google-alias.md new file mode 100644 index 000000000..47ec0a0ac --- /dev/null +++ b/.changeset/add-create-google-alias.md @@ -0,0 +1,5 @@ +--- +"ai-gateway-provider": patch +--- + +Add `createGoogle` as an alias for `createGoogleGenerativeAI` in the Google provider. This fixes the mismatch between the Cloudflare dashboard example (which uses `createGoogle()`) and the package export. Both names now work interchangeably. diff --git a/packages/ai-gateway-provider/README.md b/packages/ai-gateway-provider/README.md index b52f29eac..64d984fcc 100644 --- a/packages/ai-gateway-provider/README.md +++ b/packages/ai-gateway-provider/README.md @@ -180,6 +180,26 @@ const aigateway = createAiGateway({ - `retryDelayMs`: Delay between retries - `backoff`: Retry backoff strategy ('constant', 'linear', 'exponential') +### Google AI Studio Example + +```typescript +import { createAiGateway } from "ai-gateway-provider"; +import { createGoogle } from "ai-gateway-provider/providers/google"; +import { generateText } from "ai"; + +const aigateway = createAiGateway({ + accountId: "{CLOUDFLARE_ACCOUNT_ID}", + gateway: "{GATEWAY_NAME}", +}); + +const google = createGoogle({ apiKey: "{GOOGLE_API_KEY}" }); + +const { text } = await generateText({ + model: aigateway(google("gemini-1.5-pro")), + prompt: "Write a vegetarian lasagna recipe for 4 people.", +}); +``` + ## Supported Providers - OpenAI diff --git a/packages/ai-gateway-provider/src/providers/google.ts b/packages/ai-gateway-provider/src/providers/google.ts index 7ba580029..e00e4a2bd 100644 --- a/packages/ai-gateway-provider/src/providers/google.ts +++ b/packages/ai-gateway-provider/src/providers/google.ts @@ -4,3 +4,5 @@ import { authWrapper } from "../auth"; export const createGoogleGenerativeAI = ( ...args: Parameters ) => authWrapper(createGoogleGenerativeAIOriginal)(...args); + +export const createGoogle = createGoogleGenerativeAI; diff --git a/packages/ai-gateway-provider/src/providers/index.ts b/packages/ai-gateway-provider/src/providers/index.ts index a31b24b69..b9c7e91b1 100644 --- a/packages/ai-gateway-provider/src/providers/index.ts +++ b/packages/ai-gateway-provider/src/providers/index.ts @@ -7,7 +7,7 @@ export { createDeepgram } from "./deepgram"; export { createDeepSeek } from "./deepseek"; export { createElevenLabs } from "./elevenlabs"; export { createFireworks } from "./fireworks"; -export { createGoogleGenerativeAI } from "./google"; +export { createGoogle, createGoogleGenerativeAI } from "./google"; export { createVertex } from "./google-vertex"; export { createGroq } from "./groq"; export { createMistral } from "./mistral";