feat: date range in download report

api/db/campaign_client.py

@@ -365,12 +365,29 @@ class CampaignClient(BaseDBClient):
             result = await session.execute(query)
             return list(result.scalars().all())
 
-    async def get_completed_runs_for_report(self, campaign_id: int) -> list:
+    async def get_completed_runs_for_report(
+        self,
+        campaign_id: int,
+        start_date: Optional[datetime] = None,
+        end_date: Optional[datetime] = None,
+    ) -> list:
         """Get completed workflow runs for campaign report CSV.
 
         Returns rows with only the columns needed for report generation.
         """
         async with self.async_session() as session:
+            conditions = [
+                WorkflowRunModel.campaign_id == campaign_id,
+                WorkflowRunModel.is_completed.is_(True),
+                WorkflowRunModel.cost_info["call_duration_seconds"]
+                .as_string()
+                .isnot(None),
+            ]
+            if start_date is not None:
+                conditions.append(WorkflowRunModel.created_at >= start_date)
+            if end_date is not None:
+                conditions.append(WorkflowRunModel.created_at <= end_date)
+
             query = (
                 select(
                     WorkflowRunModel.id,
@@ -381,13 +398,7 @@ class CampaignClient(BaseDBClient):
                     WorkflowRunModel.logs,
                     WorkflowRunModel.public_access_token,
                 )
-                .where(
-                    WorkflowRunModel.campaign_id == campaign_id,
-                    WorkflowRunModel.is_completed.is_(True),
-                    WorkflowRunModel.cost_info["call_duration_seconds"]
-                    .as_string()
-                    .isnot(None),
-                )
+                .where(*conditions)
                 .order_by(WorkflowRunModel.created_at.desc())
             )
             result = await session.execute(query)

api/routes/campaign.py

@@ -706,13 +706,21 @@ async def get_campaign_source_download_url(
 async def download_campaign_report(
     campaign_id: int,
     user: UserModel = Depends(get_user),
+    start_date: Optional[datetime] = Query(
+        None, description="Filter runs created on or after this datetime (ISO 8601)"
+    ),
+    end_date: Optional[datetime] = Query(
+        None, description="Filter runs created on or before this datetime (ISO 8601)"
+    ),
 ) -> StreamingResponse:
     """Download a CSV report of completed campaign runs."""
     campaign = await db_client.get_campaign(campaign_id, user.selected_organization_id)
     if not campaign:
         raise HTTPException(status_code=404, detail="Campaign not found")
 
-    output, filename = await generate_campaign_report_csv(campaign_id)
+    output, filename = await generate_campaign_report_csv(
+        campaign_id, start_date=start_date, end_date=end_date
+    )
 
     return StreamingResponse(
         output,

api/services/campaign/report.py

@@ -1,6 +1,7 @@
 import csv
 import io
-from typing import Any, List
+from datetime import datetime
+from typing import Any, List, Optional
 
 from api.constants import BACKEND_API_ENDPOINT
 from api.db import db_client
@@ -27,12 +28,18 @@ def _collect_extracted_variable_keys(runs: List[Any]) -> list[str]:
     return list(keys)
 
 
-async def generate_campaign_report_csv(campaign_id: int) -> tuple[io.StringIO, str]:
+async def generate_campaign_report_csv(
+    campaign_id: int,
+    start_date: Optional[datetime] = None,
+    end_date: Optional[datetime] = None,
+) -> tuple[io.StringIO, str]:
     """Generate a CSV report for a campaign.
 
     Returns a tuple of (csv_output, filename).
     """
-    runs = await db_client.get_completed_runs_for_report(campaign_id)
+    runs = await db_client.get_completed_runs_for_report(
+        campaign_id, start_date=start_date, end_date=end_date
+    )
 
     # Collect dynamic extracted variable columns
     extracted_var_keys = _collect_extracted_variable_keys(runs)