fix(worker): постмортем получает транскрипт сессий dev/reviewer по session_id
Some checks failed
CI / test (push) Failing after 1m52s
CI / build-and-package (amd64, linux) (push) Failing after 1m1s
CI / build-and-package (amd64, windows) (push) Successful in 29s

Раньше постмортем-агенту передавался только session_id, без содержимого
сессии — анализировать было нечего. Теперь Runner умеет читать сообщения
сессии (Messages → renderTranscript) через SessionMessages, а postMortem
подставляет реальные шаги агента (сообщения/tool-вызовы) в промпт.
Пустая/недоступная сессия помечается явно. Сбои чтения не фатальны.
This commit is contained in:
ki.sagidullin
2026-08-24 16:48:43 +05:00
parent 89d571b483
commit 6546f18348
5 changed files with 221 additions and 16 deletions

View File

@@ -18,6 +18,9 @@ const postMortemAgent = "postmortem"
// не превращался в полные транскрипты и не переполнял контекст модели).
const traceOutputMax = 6000
// sessionTranscriptMax — обрезка транскрипта сессии dev/reviewer в промпте.
const sessionTranscriptMax = 20000
// postMortemPromptTemplate — промпт для постмортем-агента после failed/timeout:
// задача + сессии dev/reviewer. Ожидается резюме простым текстом на русском.
var postMortemPromptTemplate = template.Must(template.New("postmortem").Parse(`Ты — постмортем-аналитик в конвейере Ratatoskr. Задача завершилась неудачей ({{.Status}}). Проанализируй сессии агентов dev/reviewer и дай резюме: почему так случилось и что сделать, чтобы не повторялось.
@@ -70,8 +73,11 @@ func RenderPostMortemPrompt(data PostMortemPromptData) (string, error) {
return buf.String(), nil
}
// postMortemsText форматирует сессии dev/reviewer в секцию промпта.
func postMortemsText(traces []storage.Trace) string {
// postMortemsText форматирует сессии dev/reviewer в секцию промпта: поля
// трассы (session_id, промпт, вывод) + транскрипт сессии (шаги агента).
// transcripts — sessionID → транскрипт (пустая строка — в сессии нет сообщений);
// отсутствие ключа — транскрипт недоступен (сбой чтения).
func postMortemsText(traces []storage.Trace, transcripts map[string]string) string {
var b strings.Builder
for _, tr := range traces {
b.WriteString("\n=== Агент: " + tr.Agent + " (статус " + string(tr.Status) + ") ===\n")
@@ -88,6 +94,19 @@ func postMortemsText(traces []storage.Trace) string {
b.WriteString(truncateTrace(tr.Output, traceOutputMax))
b.WriteString("\n")
}
if tr.SessionID != "" {
tx, ok := transcripts[tr.SessionID]
b.WriteString("-- Транскрипт сессии (шаги агента) --\n")
switch {
case !ok:
b.WriteString("(транскрипт сессии недоступен)\n")
case strings.TrimSpace(tx) == "":
b.WriteString("(в сессии нет сообщений — агент не сделал ни одного шага)\n")
default:
b.WriteString(truncateTrace(tx, sessionTranscriptMax))
b.WriteString("\n")
}
}
}
if b.Len() == 0 {
return "(сессии dev/reviewer не найдены — вероятна инфраструктурная ошибка до запуска агентов)"
@@ -144,6 +163,23 @@ func (w *Worker) postMortem(ctx context.Context, task *storage.Task) {
}
}
// Достаём транскрипты сессий dev/reviewer (по session_id) — реальные шаги
// агента (сообщения/tool-вызовы), которые не попадают в трассу (там только
// финальный вывод). Сбои чтения не фатальны: промпт соберётся без них.
transcripts := make(map[string]string, len(sessions))
for i := range sessions {
tr := &sessions[i]
if tr.SessionID == "" {
continue
}
tx, tErr := w.Runner.SessionMessages(w.runCtx(ctx, task.ID), w.Worktree, tr.SessionID)
if tErr != nil {
log.Printf("worker: task %d: постмортем: транскрипт %s (%s): %v", task.ID, tr.Agent, tr.SessionID, tErr)
continue
}
transcripts[tr.SessionID] = tx
}
prompt, pErr := RenderPostMortemPrompt(PostMortemPromptData{
Title: task.Title,
Goal: task.Goal,
@@ -151,7 +187,7 @@ func (w *Worker) postMortem(ctx context.Context, task *storage.Task) {
Why: task.Why,
AC: task.AC,
Status: task.Status,
Sessions: postMortemsText(sessions),
Sessions: postMortemsText(sessions, transcripts),
})
if pErr != nil {
log.Printf("worker: task %d: постмортем: рендер промпта: %v", task.ID, pErr)