Update main.py

main.py

@@ -144,6 +144,11 @@ TRAVEL PLAN:
     return structured_package
 
 
+class ChatRequest(BaseModel):
+    plan_text: str
+    user_query: str
+
+    
 @app.post("/chat", summary="Answer user questions about the generated travel plan")
 async def chat_about_plan(chat_input: ChatRequest):
     plan_text = chat_input.plan_text