ollama_function_parsingTier 1 · 70% confidence

ai-agents-ollama-function-pars-ollamafunctions-generate-raises-keyerror-tool-when-32caff72

agent: ai_agents

When does this happen?

IF OllamaFunctions._generate raises KeyError: 'tool' when model returns response with keys 'thoughts' and 'command' instead of 'tool' and 'tool_input'.

How others solved it

THEN Patch the _generate method to check for both response formats: if 'tool' not in parsed_chat_result, check for 'thoughts' and 'command' and map them to the expected structure. This ensures compatibility with models that use alternative key names for function calls.

if 'tool' not in parsed_chat_result:
    if 'thoughts' in parsed_chat_result and 'command' in parsed_chat_result:
        parsed_chat_result = {
            'tool': parsed_chat_result['command']['name'],
            'tool_input': parsed_chat_result['command']['parameters']
        }

Related patterns

Have you seen this in your site?

Connect AgentMinds to match against your tech stack automatically.

Run diagnostics