Assistant: Add missing CreationOptional id/createdAt/updatedAt in con…

…fig models
dust-tt · Sep 6, 2023 · f3335cc · f3335cc
1 parent 689034e
commit f3335cc
Show file tree

Hide file tree

Showing 2 changed files with 54 additions and 4 deletions.
diff --git a/front/lib/models/assistant/actions/retrieval.ts b/front/lib/models/assistant/actions/retrieval.ts
@@ -1,4 +1,5 @@
 import {
+  CreationOptional,
   DataTypes,
   ForeignKey,
   InferAttributes,
@@ -18,7 +19,9 @@ export class AgentRetrievalConfiguration extends Model<
   InferAttributes<AgentRetrievalConfiguration>,
   InferCreationAttributes<AgentRetrievalConfiguration>
 > {
-  declare id: number;
+  declare id: CreationOptional<number>;
+  declare createdAt: CreationOptional<Date>;
+  declare updatedAt: CreationOptional<Date>;
 
   declare query: "auto" | "none" | "templated";
   declare queryTemplate: string | null;
@@ -36,6 +39,16 @@ AgentRetrievalConfiguration.init(
       autoIncrement: true,
       primaryKey: true,
     },
+    createdAt: {
+      type: DataTypes.DATE,
+      allowNull: false,
+      defaultValue: DataTypes.NOW,
+    },
+    updatedAt: {
+      type: DataTypes.DATE,
+      allowNull: false,
+      defaultValue: DataTypes.NOW,
+    },
     query: {
       type: DataTypes.STRING,
       allowNull: false,
@@ -100,7 +113,9 @@ export class AgentDataSourceConfiguration extends Model<
   InferAttributes<AgentDataSourceConfiguration>,
   InferCreationAttributes<AgentDataSourceConfiguration>
 > {
-  declare id: number;
+  declare id: CreationOptional<number>;
+  declare createdAt: CreationOptional<Date>;
+  declare updatedAt: CreationOptional<Date>;
 
   declare minTimestamp: number | null;
   declare maxTimestamp: number | null;
@@ -124,6 +139,16 @@ AgentDataSourceConfiguration.init(
       autoIncrement: true,
       primaryKey: true,
     },
+    createdAt: {
+      type: DataTypes.DATE,
+      allowNull: false,
+      defaultValue: DataTypes.NOW,
+    },
+    updatedAt: {
+      type: DataTypes.DATE,
+      allowNull: false,
+      defaultValue: DataTypes.NOW,
+    },
     minTimestamp: {
       type: DataTypes.INTEGER,
       allowNull: true,

diff --git a/front/lib/models/assistant/agent.ts b/front/lib/models/assistant/agent.ts
@@ -1,4 +1,5 @@
 import {
+  CreationOptional,
   DataTypes,
   ForeignKey,
   InferAttributes,
@@ -21,7 +22,9 @@ export class AgentConfiguration extends Model<
   InferAttributes<AgentConfiguration>,
   InferCreationAttributes<AgentConfiguration>
 > {
-  declare id: number;
+  declare id: CreationOptional<number>;
+  declare createdAt: CreationOptional<Date>;
+  declare updatedAt: CreationOptional<Date>;
 
   declare sId: string;
   declare status: AgentConfigurationStatus;
@@ -40,6 +43,16 @@ AgentConfiguration.init(
       autoIncrement: true,
       primaryKey: true,
     },
+    createdAt: {
+      type: DataTypes.DATE,
+      allowNull: false,
+      defaultValue: DataTypes.NOW,
+    },
+    updatedAt: {
+      type: DataTypes.DATE,
+      allowNull: false,
+      defaultValue: DataTypes.NOW,
+    },
     sId: {
       type: DataTypes.STRING,
       allowNull: false,
@@ -95,7 +108,9 @@ export class AgentGenerationConfiguration extends Model<
   InferAttributes<AgentGenerationConfiguration>,
   InferCreationAttributes<AgentGenerationConfiguration>
 > {
-  declare id: number;
+  declare id: CreationOptional<number>;
+  declare createdAt: CreationOptional<Date>;
+  declare updatedAt: CreationOptional<Date>;
 
   declare prompt: string;
   declare modelProvider: string;
@@ -110,6 +125,16 @@ AgentGenerationConfiguration.init(
       autoIncrement: true,
       primaryKey: true,
     },
+    createdAt: {
+      type: DataTypes.DATE,
+      allowNull: false,
+      defaultValue: DataTypes.NOW,
+    },
+    updatedAt: {
+      type: DataTypes.DATE,
+      allowNull: false,
+      defaultValue: DataTypes.NOW,
+    },
     prompt: {
       type: DataTypes.TEXT,
       allowNull: false,