Imporve gateway auth for all services

services/data/app/api/traffic.py

@@ -5,7 +5,7 @@
 
 from fastapi import APIRouter, Depends, HTTPException, Query
 from sqlalchemy.ext.asyncio import AsyncSession
-from typing import List, Optional
+from typing import List, Dict, Any
 from datetime import datetime, timedelta
 import structlog
 
@@ -19,6 +19,11 @@ from app.schemas.external import (
     DateRangeRequest
 )
 
+from shared.auth.decorators import (
+    get_current_user_dep,
+    get_current_tenant_id_dep
+)
+
 router = APIRouter()
 traffic_service = TrafficService()
 logger = structlog.get_logger()
@@ -27,7 +32,7 @@ logger = structlog.get_logger()
 async def get_current_traffic(
     latitude: float = Query(..., description="Latitude"),
     longitude: float = Query(..., description="Longitude"),
-    current_user: AuthInfo = Depends(get_current_user)
+    current_user: Dict[str, Any] = Depends(get_current_user_dep),
 ):
     """Get current traffic data for location"""
     try:
@@ -72,7 +77,7 @@ async def get_historical_traffic(
     start_date: datetime = Query(..., description="Start date"),
     end_date: datetime = Query(..., description="End date"),
     db: AsyncSession = Depends(get_db),
-    current_user: AuthInfo = Depends(get_current_user)
+    current_user: Dict[str, Any] = Depends(get_current_user_dep),
 ):
     """Get historical traffic data"""
     try:
@@ -116,7 +121,7 @@ async def store_traffic_data(
     latitude: float = Query(..., description="Latitude"),
     longitude: float = Query(..., description="Longitude"),
     db: AsyncSession = Depends(get_db),
-    current_user: AuthInfo = Depends(get_current_user)
+    current_user: Dict[str, Any] = Depends(get_current_user_dep)
 ):
     """Store current traffic data to database"""
     try: