query tags integration

[sreekanth-db]

What type of PR is this?

• Refactor
• Feature
• Bug Fix
• Other

Description

This PR adds Query Tags support to the Python SQL Connector. Query Tags allow users to attach key-value pairs to SQL executions that appear in the system.query.history table.

Changes Made:

1. Created Example File (examples/query_tags_example.py):

   • New file demonstrating Query Tags usage with session configuration

2. Enhanced Integration Test (tests/e2e/test_driver.py):

   • Modified test_ssp_passthrough() to include Query Tags in session configuration

3. Added Unit Test (tests/unit/test_session.py):

   • Created test_query_tags_configuration_passthrough() test
   • Verifies Query Tags are properly passed through to the backend

Implementation Details:

• No Code Changes Required: The existing python connector interface supports the query tags through session_configuration parameter

import databricks.sql

connection = databricks.sql.connect(
    server_hostname='host',
    http_path='/sql/1.0/warehouses/xxx',
    access_token='token',
    session_configuration={
        'QUERY_TAGS': 'team:engineering,test:query-tags',
        'ansi_mode': False
    }
)

How is this tested?

• Unit tests
• E2E Tests
• Manually
• N/A

Testing Details:

Unit Tests:

• test_query_tags_configuration_passthrough() - Verifies Query Tags flow through connection layers
• All existing tests still pass

E2E Tests:

• Enhanced test_ssp_passthrough() to include Query Tags

Manual Testing:

• Ran examples/query_tags_example.py with real backend connection
• Verified Query Tags are accepted and queries execute successfully

Related Tickets & Documents