enabling local files to work and type cleanup

This commit is contained in:
Lorenze Jay
2024-12-16 20:21:47 -08:00
parent 356eb07d5f
commit 10c04d54a9
2 changed files with 12 additions and 4 deletions

View File

@@ -575,3 +575,11 @@ def test_multiple_docling_sources():
assert docling_source.file_paths == urls
assert docling_source.content is not None
def test_docling_source_with_local_file():
current_dir = Path(__file__).parent
pdf_path = current_dir / "crewai_quickstart.pdf"
docling_source = DoclingSource(file_paths=[str(pdf_path.name)])
assert docling_source.file_paths == [str(pdf_path.name)]
assert docling_source.content is not None