feat(agent): discover Databricks Unity Catalog models - #6918
Merged
Merged
Conversation
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
Add Unity Catalog model-service discovery to
buzz-agentDatabricks v2, preserve full Unity Catalog FQNs through ACP model discovery, route model services through MLflow chat, and add the optionalDATABRICKS_MODEL_FILTERvisibility filter.The resilience follow-up bounds each catalog attempt across headers, bounded body reads, and JSON parsing; retries transient 499/5xx, transport/body failures, malformed JSON, and attempt timeouts; and emits only safe catalog/status diagnostics.
Validation
The live validation and test gates below were run on the pre-rewrite tip
1cf6c66e7c5bf9a00c357bda24df48e3beeb310b. The rewritten tip0364cf44f3eb5cfbd5771b0a0dd7868ce0ad2c19has identical trees for both commits, so the attribution rewrite changes no tested code.cargo check -p buzz-agentpassedcargo clippy -p buzz-agent --all-targets -- -D warningspassedcargo fmt --checkpassedgit diff --checkpassedThe full
buzz-agentpackage has one separately reproduced pre-existing failure:regressions::cancel_kills_inflight_tool_via_mcp_notification, failing on both baselineeb8e97cfeand the follow-up worktree. No Databricks path reaches that test. The live validation proves discovery and ACP exposure, not inference against every listed model.