From c2ed1f23554c9120e47a37bc11e6959a4633ca30 Mon Sep 17 00:00:00 2001 From: Lorenze Jay Date: Mon, 16 Dec 2024 19:44:29 -0800 Subject: [PATCH] added test for multiple sources for file_paths --- tests/knowledge/knowledge_test.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/tests/knowledge/knowledge_test.py b/tests/knowledge/knowledge_test.py index 518104db3..2602a5e3c 100644 --- a/tests/knowledge/knowledge_test.py +++ b/tests/knowledge/knowledge_test.py @@ -564,3 +564,14 @@ def test_docling_source(mock_vector_db): results = mock_vector_db.query(query) assert any("reward hacking" in result["context"].lower() for result in results) mock_vector_db.query.assert_called_once() + + +def test_multiple_docling_sources(): + urls = [ + "https://lilianweng.github.io/posts/2024-11-28-reward-hacking/", + "https://lilianweng.github.io/posts/2024-07-07-hallucination/", + ] + docling_source = DoclingSource(file_paths=urls) + + assert docling_source.file_paths == urls + assert docling_source.content is not None