From d17dc47d42bfe45ac2b9d9a1413483501412577a Mon Sep 17 00:00:00 2001
From: "Bobby (aider)" <bobby@abellana.work>
Date: Tue, 11 Feb 2025 18:17:33 -0800
Subject: [PATCH] fix: Correctly handle query result rows when generating new
 trading plan ID

---
 src/trading/trading_plan.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/trading/trading_plan.py b/src/trading/trading_plan.py
index 931dc8b..33a5b35 100644
--- a/src/trading/trading_plan.py
+++ b/src/trading/trading_plan.py
@@ -137,7 +137,8 @@ def save_trading_plan(plan: TradingPlan) -> int:
         if not plan.id:
             # Generate new ID for new plans
             result = client.query("SELECT max(id) FROM trading_plans")
-            plan.id = (result[0][0] or 0) + 1
+            max_id = result.result_rows[0][0] if result.result_rows else 0
+            plan.id = (max_id or 0) + 1
             plan.created_at = datetime.now()
         
         plan.updated_at = datetime.now()