Skip to content

[AIGTWY-4565] Enable Claude MPS model discovery - #551

Open
andy-xu-db wants to merge 2 commits into
mainfrom
aigtwy-4565-claude-mps-client-v2
Open

[AIGTWY-4565] Enable Claude MPS model discovery#551
andy-xu-db wants to merge 2 commits into
mainfrom
aigtwy-4565-claude-mps-client-v2

Conversation

@andy-xu-db

@andy-xu-db andy-xu-db commented Sep 10, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • Enable Claude model discovery for direct and relayed MPS launches.
  • Scope /v1/models with the existing Databricks-Model-Provider-Service custom header.
  • Keep discovery launch-scoped and remove stale system.ai.* model defaults from MPS launches.
  • Preserve explicit provider model pins, including Bedrock.

Server counterpart: https://github.com/databricks-eng/universe/pull/2523045

Usage

Pass both flags before --:

uv run ug claude \
  --provider <catalog>.<schema>.<model-provider-service> \
  --enable-model-discovery

Testing

  • Focused: 407 passed
  • Full suite: 2191 passed, 37 skipped; 2 unrelated installed-binary E2E failures
  • Ruff and diff checks passed
  • LiteSwap /model: exactly claude-fable-5, claude-haiku-4-5, and claude-opus-5
  • Selected Haiku 4.5 and received E2E_OK; LiteSwap logged the MPS route and HTTP 200
Claude Code /model showing MPS models

@andy-xu-db
andy-xu-db marked this pull request as ready for review September 10, 2026 20:18
Comment thread src/ucode/agents/claude.py

@lilly-luo lilly-luo left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

pls wait to merge until backend is fully released. or have a graceful way to handle if someone tries to use this and the backend doesn't have your functionality yet

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants