AI Agents - calling custom code and databricks jobs

[GiriSreerangam]

Hi Everyone

I am building AI Agents; my requirement is to call custom tool logic (which was not possible using a unity catalog function) and databricks jobs. I could not find much of documentation on these scenarios. If someone could share any references to documentation would be of great help.

Thank you

#AI Agents #Custom Tools

[SP_6721]

Hi @GiriSreerangam ,

You can try using agent code tools to define Python functions directly within your agent for any custom logic, including Databricks SDK or REST API calls.
https://docs.databricks.com/aws/en/generative-ai/agent-framework/agent-tool

Also, consider exploring MCP (Model Context Protocol) tools for better discoverability and enterprise integration. They allow you to wrap external or custom logic as API endpoints that your agent can easily invoke.
https://docs.databricks.com/aws/en/generative-ai/mcp

[GiriSreerangam]

thank you, @SP_6721. I have gone through this documentation, it was not detailed enough. I was trying with different approaches, could not find a solution yet.