feat(adk): expose callable provider

Author: grabbou

apps/expo-example/src/components/adapters/adkSetupAdapter.ts

@@ -27,7 +27,7 @@ export const createAdkLanguageSetupAdapter = (
       'You are a helpful assistant in the Callstack React Native AI demo app.',
     availableTools: tools,
   })
-  const model = provider.languageModel()
+  const model = provider()
 
   return {
     model,

packages/adk/README.md

@@ -41,10 +41,8 @@ Then run `npx expo prebuild --clean` so the native Android project picks up the
 import { adk } from '@react-native-ai/adk'
 import { generateText } from 'ai'
 
-const model = adk.languageModel()
-
 const { text } = await generateText({
-  model,
+  model: adk(),
   prompt: 'What time is it in New York?',
 })
 ```
@@ -72,7 +70,7 @@ Pass file parts in user messages using the standard AI SDK prompt format:
 import { generateText } from 'ai'
 
 const { text } = await generateText({
-  model: adk.languageModel(),
+  model: adk(),
   messages: [
     {
       role: 'user',
@@ -105,10 +103,8 @@ const provider = createAdkProvider({
   instruction: 'You are a helpful assistant.',
 })
 
-const model = provider.languageModel()
-
 const { text } = await generateText({
-  model,
+  model: provider(),
   prompt: 'Summarize on-device AI in one sentence.',
 })
 ```
@@ -163,7 +159,7 @@ if (!ready) {
 }
 
 await provider.prepareNano()
-const model = provider.languageModel()
+const model = provider()
 ```
 
 Both checks are cheap native calls (no model download). Safe to cache at app startup and re-check after resume.
@@ -194,7 +190,7 @@ const provider = createAdkProvider({
 })
 
 const { text } = await generateText({
-  model: provider.languageModel(),
+  model: provider(),
   tools: { getCurrentTime },
   prompt: 'What time is it in Warsaw?',
 })

packages/adk/src/ai-sdk.ts

@@ -73,18 +73,19 @@ export function createAdkProvider(
   options: AdkProviderOptions = {}
 ): AdkProvider {
   const createLanguageModel = () => new AdkChatLanguageModel(options)
-
-  return Object.assign(() => createLanguageModel(), {
-    languageModel: createLanguageModel,
-    isNanoSupported: () => checkNanoSupported(),
-    isAvailable: async (modelType: AdkModelType = 'gemini') => {
-      if (modelType === 'genai-nano' && !(await checkNanoSupported())) {
-        return false
-      }
-      return getNativeAdkEngine().isAvailable(modelType).then(Boolean)
-    },
-    prepareNano: () => getNativeAdkEngine().prepareNano(),
-  })
+  const provider = function () {
+    return createLanguageModel()
+  }
+  provider.languageModel = createLanguageModel
+  provider.isNanoSupported = () => checkNanoSupported()
+  provider.isAvailable = async (modelType: AdkModelType = 'gemini') => {
+    if (modelType === 'genai-nano' && !(await checkNanoSupported())) {
+      return false
+    }
+    return getNativeAdkEngine().isAvailable(modelType).then(Boolean)
+  }
+  provider.prepareNano = () => getNativeAdkEngine().prepareNano()
+  return provider
 }
 
 export const adk = createAdkProvider()