NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-16 04:02:55 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

Copybara import of the project:

Browse Source 
--
bdb2ddff1336db7fbeae71d3d76b4d294d7c4bea by Arpit Roopchandani <17565234+whoisarpit@users.noreply.github.com>:

Fix: Set output_schema correctly for LiteLllm
COPYBARA_INTEGRATE_REVIEW=https://github.com/google/adk-python/pull/580 from whoisarpit:fix/lite_llm_response_schema 08bdb5a993fc44adf03e985c820c2083de30ba58
PiperOrigin-RevId: 764524765
This commit is contained in:
Arpit R 2025-05-28 19:55:33 -07:00 committed by Copybara-Service
parent 9d4ca4ed44
commit 6157db77f2
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/google/adk/models/lite_llm.py
View File

@ -477,7 +477,7 @@ def _get_completion_inputs(
llm_request: The LlmRequest to convert. llm_request: The LlmRequest to convert.
Returns: Returns:
The litellm inputs (message list and tool dictionary). The litellm inputs (message list, tool dictionary and response format).
""" """
messages = [] messages = []
for content in llm_request.contents or []: for content in llm_request.contents or []:
@ -506,7 +506,13 @@ def _get_completion_inputs(
_function_declaration_to_tool_param(tool) _function_declaration_to_tool_param(tool)
for tool in llm_request.config.tools[0].function_declarations for tool in llm_request.config.tools[0].function_declarations
] ]
return messages, tools
response_format = None
if llm_request.config.response_schema:
response_format = llm_request.config.response_schema
return messages, tools, response_format
def _build_function_declaration_log( def _build_function_declaration_log(
@ -643,12 +649,13 @@ class LiteLlm(BaseLlm):
self._maybe_append_user_content(llm_request) self._maybe_append_user_content(llm_request)
logger.debug(_build_request_log(llm_request)) logger.debug(_build_request_log(llm_request))
messages, tools = _get_completion_inputs(llm_request) messages, tools, response_format = _get_completion_inputs(llm_request)
completion_args = { completion_args = {
"model": self.model, "model": self.model,
"messages": messages, "messages": messages,
"tools": tools, "tools": tools,
"response_format": response_format,
} }
completion_args.update(self._additional_args) completion_args.update(self._additional_args)