Citi interview question

System design related to GenAi. Coding question (python)