Upload 164 files

src/apis/controllers/chat_controller.py

@@ -153,6 +153,89 @@ async def chat_streaming_function(user, data: Chat, background_tasks: Background
     )
 
 
+async def chat_streaming_no_login_function(
+    data: Chat, background_tasks: BackgroundTasks
+):
+    human_message = data.message
+    history = data.history
+    lat = data.lat
+    long = data.long
+    language = data.language
+    logger.info(f"Language: {language}")
+    process_history = post_process_history(history) if history is not None else None
+    config = {
+        "configurable": {
+            "user_id": None,
+            "user_email": None,
+            "contact_number": None,
+            "session_id": None,
+            "lat": lat,
+            "long": long,
+        }
+    }
+    initial_input = {
+        "messages": [("user", human_message)],
+        "messages_history": process_history,
+        "entry_message": None,
+        "manual_save": False,
+        "intent": data.intent,
+        "language": language,
+        "tool_name": None,
+        "ever_leave_skill": False,
+    }
+    last_output_state = None
+    temp = ""
+    async for event in workflow.astream(
+        input=initial_input,
+        config=config,
+        stream_mode=["messages", "values"],
+    ):
+        event_type, event_message = event
+        if event_type == "messages":
+            message, metadata = event_message
+            if (
+                isinstance(message, AIMessageChunk)
+                and message.tool_calls
+                and message.tool_call_chunks[0]["name"] != "ClassifyUserIntent"
+            ):
+                tool_name = message.tool_call_chunks[0]["name"]
+                message_yield = json.dumps(
+                    {"type": "tool_call", "content": tool_name}, ensure_ascii=False
+                )
+                print(message_yield)
+                yield message_yield + "\n"
+            if metadata["langgraph_node"] in [
+                "primary_assistant",
+                "scheduling_agent",
+                "book_hotel_agent",
+            ]:
+
+                if message.content:
+                    temp += message.content
+                    message_yield = json.dumps(
+                        {"type": "message", "content": temp}, ensure_ascii=False
+                    )
+                    print(message_yield)
+                    yield message_yield + "\n"
+        if event_type == "values":
+            last_output_state = event_message
+
+    final_ai_output = last_output_state["messages"][-1].content
+    final_intent = last_output_state["intent"]
+    tool_name_important = last_output_state["tool_name"]
+
+    final_response = json.dumps(
+        {
+            "type": "final",
+            "content": final_ai_output,
+            "intent": final_intent,
+            "tool_name": tool_name_important,
+        },
+        ensure_ascii=False,
+    )
+    yield final_response
+
+
 async def chat_function(user, data: Chat, background_tasks: BackgroundTasks):
     message = data.message
     history = data.history

src/apis/controllers/location_controller.py

@@ -81,6 +81,12 @@ def get_weather(lat, long):
         )
 
 
+def get_weather_api(destination: str):
+    lat, long = get_lat_long_location(destination)
+    print("lat", lat, "long", long)
+    return json.loads(get_weather(lat, long).body)
+
+
 def get_lat_long_location(location_name):
     """
     Retrieve the latitude and longitude for a given location name using Geoapify's Geocoding API.

src/apis/interfaces/api_interface.py

@@ -6,10 +6,10 @@ from typing import List, Union
 
 class Chat(BaseModel):
     message: str = Field(..., title="Message from user")
-    session_id: Optional[str] = Field("6701fe32d76fde9d8df1de8e", title="Session Id")
+    session_id: Optional[str] = Field(None, title="Session Id")
     history: Optional[list] = Field(None, title="Chat history")
-    lat: Optional[float] = Field(13.717162954654036, title="Latitude")
-    long: Optional[float] = Field(109.21054173319894, title="Longitude")
+    lat: Optional[float] = Field(None, title="Latitude")
+    long: Optional[float] = Field(None, title="Longitude")
     intent: Optional[str] = Field(None, title="Intent")
     language: Optional[str] = Field("en", title="Language")
 

src/apis/routes/chat_route.py

@@ -10,6 +10,7 @@ from src.apis.controllers.chat_controller import (
     get_history_function,
     list_chat_history_function,
     delete_chat_history_function,
+    chat_streaming_no_login_function,
 )
 from src.utils.logger import logger
 from src.apis.models.user_models import User
@@ -41,6 +42,15 @@ async def chat_streaming_router(
         media_type="text/plain",
     )
 
+@router.post("/chat_streaming_guest", status_code=status.HTTP_200_OK)
+async def chat_streaming_guest_router(
+    data: Chat, background_tasks: BackgroundTasks
+):
+    return StreamingResponse(
+        chat_streaming_no_login_function(data, background_tasks),
+        media_type="text/plain",
+    )
+
 
 @router.post("/get_chat_history", status_code=status.HTTP_200_OK)
 async def get_chat_history(user: user_dependency, background_tasks: BackgroundTasks):

src/apis/routes/location_route.py

@@ -5,8 +5,10 @@ from src.apis.controllers.location_controller import (
     get_location_details,
     get_nearby_places,
     get_places,
+    get_weather,
+    get_weather_api,
 )
-from fastapi import APIRouter, status, Depends
+from fastapi import APIRouter, status, Depends, Query
 from typing import Annotated
 from fastapi.responses import JSONResponse
 
@@ -33,3 +35,14 @@ def get_nearby(body: Location, user: user_dependency):
 @router.post("/places", status_code=status.HTTP_200_OK)
 def get_near_places(body: Location):
     return get_places(body.lat, body.long, body.radius, body.categories)
+
+
+@router.post("/weather", status_code=status.HTTP_200_OK)
+def get_weather_with_geo(body: Location):
+    return get_weather(body.lat, body.long)
+
+
+@router.get("/weather_text", status_code=status.HTTP_200_OK)
+def get_weather_text(destination: str = Query(..., min_length=3)):
+    print("destination", destination)
+    return get_weather_api(destination)

src/langgraph/multi_agent/chat/chat_flow.py

@@ -16,12 +16,13 @@ from src.langgraph.utils_function.function_graph import (
     get_history,
     save_history,
 )
-from src.langgraph.tools.destination_tools import destination_suggestion
+from src.langgraph.tools.destination_tools import destination_suggestion, get_weather
 from src.utils.logger import logger
 
 primary_assistant_tools = [
     TavilySearchResults(max_results=2),
     destination_suggestion,
+    get_weather,
 ]
 
 assistant_runnable = primary_assistant_prompt | llm.bind_tools(
@@ -60,6 +61,7 @@ class ChatBot:
         self.primary_assistant_tools = [
             TavilySearchResults(max_results=2),
             destination_suggestion,
+            get_weather,
         ]
         self.assistant_runnable = assistant_runnable
 

src/langgraph/multi_agent/chat/scheduling_flow.py

@@ -1,5 +1,7 @@
+from langchain_core.runnables import RunnableConfig
 from langgraph.prebuilt import tools_condition
 from langgraph.graph import END, StateGraph
+from langchain_core.messages import HumanMessage
 from src.langgraph.tools.scheduling_tools import (
     create_a_activity,
     search_activities,
@@ -21,6 +23,7 @@ from src.langgraph.state import State
 from src.utils.logger import logger
 from src.langgraph.tools.plan_itinerary import plan_itinerary
 
+
 class SchedulingAgent:
     def __init__(self, builder: StateGraph):
         self.builder = builder
@@ -45,7 +48,9 @@ class SchedulingAgent:
             and "Do you want to run the following tool(s)?"
             in state["messages_history"][-1].content
         ):
-            if state["messages"][-1].tool_calls:
+            if state["messages"][-1].tool_calls and state["messages"][-1].tool_calls[0][
+                "name"
+            ] in [r.name for r in self.scheduling_sensitive_tools]:
                 logger.info("Sensitive tools")
                 return "scheduling_sensitive_tools"
             logger.info("Safe tools")

src/langgraph/tools/destination_tools.py

@@ -4,20 +4,7 @@ from src.apis.controllers.destination_controller import (
 )
 from langchain_core.runnables.config import RunnableConfig
 from src.utils.logger import logger
-from src.apis.controllers.location_controller import (
-    get_nearby_places,
-    get_lat_long_location,
-    get_geometry_nearby_places,
-)
-from src.langgraph.models.model_validator import NearlyDestinationRecommendation
-from src.utils.helper import get_google_map_url
-from src.langgraph.langchain.prompt import (
-    routing_recommender_chain,
-    characteristic_extractor_chain,
-    RoutingRecommender,
-    CharacteristicExtractor,
-)
-import json
+from src.apis.controllers.location_controller import get_weather_api
 
 
 @tool
@@ -31,3 +18,15 @@ async def destination_suggestion(query: str, config: RunnableConfig):
     response = await destination_recommendation_func(query, tool_chat=True)
     logger.info(f"Destination recommendation output: {response}")
     return response
+
+
+@tool
+def get_weather(destination: str):
+    """
+    Get weather information for a specific destination.
+    Args:
+        destination (str): The destination to get weather information for.
+    Returns:
+        str: Weather information for the destination.
+    """
+    return get_weather_api(destination)

src/langgraph/tools/hotel_tools.py

@@ -1,17 +1,15 @@
 from langchain_core.tools import tool
 from langchain_core.runnables.config import RunnableConfig
 from datetime import datetime
-from src.apis.controllers.location_controller import get_places
-from src.utils.helper import process_controller_output, format_accommodation_markdown
 from typing import Optional
 from src.apis.controllers.hotel_controller import (
     book_hotel_controller,
     send_booking_confirmation_email,
 )
-
+from src.utils.logger import logger
 
 @tool
-def search_hotels(option: str, config: RunnableConfig):
+async def search_hotels(option: str, config: RunnableConfig):
     """
     Call this tool directly to search hotels for the user. No need to require asking for city location.
     Args:
