Add the client-side subscriptions/listen driver (#3047)
This commit is contained in:
parent
6d2e908f2b
commit
9bdc03d54e
1 changed files with 2587 additions and 295 deletions
|
|
@ -1,28 +1,33 @@
|
|||
from mcp.server.mcpserver import Context, MCPServer
|
||||
|
||||
mcp = MCPServer("Notebook")
|
||||
mcp = MCPServer("Sprint Board")
|
||||
|
||||
NOTES = {"todo": "buy milk", "journal": "day one"}
|
||||
BOARDS = {
|
||||
"sprint": {"design": False, "build": False, "ship": False},
|
||||
"backlog": {"tidy docs": False},
|
||||
}
|
||||
|
||||
|
||||
@mcp.resource("note://{name}")
|
||||
def note(name: str) -> str:
|
||||
return NOTES[name]
|
||||
@mcp.resource("board://{name}")
|
||||
def board(name: str) -> str:
|
||||
tasks = BOARDS[name]
|
||||
return "\n".join(f"[{'x' if done else ' '}] {task}" for task, done in tasks.items())
|
||||
|
||||
|
||||
@mcp.tool()
|
||||
async def edit_note(name: str, text: str, ctx: Context) -> str:
|
||||
NOTES[name] = text
|
||||
await ctx.notify_resource_updated(f"note://{name}")
|
||||
return "saved"
|
||||
async def complete_task(board: str, task: str, ctx: Context) -> str:
|
||||
BOARDS[board][task] = True
|
||||
await ctx.notify_resource_updated(f"board://{board}")
|
||||
return f"{task}: done"
|
||||
|
||||
|
||||
def search(query: str) -> list[str]:
|
||||
return [name for name, text in NOTES.items() if query in text]
|
||||
def sprint_report() -> str:
|
||||
done = sum(done for tasks in BOARDS.values() for done in tasks.values())
|
||||
return f"{done} task(s) done"
|
||||
|
||||
|
||||
@mcp.tool()
|
||||
async def enable_search(ctx: Context) -> str:
|
||||
mcp.add_tool(search)
|
||||
async def enable_reports(ctx: Context) -> str:
|
||||
mcp.add_tool(sprint_report)
|
||||
await ctx.notify_tools_changed()
|
||||
return "search is live"
|
||||
return "reporting is live"
|
||||
|
|
|
|||
Loading…
Reference in a new issue