agrs-sequelize-sdk 1.3.14 → 1.3.16

package/models/AIGenerationLog.js

@@ -0,0 +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;
+};

package/models/AIGenerationRequest.js

@@ -0,0 +1,206 @@
+/**
+ * AI Generation Request Model
+ *
+ * Stores all AI generation requests (banners, text, etc.)
+ * Replaces Google Sheets logging with database storage.
+ */
+
+module.exports = (sequelize, DataTypes) => {
+  const AIGenerationRequest = sequelize.define(
+    "AIGenerationRequest",
+    {
+      id: {
+        type: DataTypes.UUID,
+        defaultValue: DataTypes.UUIDV4,
+        primaryKey: true,
+      },
+      requestId: {
+        type: DataTypes.STRING,
+        allowNull: false,
+        unique: true,
+        comment: "Unique request identifier",
+      },
+      generatorType: {
+        type: DataTypes.STRING,
+        allowNull: false,
+        defaultValue: "banner",
+        comment: "Type of generator: banner, text, video, etc.",
+      },
+      // Request data
+      requestText: {
+        type: DataTypes.TEXT,
+        allowNull: false,
+        comment: "Original user request text",
+      },
+      hasReferenceImage: {
+        type: DataTypes.BOOLEAN,
+        allowNull: false,
+        defaultValue: false,
+        comment: "Whether reference image was provided",
+      },
+      referenceImageUrl: {
+        type: DataTypes.STRING,
+        allowNull: true,
+        comment: "URL to reference image if provided",
+      },
+      // Parsed request data
+      subject: {
+        type: DataTypes.STRING,
+        allowNull: true,
+        comment: "Extracted subject from request",
+      },
+      language: {
+        type: DataTypes.STRING,
+        allowNull: true,
+        comment: "Detected language",
+      },
+      styleDescription: {
+        type: DataTypes.TEXT,
+        allowNull: true,
+        comment: "Style description from parsing",
+      },
+      bannerText: {
+        type: DataTypes.TEXT,
+        allowNull: true,
+        comment: "Banner text to display",
+      },
+      visualElements: {
+        type: DataTypes.TEXT,
+        allowNull: true,
+        comment: "Visual elements description",
+      },
+      // Generation prompts
+      promptVersion1: {
+        type: DataTypes.TEXT,
+        allowNull: true,
+        comment: "First prompt version",
+      },
+      promptVersion2: {
+        type: DataTypes.TEXT,
+        allowNull: true,
+        comment: "Second prompt version",
+      },
+      promptVersion3: {
+        type: DataTypes.TEXT,
+        allowNull: true,
+        comment: "Third prompt version",
+      },
+      // Storage info
+      driveFolderId: {
+        type: DataTypes.STRING,
+        allowNull: true,
+        comment: "Google Drive folder ID",
+      },
+      driveFolderName: {
+        type: DataTypes.STRING,
+        allowNull: true,
+        comment: "Google Drive folder name",
+      },
+      driveFolderLink: {
+        type: DataTypes.STRING,
+        allowNull: true,
+        comment: "Google Drive folder link",
+      },
+      // Generation results
+      imagesTotal: {
+        type: DataTypes.INTEGER,
+        allowNull: false,
+        defaultValue: 0,
+        comment: "Total number of images generated",
+      },
+      imagesSuccessful: {
+        type: DataTypes.INTEGER,
+        allowNull: false,
+        defaultValue: 0,
+        comment: "Number of successfully generated images",
+      },
+      imagesFailed: {
+        type: DataTypes.INTEGER,
+        allowNull: false,
+        defaultValue: 0,
+        comment: "Number of failed image generations",
+      },
+      // Status
+      status: {
+        type: DataTypes.ENUM(
+          "pending",
+          "processing",
+          "completed",
+          "failed",
+          "partial"
+        ),
+        allowNull: false,
+        defaultValue: "pending",
+        comment: "Request status",
+      },
+      errorMessage: {
+        type: DataTypes.TEXT,
+        allowNull: true,
+        comment: "Error message if generation failed",
+      },
+      // Metadata
+      user: {
+        type: DataTypes.STRING,
+        allowNull: true,
+        comment: "User who made the request",
+      },
+      duration: {
+        type: DataTypes.INTEGER,
+        allowNull: true,
+        comment: "Generation duration in milliseconds",
+      },
+      // Model info
+      parsingModel: {
+        type: DataTypes.STRING,
+        allowNull: true,
+        comment: "Model used for parsing",
+      },
+      promptModel: {
+        type: DataTypes.STRING,
+        allowNull: true,
+        comment: "Model used for prompt generation",
+      },
+      imageModel: {
+        type: DataTypes.STRING,
+        allowNull: true,
+        comment: "Model used for image generation",
+      },
+    },
+    {
+      tableName: "ai_generation_requests",
+      timestamps: true,
+      createdAt: "createdAt",
+      updatedAt: "updatedAt",
+      indexes: [
+        {
+          fields: ["requestId"],
+          unique: true,
+        },
+        {
+          fields: ["generatorType"],
+        },
+        {
+          fields: ["status"],
+        },
+        {
+          fields: ["createdAt"],
+        },
+        {
+          fields: ["user"],
+        },
+      ],
+    }
+  );
+
+  // Associations
+  AIGenerationRequest.associate = function (models) {
+    // One request has many logs
+    AIGenerationRequest.hasMany(models.AIGenerationLog, {
+      foreignKey: "requestId",
+      sourceKey: "requestId",
+      as: "logs",
+    });
+  };
+
+  return AIGenerationRequest;
+};

