Fix output repair diagnostics and concurrency tests

2026-08-25 11:54:50 +00:00
parent 465dc1389d
commit 36ce5a5099
3 changed files with 39 additions and 10 deletions
--- a/internal/usecase/runner_test.go
+++ b/internal/usecase/runner_test.go
@@ -295,8 +295,10 @@ func (c *controlledRepairLLM) Generate(
 	ctx context.Context,
 	req domain.GenerateRequest,
 ) (*domain.GenerateResponse, error) {
-	isRepair := len(req.Prompt.Messages) > 0 &&
-		strings.HasPrefix(req.Prompt.Messages[0].Content, "You repair invalid JSON")
+	messages := req.Prompt.Messages
+	isRepair := len(messages) >= 2 &&
+		messages[len(messages)-2].Role == "assistant" &&
+		messages[len(messages)-1].Role == "user"
 	c.mu.Lock()
 	c.calls++
 	c.active++