fix: fix the lint check errors of the main branch (#403)

tests/unit/config/test_configuration.py

@@ -1,13 +1,8 @@
 # Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
 # SPDX-License-Identifier: MIT
 
-import os
-import pytest
 import sys
 import types
-from pathlib import Path
-import builtins
-import importlib
 from src.config.configuration import Configuration
 
 # Patch sys.path so relative import works

tests/unit/config/test_loader.py

@@ -3,8 +3,6 @@
 
 import os
 import tempfile
-import yaml
-import pytest
 from src.config.loader import load_yaml_config, process_dict, replace_env_vars
 
 

tests/unit/crawler/test_article.py

@@ -1,6 +1,5 @@
 # Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
 # SPDX-License-Identifier: MIT
-import pytest
 from src.crawler.article import Article
 
 

tests/unit/crawler/test_crawler_class.py

@@ -1,9 +1,7 @@
 # Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
 # SPDX-License-Identifier: MIT
 
-import pytest
 import src.crawler as crawler_module
-from src.crawler import Crawler
 
 
 def test_crawler_sets_article_url(monkeypatch):

tests/unit/llms/test_llm.py

@@ -1,8 +1,6 @@
 # Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
 # SPDX-License-Identifier: MIT
 
-import os
-import types
 import pytest
 from src.llms import llm
 

tests/unit/prompt_enhancer/graph/test_builder.py

@@ -6,7 +6,6 @@ from unittest.mock import patch, MagicMock
 
 from src.prompt_enhancer.graph.builder import build_graph
 from src.prompt_enhancer.graph.state import PromptEnhancerState
-from src.config.report_style import ReportStyle
 
 
 class TestBuildGraph:
@@ -48,7 +47,7 @@ class TestBuildGraph:
         mock_state_graph.return_value = mock_builder
         mock_builder.compile.return_value = mock_compiled_graph
 
-        result = build_graph()
+        build_graph()
 
         # Verify the correct node function was added
         mock_builder.add_node.assert_called_once_with("enhancer", mock_enhancer_node)

tests/unit/prompt_enhancer/graph/test_state.py

@@ -1,7 +1,6 @@
 # Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
 # SPDX-License-Identifier: MIT
 
-import pytest
 from src.prompt_enhancer.graph.state import PromptEnhancerState
 from src.config.report_style import ReportStyle
 

tests/unit/rag/test_ragflow.py

@@ -1,9 +1,7 @@
 # Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
 # SPDX-License-Identifier: MIT
 
-import os
 import pytest
-import requests
 from unittest.mock import patch, MagicMock
 from src.rag.ragflow import RAGFlowProvider, parse_uri
 
@@ -144,30 +142,6 @@ def test_list_resources_success(mock_get, monkeypatch):
     assert resources[1].description == "desc2"
 
 
-@patch("src.rag.ragflow.requests.get")
-def test_list_resources_success(mock_get, monkeypatch):
-    monkeypatch.setenv("RAGFLOW_API_URL", "http://api")
-    monkeypatch.setenv("RAGFLOW_API_KEY", "key")
-    provider = RAGFlowProvider()
-    mock_response = MagicMock()
-    mock_response.status_code = 200
-    mock_response.json.return_value = {
-        "data": [
-            {"id": "123", "name": "Dataset1", "description": "desc1"},
-            {"id": "456", "name": "Dataset2", "description": "desc2"},
-        ]
-    }
-    mock_get.return_value = mock_response
-    resources = provider.list_resources()
-    assert len(resources) == 2
-    assert resources[0].uri == "rag://dataset/123"
-    assert resources[0].title == "Dataset1"
-    assert resources[0].description == "desc1"
-    assert resources[1].uri == "rag://dataset/456"
-    assert resources[1].title == "Dataset2"
-    assert resources[1].description == "desc2"
-
-
 @patch("src.rag.ragflow.requests.get")
 def test_list_resources_error(mock_get, monkeypatch):
     monkeypatch.setenv("RAGFLOW_API_URL", "http://api")

tests/unit/rag/test_vikingdb_knowledge_base.py

@@ -173,8 +173,8 @@ class TestVikingDBKnowledgeBaseProviderPrepareRequest:
         """Test basic request preparation"""
         with (
             patch("src.rag.vikingdb_knowledge_base.Request") as mock_request,
-            patch("src.rag.vikingdb_knowledge_base.Credentials") as mock_credentials,
-            patch("src.rag.vikingdb_knowledge_base.SignerV4.sign") as mock_sign,
+            patch("src.rag.vikingdb_knowledge_base.Credentials") as _mock_credentials,
+            patch("src.rag.vikingdb_knowledge_base.SignerV4.sign") as _mock_sign,
         ):
 
             mock_req_instance = MagicMock()

tests/unit/server/test_app.py

@@ -2,31 +2,17 @@
 # SPDX-License-Identifier: MIT
 
 import base64
-import json
 import os
-from unittest.mock import AsyncMock, MagicMock, patch, mock_open
-from uuid import uuid4
-from fastapi.responses import JSONResponse, StreamingResponse
+from unittest.mock import MagicMock, patch, mock_open
 import pytest
 from fastapi.testclient import TestClient
-from fastapi import HTTPException, logger
+from fastapi import HTTPException
 from src.server.app import app, _make_event, _astream_workflow_generator
-from src.server.mcp_request import MCPServerMetadataRequest
-from src.server.rag_request import RAGResourceRequest
 from src.config.report_style import ReportStyle
 from langgraph.types import Command
 from langchain_core.messages import ToolMessage
 from langchain_core.messages import AIMessageChunk
 
-from src.server.chat_request import (
-    ChatRequest,
-    TTSRequest,
-    GeneratePodcastRequest,
-    GeneratePPTRequest,
-    GenerateProseRequest,
-    EnhancePromptRequest,
-)
-
 
 @pytest.fixture
 def client():

tests/unit/server/test_chat_request.py

@@ -1,7 +1,6 @@
 # Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
 # SPDX-License-Identifier: MIT
 
-import asyncio  # Ensure asyncio is imported
 import pytest
 from pydantic import ValidationError
 from src.config.report_style import ReportStyle

tests/unit/tools/test_crawl.py

@@ -1,4 +1,3 @@
-import pytest
 from unittest.mock import Mock, patch
 from src.tools.crawl import crawl_tool
 

tests/unit/tools/test_decorators.py

@@ -1,10 +1,8 @@
 # Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
 # SPDX-License-Identifier: MIT
 
-import pytest
-import logging
-from unittest.mock import Mock, call, patch, MagicMock
-from src.tools.decorators import LoggedToolMixin, create_logged_tool
+from unittest.mock import Mock, call, patch
+from src.tools.decorators import create_logged_tool
 
 
 class MockBaseTool:
@@ -58,7 +56,7 @@ class TestLoggedToolMixin:
         tool = LoggedTool()
 
         with patch("src.tools.decorators.logger.debug") as mock_debug:
-            result = tool._run("test_arg")
+            tool._run("test_arg")
 
             # Verify debug log was called with correct message
             mock_debug.assert_has_calls(

tests/unit/tools/test_python_repl.py

@@ -2,7 +2,7 @@
 # SPDX-License-Identifier: MIT
 
 import pytest
-from unittest.mock import Mock, patch, MagicMock
+from unittest.mock import patch
 from src.tools.python_repl import python_repl_tool
 
 

tests/unit/tools/test_search.py

@@ -3,7 +3,7 @@
 
 import os
 import pytest
-from unittest.mock import patch, MagicMock
+from unittest.mock import patch
 from src.tools.search import get_web_search_tool
 from src.config import SearchEngine
 

tests/unit/tools/test_tavily_search_api_wrapper.py

@@ -3,7 +3,6 @@
 import json
 import pytest
 from unittest.mock import Mock, patch, AsyncMock, MagicMock
-import aiohttp
 import requests
 from src.tools.tavily_search.tavily_search_api_wrapper import (
     EnhancedTavilySearchAPIWrapper,

tests/unit/tools/test_tavily_search_results_with_images.py

@@ -4,7 +4,6 @@
 import json
 import pytest
 from unittest.mock import Mock, patch, AsyncMock
-from typing import Dict, Any
 from src.tools.tavily_search.tavily_search_results_with_images import (
     TavilySearchResultsWithImages,
 )

tests/unit/tools/test_tools_retriever.py

@@ -1,7 +1,7 @@
 # Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
 # SPDX-License-Identifier: MIT
 
-from unittest.mock import Mock, patch, MagicMock
+from unittest.mock import Mock, patch
 from langchain_core.callbacks import (
     CallbackManagerForToolRun,
     AsyncCallbackManagerForToolRun,

tests/unit/utils/test_json_utils.py

@@ -1,9 +1,7 @@
 # Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
 # SPDX-License-Identifier: MIT
 
-import pytest
 import json
-from unittest.mock import patch
 from src.utils.json_utils import repair_json_output