final_asnwer as Json

[lightScout]

I'm trying to get the agent's final answer in a JSON structure rather than a plain string. While defining the structure within the query works in most cases, it tends to break down in a multi-agent system. Does anyone have suggestions or best practices for handling this?

Thanks!

[grrowl]

Can you share some example code? I imagine overriding final_answer with a more complex input type would be how you'd approach this?

[lightScout]

hey @grrowl, thanks for getting back to me on this.
This an example set-up I'm using:

-Agent
agent = CodeAgent( tools=[DuckDuckGoSearchTool()], model=HfApiModel())
response = agent.run(query)
return jsonify({'response': response})

-Input
$task. The final_answer should be a json under this format and conditions. { "title": "string", "answer": "string", "sources": ["list of source URLs or leave empty if the source is you"] }. Ensure all results are combined into a single 'answer' field, even if multiple items are requested. Ensure the output is valid JSON, properly escaped, and formatted without extra characters or invalid tokens. Use double quotes (") for strings.

You mentioned overriding the 'final_answer' tool, can you expand on that please?

[grrowl]

FinalAnswerTool is defined here and MultiStepAgent always has it baked in. I guess the move would be to extend your own class ComplexMultiStepAgent(MultiStepAgent): and override __init__ to do basically the same thing but add your own FinalAnswerTool to it. You'll note FinalAnswerTool just forwards the input to output -- you might want to json.dumps(answer) in case more of the smolagents plumbing expects str