Generate only the SQL in vanna 2.0 like vn.generate_sql in vanna 0.7

[aexertier]

Hello,

Do you know what is the easiest way to switch off data query and visualisation of the agents ? I was using vanna 0.7 only as a sql generator with vn.generate_sql, I was quite happy with it, I want to migrate to vanna 2.0 because I'll be interested in the visualisation feature but as it's already in production I would like to do baby steps, thanks :)

[zainhoda]

Hi @aexertier -- the key to the agent behavior is the tools that it has access to:

This is from the default example:

tools = ToolRegistry()
tools.register_local_tool(db_tool, access_groups=['admin', 'user'])
tools.register_local_tool(SaveQuestionToolArgsTool(), access_groups=['admin'])
tools.register_local_tool(SearchSavedCorrectToolUsesTool(), access_groups=['admin', 'user'])
tools.register_local_tool(SaveTextMemoryTool(), access_groups=['admin', 'user'])
tools.register_local_tool(VisualizeDataTool(), access_groups=['admin', 'user'])

Turning off visualizations is as easy as omitting the VisualizeDataTool when regstering the tools.

However, turning off query execution isn't really possible. That would make the agent lose all of its benefits. One of the large benefits of Vanna 2 is that you can give the agent extremely high level directives like "find out why sales were down last week" and it can do as many queries as it needs to in order to answer that question. This only works if the agent is actually able to query the database. I would highly recommend trying to flex the agent in this way -- it will completely floor you with what it's able to accomplish.

If you're just using vn.generate_sql and you're stuck with that for backwards-compatibility issues, then I think the best think to do would be to pin to 0.7.x in your requirements.txt

[aswincandra]

@zainhoda will the version 0.x be deprecated sooner or later?