SuperAPI AI 工具集

databricks-testing

数据与分析

查看项目

更新于 10-16 05:05

技能介绍

用途:在 Databricks 集群上直接执行和测试代码,基于 MCP(Model Context Protocol)命令执行 API,覆盖快速校验与交互式开发两类场景。典型用途包括:打包 DAB 前测试 notebook 代码、在真实集群上调试数据转换、针对真实数据校验 SQL 查询、迭代开发并即时获得反馈、验证集群是否具备所需库、快速概念验证。

主要能力:区分无状态与有状态两种执行方式。无状态使用 databricks_command MCP 工具,单次操作、自动创建并销毁上下文,适合一次性 SQL 查询和简单校验;有状态基于上下文执行,同一会话内多次操作,变量与导入在调用间保持,适合多步工作流和迭代调试,需显式管理上下文。

使用方式:无状态模式向 databricks_command 传入 cluster_id、language(python 或 sql)和 code,MCP 自动创建上下文、执行并返回输出,随后清理。有状态模式先调用 create_context(传 cluster_id、language,返回 context_id),再用 execute_command_with_context 逐块执行代码(传 cluster_id、context_id、code),变量在调用间保留,完成后调用 destroy_context 清理。调试时展示完整错误信息、提出修复建议并重跑,直至成功。

适用场景:SQL 聚合校验、Python 数据清洗测试、迭代式特征工程、库与 Spark 版本检查等。

限制与注意事项:命令超时约 120 秒,需拆分任务或加 LIMIT;上下文空闲超时约 1 小时,需新建;应始终返回完整输出而非摘要,使用真实集群而非本地模拟;测试时避免打印访问令牌,凭据用环境变量,勿硬编码密钥。资料未提供收费、价格与平台支持等细节。