Merge pull request #506 from innogames/dalle_stats

DALL-E: add metrics for dalle requests
innogames · Dec 1, 2023 · 66fe706 · 66fe706
2 parents 6216dbf + b4afa21
commit 66fe706
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/bot/interaction.go b/bot/interaction.go
@@ -28,7 +28,7 @@ func (b *Bot) handleEvent(eventsAPIEvent slackevents.EventsAPIEvent) {
 				return
 			}
 
-			if len(ev.Files) > 0 {
+			if len(ev.Files) > 0 && ev.User != b.auth.UserID {
 				ev.Text += b.loadFileContent(ev)
 			}
 

diff --git a/bot/stats/stats.go b/bot/stats/stats.go
@@ -36,9 +36,7 @@ func Increase[T constraints.Signed](key string, count T) {
 
 		value += count
 
-		if err := storage.Write(collection, key, value); err != nil {
-			log.Warn(errors.Wrap(err, "error while increasing stats"))
-		}
+		Set(key, value)
 	})
 }
 

diff --git a/command/openai/dalle.go b/command/openai/dalle.go
@@ -8,6 +8,7 @@ import (
 
 	"github.com/innogames/slack-bot/v2/bot/matcher"
 	"github.com/innogames/slack-bot/v2/bot/msg"
+	"github.com/innogames/slack-bot/v2/bot/stats"
 	"github.com/innogames/slack-bot/v2/bot/util"
 	log "github.com/sirupsen/logrus"
 	"github.com/slack-go/slack"
@@ -95,6 +96,7 @@ func generateImages(cfg Config, prompt string) ([]DalleResponseImage, error) {
 
 	log.WithField("model", cfg.DalleModel).
 		Infof("Dall-E image generation took %s", time.Since(start))
+	stats.Increase("openai_dalle_images", len(response.Data))
 
 	return response.Data, nil
 }