diff options
| author | Xe Iaso <me@xeiaso.net> | 2024-07-27 09:18:53 -0400 |
|---|---|---|
| committer | Xe Iaso <me@xeiaso.net> | 2024-07-27 09:20:13 -0400 |
| commit | 028e417851f0e1dc1e7ecc8c473e40805c994ebb (patch) | |
| tree | 8a81d9c3138c375f9632b9d47b069e783a1e5d19 /cmd/mimi | |
| parent | 3f79590dbd36dca389b9472e09a4ea6a7943a453 (diff) | |
| download | x-028e417851f0e1dc1e7ecc8c473e40805c994ebb.tar.xz x-028e417851f0e1dc1e7ecc8c473e40805c994ebb.zip | |
cmd/mimi: be more verbose
Signed-off-by: Xe Iaso <me@xeiaso.net>
Diffstat (limited to 'cmd/mimi')
| -rw-r--r-- | cmd/mimi/modules/discord/jufra/jufra.go | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/cmd/mimi/modules/discord/jufra/jufra.go b/cmd/mimi/modules/discord/jufra/jufra.go index 1f0acd1..6b06aa6 100644 --- a/cmd/mimi/modules/discord/jufra/jufra.go +++ b/cmd/mimi/modules/discord/jufra/jufra.go @@ -106,7 +106,7 @@ func (m *Module) messageCreate(s *discordgo.Session, mc *discordgo.MessageCreate } if !lgResp.IsSafe { - msg, err := m.llamaGuardComplain(context.Background(), lgResp) + msg, err := m.llamaGuardComplain(context.Background(), "user", lgResp) if err != nil { slog.Error("error generating response", "err", err, "message_id", mc.ID, "channel_id", mc.ChannelID) s.ChannelMessageSend(mc.ChannelID, "error generating response") @@ -144,7 +144,8 @@ func (m *Module) messageCreate(s *discordgo.Session, mc *discordgo.MessageCreate } if !lgResp.IsSafe { - msg, err := m.llamaGuardComplain(context.Background(), lgResp) + slog.Error("rule violation detected", "message_id", mc.ID, "channel_id", mc.ChannelID, "categories", lgResp.ViolationCategories, "message", resp.Message.Content) + msg, err := m.llamaGuardComplain(context.Background(), "assistant", lgResp) if err != nil { slog.Error("error generating response", "err", err, "message_id", mc.ID, "channel_id", mc.ChannelID) s.ChannelMessageSend(mc.ChannelID, "error generating response") |