feat(agent): integrate Pi V2 with recovery, charts and object tools

.github/workflows/ci.yml

@@ -48,6 +48,9 @@ jobs:
       - name: Run SQL clipboard test
         run: yarn run test:sql-in-clipboard
 
+      - name: Run agent chat tests
+        run: yarn run test:agent-chat
+
       - name: Run result Markdown formatter test
         run: yarn run test:result-markdown
 

AGENTS.md

@@ -64,6 +64,9 @@ Instruction authority and factual evidence are separate:
   JCEF, SPI, plugin, tools, and startup modules.
 - `chat2db-community-server/chat2db-community-start/`: executable backend,
   configuration, assembled frontend resources, and final jar.
+- `chat2db-community-server/chat2db-community-agent/`: agent runtime adapters,
+  installation, process/RPC transport, and runtime resources. It depends only
+  on tools and third-party libraries; start assembles it with business services.
 - `chat2db-community-server/chat2db-community-plugins/`: database-specific
   implementations. Keep dialect-specific behavior inside the owning plugin.
 - `docker/`: Community image, Compose file, and local image build entry point.

chat2db-community-client/package.json

@@ -12,7 +12,7 @@
     "build": "yarn run build:web:community",
     "build:prod": "yarn run build:web:community",
     "build:web": "umi build",
-    "prebuild:web:community": "yarn test:ipc-log-redaction && yarn test:community-boundary && yarn test:database-capabilities && yarn test:table-editor && yarn test:retired-ai && yarn test:tree-title-highlight && yarn test:tree-search-lifecycle && yarn test:driver-upload && yarn test:tree-loading && yarn test:tree-node-lookup && yarn test:tree-node-path && yarn test:data-source-authorization && yarn test:active-transactions && yarn test:data-source-mutation-refresh && yarn test:ai-model-config && yarn test:ai-model-select && yarn test:export-connections && yarn test:main-page-navigation && yarn test:shortcut && yarn test:console-tab-name && yarn test:workspace-tab-scroll && yarn test:workspace-split-lifecycle && yarn test:sql-completion-context && yarn test:file-manager-label && yarn test:local-file-encoding && yarn test:local-file-save && yarn test:local-file-tree-path && yarn test:saved-console-dirty && yarn test:editor-close && yarn test:result-set-ui && yarn test:result-pagination && yarn test:sql-execution-stream && yarn test:result-table-layout && yarn test:result-status && yarn test:data-source-identity && yarn test:data-source-execution-snapshot && yarn test:data-source-watermark && yarn test:hot-update && yarn test:update-check-schedule && yarn test:task-center && yarn test:import-preview && yarn test:application-exit && yarn test:monaco-lifecycle && yarn test:result-set-editor && yarn test:account-grants-request && yarn test:connection-close-request && yarn test:ssh-connection && yarn test:operation-log-filters && yarn test:dashboard-store",
+    "prebuild:web:community": "yarn test:ipc-log-redaction && yarn test:community-boundary && yarn test:database-capabilities && yarn test:table-editor && yarn test:retired-ai && yarn test:tree-title-highlight && yarn test:tree-search-lifecycle && yarn test:driver-upload && yarn test:tree-loading && yarn test:tree-node-lookup && yarn test:tree-node-path && yarn test:data-source-authorization && yarn test:active-transactions && yarn test:data-source-mutation-refresh && yarn test:ai-model-config && yarn test:ai-model-select && yarn test:export-connections && yarn test:main-page-navigation && yarn test:shortcut && yarn test:console-tab-name && yarn test:workspace-tab-scroll && yarn test:workspace-split-lifecycle && yarn test:sql-completion-context && yarn test:file-manager-label && yarn test:local-file-encoding && yarn test:local-file-save && yarn test:local-file-tree-path && yarn test:saved-console-dirty && yarn test:editor-close && yarn test:result-set-ui && yarn test:result-pagination && yarn test:sql-execution-stream && yarn test:result-table-layout && yarn test:result-status && yarn test:data-source-identity && yarn test:data-source-execution-snapshot && yarn test:data-source-watermark && yarn test:hot-update && yarn test:update-check-schedule && yarn test:task-center && yarn test:import-preview && yarn test:application-exit && yarn test:monaco-lifecycle && yarn test:result-set-editor && yarn test:account-grants-request && yarn test:connection-close-request && yarn test:ssh-connection && yarn test:operation-log-filters && yarn test:dashboard-store && yarn test:agent-chat",
     "postbuild:web:community": "node ./scripts/verify-production-bundles.cjs",
     "build:web:2java": "yarn run build:web:community && rm -rf ../chat2db-community-server/chat2db-community-start/src/main/resources/thymeleaf/* && cp -r dist/index.html ../chat2db-community-server/chat2db-community-start/src/main/resources/thymeleaf/",
     "build:web:community": "cross-env UMI_ENV=community cross-env APP_NAME=chat2db-community cross-env APP_VERSION=${npm_config_app_version} cross-env PRINT_LOGS=${npm_config_print_logs} cross-env APP_PORT=${npm_config_app_port} umi build",
