fix(worker): постмортем получает транскрипт сессий dev/reviewer по session_id
Раньше постмортем-агенту передавался только session_id, без содержимого сессии — анализировать было нечего. Теперь Runner умеет читать сообщения сессии (Messages → renderTranscript) через SessionMessages, а postMortem подставляет реальные шаги агента (сообщения/tool-вызовы) в промпт. Пустая/недоступная сессия помечается явно. Сбои чтения не фатальны.
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user