NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-12-18 19:32:21 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix ollama issues with ChatCompletionAssistantToolCall

Browse Source 
PiperOrigin-RevId: 761732316
This commit is contained in:
Google Team Member
2025-05-21 17:26:31 -07:00
committed by Copybara-Service
parent 505d936007
commit 4542af5650
2 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -30,6 +30,7 @@ from typing import Union
from google.genai import types
from litellm import acompletion
from litellm import ChatCompletionAssistantMessage
from litellm import ChatCompletionAssistantToolCall
from litellm import ChatCompletionDeveloperMessage
from litellm import ChatCompletionImageUrlObject
from litellm import ChatCompletionMessageToolCall
@@ -180,12 +181,12 @@ def _content_to_message_param(
for part in content.parts:
if part.function_call:
tool_calls.append(
ChatCompletionMessageToolCall(
ChatCompletionAssistantToolCall(
type="function",
id=part.function_call.id,
function=Function(
name=part.function_call.name,
arguments=part.function_call.args,
arguments=json.dumps(part.function_call.args),
),
)
)