chore: updating formatting + linting (#325)

Just getting to know the repo, drive by improvements

Author: sydney-runkle

langchain_mcp_adapters/sessions.py

@@ -339,7 +339,7 @@ async def _create_websocket_session(
         ImportError: If websockets package is not installed.
     """
     try:
-        from mcp.client.websocket import websocket_client
+        from mcp.client.websocket import websocket_client  # noqa: PLC0415
     except ImportError:
         msg = (
             "Could not import websocket_client. "

langchain_mcp_adapters/tools.py

@@ -137,12 +137,13 @@ async def call_tool(
             call_tool_result = await session.call_tool(tool.name, arguments)
 
         if call_tool_result is None:
-            raise RuntimeError(
+            msg = (
                 "Tool call failed: no result returned from the underlying MCP SDK. "
                 "This may indicate that an exception was handled or suppressed "
                 "by the MCP SDK (e.g., client disconnection, network issue, "
                 "or other execution error)."
             )
+            raise RuntimeError(msg)
 
         return _convert_call_tool_result(call_tool_result)
 

tests/conftest.py

@@ -4,6 +4,7 @@
 from collections.abc import Generator
 
 import pytest
+import pytest_socket
 
 from tests.utils import run_server
 
@@ -55,8 +56,6 @@ def websocket_server(websocket_server_port: int) -> Generator[None, None, None]:
 def socket_enabled():
     """Temporarily enable socket connections for websocket tests."""
     try:
-        import pytest_socket
-
         pytest_socket.enable_socket()
         previous_state = pytest_socket.socket_allow_hosts()
         # Only allow connections to localhost

tests/test_import.py

@@ -1,3 +1,8 @@
 def test_import() -> None:
     """Test that the code can be imported"""
-    from langchain_mcp_adapters import client, prompts, resources, tools  # noqa: F401
+    from langchain_mcp_adapters import (  # noqa: F401, PLC0415
+        client,
+        prompts,
+        resources,
+        tools,
+    )

tests/test_tools.py

@@ -1,10 +1,12 @@
 from typing import Annotated
 from unittest.mock import AsyncMock, MagicMock
 
+import httpx
 import pytest
 from langchain_core.callbacks import CallbackManagerForToolRun
 from langchain_core.messages import ToolMessage
 from langchain_core.tools import BaseTool, InjectedToolArg, ToolException, tool
+from mcp.server import FastMCP
 from mcp.types import (
     CallToolResult,
     EmbeddedResource,
@@ -223,9 +225,6 @@ async def mock_call_tool(tool_name, arguments):
 
 
 def _create_annotations_server():
-    from mcp.server import FastMCP
-    from mcp.types import ToolAnnotations
-
     server = FastMCP(port=8181)
 
     @server.tool(
@@ -356,8 +355,6 @@ def test_convert_langchain_tool_to_fastmcp_tool_with_injection():
 
 
 def _create_status_server():
-    from mcp.server import FastMCP
-
     server = FastMCP(port=8182)
 
     @server.tool()
@@ -372,7 +369,6 @@ def get_status() -> str:
 @pytest.mark.asyncio
 async def test_load_mcp_tools_with_custom_httpx_client_factory(socket_enabled) -> None:
     """Test load mcp tools with custom httpx client factory."""
-    import httpx
 
     # Custom httpx client factory
     def custom_httpx_client_factory(
@@ -412,8 +408,6 @@ def custom_httpx_client_factory(
 
 
 def _create_info_server():
-    from mcp.server import FastMCP
-
     server = FastMCP(port=8183)
 
     @server.tool()
@@ -429,7 +423,6 @@ async def test_load_mcp_tools_with_custom_httpx_client_factory_sse(
     socket_enabled,
 ) -> None:
     """Test load mcp tools with custom httpx client factory using SSE transport."""
-    import httpx
 
     # Custom httpx client factory
     def custom_httpx_client_factory(