Linter fix

Author: EvilFreelancer

tests/conftest.py

@@ -1,9 +1,8 @@
 """Pytest configuration and fixtures for tests."""
 
-from unittest.mock import Mock
 from typing import Type
+from unittest.mock import Mock
 
-import httpx
 import pytest
 from openai import AsyncOpenAI
 
@@ -102,4 +101,3 @@ def test_execution_config():
         max_clarifications=3,
         max_searches=4,
     )
-

tests/test_agent_factory.py

@@ -1,6 +1,7 @@
 """Tests for agent factory and configuration-based agent creation.
 
-This module contains tests for AgentFactory and dynamic agent instantiation.
+This module contains tests for AgentFactory and dynamic agent
+instantiation.
 """
 
 from unittest.mock import Mock, patch
@@ -275,6 +276,7 @@ async def test_empty_task_creation(self):
     @pytest.mark.asyncio
     async def test_agent_creation_with_toolkit(self):
         """Test creating agent with custom toolkit."""
+
         class CustomTool(BaseTool):
             tool_name = "custom_tool"
             description = "A custom test tool"
@@ -402,6 +404,7 @@ async def test_create_agent_with_string_tool(self):
     @pytest.mark.asyncio
     async def test_create_agent_with_mixed_tools(self):
         """Test creating agent with both class and string tool names."""
+
         class CustomTool(BaseTool):
             tool_name = "custom_tool"
             description = "A custom test tool"
@@ -511,6 +514,7 @@ class TestAgentFactoryMCPIntegration:
     @pytest.mark.asyncio
     async def test_create_agent_with_mcp_tools(self):
         """Test creating agent with MCP tools."""
+
         class MockMCPTool(BaseTool):
             tool_name = "mcp_tool"
             description = "Mock MCP tool"
@@ -545,6 +549,7 @@ class MockMCPTool(BaseTool):
     @pytest.mark.asyncio
     async def test_create_agent_with_mcp_and_regular_tools(self):
         """Test creating agent with both MCP and regular tools."""
+
         class MockMCPTool1(BaseTool):
             tool_name = "mcp_tool_1"
             description = "Mock MCP tool 1"
@@ -613,4 +618,3 @@ def test_get_definitions_list_empty(self):
 
             assert len(definitions) == 0
             assert definitions == []
-

tests/test_api_endpoints.py

@@ -133,7 +133,7 @@ async def mock_execute():
         mock_agent_def = Mock()
         mock_factory.get_definitions_list.return_value = [mock_agent_def]
         mock_agent_def.name = "sgr_agent"
-        
+
         # Make create method async
         mock_factory.create = AsyncMock(return_value=mock_agent)
 

tests/test_base_agent.py

@@ -36,7 +36,8 @@ def test_initialization_basic(self):
         assert agent.max_clarifications == 3
 
     def test_initialization_with_custom_limits(self):
-        """Test initialization with custom iteration and clarification limits."""
+        """Test initialization with custom iteration and clarification
+        limits."""
         from sgr_deep_research.core.agent_definition import ExecutionConfig
 
         agent = create_test_agent(
@@ -72,6 +73,7 @@ def test_toolkit_initialization_default(self):
 
     def test_toolkit_initialization_with_custom_tools(self):
         """Test adding custom tools to toolkit."""
+
         class CustomTool(BaseTool):
             pass
 
@@ -139,7 +141,8 @@ async def test_provide_clarification_increments_counter(self):
 
     @pytest.mark.asyncio
     async def test_provide_clarification_sets_event(self):
-        """Test that providing clarification sets the clarification_received event."""
+        """Test that providing clarification sets the clarification_received
+        event."""
         agent = create_test_agent(BaseAgent, task="Test")
         agent._context.clarification_received.clear()
 

tests/test_base_tool.py

@@ -1,7 +1,7 @@
 """Tests for BaseTool base class.
 
-This module contains tests for the BaseTool base class,
-covering initialization, subclassing, and tool_name generation.
+This module contains tests for the BaseTool base class, covering
+initialization, subclassing, and tool_name generation.
 """
 
 from pydantic import BaseModel
@@ -31,20 +31,23 @@ def test_base_tool_default_description_is_none(self):
 
     def test_subclass_auto_generates_tool_name(self):
         """Test that subclass auto-generates tool_name from class name."""
+
         class MyCustomTool(BaseTool):
             pass
 
         assert MyCustomTool.tool_name == "mycustomtool"
 
     def test_subclass_preserves_custom_tool_name(self):
         """Test that subclass can override tool_name."""
+
         class MyCustomTool(BaseTool):
             tool_name = "custom_name"
 
         assert MyCustomTool.tool_name == "custom_name"
 
     def test_subclass_can_have_description(self):
         """Test that subclass can have description."""
+
         class MyCustomTool(BaseTool):
             description = "Custom tool description"
 

tests/test_prompts.py

@@ -66,8 +66,8 @@ def test_load_prompt_file_io_error(self):
             assert "Permission denied" in str(exc_info.value)
 
     def test_load_prompt_file_strips_whitespace(self):
-        """Test that loaded content is not automatically stripped (PromptsConfig
-        doesn't strip, but we can test the raw loading)."""
+        """Test that loaded content is not automatically stripped
+        (PromptsConfig doesn't strip, but we can test the raw loading)."""
         with tempfile.TemporaryDirectory() as tmpdir:
             test_file = os.path.join(tmpdir, "whitespace.txt")
             test_content = "  \n  Content with spaces  \n  "
@@ -280,7 +280,8 @@ def test_get_clarification_template_date_format(self):
             assert len(date_part) == 19  # YYYY-MM-DD HH:MM:SS
 
     def test_load_prompt_file_falls_back_to_lib_dir(self):
-        """Test that PromptsConfig can load files from default library directory."""
+        """Test that PromptsConfig can load files from default library
+        directory."""
         # This tests that PromptsConfig can load files using default paths
         # In real scenario, it should find files in the installed package
         try: