add think tool (#254)

Author: nuwandavek

apps/src/metabase/appController.ts

@@ -842,6 +842,24 @@ export class MetabaseController extends AppController<MetabaseAppState> {
     return actionContent;
   }
 
+  @Action({
+    labelRunning: "Thinking",
+    labelDone: "Formulated a plan",
+    labelTask: "Formulated a plan",
+    description: "Formulates a plan to answer the user's request.",
+    renderBody: ({ thoughts }: { thoughts: string }) => {
+      return { text: null, code: null }
+    }
+  })
+  async Think({ thoughts }: { thoughts: string[] }) {
+    const actionContent: DefaultMessageContent = {
+      type: "DEFAULT",
+      text: `Thinking:\n${thoughts}`,
+      images: [],
+    };
+    return actionContent;
+  }
+
   // 1. Internal actions -------------------------------------------
   async checkVisualizationInvalid() {
     let returnMessage = "Visualization set successfully";

web/src/components/common/TaskUI.tsx

@@ -233,22 +233,22 @@ const TaskUI = forwardRef<HTMLTextAreaElement>((_props, ref) => {
         toastDescription = "Please select a database"
         preventRunTask = true
     }
-    else if (toolContext.pageType === 'dashboard' && !drMode) {
+    else if (toolContext.pageType === 'dashboard' && (!drMode || analystMode)) {
         toastTitle = 'Dashboard is supported only in agent mode'
         toastDescription = "You can enable agent mode in settings"
         preventRunTask = true
     }
-    else if (toolContext.pageType === 'mbql' && !drMode) {
+    else if (toolContext.pageType === 'mbql' && (!drMode || analystMode)) {
         toastTitle = 'MBQL Editor is supported only in agent mode'
         toastDescription = "You can enable agent mode in settings"
         preventRunTask = true
     }
-    else if (toolContext.pageType === 'mbql' && selectedCatalog != DEFAULT_TABLES) {
+    else if (toolContext.pageType === 'mbql' && (selectedCatalog != DEFAULT_TABLES || analystMode)) {
         toastTitle = 'MBQL Editor is supported only in Default Tables catalog'
         toastDescription = "You can switch to Default Tables catalog in settings"
         preventRunTask = true
     }
-    else if (selectedCatalog === DEFAULT_TABLES && isEmpty(validAddedTables) && isEmpty(validSelectedModels)) {
+    else if (selectedCatalog === DEFAULT_TABLES && isEmpty(validAddedTables) && isEmpty(validSelectedModels) && !analystMode) {
         toastTitle = 'No Table in Default Tables'
         toastDescription = "Please select at least one table in Default Tables catalog"
         preventRunTask = true
@@ -380,6 +380,9 @@ const TaskUI = forwardRef<HTMLTextAreaElement>((_props, ref) => {
     {
         isAppEnabled && !shouldBeEnabled && <DisabledOverlay toolEnabledReason={"You're currently using MinusX Classic, which only works on SQL Editor pages. [Find out](https://minusx.ai/demo) how to enable Agent mode and unlock all the features!"}/>
     }
+    {
+        analystMode && (toolContext.pageType != 'sql') && <DisabledOverlay toolEnabledReason={"You're currently using `[badge]Analyst Mode (alpha)`, which only works on SQL Editor pages for now!"}/>
+    }
     <VStack
       justifyContent="space-between"
       alignItems="stretch"