@expandai/mcp-server 0.1.1

package/src/main.ts

@@ -0,0 +1,24 @@
+#!/usr/bin/env node
+import { McpServer } from '@effect/ai'
+import { NodeRuntime, NodeSink, NodeStream } from '@effect/platform-node'
+import { Config, Effect, Layer, Logger, LogLevel } from 'effect'
+import packageJson from '../package.json' with { type: 'json' }
+import { ExpandDocs } from './resources/ExpandDocs.js'
+import { ExpandTools } from './tools/index.js'
+
+const enableDocs = Config.withDefault(Config.boolean('EXPAND_ENABLE_DOCS'), () => false)
+
+const program = Effect.gen(function* () {
+	const docsEnabled = yield* enableDocs
+
+	const layers = docsEnabled ? Layer.mergeAll(ExpandTools, ExpandDocs) : ExpandTools
+
+	yield* McpServer.layerStdio({
+		name: 'expandai-mcp-server',
+		version: packageJson.version,
+		stdin: NodeStream.stdin,
+		stdout: NodeSink.stdout,
+	}).pipe(Layer.provide(layers), Layer.provide(Logger.minimumLogLevel(LogLevel.None)), Layer.launch)
+})
+
+program.pipe(NodeRuntime.runMain)

package/src/resources/ExpandDocs.ts

@@ -0,0 +1,14 @@
+import { McpServer } from '@effect/ai'
+import { Effect } from 'effect'
+
+export const ExpandDocs = McpServer.resource({
+	uri: 'expand://about',
+	name: 'About expand.ai',
+	description: 'Key links for the expand.ai platform',
+	content: Effect.succeed(`# expand.ai
+
+- Website: https://expand.ai
+- Quickstart: https://expand.ai/docs
+- Dashboard: https://expand.ai/dashboard
+`),
+})

package/src/tools/Fetch.ts

@@ -0,0 +1,94 @@
+import { Tool } from '@effect/ai'
+import { Effect, Schema } from 'effect'
+import { ExpandClient } from '../ExpandClient.js'
+import { PageMeta } from '../generated/ExpandClient.js'
+
+export class SnippetsFormat extends Schema.Class<SnippetsFormat>('SnippetsFormat')({
+	query: Schema.String.annotations({
+		description: 'Query to find relevant content snippets from the page',
+	}),
+	maxSnippets: Schema.optional(
+		Schema.Int.pipe(Schema.greaterThanOrEqualTo(1), Schema.lessThanOrEqualTo(50)),
+	).annotations({
+		description: 'Maximum number of snippets to return (1-50, default: 5)',
+	}),
+	targetSnippetSize: Schema.optional(
+		Schema.Int.pipe(Schema.greaterThanOrEqualTo(100), Schema.lessThanOrEqualTo(2000)),
+	).annotations({
+		description: 'Target snippet size in characters (100-2000, default: 384)',
+	}),
+}) {}
+
+export const Format = Schema.Union(
+	Schema.Literal('markdown').annotations({
+		description: 'Return as cleaned markdown content',
+	}),
+	Schema.Literal('html').annotations({
+		description: 'Return as raw HTML content',
+	}),
+	Schema.Literal('summary').annotations({
+		description: 'Return AI-generated summary of the page content',
+	}),
+	SnippetsFormat,
+).annotations({
+	description: 'Output format for the fetched content',
+})
+
+export class FetchParams extends Schema.Class<FetchParams>('FetchParams')({
+	url: Schema.String.annotations({
+		description: 'The URL to fetch content from',
+	}),
+	format: Format,
+	includeMeta: Schema.Boolean.annotations({
+		description: 'Whether to include page meta tags in the response',
+	}),
+}) {}
+
+export class FetchResult extends Schema.Class<FetchResult>('FetchResult')({
+	content: Schema.String,
+	url: Schema.String,
+	meta: Schema.optional(PageMeta),
+}) {}
+
+export const FetchTool = Tool.make('fetch', {
+	description: 'Fetch and extract content from any URL.',
+	parameters: FetchParams.fields,
+	success: FetchResult,
+})
+	.annotate(Tool.Readonly, true)
+	.annotate(Tool.Destructive, false)
+
+export class Fetch extends Effect.Service<Fetch>()('Fetch', {
+	dependencies: [ExpandClient.Default],
+	scoped: Effect.gen(function* () {
+		const client = yield* ExpandClient
+
+		const fetch = Effect.fn('Fetch.fetch')(function* ({ url, format, includeMeta }: FetchParams) {
+			const result = yield* client.fetch({
+				url,
+				select: {
+					markdown: format === 'markdown',
+					html: format === 'html',
+					summary: format === 'summary',
+					snippets: format instanceof SnippetsFormat ? { ...format } : undefined,
+					meta: includeMeta,
+				},
+			})
+
+			const content =
+				result.data.markdown ??
+				result.data.html ??
+				result.data.summary ??
+				result.data.snippets?.map((snippet) => `${snippet.text} (Score: ${snippet.score})`).join('\n') ??
+				''
+
+			return new FetchResult({
+				content,
+				url: result.data.response.url,
+				meta: result.data.meta,
+			})
+		}, Effect.orDie)
+
+		return { fetch } as const
+	}),
+}) {}

package/src/tools/index.ts

@@ -0,0 +1,33 @@
+import { McpServer, Toolkit } from '@effect/ai'
+import { NodeHttpClient } from '@effect/platform-node'
+import { Effect, Layer } from 'effect'
+import { ExpandClient } from '../ExpandClient.js'
+import { Fetch, FetchTool } from './Fetch.js'
+
+// =============================================================================
+// Toolkit (add more tools here)
+// =============================================================================
+
+const toolkit = Toolkit.make(FetchTool)
+
+// =============================================================================
+// Toolkit Layer
+// =============================================================================
+
+const ToolkitLayer = toolkit
+	.toLayer(
+		Effect.gen(function* () {
+			const fetchService = yield* Fetch
+
+			return toolkit.of({
+				fetch: fetchService.fetch,
+			})
+		}),
+	)
+	.pipe(Layer.provide(Fetch.Default), Layer.provide(ExpandClient.Default), Layer.provide(NodeHttpClient.layerUndici))
+
+// =============================================================================
+// Export
+// =============================================================================
+
+export const ExpandTools = McpServer.toolkit(toolkit).pipe(Layer.provide(ToolkitLayer))