@@ -35,6 +35,9 @@
     "test:tree-node-path": "tsx src/blocks/NewTree/utils/treeNodePath.test.ts",
     "test:ai-model-config": "tsx src/blocks/AI/components/AIModelConfigModal/modelConfigDefaults.test.ts",
     "test:ai-model-select": "tsx src/blocks/AI/components/AIModelSelect/modelSelectOptions.test.ts",
+    "test:desktop-request": "tsx src/service/commandLine/commandLineRequest.test.ts",
+    "test:agent-chat": "cross-env NODE_ENV=test yarn run test:agent-chat:suite",
+    "test:agent-chat:suite": "yarn test:desktop-request && yarn test:pi-adapters && tsx src/blocks/AI/agentOutput.test.ts && tsx src/blocks/AI/components/AgentV2Session/useOutputPage.test.tsx && tsx src/blocks/AI/components/AIAtMetion/useActive.test.tsx && tsx src/blocks/AI/components/AgentV2Session/presentation.test.ts && tsx src/blocks/AI/components/AIChatInput/inputSuggestions.test.ts && tsx src/blocks/AI/agentContext.test.ts && tsx src/blocks/AI/components/AIChatInput/mentionSelection.test.ts && tsx src/blocks/AI/agentCharts.test.ts && tsx src/blocks/AI/components/AgentChartCard/option.test.ts && tsx src/blocks/AI/agentEvents.test.ts && tsx src/blocks/AI/agentQuestions.test.ts && tsx src/blocks/AI/agentEventStream.test.ts && tsx src/blocks/AI/chatSessionRoute.test.ts && tsx src/blocks/AI/components/PiToolSettings/model.test.ts && tsx src/utils/confirmBetaFeature.test.ts",
     "test:active-tab-locator": "tsx src/pages/main/workspace/utils/activeTabLocator.test.ts",
     "test:application-exit": "tsx src/layouts/init/applicationExitCoordinator.test.ts",
     "test:base-table-interaction": "tsx src/components/BaseTable/treeInteraction.test.ts",
@@ -109,7 +112,8 @@
     "start": "yarn run start:community:hot",
     "start:community:hot": "cross-env UMI_ENV=community cross-env APP_NAME=chat2db-community cross-env DISABLE_MFSU=true cross-env UMI_DEV_SERVER_COMPRESS=none cross-env HOST=127.0.0.1 cross-env PORT=8889 node --require ./scripts/bind-dev-server-loopback.cjs ./node_modules/umi/bin/umi.js dev --public_path=/",
     "ui": "yarn add @chat2db/ui",
-    "test:create-stream-index": "tsx src/blocks/RedisAllData/CreateStream/mapDisplayedIndex.test.ts"
+    "test:create-stream-index": "tsx src/blocks/RedisAllData/CreateStream/mapDisplayedIndex.test.ts",
+    "test:pi-adapters": "tsx src/service/pi/client.test.ts"
   },
   "dependencies": {
     "@chat2db/ui": "^1.45.3",

chat2db-community-client/scripts/i18n-source-hashes.json

@@ -17,10 +17,10 @@
       "notification.ts": "500751da2f02391622f074448eb46611fd41da2e5530d6c7d65f17a1382bd236",
       "plugin.ts": "01f15b6342cc5b29899ee4144d67b04bb4bc02c6bb10779806be322dc5e9a2bc",
       "redis.ts": "cacf9cf685bf3bf6d019f18eead0f63713d4727be741a87d8dc1d22807ebd9ad",
-      "setting.ts": "c256e93a1c6ee99eac34f0d22cea8b5c0609b4cb0960af56198e663bf3e3f93c",
+      "setting.ts": "8422043f7352260c681fac02550af8cec5524df0409ff2fdaa9434bda04957c9",
       "spaceSetting.ts": "defdf3138e5e1782f0282760bcef4ce1b8550cc80b03052fea468d97165cb980",
       "sqlEditor.ts": "5c41e5f24d5ba78909795bcc681cabb8460cca075233b665dc4940afe82ce0c9",
-      "stream.ts": "2e2fca0b1ba1ecd5a1501b767be8857cf0b72d800d69a4acf9cd62a61d5ceaf5",
+      "stream.ts": "8c870a19ec607a359ed9a0fbfe711554cbb07c9c460414c371a350fb7b0ea405",
       "workspace.ts": "f41dcc31f6fa7de500cda95ef550737c99606569077d26150239b4ce14db7ef1"
     },
     "ko-KR": {
@@ -38,10 +38,10 @@
       "notification.ts": "500751da2f02391622f074448eb46611fd41da2e5530d6c7d65f17a1382bd236",
       "plugin.ts": "01f15b6342cc5b29899ee4144d67b04bb4bc02c6bb10779806be322dc5e9a2bc",
       "redis.ts": "cacf9cf685bf3bf6d019f18eead0f63713d4727be741a87d8dc1d22807ebd9ad",
-      "setting.ts": "c256e93a1c6ee99eac34f0d22cea8b5c0609b4cb0960af56198e663bf3e3f93c",
+      "setting.ts": "8422043f7352260c681fac02550af8cec5524df0409ff2fdaa9434bda04957c9",
       "spaceSetting.ts": "defdf3138e5e1782f0282760bcef4ce1b8550cc80b03052fea468d97165cb980",
       "sqlEditor.ts": "5c41e5f24d5ba78909795bcc681cabb8460cca075233b665dc4940afe82ce0c9",
-      "stream.ts": "2e2fca0b1ba1ecd5a1501b767be8857cf0b72d800d69a4acf9cd62a61d5ceaf5",
+      "stream.ts": "8c870a19ec607a359ed9a0fbfe711554cbb07c9c460414c371a350fb7b0ea405",
       "workspace.ts": "f41dcc31f6fa7de500cda95ef550737c99606569077d26150239b4ce14db7ef1"
     }
   }

chat2db-community-client/src/blocks/AI/agentCharts.test.ts

@@ -0,0 +1,88 @@
+import assert from 'node:assert/strict';
+import { readFileSync } from 'node:fs';
+import type { AgentEvent } from '@/service/agent';
+import { agentChartDetail, isPartialChart, updateAgentCharts, usesGroupedAgentChart } from './agentCharts';
+import { buildAgentChartOption } from './components/AgentChartCard/option';
+
+const payload = {
+  id: 'chart-1', runId: 'run-1', resultId: 'result-1', chartType: 'Line', title: 'Same title',
+  xField: 'month', yField: 'value', series: [], warnings: [],
+  data: [{ month: 'Jan', value: 0.1 }, { month: 'Feb', value: null }], page: { number: 1, hasMore: false },
+};
+const event = (chart: unknown, sequence = 1): AgentEvent => ({
+  id: `event-${sequence}`, sessionId: 'session', runId: 'run-1', sequence,
+  type: 'CHART_CREATED', payload: { chart }, occurredAt: '',
+});
+const first = updateAgentCharts([], [event(payload)]);
+assert.equal(first.length, 1);
+assert.strictEqual(updateAgentCharts(first, [event(payload)]), first);
+const second = updateAgentCharts(first, [event({ ...payload, id: 'chart-2' }, 2)]);
+assert.equal(second.length, 2, 'Equal titles do not merge distinct charts');
+assert.deepEqual(agentChartDetail(first[0]).chartSchema?.data, payload.data);
+assert.equal(agentChartDetail(first[0]).chartSchema?.data[1].value, null);
+assert.equal(agentChartDetail({ ...first[0], chartType: 'Combo' }).chartSchema?.chartOptionCheckbox.includes('showLabel'), false);
+assert.equal(isPartialChart(first[0]), false);
+assert.equal(isPartialChart({ ...first[0], page: { number: 2, hasMore: false } }), true);
+assert.equal(isPartialChart({ ...first[0], page: { number: 1, hasMore: null } }), true);
+assert.deepEqual(updateAgentCharts([], [event({ ...payload, chartType: 'Bad' }), event({ ...payload, runId: 'foreign' })]), []);
+assert.equal(updateAgentCharts([], [event({ ...payload, chartType: 'Statistics', xField: undefined, page: undefined })]).length, 1);
+assert.equal(updateAgentCharts([], [event({ ...payload, data: [{ value: Number.NaN }] })]).length, 0);
+assert.equal(usesGroupedAgentChart(first[0]), false, 'Historic charts keep the existing renderer');
+assert.equal(updateAgentCharts([], [event({ ...payload, groupBy: null, stack: null })]).length, 1);
+const groupedPayload = {
+  ...payload, chartType: 'Column', groupBy: ['region'], stack: true,
+  data: [{ month: 'Jan', region: '华东', value: 10 }, { month: 'Jan', region: '华西', value: 20 }],
+};
+const restored = updateAgentCharts([], JSON.parse(JSON.stringify([event(groupedPayload)])));
+assert.equal(restored.length, 1);
+assert.deepEqual(restored[0].groupBy, ['region']);
+assert.equal(restored[0].stack, true);
+assert.equal(usesGroupedAgentChart(restored[0]), true);
+assert.deepEqual(buildAgentChartOption(restored[0], { border: '#ddd', text: '#111' })
+  .series.map((series) => series.data), [[10], [20]], 'Restored event payloads render the same groups');
+for (const invalid of [
+  { groupBy: 'region' }, { groupBy: ['region', 'region'] }, { groupBy: [''] }, { groupBy: [null] },
+  { groupBy: ['a', 'b', 'c', 'd'] }, { groupBy: ['month'] }, { groupBy: ['value'] }, { groupBy: ['missing'] },
+  { stack: 'true' }, { stack: 1 }, { chartType: 'Pie' }, { chartType: 'Statistics' },
+  { chartType: 'Line' }, { chartType: 'Scatter' },
+  { data: [...groupedPayload.data, groupedPayload.data[0]] },
+]) {
+  assert.equal(updateAgentCharts([], [event({ ...groupedPayload, ...invalid })]).length, 0,
+    `Reject invalid group/stack payload ${JSON.stringify(invalid)}`);
+}
+for (const chartType of ['Line', 'Scatter', 'AreaLine', 'Bar']) {
+  assert.equal(updateAgentCharts([], [event({ ...groupedPayload, chartType, stack: false })]).length, 1);
+}
+assert.equal(updateAgentCharts([], [event({
+  ...groupedPayload, chartType: 'Scatter', stack: false,
+  data: [...groupedPayload.data, { ...groupedPayload.data[0], value: 15 }],
+})]).length, 1, 'Scatter allows repeated X values for the same group');
+const seriesLimit = (count: number) => ({
+  ...groupedPayload,
+  data: Array.from({ length: count }, (_, index) => ({ month: 'Jan', region: index, value: 1 })),
+});
+assert.equal(updateAgentCharts([], [event(seriesLimit(32))]).length, 1);
+assert.equal(updateAgentCharts([], [event(seriesLimit(33))]).length, 0);
+const comboPayload = {
+  ...groupedPayload, chartType: 'Combo',
+  series: [
+    { field: 'value', chartType: 'Column', axisPosition: 'left' },
+    { field: 'count', chartType: 'Line', axisPosition: 'right' },
+  ],
+};
+assert.equal(updateAgentCharts([], [event({ ...comboPayload, data: seriesLimit(16).data })]).length, 1);
+assert.equal(updateAgentCharts([], [event({ ...comboPayload, data: seriesLimit(17).data })]).length, 0);
+assert.equal(updateAgentCharts([], [event({
+  ...comboPayload, groupBy: [], data: [groupedPayload.data[0]],
+})]).length, 1);
+assert.equal(updateAgentCharts([], [event({
+  ...comboPayload, series: [comboPayload.series[0], comboPayload.series[0]],
+})]).length, 0, 'Duplicate metric fields cannot create duplicate ECharts series IDs');
+assert.equal(updateAgentCharts([], [event({
+  ...comboPayload, series: [{ field: 'value', chartType: 'Line', axisPosition: 'left' }],
+})]).length, 0, 'Stacked Combo must contain a column or area series');
+for (const locale of ['zh-CN', 'en-US', 'es-ES', 'ja-JP', 'ko-KR']) {
+  const source = readFileSync(new URL(`../../i18n/${locale}/stream.ts`, import.meta.url), 'utf8');
+  for (const key of ['partialResult', 'viewQueryData', 'queryData']) assert.ok(source.includes(`stream.chart.${key}`));
+}
+console.log('Agent chart restoration, group/stack schema, identity, series limits and locale coverage passed.');