fix test error

Signed-off-by: Akihiko Kuroda <akihikokuroda2020@gmail.com>

Author: akihikokuroda

docs/examples/citation_requirement_example.py

@@ -98,5 +98,3 @@ async def main():
 
 if __name__ == "__main__":
     asyncio.run(main())
-
-# Made with Bob

mellea/stdlib/requirements/rag.py

@@ -192,6 +192,13 @@ async def validate(
 
             context_before_response = ChatContext()
 
+        # Handle empty response before calling intrinsic
+        total_chars = len(response)
+        if total_chars == 0:
+            return ValidationResult(
+                True, reason="Empty response has 100% citation coverage", score=1.0
+            )
+
         # Call find_citations intrinsic
         try:
             # Import here to avoid circular dependency
@@ -206,11 +213,6 @@ async def validate(
             )
 
         # Calculate citation coverage
-        total_chars = len(response)
-        if total_chars == 0:
-            return ValidationResult(
-                True, reason="Empty response has 100% citation coverage", score=1.0
-            )
 
         cited_chars = sum(
             citation["response_end"] - citation["response_begin"]
@@ -336,6 +338,3 @@ def citation_check(
         documents=documents,
         description=description,
     )
-
-
-# Made with Bob

test/stdlib/requirements/test_rag.py

@@ -14,7 +14,7 @@
 @pytest.mark.requires_heavy_ram
 async def test_citation_requirement_basic():
     """Test basic citation requirement functionality."""
-    backend = LocalHFBackend(model_id="meta-llama/Llama-3.2-1B-Instruct")
+    backend = LocalHFBackend(model_id="ibm-granite/granite-4.0-micro")
 
     # Create documents
     docs = [
@@ -46,7 +46,7 @@ async def test_citation_requirement_basic():
 @pytest.mark.requires_heavy_ram
 async def test_citation_requirement_with_constructor_documents():
     """Test citation requirement with documents in constructor."""
-    backend = LocalHFBackend(model_id="meta-llama/Llama-3.2-1B-Instruct")
+    backend = LocalHFBackend(model_id="ibm-granite/granite-4.0-micro")
 
     # Create documents
     docs = [
@@ -77,7 +77,7 @@ async def test_citation_requirement_with_constructor_documents():
 @pytest.mark.requires_heavy_ram
 async def test_citation_check_factory():
     """Test citation_check factory function."""
-    backend = LocalHFBackend(model_id="meta-llama/Llama-3.2-1B-Instruct")
+    backend = LocalHFBackend(model_id="ibm-granite/granite-4.0-micro")
 
     # Create documents
     docs = [Document(doc_id="doc1", text="The sky is blue during the day.")]
@@ -241,7 +241,7 @@ def test_citation_requirement_default_description():
 @pytest.mark.requires_heavy_ram
 async def test_citation_requirement_empty_response():
     """Test citation requirement with empty response."""
-    backend = LocalHFBackend(model_id="meta-llama/Llama-3.2-1B-Instruct")
+    backend = LocalHFBackend(model_id="ibm-granite/granite-4.0-micro")
 
     # Create documents
     docs = [Document(doc_id="doc1", text="The sky is blue.")]
@@ -266,7 +266,7 @@ async def test_citation_requirement_empty_response():
 @pytest.mark.requires_heavy_ram
 async def test_citation_requirement_threshold_boundary():
     """Test citation requirement at exact threshold boundary."""
-    backend = LocalHFBackend(model_id="meta-llama/Llama-3.2-1B-Instruct")
+    backend = LocalHFBackend(model_id="ibm-granite/granite-4.0-micro")
 
     # Create documents
     docs = [Document(doc_id="doc1", text="The sky is blue during the day.")]
@@ -293,6 +293,3 @@ async def test_citation_requirement_threshold_boundary():
         assert result.as_bool()
     else:
         assert not result.as_bool()
-
-
-# Made with Bob