Fix token issue

services/data/app/api/traffic.py

@@ -9,7 +9,7 @@ from typing import List, Optional
 from datetime import datetime, timedelta
 
 from app.core.database import get_db
-from app.core.auth import verify_token
+from app.core.auth import get_current_user, AuthInfo
 from app.services.traffic_service import TrafficService
 from app.services.messaging import data_publisher
 from app.schemas.external import (
@@ -25,7 +25,7 @@ traffic_service = TrafficService()
 async def get_current_traffic(
     latitude: float = Query(..., description="Latitude"),
     longitude: float = Query(..., description="Longitude"),
-    current_user: dict = Depends(verify_token)
+    current_user: AuthInfo = Depends(get_current_user)
 ):
     """Get current traffic data for location"""
     try:
@@ -52,7 +52,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: dict = Depends(verify_token)
+    current_user: AuthInfo = Depends(get_current_user)
 ):
     """Get historical traffic data"""
     try: