Fetch data source view in agent configuration (#6708)

* Fetch data source view in agent configuration * Setup Sequelize relationship * Return both data source id and data source view id. * Refactor process action * Address naming comments from review * Refactor dust app fetching logic * Refactor table query actions fetch * Refactor websearch and browse fetch logic * ✨ * ✨
dust-tt · Aug 9, 2024 · 362b099 · 362b099
1 parent b0d41ba
commit 362b099
Show file tree

Hide file tree

Showing 16 changed files with 663 additions and 291 deletions.
diff --git a/front/components/assistant_builder/AssistantBuilderDataSourceModal.tsx b/front/components/assistant_builder/AssistantBuilderDataSourceModal.tsx
@@ -225,6 +225,8 @@ export default function AssistantBuilderDataSourceModal({
                         ...currentConfigurations,
                         [ds.name]: {
                           dataSource: ds,
+                          // TODO(GROUPS_INFRA) Replace with DataSourceViewType once the UI has it.
+                          dataSourceViewId: null,
                           selectedResources: [],
                           isSelectAll: true,
                         },

diff --git a/front/components/assistant_builder/server_side_props_helpers.ts b/front/components/assistant_builder/server_side_props_helpers.ts
@@ -73,7 +73,9 @@ export async function buildInitialActions({
             const dataSource = dataSourcesByName[ds.dataSourceName];
             if (!dataSource.connectorId || !ds.resources) {
               return {
-                dataSource: dataSource,
+                dataSource,
+                // TODO(GROUPS_INFRA) Replace with DataSourceViewType once the UI has it.
+                dataSourceViewId: null,
                 selectedResources: [],
                 isSelectAll: ds.isSelectAll,
               };
@@ -92,7 +94,9 @@ export async function buildInitialActions({
             }
 
             return {
-              dataSource: dataSource,
+              dataSource,
+              // TODO(GROUPS_INFRA) Replace with DataSourceViewType once the UI has it.
+              dataSourceViewId: null,
               selectedResources: response.value.nodes,
               isSelectAll: ds.isSelectAll,
             };

diff --git a/front/components/assistant_builder/submitAssistantBuilderForm.ts b/front/components/assistant_builder/submitAssistantBuilderForm.ts
@@ -84,6 +84,8 @@ export async function submitAssistantBuilderForm({
               a.configuration.dataSourceConfigurations
             ).map(({ dataSource, selectedResources, isSelectAll }) => ({
               dataSourceId: dataSource.name,
+              // TODO(GROUPS_INFRA) Replace with DataSourceViewType once the UI has it.
+              dataSourceViewId: null,
               workspaceId: owner.sId,
               filter: {
                 parents: !isSelectAll
@@ -150,6 +152,8 @@ export async function submitAssistantBuilderForm({
               a.configuration.dataSourceConfigurations
             ).map(({ dataSource, selectedResources, isSelectAll }) => ({
               dataSourceId: dataSource.name,
+              // TODO(GROUPS_INFRA) Replace with DataSourceViewType once the UI has it.
+              dataSourceViewId: null,
               workspaceId: owner.sId,
               filter: {
                 parents: !isSelectAll

diff --git a/front/components/assistant_builder/types.ts b/front/components/assistant_builder/types.ts
@@ -38,6 +38,8 @@ export const ACTION_MODES = [
 
 export type AssistantBuilderDataSourceConfiguration = {
   dataSource: DataSourceType;
+  // TODO(GROUPS_INFRA) Replace with DataSourceViewType once the UI has it.
+  dataSourceViewId: null;
   selectedResources: ContentNode[];
   isSelectAll: boolean;
 };