Release version 0.6.2

- Refactored transmit module for improved maintainability and testability
- Added CI/CD safe smoke tests (no real database connections needed)
- Obfuscated all PII and API keys in test data
- Added SQLite support to all database operations
- Fixed CLI error handling for proper exit codes
- Fixed date filtering in SQLite implementation
- Fixed CLI mode mapping for transmit functionality
- Created comprehensive test suite with 97 passing tests

Co-Authored-By: Erik Peterson <erik@cloudzero.com>

Author: silvexis

CHANGELOG.md

@@ -7,21 +7,57 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.6.2] - 2025-01-29
+
+### Added
+- **Transmit module refactoring** for improved maintainability and testability
+  - Split monolithic `DataTransmitter` into focused components with single responsibilities
+  - Introduced dependency injection for all components
+  - Created protocol-based design with `OutputHandler` and `Transmitter` interfaces
+  - Added test-friendly implementations: `NullOutput`, `MockTransmitter`, `CollectingOutput`
+  - Comprehensive test suite with 27 new tests demonstrating improved testability
+  - Full backward compatibility maintained
+- **CI/CD safe smoke tests** for all CLI commands
+  - Updated smoke tests to not require real database connections
+  - All tests now safe for automated CI/CD pipelines
+- **PII obfuscation in test data**
+  - Anonymized all names, emails, and API keys in test.sqlite
+  - Updated create_test_sqlite.py script to generate only anonymized test data
+  - Reviewed entire codebase to ensure no PII is present
+
+### Changed
+- Updated CLI to use refactored transmit module (`transmit_refactored.py`)
+- Fixed CLI error handling to properly exit with error code when transmit fails
+- Improved BatchAnalyzer to correctly group data by date in refactored transmit module
+
+### Fixed
+- Fixed date filtering in SQLite implementation for data source strategies
+- Fixed CLI mode mapping to correctly handle 'today', 'yesterday', 'date-range' modes
+
+### Testing
+- All 97 tests passing including new refactored transmit tests
+- SQLite integration fully tested with transmit functionality
+- All smoke tests are now CI/CD safe
+
 ## [0.6.1] - 2025-01-28
 
 ### Fixed
 - Fixed import error in `data_processor.py` - `extract_model_name` now correctly imported from `model_name_strategies`
 - Added `__main__.py` to enable running package as module: `python -m ll2cz`
 
 ### Added
+- **SQLite database support** - All commands now accept `sqlite://` connection strings
 - Comprehensive import tests (`test_imports.py`) to prevent future import errors
 - CLI smoke tests (`test_cli_smoke.py`) to ensure all commands work without crashing
 - Tests for all critical imports and circular dependency detection
+- `scripts/create_test_sqlite.py` to generate test SQLite databases with sample data
+- SQLite-specific tests (`test_sqlite.py`) for database functionality
 
 ### Testing
 - Verified all CLI commands work correctly: `transmit`, `analyze`, `transform`, `cache`, `config`
 - All 21 core tests passing
-- All 16 new import and smoke tests passing
+- All 24 new tests passing (import, smoke, and SQLite tests)
+- SQLite support tested with both test databases and production-like schemas
 
 ## [0.6.0] - 2025-01-28
 

README.md

@@ -4,10 +4,11 @@ Transform LiteLLM database data into CloudZero AnyCost CBF format for cost track
 
 ## Features
 
-- Extract usage data from LiteLLM PostgreSQL database
+- Extract usage data from LiteLLM PostgreSQL or SQLite database
 - Transform data into CloudZero Billing Format (CBF)
 - **Cost comparison analysis** between SpendLogs and user tables (v0.4.0+)
 - **Dual data source support** with `--source` option: transaction-level SpendLogs or daily aggregated user tables
+- **SQLite support** for testing and offline analysis (v0.6.1+)
 - Analysis mode with beautiful terminal output using Rich
 - Multiple output options: CSV files or direct CloudZero API streaming
 - Built with modern Python tools: uv, ruff, pytest, polars, httpx
@@ -84,9 +85,12 @@ ll2cz config edit        # Interactively edit configuration
 Transform LiteLLM data to CloudZero CBF format:
 
 ```bash
-# Display data on screen (formatted table)
+# Display data on screen (formatted table) - PostgreSQL
 ll2cz transform --input "postgresql://user:pass@host:5432/litellm_db" --screen
 
+# Display data on screen - SQLite
+ll2cz transform --input "sqlite://path/to/litellm.sqlite" --screen
+
 # Export to CSV file
 ll2cz transform --input "postgresql://user:pass@host:5432/litellm_db" --output data.csv
 
@@ -219,10 +223,28 @@ uv run ruff check --fix src/ tests/
 uv build
 ```
 
+## Testing with SQLite
+
+For testing and development, you can use SQLite instead of PostgreSQL:
+
+```bash
+# Create a test SQLite database with sample data
+python scripts/create_test_sqlite.py
+
+# Use the test database
+ll2cz transmit all --test --input "sqlite://test.sqlite"
+```
+
+The test database includes:
+- Sample organizations, teams, and users
+- API keys with realistic naming
+- 30 days of usage data
+- Multiple model providers (OpenAI, Anthropic, etc.)
+
 ## Requirements
 
 - Python ≥ 3.12
-- PostgreSQL database with LiteLLM data
+- PostgreSQL or SQLite database with LiteLLM data
 - CloudZero API key and connection ID (for streaming mode)
 
 ## License

docs/CLI_Reference.md

@@ -31,19 +31,25 @@ ll2cz transform [OPTIONS]
 ```
 
 ### Options
-- `--input TEXT` - LiteLLM PostgreSQL database connection URL
+- `--input TEXT` - LiteLLM database connection URL (PostgreSQL or SQLite)
 - `--output TEXT` - Output CSV file name  
 - `--screen` - Display transformed data on screen in formatted table
 - `--limit INTEGER` - Limit number of records for screen output (default: 50)
 
 ### Examples
 ```bash
-# Display data on screen (formatted table)
+# Display data on screen (formatted table) - PostgreSQL
 ll2cz transform --input "postgresql://user:pass@host:5432/litellm_db" --screen
 
-# Export to CSV file
+# Display data on screen - SQLite
+ll2cz transform --input "sqlite://path/to/litellm.sqlite" --screen
+
+# Export to CSV file - PostgreSQL
 ll2cz transform --input "postgresql://user:pass@host:5432/litellm_db" --output data.csv
 
+# Export to CSV file - SQLite
+ll2cz transform --input "sqlite://test.sqlite" --output data.csv
+
 # Limit records for screen display
 ll2cz transform --screen --limit 25
 
@@ -69,7 +75,7 @@ ll2cz transmit [OPTIONS] MODE [DATE_SPEC]
   - For `all` mode: ignored
 
 ### Options
-- `--input TEXT` - LiteLLM PostgreSQL database connection URL
+- `--input TEXT` - LiteLLM database connection URL (PostgreSQL or SQLite)
 - `--cz-api-key TEXT` - CloudZero API key
 - `--cz-connection-id TEXT` - CloudZero connection ID
 - `--append` - Use 'sum' operation instead of 'replace_hourly'
@@ -127,6 +133,20 @@ ll2cz transmit day --test
 
 # Test specific month with API keys still required
 ll2cz transmit month 01-2024 --test --cz-api-key "key" --cz-connection-id "id"
+
+# Test with SQLite database
+ll2cz transmit all --test --input "sqlite://test.sqlite"
+```
+
+### Database Support
+Both PostgreSQL and SQLite databases are supported:
+
+```bash
+# PostgreSQL
+ll2cz transmit all --input "postgresql://user:pass@host:5432/litellm_db"
+
+# SQLite
+ll2cz transmit all --input "sqlite://path/to/database.sqlite"
 ```
 
 ---