@@ -62,6 +60,9 @@ async def book_hotel(
 
     configuration = config.get("configurable", {})
     user_id = configuration.get("user_id", None)
+    if not user_id:
+        logger.warning(f"User not login")
+        return "User not login. Please login"
     user_email = configuration.get("user_email", None)
     user_contact_number = configuration.get(
         "contact_number", "Does not have contact number"
@@ -102,9 +103,13 @@ async def cancel_hotel(
     Returns:
         str: Confirmation message of the cancellation request.
     """
+    configuration = config.get("configurable", {})
+    user_id = configuration.get("user_id", None)
+    if not user_id:
+        logger.warning(f"User not login")
+        return "User not login. Please login"
     if receiver_email is None:
         return "The hotel booking email is required"
-
     return f"Email sent to {receiver_email} to cancel booking for room {room_number} with reason: {reason}"
 
 
@@ -122,6 +127,11 @@ async def update_hotel(
         room_number (Optional[str]): Room number
         content (Optional[str]): Updated content
     """
+    configuration = config.get("configurable", {})
+    user_id = configuration.get("user_id", None)
+    if not user_id:
+        logger.warning(f"User not login")
+        return "User not login. Please login"
     if receiver_email is None:
         return "The hotel booking email is required"
     return f"Email sent to {receiver_email} to update booking for room {room_number} with content: {content}"

src/langgraph/tools/plan_itinerary.py

@@ -1,8 +1,6 @@
 from langchain_core.tools import tool
 
 from langchain_core.runnables.config import RunnableConfig
-from src.utils.logger import logger
-from typing import Optional
 from src.langgraph.multi_agent.planner.planner_flow import planner_app
 
 @tool

src/langgraph/tools/scheduling_tools.py

@@ -9,7 +9,6 @@ from src.apis.controllers.scheduling_controller import (
     search_activities_controller,
 )
 
-
 @tool
 async def create_a_activity(
     description: str,
@@ -30,6 +29,9 @@ async def create_a_activity(
         # Convert start_time and end_time to the required format
         configuration = config.get("configurable", {})
         user_id = configuration.get("user_id", None)
+        if not user_id:
+            logger.warning(f"User not login")
+            return "You are not logged in. Please log in to use this feature"
         response = await create_a_activity_controller(
             None,
             activity_category,
@@ -60,6 +62,9 @@ async def search_activities(
     try:
         configuration = config.get("configurable", {})
         user_id = configuration.get("user_id", None)
+        if not user_id:
+            logger.warning(f"User not login")
+            return "You are not logged in. Please log in to use this feature"
         response = await search_activities_controller(start_time, end_time, user_id)
         return response["message"]
     except Exception as e:
@@ -85,6 +90,9 @@ async def update_a_activiy(
     try:
         configuration = config.get("configurable", {})
         user_id = configuration.get("user_id", None)
+        if not user_id:
+            logger.warning(f"User not login")
+            return "You are not logged in. Please log in to use this feature"
         response = await update_a_activity_controller(
             None,
             activity_category,
@@ -113,6 +121,9 @@ async def delete_a_activity(
     try:
         configuration = config.get("configurable", {})
         user_id = configuration.get("user_id", None)
+        if not user_id:
+            logger.warning(f"User not login")
+            return "You are not logged in. Please log in to use this feature"
         response = await delete_activities_controller(
             None, start_time, end_time, user_id
         )
@@ -120,3 +131,5 @@ async def delete_a_activity(
     except Exception as e:
         logger.error(f"Error deleting activity: {e}")
         return f"Error deleting activity {e}"
+        logger.error(f"Error deleting activity: {e}")
+        return f"Error deleting activity {e}"

src/langgraph/tools/search_tools.py

@@ -3,6 +3,7 @@ from langchain_community.document_loaders.recursive_url_loader import RecursiveU
 from langchain_core.prompts import PromptTemplate
 from langchain_community.tools.tavily_search import TavilySearchResults
 from langchain_core.tools import tool
+from src.apis.controllers.location_controller import get_weather_api
 from src.langgraph.langchain.llm import llm_flash
 from src.utils.logger import logger
 from langchain_community.tools import DuckDuckGoSearchResults

src/utils/helper.py

@@ -56,9 +56,13 @@ def format_weather_data(weather_data):
         lat = weather_data["lat"]
         lon = weather_data["lon"]
         location = f"Latitude: {lat}, Longitude: {lon}"
+        icon_url = f"http://openweathermap.org/img/wn/{current_weather['weather'][0]['icon']}@2x.png"
         formatted_weather = f"In {location}, the current weather is as follows:\n"
         formatted_weather += (
-            f"Detailed status: {current_weather['weather'][0]['description']}\n"
+            f"  <img src='{icon_url}' width='100' height='100'/>\n"
+        )
+        formatted_weather += (
+            f"  Detailed status: {current_weather['weather'][0]['description']}\n"
         )
         formatted_weather += f"Wind speed: {current_weather['wind_speed']} m/s, direction: {current_weather['wind_deg']}°\n"
         formatted_weather += f"Humidity: {current_weather['humidity']}%\n"