From 9ed2f029208364c1c15c3f475e3be05dc369f761 Mon Sep 17 00:00:00 2001
From: Chris Estreich <cestreich@gmail.com>
Date: Fri, 30 May 2025 21:07:08 -0700
Subject: [PATCH] Add mode to messages (#57)

---
 package.json                    |  2 +-
 pnpm-lock.yaml                  | 10 +++++-----
 src/actions/analytics/events.ts | 15 +++++++++++++--
 src/db/clickhouse/messages.sql  |  2 ++
 4 files changed, 21 insertions(+), 8 deletions(-)

diff --git a/package.json b/package.json
index 5cccd96128..b18afeed45 100644
--- a/package.json
+++ b/package.json
@@ -46,7 +46,7 @@
     "@radix-ui/react-tabs": "^1.1.12",
     "@radix-ui/react-toast": "^1.2.14",
     "@radix-ui/react-tooltip": "^1.2.7",
-    "@roo-code/types": "^1.23.0",
+    "@roo-code/types": "^1.24.0",
     "@sentry/nextjs": "^9.23.0",
     "@t3-oss/env-nextjs": "^0.13.6",
     "@tailwindcss/postcss": "^4.1.8",
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
index 4e831679cf..2cba7371ae 100644
--- a/pnpm-lock.yaml
+++ b/pnpm-lock.yaml
@@ -66,8 +66,8 @@ importers:
         specifier: ^1.2.7
         version: 1.2.7(@types/react@19.1.6)(react-dom@19.1.0(react@19.1.0))(react@19.1.0)
       '@roo-code/types':
-        specifier: ^1.23.0
-        version: 1.23.0
+        specifier: ^1.24.0
+        version: 1.24.0
       '@sentry/nextjs':
         specifier: ^9.23.0
         version: 9.23.0(@opentelemetry/context-async-hooks@1.30.1(@opentelemetry/api@1.9.0))(@opentelemetry/core@1.30.1(@opentelemetry/api@1.9.0))(@opentelemetry/instrumentation@0.57.2(@opentelemetry/api@1.9.0))(@opentelemetry/sdk-trace-base@1.30.1(@opentelemetry/api@1.9.0))(next@15.3.3(@babel/core@7.27.3)(@opentelemetry/api@1.9.0)(@playwright/test@1.52.0)(react-dom@19.1.0(react@19.1.0))(react@19.1.0))(react@19.1.0)(webpack@5.99.9(@swc/core@1.11.29)(esbuild@0.25.5))
@@ -2917,8 +2917,8 @@ packages:
     cpu: [x64]
     os: [win32]
 
-  '@roo-code/types@1.23.0':
-    resolution: {integrity: sha512-CFZ2UFlrNdHymUKrqB7iu7U6X6WwoDncvkrAidMcMzQU175o/GkLLgetIIVFVuYXyvRu01xUkReU0r9esQJVeg==}
+  '@roo-code/types@1.24.0':
+    resolution: {integrity: sha512-Fo8/k2EqovcDji7L674MbrpJ+Prw18jag4j2S1+xQrmRDTgmJ7nBZXjFxi8MtY/Us3AIA6sufS9Gi5/czwye6Q==}
 
   '@schummar/icu-type-parser@1.21.5':
     resolution: {integrity: sha512-bXHSaW5jRTmke9Vd0h5P7BtWZG9Znqb8gSDxZnxaGSJnGwPLDPfS+3g0BKzeWqzgZPsIVZkM7m2tbo18cm5HBw==}
@@ -11656,7 +11656,7 @@ snapshots:
   '@rollup/rollup-win32-x64-msvc@4.41.1':
     optional: true
 
-  '@roo-code/types@1.23.0': {}
+  '@roo-code/types@1.24.0': {}
 
   '@schummar/icu-type-parser@1.21.5': {}
 
diff --git a/src/actions/analytics/events.ts b/src/actions/analytics/events.ts
index 269e506e23..4e756b88a8 100644
--- a/src/actions/analytics/events.ts
+++ b/src/actions/analytics/events.ts
@@ -33,9 +33,20 @@ export const captureEvent = async ({ event, ...rest }: AnalyticsEvent) => {
   switch (event.type) {
     case TelemetryEventName.TASK_MESSAGE: {
       table = 'messages';
-      const { taskId, message } = event.properties;
+      const { taskId, mode, message } = event.properties;
       const { ts, type, ask, say, text, reasoning, partial } = message;
-      value = { ...rest, taskId, ts, type, ask, say, text, reasoning, partial };
+      value = {
+        ...rest,
+        taskId,
+        mode,
+        ts,
+        type,
+        ask,
+        say,
+        text,
+        reasoning,
+        partial,
+      };
       break;
     }
     default: {
diff --git a/src/db/clickhouse/messages.sql b/src/db/clickhouse/messages.sql
index 70d95a6cf1..fbe4343f38 100644
--- a/src/db/clickhouse/messages.sql
+++ b/src/db/clickhouse/messages.sql
@@ -16,3 +16,5 @@ CREATE TABLE default.messages
 ENGINE = ReplicatedMergeTree('/clickhouse/tables/{uuid}/{shard}', '{replica}')
 ORDER BY (id, timestamp)
 SETTINGS index_granularity = 8192;
+
+ALTER TABLE default.messages ADD COLUMN mode Nullable(String);