fix: formated and linted code

Author: zlendo1

app/services/ai_agents_service.py

@@ -5,7 +5,7 @@
 
 from app.core.config import settings
 from app.schemas.health import ServiceHealth
-from app.schemas.itinary import Itinerary, Leg
+from app.schemas.itinary import Itinerary
 
 logger = logging.getLogger(__name__)
 
@@ -68,11 +68,11 @@ async def health_check(self) -> ServiceHealth:
     async def get_itinerary_insight(self, itinerary: Itinerary) -> None:
         """
         Get AI-generated insights for a complete itinerary and enrich it with AI data.
-        
+
         This method sends the entire itinerary to the AI service and receives back:
         - ai_description: Overall description of the itinerary
         - ai_insights: List of insights for each leg
-        
+
         The method directly modifies the itinerary object in place, setting:
         - itinerary.ai_description
         - leg.ai_insight for each leg
@@ -120,18 +120,18 @@ async def get_itinerary_insight(self, itinerary: Itinerary) -> None:
 
             if response.status_code == 200:
                 data = response.json()
-                
+
                 # Set the itinerary-level AI description
                 itinerary.ai_description = data.get("ai_description")
-                
+
                 # Set AI insights for each leg
                 ai_insights = data.get("ai_insights", [])
                 for idx, leg in enumerate(itinerary.legs):
                     if idx < len(ai_insights):
                         leg.ai_insight = ai_insights[idx]
                     else:
                         leg.ai_insight = None
-                
+
                 return
 
             logger.warning(

tests/endpoints/test_routes.py

@@ -442,11 +442,15 @@ def test_search_routes_with_ai_insights_success(client: TestClient, sample_itine
 
             # Mock AI service to enrich itinerary in place
             async def mock_get_itinerary_insight(itinerary):
-                itinerary.ai_description = "This route offers a good balance of walking and public transport."
+                itinerary.ai_description = (
+                    "This route offers a good balance of walking and public transport."
+                )
                 itinerary.legs[0].ai_insight = "Short walk to the bus stop."
                 itinerary.legs[1].ai_insight = "Express bus with comfortable seats."
 
-            mock_ai_service.get_itinerary_insight = AsyncMock(side_effect=mock_get_itinerary_insight)
+            mock_ai_service.get_itinerary_insight = AsyncMock(
+                side_effect=mock_get_itinerary_insight
+            )
 
             response = client.post(
                 "/api/v1/routes/search",
@@ -490,7 +494,9 @@ async def mock_get_itinerary_insight_unavailable(itinerary):
                 # Service unavailable - does not modify itinerary
                 pass
 
-            mock_ai_service.get_itinerary_insight = AsyncMock(side_effect=mock_get_itinerary_insight_unavailable)
+            mock_ai_service.get_itinerary_insight = AsyncMock(
+                side_effect=mock_get_itinerary_insight_unavailable
+            )
 
             response = client.post(
                 "/api/v1/routes/search",
@@ -527,7 +533,9 @@ async def mock_get_itinerary_insight_partial(itinerary):
                 # Only set description, not leg insights
                 itinerary.ai_description = "This is a good route."
 
-            mock_ai_service.get_itinerary_insight = AsyncMock(side_effect=mock_get_itinerary_insight_partial)
+            mock_ai_service.get_itinerary_insight = AsyncMock(
+                side_effect=mock_get_itinerary_insight_partial
+            )
 
             response = client.post(
                 "/api/v1/routes/search",

tests/services/test_ai_agents_service.py

@@ -185,10 +185,7 @@ async def test_get_itinerary_insight_success(ai_service, sample_itinerary):
     mock_response.status_code = 200
     mock_response.json.return_value = {
         "ai_description": "This route combines a short walk with a direct bus connection.",
-        "ai_insights": [
-            "Short walk to the bus stop.",
-            "Express bus with comfortable seats."
-        ]
+        "ai_insights": ["Short walk to the bus stop.", "Express bus with comfortable seats."],
     }
 
     with patch.object(ai_service, "_get_client") as mock_get_client:
@@ -210,9 +207,12 @@ async def test_get_itinerary_insight_success(ai_service, sample_itinerary):
         assert len(payload["legs"]) == 2
         assert payload["legs"][0]["mode"] == "WALK"
         assert payload["legs"][1]["mode"] == "BUS"
-        
+
         # Verify the itinerary was enriched
-        assert sample_itinerary.ai_description == "This route combines a short walk with a direct bus connection."
+        assert (
+            sample_itinerary.ai_description
+            == "This route combines a short walk with a direct bus connection."
+        )
         assert sample_itinerary.legs[0].ai_insight == "Short walk to the bus stop."
         assert sample_itinerary.legs[1].ai_insight == "Express bus with comfortable seats."