diff options
Diffstat (limited to 'cmd/mimi/modules')
| -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()) |