From 9429253ac222a4e5f92b737e2443845247c9aef5 Mon Sep 17 00:00:00 2001 From: Brandon Hancock Date: Thu, 2 Jan 2025 15:18:43 -0500 Subject: [PATCH] more test fixes --- tests/knowledge/knowledge_test.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/tests/knowledge/knowledge_test.py b/tests/knowledge/knowledge_test.py index 440ffe782..2af4f2110 100644 --- a/tests/knowledge/knowledge_test.py +++ b/tests/knowledge/knowledge_test.py @@ -6,9 +6,6 @@ from unittest.mock import patch 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.csv_knowledge_source import CSVKnowledgeSource 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 -@pytest.mark.vcr(record_mode="new_episodes", filter_headers=["authorization"]) +@pytest.mark.vcr(filter_headers=["authorization"]) def test_docling_source_with_local_file(): current_dir = Path(__file__).parent pdf_path = current_dir / "crewai_quickstart.pdf"