Loki/internal/loki/llm_repeat_test.go

package loki

import (
	"strings"
	"testing"
)

// Un appel d'outil redemandé à l'identique ne doit rendre son contenu qu'UNE
// fois, et l'avertissement doit être en TÊTE (collé après un résultat de
// plusieurs milliers de caractères, il passait inaperçu et le modèle rebouclait).
func TestRepeatedCallResultEscalates(t *testing.T) {
	page := strings.Repeat("contenu de la page ", 300)

	first := repeatedCallResult(page, 1)
	if !strings.HasPrefix(first, "[déjà fait]") {
		t.Fatalf("1re redemande : l'avertissement doit ouvrir le résultat, obtenu %.40q", first)
	}
	if !strings.Contains(first, page) {
		t.Fatal("1re redemande : le contenu doit encore être rendu")
	}

	for _, n := range []int{2, 3, 7} {
		again := repeatedCallResult(page, n)
		if strings.Contains(again, page) {
			t.Fatalf("redemande n°%d : le contenu ne doit plus être renvoyé", n)
		}
		if !strings.HasPrefix(again, "[déjà fait]") {
			t.Fatalf("redemande n°%d : avertissement manquant", n)
		}
	}
}