# Add the user query
query = "What's your remote-working policy?"

query_input = [{"content": [{"type": "text", "text": query}]}]