Remove chat rate limit

apps/ai-service/tests/test_clinical_condition_flow.py

@@ -15,6 +15,7 @@ from rag.clinical import (
     RenalContext,
 )
 from rag.models import Evidence, SourceRef
+from rag.sections import SECTION_PHRASES
 from rag.understanding import (
     LlmQueryUnderstander,
     QueryFrame,
@@ -173,12 +174,57 @@ def test_named_drug_safety_and_purpose_cues_override_noisy_relation_frames():
         replace(noisy, drugs=("paracetamol_acetaminophen",)),
         "Paracetamol có tác dụng gì?",
     )
+    golden_contraindication = _apply_named_drug_cues(
+        replace(noisy, drugs=("paracetamol_acetaminophen",)),
+        "Chống chỉ định của Paracetamol là gì?",
+    )
+    golden_contraindication_with_omitted_llm_drug = _apply_named_drug_cues(
+        replace(noisy, drugs=()),
+        "Chống chỉ định của Paracetamol là gì?",
+        resolved_turn_drug="paracetamol_acetaminophen",
+    )
+    reverse_contraindication = _apply_named_drug_cues(
+        replace(noisy, drugs=("warfarin",)),
+        "Thuốc nào chống chỉ định ở bệnh nhân đang dùng warfarin?",
+    )
 
     assert safety.turn_type == "drug_attribute"
     assert safety.attribute == "chong_chi_dinh"
     assert safety.needs_clarify is False
     assert purpose.turn_type == "drug_to_condition"
     assert purpose.attribute == "chi_dinh"
+    assert golden_contraindication.turn_type == "drug_attribute"
+    assert golden_contraindication.attribute == "chong_chi_dinh"
+    assert golden_contraindication.condition_relation == ConditionRelation.CONTRAINDICATION
+    assert golden_contraindication_with_omitted_llm_drug.turn_type == "drug_attribute"
+    assert golden_contraindication_with_omitted_llm_drug.drugs == (
+        "paracetamol_acetaminophen",
+    )
+    assert reverse_contraindication.turn_type == "condition_relation"
+
+
+def test_owned_section_phrase_routes_every_monograph_section_without_llm_help():
+    noisy = QueryFrame(
+        turn_type="condition_relation",
+        condition_relation=ConditionRelation.CONTRAINDICATION,
+        needs_clarify=True,
+        clarify_reason="Cần làm rõ",
+    )
+
+    for section_key, phrases in SECTION_PHRASES.items():
+        phrase = phrases[0]
+        corrected = _apply_named_drug_cues(
+            noisy,
+            f"{phrase} của Paracetamol là gì?",
+            resolved_turn_drug="paracetamol_acetaminophen",
+            resolved_section_key=section_key,
+            resolved_section_phrase=phrase,
+        )
+
+        assert corrected.turn_type == "drug_attribute", section_key
+        assert corrected.drugs == ("paracetamol_acetaminophen",), section_key
+        assert corrected.attribute == section_key
+        assert corrected.needs_clarify is False
 
 
 def test_ambiguous_condition_clarifies_before_retrieval():