feat: Trino oauth support

[equiet]

Summary by CodeRabbit

• New Features
  • Added Trino OAuth authentication support for SQL connections. When connecting to Trino with OAuth credentials, the system now automatically handles credential injection for seamless authentication.

[coderabbitai]

📝 Walkthrough

Walkthrough

The change adds Trino OAuth authentication support to SQL connection initialization. When a Trino database connection is detected (URL scheme trino://) and the TRINO_OAUTH_TOKEN environment variable is available, the code now constructs a JWT authentication object and injects it into connection parameters. A new private helper function _build_params_for_trino_oauth() handles credential assembly, integrating into the existing parameter-building flow alongside BigQuery OAuth. No public API changes or modifications to error handling occur.

Pre-merge checks

❌ Failed checks (1 warning)

Check name	Status	Explanation	Resolution
Docstring Coverage	⚠️ Warning	Docstring coverage is 50.00% which is insufficient. The required threshold is 80.00%.	You can run @coderabbitai generate docstrings to improve docstring coverage.

✅ Passed checks (2 passed)

Check name	Status	Explanation
Description Check	✅ Passed	Check skipped - CodeRabbit’s high-level summary is enabled.
Title Check	✅ Passed	The pull request title "feat: Trino oauth support" directly and accurately describes the main change in the changeset. The PR adds Trino OAuth handling to the SQL execution module, including a new helper function for building OAuth credentials, which aligns precisely with what the title conveys. The title is concise, clear, and specific enough for a teammate reviewing the repository history to immediately understand the primary objective without ambiguity.

---

_{Comment @coderabbitai help to get the list of available commands and usage tips.}

[github-actions]

📦 Python package built successfully!

• Version: 1.0.0.dev6+1802f08
• Wheel: deepnote_toolkit-1.0.0.dev6+1802f08-py3-none-any.whl
• Install:

  pip install "deepnote-toolkit @ https://deepnote-staging-runtime-artifactory.s3.amazonaws.com/deepnote-toolkit-packages/1.0.0.dev6%2B1802f08/deepnote_toolkit-1.0.0.dev6%2B1802f08-py3-none-any.whl"

[codecov]

Codecov Report

❌ Patch coverage is 20.00000% with 12 lines in your changes missing coverage. Please review.
✅ Project coverage is 76.45%. Comparing base (1b4d41d) to head (19f0af6).
✅ All tests successful. No failed tests found.

Files with missing lines	Patch %	Lines
deepnote_toolkit/sql/sql_execution.py	20.00%	12 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main       #8      +/-   ##
==========================================
- Coverage   76.60%   76.45%   -0.16%     
==========================================
  Files          99       99              
  Lines        5476     5491      +15     
  Branches      748      750       +2     
==========================================
+ Hits         4195     4198       +3     
- Misses       1281     1293      +12     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[coderabbitai]

Actionable comments posted: 1

📜 Review details

Configuration used: CodeRabbit UI

Review profile: ASSERTIVE

Plan: Pro

Disabled knowledge base sources:

• Linear integration is disabled by default for public repositories

You can enable these sources in your CodeRabbit configuration.

📥 Commits

Reviewing files that changed from the base of the PR and between 1b4d41d and 19f0af6.

📒 Files selected for processing (1)

• deepnote_toolkit/sql/sql_execution.py (2 hunks)

🧰 Additional context used

🪛 Ruff (0.14.2)

deepnote_toolkit/sql/sql_execution.py

466-466: Missing return type annotation for private function _build_params_for_trino_oauth

(ANN202)

---

471-471: Create your own exception

(TRY002)

---

471-471: Avoid specifying long messages outside the exception class

(TRY003)

🔇 Additional comments (1)

deepnote_toolkit/sql/sql_execution.py (1)

110-114: Trino dependency confirmed—code approved.

Trino is declared in main dependencies at line 95 of pyproject.toml as "trino[sqlalchemy]>=0.327.0,<=0.330.0". The OAuth detection logic at lines 110-114 correctly mirrors the BigQuery pattern, and the helper function includes proper defensive error handling. The redundant token check (at lines 110 and 470) is benign.

[deepnote-bot]

🚀 Review App Deployment Started

📝 Description	🌐 Link / Info
🌍 Review application	ra-8
🔑 Sign-in URL	Click to sign-in
📊 Application logs	View logs
🔄 Actions	Click to redeploy
🚀 ArgoCD deployment	View deployment
⏰ Last deployed	2025-10-31 05:04:34 (UTC)
📜 Deployed commit	62552543c69d520114d2d0a0477e750a5ea29797
🛠️ Toolkit version	1802f08