import { McpAgent } from "agents/mcp";
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { registerAllTools } from "./tools";
// Define our MCP agent with tools
export class MyMCP extends McpAgent {
server = new McpServer({
name: "TripGo MCP Toolkit",
version: "1.0.0",
});
// Store environment variables
env: any;
// Constructor to receive environment
constructor(state: any, env: any) {
super(state, env);
this.env = env;
}
async init() {
registerAllTools(this.server, this.env);
}
}
export default {
fetch(request: Request, env: Env, ctx: ExecutionContext) {
const url = new URL(request.url);
if (url.pathname === "/sse" || url.pathname === "/sse/message") {
return MyMCP.serveSSE("/sse").fetch(request, env, ctx);
}
if (url.pathname === "/mcp") {
return MyMCP.serve("/mcp").fetch(request, env, ctx);
}
return new Response("Not found", { status: 404 });
},
};
