scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 85

Add gemini tools support for responses api

Browse source
This commit is contained in:
Sameer Kankute 2025-12-15 16:45:23 +05:30
parent bba229f922
commit f59c2c1177
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/responses/litellm_completion_transformation/transformation.py
View file

@ -55,6 +55,8 @@ from litellm.types.utils import (
ModelResponse,
Usage,
)
from litellm.types.llms.vertex_ai import VertexToolName
########### Initialize Classes used for Responses API ###########
TOOL_CALLS_CACHE = InMemoryCache()
@ -691,6 +693,8 @@ class LiteLLMCompletionResponsesConfig:
search_context_size=_search_context_size,
user_location=_user_location,
)
elif len(tool) == 1 and next(iter(tool)) in {e.value for e in VertexToolName}:
chat_completion_tools.append(tool)
else:
typed_tool = cast(FunctionToolParam, tool)
# Ensure parameters has "type": "object" as required by providers like Anthropic