Fix Google provider tool formatting - import FunctionDeclaration from genai.types

parent 58dd90df
......@@ -304,8 +304,8 @@ class GoogleProviderHandler(BaseProviderHandler):
if tool.get("type") == "function":
function = tool.get("function", {})
# Use Google's SDK types for proper validation
from google import genai
function_declaration = genai.FunctionDeclaration(
from google.genai import types
function_declaration = types.FunctionDeclaration(
name=function.get("name"),
description=function.get("description", ""),
parameters=function.get("parameters", {})
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment