@lightcone-ai/daemon 0.15.56 → 0.15.58

package/mcp-servers/mysql/core.js

@@ -4,6 +4,7 @@ export const DEFAULT_LIMIT = 200;
 export const MAX_LIMIT = 1000;
 
 const READ_QUERY_PREFIX = /^(SELECT|SHOW|DESCRIBE|DESC|EXPLAIN|WITH)\b/i;
+const AUTO_LIMIT_PREFIX = /^(SELECT|WITH)\b/i;
 const REQUIRED_DB_KEYS = ['DB_HOST', 'DB_PORT', 'DB_USER', 'DB_PASSWORD', 'DB_NAME'];
 
 function trimTrailingSemicolons(sql) {
@@ -27,6 +28,10 @@ export function isReadQuery(sql) {
   return READ_QUERY_PREFIX.test(sql);
 }
 
+export function supportsAutoLimit(sql) {
+  return AUTO_LIMIT_PREFIX.test(sql);
+}
+
 export function hasLimitClause(sql) {
   return /\bLIMIT\b/i.test(sql);
 }
@@ -36,7 +41,7 @@ export function prepareSql(sql, limit) {
   if (!normalizedSql) throw new Error('sql must be a non-empty string');
 
   const effectiveLimit = normalizeLimit(limit);
-  if (!isReadQuery(normalizedSql) || hasLimitClause(normalizedSql)) {
+  if (!supportsAutoLimit(normalizedSql) || hasLimitClause(normalizedSql)) {
     return {
       sql: normalizedSql,
       limit: effectiveLimit,
@@ -111,9 +116,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 +128,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 +177,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 +195,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 +223,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 +245,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 +257,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 +266,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 broker 会在运行时解析到绑定凭据。'),
+  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.56",
+  "version": "0.15.58",
   "type": "module",
   "main": "src/index.js",
   "bin": {

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` +