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

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions