Max
20 lines
515 B
Python
20 lines
515 B
Python
from typing import Annotated
|
|
|
|
from mcp.server import MCPServer
|
|
from mcp.server.mcpserver import ListRoots, Resolve
|
|
from mcp.types import ListRootsResult
|
|
|
|
mcp = MCPServer("Bookshop")
|
|
|
|
|
|
def workspace_roots() -> ListRoots:
|
|
return ListRoots()
|
|
|
|
|
|
@mcp.tool()
|
|
async def catalog_folder(roots: Annotated[ListRootsResult, Resolve(workspace_roots)]) -> str:
|
|
"""Pick the folder the catalog export should go to."""
|
|
if not roots.roots:
|
|
return "No workspace folders shared."
|
|
return str(roots.roots[0].uri)
|