closes #6072

Greptile Overview

Greptile Summary

This PR adds support for custom SQLAlchemy connection arguments through a new connect_args configuration field. The implementation enables advanced database configurations like Snowflake private key authentication.

Changes Made

Configuration (reflex/config.py)

• Added connect_args: dict[str, Any] field to BaseConfig with proper default factory initialization

Database Engine (reflex/model.py)

• Updated get_engine_args() to merge user-provided connect_args with Reflex's internal requirements
• Preserved existing SQLite-specific behavior: check_same_thread: False is always set for SQLite databases (required for admin dashboard)
• Both sync (create_engine) and async (create_async_engine) database engines benefit from this change

Testing (tests/units/test_connect_args.py)

• Comprehensive test coverage for PostgreSQL and SQLite scenarios
• Verifies custom args are passed through correctly
• Confirms SQLite-specific settings are preserved alongside custom args

Implementation Quality

The implementation is solid with proper handling of edge cases:

• Uses .copy() to avoid mutating the original config dictionary
• Correctly handles empty dictionaries and preserves SQLite requirements
• Short-circuit evaluation prevents errors with falsy values
• Both sync and async engines use the same get_engine_args() function, ensuring consistency

Minor Improvement

The Config class docstring could be updated to mention connect_args in the "Key Configuration Areas" section for better discoverability.

Confidence Score: 4/5

• This PR is safe to merge with only minor documentation improvements suggested
• The implementation is well-designed with proper edge case handling, comprehensive tests, and no logic errors or security issues. The score is 4/5 (not 5/5) only due to a minor documentation gap in the Config class docstring - the new field should be mentioned in the "Key Configuration Areas" section for better discoverability
• No files require special attention - all changes are straightforward and well-tested

Important Files Changed

File Analysis

Sequence Diagram

sequenceDiagram
    participant User
    participant Config
    participant Model
    participant SQLAlchemy

    User->>Config: Set connect_args in rxconfig.py
    Note over Config: connect_args = {"application_name": "myapp"}
    
    User->>Model: Request database connection
    Model->>Config: get_config()
    Config-->>Model: Returns config with connect_args
    
    Model->>Model: get_engine_args(url)
    Note over Model: Copy connect_args from config
    
    alt URL is SQLite
        Model->>Model: Add check_same_thread: False
        Note over Model: Required for admin dashboard
    end
    
    Model->>Model: Merge connect_args into kwargs
    Model->>SQLAlchemy: create_engine(url, **kwargs)
    Note over SQLAlchemy: Receives custom connect_args
    SQLAlchemy-->>Model: Returns engine with custom settings
    Model-->>User: Database connection ready

CodSpeed Performance Report

Merging this PR will not alter performance

_{Comparing theradP:feature/set_connect_args (5075b84) with main (ce9b21a)}

Summary

✅ 8 untouched benchmarks