From 3f79590dbd36dca389b9472e09a4ea6a7943a453 Mon Sep 17 00:00:00 2001 From: Xe Iaso Date: Sat, 27 Jul 2024 09:18:53 -0400 Subject: cmd/mimi: be more verbose Signed-off-by: Xe Iaso --- cmd/mimi/modules/discord/jufra/jufra.go | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'cmd') 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()) -- cgit v1.2.3