t3tra
cb421cf9ea
* refactor: remove unused imports * fix: replace NotImplemented with NotImplementedError * refactor: resolve E722 (do not use bare 'except') * refactor: remove unused variable * refactor: remove unused imports * refactor: ignore unused imports that will be used in the future * refactor: resolve W293 (blank line contains whitespace) * refactor: resolve F541 (f-string is missing placeholders) --------- Co-authored-by: afourney <adamfo@microsoft.com>
83 lines
2.3 KiB
Python
83 lines
2.3 KiB
Python
import sys
|
|
from mcp.server.fastmcp import FastMCP
|
|
from starlette.applications import Starlette
|
|
from mcp.server.sse import SseServerTransport
|
|
from starlette.requests import Request
|
|
from starlette.routing import Mount, Route
|
|
from mcp.server import Server
|
|
from markitdown import MarkItDown
|
|
import uvicorn
|
|
|
|
# Initialize FastMCP server for MarkItDown (SSE)
|
|
mcp = FastMCP("markitdown")
|
|
|
|
|
|
@mcp.tool()
|
|
async def convert_to_markdown(uri: str) -> str:
|
|
"""Convert a resource described by an http:, https:, file: or data: URI to markdown"""
|
|
return MarkItDown().convert_uri(uri).markdown
|
|
|
|
|
|
def create_starlette_app(mcp_server: Server, *, debug: bool = False) -> Starlette:
|
|
sse = SseServerTransport("/messages/")
|
|
|
|
async def handle_sse(request: Request) -> None:
|
|
async with sse.connect_sse(
|
|
request.scope,
|
|
request.receive,
|
|
request._send,
|
|
) as (read_stream, write_stream):
|
|
await mcp_server.run(
|
|
read_stream,
|
|
write_stream,
|
|
mcp_server.create_initialization_options(),
|
|
)
|
|
|
|
return Starlette(
|
|
debug=debug,
|
|
routes=[
|
|
Route("/sse", endpoint=handle_sse),
|
|
Mount("/messages/", app=sse.handle_post_message),
|
|
],
|
|
)
|
|
|
|
|
|
# Main entry point
|
|
def main():
|
|
import argparse
|
|
|
|
mcp_server = mcp._mcp_server
|
|
|
|
parser = argparse.ArgumentParser(description="Run MCP SSE-based MarkItDown server")
|
|
|
|
parser.add_argument(
|
|
"--sse",
|
|
action="store_true",
|
|
help="Run the server with SSE transport rather than STDIO (default: False)",
|
|
)
|
|
parser.add_argument(
|
|
"--host", default=None, help="Host to bind to (default: 127.0.0.1)"
|
|
)
|
|
parser.add_argument(
|
|
"--port", type=int, default=None, help="Port to listen on (default: 3001)"
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
if not args.sse and (args.host or args.port):
|
|
parser.error("Host and port arguments are only valid when using SSE transport.")
|
|
sys.exit(1)
|
|
|
|
if args.sse:
|
|
starlette_app = create_starlette_app(mcp_server, debug=True)
|
|
uvicorn.run(
|
|
starlette_app,
|
|
host=args.host if args.host else "127.0.0.1",
|
|
port=args.port if args.port else 3001,
|
|
)
|
|
else:
|
|
mcp.run()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|