andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-02 07:42:40 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

more test fixes

Browse Source
This commit is contained in:
Brandon Hancock
2025-01-02 15:18:43 -05:00
parent a60c66a9af
commit 9429253ac2
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/knowledge/knowledge_test.py
View File

@@ -6,9 +6,6 @@ from unittest.mock import patch
import pytest import pytest
# Attempt to import the module and skip the test if it's not available
docling = pytest.importorskip("docling")
from crewai.knowledge.source.crew_docling_source import CrewDoclingSource from crewai.knowledge.source.crew_docling_source import CrewDoclingSource
from crewai.knowledge.source.csv_knowledge_source import CSVKnowledgeSource from crewai.knowledge.source.csv_knowledge_source import CSVKnowledgeSource
from crewai.knowledge.source.excel_knowledge_source import ExcelKnowledgeSource from crewai.knowledge.source.excel_knowledge_source import ExcelKnowledgeSource
@@ -581,7 +578,7 @@ def test_multiple_docling_sources():
assert docling_source.content is not None assert docling_source.content is not None
@pytest.mark.vcr(record_mode="new_episodes", filter_headers=["authorization"]) @pytest.mark.vcr(filter_headers=["authorization"])
def test_docling_source_with_local_file(): def test_docling_source_with_local_file():
current_dir = Path(__file__).parent current_dir = Path(__file__).parent
pdf_path = current_dir / "crewai_quickstart.pdf" pdf_path = current_dir / "crewai_quickstart.pdf"