@lightcone-ai/daemon 0.15.55 → 0.15.57

package/mcp-servers/mysql/core.js

@@ -111,9 +111,9 @@ export function createDataSourcePoolRegistry({ createPool, logger = () => {} } =
   const pools = new Map();
 
   return {
-    getOrCreate(dataSourceId, dbConfig) {
-      const id = String(dataSourceId ?? '').trim();
-      if (!id) throw new Error('data_source_id must be a non-empty string');
+    getOrCreate(dataSource, dbConfig) {
+      const id = String(dataSource ?? '').trim();
+      if (!id) throw new Error('data_source must be a non-empty string');
       const nextFingerprint = fingerprintDbConfig(dbConfig);
       const existing = pools.get(id);
 
@@ -123,7 +123,7 @@ export function createDataSourcePoolRegistry({ createPool, logger = () => {} } =
 
       if (existing) {
         Promise.resolve(existing.pool.end())
-          .catch((error) => logger(`[mysql-mcp] failed to close previous pool for data_source_id=${id}: ${error.message}`));
+          .catch((error) => logger(`[mysql-mcp] failed to close previous pool for data_source=${id}: ${error.message}`));
       }
 
       const pool = createPool(dbConfig);
@@ -172,10 +172,10 @@ export function createCredentialBrokerClient({
   const normalizedWorkspaceId = String(workspaceId ?? '').trim();
   const normalizedBundleId = String(bundleId ?? '').trim();
 
-  return async function fetchCredentialEnvVars(dataSourceId) {
-    const normalizedDataSourceId = String(dataSourceId ?? '').trim();
-    if (!normalizedDataSourceId) {
-      throw new Error('data_source_id is required');
+  return async function fetchCredentialEnvVars(dataSource) {
+    const normalizedDataSource = String(dataSource ?? '').trim();
+    if (!normalizedDataSource) {
+      throw new Error('data_source is required');
     }
     if (!normalizedServerUrl || !normalizedMachineApiKey || !normalizedAgentId) {
       throw new Error('mysql MCP missing SERVER_URL/MACHINE_API_KEY/AGENT_ID runtime context');
@@ -190,7 +190,7 @@ export function createCredentialBrokerClient({
       body: JSON.stringify({
         agent_id: normalizedAgentId,
         workspace_id: normalizedWorkspaceId || null,
-        required_credentials: [normalizedDataSourceId],
+        required_credentials: [normalizedDataSource],
         bundle_id: normalizedBundleId || null,
       }),
     });
@@ -218,16 +218,16 @@ export function createCredentialBrokerClient({
 }
 
 export async function executeSql({
-  dataSourceId,
+  dataSource,
   sql,
   params = [],
   limit,
   fetchCredentialEnvVars,
   poolRegistry,
 } = {}) {
-  const normalizedDataSourceId = String(dataSourceId ?? '').trim();
-  if (!normalizedDataSourceId) {
-    throw new Error('data_source_id is required');
+  const normalizedDataSource = String(dataSource ?? '').trim();
+  if (!normalizedDataSource) {
+    throw new Error('data_source is required');
   }
   if (!Array.isArray(params)) {
     throw new Error('params must be an array');
@@ -240,9 +240,9 @@ export async function executeSql({
   }
 
   const prepared = prepareSql(sql, limit);
-  const credentialEnv = await fetchCredentialEnvVars(normalizedDataSourceId);
+  const credentialEnv = await fetchCredentialEnvVars(normalizedDataSource);
   const dbConfig = parseDbConfig(credentialEnv);
-  const pool = poolRegistry.getOrCreate(normalizedDataSourceId, dbConfig);
+  const pool = poolRegistry.getOrCreate(normalizedDataSource, dbConfig);
 
   const queryParams = [...params, ...prepared.limitParams];
   const [result] = await pool.query(prepared.sql, queryParams);
@@ -252,7 +252,7 @@ export async function executeSql({
       ? result.slice(0, prepared.limit)
       : result;
     return {
-      data_source_id: normalizedDataSourceId,
+      data_source: normalizedDataSource,
       row_count: rows.length,
       limit: prepared.limit,
       limit_applied: prepared.limitInjected || rows.length < result.length,
@@ -261,7 +261,7 @@ export async function executeSql({
   }
 
   return {
-    data_source_id: normalizedDataSourceId,
+    data_source: normalizedDataSource,
     affected_rows: Number(result?.affectedRows ?? 0),
     changed_rows: Number(result?.changedRows ?? 0),
     insert_id: result?.insertId ?? null,

package/mcp-servers/mysql/index.js

@@ -11,7 +11,7 @@ import {
 } from './core.js';
 
 const TOOL_SCHEMA = {
-  data_source_id: z.string().describe('数据源 ID（当前实现映射为 credential_id）'),
+  data_source: z.string().describe('工作区数据源名称，例如 sophon_cub。不要填写数据库密码或内部 ID。'),
   sql: z.string().describe('要执行的 SQL。支持参数占位符 ?'),
   params: z.array(z.any()).optional().describe('SQL 参数数组（按顺序对应 ? 占位符）'),
   limit: z.number().int().positive().optional().describe('结果行数上限。默认 200，最大 1000'),
@@ -50,12 +50,12 @@ export function createMysqlMcpServer({
 
   server.tool(
     'query',
-    '执行通用 SQL。调用时通过 data_source_id 走 credential broker 动态获取 DB 凭证并路由连接。',
+    '执行通用 SQL。调用时通过数据源名称走 credential broker 动态获取 DB 凭证并路由连接。',
     TOOL_SCHEMA,
-    async ({ data_source_id, sql, params = [], limit }) => {
+    async ({ data_source, sql, params = [], limit }) => {
       try {
         const payload = await executeSql({
-          dataSourceId: data_source_id,
+          dataSource: data_source,
           sql,
           params,
           limit,
@@ -64,7 +64,7 @@ export function createMysqlMcpServer({
         });
         return { content: [{ type: 'text', text: JSON.stringify(payload, null, 2) }] };
       } catch (error) {
-        logger(`[mysql-mcp] query failed for data_source_id=${data_source_id}: ${error.message}`);
+        logger(`[mysql-mcp] query failed for data_source=${data_source}: ${error.message}`);
         return {
           isError: true,
           content: [{ type: 'text', text: `query failed: ${error.message}` }],

package/mcp-servers/mysql/manifest.json

@@ -10,7 +10,7 @@
   "smoke_test": {
     "tool": "query",
     "arguments": {
-      "data_source_id": "smoke-test-data-source",
+      "data_source": "smoke-test-data-source",
       "sql": "SELECT 1",
       "limit": 1
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lightcone-ai/daemon",
-  "version": "0.15.55",
+  "version": "0.15.57",
   "type": "module",
   "main": "src/index.js",
   "bin": {

package/src/agent-manager.js

@@ -184,6 +184,7 @@ export class AgentManager {
       case 'agent:start':       return this._startAgent(msg, connection);
       case 'agent:stop':        return this._stopAgent(msg.agentId, msg.workspaceId, connection);
       case 'agent:deliver':     return this._deliverMessage(msg, connection);
+      case 'agent:reprobe':     return this._reprobeCapability(msg, connection);
       case 'publish:job':       return this._handlePublishJob(msg, connection);
       case 'browser:start_login': return this._startBrowserLogin(msg, connection);
       case 'browser:stop_login':  return this._stopBrowserLogin(msg);
@@ -504,6 +505,49 @@ export class AgentManager {
     });
   }
 
+  async _reprobeCapability({ agentId, workspaceId }, connection) {
+    const normalizedAgentId = String(agentId ?? '').trim();
+    if (!normalizedAgentId) {
+      console.log('[AgentManager] agent:reprobe ignored: missing agentId');
+      return;
+    }
+    const normalizedWorkspaceId = String(workspaceId ?? '').trim();
+    const workspaceDir = this._workspaceDir(normalizedAgentId, normalizedWorkspaceId);
+    const chatBridgePath = new URL('./chat-bridge.js', import.meta.url).pathname;
+
+    this._emitCapabilityProbe(connection, {
+      agentId: normalizedAgentId,
+      workspaceId: normalizedWorkspaceId,
+      capabilityId: CHAT_CAPABILITY_ID,
+      state: 'unknown',
+      reason: 'reprobe_pending',
+      details: { stage: 'reprobe' },
+    });
+
+    const result = await probeChatBridgeCapability({
+      chatBridgePath,
+      cwd: workspaceDir,
+      timeoutMs: CAPABILITY_PROBE_TIMEOUT_MS,
+      env: {
+        SERVER_URL: this.serverUrl,
+        MACHINE_API_KEY: this.machineApiKey,
+        AGENT_ID: normalizedAgentId,
+        WORKSPACE_ID: normalizedWorkspaceId,
+        WORKSPACE_DIR: workspaceDir,
+      },
+    });
+
+    this._emitCapabilityProbe(connection, {
+      agentId: normalizedAgentId,
+      workspaceId: normalizedWorkspaceId,
+      capabilityId: CHAT_CAPABILITY_ID,
+      state: result.state,
+      reason: result.reason,
+      details: { ...(result.details ?? {}), stage: 'reprobe' },
+      latencyMs: result.latencyMs,
+    });
+  }
+
   _emitCapabilityProbe(connection, {
     agentId,
     workspaceId,

package/src/chat-bridge.js

@@ -188,6 +188,7 @@ const DEFAULT_TOOL_CLASSIFICATION = {
   query_user_history: 'cacheable',
   list_accounts: 'cacheable',
   list_credentials: 'cacheable',
+  list_data_sources: 'cacheable',
   get_data_locations: 'cacheable',
   explain_concept: 'cacheable',
   get_release_notes: 'cacheable',
@@ -296,6 +297,7 @@ function inferToolForApi(method, apiPath, body) {
   if (method === 'POST' && cleanPath === '/credential-auth/request') return 'request_credential_auth';
   if (method === 'POST' && cleanPath === '/content-library/submit') return 'submit_to_library';
   if (method === 'POST' && cleanPath === '/api/data-sources') return 'register_data_source';
+  if (method === 'GET' && cleanPath === '/api/data-sources') return 'list_data_sources';
   if (method === 'POST' && cleanPath === '/orchestrate/decision') return 'write_governance_decision';
   if (method === 'POST' && cleanPath === '/orchestrate/correction') return 'write_governance_correction';
   if (method === 'GET'  && cleanPath === '/orchestrate/context') return 'get_orchestrate_context';
@@ -933,6 +935,29 @@ server.tool('list_server', 'List workspaces, agents, and humans on the server',
   return { content: [{ type: 'text', text: `Workspaces:\n${workspaces}\n\nAgents:\n${agents}\n\nHumans:\n${humans}` }] };
 });
 
+// ── list_data_sources ─────────────────────────────────────────────────────────
+server.tool('list_data_sources', 'List named data sources available in the current workspace. Returns names and status only; no internal IDs or secrets.', {
+  workspace_id: z.string().optional().describe('Target workspace id. Defaults to current workspace context if omitted.'),
+}, async ({ workspace_id }) => {
+  const targetWorkspaceId = (workspace_id ?? currentWorkspaceId ?? WORKSPACE_ID ?? '').trim();
+  if (!targetWorkspaceId) {
+    return { isError: true, content: [{ type: 'text', text: 'workspace_id is required (no current workspace context).' }] };
+  }
+  const params = new URLSearchParams({ workspace_id: targetWorkspaceId });
+  const data = await api('GET', `/api/data-sources?${params}`);
+  const sources = Array.isArray(data.data_sources) ? data.data_sources : [];
+  if (sources.length === 0) {
+    return { content: [{ type: 'text', text: 'No data sources are configured for this workspace.' }] };
+  }
+  const text = sources.map((source) => {
+    const name = String(source.name ?? '未命名数据源').trim();
+    const type = String(source.source_type ?? 'unknown').trim();
+    const status = String(source.credential_status ?? 'unknown').trim();
+    return `- ${name} (${type}) - connection ${status}`;
+  }).join('\n');
+  return { content: [{ type: 'text', text }] };
+});
+
 // ── list_tasks ────────────────────────────────────────────────────────────────
 server.tool('list_tasks', 'List tasks in a workspace', {
   workspace: z.string().describe('Target: #workspace-name'),
@@ -1457,7 +1482,7 @@ server.tool('submit_to_library',
 
 // ── register_data_source ───────────────────────────────────────────────────────
 server.tool('register_data_source',
-  'Register a workspace data source without binding credential yet. Returns a one-time secure auth URL (10-minute expiry) that should be sent to the user via IM.',
+  'Register a named workspace data source without binding credential yet. Returns a one-time secure auth URL (10-minute expiry) that should be sent to the user via IM.',
   {
     workspace_id: z.string().optional().describe('Target workspace id. Defaults to current workspace context if omitted.'),
     display_name: z.string().describe('Human-readable data source name.'),
@@ -1484,7 +1509,7 @@ server.tool('register_data_source',
         text:
           `Data source registered.\n` +
           `workspace_id=${data.workspace_id ?? targetWorkspaceId}\n` +
-          `data_source_id=${data.data_source_id}\n` +
+          `display_name=${display_name}\n` +
           `source_type=${data.source_type ?? source_type}\n` +
           `expires_at=${data.expires_at ?? 'unknown'}\n` +
           `secure_auth_url=${data.secure_auth_url}\n\n` +