Update travel.py

travel.py

@@ -384,6 +384,26 @@ def get_user_input() -> dict:
         "special_requirements": special_requirements
     }
 
+def update_llm(api_key: str):
+    global llm
+    llm = ChatGoogleGenerativeAI(model="gemini-2.0-pro-exp-02-05", google_api_key=api_key)
+    if destination_research_agent:
+        destination_research_agent.llm = llm
+    if accommodation_agent:
+        accommodation_agent.llm = llm
+    if transportation_agent:
+        transportation_agent.llm = llm
+    if activities_agent:
+        activities_agent.llm = llm
+    if dining_agent:
+        dining_agent.llm = llm
+    if itinerary_agent:
+        itinerary_agent.llm = llm
+    if chatbot_agent:
+        chatbot_agent.llm = llm
+    logging.info("LLM updated with new API key.")
+
+
 def generate_travel_itinerary(user_input: dict) -> str:
     print("\nGenerating your personalized travel itinerary...\n")