from customer_service_agent.engines import calculate_price

def calculate_price_node(state: AgentState) -> dict[str, Any]:
    return {
        "calculated_price": calculate_price(
            state["booking_details"]
        )
    }

graph.add_node("calculate_price", calculate_price_node)