Initial open-source release

2026-07-01 08:59:39 +02:00 · 2026-05-10 23:12:26 +02:00 · 2026-05-10 23:12:26 +02:00 · 1a42152e6f
commit 1a42152e6f
1199 changed files with 257054 additions and 0 deletions
--- a/packages/context/src/mcp/server.ts
+++ b/packages/context/src/mcp/server.ts
@ -0,0 +1,94 @@
+import { randomUUID } from 'node:crypto';
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { z } from 'zod';
+import type { MemoryAgentInput } from '../memory/index.js';
+import { jsonErrorToolResult, jsonToolResult, registerKloContextTools } from './context-tools.js';
+import type { KloMcpServerDeps, KloMcpServerLike, MemoryCapturePort } from './types.js';
+
+const memoryCaptureInputSchema = {
+  userMessage: z.string().min(1).describe('The user message that may contain durable knowledge.'),
+  assistantMessage: z.string().optional().describe('The assistant response that concluded the exchange.'),
+  connectionId: z.string().min(1).optional().describe('Optional connection id for semantic-layer capture.'),
+};
+
+const memoryCaptureStatusInputSchema = {
+  runId: z.string().min(1).describe('The memory capture run id returned by memory_capture.'),
+};
+
+function registerMemoryCaptureTools(deps: {
+  server: KloMcpServerLike;
+  memoryCapture: MemoryCapturePort;
+  userContext: KloMcpServerDeps['userContext'];
+}): void {
+  deps.server.registerTool(
+    'memory_capture',
+    {
+      title: 'Memory Capture',
+      description:
+        'Capture durable knowledge and semantic-layer updates from the final user/assistant exchange. Returns a run id for polling.',
+      inputSchema: memoryCaptureInputSchema,
+    },
+    async (input) => {
+      const captureInput: MemoryAgentInput = {
+        userId: deps.userContext.userId,
+        chatId: `mcp-${randomUUID()}`,
+        userMessage: String(input.userMessage),
+        assistantMessage: typeof input.assistantMessage === 'string' ? input.assistantMessage : undefined,
+        connectionId: typeof input.connectionId === 'string' ? input.connectionId : undefined,
+        sourceType: 'external_ingest',
+      };
+      const result = await deps.memoryCapture.capture(captureInput);
+      return jsonToolResult(result);
+    },
+  );
+
+  deps.server.registerTool(
+    'memory_capture_status',
+    {
+      title: 'Memory Capture Status',
+      description: 'Read the current or final status for a memory capture run.',
+      inputSchema: memoryCaptureStatusInputSchema,
+    },
+    async (input) => {
+      const runId = String(input.runId);
+      const status = await deps.memoryCapture.status(runId);
+      return status ? jsonToolResult(status) : jsonErrorToolResult(`Memory capture run "${runId}" was not found.`);
+    },
+  );
+}
+
+export function createKloMcpServer(deps: KloMcpServerDeps): KloMcpServerDeps['server'] {
+  if (deps.memoryCapture) {
+    registerMemoryCaptureTools({
+      server: deps.server,
+      memoryCapture: deps.memoryCapture,
+      userContext: deps.userContext,
+    });
+  }
+
+  if (deps.contextTools) {
+    registerKloContextTools({
+      server: deps.server,
+      ports: deps.contextTools,
+      userContext: deps.userContext,
+    });
+  }
+
+  return deps.server;
+}
+
+export function createDefaultKloMcpServer(
+  deps: Omit<KloMcpServerDeps, 'server'> & { name?: string; version?: string },
+): McpServer {
+  const server = new McpServer({
+    name: deps.name ?? 'klo',
+    version: deps.version ?? '0.0.0-private',
+  });
+  createKloMcpServer({
+    server: server as KloMcpServerLike,
+    memoryCapture: deps.memoryCapture,
+    userContext: deps.userContext,
+    contextTools: deps.contextTools,
+  });
+  return server;
+}