Citi Interview Question

System design related to GenAi. Coding question (python)