Skip to content

Commit 2dd57a1

Browse files
committed
fix ruff checks
1 parent f425cc4 commit 2dd57a1

File tree

4 files changed

+35
-12
lines changed

4 files changed

+35
-12
lines changed

src/mcp/server/fastmcp/authorizer.py

Lines changed: 21 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,9 @@ class Authorizer:
1717
__metaclass__ = abc.ABCMeta
1818

1919
@abc.abstractmethod
20-
def permit_get_tool(self, name: str, context: Context[ServerSession, LifespanResultT, Request] | None = None) -> bool:
20+
def permit_get_tool(
21+
self, name: str, context: Context[ServerSession, LifespanResultT, Request] | None = None
22+
) -> bool:
2123
"""Check if the specified tool can be retrieved from the associated mcp server"""
2224
return False
2325

@@ -69,12 +71,16 @@ def permit_list_template(
6971
return False
7072

7173
@abc.abstractmethod
72-
def permit_get_prompt(self, name: str, context: Context[ServerSession, LifespanResultT, Request] | None = None) -> bool:
74+
def permit_get_prompt(
75+
self, name: str, context: Context[ServerSession, LifespanResultT, Request] | None = None
76+
) -> bool:
7377
"""Check if the specified prompt can be retrieved from the associated mcp server"""
7478
return False
7579

7680
@abc.abstractmethod
77-
def permit_list_prompt(self, name: str, context: Context[ServerSession, LifespanResultT, Request] | None = None) -> bool:
81+
def permit_list_prompt(
82+
self, name: str, context: Context[ServerSession, LifespanResultT, Request] | None = None
83+
) -> bool:
7884
"""Check if the specified prompt can be listed from the associated mcp server"""
7985
return False
8086

@@ -90,10 +96,14 @@ def permit_render_prompt(
9096

9197

9298
class AllowAllAuthorizer(Authorizer):
93-
def permit_get_tool(self, name: str, context: Context[ServerSession, LifespanResultT, Request] | None = None) -> bool:
99+
def permit_get_tool(
100+
self, name: str, context: Context[ServerSession, LifespanResultT, Request] | None = None
101+
) -> bool:
94102
return True
95103

96-
def permit_list_tool(self, name: str, context: Context[ServerSession, LifespanResultT, Request] | None = None) -> bool:
104+
def permit_list_tool(
105+
self, name: str, context: Context[ServerSession, LifespanResultT, Request] | None = None
106+
) -> bool:
97107
return True
98108

99109
def permit_call_tool(
@@ -124,10 +134,14 @@ def permit_list_template(
124134
) -> bool:
125135
return True
126136

127-
def permit_get_prompt(self, name: str, context: Context[ServerSession, LifespanResultT, Request] | None = None) -> bool:
137+
def permit_get_prompt(
138+
self, name: str, context: Context[ServerSession, LifespanResultT, Request] | None = None
139+
) -> bool:
128140
return True
129141

130-
def permit_list_prompt(self, name: str, context: Context[ServerSession, LifespanResultT, Request] | None = None) -> bool:
142+
def permit_list_prompt(
143+
self, name: str, context: Context[ServerSession, LifespanResultT, Request] | None = None
144+
) -> bool:
131145
return True
132146

133147
def permit_render_prompt(

src/mcp/server/fastmcp/prompts/manager.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,12 @@ def __init__(
2929
self._authorizer = authorizer
3030
self.warn_on_duplicate_prompts = warn_on_duplicate_prompts
3131

32-
def get_prompt(self, name: str, arguments: dict[str, Any] | None = None,context: Context[ServerSession, object, Request] | None = None) -> Prompt | None:
32+
def get_prompt(
33+
self,
34+
name: str,
35+
arguments: dict[str, Any] | None = None,
36+
context: Context[ServerSession, object, Request] | None = None,
37+
) -> Prompt | None:
3338
"""Get prompt by name."""
3439
if self._authorizer.permit_get_prompt(name, context):
3540
return self._prompts.get(name)

src/mcp/server/fastmcp/resources/resource_manager.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,11 +8,11 @@
88
from pydantic import AnyUrl
99
from starlette.requests import Request
1010

11-
from mcp.server.lowlevel.server import LifespanResultT
1211
from mcp.server.fastmcp.authorizer import AllowAllAuthorizer, Authorizer
1312
from mcp.server.fastmcp.resources.base import Resource
1413
from mcp.server.fastmcp.resources.templates import ResourceTemplate
1514
from mcp.server.fastmcp.utilities.logging import get_logger
15+
from mcp.server.lowlevel.server import LifespanResultT
1616
from mcp.server.session import ServerSession
1717

1818
if TYPE_CHECKING:
@@ -115,7 +115,9 @@ def list_resources(self, context: Context[ServerSession, LifespanResultT, Reques
115115
resource for uri, resource in self._resources.items() if self._authorizer.permit_list_resource(uri, context)
116116
]
117117

118-
def list_templates(self, context: Context[ServerSession, LifespanResultT, Request] | None = None) -> list[ResourceTemplate]:
118+
def list_templates(
119+
self, context: Context[ServerSession, LifespanResultT, Request] | None = None
120+
) -> list[ResourceTemplate]:
119121
"""List all registered templates."""
120122
logger.debug("Listing templates", extra={"count": len(self._templates)})
121123
return [

src/mcp/server/fastmcp/tools/tool_manager.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,11 @@
55

66
from starlette.requests import Request
77

8-
from mcp.server.lowlevel.server import LifespanResultT
98
from mcp.server.fastmcp.authorizer import AllowAllAuthorizer, Authorizer
109
from mcp.server.fastmcp.exceptions import ToolError
1110
from mcp.server.fastmcp.tools.base import Tool
1211
from mcp.server.fastmcp.utilities.logging import get_logger
12+
from mcp.server.lowlevel.server import LifespanResultT
1313
from mcp.server.session import ServerSession
1414
from mcp.types import ToolAnnotations
1515

@@ -39,7 +39,9 @@ def __init__(
3939
self.warn_on_duplicate_tools = (warn_on_duplicate_tools,)
4040
self._authorizer = authorizer
4141

42-
def get_tool(self, name: str, context: Context[ServerSession, LifespanResultT, Request] | None = None) -> Tool | None:
42+
def get_tool(
43+
self, name: str, context: Context[ServerSession, LifespanResultT, Request] | None = None
44+
) -> Tool | None:
4345
"""Get tool by name."""
4446
if self._authorizer.permit_get_tool(name, context):
4547
return self._tools.get(name)

0 commit comments

Comments
 (0)