👕

dust-tt · Jan 17, 2024 · 4020c66 · 4020c66
1 parent bfde1e1
commit 4020c66
Show file tree

Hide file tree

Showing 29 changed files with 104 additions and 130 deletions.
diff --git a/front/components/app/blocks/Block.tsx b/front/components/app/blocks/Block.tsx
@@ -112,8 +112,8 @@ export default function Block({
                 readOnly
                   ? "border-white ring-0 focus:border-white focus:ring-0"
                   : nameError != ""
-                    ? "border-orange-400 focus:border-orange-400 focus:ring-0"
-                    : "border-white focus:border-gray-300 focus:ring-0"
+                  ? "border-orange-400 focus:border-orange-400 focus:ring-0"
+                  : "border-white focus:border-gray-300 focus:ring-0"
               )}
               readOnly={readOnly}
               value={block.name}

diff --git a/front/components/assistant/RenderMessageMarkdown.tsx b/front/components/assistant/RenderMessageMarkdown.tsx
@@ -268,8 +268,8 @@ function MentionBlock({
     !agentConfiguration || agentConfiguration?.status === "archived"
       ? "(This assistant was deleted)"
       : agentConfiguration?.status === "active"
-        ? ""
-        : "(This assistant is deactivated for this workspace)";
+      ? ""
+      : "(This assistant is deactivated for this workspace)";
   const tooltipLabel = agentConfiguration?.description || "" + " " + statusText;
   return (
     <span className="inline-block cursor-default font-medium text-brand">

diff --git a/front/components/assistant/conversation/AgentMessage.tsx b/front/components/assistant/conversation/AgentMessage.tsx
@@ -289,13 +289,10 @@ export function AgentMessage({
       agentMessageToRender.action.documents
     ) {
       setReferences(
-        agentMessageToRender.action.documents.reduce(
-          (acc, d) => {
-            acc[d.reference] = d;
-            return acc;
-          },
-          {} as { [key: string]: RetrievalDocumentType }
-        )
+        agentMessageToRender.action.documents.reduce((acc, d) => {
+          acc[d.reference] = d;
+          return acc;
+        }, {} as { [key: string]: RetrievalDocumentType })
       );
     }
   }, [agentMessageToRender.action]);

diff --git a/front/components/assistant_builder/AssistantBuilderDataSourceModal.tsx b/front/components/assistant_builder/AssistantBuilderDataSourceModal.tsx
@@ -215,13 +215,10 @@ function DataSourceResourceSelector({
       }
       const json: GetConnectorResourceParentsResponseBody = await res.json();
       setParentsById(
-        json.resources.reduce(
-          (acc, r) => {
-            acc[r.internalId] = new Set(r.parents);
-            return acc;
-          },
-          {} as Record<string, Set<string>>
-        )
+        json.resources.reduce((acc, r) => {
+          acc[r.internalId] = new Set(r.parents);
+          return acc;
+        }, {} as Record<string, Set<string>>)
       );
     } catch (e) {
       setParentsAreError(true);

diff --git a/front/components/providers/AI21Setup.tsx b/front/components/providers/AI21Setup.tsx
@@ -183,8 +183,8 @@ export default function AI21Setup({
                               ? "Updating..."
                               : "Update"
                             : enableRunning
-                              ? "Enabling..."
-                              : "Enable"
+                            ? "Enabling..."
+                            : "Enable"
                         }
                       />
                     ) : (

diff --git a/front/components/providers/AnthropicSetup.tsx b/front/components/providers/AnthropicSetup.tsx
@@ -186,8 +186,8 @@ export default function AnthropicSetup({
                               ? "Updating..."
                               : "Update"
                             : enableRunning
-                              ? "Enabling..."
-                              : "Enable"
+                            ? "Enabling..."
+                            : "Enable"
                         }
                       />
                     ) : (

diff --git a/front/components/providers/AzureOpenAISetup.tsx b/front/components/providers/AzureOpenAISetup.tsx
@@ -197,8 +197,8 @@ export default function AzureOpenAISetup({
                               ? "Updating..."
                               : "Update"
                             : enableRunning
-                              ? "Enabling..."
-                              : "Enable"
+                            ? "Enabling..."
+                            : "Enable"
                         }
                       />
                     ) : (

diff --git a/front/components/providers/BrowserlessAPISetup.tsx b/front/components/providers/BrowserlessAPISetup.tsx
@@ -191,8 +191,8 @@ export default function BrowserlessAPISetup({
                               ? "Updating..."
                               : "Update"
                             : enableRunning
-                              ? "Enabling..."
-                              : "Enable"
+                            ? "Enabling..."
+                            : "Enable"
                         }
                       />
                     ) : (

diff --git a/front/components/providers/CohereSetup.tsx b/front/components/providers/CohereSetup.tsx
@@ -183,8 +183,8 @@ export default function CohereSetup({
                               ? "Updating..."
                               : "Update"
                             : enableRunning
-                              ? "Enabling..."
-                              : "Enable"
+                            ? "Enabling..."
+                            : "Enable"
                         }
                       />
                     ) : (

diff --git a/front/components/providers/GoogleVertexAISetup.tsx b/front/components/providers/GoogleVertexAISetup.tsx
@@ -201,8 +201,8 @@ export default function GoogleVertexAISetup({
                               ? "Updating..."
                               : "Update"
                             : enableRunning
-                              ? "Enabling..."
-                              : "Enable"
+                            ? "Enabling..."
+                            : "Enable"
                         }
                       />
                     ) : (

diff --git a/front/components/providers/MistralAISetup.tsx b/front/components/providers/MistralAISetup.tsx
@@ -184,8 +184,8 @@ export default function MistralAISetup({
                               ? "Updating..."
                               : "Update"
                             : enableRunning
-                              ? "Enabling..."
-                              : "Enable"
+                            ? "Enabling..."
+                            : "Enable"
                         }
                       />
                     ) : (

diff --git a/front/components/providers/OpenAISetup.tsx b/front/components/providers/OpenAISetup.tsx
@@ -183,8 +183,8 @@ export default function OpenAISetup({
                               ? "Updating..."
                               : "Update"
                             : enableRunning
-                              ? "Enabling..."
-                              : "Enable"
+                            ? "Enabling..."
+                            : "Enable"
                         }
                       />
                     ) : (

diff --git a/front/components/providers/SerpAPISetup.tsx b/front/components/providers/SerpAPISetup.tsx
@@ -183,8 +183,8 @@ export default function SerpAPISetup({
                               ? "Updating..."
                               : "Update"
                             : enableRunning
-                              ? "Enabling..."
-                              : "Enable"
+                            ? "Enabling..."
+                            : "Enable"
                         }
                       />
                     ) : (

diff --git a/front/components/providers/SerperSetup.tsx b/front/components/providers/SerperSetup.tsx
@@ -183,8 +183,8 @@ export default function SerperSetup({
                               ? "Updating..."
                               : "Update"
                             : enableRunning
-                              ? "Enabling..."
-                              : "Enable"
+                            ? "Enabling..."
+                            : "Enable"
                         }
                       />
                     ) : (

diff --git a/front/components/providers/TextSynthSetup.tsx b/front/components/providers/TextSynthSetup.tsx
@@ -185,8 +185,8 @@ export default function TextSynthSetup({
                               ? "Updating..."
                               : "Update"
                             : enableRunning
-                              ? "Enabling..."
-                              : "Enable"
+                            ? "Enabling..."
+                            : "Enable"
                         }
                       />
                     ) : (

diff --git a/...ost_process_hooks/hooks/document_tracker/suggest_changes/actions/doc_tracker_retrieval.ts b/...ost_process_hooks/hooks/document_tracker/suggest_changes/actions/doc_tracker_retrieval.ts
@@ -15,8 +15,9 @@ export async function callDocTrackerRetrievalAction(
   const action = DustProdActionRegistry["doc-tracker-retrieval"];
   const config = cloneBaseConfig(action.config);
 
-  config.SEMANTIC_SEARCH.data_sources =
-    await getTrackableDataSources(workspaceId);
+  config.SEMANTIC_SEARCH.data_sources = await getTrackableDataSources(
+    workspaceId
+  );
   config.SEMANTIC_SEARCH.target_document_tokens = targetDocumentTokens;
 
   const res = await callAction({

diff --git a/front/documents_post_process_hooks/hooks/index.ts b/front/documents_post_process_hooks/hooks/index.ts
@@ -89,13 +89,10 @@ export const DOCUMENTS_POST_PROCESS_HOOKS = [
 export const DOCUMENTS_POST_PROCESS_HOOK_BY_TYPE: Record<
   DocumentsPostProcessHookType,
   DocumentsPostProcessHook
-> = DOCUMENTS_POST_PROCESS_HOOKS.reduce(
-  (acc, hook) => {
-    acc[hook.type] = hook;
-    return acc;
-  },
-  {} as Record<DocumentsPostProcessHookType, DocumentsPostProcessHook>
-);
+> = DOCUMENTS_POST_PROCESS_HOOKS.reduce((acc, hook) => {
+  acc[hook.type] = hook;
+  return acc;
+}, {} as Record<DocumentsPostProcessHookType, DocumentsPostProcessHook>);
 
 export async function getDocumentsPostUpsertHooksToRun(
   params: DocumentsPostProcessHookOnUpsertParams

diff --git a/front/lib/api/assistant/configuration.ts b/front/lib/api/assistant/configuration.ts
@@ -212,13 +212,10 @@ export async function getAgentConfigurations<V extends "light" | "full">({
   })();
 
   function byId<T extends { id: number }>(list: T[]): Record<string, T> {
-    return list.reduce(
-      (acc, item) => {
-        acc[item.id] = item;
-        return acc;
-      },
-      {} as Record<number, T>
-    );
+    return list.reduce((acc, item) => {
+      acc[item.id] = item;
+      return acc;
+    }, {} as Record<number, T>);
   }
 
   const configurationIds = agentConfigurations.map((a) => a.id);
@@ -270,13 +267,10 @@ export async function getAgentConfigurations<V extends "light" | "full">({
             userId: user.id,
           },
         }).then((relations) =>
-          relations.reduce(
-            (acc, relation) => {
-              acc[relation.agentConfiguration] = relation;
-              return acc;
-            },
-            {} as Record<string, AgentUserRelation>
-          )
+          relations.reduce((acc, relation) => {
+            acc[relation.agentConfiguration] = relation;
+            return acc;
+          }, {} as Record<string, AgentUserRelation>)
         )
       : Promise.resolve({} as Record<string, AgentUserRelation>),
   ]);
@@ -291,15 +285,12 @@ export async function getAgentConfigurations<V extends "light" | "full">({
           },
         })
       : []
-  ).reduce(
-    (acc, dsConfig) => {
-      acc[dsConfig.retrievalConfigurationId] =
-        acc[dsConfig.retrievalConfigurationId] || [];
-      acc[dsConfig.retrievalConfigurationId].push(dsConfig);
-      return acc;
-    },
-    {} as Record<number, AgentDataSourceConfiguration[]>
-  );
+  ).reduce((acc, dsConfig) => {
+    acc[dsConfig.retrievalConfigurationId] =
+      acc[dsConfig.retrievalConfigurationId] || [];
+    acc[dsConfig.retrievalConfigurationId].push(dsConfig);
+    return acc;
+  }, {} as Record<number, AgentDataSourceConfiguration[]>);
 
   const dataSourceIds = Object.values(agentDatasourceConfigurations)
     .flat()
@@ -314,13 +305,10 @@ export async function getAgentConfigurations<V extends "light" | "full">({
           },
         })
       : []
-  ).reduce(
-    (acc, ds) => {
-      acc[ds.id] = ds;
-      return acc;
-    },
-    {} as Record<number, DataSource>
-  );
+  ).reduce((acc, ds) => {
+    acc[ds.id] = ds;
+    return acc;
+  }, {} as Record<number, DataSource>);
 
   const workspaceIds = Object.values(dataSources).map((ds) => ds.workspaceId);
   const dataSourceWorkspaces = (
@@ -333,13 +321,10 @@ export async function getAgentConfigurations<V extends "light" | "full">({
           },
         })
       : []
-  ).reduce(
-    (acc, ws) => {
-      acc[ws.id] = ws;
-      return acc;
-    },
-    {} as Record<number, Workspace>
-  );
+  ).reduce((acc, ws) => {
+    acc[ws.id] = ws;
+    return acc;
+  }, {} as Record<number, Workspace>);
 
   let agentConfigurationTypes: AgentConfigurationType[] = [];
   for (const agent of agentConfigurations) {

diff --git a/front/lib/api/assistant/recent_authors.ts b/front/lib/api/assistant/recent_authors.ts
@@ -79,8 +79,9 @@ async function populateAuthorIdsFromDb({
   agentId: string;
   workspaceId: string;
 }) {
-  const recentAuthorIdsWithVersion =
-    await fetchRecentAuthorIdsWithVersion(agentId);
+  const recentAuthorIdsWithVersion = await fetchRecentAuthorIdsWithVersion(
+    agentId
+  );
 
   if (recentAuthorIdsWithVersion.length === 0) {
     return [];

diff --git a/front/migrations/helpers.ts b/front/migrations/helpers.ts
@@ -14,10 +14,10 @@ type InferArgs<T> = {
   [P in keyof T]: T[P] extends { type: "number" }
     ? number
     : T[P] extends { type: "boolean" }
-      ? boolean
-      : T[P] extends { type: "string" }
-        ? string
-        : never;
+    ? boolean
+    : T[P] extends { type: "string" }
+    ? string
+    : never;
 } & { execute?: boolean };
 
 const defaultArgumentSpecs: ArgumentSpecs = {

diff --git a/front/pages/api/poke/plans.ts b/front/pages/api/poke/plans.ts
@@ -123,13 +123,10 @@ async function handler(
         await Promise.all(
           stripeProductIds.map((stripeProductId) => getProduct(stripeProductId))
         )
-      ).reduce(
-        (acc, product) => {
-          acc[product.id] = product;
-          return acc;
-        },
-        {} as { [key: string]: Stripe.Product }
-      );
+      ).reduce((acc, product) => {
+        acc[product.id] = product;
+        return acc;
+      }, {} as { [key: string]: Stripe.Product });
 
       res.status(200).json({
         plans: plans.map((plan) => ({

diff --git a/front/pages/api/stripe/webhook.ts b/front/pages/api/stripe/webhook.ts
@@ -513,8 +513,9 @@ async function handler(
               endDate: new Date(),
             });
             const workspaceId = activeSubscription.workspace.sId;
-            const auth =
-              await Authenticator.internalAdminForWorkspace(workspaceId);
+            const auth = await Authenticator.internalAdminForWorkspace(
+              workspaceId
+            );
             await revokeUsersForDowngrade(auth);
             await archiveConnectedAgents(auth);
             await deleteConnectedDatasources(auth);

diff --git a/front/pages/api/w/[wId]/assistant/agent_configurations/index.ts b/front/pages/api/w/[wId]/assistant/agent_configurations/index.ts
@@ -91,8 +91,8 @@ async function handler(
       const viewParam = view
         ? view
         : conversationId
-          ? { conversationId }
-          : "all";
+        ? { conversationId }
+        : "all";
       if (viewParam === "admin_internal" && !auth.isDustSuperUser()) {
         return apiError(req, res, {
           status_code: 404,