fix: adding wait group so we can concurrently call these

This commit is contained in:
2025-07-29 15:40:18 +01:00
parent 5024933852
commit 13170a33e8

View File

@ -10,6 +10,7 @@ import (
"screenmark/screenmark/agents"
"screenmark/screenmark/models"
"strconv"
"sync"
"time"
"github.com/charmbracelet/log"
@ -64,14 +65,22 @@ func ListenNewImageEvents(db *sql.DB, notifier *Notifier[Notification]) {
}
descriptionAgent := agents.NewDescriptionAgent(createLogger("Description 📝", splitWriter), imageModel)
err = descriptionAgent.Describe(createLogger("Description 📓", splitWriter), image.Image.ID, image.Image.ImageName, image.Image.Image)
if err != nil {
log.Error(err)
}
listAgent := agents.NewListAgent(createLogger("Lists 🖋️", splitWriter), listModel)
listAgent.RunAgent(image.UserID, image.ImageID, image.Image.ImageName, image.Image.Image)
var wg sync.WaitGroup
wg.Add(2)
go func() {
descriptionAgent.Describe(createLogger("Description 📓", splitWriter), image.Image.ID, image.Image.ImageName, image.Image.Image)
wg.Done()
}()
go func() {
listAgent.RunAgent(image.UserID, image.ImageID, image.Image.ImageName, image.Image.Image)
wg.Done()
}()
wg.Wait()
_, err = imageModel.FinishProcessing(ctx, image.ID)
if err != nil {