feat: teammates feature (#1433)

* feat: coach ai teammate * fix: integration * fix: integration * fix: integration * fix: integration * fix: coach form UI * fix: success message after save coach info * fix: coach ai in model or column context * fix: doc model ai coach * fix: citations * fix: model generation * fix: config for teammates feature * fix: coach ai * fix: make citations link to frontend * fix: teammates enabled flag from backend * fix: interactions * fix: learning redirection * revert: unnecessary changes * revert: generated files * Apply suggestions from code review Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> * Update src/webview_provider/altimateWebviewProvider.ts Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> * chore: cleanup * chore: cleanup * fix: tasklabel for citations * fix: content update * fix: lib updates * fix: ts error * fix: ts error --------- Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
AltimateAI · Oct 3, 2024 · eb1344b · eb1344b
1 parent 20ca78d
commit eb1344b
Show file tree

Hide file tree

Showing 43 changed files with 47,460 additions and 28,004 deletions.
diff --git a/package.json b/package.json
@@ -97,6 +97,11 @@
             "description": "Enable dbt docs collaboration.",
             "default": true
           },
+          "dbt.enableTeammates": {
+            "type": "boolean",
+            "description": "Enable AltimateAI teammates feature.",
+            "default": false
+          },
           "dbt.enableQueryBookmarks": {
             "type": "boolean",
             "description": "Enable Query history and bookmarks feature",

diff --git a/src/altimate.ts b/src/altimate.ts
@@ -236,8 +236,10 @@ export interface DocsGenerateResponse {
   column_descriptions?: {
     column_name: string;
     column_description: string;
+    column_citations?: { id: string; content: string }[];
   }[];
   model_description?: string;
+  model_citations?: { id: string; content: string }[];
 }
 
 export interface DBTCoreIntegration {

diff --git a/src/services/docGenService.ts b/src/services/docGenService.ts
@@ -160,19 +160,19 @@ export class DocGenService {
   }
 
   private async transmitAIGeneratedModelDocs(
-    description: string,
+    response: DocsGenerateResponse,
     syncRequestId?: string,
     panel?: WebviewView | WebviewPanel,
   ) {
     if (panel) {
       const result = syncRequestId
         ? {
             command: "response",
-            args: { body: { description }, syncRequestId, status: true },
+            args: { body: response, syncRequestId, status: true },
           }
         : {
             command: "renderAIGeneratedModelDocs",
-            description,
+            response,
           };
       await panel.webview.postMessage(result);
     }
@@ -375,6 +375,7 @@ export class DocGenService {
             generatedDocsForColumn.column_descriptions.map((entry) => ({
               name: entry.column_name,
               description: entry.column_description,
+              citations: entry.column_citations,
             })),
             message.syncRequestId,
           );
@@ -471,7 +472,7 @@ export class DocGenService {
             return;
           }
           this.transmitAIGeneratedModelDocs(
-            generateDocsForModel.model_description,
+            generateDocsForModel,
             message.syncRequestId,
             panel,
           );

diff --git a/src/webview_provider/altimateWebviewProvider.ts b/src/webview_provider/altimateWebviewProvider.ts
@@ -90,6 +90,26 @@ export class AltimateWebviewProvider implements WebviewViewProvider {
         t.onEvent(d as SharedStateEventEmitterProps),
       ),
     );
+
+    workspace.onDidChangeConfiguration(
+      (e) => {
+        if (e.affectsConfiguration("dbt.enableTeammates")) {
+          const isEnabled = workspace
+            .getConfiguration("dbt")
+            .get<boolean>("enableTeammates", false);
+          const event = isEnabled ? "TeammatesEnabled" : "TeammatesDisabled";
+          this.telemetry.sendTelemetryEvent(event);
+          if (this._panel) {
+            this.sendResponseToWebview({
+              command: "teammatesUpdated",
+              data: isEnabled,
+            });
+          }
+        }
+      },
+      this,
+      this._disposables,
+    );
   }
 
   public isWebviewView(
@@ -223,6 +243,17 @@ export class AltimateWebviewProvider implements WebviewViewProvider {
 
     try {
       switch (command) {
+        case "getTeammatesStatus":
+          {
+            const isEnabled = workspace
+              .getConfiguration("dbt")
+              .get<boolean>("enableTeammates", false);
+            this.sendResponseToWebview({
+              command: "teammatesUpdated",
+              data: isEnabled,
+            });
+            break;
+          }
         case "configEnabled":
           this.handleSyncRequestFromWebview(
             syncRequestId,

diff --git a/src/webview_provider/docsEditPanel.ts b/src/webview_provider/docsEditPanel.ts
@@ -74,6 +74,7 @@ export interface DBTDocumentation {
 export interface AIColumnDescription {
   name: string;
   description: string;
+  citations?: { id: string; content: string }[];
 }
 
 export interface DocsGenPanelView extends WebviewViewProvider {

diff --git a/webview_panels/.storybook/__mocks__/crypto.ts b/webview_panels/.storybook/__mocks__/crypto.ts
@@ -0,0 +1 @@
+export {};
diff --git a/webview_panels/.storybook/__mocks__/vscode.ts b/webview_panels/.storybook/__mocks__/vscode.ts
@@ -5,14 +5,25 @@ let func = undefined;
 export const vscode = {
   postMessage: async (args: Record<string, unknown>) => {
     if (args.syncRequestId) {
-      const body = typeof func === 'function' ? (func as Function)(args) : data;
+      let response;
+      let status = true;
+      try {
+        const body =
+          typeof func === "function" ? await (func as (args: Record<string, unknown>) => unknown)(args) : data;
+        response = { body };
+      } catch (error) {
+        response = {
+          error: (error as Error).message,
+        };
+        status = false;
+      }
       setTimeout(() => {
         window.postMessage({
           command: "response",
           args: {
             syncRequestId: args.syncRequestId,
-            body,
-            status: true,
+            ...response,
+            status,
           },
         });
       }, timer);

diff --git a/webview_panels/.storybook/main.ts b/webview_panels/.storybook/main.ts
@@ -8,6 +8,7 @@ const config: StorybookConfig = {
     "@storybook/addon-essentials",
     "@storybook/addon-interactions",
   ],
+  staticDirs: ["../src/assets"],
   framework: {
     name: "@storybook/react-vite",
     options: {},
@@ -23,6 +24,10 @@ const config: StorybookConfig = {
       config.resolve.alias["@vscodeApi"] = require.resolve(
         "./__mocks__/vscode.ts"
       );
+
+      config.resolve.alias["crypto"] = require.resolve(
+        "./__mocks__/crypto.ts"
+      );
     }
     // Merge custom configuration into the default config
     return mergeConfig(config, {