Skip to content

Migrate agent-bricks tools to official Databricks SDK when public API is available #168

@jacksandom

Description

@jacksandom

Summary

The manage_ka, manage_mas MCP tools currently use private REST endpoints (/api/2.0/knowledge-assistants, /api/2.0/multi-agent-supervisors, /api/2.0/data-rooms) via raw requests calls in agent_bricks/manager.py. These can break without notice.

Changes needed

  • KA / MAS: Replace all raw requests calls with WorkspaceClient SDK methods
    when w.knowledge_assistants / w.multi_agent_supervisors are publicly available
  • Genie: w.genie SDK already exists for listing/querying — migrate remaining
    create/update/delete/curation operations from private /api/2.0/data-rooms/ to SDK
  • Remove the TileExampleQueue background polling thread in favour of SDK wait patterns
  • Remove manual _get / _post / _patch / _delete HTTP wrappers
  • Update pyproject.toml SDK minimum version

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or requesttools-libraryRelated to databricks-tools-core

    Type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions