1 files changed, 10 insertions, 8 deletions
diff --git a/cli/src/trading_cli/commands/portfolio.py b/cli/src/trading_cli/commands/portfolio.py
index ad9a6b4..fd3ebd6 100644
--- a/cli/src/trading_cli/commands/portfolio.py
+++ b/cli/src/trading_cli/commands/portfolio.py
@@ -1,6 +1,6 @@
 import asyncio
 import sys
-from datetime import datetime, timedelta, timezone
+from datetime import UTC, datetime, timedelta
 
 import click
 from rich.console import Console
@@ -17,17 +17,18 @@ def portfolio():
 def show():
     """Show the current portfolio holdings and balances."""
     try:
-        from shared.db import Database
+        from sqlalchemy import select
+
         from shared.config import Settings
+        from shared.db import Database
         from shared.sa_models import PositionRow
-        from sqlalchemy import select
     except ImportError as e:
         click.echo(f"Error: Could not import required modules: {e}", err=True)
         sys.exit(1)
 
     async def _show():
         settings = Settings()
-        db = Database(settings.database_url)
+        db = Database(settings.database_url.get_secret_value())
         await db.connect()
         try:
             async with db.get_session() as session:
@@ -71,20 +72,21 @@ def show():
 def history(days):
     """Show PnL history for the portfolio."""
     try:
-        from shared.db import Database
+        from sqlalchemy import select
+
         from shared.config import Settings
+        from shared.db import Database
         from shared.sa_models import PortfolioSnapshotRow
-        from sqlalchemy import select
     except ImportError as e:
         click.echo(f"Error: Could not import required modules: {e}", err=True)
         sys.exit(1)
 
     async def _history():
         settings = Settings()
-        db = Database(settings.database_url)
+        db = Database(settings.database_url.get_secret_value())
         await db.connect()
         try:
-            since = datetime.now(timezone.utc) - timedelta(days=days)
+            since = datetime.now(UTC) - timedelta(days=days)
             stmt = (
                 select(PortfolioSnapshotRow)
                 .where(PortfolioSnapshotRow.snapshot_at >= since)