Skip to content

Let users choose a usage SQL warehouse - #388

Merged
andy-xu-db merged 7 commits into
mainfrom
usage-warehouse-picker
Aug 31, 2026
Merged

Let users choose a usage SQL warehouse#388
andy-xu-db merged 7 commits into
mainfrom
usage-warehouse-picker

Conversation

@andy-xu-db

@andy-xu-db andy-xu-db commented Aug 25, 2026 •
edited
Loading

Copy link
Copy Markdown
Collaborator

Summary

  • list the SQL warehouses visible to the user before running the detailed usage query
  • let the user choose which warehouse runs the query
  • if the selected warehouse fails, reopen the picker with the remaining warehouses
  • let the user cancel cleanly after a failed selection
  • preserve the existing --warehouse-id override without prompting

Testing

  • uv run pytest tests/test_usage.py (89 passed)
  • uv run ruff check src/ucode/usage.py tests/test_usage.py
  • uv run ruff format --check src/ tests/
Screen.Recording.2026-08-31.at.6.00.42.PM.mov

This was referenced Aug 25, 2026
Merged
Merged
@andy-xu-db
andy-xu-db marked this pull request as ready for review August 25, 2026 15:13
@andy-xu-db
andy-xu-db force-pushed the feature/usage-budget-first branch from c353f45 to b41c1d1 Compare August 25, 2026 20:26
@andy-xu-db
andy-xu-db force-pushed the usage-warehouse-picker branch from 26726a6 to f952464 Compare August 25, 2026 20:26
Base automatically changed from feature/usage-budget-first to main August 25, 2026 20:32
@andy-xu-db
andy-xu-db force-pushed the usage-warehouse-picker branch from f952464 to 9ef700d Compare August 25, 2026 20:57
@andy-xu-db
andy-xu-db force-pushed the usage-warehouse-picker branch from 9ef700d to 3663d3d Compare August 25, 2026 20:57
@lilly-luo

lilly-luo commented Aug 31, 2026

Copy link
Copy Markdown
Collaborator

can you merge main and include a video of it working in the PR desc please?

@andy-xu-db
andy-xu-db requested a review from lilly-luo August 31, 2026 22:23
@andy-xu-db
andy-xu-db merged commit 1c23681 into main Aug 31, 2026
2 checks passed
@andy-xu-db
andy-xu-db deleted the usage-warehouse-picker branch August 31, 2026 22:34
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@lilly-luo lilly-luo lilly-luo approved these changes
@AarushiShah-db AarushiShah-db Awaiting requested review from AarushiShah-db

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@andy-xu-db @lilly-luo