@tanstack/ai-gemini 0.24.0 → 0.24.2

package/src/tools/google-search-tool.ts

@@ -1,4 +1,7 @@
-import { brandProviderTool } from '@tanstack/ai'
+import {
+  brandGeminiProviderTool,
+  getGeminiProviderToolMetadata,
+} from './gemini-provider-tool'
 import type { GoogleSearch } from '@google/genai'
 import type { ProviderTool, Tool } from '@tanstack/ai'
 
@@ -10,7 +13,7 @@ export type GoogleSearchTool = GoogleSearchToolConfig
 export type GeminiGoogleSearchTool = ProviderTool<'gemini', 'google_search'>
 
 export function convertGoogleSearchToolToAdapterFormat(tool: Tool) {
-  const metadata = tool.metadata as GoogleSearchToolConfig
+  const metadata = getGeminiProviderToolMetadata(tool) as GoogleSearchToolConfig
   return {
     googleSearch: metadata,
   }
@@ -19,9 +22,12 @@ export function convertGoogleSearchToolToAdapterFormat(tool: Tool) {
 export function googleSearchTool(
   config?: GoogleSearchToolConfig,
 ): GeminiGoogleSearchTool {
-  return brandProviderTool<GeminiGoogleSearchTool>({
-    name: 'google_search',
-    description: '',
-    metadata: config,
-  })
+  return brandGeminiProviderTool<GeminiGoogleSearchTool>(
+    {
+      name: 'google_search',
+      description: '',
+      metadata: config,
+    },
+    'google_search',
+  )
 }

package/src/tools/tool-converter.ts

@@ -1,6 +1,8 @@
+import { assertUniqueToolNames } from '@tanstack/ai/adapter-internals'
 import { convertCodeExecutionToolToAdapterFormat } from './code-execution-tool'
 import { convertComputerUseToolToAdapterFormat } from './computer-use-tool'
 import { convertFileSearchToolToAdapterFormat } from './file-search-tool'
+import { getGeminiProviderToolKind } from './gemini-provider-tool'
 import { convertGoogleMapsToolToAdapterFormat } from './google-maps-tool'
 import { convertGoogleSearchRetrievalToolToAdapterFormat } from './google-search-retriveal-tool'
 import { convertGoogleSearchToolToAdapterFormat } from './google-search-tool'
@@ -33,6 +35,7 @@ export function convertToolsToProviderFormat<TTool extends Tool>(
   if (!tools || tools.length === 0) {
     return []
   }
+  assertUniqueToolNames(tools)
   const result: Array<ToolUnion> = []
   const functionDeclarations: Array<{
     name: string
@@ -42,9 +45,7 @@ export function convertToolsToProviderFormat<TTool extends Tool>(
 
   // Process each tool and group function declarations together
   for (const tool of tools) {
-    const name = tool.name
-
-    switch (name) {
+    switch (getGeminiProviderToolKind(tool)) {
       case 'code_execution':
         result.push(convertCodeExecutionToolToAdapterFormat(tool))
         break
@@ -66,7 +67,7 @@ export function convertToolsToProviderFormat<TTool extends Tool>(
       case 'url_context':
         result.push(convertUrlContextToolToAdapterFormat(tool))
         break
-      default:
+      case undefined:
         // Collect function declarations to group together
         // Description is required for Gemini function declarations
         if (!tool.description) {

package/src/tools/url-context-tool.ts

@@ -1,4 +1,4 @@
-import { brandProviderTool } from '@tanstack/ai'
+import { brandGeminiProviderTool } from './gemini-provider-tool'
 import type { ProviderTool, Tool } from '@tanstack/ai'
 
 export interface UrlContextToolConfig {}
@@ -15,9 +15,12 @@ export function convertUrlContextToolToAdapterFormat(_tool: Tool) {
 }
 
 export function urlContextTool(): GeminiUrlContextTool {
-  return brandProviderTool<GeminiUrlContextTool>({
-    name: 'url_context',
-    description: '',
-    metadata: {},
-  })
+  return brandGeminiProviderTool<GeminiUrlContextTool>(
+    {
+      name: 'url_context',
+      description: '',
+      metadata: {},
+    },
+    'url_context',
+  )
 }