agrs-sequelize-sdk 1.4.16 → 1.4.18

package/models/AICampaignQueue.js

@@ -1,137 +1,144 @@
-module.exports = (sequelize, DataTypes) => {
-    const AICampaignQueue = sequelize.define(
-      "AICampaignQueue",
-      {
-        id: {
-          type: DataTypes.INTEGER,
-          primaryKey: true,
-          autoIncrement: true,
-        },
-        article_id: {
-          type: DataTypes.STRING(255),
-          allowNull: false,
-          field: "article_id",
-        },
-        feed_provider: {
-          type: DataTypes.STRING(100),
-          allowNull: false,
-          field: "feed_provider",
-        },
-        payload: {
-          type: DataTypes.JSONB,
-          allowNull: false,
-        },
-        status: {
-          type: DataTypes.STRING(50),
-          allowNull: false,
-          defaultValue: "pending",
-          validate: {
-            isIn: [["pending", "queued", "processing", "completed", "failed"]],
-          },
-        },
-        created_at: {
-          type: DataTypes.DATE,
-          allowNull: false,
-          defaultValue: DataTypes.NOW,
-          field: "created_at",
-        },
-        processed_at: {
-          type: DataTypes.DATE,
-          allowNull: true,
-          field: "processed_at",
-        },
-        retry_count: {
-          type: DataTypes.INTEGER,
-          allowNull: false,
-          defaultValue: 0,
-          field: "retry_count",
-        },
-        error_message: {
-          type: DataTypes.TEXT,
-          allowNull: true,
-          field: "error_message",
-        },
-        facebook_campaign_id: {
-          type: DataTypes.STRING(255),
-          allowNull: true,
-          field: "facebook_campaign_id",
-        },
-        process_id: {
-          type: DataTypes.STRING(255),
-          allowNull: true,
-          field: "process_id",
-        },
-        creation_logs_process_id: {
-          type: DataTypes.UUID,
-          allowNull: true,
-          field: "creation_logs_process_id",
-          comment: "Process ID from CampaignCreationLogV2 (UUID)",
-        },
-        agrs_cid: {
-          type: DataTypes.STRING(255),
-          allowNull: true,
-          field: "agrs_cid",
-          comment: "AGRS Campaign ID",
-        },
-        preset_id: {
-          type: DataTypes.STRING(255),
-          allowNull: true,
-          field: "preset_id",
-          comment: "Preset ID",
-        },
-        article_creation_payload: {
-          type: DataTypes.JSONB,
-          allowNull: true,
-          field: "article_creation_payload",
-          comment: "Payload sent for article creation (Mine/Predicto/Mobitech specific parameters)",
-        },
-        original_payload: {
-          type: DataTypes.JSONB,
-          allowNull: true,
-          field: "original_payload",
-          comment: "Original createAICampaign request params for this campaign (preset, feed config, AI settings, combo values)",
-        },
-      },
-      {
-        tableName: "ai_campaign_queue",
-        timestamps: false, // We manage created_at manually
-        indexes: [
-          {
-            fields: ["status"],
-          },
-          {
-            fields: ["article_id"],
-          },
-          {
-            fields: ["created_at"],
-          },
-          {
-            fields: ["feed_provider"],
-          },
-          {
-            fields: ["process_id"],
-          },
-          {
-            fields: ["creation_logs_process_id"],
-          },
-          {
-            fields: ["agrs_cid"],
-          },
-          {
-            fields: ["preset_id"],
-          },
-        ],
-      }
-    );
-  
-    AICampaignQueue.associate = (models) => {
-      // Optional: Add association to Article model if needed
-      // AICampaignQueue.belongsTo(models.Article, {
-      //   foreignKey: "article_id",
-      //   targetKey: "AGRSAID",
-      //   as: "Article",
-      // });
-    };
-  
-    return AICampaignQueue;
+module.exports = (sequelize, DataTypes) => {
+    const AICampaignQueue = sequelize.define(
+      "AICampaignQueue",
+      {
+        id: {
+          type: DataTypes.INTEGER,
+          primaryKey: true,
+          autoIncrement: true,
+        },
+        article_id: {
+          type: DataTypes.STRING(255),
+          allowNull: false,
+          field: "article_id",
+        },
+        feed_provider: {
+          type: DataTypes.STRING(100),
+          allowNull: false,
+          field: "feed_provider",
+        },
+        payload: {
+          type: DataTypes.JSONB,
+          allowNull: false,
+        },
+        status: {
+          type: DataTypes.STRING(50),
+          allowNull: false,
+          defaultValue: "pending",
+          validate: {
+            isIn: [["pending", "queued", "processing", "completed", "failed"]],
+          },
+        },
+        created_at: {
+          type: DataTypes.DATE,
+          allowNull: false,
+          defaultValue: DataTypes.NOW,
+          field: "created_at",
+        },
+        processed_at: {
+          type: DataTypes.DATE,
+          allowNull: true,
+          field: "processed_at",
+        },
+        retry_count: {
+          type: DataTypes.INTEGER,
+          allowNull: false,
+          defaultValue: 0,
+          field: "retry_count",
+        },
+        error_message: {
+          type: DataTypes.TEXT,
+          allowNull: true,
+          field: "error_message",
+        },
+        facebook_campaign_id: {
+          type: DataTypes.STRING(255),
+          allowNull: true,
+          field: "facebook_campaign_id",
+        },
+        process_id: {
+          type: DataTypes.STRING(255),
+          allowNull: true,
+          field: "process_id",
+        },
+        creation_logs_process_id: {
+          type: DataTypes.UUID,
+          allowNull: true,
+          field: "creation_logs_process_id",
+          comment: "Process ID from CampaignCreationLogV2 (UUID)",
+        },
+        agrs_cid: {
+          type: DataTypes.STRING(255),
+          allowNull: true,
+          field: "agrs_cid",
+          comment: "AGRS Campaign ID",
+        },
+        preset_id: {
+          type: DataTypes.STRING(255),
+          allowNull: true,
+          field: "preset_id",
+          comment: "Preset ID",
+        },
+        article_creation_payload: {
+          type: DataTypes.JSONB,
+          allowNull: true,
+          field: "article_creation_payload",
+          comment: "Payload sent for article creation (Mine/Predicto/Mobitech specific parameters)",
+        },
+        original_payload: {
+          type: DataTypes.JSONB,
+          allowNull: true,
+          field: "original_payload",
+          comment: "Original createAICampaign request params for this campaign (preset, feed config, AI settings, combo values)",
+        },
+        platform: {
+          type: DataTypes.ENUM("fb", "tt", "snp"),
+          allowNull: false,
+          defaultValue: "fb",
+          field: "platform",
+          comment: "Ad platform identifier: fb=Facebook, tt=TikTok, snp=Snapchat",
+        },
+      },
+      {
+        tableName: "ai_campaign_queue",
+        timestamps: false, // We manage created_at manually
+        indexes: [
+          {
+            fields: ["status"],
+          },
+          {
+            fields: ["article_id"],
+          },
+          {
+            fields: ["created_at"],
+          },
+          {
+            fields: ["feed_provider"],
+          },
+          {
+            fields: ["process_id"],
+          },
+          {
+            fields: ["creation_logs_process_id"],
+          },
+          {
+            fields: ["agrs_cid"],
+          },
+          {
+            fields: ["preset_id"],
+          },
+        ],
+      }
+    );
+  
+    AICampaignQueue.associate = (models) => {
+      // Optional: Add association to Article model if needed
+      // AICampaignQueue.belongsTo(models.Article, {
+      //   foreignKey: "article_id",
+      //   targetKey: "AGRSAID",
+      //   as: "Article",
+      // });
+    };
+  
+    return AICampaignQueue;
   };

package/models/AIGenerationLog.js

@@ -1,85 +1,85 @@
-/**
- * AI Generation Log Model
- *
- * Stores detailed logs for each step of the generation process.
- * Useful for debugging and monitoring.
- */
-
-module.exports = (sequelize, DataTypes) => {
-  const AIGenerationLog = sequelize.define(
-    "AIGenerationLog",
-    {
-      id: {
-        type: DataTypes.UUID,
-        defaultValue: DataTypes.UUIDV4,
-        primaryKey: true,
-      },
-      requestId: {
-        type: DataTypes.STRING,
-        allowNull: false,
-        comment: "Reference to AIGenerationRequest.requestId",
-        indexes: [
-          {
-            fields: ["requestId"],
-          },
-        ],
-      },
-      step: {
-        type: DataTypes.STRING,
-        allowNull: false,
-        comment:
-          "Step name: parsing, prompt_generation, image_generation, storage, logging",
-      },
-      level: {
-        type: DataTypes.ENUM("info", "warning", "error", "success"),
-        allowNull: false,
-        defaultValue: "info",
-        comment: "Log level",
-      },
-      message: {
-        type: DataTypes.TEXT,
-        allowNull: false,
-        comment: "Log message",
-      },
-      data: {
-        type: DataTypes.JSONB,
-        allowNull: true,
-        comment: "Additional log data (JSON)",
-      },
-      duration: {
-        type: DataTypes.INTEGER,
-        allowNull: true,
-        comment: "Step duration in milliseconds",
-      },
-    },
-    {
-      tableName: "ai_generation_logs",
-      timestamps: true,
-      createdAt: "createdAt",
-      updatedAt: false, // Logs are immutable
-      indexes: [
-        {
-          fields: ["requestId", "step"],
-        },
-        {
-          fields: ["level"],
-        },
-        {
-          fields: ["createdAt"],
-        },
-      ],
-    }
-  );
-
-  // Associations
-  AIGenerationLog.associate = function (models) {
-    // Each log belongs to a request
-    AIGenerationLog.belongsTo(models.AIGenerationRequest, {
-      foreignKey: "requestId",
-      targetKey: "requestId",
-      as: "request",
-    });
-  };
-
-  return AIGenerationLog;
-};
+/**
+ * AI Generation Log Model
+ *
+ * Stores detailed logs for each step of the generation process.
+ * Useful for debugging and monitoring.
+ */
+
+module.exports = (sequelize, DataTypes) => {
+  const AIGenerationLog = sequelize.define(
+    "AIGenerationLog",
+    {
+      id: {
+        type: DataTypes.UUID,
+        defaultValue: DataTypes.UUIDV4,
+        primaryKey: true,
+      },
+      requestId: {
+        type: DataTypes.STRING,
+        allowNull: false,
+        comment: "Reference to AIGenerationRequest.requestId",
+        indexes: [
+          {
+            fields: ["requestId"],
+          },
+        ],
+      },
+      step: {
+        type: DataTypes.STRING,
+        allowNull: false,
+        comment:
+          "Step name: parsing, prompt_generation, image_generation, storage, logging",
+      },
+      level: {
+        type: DataTypes.ENUM("info", "warning", "error", "success"),
+        allowNull: false,
+        defaultValue: "info",
+        comment: "Log level",
+      },
+      message: {
+        type: DataTypes.TEXT,
+        allowNull: false,
+        comment: "Log message",
+      },
+      data: {
+        type: DataTypes.JSONB,
+        allowNull: true,
+        comment: "Additional log data (JSON)",
+      },
+      duration: {
+        type: DataTypes.INTEGER,
+        allowNull: true,
+        comment: "Step duration in milliseconds",
+      },
+    },
+    {
+      tableName: "ai_generation_logs",
+      timestamps: true,
+      createdAt: "createdAt",
+      updatedAt: false, // Logs are immutable
+      indexes: [
+        {
+          fields: ["requestId", "step"],
+        },
+        {
+          fields: ["level"],
+        },
+        {
+          fields: ["createdAt"],
+        },
+      ],
+    }
+  );
+
+  // Associations
+  AIGenerationLog.associate = function (models) {
+    // Each log belongs to a request
+    AIGenerationLog.belongsTo(models.AIGenerationRequest, {
+      foreignKey: "requestId",
+      targetKey: "requestId",
+      as: "request",
+    });
+  };
+
+  return AIGenerationLog;
+};