phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix gemini test

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-26 19:00:30 -07:00
parent 3dc578555c
commit 4707c4fccd
1 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/llms/vertex_httpx.py
View file

@ -183,10 +183,17 @@ class GoogleAIStudioGeminiConfig: # key diff from VertexAI - 'frequency_penalty
if param == "tools" and isinstance(value, list): if param == "tools" and isinstance(value, list):
gtool_func_declarations = [] gtool_func_declarations = []
for tool in value: for tool in value:
_parameters = tool.get("function", {}).get("parameters", {})
_properties = _parameters.get("properties", {})
if isinstance(_properties, dict):
for _, _property in _properties.items():
if "enum" in _property and "format" not in _property:
_property["format"] = "enum"
gtool_func_declaration = FunctionDeclaration( gtool_func_declaration = FunctionDeclaration(
name=tool["function"]["name"], name=tool["function"]["name"],
description=tool["function"].get("description", ""), description=tool["function"].get("description", ""),
parameters=tool["function"].get("parameters", {}), parameters=_parameters,
) )
gtool_func_declarations.append(gtool_func_declaration) gtool_func_declarations.append(gtool_func_declaration)
optional_params["tools"] = [ optional_params["tools"] = [