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:18:53 -0400 |
| commit | 3f79590dbd36dca389b9472e09a4ea6a7943a453 (patch) | |
| tree | 925c30e1b14d9b96966039d60b006ffa2cd667d6 /cmd/mimi | |
| parent | 301998a7fb21579155335e0abfba5ae998cee76a (diff) | |
| download | x-3f79590dbd36dca389b9472e09a4ea6a7943a453.tar.xz x-3f79590dbd36dca389b9472e09a4ea6a7943a453.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 | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/cmd/mimi/modules/discord/jufra/jufra.go b/cmd/mimi/modules/discord/jufra/jufra.go index 1033672..1f0acd1 100644 --- a/cmd/mimi/modules/discord/jufra/jufra.go +++ b/cmd/mimi/modules/discord/jufra/jufra.go @@ -164,9 +164,11 @@ func (m *Module) llamaGuardCheck(ctx context.Context, role string, messages []ol return llamaguard.Check(ctx, m.ollama, role, *llamaGuardModel, messages) } -func (m *Module) llamaGuardComplain(ctx context.Context, lgResp *llamaguard.Response) (string, error) { +func (m *Module) llamaGuardComplain(ctx context.Context, from string, lgResp *llamaguard.Response) (string, error) { var sb strings.Builder - sb.WriteString("⚠️ Rule violation detected ⚠️\n") + sb.WriteString("⚠️ Rule violation detected from ") + sb.WriteString(from) + sb.WriteString(":\n") for _, cat := range lgResp.ViolationCategories { sb.WriteString("- ") sb.WriteString(cat.String()) |