apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-23 20:21:03 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Add Effect stdio MCP entrypoint

Browse source
This commit is contained in:
elpresidank 2026-06-02 08:59:53 -05:00
parent 0fb10aca73
commit e311315556
5 changed files with 136 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

40
ts/packages/mcp/src/server-effect.ts
View file

@ -1,5 +1,6 @@
import {OpenAiClient, OpenAiLanguageModel} from "@effect/ai-openai";
import {BunHttpServer, BunRuntime} from "@effect/platform-bun";
import {NodeRuntime, NodeStdio} from "@effect/platform-node";
import {createTrustGraphSocket, type BaseApi, type Term as ClientTerm} from "@trustgraph/client";
import {Config, Context, Effect, Layer, Redacted} from "effect";
import * as O from "effect/Option";
@ -1725,10 +1726,7 @@ export const TrustGraphMcpHttpApiRoutes = HttpApiBuilder.layer(
const makeTrustGraphMcpHttpLayerFromConfig = (
config: TrustGraphMcpConfigShape,
) => {
const tools = McpServer.toolkit(TrustGraphMcpToolkit).pipe(
Layer.provide(TrustGraphMcpToolkitLive),
Layer.provide(makeOpenAiProviderLayerFromConfig(config)),
)
const tools = makeTrustGraphMcpToolkitLayerFromConfig(config)
return Layer.mergeAll(
TrustGraphMcpHttpApiRoutes,
@ -1744,6 +1742,27 @@ const makeTrustGraphMcpHttpLayerFromConfig = (
)
}
const makeTrustGraphMcpToolkitLayerFromConfig = (
config: TrustGraphMcpConfigShape,
) =>
McpServer.toolkit(TrustGraphMcpToolkit).pipe(
Layer.provide(TrustGraphMcpToolkitLive),
Layer.provide(makeOpenAiProviderLayerFromConfig(config)),
)
const makeTrustGraphMcpStdioLayerFromConfig = (
config: TrustGraphMcpConfigShape,
) =>
makeTrustGraphMcpToolkitLayerFromConfig(config).pipe(
Layer.provide(McpServer.layerStdio({
name: config.name,
version: config.version,
})),
Layer.provide(NodeStdio.layer),
Layer.provide(TrustGraphSocket.layer),
Layer.provide(Layer.succeed(TrustGraphMcpConfig, TrustGraphMcpConfig.of(config))),
)
export const makeTrustGraphMcpHttpServerLayer = (
options: TrustGraphMcpOptions = {},
) =>
@ -1766,6 +1785,19 @@ export const makeTrustGraphMcpHttpLayer = (
),
)
export const makeTrustGraphMcpStdioLayer = (
options: TrustGraphMcpOptions = {},
) =>
Layer.unwrap(
loadTrustGraphMcpConfig(options).pipe(
Effect.map(makeTrustGraphMcpStdioLayerFromConfig),
),
)
export const runHttp = (options: TrustGraphMcpOptions = {}): void => {
Layer.launch(makeTrustGraphMcpHttpServerLayer(options)).pipe(BunRuntime.runMain)
}
export const runStdio = (options: TrustGraphMcpOptions = {}): void => {
Layer.launch(makeTrustGraphMcpStdioLayer(options)).pipe(NodeRuntime.runMain)
}