administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-01 19:17:41 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix gemini tools call

Browse Source
This commit is contained in:
Nicolas Mowen 2026-04-29 16:14:26 -06:00
parent 11bb9fed4c
commit 3c19eba7c4
1 changed files with 14 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
frigate/genai/gemini.py
View File

@ -143,15 +143,17 @@ class GeminiClient(GenAIClient):
) )
elif role == "tool": elif role == "tool":
# Handle tool response # Handle tool response
function_response = { response_payload = (
"name": msg.get("name", ""), content if isinstance(content, dict) else {"result": content}
"response": content, )
}
gemini_messages.append( gemini_messages.append(
types.Content( types.Content(
role="function", role="function",
parts=[ parts=[
types.Part.from_function_response(function_response) # type: ignore[misc,call-arg,arg-type] types.Part.from_function_response(
name=msg.get("name", ""),
response=response_payload,
)
], ],
) )
) )
@ -350,15 +352,17 @@ class GeminiClient(GenAIClient):
) )
elif role == "tool": elif role == "tool":
# Handle tool response # Handle tool response
function_response = { response_payload = (
"name": msg.get("name", ""), content if isinstance(content, dict) else {"result": content}
"response": content, )
}
gemini_messages.append( gemini_messages.append(
types.Content( types.Content(
role="function", role="function",
parts=[ parts=[
types.Part.from_function_response(function_response) # type: ignore[misc,call-arg,arg-type] types.Part.from_function_response(
name=msg.get("name", ""),
response=response_payload,
)
], ],
) )
) )