index.ts•1.2 kB
#!/usr/bin/env node
import { config as loadEnv } from 'dotenv';
loadEnv();
import { loadConfig } from './config.js';
import { parseArgs } from './cli.js';
import { HuggingFaceServer } from './server.js';
import { runStdioTransport, startHttpTransport } from './transport/index.js';
/**
* Main entry point for the Hugging Face Hub MCP Server
*
* Transport selection logic:
* 1. --stdio flag forces STDIO transport
* 2. --port flag or PORT env var triggers HTTP transport
* 3. Default: STDIO for local development
*/
async function main() {
try {
const config = loadConfig();
const cliOptions = parseArgs();
const shouldUseHttp = cliOptions.port || (process.env.PORT && !cliOptions.stdio);
const port = cliOptions.port || parseInt(process.env.PORT || '3003', 10);
if (shouldUseHttp) {
await startHttpTransport({ ...config, port });
} else {
const server = new HuggingFaceServer(config);
await runStdioTransport(server.getServer());
}
} catch (error) {
console.error("Fatal error running Hugging Face Hub server:", error);
process.exit(1);
}
}
main();