Skip to content

A
aisbf
Commit 8628e8d3 authored by Stefy Lanza (nextime / spora )'s avatar Stefy Lanza (nextime / spora )
Browse files
Options

Fix Google provider tool formatting - use genai_types alias for types module

parent e73e4a1e
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
aisbf/providers.py
View file @ 8628e8d3
...@@ -304,8 +304,8 @@ class GoogleProviderHandler(BaseProviderHandler): ...@@ -304,8 +304,8 @@ class GoogleProviderHandler(BaseProviderHandler):
if tool.get("type") == "function": if tool.get("type") == "function":
function = tool.get("function", {}) function = tool.get("function", {})
# Use Google's SDK types for proper validation # Use Google's SDK types for proper validation
from google.genai import types from google.genai import types as genai_types
function_declaration = types.FunctionDeclaration( function_declaration = genai_types.FunctionDeclaration(
name=function.get("name"), name=function.get("name"),
description=function.get("description", ""), description=function.get("description", ""),
parameters=function.get("parameters", {}) 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