Merge pull request #89 from olivier-lacroix/fix/native-detection

owndev · web-flow · commit 1a3b37b81263 · 2025-11-17T12:11:11.000+01:00
Fix native mode detection
diff --git a/pipelines/google/google_gemini.py b/pipelines/google/google_gemini.py
@@ -1413,8 +1413,10 @@ def _configure_generation(
                 types.Tool(google_search=types.GoogleSearch())
             )
 
-        if __tools__ is not None and __metadata__.get("function_calling") == "native":
-            for name, tool in __tools__.items():
+        params = __metadata__.get("params", {})
+        if __tools__ is not None and params.get("function_calling") == "native":
+            for name, tool_def in __tools__.items():
+                tool = self._create_tool(tool_def)
                 self.log.debug(
                     f"Adding tool '{name}' with signature {tool.__signature__}"
                 )

Original file line number	Diff line number	Diff line change
`@@ -1413,8 +1413,10 @@ def _configure_generation(`
`1413`	`1413`	`types.Tool(google_search=types.GoogleSearch())`
`1414`	`1414`	`)`
`1415`	`1415`
`1416`		`- if __tools__ is not None and __metadata__.get("function_calling") == "native":`
`1417`		`- for name, tool in __tools__.items():`
	`1416`	`+ params = __metadata__.get("params", {})`
	`1417`	`+ if __tools__ is not None and params.get("function_calling") == "native":`
	`1418`	`+ for name, tool_def in __tools__.items():`
	`1419`	`+ tool = self._create_tool(tool_def)`
`1418`	`1420`	`self.log.debug(`
`1419`	`1421`	`f"Adding tool '{name}' with signature {tool.__signature__}"`
`1420`	`1422`	`)`