wehub-resource-sync
49b9bb6724
Deploy Docs / deploy-docs (push) Failing after 1s
Conformance Tests / client-conformance (push) Failing after 3s
Conformance Tests / server-conformance (push) Failing after 1s
GitHub Actions Security Analysis / zizmor (push) Failing after 1s
CI / checks (push) Failing after 59m20s
CI / all-green (push) Waiting to run
40 lines
932 B
Python
40 lines
932 B
Python
from collections.abc import AsyncIterator
|
|
from contextlib import AsyncExitStack, asynccontextmanager
|
|
|
|
from starlette.applications import Starlette
|
|
from starlette.routing import Mount
|
|
|
|
from mcp.server import MCPServer
|
|
|
|
notes = MCPServer("Notes")
|
|
tasks = MCPServer("Tasks")
|
|
|
|
|
|
@notes.tool()
|
|
def add_note(text: str) -> str:
|
|
"""Save a note."""
|
|
return f"Saved: {text}"
|
|
|
|
|
|
@tasks.tool()
|
|
def add_task(title: str) -> str:
|
|
"""Create a task."""
|
|
return f"Created: {title}"
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: Starlette) -> AsyncIterator[None]:
|
|
async with AsyncExitStack() as stack:
|
|
await stack.enter_async_context(notes.session_manager.run())
|
|
await stack.enter_async_context(tasks.session_manager.run())
|
|
yield
|
|
|
|
|
|
app = Starlette(
|
|
routes=[
|
|
Mount("/notes", app=notes.streamable_http_app()),
|
|
Mount("/tasks", app=tasks.streamable_http_app()),
|
|
],
|
|
lifespan=lifespan,
|
|
)
|