github/home-llm
1
1
Fork 0
mirror of https://github.com/acon96/home-llm.git synced 2026-01-08 05:14:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

extract tool calls from multiple keys

Browse Source
This commit is contained in:
Alex O'Connell
2025-12-20 18:07:27 -05:00
parent ee2d2e7640
commit 672a9de65c
1 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
custom_components/llama_conversation/entity.py
View File

@@ -288,7 +288,12 @@ class LocalLLMClient:
if isinstance(raw_tool_call, str):
tool_call, to_say = parse_raw_tool_call(raw_tool_call, agent_id)
else:
tool_call, to_say = parse_raw_tool_call(raw_tool_call["function"], agent_id)
# try multiple dict key names
function_content = raw_tool_call.get("function") or raw_tool_call.get("function_call") or raw_tool_call.get("tool")
if not function_content:
_LOGGER.warning("Received tool call dict without 'function', 'function_call' or 'tool' key: %s", raw_tool_call)
continue
tool_call, to_say = parse_raw_tool_call(function_content, agent_id)
if tool_call:
_LOGGER.debug("Tool call parsed: %s", tool_call)
@@ -336,8 +341,12 @@ class LocalLLMClient:
if isinstance(raw_tool_call, str):
tool_call, to_say = parse_raw_tool_call(raw_tool_call, agent_id)
else:
tool_call, to_say = parse_raw_tool_call(raw_tool_call["function"], agent_id)
# try multiple dict key names
function_content = raw_tool_call.get("function") or raw_tool_call.get("function_call") or raw_tool_call.get("tool")
if not function_content:
_LOGGER.warning("Received tool call dict without 'function', 'function_call' or 'tool' key: %s", raw_tool_call)
continue
tool_call, to_say = parse_raw_tool_call(function_content, agent_id)
if tool_call:
_LOGGER.debug("Tool call parsed: %s", tool_call)
parsed_tool_calls.append(tool_call)