tests: add assertion for correct RAG results

defenseunicorns · Sep 4, 2024 · 9e31599 · 9e31599
1 parent 8a13f33
commit 9e31599
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/integration/api/test_rag_files.py b/tests/integration/api/test_rag_files.py
@@ -70,6 +70,7 @@ def test_rag_needle_haystack():
     # Get the response content from the last message
     message_content = messages[-1].content[0].text
     assert isinstance(message_content, Text)
+    assert "orange" in message_content.value
     assert len(message_content.annotations) > 0
 
     for a in message_content.annotations: