Upload 157 files

src/apis/controllers/destination_controller.py

@@ -54,16 +54,23 @@ async def destination_suggestion_controller(
             raise HTTPException(status_code=500, detail=f"Request failed: {str(e)}")
 
 
-async def destination_recommendation_func(query, top_k=5):
+async def destination_recommendation_func(query, top_k=5, tool_chat=False):
     routing: RoutingRecommender = routing_recommender_chain.invoke({"query": query})
     print("label", routing.label)
     if routing.label == "characteristic":
         output = await destination_suggestion_controller(query, top_k)
-        return output
+        if tool_chat:
+            return output
+        output = [
+            {
+                "name": i,
+                "map_url": "https://www.google.com/maps/search/109.23333,13.76667",
+            }
+            for i in output
+        ]
     characteristic_extract_response: CharacteristicExtractor = (
         await characteristic_extractor_chain.ainvoke({"query": query})
     )
-    print("characterist", characteristic_extract_response)
     lat, lon = get_lat_long_location(characteristic_extract_response.main_place)
     response = get_places(
         lat,
@@ -72,7 +79,15 @@ async def destination_recommendation_func(query, top_k=5):
         available_categories.get(characteristic_extract_response.kind, None),
         top_k,
     )
-    response = json.loads(response.body)
-    output = [i.get("Accommodation_Name", None) for i in response]
+    output = json.loads(response.body)
+    if tool_chat:
+        output = [
+            {
+                "name": i["name"],
+                "address": i["address"],
+                "distance_km": i["distance_km"],
+            }
+            for i in output
+        ]
 
     return output

src/langgraph/tools/destination_tools.py

@@ -29,7 +29,6 @@ async def destination_recommendation(query: str, config: RunnableConfig):
     Using Vietnamese language for better results
     """
     logger.info(f"Destination recommendation query: {query}")
-    output = await destination_recommendation_func(query)
-    logger.info(f"Destination recommendation output: {output}")
-    return output
-
+    response = await destination_recommendation_func(query)
+    logger.info(f"Destination recommendation output: {response}")
+    return response

src/utils/helper.py

@@ -104,8 +104,8 @@ def format_geoapify_response(
         if include_latnlong:
             formatted_item["lat"] = place_lat
             formatted_item["lon"] = place_lon
-        formatted_item["Accommodation_Name"] = feature["properties"]["address_line1"]
-        formatted_item["Address"] = feature["properties"]["formatted"]
+        formatted_item["name"] = feature["properties"]["address_line1"]
+        formatted_item["address"] = feature["properties"]["formatted"]
         formatted_item["distance_km"] = str(round(distance, 2)) + " km"
 
         if "contact" in feature["properties"]:
@@ -116,6 +116,7 @@ def format_geoapify_response(
 
         if "accommodation" in feature["properties"]:
             formatted_item["accommodation"] = feature["properties"]["accommodation"]
+        formatted_item["map_url"] = get_google_map_url(place_lat, place_lon)
 
         formatted_data.append(formatted_item)
 
@@ -302,5 +303,6 @@ def serialize_datetime(obj):
         return str(obj)
     return obj
 
-def get_google_map_url(geometry:list):
-    return f"https://www.google.com/maps/search/{geometry[1]},{geometry[0]}"
+
+def get_google_map_url(lat, long):
+    return f"https://www.google.com/maps/search/{lat},{long}"