Fix lint

Author: amri369

examples/mcp/caching/main.py

@@ -10,19 +10,19 @@
 
 
 async def run(mcp_server: MCPServerStreamableHttp):
-    print(f"Cached tools before invoking tool_list")
+    print("Cached tools before invoking tool_list")
     print(mcp_server._tools_list)
 
-    print(f"Cached tools names after invoking list_tools")
+    print("Cached tools names after invoking list_tools")
     await mcp_server.list_tools()
     cached_tools_list = mcp_server._tools_list
     for tool in cached_tools_list:
         print(f"name: {tool.name}")
 
-    print(f"Cached prompts before invoking list_prompts")
+    print("Cached prompts before invoking list_prompts")
     print(mcp_server._prompts_list)
 
-    print(f"\nCached prompts after invoking list_prompts")
+    print("\nCached prompts after invoking list_prompts")
     await mcp_server.list_prompts()
     cached_prompts_list = mcp_server._prompts_list
     for prompt in cached_prompts_list.prompts:

src/agents/mcp/server.py

@@ -91,18 +91,19 @@ def __init__(
         """
         Args:
             cache_tools_list: Whether to cache the tools list. If `True`, the tools list will be
-            cached and only fetched from the server once. If `False`, the tools list will be
-            fetched from the server on each call to `list_tools()`. The cache can be invalidated
-            by calling `invalidate_tools_cache()`. You should set this to `True` if you know the
-            server will not change its tools list, because it can drastically improve latency
-            (by avoiding a round-trip to the server every time).
-
-            cache_prompts_list: Whether to cache the prompts list. If `True`, the prompts list will be
-            cached and only fetched from the server once. If `False`, the prompts list will be
-            fetched from the server on each call to `list_prompts()`. The cache can be invalidated
-            by calling `invalidate_prompts_cache()`. You should set this to `True` if you know the
-            server will not change its prompts list, because it can drastically improve latency
-            (by avoiding a round-trip to the server every time).
+                cached and only fetched from the server once. If `False`, the tools list will be
+                fetched from the server on each call to `list_tools()`. The cache can be invalidated
+                by calling `invalidate_tools_cache()`. You should set this to `True` if you know the
+                server will not change its tools list, because it can drastically improve latency
+                (by avoiding a round-trip to the server every time).
+
+            cache_prompts_list: Whether to cache the prompts list. If `True`, the prompts list
+                will be cached and only fetched from the server once. If `False`, the prompts
+                list will be fetched from the server on each call to `list_prompts()`.
+                The cache can be invalidated by calling `invalidate_prompts_cache()`.
+                You should set this to `True` if you know the server will not change
+                its prompts list, because it can drastically improve latency
+                (by avoiding a round-trip to the server every time).
 
             client_session_timeout_seconds: the read timeout passed to the MCP ClientSession.
             tool_filter: The tool filter to use for filtering tools.
@@ -387,11 +388,12 @@ def __init__(
                 if you know the server will not change its tools list, because it can drastically
                 improve latency (by avoiding a round-trip to the server every time).
 
-            cache_prompts_list: Whether to cache the prompts list. If `True`, the prompts list will be
-                cached and only fetched from the server once. If `False`, the prompts list will be
-                fetched from the server on each call to `list_prompts()`. The cache can be invalidated
-                by calling `invalidate_prompts_cache()`. You should set this to `True` if you know the
-                server will not change its prompts list, because it can drastically improve latency
+            cache_prompts_list: Whether to cache the prompts list. If `True`, the prompts list
+                will be cached and only fetched from the server once. If `False`, the prompts
+                list will be fetched from the server on each call to `list_prompts()`.
+                The cache can be invalidated by calling `invalidate_prompts_cache()`.
+                You should set this to `True` if you know the server will not change
+                its prompts list, because it can drastically improve latency
                 (by avoiding a round-trip to the server every time).
 
             name: A readable name for the server. If not provided, we'll create one from the
@@ -480,11 +482,12 @@ def __init__(
                 if you know the server will not change its tools list, because it can drastically
                 improve latency (by avoiding a round-trip to the server every time).
 
-            cache_prompts_list: Whether to cache the prompts list. If `True`, the prompts list will be
-                cached and only fetched from the server once. If `False`, the prompts list will be
-                fetched from the server on each call to `list_prompts()`. The cache can be invalidated
-                by calling `invalidate_prompts_cache()`. You should set this to `True` if you know the
-                server will not change its prompts list, because it can drastically improve latency
+            cache_prompts_list: Whether to cache the prompts list. If `True`, the prompts list
+                will be cached and only fetched from the server once. If `False`, the prompts
+                list will be fetched from the server on each call to `list_prompts()`.
+                The cache can be invalidated by calling `invalidate_prompts_cache()`.
+                You should set this to `True` if you know the server will not change
+                its prompts list, because it can drastically improve latency
                 (by avoiding a round-trip to the server every time).
 
             name: A readable name for the server. If not provided, we'll create one from the
@@ -575,11 +578,12 @@ def __init__(
                 if you know the server will not change its tools list, because it can drastically
                 improve latency (by avoiding a round-trip to the server every time).
 
-            cache_prompts_list: Whether to cache the prompts list. If `True`, the prompts list will be
-                cached and only fetched from the server once. If `False`, the prompts list will be
-                fetched from the server on each call to `list_prompts()`. The cache can be invalidated
-                by calling `invalidate_prompts_cache()`. You should set this to `True` if you know the
-                server will not change its prompts list, because it can drastically improve latency
+            cache_prompts_list: Whether to cache the prompts list. If `True`, the prompts list
+                will be cached and only fetched from the server once. If `False`, the prompts
+                list will be fetched from the server on each call to `list_prompts()`.
+                The cache can be invalidated by calling `invalidate_prompts_cache()`.
+                You should set this to `True` if you know the server will not change
+                its prompts list, because it can drastically improve latency
                 (by avoiding a round-trip to the server every time).
 
             name: A readable name for the server. If not provided, we'll create one from the

tests/mcp/test_caching.py

@@ -69,8 +69,8 @@ async def test_server_caching_tools_works(
 async def test_server_caching_prompts_works(
     mock_list_prompts: AsyncMock, mock_initialize: AsyncMock, mock_stdio_client
 ):
-    """Test that if we turn caching on, the list of prompts is cached and not fetched from the server
-    on each call to `list_prompts()`.
+    """Test that if we turn caching on, the list of prompts is cached and not fetched
+    from the server on each call to `list_prompts()`.
     """
     server = MCPServerStdio(
         params={
@@ -98,15 +98,18 @@ async def test_server_caching_prompts_works(
         result_prompts = await server.list_prompts()
         assert result_prompts == prompts
 
-        assert mock_list_prompts.call_count == 1, "list_prompts() should not have been called again"
+        assert mock_list_prompts.call_count == 1, ("list_prompts() "
+                                                   "should not have been called again")
 
         # Invalidate the cache and call list_prompts() again
         server.invalidate_prompts_cache()
         result_prompts = await server.list_prompts()
         assert result_prompts == prompts
 
-        assert mock_list_prompts.call_count == 2, "list_prompts() should be called again"
+        assert mock_list_prompts.call_count == 2, ("list_prompts() "
+                                                   "should be called again")
 
-        # Without invalidating the cache, calling list_prompts() again should return the cached value
+        # Without invalidating the cache, calling list_prompts()
+        # again should return the cached value
         result_prompts = await server.list_prompts()
         assert result_prompts == prompts