feat: add shadow LangGraph orchestrator runtime

2026-07-24 21:25:40 +08:00
parent 46ea22b342
commit 24336d0178
29 changed files with 3247 additions and 25 deletions
@@ -0,0 +1,111 @@
+import {
+  WORKFLOW_ENGINE,
+  normalizeRunSpec,
+} from './contracts.mjs';
+import { createRemoteWorkflowEngine } from './engine-registry.mjs';
+
+const MAX_INSTRUCTION_CHARACTERS = 16_000;
+
+function extractMessageText(message) {
+  if (typeof message === 'string') return message;
+  if (!message || typeof message !== 'object') return '';
+  if (typeof message.content === 'string') return message.content;
+  if (!Array.isArray(message.content)) return '';
+  return message.content
+    .filter((part) => part?.type === 'text')
+    .map((part) => String(part.text ?? ''))
+    .join('\n');
+}
+
+function safeError(error) {
+  return {
+    code: String(error?.code ?? 'WORKFLOW_SHADOW_FAILED').slice(0, 128),
+    message: String(error instanceof Error ? error.message : error).slice(0, 1000),
+  };
+}
+
+export function createWorkflowShadowObserver({
+  configService,
+  serviceToken = process.env.MEMIND_ORCHESTRATOR_SERVICE_TOKEN,
+  fetchImpl = globalThis.fetch,
+  logger = console,
+} = {}) {
+  if (!configService?.selectEngine || !configService?.getRuntimeState) {
+    throw new Error('Workflow shadow observer requires orchestrator config service');
+  }
+
+  return async function observeWorkflowRun({
+    runId,
+    requestId,
+    userId,
+    sessionId = null,
+    workflowName = 'code-run-v1',
+    userMessage,
+    taskType = null,
+  } = {}) {
+    const selection = await configService.selectEngine({
+      runId,
+      requestId,
+      userId,
+      workflowName,
+    });
+    if (selection.shadowEngine !== WORKFLOW_ENGINE.LANGGRAPH) {
+      return {
+        observed: false,
+        reason: selection.reason,
+        mode: selection.mode,
+      };
+    }
+
+    const state = await configService.getRuntimeState();
+    const engine = createRemoteWorkflowEngine({
+      id: WORKFLOW_ENGINE.LANGGRAPH,
+      baseUrl: state.config.serviceUrl,
+      serviceToken,
+      timeoutMs: state.config.requestTimeoutMs,
+      fetchImpl,
+    });
+    const instruction = extractMessageText(userMessage).slice(0, MAX_INSTRUCTION_CHARACTERS);
+    const spec = normalizeRunSpec({
+      runId,
+      requestId,
+      workflow: { name: workflowName, version: 1 },
+      subject: { userId },
+      input: {
+        instruction,
+        taskType,
+        toolMode: 'code',
+        sessionRef: sessionId
+          ? { kind: 'goose-session', id: String(sessionId) }
+          : null,
+      },
+      policy: {
+        executionMode: 'observe-only',
+        sideEffectsAllowed: false,
+      },
+      metadata: {
+        source: 'memind-agent-run',
+        configVersion: selection.configVersion,
+      },
+    });
+    try {
+      const result = await engine.start(spec);
+      return {
+        observed: true,
+        engine: WORKFLOW_ENGINE.LANGGRAPH,
+        mode: selection.mode,
+        configVersion: selection.configVersion,
+        shadowRun: result,
+      };
+    } catch (error) {
+      logger.warn('[orchestrator-shadow] observation failed:', safeError(error));
+      throw error;
+    }
+  };
+}
+
+export const workflowShadowObserverInternals = {
+  MAX_INSTRUCTION_CHARACTERS,
+  extractMessageText,
+  safeError,
+};