fix: add print statements with flush=True for Cloud Run logs (#33)

Author: nikhilwoodruff

src/policyengine_api/agent_sandbox.py

@@ -46,6 +46,7 @@ def run_claude_code_in_sandbox(
 
     from policyengine_api.config import settings
 
+    print("[SANDBOX] run_claude_code_in_sandbox starting", flush=True)
     logfire.info(
         "run_claude_code_in_sandbox: starting",
         question=question[:100],
@@ -63,10 +64,13 @@ def run_claude_code_in_sandbox(
 }}"""
 
     # Get reference to deployed app (required when calling from outside Modal)
+    print("[SANDBOX] looking up Modal app", flush=True)
     logfire.info("run_claude_code_in_sandbox: looking up Modal app")
     sandbox_app = modal.App.lookup("policyengine-sandbox", create_if_missing=True)
+    print("[SANDBOX] Modal app found", flush=True)
     logfire.info("run_claude_code_in_sandbox: Modal app found")
 
+    print("[SANDBOX] creating sandbox", flush=True)
     logfire.info("run_claude_code_in_sandbox: creating sandbox")
     sb = modal.Sandbox.create(
         app=sandbox_app,
@@ -75,6 +79,7 @@ def run_claude_code_in_sandbox(
         timeout=600,
         workdir="/tmp",
     )
+    print("[SANDBOX] sandbox created", flush=True)
     logfire.info("run_claude_code_in_sandbox: sandbox created")
 
     # Log from inside the sandbox via Python

src/policyengine_api/api/agent.py

@@ -89,14 +89,16 @@ async def _stream_modal_sandbox(question: str, api_base_url: str):
     """Stream output from Claude Code running in Modal Sandbox."""
     from concurrent.futures import ThreadPoolExecutor
 
-    # Immediate log and yield
+    # Immediate log
+    print("[AGENT] _stream_modal_sandbox started", flush=True)
     logfire.info("_stream_modal_sandbox: started", question=question[:100])
 
     sb = None
     executor = ThreadPoolExecutor(max_workers=1)
     try:
         from policyengine_api.agent_sandbox import run_claude_code_in_sandbox
 
+        print("[AGENT] creating sandbox", flush=True)
         logfire.info(
             "_stream_modal_sandbox: creating sandbox", api_base_url=api_base_url
         )
@@ -106,6 +108,7 @@ async def _stream_modal_sandbox(question: str, api_base_url: str):
         sb, process = await loop.run_in_executor(
             executor, run_claude_code_in_sandbox, question, api_base_url
         )
+        print("[AGENT] sandbox created", flush=True)
         logfire.info("_stream_modal_sandbox: sandbox created")
 
         # Poll for lines with timeout to allow other async tasks
@@ -215,6 +218,7 @@ async def stream_analysis(request: AskRequest):
     data: {"type": "done", "returncode": 0}
     ```
     """
+    print(f"[AGENT] /stream called, use_modal={settings.agent_use_modal}", flush=True)
     api_base_url = settings.policyengine_api_url
     logfire.info(
         "stream_analysis: called",