fix

dust-tt · Aug 9, 2024 · 9ede7b4 · 9ede7b4
1 parent e4c878f
commit 9ede7b4
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 12 deletions.
diff --git a/front/pages/api/w/[wId]/vaults/[vId]/categories.ts b/front/pages/api/w/[wId]/vaults/[vId]/categories.ts
@@ -8,7 +8,7 @@ import { apiError } from "@app/logger/withlogging";
 import { getDataSourceViewsInfo } from "@app/pages/api/w/[wId]/vaults/[vId]/data_source_views";
 import { getDataSourceInfos } from "@app/pages/api/w/[wId]/vaults/[vId]/data_sources";
 
-type VaultCategoryInfo = {
+export type VaultCategoryInfo = {
   category: ResourceCategory;
   usage: number;
   count: number;

diff --git a/front/pages/api/w/[wId]/vaults/[vId]/data_source_views/index.ts b/front/pages/api/w/[wId]/vaults/[vId]/data_source_views/index.ts
@@ -12,8 +12,8 @@ import { DataSourceViewResource } from "@app/lib/resources/data_source_view_reso
 import { VaultResource } from "@app/lib/resources/vault_resource";
 import { apiError } from "@app/logger/withlogging";
 
-export type GetVaultDataSourcesResponseBody = {
-  dataSources: ResourceInfo[];
+export type GetVaultDataSourceViewsResponseBody = {
+  dataSourceViews: ResourceInfo[];
 };
 
 export const getDataSourceCategory = (
@@ -38,7 +38,7 @@ export const getDataSourceViewsInfo = async (
 
   return dataSourceViews.map((view) => {
     return {
-      ...view.dataSource?.toJSON(),
+      ...(view.dataSource as DataSourceResource).toJSON(),
       ...view.toJSON(),
       usage: 0,
       category: getDataSourceCategory(view.dataSource as DataSourceResource),
@@ -48,7 +48,9 @@ export const getDataSourceViewsInfo = async (
 
 async function handler(
   req: NextApiRequest,
-  res: NextApiResponse<WithAPIErrorResponse<GetVaultDataSourcesResponseBody>>,
+  res: NextApiResponse<
+    WithAPIErrorResponse<GetVaultDataSourceViewsResponseBody>
+  >,
   auth: Authenticator
 ): Promise<void> {
   const owner = auth.workspace();
@@ -87,7 +89,7 @@ async function handler(
       const all = await getDataSourceViewsInfo(auth, vault);
 
       res.status(200).json({
-        dataSources: all.filter(
+        dataSourceViews: all.filter(
           (dataSourceInfo) => !category || dataSourceInfo.category === category
         ),
       });

diff --git a/front/pages/api/w/[wId]/vaults/[vId]/data_sources/index.ts b/front/pages/api/w/[wId]/vaults/[vId]/data_sources/index.ts
@@ -11,8 +11,8 @@ import { DataSourceResource } from "@app/lib/resources/data_source_resource";
 import { VaultResource } from "@app/lib/resources/vault_resource";
 import { apiError } from "@app/logger/withlogging";
 
-export type GetVaultDataSourceViewsResponseBody = {
-  dataSourceViews: ResourceInfo[];
+export type GetVaultDataSourcesResponseBody = {
+  dataSources: ResourceInfo[];
 };
 
 export const getDataSourceCategory = (
@@ -45,9 +45,7 @@ export const getDataSourceInfos = async (
 
 async function handler(
   req: NextApiRequest,
-  res: NextApiResponse<
-    WithAPIErrorResponse<GetVaultDataSourceViewsResponseBody>
-  >,
+  res: NextApiResponse<WithAPIErrorResponse<GetVaultDataSourcesResponseBody>>,
   auth: Authenticator
 ): Promise<void> {
   const owner = auth.workspace();
@@ -86,7 +84,7 @@ async function handler(
       const all = await getDataSourceInfos(auth, vault);
 
       res.status(200).json({
-        dataSourceViews: all.filter(
+        dataSources: all.filter(
           (dataSourceInfo) => !category || dataSourceInfo.category === category
         ),
       });