From 464ca30e4634f7a41a5e2b273ab04edc1ce1b3d8 Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Mon, 10 Mar 2025 17:12:27 +0000 Subject: [PATCH] Fix linting issues in test file Co-Authored-By: Joe Moura --- tests/knowledge/test_string_knowledge_source_fix.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/knowledge/test_string_knowledge_source_fix.py b/tests/knowledge/test_string_knowledge_source_fix.py index 905892de6..690b9f640 100644 --- a/tests/knowledge/test_string_knowledge_source_fix.py +++ b/tests/knowledge/test_string_knowledge_source_fix.py @@ -1,9 +1,12 @@ +from unittest.mock import MagicMock, patch + import pytest -from unittest.mock import patch, MagicMock -from crewai import Agent, Task, Crew, Process + +from crewai import Agent, Crew, Process, Task from crewai.knowledge.source.string_knowledge_source import StringKnowledgeSource from crewai.knowledge.storage.knowledge_storage import KnowledgeStorage + def test_knowledge_storage_search_filtering(): """Test that KnowledgeStorage.search() correctly filters results based on distance scores.""" # Create a mock collection to simulate ChromaDB behavior