fuli-context 0.7.4-beta.2 → 0.7.5

package/src/mcp/runtime.js

@@ -1,16 +1,28 @@
 import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
 
+import {
+  bindRuntimeLeaseAgentTools,
+  createRuntimeLeaseClient
+} from '../adaptive-runtime/lease-client.js';
 import { openFederatedGraphApplication } from '../graphiti/federated-application.js';
 import { createMcpServer } from './create-mcp-server.js';
 
-export async function runStdio({ runtimeConfigPath }) {
+export async function runStdio({ runtimeConfigPath, sourceApplication }) {
   const app = openFederatedGraphApplication({ runtimeConfigPath });
-  const server = createServerOrClose(app);
+  const leaseClient = createRuntimeLeaseClient({ runtimeConfigPath });
+  bindRuntimeLeaseAgentTools(app, leaseClient);
+  const server = createServerOrClose(app, createMcpServer, {
+    sourceApplication,
+    withRuntimeLease: leaseClient.withGraphLease
+  });
   const transport = new StdioServerTransport();
   const signalHandlers = new Map();
   const close = createCloseOnce({
     closeServer: () => server.close(),
-    closeApplication: () => app.close(),
+    closeApplication: async () => {
+      await leaseClient.close();
+      return app.close();
+    },
     afterClose: () => removeHandlers(signalHandlers)
   });
   const quietClose = createQuietCloser(close, () => { process.exitCode = 1; });
@@ -31,9 +43,9 @@ export async function runStdio({ runtimeConfigPath }) {
   }
 }
 
-export function createServerOrClose(app, factory = createMcpServer) {
+export function createServerOrClose(app, factory = createMcpServer, options) {
   try {
-    return factory(app);
+    return factory(app, options);
   } catch (error) {
     void app.close();
     throw error;

package/src/mcp/session-id.js

@@ -2,6 +2,17 @@ import { randomUUID } from 'node:crypto';
 
 const UUID = /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i;
 const WORKFLOW_OBSERVATION_TOOL = 'record_workflow_transition_observation';
+const PROJECT_AGENT_SOURCE_TOOLS = new Set([
+  'create_project_agent_assignment',
+  'replace_project_agent_assignment',
+  'coordinate_project_agent_task',
+  'submit_project_agent_task',
+  'get_project_agent_context',
+  'record_project_agent_task_activity',
+  'preflight_executor',
+  'report_executor_health',
+  'record_project_agent_executor_actual'
+]);
 
 const SESSION_ID_TOOLS = new Set([
   'begin_task_context',
@@ -31,7 +42,8 @@ export function normalizeAgentSessionInput(
   input,
   nativeThreadId,
   hostSessionId = nativeThreadId,
-  clock = () => new Date()
+  clock = () => new Date(),
+  sourceApplication = nativeThreadId ? 'codex' : 'other'
 ) {
   if (name === WORKFLOW_OBSERVATION_TOOL) {
     if (!hostSessionId) {
@@ -51,6 +63,24 @@ export function normalizeAgentSessionInput(
       hostObservedAt: clock().toISOString()
     };
   }
+  if (PROJECT_AGENT_SOURCE_TOOLS.has(name)) {
+    if (!hostSessionId) {
+      throw new TypeError('Project Agent calls require an MCP host session');
+    }
+    const safeSourceApplication = normalizeMcpSourceApplication(sourceApplication);
+    const {
+      sourceApplication: _sourceApplication,
+      source_application: _source_application,
+      sourceSessionId: _sourceSessionId,
+      source_session_id: _source_session_id,
+      ...safeInput
+    } = input ?? {};
+    return {
+      ...safeInput,
+      sourceApplication: safeSourceApplication,
+      sourceSessionId: hostSessionId
+    };
+  }
   if (!nativeThreadId || !SESSION_ID_TOOLS.has(name)) return input;
   if (!input || typeof input !== 'object' || Array.isArray(input)) return input;
   if (typeof input.sessionId !== 'string' || input.sessionId.trim() === nativeThreadId) {
@@ -58,3 +88,10 @@ export function normalizeAgentSessionInput(
   }
   return { ...input, sessionId: nativeThreadId };
 }
+
+export function normalizeMcpSourceApplication(value) {
+  if (!['codex', 'claude_code', 'cursor', 'kiro', 'other'].includes(value)) {
+    throw new TypeError('Unsupported MCP source application');
+  }
+  return value;
+}

package/src/mcp/tool-annotations.js

@@ -7,6 +7,19 @@ const READ_TOOLS = new Set([
   'preview_common_knowledge_promotion',
   'list_knowledge_spaces',
   'list_personal_projects',
+  'list_project_agents',
+  'get_project_agent',
+  'list_project_agent_assignments',
+  'get_project_agent_context',
+  'view_project_agent_task',
+  'view_project_agent_activity',
+  'get_project_agent_recruitment_policy',
+  'list_project_agent_recruitments',
+  'list_executors',
+  'get_executor',
+  'list_executor_routing_rules',
+  'get_executor_routing_rule',
+  'list_project_agent_routing_learning',
   'list_knowledge_review_candidates',
   'list_workflow_candidates',
   'recommend_next_workflow_steps',
@@ -36,6 +49,32 @@ const WRITE_TOOLS = new Set([
   'search_human_knowledge_changes',
   'review_human_knowledge_change',
   'upsert_personal_project',
+  'upsert_project_agent',
+  'delete_project_agent',
+  'cleanup_test_project_agents',
+  'create_project_agent_assignment',
+  'end_project_agent_assignment',
+  'replace_project_agent_assignment',
+  'coordinate_project_agent_task',
+  'acquire_runtime_lease',
+  'refresh_runtime_lease',
+  'release_runtime_lease',
+  'submit_project_agent_task',
+  'record_project_agent_task_activity',
+  'update_project_agent_recruitment_policy',
+  'decide_project_agent_recruitment',
+  'upsert_executor',
+  'delete_executor',
+  'preflight_executor',
+  'authorize_executor',
+  'report_executor_health',
+  'upsert_executor_routing_rule',
+  'update_executor_routing_rule',
+  'delete_executor_routing_rule',
+  'record_project_agent_task_outcome',
+  'record_project_agent_executor_actual',
+  'ignore_project_agent_routing_learning',
+  'reset_project_agent_routing_learning',
   'start_knowledge_review',
   'record_knowledge_review_progress',
   'finish_knowledge_review',
@@ -69,12 +108,42 @@ export function annotationsFor(name) {
     readOnlyHint: readOnly,
     destructiveHint: DESTRUCTIVE_TOOLS.has(name),
     idempotentHint: readOnly || name === 'upsert_personal_project' ||
+      name === 'upsert_project_agent' ||
+      name === 'delete_project_agent' ||
+      name === 'cleanup_test_project_agents' ||
+      name === 'create_project_agent_assignment' ||
+      name === 'coordinate_project_agent_task' ||
+      name === 'refresh_runtime_lease' ||
+      name === 'release_runtime_lease' ||
+      name === 'submit_project_agent_task' ||
+      name === 'record_project_agent_task_activity' ||
+      name === 'record_project_agent_task_outcome' ||
+      name === 'record_project_agent_executor_actual' ||
+      name === 'upsert_executor' ||
+      name === 'preflight_executor' ||
+      name === 'authorize_executor' ||
+      name === 'report_executor_health' ||
+      name === 'upsert_executor_routing_rule' ||
+      name === 'update_executor_routing_rule' ||
       name === 'publish_personal_project' || name === 'unsubscribe_public_project' ||
       name === 'record_knowledge_usage' || name === 'record_decision_trace' ||
       name === 'record_knowledge_feedback' ||
       name === 'record_workflow_transition_observation' ||
       name === 'checkpoint_task_knowledge',
     openWorldHint: name === 'capture_session_knowledge' ||
+      name === 'coordinate_project_agent_task' ||
+      name === 'submit_project_agent_task' ||
+      name === 'record_project_agent_task_activity' ||
+      name === 'preflight_executor' ||
+      name === 'record_project_agent_task_outcome' ||
+      name === 'record_project_agent_executor_actual' ||
+      name === 'upsert_executor' ||
+      name === 'delete_executor' ||
+      name === 'authorize_executor' ||
+      name === 'report_executor_health' ||
+      name === 'upsert_executor_routing_rule' ||
+      name === 'update_executor_routing_rule' ||
+      name === 'delete_executor_routing_rule' ||
       name === 'record_workflow_transition_observation' ||
       name === 'search_connected_knowledge' ||
       name === 'record_decision_trace' ||

package/src/mcp-server.js

@@ -1,5 +1,9 @@
 #!/usr/bin/env node
 import { callAgentTool, listAgentTools } from './agent-tools.js';
+import {
+  bindRuntimeLeaseAgentTools,
+  createRuntimeLeaseClient
+} from './adaptive-runtime/lease-client.js';
 import { ApplicationError } from './app/application-error.js';
 import { openFederatedGraphApplication } from './graphiti/federated-application.js';
 import {
@@ -7,6 +11,7 @@ import {
   resolveGraphRuntimeOptions
 } from './graphiti/runtime-config.js';
 import { runStdio } from './mcp/runtime.js';
+import { normalizeMcpSourceApplication } from './mcp/session-id.js';
 import { applicationErrorMessage } from './mcp/tool-result.js';
 
 await main(process.argv.slice(2)).catch((error) => {
@@ -29,19 +34,30 @@ async function main(args) {
     await runCall({ args, runtimeConfigPath });
     return;
   }
-  await runStdio({ runtimeConfigPath });
+  const configuredSourceApplication = args.includes('--source-application')
+    ? requiredOption(args, '--source-application')
+    : null;
+  await runStdio({
+    runtimeConfigPath,
+    sourceApplication: configuredSourceApplication
+      ? normalizeMcpSourceApplication(configuredSourceApplication)
+      : undefined
+  });
 }
 
 async function runCall({ args, runtimeConfigPath }) {
   const app = openFederatedGraphApplication({ runtimeConfigPath });
+  const leaseClient = createRuntimeLeaseClient({ runtimeConfigPath });
+  bindRuntimeLeaseAgentTools(app, leaseClient);
   try {
-    const result = await callAgentTool(
-      app,
-      requiredOption(args, '--call'),
-      parseInput(args)
+    const toolName = requiredOption(args, '--call');
+    const result = await leaseClient.withGraphLease(
+      `cli-call:${toolName}`,
+      () => callAgentTool(app, toolName, parseInput(args))
     );
     process.stdout.write(`${JSON.stringify(result, null, 2)}\n`);
   } finally {
+    await leaseClient.close();
     await app.close();
   }
 }
@@ -69,7 +85,8 @@ function startupMessage(error) {
 }
 
 function helpText() {
-  return 'Usage: node src/mcp-server.js [--runtime-config <graph-runtime.json>]\n' +
+  return 'Usage: node src/mcp-server.js [--runtime-config <graph-runtime.json>] ' +
+    '[--source-application <codex|claude_code|cursor|kiro|other>]\n' +
     '       node src/mcp-server.js --tools\n' +
     '       node src/mcp-server.js --runtime-config <graph-runtime.json> ' +
     '--call <tool> [--input <json>]\n';