http.js•3.12 kB
import { createServer } from 'http';
import { StreamableHTTPServerTransport } from '@modelcontextprotocol/sdk/server/streamableHttp.js';
import { randomUUID } from 'crypto';
import { createStandaloneServer } from '../server.js';
const sessions = new Map();
/**
* Starts the HTTP transport server
* @param {HttpConfig} config - Configuration including port
*/
export function startHttpTransport(config) {
const httpServer = createServer();
httpServer.on('request', async (req, res) => {
const url = new URL(req.url, `http://${req.headers.host}`);
switch (url.pathname) {
case '/mcp':
await handleMcpRequest(req, res, config);
break;
case '/health':
handleHealthCheck(res);
break;
default:
handleNotFound(res);
}
});
const host = 'localhost';
httpServer.listen(config.port, host, () => {
console.log(`Hugging Face MCP Server listening on http://localhost:${config.port}`);
console.log('Put this in your client config:');
console.log(JSON.stringify({
"mcpServers": {
"huggingface-hub": {
"url": `http://localhost:${config.port}/mcp`
}
}
}, null, 2));
});
}
async function handleMcpRequest(req, res, config) {
const sessionId = req.headers['mcp-session-id'];
if (sessionId) {
const session = sessions.get(sessionId);
if (!session) {
res.statusCode = 404;
res.end('Session not found');
return;
}
return await session.transport.handleRequest(req, res);
}
if (req.method === 'POST') {
await createNewSession(req, res, config);
return;
}
res.statusCode = 400;
res.end('Invalid request');
}
async function createNewSession(req, res, config) {
const serverInstance = createStandaloneServer(config);
const transport = new StreamableHTTPServerTransport({
sessionIdGenerator: () => randomUUID(),
onsessioninitialized: (sessionId) => {
sessions.set(sessionId, { transport, server: serverInstance });
console.log('New Hugging Face session created:', sessionId);
}
});
transport.onclose = () => {
if (transport.sessionId) {
sessions.delete(transport.sessionId);
console.log('Hugging Face session closed:', transport.sessionId);
}
};
try {
await serverInstance.connect(transport);
await transport.handleRequest(req, res);
}
catch (error) {
console.error('Streamable HTTP connection error:', error);
res.statusCode = 500;
res.end('Internal server error');
}
}
function handleHealthCheck(res) {
res.writeHead(200, { 'Content-Type': 'application/json' });
res.end(JSON.stringify({
status: 'healthy',
timestamp: new Date().toISOString()
}));
}
function handleNotFound(res) {
res.writeHead(404, { 'Content-Type': 'text/plain' });
res.end('Not Found');
}