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
}