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 agents = sequelize.define( 'agents', { id: { type: DataTypes.UUID, defaultValue: DataTypes.UUIDV4, primaryKey: true, }, name: { type: DataTypes.TEXT, }, description: { type: DataTypes.TEXT, }, model: { type: DataTypes.TEXT, }, system_prompt: { type: DataTypes.TEXT, }, temperature: { type: DataTypes.DECIMAL, }, max_output_tokens: { type: DataTypes.INTEGER, }, is_default: { type: DataTypes.BOOLEAN, allowNull: false, defaultValue: false, }, is_active: { type: DataTypes.BOOLEAN, allowNull: false, defaultValue: false, }, metadata_json: { type: DataTypes.TEXT, }, importHash: { type: DataTypes.STRING(255), allowNull: true, unique: true, }, }, { timestamps: true, paranoid: true, freezeTableName: true, }, ); agents.associate = (db) => { /// loop through entities and it's fields, and if ref === current e[name] and create relation has many on parent entity db.agents.hasMany(db.conversations, { as: 'conversations_agent', foreignKey: { name: 'agentId', }, constraints: false, }); db.agents.hasMany(db.usage_events, { as: 'usage_events_agent', foreignKey: { name: 'agentId', }, constraints: false, }); //end loop db.agents.belongsTo(db.users, { as: 'createdBy', }); db.agents.belongsTo(db.users, { as: 'updatedBy', }); }; return agents; };