docs-autorag.app.ts•919 B
import { McpAgent } from 'agents/mcp'
import { createApiHandler } from '@repo/mcp-common/src/api-handler'
import { getEnv } from '@repo/mcp-common/src/env'
import { CloudflareMCPServer } from '@repo/mcp-common/src/server'
import { registerDocsTools } from './tools/docs-autorag.tools'
import type { Env } from './docs-autorag.context'
const env = getEnv<Env>()
// The docs MCP server isn't stateful, so we don't have state/props
export type Props = never
export type State = never
export class CloudflareDocumentationMCP extends McpAgent<Env, State, Props> {
	server = new CloudflareMCPServer({
		wae: env.MCP_METRICS,
		serverInfo: {
			name: env.MCP_SERVER_NAME,
			version: env.MCP_SERVER_VERSION,
		},
	})
	constructor(
		public ctx: DurableObjectState,
		public env: Env
	) {
		super(ctx, env)
	}
	async init() {
		registerDocsTools(this)
	}
}
export default createApiHandler(CloudflareDocumentationMCP)