jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-28 07:37:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

Modify ollama tool calls

Browse Source
This commit is contained in:
Allen Porter 2024-07-21 21:30:23 +00:00
parent eaeca423d4
commit 4879e02839
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
homeassistant/components/ollama/conversation.py
View File

@ -256,16 +256,18 @@ class OllamaConversationEntity(
message_history.messages.append(message_convert(response_message)) message_history.messages.append(message_convert(response_message))
if not tool_calls or not llm_api: if not tool_calls or not llm_api:
_LOGGER.debug("tool_calls=%s", tool_calls)
_LOGGER.debug("llm_api=%s", llm_api)
break break
_LOGGER.debug("Response: %s", response_message.get("content"))
_LOGGER.debug("Tools calls: %s", tool_calls)
for tool_call in tool_calls: for tool_call in tool_calls:
tool_args = tool_call["function"]["arguments"] or {}
tool_args = {
k: v
for k, v in tool_args.items()
if v # Don't pass empty args
}
tool_input = llm.ToolInput( tool_input = llm.ToolInput(
tool_name=tool_call["function"]["name"], tool_name=tool_call["function"]["name"],
tool_args=tool_call["function"]["arguments"], tool_args=tool_args,
) )
_LOGGER.debug( _LOGGER.debug(
"Tool call: %s(%s)", tool_input.tool_name, tool_input.tool_args "Tool call: %s(%s)", tool_input.tool_name, tool_input.tool_args