Skip to content

Commit e873eed

Browse files
committed
add an include or ignore list
1 parent 5130d77 commit e873eed

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

src/planet_mcp/server.py

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,10 +25,19 @@ def __init__(
2525
mcp: FastMCP,
2626
session: planet.Session,
2727
ignore: set[str] | None = None,
28+
include: set[str] | None = None,
2829
):
2930
self.mcp = mcp
3031
self.session = session
3132

33+
if ignore and include:
34+
raise ValueError("Cannot specify both ignore and include sets.")
35+
36+
self.include = include
37+
self.ignore = ignore
38+
39+
if self.ignore is None and self.include is None:
40+
self.ignore = DEFAULT_IGNORE
3241
self.ignore = ignore if ignore is not None else DEFAULT_IGNORE
3342

3443
self.make_tools(planet.FeaturesClient, "features")
@@ -46,7 +55,9 @@ def make_tools(self, client_class, prefix: str):
4655
if inspect.ismethod(func) and name[0] != "_":
4756
full_name = f"{prefix}_{name}"
4857

49-
if full_name in self.ignore:
58+
if self.ignore is not None and full_name in self.ignore:
59+
continue
60+
if self.include is not None and name not in self.include:
5061
continue
5162

5263
# extended tool options

0 commit comments

Comments
 (0)