Remove corpus counts from chat chrome

apps/ai-service/tests/test_citation_and_intro.py

@@ -84,7 +84,8 @@ def test_only_cited_sources_are_returned():
     result = _answerable(_evidence(0, 100), _evidence(1, 200), _evidence(2, 300))
     service = GroundedAnswerService(
         _Routing(result),
-        _Generator({"answer": "Chỉ dùng đoạn hai [2].", "evidence_sufficient": True}),
+        _Generator({"claims": [{"text": "Chỉ dùng đoạn hai", "citations": [2]}],
+                    "evidence_sufficient": True}),
     )
 
     grounded = service.answer("q", SubjectScope.HUMAN, QueryIntent.FACT_LOOKUP)
@@ -105,7 +106,8 @@ def test_answer_citing_nothing_is_rejected_not_dressed_up_with_borrowed_citation
         # and it must not silently degrade to a raw extractive quote either
         # (owner correction, 2026-08-06: no fallback to the retired
         # offline-extractive shape when a real generator is configured).
-        _Generator({"answer": "Không có trích dẫn.", "evidence_sufficient": True}),
+        _Generator({"claims": [{"text": "Không có trích dẫn.", "citations": []}],
+                    "evidence_sufficient": True}),
     )
 
     grounded = service.answer("q", SubjectScope.HUMAN, QueryIntent.FACT_LOOKUP)
@@ -157,8 +159,8 @@ def test_sufficiency_check_outage_fails_open_to_generation_not_abstain():
     the real safety net on whatever gets generated next."""
     result = _answerable(_evidence(0, 100), _evidence(1, 200))
     gen = _Generator(
-        {"answer": "Đoạn bằng chứng 0 [1].", "evidence_sufficient": True,
-         "clarifying_question": None},
+        {"claims": [{"text": "Đoạn bằng chứng 0", "citations": [1]}],
+         "evidence_sufficient": True, "clarifying_question": None},
         sufficiency_payload=AnswerGenerationUnavailable("Bedrock unreachable"),
     )
     service = GroundedAnswerService(_Routing(result), gen)
@@ -177,8 +179,8 @@ def test_sufficient_query_is_not_turned_into_a_clarification():
 
     g = service.answer("liều người lớn", SubjectScope.HUMAN, QueryIntent.FACT_LOOKUP)
 
-    # sufficiency passes; generation then runs (its payload lacks answer keys, so
-    # it falls back to the source text) — the point is no clarification fired.
+    # Sufficiency passes; generation then runs (its payload lacks answer keys,
+    # so it fails closed) — the point is no clarification fired.
     assert g.clarification is None
 
 
@@ -208,8 +210,10 @@ def test_list_mode_skips_the_sufficiency_clarify():
     must never call it at all, so only the real answer payload is ever read."""
     result = _answerable(_evidence(0, 100), _evidence(1, 200))
     gen = _Generator(
-        {"answer": "Đoạn bằng chứng 0 [1]. Đoạn bằng chứng 1 [2].",
-         "evidence_sufficient": True, "clarifying_question": None},
+        {"claims": [
+            {"text": "Đoạn bằng chứng 0", "citations": [1]},
+            {"text": "Đoạn bằng chứng 1", "citations": [2]},
+         ], "evidence_sufficient": True, "clarifying_question": None},
         sufficiency_payload={
             "sufficient": False, "clarifying_question": "Loại nào?", "quick_replies": [],
         },