Specify

wmde · RickiJay-WMDE · Sep 8, 2025 · Aug 27, 2025 · Aug 28, 2025 · Aug 28, 2025
commit bd8f5a5e98d942177af093405b154143e0a9c89c
diff --git a/app.py b/app.py
@@ -50,7 +50,9 @@ def read_root():
 
 
 @app.get("/csv/metrics", response_class=StreamingResponse)
-async def metric_csv(request: Request, background_tasks: BackgroundTasks):
+async def metric_csv(
+    request: Request, background_tasks: BackgroundTasks
+) -> StreamingResponse:
     """Quantity CSV"""
 
     # authenticate_request(request)

diff --git a/export_csv/metric.py b/export_csv/metric.py
@@ -1,7 +1,8 @@
 """Quantity CSV"""
 
 from fastapi import BackgroundTasks
-from sqlalchemy import and_, func, or_, select
+from fastapi.responses import StreamingResponse
+from sqlalchemy import Select, and_, func, or_, select
 
 from export_csv.util import export_csv
 from model.database import (
@@ -15,7 +16,7 @@
 from model.enum import WikibaseType
 
 
-async def export_metric_csv(background_tasks: BackgroundTasks):
+async def export_metric_csv(background_tasks: BackgroundTasks) -> StreamingResponse:
     """CSV with Requested Metrics"""
 
     query = get_metrics_query()
@@ -28,7 +29,7 @@ async def export_metric_csv(background_tasks: BackgroundTasks):
     )
 
 
-def get_metrics_query():
+def get_metrics_query() -> Select:
     """
     Filter Out Offline and Test Wikis