refactor: rename demo to agent

nikhilwoodruff · nikhilwoodruff · commit 5a93004c8d7f · 2025-12-28T16:18:49.000Z
diff --git a/.github/workflows/deploy.yml b/.github/workflows/deploy.yml
@@ -119,4 +119,4 @@ jobs:
           MODAL_TOKEN_SECRET: ${{ secrets.MODAL_TOKEN_SECRET }}
         run: |
           uv run modal deploy src/policyengine_api/modal_app.py
-          uv run modal deploy src/policyengine_api/demo_sandbox.py
+          uv run modal deploy src/policyengine_api/agent_sandbox.py
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -14,7 +14,7 @@ services:
       API_PORT: ${API_PORT:-8000}
       ANTHROPIC_API_KEY: ${ANTHROPIC_API_KEY}
       POLICYENGINE_API_URL: http://localhost:${API_PORT:-8000}
-      DEMO_USE_MODAL: "false"
+      AGENT_USE_MODAL: "false"
     volumes:
       - ./src:/app/src
       - ./docs/out:/app/docs/out
diff --git a/docs/src/app/agent/page.tsx b/docs/src/app/agent/page.tsx
diff --git a/docs/src/components/policy-chat.tsx b/docs/src/components/policy-chat.tsx
@@ -93,7 +93,7 @@ export function PolicyChat() {
     ]);
 
     try {
-      const res = await fetch(`${baseUrl}/demo/stream`, {
+      const res = await fetch(`${baseUrl}/agent/stream`, {
         method: "POST",
         headers: { "Content-Type": "application/json" },
         body: JSON.stringify({ question: userMessage }),
diff --git a/docs/src/components/sidebar.tsx b/docs/src/components/sidebar.tsx
@@ -13,7 +13,7 @@ const navigation = [
       { name: "Quick start", href: "/quickstart" },
       { name: "Environment setup", href: "/setup" },
       { name: "MCP integration", href: "/mcp" },
-      { name: "AI demo", href: "/demo" },
+      { name: "AI agent", href: "/agent" },
     ],
   },
   {
diff --git a/src/policyengine_api/agent_sandbox.py b/src/policyengine_api/agent_sandbox.py
diff --git a/src/policyengine_api/api/__init__.py b/src/policyengine_api/api/__init__.py
@@ -3,10 +3,10 @@
 from fastapi import APIRouter
 
 from . import (
+    agent,
     analysis,
     change_aggregates,
     datasets,
-    demo,
     dynamics,
     household,
     outputs,
@@ -34,6 +34,6 @@
 api_router.include_router(change_aggregates.router)
 api_router.include_router(household.router)
 api_router.include_router(analysis.router)
-api_router.include_router(demo.router)
+api_router.include_router(agent.router)
 
 __all__ = ["api_router"]
diff --git a/src/policyengine_api/api/agent.py b/src/policyengine_api/api/agent.py
@@ -1,4 +1,4 @@
-"""Demo endpoint for AI-powered policy analysis.
+"""Agent endpoint for AI-powered policy analysis.
 
 This endpoint lets users ask natural language questions about tax/benefit policy
 and get AI-generated reports using Claude Code connected to the PolicyEngine MCP server.
@@ -16,7 +16,7 @@
 
 from policyengine_api.config import settings
 
-router = APIRouter(prefix="/demo", tags=["demo"])
+router = APIRouter(prefix="/agent", tags=["agent"])
 
 
 class AskRequest(BaseModel):
@@ -33,7 +33,7 @@ class AskResponse(BaseModel):
 
 
 class JobStatusResponse(BaseModel):
-    """Status of a demo job."""
+    """Status of an agent job."""
 
     job_id: str
     status: str
@@ -87,7 +87,7 @@ async def _stream_claude_code(question: str, api_base_url: str):
 async def _stream_modal_sandbox(question: str, api_base_url: str):
     """Stream output from Claude Code running in Modal Sandbox."""
 
-    from policyengine_api.demo_sandbox import run_claude_code_in_sandbox
+    from policyengine_api.agent_sandbox import run_claude_code_in_sandbox
 
     sb, process = run_claude_code_in_sandbox(question, api_base_url)
 
@@ -129,7 +129,7 @@ async def stream_analysis(request: AskRequest):
     """
     api_base_url = settings.policyengine_api_url
 
-    if settings.demo_use_modal:
+    if settings.agent_use_modal:
         return StreamingResponse(
             _stream_modal_sandbox(request.question, api_base_url),
             media_type="text/event-stream",
@@ -145,8 +145,8 @@ async def stream_analysis(request: AskRequest):
 async def ask_question(request: AskRequest) -> AskResponse:
     """Ask a policy question (non-streaming).
 
-    Starts the analysis in the background. Poll GET /demo/status/{job_id} for results.
-    For real-time streaming, use POST /demo/stream instead.
+    Starts the analysis in the background. Poll GET /agent/status/{job_id} for results.
+    For real-time streaming, use POST /agent/stream instead.
     """
     job_id = str(uuid4())
     api_base_url = settings.policyengine_api_url
@@ -162,7 +162,7 @@ async def ask_question(request: AskRequest) -> AskResponse:
     async def run_job():
         _jobs[job_id]["status"] = "running"
         try:
-            if settings.demo_use_modal:
+            if settings.agent_use_modal:
                 import modal
 
                 run_policy_analysis = modal.Function.lookup(
@@ -203,7 +203,7 @@ async def run_job():
 
 @router.get("/status/{job_id}", response_model=JobStatusResponse)
 async def get_job_status(job_id: str) -> JobStatusResponse:
-    """Get the status of a demo job."""
+    """Get the status of an agent job."""
     if job_id not in _jobs:
         raise HTTPException(status_code=404, detail="Job not found")
 
diff --git a/src/policyengine_api/api/analysis.py b/src/policyengine_api/api/analysis.py
@@ -522,7 +522,7 @@ def _trigger_economy_comparison(
     """Trigger economy comparison analysis (local or Modal)."""
     from policyengine_api.config import settings
 
-    if not settings.demo_use_modal and session is not None:
+    if not settings.agent_use_modal and session is not None:
         # Run locally
         if tax_benefit_model_name == "policyengine_uk":
             _run_local_economy_comparison_uk(job_id, session)
diff --git a/src/policyengine_api/api/household.py b/src/policyengine_api/api/household.py
@@ -371,7 +371,7 @@ def _trigger_modal_household(
     """Trigger household simulation - Modal or local based on settings."""
     from policyengine_api.config import settings
 
-    if not settings.demo_use_modal and session is not None:
+    if not settings.agent_use_modal and session is not None:
         # Run locally
         if request.tax_benefit_model_name == "policyengine_uk":
             _run_local_household_uk(
diff --git a/src/policyengine_api/config/settings.py b/src/policyengine_api/config/settings.py
@@ -32,9 +32,9 @@ class Settings(BaseSettings):
     # Seeding
     limit_seed_parameters: bool = False
 
-    # Demo agent
+    # Agent
     anthropic_api_key: str = ""
-    demo_use_modal: bool = False
+    agent_use_modal: bool = False
     policyengine_api_url: str = "https://v2.api.policyengine.org"
 
     @property
diff --git a/tests/test_agent.py b/tests/test_agent.py
@@ -1,4 +1,4 @@
-"""Tests for the demo streaming API endpoints.
+"""Tests for the agent streaming API endpoints.
 
 Tests verify that Claude Code is invoked correctly with proper MCP configuration.
 """
@@ -14,24 +14,24 @@
 client = TestClient(app)
 
 
-class TestDemoEndpoints:
-    """Test the /demo API endpoints."""
+class TestAgentEndpoints:
+    """Test the /agent API endpoints."""
 
     def test_status_not_found(self):
-        """GET /demo/status/{job_id} should return 404 for unknown job."""
-        response = client.get("/demo/status/nonexistent-job-id")
+        """GET /agent/status/{job_id} should return 404 for unknown job."""
+        response = client.get("/agent/status/nonexistent-job-id")
         assert response.status_code == 404
 
     def test_ask_request_model(self):
         """AskRequest model should accept question field."""
-        from policyengine_api.api.demo import AskRequest
+        from policyengine_api.api.agent import AskRequest
 
         req = AskRequest(question="Test question")
         assert req.question == "Test question"
 
     def test_ask_response_model(self):
         """AskResponse model should have job_id and status fields."""
-        from policyengine_api.api.demo import AskResponse
+        from policyengine_api.api.agent import AskResponse
 
         resp = AskResponse(job_id="test-123", status="pending")
         assert resp.job_id == "test-123"
@@ -57,7 +57,7 @@ def test_claude_cli_runs(self):
     @pytest.mark.asyncio
     async def test_stream_claude_code_invokes_claude_cli(self):
         """_stream_claude_code should invoke the claude CLI with correct args."""
-        from policyengine_api.api.demo import _stream_claude_code
+        from policyengine_api.api.agent import _stream_claude_code
 
         captured_args = []
 
@@ -105,7 +105,7 @@ async def mock_stdout_iter():
     @pytest.mark.asyncio
     async def test_stream_claude_code_yields_sse_events(self):
         """_stream_claude_code should yield properly formatted SSE events."""
-        from policyengine_api.api.demo import _stream_claude_code
+        from policyengine_api.api.agent import _stream_claude_code
 
         async def mock_create_subprocess(*args, **kwargs):
             mock_process = MagicMock()
@@ -150,7 +150,7 @@ async def mock_stdout_iter():
     @pytest.mark.asyncio
     async def test_stream_claude_code_handles_errors(self):
         """_stream_claude_code should yield error events on non-zero exit."""
-        from policyengine_api.api.demo import _stream_claude_code
+        from policyengine_api.api.agent import _stream_claude_code
 
         async def mock_create_subprocess(*args, **kwargs):
             mock_process = MagicMock()
@@ -184,7 +184,7 @@ async def mock_stdout_iter():
     @pytest.mark.asyncio
     async def test_stream_claude_code_passes_anthropic_api_key(self):
         """_stream_claude_code should pass ANTHROPIC_API_KEY in env."""
-        from policyengine_api.api.demo import _stream_claude_code
+        from policyengine_api.api.agent import _stream_claude_code
 
         captured_kwargs = []
 
@@ -214,12 +214,12 @@ async def mock_stdout_iter():
         assert "ANTHROPIC_API_KEY" in captured_kwargs[0]["env"]
 
 
-class TestDemoSandbox:
+class TestAgentSandbox:
     """Test the Modal sandbox configuration."""
 
     def test_sandbox_image_uses_bun(self):
         """Sandbox image should use bun, not npm."""
-        from policyengine_api.demo_sandbox import sandbox_image
+        from policyengine_api.agent_sandbox import sandbox_image
 
         # Just verify the image is defined - actual bun installation
         # is tested when deploying to Modal
@@ -229,7 +229,7 @@ def test_run_function_signature(self):
         """run_claude_code_in_sandbox should accept question and api_base_url."""
         import inspect
 
-        from policyengine_api.demo_sandbox import run_claude_code_in_sandbox
+        from policyengine_api.agent_sandbox import run_claude_code_in_sandbox
 
         sig = inspect.signature(run_claude_code_in_sandbox)
         params = list(sig.parameters.keys())
@@ -238,7 +238,7 @@ def test_run_function_signature(self):
 
     def test_modal_function_defined(self):
         """run_policy_analysis Modal function should be defined."""
-        from policyengine_api.demo_sandbox import run_policy_analysis
+        from policyengine_api.agent_sandbox import run_policy_analysis
 
         # Modal functions are wrapped, so check it exists and has expected attributes
         assert run_policy_analysis is not None

Original file line number	Diff line number	Diff line change
`@@ -13,7 +13,7 @@ const navigation = [`
`13`	`13`	`{ name: "Quick start", href: "/quickstart" },`
`14`	`14`	`{ name: "Environment setup", href: "/setup" },`
`15`	`15`	`{ name: "MCP integration", href: "/mcp" },`
`16`		`- { name: "AI demo", href: "/demo" },`
	`16`	`+ { name: "AI agent", href: "/agent" },`
`17`	`17`	`],`
`18`	`18`	`},`
`19`	`19`	`{`