Skip to content

Feat: Add request-time Databricks U2M OAuth for Pi and OpenCode - #257

Open
anthonyivn2 wants to merge 2 commits into
databricks:mainfrom
anthonyivn2:feat/add-dbx-u2m-oauth-all-harness
Open

Feat: Add request-time Databricks U2M OAuth for Pi and OpenCode#257
anthonyivn2 wants to merge 2 commits into
databricks:mainfrom
anthonyivn2:feat/add-dbx-u2m-oauth-all-harness

Conversation

@anthonyivn2

@anthonyivn2 anthonyivn2 commented Aug 2, 2026

Copy link
Copy Markdown
Contributor

Summary

  • configure Pi to resolve ucode auth-token through its native request-time command API
  • install an OpenCode auth plugin that injects a fresh Databricks bearer per request
  • remove static token persistence and background refresh threads
  • use Pi's native PI_CODING_AGENT_DIR override so Databricks OAuth continues using the user's normal home

Testing

  • uv run --frozen ruff format --check src/ tests/
  • uv run --frozen ruff check .
  • focused tests: 450 passed
  • Pi/OpenCode capture tests: 2 passed
  • broad suite: 1060 passed, 37 skipped, 1 unrelated Claude cache test deselected
  • live Pi and OpenCode requests passed with Databricks U2M OAuth profile

@anthonyivn2 anthonyivn2 changed the title Add request-time OAuth for Pi and OpenCode Feat: Add request-time OAuth for Pi and OpenCode Aug 2, 2026
@anthonyivn2 anthonyivn2 changed the title Feat: Add request-time OAuth for Pi and OpenCode Feat: Add request-time Databricks U2M OAuth for Pi and OpenCode Aug 2, 2026
@anthonyivn2
anthonyivn2 marked this pull request as ready for review August 2, 2026 12:12
@anthonyivn2

Copy link
Copy Markdown
Contributor Author

@rohita5l can I get you or your team to review this?

Signed-off-by: Anthony Ivan <anthony.ivan@databricks.com>

# Conflicts:
#	src/ucode/cli.py
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.

1 participant