add parent_id in returns

dust-tt · Dec 11, 2024 · 463f7e7 · 463f7e7
1 parent ee57b59
commit 463f7e7
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 0 deletions.
diff --git a/front/pages/api/v1/w/[wId]/spaces/[spaceId]/data_sources/[dsId]/tables/[tId]/index.ts b/front/pages/api/v1/w/[wId]/spaces/[spaceId]/data_sources/[dsId]/tables/[tId]/index.ts
@@ -205,6 +205,7 @@ async function handler(
           timestamp: table.timestamp,
           tags: table.tags,
           parents: table.parents,
+          parent_id: table.parent_id,
           mime_type: table.mime_type,
           title: table.title,
         },

diff --git a/front/pages/api/v1/w/[wId]/spaces/[spaceId]/data_sources/[dsId]/tables/index.ts b/front/pages/api/v1/w/[wId]/spaces/[spaceId]/data_sources/[dsId]/tables/index.ts
@@ -236,6 +236,7 @@ async function handler(
             timestamp: table.timestamp,
             tags: table.tags,
             parents: table.parents,
+            parent_id: table.parent_id,
             mime_type: table.mime_type,
             title: table.title,
           };

diff --git a/front/pages/api/v1/w/[wId]/swagger_schemas.ts b/front/pages/api/v1/w/[wId]/swagger_schemas.ts
@@ -471,6 +471,12 @@
  *           description: Array of tags associated with the table
  *           items:
  *             type: string
+ *         parent_id:
+ *           type: string
+ *           description: ID of the table parent
+ *           items:
+ *             type: string
+ *           example: "1234f4567c"
  *         parents:
  *           type: array
  *           description: Array of parent table IDs
@@ -594,6 +600,12 @@
  *           items:
  *             type: string
  *           example: ["customer_support", "faq"]
+ *         parent_id:
+ *           type: string
+ *           description: ID of the document parent
+ *           items:
+ *             type: string
+ *           example: "1234f4567c"
  *         parents:
  *           type: array
  *           items:

diff --git a/sdks/js/src/types.ts b/sdks/js/src/types.ts
@@ -268,6 +268,7 @@ const CoreAPIDocumentSchema = z.object({
   created: z.number(),
   document_id: z.string(),
   timestamp: z.number(),
+  parent_id: z.string().nullable().optional(),
   tags: z.array(z.string()),
   source_url: z.string().nullable().optional(),
   hash: z.string(),
@@ -318,6 +319,7 @@ const CoreAPITablePublicSchema = z.object({
   timestamp: z.number(),
   tags: z.array(z.string()),
   parents: z.array(z.string()),
+  parent_id: z.string().nullable().optional(),
   mime_type: z.string().optional(),
   title: z.string().optional(),
 });

diff --git a/types/src/core/data_source.ts b/types/src/core/data_source.ts
@@ -48,6 +48,7 @@ export type CoreAPIDocument = {
   data_source_id: string;
   created: number;
   document_id: string;
+  parent_id: string | null;
   timestamp: number;
   tags: string[];
   source_url?: string | null;
@@ -79,6 +80,7 @@ export type CoreAPIFolder = {
   folder_id: string;
   timestamp: number;
   title: string;
+  parent_id: string | null;
   parents: string[];
 };