Revert "changes to implement instructions"

This reverts commit 916fe7ace3.
This commit is contained in:
2026-10-08 15:19:37 -04:00
parent 916fe7ace3
commit 4cf25af5a2
9 changed files with 66 additions and 279 deletions

View File

@@ -10,7 +10,6 @@ const sqs = require('../app/services/sqs')
const s3 = require('../app/services/s3')
const voiceCloningService = require('./voice_cloning')
const userAudioProfileService = require('./user_audio_profile')
const { requireUserId, requireDocumentId } = require('../worker-tenant')
AWS.config.update({ region: 'us-west-2' })
const sqsQueueUrl = process.env.SQS_URL
@@ -102,15 +101,14 @@ const processQueue = () => {
const receiptHandle = response.Messages[0].ReceiptHandle
console.log('job===', job)
const { _id, userAudioProfileId, userId } = job._doc
requireUserId(userId)
requireDocumentId(_id)
requireDocumentId(userAudioProfileId)
const { metadata, input, _id, userAudioProfileId } = job._doc
console.log('userAudioProfileId', userAudioProfileId)
console.log('_id', _id)
const { env } = job
console.log('env', env)
console.log('metadata------', metadata)
console.log('input', input)
const DB_URI =
env === 'production'
? mongoUriProd
@@ -128,26 +126,9 @@ const processQueue = () => {
? cloudFrontUrlStaging
: cloudFrontUrlDev
let authorized = false
try {
await sqs.deleteMessageFromSQS(sqsQueueUrl, receiptHandle)
const cloningJob = await voiceCloningService.read({
_id,
userId,
userAudioProfileId,
})
const audioProfile = await userAudioProfileService.read({
_id: userAudioProfileId,
userId,
})
if (!cloningJob || !audioProfile) {
throw new Error('Cloning job or audio profile does not belong to user')
}
authorized = true
const { metadata, input } = cloningJob
const { directoryName } = metadata
console.log('directoryName', directoryName)
const logPath = `/mnt/efs/potion-voice/${env}/${directoryName}`
@@ -155,10 +136,9 @@ const processQueue = () => {
fs.mkdirSync(logPath, { recursive: true })
}
// update the db model to processing
await voiceCloningService.update({ _id, userId, status: 'processing' })
await voiceCloningService.update({ _id, status: 'processing' })
await userAudioProfileService.update({
_id: userAudioProfileId,
userId,
status: 'processing',
})
@@ -260,7 +240,7 @@ const processQueue = () => {
console.timeEnd(VOICE_MINIMIZE_LABEL)
// Add the code to update location of generated model and status into DB
await voiceCloningService.update({ _id, userId, status: 'completed' })
await voiceCloningService.update({ _id, status: 'completed' })
const training_model_path = {
voice_model_path: `${resultsPath}/${generatedDirectoryName}/checkpoint_365200.pth`,
@@ -272,7 +252,6 @@ const processQueue = () => {
await userAudioProfileService.update({
_id: userAudioProfileId,
userId,
status: 'completed',
training_model_path,
})
@@ -294,7 +273,6 @@ const processQueue = () => {
// add S3 path to user audio profile model
await userAudioProfileService.update({
_id: userAudioProfileId,
userId,
training_model_s3_path,
})
} catch (error) {
@@ -307,14 +285,11 @@ const processQueue = () => {
Bugsnag.notify(error)
// update the db to set status as error
if (authorized) {
await voiceCloningService.update({ _id, userId, status: 'error' })
await userAudioProfileService.update({
_id: userAudioProfileId,
userId,
status: 'error',
})
}
await voiceCloningService.update({ _id, status: 'error' })
await userAudioProfileService.update({
_id: userAudioProfileId,
status: 'error',
})
resolve() // to continue working on new jobs
}

View File

@@ -1,9 +1,8 @@
const { requireUserId } = require('../../worker-tenant')
const StringifyUtils = require('../../app/services/utils/logService')
const create = (UserAudioProfileModel) => async (data) => {
try {
const newModel = new UserAudioProfileModel({ ...data, userId: requireUserId(data.userId) })
const newModel = new UserAudioProfileModel({ ...data })
const savedModel = await newModel.save()
return savedModel
} catch (error) {
@@ -18,9 +17,7 @@ const create = (UserAudioProfileModel) => async (data) => {
const insertMany = (UserAudioProfileModel) => async (data) => {
try {
const inserted = await UserAudioProfileModel.insertMany(
data.map((item) => ({ ...item, userId: requireUserId(item.userId) }))
)
const inserted = await UserAudioProfileModel.insertMany(data)
return inserted
} catch (error) {
const details = { data }
@@ -36,7 +33,6 @@ const read = (UserAudioProfileModel) => async (filter) => {
try {
const foundModel = await UserAudioProfileModel.findOne({
...filter,
userId: requireUserId(filter.userId),
deleted: false,
})
return foundModel
@@ -54,7 +50,6 @@ const find = (UserAudioProfileModel) => async (filter) => {
try {
const foundModels = await UserAudioProfileModel.find({
...filter,
userId: requireUserId(filter.userId),
deleted: false,
})
return foundModels
@@ -70,11 +65,12 @@ const find = (UserAudioProfileModel) => async (filter) => {
const update = (UserAudioProfileModel) => async (data) => {
try {
const { _id, userId, ...changes } = data
const updatedModel = await UserAudioProfileModel.findOneAndUpdate(
{ _id, userId: requireUserId(userId), deleted: false },
{ $set: changes },
{ new: true }
{ _id: data._id },
data,
{
new: true,
}
)
return updatedModel
} catch (error) {
@@ -90,7 +86,7 @@ const update = (UserAudioProfileModel) => async (data) => {
const remove = (UserAudioProfileModel) => async (filter) => {
try {
const updatedModel = await UserAudioProfileModel.findOneAndUpdate(
{ ...filter, userId: requireUserId(filter.userId) },
{ ...filter },
{
$set: {
deleted: true,
@@ -112,7 +108,7 @@ const remove = (UserAudioProfileModel) => async (filter) => {
const removeMany = (UserAudioProfileModel) => async (filter) => {
try {
const updatedModel = await UserAudioProfileModel.updateMany(
{ ...filter, userId: requireUserId(filter.userId) },
{ ...filter },
{
$set: {
deleted: true,

View File

@@ -1,9 +1,8 @@
const { requireUserId } = require('../../worker-tenant')
const StringifyUtils = require('../../app/services/utils/logService')
const create = (VoiceCloningModel) => async (data) => {
try {
const newModel = new VoiceCloningModel({ ...data, userId: requireUserId(data.userId) })
const newModel = new VoiceCloningModel({ ...data })
const savedModel = await newModel.save()
return savedModel
} catch (error) {
@@ -18,9 +17,7 @@ const create = (VoiceCloningModel) => async (data) => {
const insertMany = (VoiceCloningModel) => async (data) => {
try {
const inserted = await VoiceCloningModel.insertMany(
data.map((item) => ({ ...item, userId: requireUserId(item.userId) }))
)
const inserted = await VoiceCloningModel.insertMany(data)
return inserted
} catch (error) {
const details = { data }
@@ -36,7 +33,6 @@ const read = (VoiceCloningModel) => async (filter) => {
try {
const foundModel = await VoiceCloningModel.findOne({
...filter,
userId: requireUserId(filter.userId),
deleted: false,
})
return foundModel
@@ -54,7 +50,6 @@ const find = (VoiceCloningModel) => async (filter) => {
try {
const foundModels = await VoiceCloningModel.find({
...filter,
userId: requireUserId(filter.userId),
deleted: false,
})
return foundModels
@@ -70,11 +65,12 @@ const find = (VoiceCloningModel) => async (filter) => {
const update = (VoiceCloningModel) => async (data) => {
try {
const { _id, userId, ...changes } = data
const updatedModel = await VoiceCloningModel.findOneAndUpdate(
{ _id, userId: requireUserId(userId), deleted: false },
{ $set: changes },
{ new: true }
{ _id: data._id },
data,
{
new: true,
}
)
return updatedModel
@@ -91,7 +87,7 @@ const update = (VoiceCloningModel) => async (data) => {
const remove = (VoiceCloningModel) => async (filter) => {
try {
const updatedModel = await VoiceCloningModel.findOneAndUpdate(
{ ...filter, userId: requireUserId(filter.userId) },
{ ...filter },
{
$set: {
deleted: true,
@@ -113,7 +109,7 @@ const remove = (VoiceCloningModel) => async (filter) => {
const removeMany = (VoiceCloningModel) => async (filter) => {
try {
const updatedModel = await VoiceCloningModel.updateMany(
{ ...filter, userId: requireUserId(filter.userId) },
{ ...filter },
{
$set: {
deleted: true,