package/models/GenericFlowRequest.js

@@ -0,0 +1,115 @@
+// models/GenericFlowRequest.js
+module.exports = (sequelize, DataTypes) => {
+    const GenericFlowRequest = sequelize.define(
+      "GenericFlowRequest",
+      {
+        id: {
+          type: DataTypes.BIGINT,
+          autoIncrement: true,
+          primaryKey: true,
+        },
+        request_id: {
+          type: DataTypes.UUID,
+          allowNull: false,
+          unique: true,
+          field: "request_id",
+        },
+        flow_type: {
+          type: DataTypes.ENUM("article", "texts", "keywords", "creative"),
+          allowNull: false,
+          field: "flow_type",
+        },
+        status: {
+          type: DataTypes.ENUM("pending", "processing", "completed", "failed", "retrying"),
+          defaultValue: "pending",
+          allowNull: false,
+        },
+        retry_count: {
+          type: DataTypes.INTEGER,
+          defaultValue: 0,
+          field: "retry_count",
+        },
+        max_retries: {
+          type: DataTypes.INTEGER,
+          defaultValue: 3,
+          field: "max_retries",
+        },
+        input_params: {
+          type: DataTypes.JSONB,
+          allowNull: true,
+          field: "input_params",
+        },
+        output_result: {
+          type: DataTypes.JSONB,
+          allowNull: true,
+          field: "output_result",
+        },
+        error_message: {
+          type: DataTypes.TEXT,
+          allowNull: true,
+          field: "error_message",
+        },
+        error_stack: {
+          type: DataTypes.TEXT,
+          allowNull: true,
+          field: "error_stack",
+        },
+        AGRS_CID: {
+          type: DataTypes.UUID,
+          allowNull: true,
+          field: "AGRS_CID",
+        },
+        process_id: {
+          type: DataTypes.UUID,
+          allowNull: true,
+          field: "process_id",
+        },
+        parent_request_id: {
+          type: DataTypes.UUID,
+          allowNull: true,
+          field: "parent_request_id",
+        },
+        related_request_ids: {
+          type: DataTypes.JSONB,
+          allowNull: true,
+          field: "related_request_ids",
+        },
+        metadata: {
+          type: DataTypes.JSONB,
+          allowNull: true,
+        },
+        processing_time_ms: {
+          type: DataTypes.INTEGER,
+          allowNull: true,
+          field: "processing_time_ms",
+        },
+        started_at: {
+          type: DataTypes.DATE,
+          allowNull: true,
+          field: "started_at",
+        },
+        completed_at: {
+          type: DataTypes.DATE,
+          allowNull: true,
+          field: "completed_at",
+        },
+        failed_at: {
+          type: DataTypes.DATE,
+          allowNull: true,
+          field: "failed_at",
+        },
+        last_retry_at: {
+          type: DataTypes.DATE,
+          allowNull: true,
+          field: "last_retry_at",
+        },
+      },
+      {
+        tableName: "generic_flow_requests",
+        timestamps: true, // Adds createdAt and updatedAt
+        underscored: false, // We're using explicit field mappings
+      }
+    );
+  
+    return GenericFlowRequest;
+  };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agrs-sequelize-sdk",
-  "version": "1.3.14",
+  "version": "1.3.16",
   "main": "index.js",
   "scripts": {
     "start": "node index.js",