fix(worker): постмортем получает транскрипт сессий dev/reviewer по session_id
Раньше постмортем-агенту передавался только session_id, без содержимого сессии — анализировать было нечего. Теперь Runner умеет читать сообщения сессии (Messages → renderTranscript) через SessionMessages, а postMortem подставляет реальные шаги агента (сообщения/tool-вызовы) в промпт. Пустая/недоступная сессия помечается явно. Сбои чтения не фатальны.
This commit is contained in:
@@ -59,6 +59,11 @@ type mockRunnerWorker struct {
|
||||
// резюме. postMortemCount — сколько раз постмортем вызывался.
|
||||
postMortemResult *opencode.Result
|
||||
postMortemCount int
|
||||
|
||||
// sessionTranscripts — sessionID → транскрипт для постмортем-агента;
|
||||
// sessionMsgsErr — ошибка чтения транскрипта.
|
||||
sessionTranscripts map[string]string
|
||||
sessionMsgsErr error
|
||||
}
|
||||
|
||||
func (m *mockRunnerWorker) Run(_ context.Context, _, _, agent, _ string) (*opencode.Result, error) {
|
||||
@@ -86,6 +91,13 @@ func (m *mockRunnerWorker) Run(_ context.Context, _, _, agent, _ string) (*openc
|
||||
return m.result, m.err
|
||||
}
|
||||
|
||||
func (m *mockRunnerWorker) SessionMessages(_ context.Context, _, sessionID string) (string, error) {
|
||||
if m.sessionMsgsErr != nil {
|
||||
return "", m.sessionMsgsErr
|
||||
}
|
||||
return m.sessionTranscripts[sessionID], nil
|
||||
}
|
||||
|
||||
// reviewFailedRunner возвращает вердикт not-passed с комментариями.
|
||||
func reviewFailedRunner() *opencode.Result {
|
||||
return &opencode.Result{RC: 0, Stdout: `{"passed":false,"critical_issues":[],"solid_violations":["DIP: высокая связанность"],"comments":["исправь связанность"]}`}
|
||||
@@ -987,7 +999,7 @@ func TestWorkerPostMortemFailureDoesNotChangeStatus(t *testing.T) {
|
||||
|
||||
runner := &mockRunnerWorker{
|
||||
// dev падает при спавне → failed; постмортем тоже падает.
|
||||
err: errors.New("opencode not found"),
|
||||
err: errors.New("opencode not found"),
|
||||
postMortemResult: &opencode.Result{RC: 1, Stdout: ""},
|
||||
}
|
||||
w := &Worker{
|
||||
@@ -1065,13 +1077,66 @@ func TestWorkerPostMortemNotRepeated(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
// TestRenderPostMortemPrompt — промпт постмортема включает задачу и сессии.
|
||||
// TestWorkerPostMortemUsesTranscript — постмортем достаёт транскрипт dev-сессии
|
||||
// (по session_id) и вставляет шаги агента в промпт постмортем-агента.
|
||||
func TestWorkerPostMortemUsesTranscript(t *testing.T) {
|
||||
s := setupWorkerDB(t)
|
||||
task := createReadyTask(t, s, "pm-transcript")
|
||||
|
||||
runner := &mockRunnerWorker{
|
||||
result: &opencode.Result{RC: -1, Stdout: "", SessionID: "sess-dev"},
|
||||
sessionTranscripts: map[string]string{
|
||||
"sess-dev": "[Инструмент]\nчитает requirements.md\n[Ассистент]\nправлю main.go",
|
||||
},
|
||||
}
|
||||
w := &Worker{
|
||||
Store: s,
|
||||
Runner: runner,
|
||||
Agent: "dev",
|
||||
Worktree: t.TempDir(),
|
||||
}
|
||||
seedFakeRepo(t, w.Worktree, "pm-transcript")
|
||||
|
||||
ctx := context.Background()
|
||||
_ = w.runTask(ctx, task)
|
||||
|
||||
task, _ = s.GetTask(ctx, task.ID)
|
||||
if task.Status != storage.StatusTimeout {
|
||||
t.Fatalf("status = %q, want timeout", task.Status)
|
||||
}
|
||||
|
||||
traces, err := s.GetTraces(ctx, task.ID)
|
||||
if err != nil {
|
||||
t.Fatalf("get traces: %v", err)
|
||||
}
|
||||
var pm *storage.Trace
|
||||
for _, tr := range traces {
|
||||
if tr.Agent == postMortemAgent {
|
||||
pm = tr
|
||||
}
|
||||
}
|
||||
if pm == nil {
|
||||
t.Fatal("нет постмортем-трассы")
|
||||
}
|
||||
for _, want := range []string{"Транскрипт сессии", "читает requirements.md", "правлю main.go"} {
|
||||
if !strings.Contains(pm.Prompt, want) {
|
||||
t.Errorf("промпт постмортема не содержит %q", want)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// TestRenderPostMortemPrompt — промпт постмортема включает задачу, сессии и
|
||||
// транскрипт сессий (шаги агентов), а не только финальный вывод.
|
||||
func TestRenderPostMortemPrompt(t *testing.T) {
|
||||
tr := storage.Trace{
|
||||
Agent: "dev",
|
||||
Status: storage.TraceTimeout,
|
||||
Prompt: "промпт dev",
|
||||
Output: "вывод dev",
|
||||
Agent: "dev",
|
||||
Status: storage.TraceTimeout,
|
||||
SessionID: "sess-dev-1",
|
||||
Prompt: "промпт dev",
|
||||
Output: "вывод dev",
|
||||
}
|
||||
transcripts := map[string]string{
|
||||
"sess-dev-1": "[Инструмент]\nпрочитал файл a.go\n[Ассистент]\nправлю код",
|
||||
}
|
||||
prompt, err := RenderPostMortemPrompt(PostMortemPromptData{
|
||||
Title: "Таймаут-задача",
|
||||
@@ -1079,18 +1144,41 @@ func TestRenderPostMortemPrompt(t *testing.T) {
|
||||
Repos: []string{"calc"},
|
||||
AC: "работает",
|
||||
Status: storage.StatusTimeout,
|
||||
Sessions: postMortemsText([]storage.Trace{tr}),
|
||||
Sessions: postMortemsText([]storage.Trace{tr}, transcripts),
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("render: %v", err)
|
||||
}
|
||||
for _, want := range []string{"Таймаут-задача", "timeout", "=== Агент: dev", "промпт dev", "вывод dev"} {
|
||||
for _, want := range []string{
|
||||
"Таймаут-задача", "timeout", "=== Агент: dev", "промпт dev", "вывод dev",
|
||||
"Транскрипт сессии", "прочитал файл a.go", "правлю код",
|
||||
} {
|
||||
if !strings.Contains(prompt, want) {
|
||||
t.Errorf("промпт не содержит %q", want)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// TestPostMortemsTextTranscriptUnavailable — сессия с session_id, для которой
|
||||
// транскрипт не загружен, помечается как недоступный, а не падает.
|
||||
func TestPostMortemsTextTranscriptUnavailable(t *testing.T) {
|
||||
tr := storage.Trace{Agent: "reviewer", Status: storage.TraceFailed, SessionID: "sess-r"}
|
||||
text := postMortemsText([]storage.Trace{tr}, nil)
|
||||
if !strings.Contains(text, "транскрипт сессии недоступен") {
|
||||
t.Errorf("нет пометки о недоступном транскрипте: %q", text)
|
||||
}
|
||||
}
|
||||
|
||||
// TestPostMortemsTextTranscriptEmpty — в сессии нет сообщений: агент не сделал
|
||||
// ни одного шага — это пишется явно, чтобы постмортем не строил догадок.
|
||||
func TestPostMortemsTextTranscriptEmpty(t *testing.T) {
|
||||
tr := storage.Trace{Agent: "dev", Status: storage.TraceTimeout, SessionID: "sess-e"}
|
||||
text := postMortemsText([]storage.Trace{tr}, map[string]string{"sess-e": ""})
|
||||
if !strings.Contains(text, "не сделал ни одного шага") {
|
||||
t.Errorf("нет пометки о пустой сессии: %q", text)
|
||||
}
|
||||
}
|
||||
|
||||
// TestFormatReviewVerdict — человекочитаемое описание вердикта ревьюера.
|
||||
func TestFormatReviewVerdict(t *testing.T) {
|
||||
tests := []struct {
|
||||
|
||||
Reference in New Issue
Block a user