Tích hợp luồng tạo quiz vào luồng sẵn có, route qua args, không đi vào orchestrator

core.py

@@ -18,6 +18,7 @@ from src.graph import run
 from src.nodes import final_response_node, image_response_node
 from src.pdf_processing import format_chat_history, pdf_to_markdown
 from src.qdrant_store import get_custom_prompt
+from src.quiz import generate_quiz
 from src.redis_client import redis_client
 from src.state import MAX_ITERS, AgentState
 
@@ -29,6 +30,8 @@ def final_answer(
     pdf_path: Optional[str] = None,
     image_path: Optional[str] = None,
     pdf_name: Optional[str] = None,
+    gen_quiz: bool = False,
+    k_question: Optional[int] = None,
 ) -> tuple[str, str, str | None, str | None]:
     """
     Khởi tạo AgentState, chạy graph, trả về 4 giá trị.
@@ -52,6 +55,11 @@ def final_answer(
     if not query:
         raise ValueError("query không được để trống.")
 
+    if gen_quiz:
+        t0 = time.perf_counter()
+        answer = generate_quiz(query, k_question or 10)
+        return answer, f"{time.perf_counter() - t0:.2f}s", None, None
+
     custom_prompt = get_custom_prompt(sender_id)
 
     if pdf_path is not None: