From c5cf8480fcd0a6fb181d004e22bd0e4def014b6f Mon Sep 17 00:00:00 2001
From: donteatfriedrice <chs@bupt.cn>
Date: Mon, 29 Jul 2024 09:42:29 +0000
Subject: [PATCH] feat: add event tracker for ai chat block (#7637)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

[BS-940](https://linear.app/affine-design/issue/BS-940/ai-chat-block-相关埋点)

related: https://github.com/toeverything/blocksuite/pull/7733
---
 .../blocksuite/presets/ai/_common/chat-actions-handle.ts | 9 +++++++++
 .../presets/ai/peek-view/chat-block-peek-view.ts         | 9 +++++++++
 2 files changed, 18 insertions(+)

diff --git a/packages/frontend/core/src/blocksuite/presets/ai/_common/chat-actions-handle.ts b/packages/frontend/core/src/blocksuite/presets/ai/_common/chat-actions-handle.ts
index 28c1f869a1..8da0b3a21a 100644
--- a/packages/frontend/core/src/blocksuite/presets/ai/_common/chat-actions-handle.ts
+++ b/packages/frontend/core/src/blocksuite/presets/ai/_common/chat-actions-handle.ts
@@ -337,6 +337,15 @@ const SAVE_CHAT_TO_BLOCK_ACTION: ChatAction = {
       if (!blockId) {
         return false;
       }
+
+      rootService.telemetryService?.track('CanvasElementAdded', {
+        control: 'manually save',
+        page: 'whiteboard editor',
+        module: 'ai chat panel',
+        segment: 'right sidebar',
+        type: 'chat block',
+        category: 'root',
+      });
       return true;
     } catch (err) {
       console.error(err);
diff --git a/packages/frontend/core/src/blocksuite/presets/ai/peek-view/chat-block-peek-view.ts b/packages/frontend/core/src/blocksuite/presets/ai/peek-view/chat-block-peek-view.ts
index 4fc06e14f9..ecd4713356 100644
--- a/packages/frontend/core/src/blocksuite/presets/ai/peek-view/chat-block-peek-view.ts
+++ b/packages/frontend/core/src/blocksuite/presets/ai/peek-view/chat-block-peek-view.ts
@@ -204,6 +204,15 @@ export class AIChatBlockPeekView extends LitElement {
       source: { id: this.parentChatBlockId },
       target: { id: aiChatBlockId },
     });
+
+    edgelessService.telemetryService?.track('CanvasElementAdded', {
+      control: 'conversation',
+      page: 'whiteboard editor',
+      module: 'canvas',
+      segment: 'whiteboard',
+      type: 'chat block',
+      category: 'branch',
+    });
   };
 
   /**