const config = require('../../config'); const providers = config.providers; const crypto = require('crypto'); const bcrypt = require('bcrypt'); const moment = require('moment'); module.exports = function(sequelize, DataTypes) { const usage_events = sequelize.define( 'usage_events', { id: { type: DataTypes.UUID, defaultValue: DataTypes.UUIDV4, primaryKey: true, }, event_type: { type: DataTypes.ENUM, values: [ "message_sent", "message_generated", "tokens_counted", "cost_incurred", "attachment_uploaded" ], }, occurred_at: { type: DataTypes.DATE, }, input_tokens: { type: DataTypes.INTEGER, }, output_tokens: { type: DataTypes.INTEGER, }, total_tokens: { type: DataTypes.INTEGER, }, cost_usd: { type: DataTypes.DECIMAL, }, provider: { type: DataTypes.TEXT, }, model: { type: DataTypes.TEXT, }, metadata_json: { type: DataTypes.TEXT, }, importHash: { type: DataTypes.STRING(255), allowNull: true, unique: true, }, }, { timestamps: true, paranoid: true, freezeTableName: true, }, ); usage_events.associate = (db) => { /// loop through entities and it's fields, and if ref === current e[name] and create relation has many on parent entity //end loop db.usage_events.belongsTo(db.users, { as: 'user', foreignKey: { name: 'userId', }, constraints: false, }); db.usage_events.belongsTo(db.conversations, { as: 'conversation', foreignKey: { name: 'conversationId', }, constraints: false, }); db.usage_events.belongsTo(db.messages, { as: 'message', foreignKey: { name: 'messageId', }, constraints: false, }); db.usage_events.belongsTo(db.agents, { as: 'agent', foreignKey: { name: 'agentId', }, constraints: false, }); db.usage_events.belongsTo(db.users, { as: 'createdBy', }); db.usage_events.belongsTo(db.users, { as: 'updatedBy', }); }; return usage_events; };