feat: 1. replace black with ruff for fomatting and sort import (#489)

2. use tavily from`langchain-tavily` rather than the older one from `langchain-community`

Co-authored-by: Willem Jiang <willem.jiang@gmail.com>

tests/unit/tools/test_crawl.py

@@ -1,9 +1,9 @@
 from unittest.mock import Mock, patch
+
 from src.tools.crawl import crawl_tool
 
 
 class TestCrawlTool:
-
     @patch("src.tools.crawl.Crawler")
     def test_crawl_tool_success(self, mock_crawler_class):
         # Arrange

tests/unit/tools/test_decorators.py

@@ -2,6 +2,7 @@
 # SPDX-License-Identifier: MIT
 
 from unittest.mock import Mock, call, patch
+
 from src.tools.decorators import create_logged_tool
 
 
@@ -13,7 +14,6 @@ class MockBaseTool:
 
 
 class TestLoggedToolMixin:
-
     def test_run_calls_log_operation(self):
         """Test that _run calls _log_operation with correct parameters."""
         # Create a logged tool instance

tests/unit/tools/test_python_repl.py

@@ -2,13 +2,14 @@
 # SPDX-License-Identifier: MIT
 
 import os
-import pytest
 from unittest.mock import patch
+
+import pytest
+
 from src.tools.python_repl import python_repl_tool
 
 
 class TestPythonReplTool:
-
     @patch.dict(os.environ, {"ENABLE_PYTHON_REPL": "true"})
     @patch("src.tools.python_repl.repl")
     @patch("src.tools.python_repl.logger")

tests/unit/tools/test_search.py

@@ -2,14 +2,15 @@
 # SPDX-License-Identifier: MIT
 
 import os
-import pytest
 from unittest.mock import patch
-from src.tools.search import get_web_search_tool
+
+import pytest
+
 from src.config import SearchEngine
+from src.tools.search import get_web_search_tool
 
 
 class TestGetWebSearchTool:
-
     @patch("src.tools.search.SELECTED_SEARCH_ENGINE", SearchEngine.TAVILY.value)
     def test_get_web_search_tool_tavily(self):
         tool = get_web_search_tool(max_search_results=5)

tests/unit/tools/test_tavily_search_api_wrapper.py

@@ -1,16 +1,17 @@
 # Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
 # SPDX-License-Identifier: MIT
 import json
+from unittest.mock import AsyncMock, MagicMock, Mock, patch
+
 import pytest
-from unittest.mock import Mock, patch, AsyncMock, MagicMock
 import requests
+
 from src.tools.tavily_search.tavily_search_api_wrapper import (
     EnhancedTavilySearchAPIWrapper,
 )
 
 
 class TestEnhancedTavilySearchAPIWrapper:
-
     @pytest.fixture
     def wrapper(self):
         with patch(

tests/unit/tools/test_tavily_search_results_with_images.py

@@ -1,18 +1,19 @@
 # Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
 # SPDX-License-Identifier: MIT
 
+from unittest.mock import AsyncMock, Mock, patch
+
 import pytest
-from unittest.mock import Mock, AsyncMock
-from src.tools.tavily_search.tavily_search_results_with_images import (
-    TavilySearchResultsWithImages,
-)
+
 from src.tools.tavily_search.tavily_search_api_wrapper import (
     EnhancedTavilySearchAPIWrapper,
 )
+from src.tools.tavily_search.tavily_search_results_with_images import (
+    TavilySearchWithImages,
+)
 
 
-class TestTavilySearchResultsWithImages:
-
+class TestTavilySearchWithImages:
     @pytest.fixture
     def mock_api_wrapper(self):
         """Create a mock API wrapper."""
@@ -21,8 +22,8 @@ class TestTavilySearchResultsWithImages:
 
     @pytest.fixture
     def search_tool(self, mock_api_wrapper):
-        """Create a TavilySearchResultsWithImages instance with mocked dependencies."""
-        tool = TavilySearchResultsWithImages(
+        """Create a TavilySearchWithImages instance with mocked dependencies."""
+        tool = TavilySearchWithImages(
             max_results=5,
             include_answer=True,
             include_raw_content=True,
@@ -64,15 +65,13 @@ class TestTavilySearchResultsWithImages:
 
     def test_init_default_values(self):
         """Test initialization with default values."""
-        tool = TavilySearchResultsWithImages()
+        tool = TavilySearchWithImages()
         assert tool.include_image_descriptions is False
         assert isinstance(tool.api_wrapper, EnhancedTavilySearchAPIWrapper)
 
     def test_init_custom_values(self):
         """Test initialization with custom values."""
-        tool = TavilySearchResultsWithImages(
-            max_results=10, include_image_descriptions=True
-        )
+        tool = TavilySearchWithImages(max_results=10, include_image_descriptions=True)
         assert tool.max_results == 10
         assert tool.include_image_descriptions is True
 

tests/unit/tools/test_tools_retriever.py

@@ -2,13 +2,15 @@
 # SPDX-License-Identifier: MIT
 
 from unittest.mock import Mock, patch
-from langchain_core.callbacks import (
-    CallbackManagerForToolRun,
-    AsyncCallbackManagerForToolRun,
-)
+
 import pytest
+from langchain_core.callbacks import (
+    AsyncCallbackManagerForToolRun,
+    CallbackManagerForToolRun,
+)
+
+from src.rag import Chunk, Document, Resource, Retriever
 from src.tools.retriever import RetrieverInput, RetrieverTool, get_retriever_tool
-from src.rag import Document, Retriever, Resource, Chunk
 
 
 def test_retriever_input_model():