JWT-update_3

app.py

@@ -35,6 +35,7 @@ SUPABASE_HEADERS = {
 
 class ChatRequest(BaseModel):
     message: str
+    access_token: str = "" 
 
 class ChatResponse(BaseModel):
     answer: str
@@ -75,7 +76,8 @@ def chat(
     if not user:
         raise HTTPException(status_code=401, detail="Invalid or expired token")
 
-    access_token = get_user_access_token(user["sub"])
+    # Use token from request if provided, fallback to sandbox
+    access_token = request.access_token if request.access_token else SANDBOX_ACCESS_TOKEN
     financial_context = get_financial_snapshot(access_token)
 
     answer = get_answer(
@@ -106,7 +108,7 @@ def chat_stream(
 
     financial_context = ""
     try:
-        access_token = get_user_access_token(user["sub"])
+        access_token = request.access_token if request.access_token else SANDBOX_ACCESS_TOKEN
         financial_context = get_financial_snapshot(access_token)
         financial_context = financial_context.replace("{", "(").replace("}", ")")
     except Exception as e:
@@ -154,7 +156,7 @@ def plaid_exchange_token(
     except Exception as e:
         print(f"Exchange token error: {e}")
         raise HTTPException(status_code=500, detail=str(e))
-        
+
 
 @app.delete("/account")
 def delete_account(