This repository is maintained by Databricks and intended for contributions from Databricks Field Engineers. While the repository is public and meant to help anyone developing projects that use Databricks, external contributions are not currently accepted. Feel free to open an issue with requests or suggestions.
-
Clone the repository:
git clone https://github.com/databricks-solutions/ai-dev-kit.git cd ai-dev-kit -
Set up the MCP server (includes databricks-tools-core):
./databricks-mcp-server/setup.sh
-
Configure authentication:
export DATABRICKS_HOST="https://your-workspace.cloud.databricks.com" export DATABRICKS_TOKEN="your-token"
or
export DATABRICKS_CONFIG_PROFILE="your-profile"
- Python: Follow PEP 8 conventions
- Documentation: Update relevant SKILL.md files when adding or modifying functionality
- Type hints: Include type annotations for public functions
- Naming: Use lowercase with hyphens for directories (e.g.,
databricks-tools-core)
Run integration tests before submitting changes:
cd databricks-tools-core
uv run pytest tests/integration/ -vEnsure your changes work with a live Databricks workspace.
- Create a feature branch from
main(fork repo is necessary) - Make your changes with clear, descriptive commits
- Test your changes against a Databricks workspace
- Open a PR with:
- Brief description of the change
- Any relevant context or motivation
- Testing performed
- Address review feedback
When adding a new skill to databricks-skills/:
- Create a directory with a descriptive name
- Include a
SKILL.mdfile that defines:- Trigger conditions (when the skill activates)
- Core patterns and best practices
- Code examples
- Add supporting documentation files as needed
- Update the skills table in the main README.md
- Never commit credentials, tokens, or sensitive data
- Use synthetic data for examples and tests
- Review changes for potential security issues before submitting
By submitting a contribution, you agree that your contributions will be licensed under the same terms as the project (see LICENSE.md).
You certify that:
- You have the right to submit the contribution
- Your contribution does not include confidential or proprietary information
- You grant Databricks the right to use, modify, and distribute your contribution