Compare commits
17 Commits
0ad53e338d
...
feat/1b452
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
6c0b903794 | ||
|
|
6546f18348 | ||
|
|
89d571b483 | ||
|
|
ed13386612 | ||
|
|
9cb174a5d3 | ||
|
|
5b43da4564 | ||
|
|
46be641967 | ||
|
|
6fae81bac6 | ||
|
|
8545315f49 | ||
|
|
6586285992 | ||
|
|
94521488b6 | ||
|
|
c9ab753c1e | ||
|
|
88b455203f | ||
|
|
3c51ffee4e | ||
|
|
8c91c83024 | ||
|
|
d731a7429d | ||
|
|
4c0dc80aa8 |
@@ -40,12 +40,13 @@ docs/ ui-spec.md — спека Fyne UI (слои, event-bus, fyne.Do
|
|||||||
- **gitops (worker):** worktree-режим; feature-ветка `feat/<taskTag>` от origin/main; push через http.extraHeader, токен Bearer.
|
- **gitops (worker):** worktree-режим; feature-ветка `feat/<taskTag>` от origin/main; push через http.extraHeader, токен Bearer.
|
||||||
- **Пути «всё рядом с .exe»:** db/worktree резолвятся от ExeDir; config.yaml — рядом с бинарём, фоллбэк cwd.
|
- **Пути «всё рядом с .exe»:** db/worktree резолвятся от ExeDir; config.yaml — рядом с бинарём, фоллбэк cwd.
|
||||||
- **Автообновление:** авто = только Check+уведомление; замена — по /update; версии в `commit-<sha7>/` (не `latest/`); Verify сверяет контрольную сумму бинаря против `.sha256` той же версии. **Perf:** `ResolveLatest` проверяет наличие бинаря версии **HEAD-пробой без скачивания тела** (405/501 → fallback `GET Range: bytes=0-0`), сортировка версий по id убыв.; один общий `http.Client` (keep-alive). Ошибка U4 — только при несовпадении суммы (пустой/отсутствующий `.sha256` пропускает проверку — M1-известное замечание).
|
- **Автообновление:** авто = только Check+уведомление; замена — по /update; версии в `commit-<sha7>/` (не `latest/`); Verify сверяет контрольную сумму бинаря против `.sha256` той же версии. **Perf:** `ResolveLatest` проверяет наличие бинаря версии **HEAD-пробой без скачивания тела** (405/501 → fallback `GET Range: bytes=0-0`), сортировка версий по id убыв.; один общий `http.Client` (keep-alive). Ошибка U4 — только при несовпадении суммы (пустой/отсутствующий `.sha256` пропускает проверку — M1-известное замечание).
|
||||||
|
- **Telegram toggle:** `telegram.enabled` (дефолт true, preset в `Load` до unmarshal, т.к. `applyDefaults` для bool не различает явный false). При `false` канал не создаётся/не крепится в Router и long-poll не стартует — сетевых вызовов к api.telegram.org нет; token/chat_id не валидируются.
|
||||||
|
|
||||||
## opencode (v2 HTTP API, >= 1.18.18)
|
## opencode (v2 HTTP API, >= 1.18.18)
|
||||||
|
|
||||||
- Интеграция с субагентами — через headless `opencode serve`, **v2 API** (`/api/*`). Версия opencode >= 1.18.18.
|
- Интеграция с субагентами — через headless `opencode serve`, **v2 API** (`/api/*`). Версия opencode >= 1.18.18.
|
||||||
- **Хардпин модели:** при создании сессии читается top-level `model` из конфига opencode (`internal/opencode/config.go`, JSONC-стрип) и передаётся в `POST /api/session` как `{"model":{providerID,id}}`.
|
- **Модель — только глобальный конфиг opencode.** Ratatoskr модель не выбирает и про неё не знает: opencode сам берёт модель по умолчанию из своего глобального конфига. Код opencode-конфиг не читает (config.go удалён).
|
||||||
- **О5 WARN (устойчивость к v1-конфигу):** конфиг по старой схеме молча игнорируется v2; провайдер без api → unsupported модели → fallback. Ratatoskr не чинит сам, но логирует warning; фактическая модель ответа сравнивается с ожидаемой. Правильный v2-вид: `api:{type:"aisdk",package,url}`, `request.headers` вместо `options.headers`.
|
- **Свой агент:** при создании сессии в `POST /api/session` передаётся `agent` (analyst/dev/reviewer/chat/postmortem) из встроенных определений. Агенты распаковываются в `<worktree>/.opencode/agent/*.md` (opencode находит через project-каталог `.opencode`); `OPENCODE_CONFIG_DIR` НЕ выставлять — он перенаправляет Global.Path.config и ломает глобальный конфиг (модель/провайдеры) → падение в fallback.
|
||||||
- **Поллинг вердикта:** `POST /api/session/:id/prompt` (durable admit) → `GET /api/session/:id/message?order=desc&limit=200` (новые assistant-сообщения, текст в `content[].type=="text"`) → завершение = `GET /api/session/active` без сессии + финальное assistant-сообщение, стабильное `settlePolls=2` опроса. `POST .../interrupt` вместо abort.
|
- **Поллинг вердикта:** `POST /api/session/:id/prompt` (durable admit) → `GET /api/session/:id/message?order=desc&limit=200` (новые assistant-сообщения, текст в `content[].type=="text"`) → завершение = `GET /api/session/active` без сессии + финальное assistant-сообщение, стабильное `settlePolls=2` опроса. `POST .../interrupt` вместо abort.
|
||||||
|
|
||||||
## Контракты (не ломать)
|
## Контракты (не ломать)
|
||||||
|
|||||||
27
README.md
27
README.md
@@ -95,11 +95,11 @@ make run # build + ./ratatoskr -config config.yaml
|
|||||||
|
|
||||||
```yaml
|
```yaml
|
||||||
telegram:
|
telegram:
|
||||||
token: "..." # TG_TOKEN
|
enabled: true # false — отключить Telegram (без сетевых вызовов к api.telegram.org)
|
||||||
chat_id: "..." # TG_CHAT_ID
|
token: "..." # TG_TOKEN (обязательно, если enabled)
|
||||||
|
chat_id: "..." # TG_CHAT_ID (обязательно, если enabled)
|
||||||
opencode:
|
opencode:
|
||||||
bin: "opencode"
|
bin: "opencode"
|
||||||
config_dir: "./agents" # каталог, куда распаковываются встроенные агенты
|
|
||||||
paths:
|
paths:
|
||||||
db: "./ratatoskr.db" # дефолт; резолвится от каталога бинаря (не от cwd)
|
db: "./ratatoskr.db" # дефолт; резолвится от каталога бинаря (не от cwd)
|
||||||
worktree: "./worktrees" # то же правило «всё рядом с .exe»
|
worktree: "./worktrees" # то же правило «всё рядом с .exe»
|
||||||
@@ -138,18 +138,15 @@ update:
|
|||||||
|
|
||||||
Что делает обёртка (`internal/opencode`):
|
Что делает обёртка (`internal/opencode`):
|
||||||
|
|
||||||
- **Хардпин модели.** При создании сессии в конфиге opencode ищется top-level
|
- **Модель — только глобальный конфиг opencode.** ratatoskr модель не выбирает
|
||||||
`"model"` (`internal/opencode/config.go`) и передаётся в `POST /api/session`
|
и про неё не знает: opencode сам берёт модель по умолчанию из своего
|
||||||
как `{"model":{providerID,id}}`. Это убирает зависимость от fallback-логики
|
глобального конфига (`~/.config/opencode/opencode.jsonc`). Наш код конфиг
|
||||||
opencode (которая молча выбирает «дефолтную» запись, если модель не задана).
|
opencode не читает.
|
||||||
- **Весь код резолва модели устойчив к этому классу проблем (класс O5 WARN):**
|
- **Свой агент.** При создании сессии в `POST /api/session` передаётся имя
|
||||||
- если конфиг не читается / в нём нет `model` — в логи пишется warning;
|
встроенного агента ratatoskr (analyst/dev/reviewer/chat/postmortem). Агенты
|
||||||
- фактическая модель ответа (из финального assistant-сообщения) сравнивается
|
распаковываются в `<worktree>/.opencode/agent/*.md`, где opencode находит
|
||||||
с ожидаемой; расхождение логируется как warning;
|
их через project-каталог `.opencode` (см. `internal/agents`); глобальный
|
||||||
- конфиг, написанный по **старой v1-схеме** (`provider.X.npm` / `options`),
|
конфиг при этом не трогается.
|
||||||
молча игнорируется v2 — обёртка этого не «чинит» сама, но предупреждает.
|
|
||||||
Правильный v2-вид провайдера — `api: { type:"aisdk", package, url }` и
|
|
||||||
`request.headers` вместо `options.headers`.
|
|
||||||
- **Поллинг вердикта.** Промпт отправляется неблокирующе (`POST .../prompt` →
|
- **Поллинг вердикта.** Промпт отправляется неблокирующе (`POST .../prompt` →
|
||||||
durable admit), вердикт собирается из новых assistant-сообщений
|
durable admit), вердикт собирается из новых assistant-сообщений
|
||||||
(`GET .../message`); завершение ответа — сессия ушла из активных дренажей
|
(`GET .../message`); завершение ответа — сессия ушла из активных дренажей
|
||||||
|
|||||||
@@ -1,18 +1,20 @@
|
|||||||
# ratatoskr-go пример конфигурации.
|
# ratatoskr-go пример конфигурации.
|
||||||
# Можно не указывать поля, у которых есть дефолт (см. internal/config/types.go).
|
# Можно не указывать поля, у которых есть дефолт (см. internal/config/types.go).
|
||||||
# Обязательно: telegram.token + telegram.chat_id
|
# Если telegram.enabled: true (дефолт) — обязательно: telegram.token + telegram.chat_id.
|
||||||
|
|
||||||
telegram:
|
telegram:
|
||||||
token: "${TG_TOKEN}" # токен Telegram бота (обязательно)
|
enabled: true # false — полностью отключить Telegram (без сетевых вызовов)
|
||||||
chat_id: "${TG_CHAT_ID}" # ID чата (обязательно)
|
token: "${TG_TOKEN}" # токен Telegram бота (обязательно, если enabled)
|
||||||
|
chat_id: "${TG_CHAT_ID}" # ID чата (обязательно, если enabled)
|
||||||
|
|
||||||
# Всё ниже — опционально, показаны дефолты:
|
# Всё ниже — опционально, показаны дефолты:
|
||||||
# opencode:
|
# opencode:
|
||||||
# bin: "opencode"
|
# bin: "opencode"
|
||||||
# config: "/путь/к/opencode.json" # файл-конфиг модели (OPENCODE_CONFIG)
|
|
||||||
# config_dir: "./agents" # каталог с агентами (OPENCODE_CONFIG_DIR)
|
|
||||||
# hard_timeout: "20m"
|
# hard_timeout: "20m"
|
||||||
# idle_timeout: "5m"
|
# idle_timeout: "5m"
|
||||||
|
#
|
||||||
|
# Агенты (analyst/dev/reviewer/...) распаковываются автоматически в
|
||||||
|
# <worktree>/.opencode/agent/*.md; конфиг opencode для этого не нужен.
|
||||||
|
|
||||||
# git (источник репозиториев для git clone)
|
# git (источник репозиториев для git clone)
|
||||||
# base_url: "http://gitea.hal9000.home" # базовый URL git-хоста (обязательно для клонирования)
|
# base_url: "http://gitea.hal9000.home" # базовый URL git-хоста (обязательно для клонирования)
|
||||||
|
|||||||
@@ -1,12 +1,10 @@
|
|||||||
// Package agents — встроенные агенты opencode (analyst, dev, ...).
|
// Package agents — встроенные агенты opencode (analyst, dev, ...).
|
||||||
//
|
//
|
||||||
// Агенты зашиты в бинарь через go:embed и распаковываются при запуске
|
// Агенты зашиты в бинарь через go:embed и распаковываются при запуске
|
||||||
// в каталог OPENCODE_CONFIG_DIR (по умолчанию ./agents рядом с бинарём),
|
// в project-каталог <worktree>/.opencode/agent/*.md. opencode при serve
|
||||||
// чтобы opencode находил их независимо от того, где запущен ratatoskr.
|
// (cwd = worktree) обнаруживает .opencode поднятием вверх (ConfigPaths) и
|
||||||
//
|
// находит агентов по шаблону {agent,agents}/**/*.md. Глобальный конфиг
|
||||||
// opencode сканирует OPENCODE_CONFIG_DIR как каталог .opencode: агенты
|
// opencode остаётся нетронутым (OPENCODE_CONFIG_DIR не выставляем).
|
||||||
// ищутся в поддиректории agents/ (мн.ч.), а не в корне. Поэтому распаковка
|
|
||||||
// идёт в <dir>/agents/*.md.
|
|
||||||
package agents
|
package agents
|
||||||
|
|
||||||
import (
|
import (
|
||||||
@@ -23,7 +21,7 @@ var FS embed.FS
|
|||||||
// в FS, но для детерминированной распаковки список лучше дополнять здесь.
|
// в FS, но для детерминированной распаковки список лучше дополнять здесь.
|
||||||
var Names = []string{"analyst", "dev", "reviewer", "chat", "postmortem"}
|
var Names = []string{"analyst", "dev", "reviewer", "chat", "postmortem"}
|
||||||
|
|
||||||
// WriteTo распаковывает всех встроенных агентов в каталог dir/agents
|
// WriteTo распаковывает всех встроенных агентов в каталог dir/agent
|
||||||
// (создаёт его). Файлы перезаписываются — встроенная копия всегда актуальна.
|
// (создаёт его). Файлы перезаписываются — встроенная копия всегда актуальна.
|
||||||
func WriteTo(dir string) error {
|
func WriteTo(dir string) error {
|
||||||
agentsDir := filepath.Join(dir, "agents")
|
agentsDir := filepath.Join(dir, "agents")
|
||||||
|
|||||||
@@ -13,7 +13,7 @@ func TestWriteTo(t *testing.T) {
|
|||||||
t.Fatalf("WriteTo: %v", err)
|
t.Fatalf("WriteTo: %v", err)
|
||||||
}
|
}
|
||||||
for _, name := range Names {
|
for _, name := range Names {
|
||||||
// opencode ищет агентов в поддиректории agents/, как у .opencode
|
// opencode ищет агентов в поддиректории agent/ (или agents/) внутри .opencode
|
||||||
p := filepath.Join(dir, "agents", name+".md")
|
p := filepath.Join(dir, "agents", name+".md")
|
||||||
data, err := os.ReadFile(p)
|
data, err := os.ReadFile(p)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|||||||
@@ -2,6 +2,20 @@
|
|||||||
name: analyst
|
name: analyst
|
||||||
description: Аналитик Ratatoskr — помогает сформулировать задачу разработки и возвращает структурированное решение в JSON
|
description: Аналитик Ratatoskr — помогает сформулировать задачу разработки и возвращает структурированное решение в JSON
|
||||||
mode: primary
|
mode: primary
|
||||||
|
tools:
|
||||||
|
read: true
|
||||||
|
write: true
|
||||||
|
edit: false
|
||||||
|
bash: true
|
||||||
|
grep: true
|
||||||
|
glob: true
|
||||||
|
websearch: false
|
||||||
|
webfetch: false
|
||||||
|
permission:
|
||||||
|
question: deny
|
||||||
|
edit: deny
|
||||||
|
bash:
|
||||||
|
"*": allow
|
||||||
---
|
---
|
||||||
|
|
||||||
Ты — аналитик в конвейере Ratatoskr. Пользователь ставит задачу разработки. Ты помогаешь
|
Ты — аналитик в конвейере Ratatoskr. Пользователь ставит задачу разработки. Ты помогаешь
|
||||||
|
|||||||
@@ -2,6 +2,20 @@
|
|||||||
name: dev
|
name: dev
|
||||||
description: Разработчик Ratatoskr — реализует задачу в репозитории по готовым критериям
|
description: Разработчик Ratatoskr — реализует задачу в репозитории по готовым критериям
|
||||||
mode: primary
|
mode: primary
|
||||||
|
tools:
|
||||||
|
read: true
|
||||||
|
write: true
|
||||||
|
edit: true
|
||||||
|
bash: true
|
||||||
|
grep: true
|
||||||
|
glob: true
|
||||||
|
websearch: false
|
||||||
|
webfetch: false
|
||||||
|
permission:
|
||||||
|
question: deny
|
||||||
|
edit: allow
|
||||||
|
bash:
|
||||||
|
"*": allow
|
||||||
---
|
---
|
||||||
|
|
||||||
Ты — dev-агент в конвейере Ratatoskr. Получаешь готовое задание и реализуешь его в коде.
|
Ты — dev-агент в конвейере Ratatoskr. Получаешь готовое задание и реализуешь его в коде.
|
||||||
|
|||||||
@@ -2,6 +2,27 @@
|
|||||||
name: postmortem
|
name: postmortem
|
||||||
description: Постмортем-аналитик Ratatoskr — анализирует сессии dev/reviewer после failed/timeout задачи и даёт резюме: почему так и что сделать, чтобы не повторялось
|
description: Постмортем-аналитик Ratatoskr — анализирует сессии dev/reviewer после failed/timeout задачи и даёт резюме: почему так и что сделать, чтобы не повторялось
|
||||||
mode: primary
|
mode: primary
|
||||||
|
tools:
|
||||||
|
read: false
|
||||||
|
write: false
|
||||||
|
edit: false
|
||||||
|
bash: false
|
||||||
|
grep: false
|
||||||
|
glob: false
|
||||||
|
list: false
|
||||||
|
websearch: false
|
||||||
|
webfetch: false
|
||||||
|
permission:
|
||||||
|
question: deny
|
||||||
|
edit: deny
|
||||||
|
read: deny
|
||||||
|
glob: deny
|
||||||
|
grep: deny
|
||||||
|
list: deny
|
||||||
|
bash:
|
||||||
|
"*": deny
|
||||||
|
webfetch: deny
|
||||||
|
websearch: deny
|
||||||
---
|
---
|
||||||
|
|
||||||
Ты — постмортем-аналитик в конвейере Ratatoskr. Задача завершилась неудачей (failed) или таймаутом (timeout). Ты анализируешь, что пошло не так, и даёшь резюме, которое поможет не допускать этого впредь.
|
Ты — постмортем-аналитик в конвейере Ratatoskr. Задача завершилась неудачей (failed) или таймаутом (timeout). Ты анализируешь, что пошло не так, и даёшь резюме, которое поможет не допускать этого впредь.
|
||||||
|
|||||||
@@ -2,6 +2,20 @@
|
|||||||
name: reviewer
|
name: reviewer
|
||||||
description: Ревьюер Ratatoskr — строго проверяет ветку работы dev-агента (качество, безопасность, SOLID) и возвращает вердикт в JSON
|
description: Ревьюер Ratatoskr — строго проверяет ветку работы dev-агента (качество, безопасность, SOLID) и возвращает вердикт в JSON
|
||||||
mode: primary
|
mode: primary
|
||||||
|
tools:
|
||||||
|
read: true
|
||||||
|
write: true
|
||||||
|
edit: false
|
||||||
|
bash: true
|
||||||
|
grep: true
|
||||||
|
glob: true
|
||||||
|
websearch: false
|
||||||
|
webfetch: false
|
||||||
|
permission:
|
||||||
|
question: deny
|
||||||
|
edit: deny
|
||||||
|
bash:
|
||||||
|
"*": allow
|
||||||
---
|
---
|
||||||
|
|
||||||
Ты — ревьюер в конвейере Ratatoskr. Проверяешь работу dev-агента в feature-ветке **жёстко и придирчиво**. Твоя цель — не дать плохому коду попасть в основную ветку.
|
Ты — ревьюер в конвейере Ratatoskr. Проверяешь работу dev-агента в feature-ветке **жёстко и придирчиво**. Твоя цель — не дать плохому коду попасть в основную ветку.
|
||||||
|
|||||||
@@ -50,16 +50,57 @@ func (a *Analyst) publish(e events.Event) {
|
|||||||
|
|
||||||
// AnalystResponse — структура JSON-ответа аналитика.
|
// AnalystResponse — структура JSON-ответа аналитика.
|
||||||
type AnalystResponse struct {
|
type AnalystResponse struct {
|
||||||
Phase string `json:"phase"`
|
Phase string `json:"phase"`
|
||||||
Title string `json:"title"`
|
Title string `json:"title"`
|
||||||
Goal string `json:"goal"`
|
Goal string `json:"goal"`
|
||||||
Repo string `json:"repo"` // одиночный репо (обратная совместимость)
|
Repo string `json:"repo"` // одиночный репо (обратная совместимость)
|
||||||
Repos []string `json:"repos"` // список репо (основной)
|
Repos json.RawMessage `json:"repos"` // список репо (основной); устойчив к строке
|
||||||
Why string `json:"why"`
|
Why string `json:"why"`
|
||||||
AC string `json:"ac"`
|
AC string `json:"ac"`
|
||||||
Questions []string `json:"questions"`
|
Steps []storage.Step `json:"steps"`
|
||||||
ChatReply string `json:"chat_reply"`
|
Questions []string `json:"questions"`
|
||||||
AbortReason string `json:"abort_reason"`
|
ChatReply string `json:"chat_reply"`
|
||||||
|
AbortReason string `json:"abort_reason"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// reposList нормализует поле repos: модель может вернуть либо массив
|
||||||
|
// ["a","b"], либо строку "a,b" (иногда с пробелами). Пустое значение → nil.
|
||||||
|
func (r *AnalystResponse) reposList() []string {
|
||||||
|
if r.Repos == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
var arr []string
|
||||||
|
if err := json.Unmarshal(r.Repos, &arr); err == nil {
|
||||||
|
return arr
|
||||||
|
}
|
||||||
|
var s string
|
||||||
|
if err := json.Unmarshal(r.Repos, &s); err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
var out []string
|
||||||
|
for _, p := range strings.Split(s, ",") {
|
||||||
|
p = strings.TrimSpace(p)
|
||||||
|
if p != "" {
|
||||||
|
out = append(out, p)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
|
||||||
|
// stepsList возвращает этапы: модель может вернуть массив объектов либо
|
||||||
|
// пустой/отсутствующий — тогда nil. Записи без title отбрасываются.
|
||||||
|
func (r *AnalystResponse) stepsList() []storage.Step {
|
||||||
|
if r.Steps == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
var out []storage.Step
|
||||||
|
for _, st := range r.Steps {
|
||||||
|
if st.Title == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
out = append(out, st)
|
||||||
|
}
|
||||||
|
return out
|
||||||
}
|
}
|
||||||
|
|
||||||
// Decide реализует core.Decider через открытый код.
|
// Decide реализует core.Decider через открытый код.
|
||||||
@@ -85,6 +126,7 @@ func (a *Analyst) Decide(ctx context.Context, history []core.Message, draft stor
|
|||||||
Repos: draft.EffectiveRepos(),
|
Repos: draft.EffectiveRepos(),
|
||||||
Why: draft.Why,
|
Why: draft.Why,
|
||||||
AC: draft.AC,
|
AC: draft.AC,
|
||||||
|
Steps: draft.Steps,
|
||||||
History: hist,
|
History: hist,
|
||||||
Force: force,
|
Force: force,
|
||||||
}
|
}
|
||||||
@@ -104,12 +146,7 @@ func (a *Analyst) Decide(ctx context.Context, history []core.Message, draft stor
|
|||||||
|
|
||||||
// 4. парсим вердикт
|
// 4. парсим вердикт
|
||||||
verdict := opencode.ExtractVerdict(res.Stdout)
|
verdict := opencode.ExtractVerdict(res.Stdout)
|
||||||
log.Printf("analyst: verdict tail: %s", truncate(verdict, 2000))
|
|
||||||
obj, ok := opencode.ExtractJSON(verdict)
|
obj, ok := opencode.ExtractJSON(verdict)
|
||||||
if ok {
|
|
||||||
b, _ := json.Marshal(obj)
|
|
||||||
log.Printf("analyst: extracted json: %s", truncate(string(b), 1000))
|
|
||||||
}
|
|
||||||
if !ok {
|
if !ok {
|
||||||
return core.Decision{}, fmt.Errorf("%w: нет JSON в выводе аналитика", ErrDecodeFail)
|
return core.Decision{}, fmt.Errorf("%w: нет JSON в выводе аналитика", ErrDecodeFail)
|
||||||
}
|
}
|
||||||
@@ -123,6 +160,8 @@ func (a *Analyst) Decide(ctx context.Context, history []core.Message, draft stor
|
|||||||
return core.Decision{}, fmt.Errorf("%w: %v", ErrDecodeFail, err)
|
return core.Decision{}, fmt.Errorf("%w: %v", ErrDecodeFail, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
log.Printf("analyst: вердикт: %s", truncate(formatVerdict(&ar), 2000))
|
||||||
|
|
||||||
// 5. валидация
|
// 5. валидация
|
||||||
if err := validateResponse(&ar); err != nil {
|
if err := validateResponse(&ar); err != nil {
|
||||||
return core.Decision{}, fmt.Errorf("%w: %v", ErrValidation, err)
|
return core.Decision{}, fmt.Errorf("%w: %v", ErrValidation, err)
|
||||||
@@ -144,10 +183,10 @@ func (a *Analyst) Decide(ctx context.Context, history []core.Message, draft stor
|
|||||||
if ar.Repo != "" {
|
if ar.Repo != "" {
|
||||||
dec.Draft.Repo = ar.Repo
|
dec.Draft.Repo = ar.Repo
|
||||||
}
|
}
|
||||||
if len(ar.Repos) > 0 {
|
if repos := ar.reposList(); len(repos) > 0 {
|
||||||
dec.Draft.Repos = ar.Repos
|
dec.Draft.Repos = repos
|
||||||
// Синхронизируем одиночный repo для старых потребителей.
|
// Синхронизируем одиночный repo для старых потребителей.
|
||||||
dec.Draft.Repo = strings.Join(ar.Repos, ",")
|
dec.Draft.Repo = strings.Join(repos, ",")
|
||||||
}
|
}
|
||||||
if ar.Why != "" {
|
if ar.Why != "" {
|
||||||
dec.Draft.Why = ar.Why
|
dec.Draft.Why = ar.Why
|
||||||
@@ -155,6 +194,9 @@ func (a *Analyst) Decide(ctx context.Context, history []core.Message, draft stor
|
|||||||
if ar.AC != "" {
|
if ar.AC != "" {
|
||||||
dec.Draft.AC = ar.AC
|
dec.Draft.AC = ar.AC
|
||||||
}
|
}
|
||||||
|
if steps := ar.stepsList(); len(steps) > 0 {
|
||||||
|
dec.Draft.Steps = steps
|
||||||
|
}
|
||||||
|
|
||||||
return dec, nil
|
return dec, nil
|
||||||
}
|
}
|
||||||
@@ -183,7 +225,7 @@ func validateResponse(ar *AnalystResponse) error {
|
|||||||
return fmt.Errorf("phase=ask, но нет ни chat_reply, ни questions")
|
return fmt.Errorf("phase=ask, но нет ни chat_reply, ни questions")
|
||||||
}
|
}
|
||||||
case "propose":
|
case "propose":
|
||||||
if ar.Title == "" && ar.Goal == "" && len(ar.Repos) == 0 && ar.Why == "" && ar.AC == "" {
|
if ar.Title == "" && ar.Goal == "" && len(ar.Repos) == 0 && ar.Why == "" && ar.AC == "" && len(ar.stepsList()) == 0 {
|
||||||
return fmt.Errorf("phase=propose, но нет ни одного изменённого поля")
|
return fmt.Errorf("phase=propose, но нет ни одного изменённого поля")
|
||||||
}
|
}
|
||||||
case "ready":
|
case "ready":
|
||||||
@@ -194,4 +236,64 @@ func validateResponse(ar *AnalystResponse) error {
|
|||||||
return fmt.Errorf("неизвестный phase=%q", ar.Phase)
|
return fmt.Errorf("неизвестный phase=%q", ar.Phase)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// formatVerdict собирает человекочитаемое однострочное описание вердикта
|
||||||
|
// аналитика (без JSON-разметки) для панели «Логи».
|
||||||
|
func formatVerdict(ar *AnalystResponse) string {
|
||||||
|
var b strings.Builder
|
||||||
|
b.WriteString("phase=" + ar.Phase)
|
||||||
|
|
||||||
|
if ar.ChatReply != "" {
|
||||||
|
b.WriteString(", chat_reply=")
|
||||||
|
b.WriteString(ar.ChatReply)
|
||||||
|
}
|
||||||
|
if len(ar.Questions) > 0 {
|
||||||
|
b.WriteString(", questions=[")
|
||||||
|
b.WriteString(strings.Join(ar.Questions, " | "))
|
||||||
|
b.WriteString("]")
|
||||||
|
}
|
||||||
|
if ar.Title != "" {
|
||||||
|
b.WriteString(", title=")
|
||||||
|
b.WriteString(ar.Title)
|
||||||
|
}
|
||||||
|
if ar.Goal != "" {
|
||||||
|
b.WriteString(", goal=")
|
||||||
|
b.WriteString(ar.Goal)
|
||||||
|
}
|
||||||
|
if ar.Repo != "" {
|
||||||
|
b.WriteString(", repo=")
|
||||||
|
b.WriteString(ar.Repo)
|
||||||
|
}
|
||||||
|
if repos := ar.reposList(); len(repos) > 0 {
|
||||||
|
b.WriteString(", repos=[")
|
||||||
|
b.WriteString(strings.Join(repos, ", "))
|
||||||
|
b.WriteString("]")
|
||||||
|
}
|
||||||
|
if ar.Why != "" {
|
||||||
|
b.WriteString(", why=")
|
||||||
|
b.WriteString(ar.Why)
|
||||||
|
}
|
||||||
|
if ar.AC != "" {
|
||||||
|
b.WriteString(", ac=")
|
||||||
|
b.WriteString(ar.AC)
|
||||||
|
}
|
||||||
|
if steps := ar.stepsList(); len(steps) > 0 {
|
||||||
|
b.WriteString(", steps=[")
|
||||||
|
var parts []string
|
||||||
|
for _, st := range steps {
|
||||||
|
s := st.Title
|
||||||
|
if st.AC != "" {
|
||||||
|
s += " → " + st.AC
|
||||||
|
}
|
||||||
|
parts = append(parts, s)
|
||||||
|
}
|
||||||
|
b.WriteString(strings.Join(parts, " | "))
|
||||||
|
b.WriteString("]")
|
||||||
|
}
|
||||||
|
if ar.AbortReason != "" {
|
||||||
|
b.WriteString(", abort_reason=")
|
||||||
|
b.WriteString(ar.AbortReason)
|
||||||
|
}
|
||||||
|
return b.String()
|
||||||
|
}
|
||||||
|
|||||||
@@ -2,7 +2,9 @@ package analyst
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"encoding/json"
|
||||||
"errors"
|
"errors"
|
||||||
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/kamelion/ratatoskr-go/internal/core"
|
"github.com/kamelion/ratatoskr-go/internal/core"
|
||||||
@@ -22,7 +24,7 @@ func (m *mockRunner) Run(_ context.Context, _, _, _, _ string) (*opencode.Result
|
|||||||
|
|
||||||
func TestDecideAsk(t *testing.T) {
|
func TestDecideAsk(t *testing.T) {
|
||||||
a := &Analyst{Runner: &mockRunner{result: &opencode.Result{
|
a := &Analyst{Runner: &mockRunner{result: &opencode.Result{
|
||||||
RC: 0,
|
RC: 0,
|
||||||
Stdout: `{"type":"text","part":{"text":"{\"phase\":\"ask\",\"chat_reply\":\"Уточню про репозиторий\",\"questions\":[\"Где лежит код?\",\"Какая цель?\"]}"}}`,
|
Stdout: `{"type":"text","part":{"text":"{\"phase\":\"ask\",\"chat_reply\":\"Уточню про репозиторий\",\"questions\":[\"Где лежит код?\",\"Какая цель?\"]}"}}`,
|
||||||
}}, Worktree: "/tmp"}
|
}}, Worktree: "/tmp"}
|
||||||
|
|
||||||
@@ -44,7 +46,7 @@ func TestDecideAsk(t *testing.T) {
|
|||||||
|
|
||||||
func TestDecidePropose(t *testing.T) {
|
func TestDecidePropose(t *testing.T) {
|
||||||
a := &Analyst{Runner: &mockRunner{result: &opencode.Result{
|
a := &Analyst{Runner: &mockRunner{result: &opencode.Result{
|
||||||
RC: 0,
|
RC: 0,
|
||||||
Stdout: `{"type":"text","part":{"text":"{\"phase\":\"propose\",\"title\":\"Калькулятор\",\"goal\":\"Сделать веб-калькулятор\",\"repo\":\"tools/calc\",\"why\":\"Нужен для учёта\",\"ac\":\"Работает + - * /\",\"chat_reply\":\"Готово!\"}"}}`,
|
Stdout: `{"type":"text","part":{"text":"{\"phase\":\"propose\",\"title\":\"Калькулятор\",\"goal\":\"Сделать веб-калькулятор\",\"repo\":\"tools/calc\",\"why\":\"Нужен для учёта\",\"ac\":\"Работает + - * /\",\"chat_reply\":\"Готово!\"}"}}`,
|
||||||
}}, Worktree: "/tmp"}
|
}}, Worktree: "/tmp"}
|
||||||
|
|
||||||
@@ -70,7 +72,7 @@ func TestDecidePropose(t *testing.T) {
|
|||||||
|
|
||||||
func TestDecideAbort(t *testing.T) {
|
func TestDecideAbort(t *testing.T) {
|
||||||
a := &Analyst{Runner: &mockRunner{result: &opencode.Result{
|
a := &Analyst{Runner: &mockRunner{result: &opencode.Result{
|
||||||
RC: 0,
|
RC: 0,
|
||||||
Stdout: `{"type":"text","part":{"text":"{\"phase\":\"abort\",\"chat_reply\":\"Это не про код.\",\"abort_reason\":\"Тема не подходит opencode\"}"}}`,
|
Stdout: `{"type":"text","part":{"text":"{\"phase\":\"abort\",\"chat_reply\":\"Это не про код.\",\"abort_reason\":\"Тема не подходит opencode\"}"}}`,
|
||||||
}}, Worktree: "/tmp"}
|
}}, Worktree: "/tmp"}
|
||||||
|
|
||||||
@@ -90,7 +92,7 @@ func TestDecideAbort(t *testing.T) {
|
|||||||
func TestDecideReady(t *testing.T) {
|
func TestDecideReady(t *testing.T) {
|
||||||
// ready с пустыми изменёнными полями — ВАЛИДНО (черновик готов как есть)
|
// ready с пустыми изменёнными полями — ВАЛИДНО (черновик готов как есть)
|
||||||
a := &Analyst{Runner: &mockRunner{result: &opencode.Result{
|
a := &Analyst{Runner: &mockRunner{result: &opencode.Result{
|
||||||
RC: 0,
|
RC: 0,
|
||||||
Stdout: `{"type":"text","part":{"text":"{\"phase\":\"ready\",\"chat_reply\":\"Черновик готов, запускаю.\"}"}}`,
|
Stdout: `{"type":"text","part":{"text":"{\"phase\":\"ready\",\"chat_reply\":\"Черновик готов, запускаю.\"}"}}`,
|
||||||
}}, Worktree: "/tmp"}
|
}}, Worktree: "/tmp"}
|
||||||
|
|
||||||
@@ -145,7 +147,7 @@ func TestEmptyHistory(t *testing.T) {
|
|||||||
|
|
||||||
func TestForceEmptyHistory(t *testing.T) {
|
func TestForceEmptyHistory(t *testing.T) {
|
||||||
a := &Analyst{Runner: &mockRunner{result: &opencode.Result{
|
a := &Analyst{Runner: &mockRunner{result: &opencode.Result{
|
||||||
RC: 0,
|
RC: 0,
|
||||||
Stdout: `{"type":"text","part":{"text":"{\"phase\":\"ask\",\"chat_reply\":\"Опишите задачу.\"}"}}`,
|
Stdout: `{"type":"text","part":{"text":"{\"phase\":\"ask\",\"chat_reply\":\"Опишите задачу.\"}"}}`,
|
||||||
}}, Worktree: "/tmp"}
|
}}, Worktree: "/tmp"}
|
||||||
|
|
||||||
@@ -157,7 +159,7 @@ func TestForceEmptyHistory(t *testing.T) {
|
|||||||
|
|
||||||
func TestInvalidPhase(t *testing.T) {
|
func TestInvalidPhase(t *testing.T) {
|
||||||
a := &Analyst{Runner: &mockRunner{result: &opencode.Result{
|
a := &Analyst{Runner: &mockRunner{result: &opencode.Result{
|
||||||
RC: 0,
|
RC: 0,
|
||||||
Stdout: `{"type":"text","part":{"text":"{\"phase\":\"unknown\"}"}}`,
|
Stdout: `{"type":"text","part":{"text":"{\"phase\":\"unknown\"}"}}`,
|
||||||
}}, Worktree: "/tmp"}
|
}}, Worktree: "/tmp"}
|
||||||
|
|
||||||
@@ -186,7 +188,7 @@ func TestNonZeroExit(t *testing.T) {
|
|||||||
func TestProposeEmptyFields(t *testing.T) {
|
func TestProposeEmptyFields(t *testing.T) {
|
||||||
// propose без изменённых полей — A3
|
// propose без изменённых полей — A3
|
||||||
a := &Analyst{Runner: &mockRunner{result: &opencode.Result{
|
a := &Analyst{Runner: &mockRunner{result: &opencode.Result{
|
||||||
RC: 0,
|
RC: 0,
|
||||||
Stdout: `{"type":"text","part":{"text":"{\"phase\":\"propose\",\"chat_reply\":\"ok\"}"}}`,
|
Stdout: `{"type":"text","part":{"text":"{\"phase\":\"propose\",\"chat_reply\":\"ok\"}"}}`,
|
||||||
}}, Worktree: "/tmp"}
|
}}, Worktree: "/tmp"}
|
||||||
|
|
||||||
@@ -203,7 +205,7 @@ func TestProposeEmptyFields(t *testing.T) {
|
|||||||
func TestAskEmptyReplyAndQuestions(t *testing.T) {
|
func TestAskEmptyReplyAndQuestions(t *testing.T) {
|
||||||
// ask без chat_reply и questions — A3
|
// ask без chat_reply и questions — A3
|
||||||
a := &Analyst{Runner: &mockRunner{result: &opencode.Result{
|
a := &Analyst{Runner: &mockRunner{result: &opencode.Result{
|
||||||
RC: 0,
|
RC: 0,
|
||||||
Stdout: `{"type":"text","part":{"text":"{\"phase\":\"ask\"}"}}`,
|
Stdout: `{"type":"text","part":{"text":"{\"phase\":\"ask\"}"}}`,
|
||||||
}}, Worktree: "/tmp"}
|
}}, Worktree: "/tmp"}
|
||||||
|
|
||||||
@@ -215,4 +217,134 @@ func TestAskEmptyReplyAndQuestions(t *testing.T) {
|
|||||||
if !errors.Is(err, ErrValidation) {
|
if !errors.Is(err, ErrValidation) {
|
||||||
t.Errorf("err = %v, want A3", err)
|
t.Errorf("err = %v, want A3", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TestDecideProposeStringRepos — модель вернула repos строкой (а не массивом):
|
||||||
|
// парсер должен нормализовать и не падать.
|
||||||
|
func TestDecideProposeStringRepos(t *testing.T) {
|
||||||
|
a := &Analyst{Runner: &mockRunner{result: &opencode.Result{
|
||||||
|
RC: 0,
|
||||||
|
Stdout: `{"type":"text","part":{"text":"{\"phase\":\"propose\",\"title\":\"Калькулятор\",\"repos\":\"tools/calc, tools/ui\"}"}}`,
|
||||||
|
}}, Worktree: "/tmp"}
|
||||||
|
|
||||||
|
history := []core.Message{{Role: "user", Content: "Сделай калькулятор"}}
|
||||||
|
dec, err := a.Decide(context.Background(), history, storage.Task{}, false)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Decide err: %v", err)
|
||||||
|
}
|
||||||
|
if dec.Phase != "propose" {
|
||||||
|
t.Errorf("Phase = %q, want propose", dec.Phase)
|
||||||
|
}
|
||||||
|
if dec.Draft.Repos[0] != "tools/calc" || dec.Draft.Repos[1] != "tools/ui" {
|
||||||
|
t.Errorf("Repos = %#v, want [tools/calc tools/ui]", dec.Draft.Repos)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestDecideProposeSteps — аналитик разложил задачу на этапы с критериями.
|
||||||
|
func TestDecideProposeSteps(t *testing.T) {
|
||||||
|
a := &Analyst{Runner: &mockRunner{result: &opencode.Result{
|
||||||
|
RC: 0,
|
||||||
|
Stdout: `{"type":"text","part":{"text":"{\"phase\":\"propose\",\"title\":\"Калькулятор\",\"steps\":[{\"title\":\"Модель\",\"ac\":\"операции + - * /\"},{\"title\":\"UI\"}]}"}}`,
|
||||||
|
}}, Worktree: "/tmp"}
|
||||||
|
|
||||||
|
history := []core.Message{{Role: "user", Content: "Сделай калькулятор"}}
|
||||||
|
dec, err := a.Decide(context.Background(), history, storage.Task{}, false)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Decide err: %v", err)
|
||||||
|
}
|
||||||
|
if dec.Phase != "propose" {
|
||||||
|
t.Errorf("Phase = %q, want propose", dec.Phase)
|
||||||
|
}
|
||||||
|
if len(dec.Draft.Steps) != 2 {
|
||||||
|
t.Fatalf("len(Steps) = %d, want 2", len(dec.Draft.Steps))
|
||||||
|
}
|
||||||
|
if dec.Draft.Steps[0].Title != "Модель" || dec.Draft.Steps[0].AC != "операции + - * /" {
|
||||||
|
t.Errorf("Steps[0] = %q/%q, want Модель/операции + - * /", dec.Draft.Steps[0].Title, dec.Draft.Steps[0].AC)
|
||||||
|
}
|
||||||
|
if dec.Draft.Steps[1].Title != "UI" || dec.Draft.Steps[1].AC != "" {
|
||||||
|
t.Errorf("Steps[1] = %q/%q, want UI/(пусто)", dec.Draft.Steps[1].Title, dec.Draft.Steps[1].AC)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestProposeOnlyStepsValid — propose меняет только steps → валидно.
|
||||||
|
func TestProposeOnlyStepsValid(t *testing.T) {
|
||||||
|
a := &Analyst{Runner: &mockRunner{result: &opencode.Result{
|
||||||
|
RC: 0,
|
||||||
|
Stdout: `{"type":"text","part":{"text":"{\"phase\":\"propose\",\"steps\":[{\"title\":\"Шаг 1\"}],\"chat_reply\":\"Разбил на этапы\"}"}}`,
|
||||||
|
}}, Worktree: "/tmp"}
|
||||||
|
|
||||||
|
history := []core.Message{{Role: "user", Content: "test"}}
|
||||||
|
dec, err := a.Decide(context.Background(), history, storage.Task{}, false)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Decide err: %v", err)
|
||||||
|
}
|
||||||
|
if dec.Phase != "propose" {
|
||||||
|
t.Errorf("Phase = %q, want propose", dec.Phase)
|
||||||
|
}
|
||||||
|
if len(dec.Draft.Steps) != 1 {
|
||||||
|
t.Fatalf("len(Steps) = %d, want 1", len(dec.Draft.Steps))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestFormatVerdict — человекочитаемое описание вердикта аналитика.
|
||||||
|
func TestFormatVerdict(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
ar *AnalystResponse
|
||||||
|
want []string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "propose with fields",
|
||||||
|
ar: &AnalystResponse{
|
||||||
|
Phase: "propose",
|
||||||
|
Title: "Калькулятор",
|
||||||
|
Goal: "Сделать веб-калькулятор",
|
||||||
|
Repos: json.RawMessage(`["tools/calc","tools/ui"]`),
|
||||||
|
Why: "Нужен для учёта",
|
||||||
|
AC: "Работает + - * /",
|
||||||
|
ChatReply: "Готово!",
|
||||||
|
},
|
||||||
|
want: []string{
|
||||||
|
"phase=propose",
|
||||||
|
"chat_reply=Готово!",
|
||||||
|
"repos=[tools/calc, tools/ui]",
|
||||||
|
"title=Калькулятор",
|
||||||
|
"goal=Сделать веб-калькулятор",
|
||||||
|
"why=Нужен для учёта",
|
||||||
|
"ac=Работает + - * /",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "ask with questions",
|
||||||
|
ar: &AnalystResponse{
|
||||||
|
Phase: "ask",
|
||||||
|
ChatReply: "Уточню",
|
||||||
|
Questions: []string{"Где код?", "Какая цель?"},
|
||||||
|
},
|
||||||
|
want: []string{"phase=ask", "chat_reply=Уточню", "questions=[Где код? | Какая цель?]"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "abort with reason",
|
||||||
|
ar: &AnalystResponse{
|
||||||
|
Phase: "abort",
|
||||||
|
AbortReason: "Тема не про код",
|
||||||
|
},
|
||||||
|
want: []string{"phase=abort", "abort_reason=Тема не про код"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "empty verdict",
|
||||||
|
ar: &AnalystResponse{},
|
||||||
|
want: []string{"phase="},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
got := formatVerdict(tt.ar)
|
||||||
|
for _, w := range tt.want {
|
||||||
|
if !strings.Contains(got, w) {
|
||||||
|
t.Errorf("formatVerdict = %q, want contain %q", got, w)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -3,6 +3,8 @@ package analyst
|
|||||||
import (
|
import (
|
||||||
"strings"
|
"strings"
|
||||||
"text/template"
|
"text/template"
|
||||||
|
|
||||||
|
"github.com/kamelion/ratatoskr-go/internal/storage"
|
||||||
)
|
)
|
||||||
|
|
||||||
// promptTemplate — шаблон промпта для аналитика (opencode analyst-agent).
|
// promptTemplate — шаблон промпта для аналитика (opencode analyst-agent).
|
||||||
@@ -23,6 +25,7 @@ var promptTemplate = template.Must(template.New("analyst").Parse(`Ты — ан
|
|||||||
| repos | список репозиториев (имена на git-хосте; для связанных — все сразу) |
|
| repos | список репозиториев (имена на git-хосте; для связанных — все сразу) |
|
||||||
| why | зачем это нужно, контекст |
|
| why | зачем это нужно, контекст |
|
||||||
| ac | acceptance criteria — конкретный результат, что считается готовым |
|
| ac | acceptance criteria — конкретный результат, что считается готовым |
|
||||||
|
| steps | (опционально) разбиение задачи на этапы: список {title, ac} с критерием готовности каждого этапа |
|
||||||
|
|
||||||
Изменяй в JSON только те поля, которые надо поменять; что менять не надо — пустой строкой.
|
Изменяй в JSON только те поля, которые надо поменять; что менять не надо — пустой строкой.
|
||||||
|
|
||||||
@@ -51,6 +54,12 @@ var promptTemplate = template.Must(template.New("analyst").Parse(`Ты — ан
|
|||||||
{{- else}} repos: (не задано){{end}}
|
{{- else}} repos: (не задано){{end}}
|
||||||
{{if .Why}} why: {{.Why}}{{else}} why: (не задано){{end}}
|
{{if .Why}} why: {{.Why}}{{else}} why: (не задано){{end}}
|
||||||
{{if .AC}} ac: {{.AC}}{{else}} ac: (не задано){{end}}
|
{{if .AC}} ac: {{.AC}}{{else}} ac: (не задано){{end}}
|
||||||
|
{{if .Steps}}
|
||||||
|
steps:
|
||||||
|
{{- range .Steps}}
|
||||||
|
- {{.Title}}{{if .AC}} → {{.AC}}{{end}}
|
||||||
|
{{- end}}
|
||||||
|
{{- else}} steps: (не задано){{end}}
|
||||||
|
|
||||||
**Ответь строго JSON-объектом, без лишнего текста:**
|
**Ответь строго JSON-объектом, без лишнего текста:**
|
||||||
{
|
{
|
||||||
@@ -60,6 +69,7 @@ var promptTemplate = template.Must(template.New("analyst").Parse(`Ты — ан
|
|||||||
"repos": ["имя_репо_1", "имя_репо_2"],
|
"repos": ["имя_репо_1", "имя_репо_2"],
|
||||||
"why": "зачем (только если меняешь)",
|
"why": "зачем (только если меняешь)",
|
||||||
"ac": "критерии (только если меняешь)",
|
"ac": "критерии (только если меняешь)",
|
||||||
|
"steps": [{"title": "этап 1", "ac": "критерий этапа 1"}],
|
||||||
"questions": ["вопрос 1", "вопрос 2"],
|
"questions": ["вопрос 1", "вопрос 2"],
|
||||||
"chat_reply": "твой ответ пользователю (на русском, естественно)",
|
"chat_reply": "твой ответ пользователю (на русском, естественно)",
|
||||||
"abort_reason": "если phase=abort — причина"
|
"abort_reason": "если phase=abort — причина"
|
||||||
@@ -73,6 +83,7 @@ type TemplateData struct {
|
|||||||
Repos []string
|
Repos []string
|
||||||
Why string
|
Why string
|
||||||
AC string
|
AC string
|
||||||
|
Steps []storage.Step
|
||||||
History string // отформатированная переписка
|
History string // отформатированная переписка
|
||||||
Force bool
|
Force bool
|
||||||
}
|
}
|
||||||
@@ -84,4 +95,4 @@ func RenderPrompt(data TemplateData) (string, error) {
|
|||||||
return "", err
|
return "", err
|
||||||
}
|
}
|
||||||
return buf.String(), nil
|
return buf.String(), nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -63,7 +63,7 @@ type App struct {
|
|||||||
CoreCtx *core.Core
|
CoreCtx *core.Core
|
||||||
Worker *worker.Worker
|
Worker *worker.Worker
|
||||||
Updater *update.Updater
|
Updater *update.Updater
|
||||||
tg *telegram.Channel // сохранена для Run
|
tg *telegram.Channel // сохранена для Run (nil, если telegram.enabled=false)
|
||||||
pool *opencode.Pool // пул opencode serve-серверов (API-режим)
|
pool *opencode.Pool // пул opencode serve-серверов (API-режим)
|
||||||
|
|
||||||
// Events — доменная шина UI; LogEvents — шина логов (панель «Логи»).
|
// Events — доменная шина UI; LogEvents — шина логов (панель «Логи»).
|
||||||
@@ -156,8 +156,6 @@ func New(configPath, version, updateToken string, noUI bool) (*App, error) {
|
|||||||
// Служебный root-сервер (worktree) живёт всё время app; остальные лениво.
|
// Служебный root-сервер (worktree) живёт всё время app; остальные лениво.
|
||||||
ocPool := opencode.NewPool(cfg.Paths.Worktree)
|
ocPool := opencode.NewPool(cfg.Paths.Worktree)
|
||||||
ocPool.Bin = cfg.OpenCode.Bin
|
ocPool.Bin = cfg.OpenCode.Bin
|
||||||
ocPool.Config = cfg.OpenCode.Config
|
|
||||||
ocPool.ConfigDir = cfg.OpenCode.ConfigDir
|
|
||||||
ocPool.DBPath = cfg.OpenCode.DBPath
|
ocPool.DBPath = cfg.OpenCode.DBPath
|
||||||
ocPool.Host = cfg.OpenCode.Serve.Hostname
|
ocPool.Host = cfg.OpenCode.Serve.Hostname
|
||||||
ocPool.BasePort = cfg.OpenCode.Serve.Port
|
ocPool.BasePort = cfg.OpenCode.Serve.Port
|
||||||
@@ -224,12 +222,14 @@ func New(configPath, version, updateToken string, noUI bool) (*App, error) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Telegram-канал
|
// Telegram-канал (если включён; иначе сетевые вызовы к Telegram не идут)
|
||||||
tg := telegram.New(cfg.Telegram.Token, cfg.Chat.PollInterval.Duration())
|
if cfg.Telegram.Enabled {
|
||||||
a.tg = tg
|
tg := telegram.New(cfg.Telegram.Token, cfg.Chat.PollInterval.Duration())
|
||||||
if err := router.Attach(tg); err != nil {
|
a.tg = tg
|
||||||
store.Close()
|
if err := router.Attach(tg); err != nil {
|
||||||
return nil, fmt.Errorf("attach telegram: %w", err)
|
store.Close()
|
||||||
|
return nil, fmt.Errorf("attach telegram: %w", err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Worker — polling-планировщик dev-агента
|
// Worker — polling-планировщик dev-агента
|
||||||
@@ -291,15 +291,15 @@ func (a *App) Run(ctx context.Context) error {
|
|||||||
}
|
}
|
||||||
defer a.pool.Close()
|
defer a.pool.Close()
|
||||||
|
|
||||||
// Канал для проверки Telegram-ошибки (горутина оборачивает Run)
|
// Telegram: long-poll цикл в горутине (только если канал включён)
|
||||||
tgErr := make(chan error, 1)
|
tgErr := make(chan error, 1)
|
||||||
|
if a.telegramChannel() != nil {
|
||||||
// Telegram: long-poll цикл в горутине
|
go func() {
|
||||||
go func() {
|
tg := a.telegramChannel()
|
||||||
tg := a.telegramChannel()
|
log.Print("app: telegram poll started")
|
||||||
log.Print("app: telegram poll started")
|
tgErr <- tg.Run(ctx)
|
||||||
tgErr <- tg.Run(ctx)
|
}()
|
||||||
}()
|
}
|
||||||
|
|
||||||
// Worker: poll-цикл (неблокирующий — стартует свою горутину)
|
// Worker: poll-цикл (неблокирующий — стартует свою горутину)
|
||||||
a.Worker.Start(ctx)
|
a.Worker.Start(ctx)
|
||||||
@@ -320,6 +320,23 @@ func (a *App) Run(ctx context.Context) error {
|
|||||||
sigCh := make(chan os.Signal, 1)
|
sigCh := make(chan os.Signal, 1)
|
||||||
signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)
|
signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)
|
||||||
|
|
||||||
|
if a.telegramChannel() != nil {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
log.Print("app: context cancelled")
|
||||||
|
return nil
|
||||||
|
case sig := <-sigCh:
|
||||||
|
log.Printf("app: signal %s — shutting down", sig)
|
||||||
|
cancel()
|
||||||
|
return nil
|
||||||
|
case err := <-tgErr:
|
||||||
|
if err != nil && !errors.Is(err, context.Canceled) {
|
||||||
|
return fmt.Errorf("%w: %v", ErrChannelFatal, err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
select {
|
select {
|
||||||
case <-ctx.Done():
|
case <-ctx.Done():
|
||||||
log.Print("app: context cancelled")
|
log.Print("app: context cancelled")
|
||||||
@@ -328,11 +345,6 @@ func (a *App) Run(ctx context.Context) error {
|
|||||||
log.Printf("app: signal %s — shutting down", sig)
|
log.Printf("app: signal %s — shutting down", sig)
|
||||||
cancel()
|
cancel()
|
||||||
return nil
|
return nil
|
||||||
case err := <-tgErr:
|
|
||||||
if err != nil && !errors.Is(err, context.Canceled) {
|
|
||||||
return fmt.Errorf("%w: %v", ErrChannelFatal, err)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -651,23 +663,21 @@ func (a *App) handleUpdateCmd(ctx context.Context, uid chat.UserID) {
|
|||||||
// сюда не возвращаемся — SwapAndRestart завершил процесс (os.Exit)
|
// сюда не возвращаемся — SwapAndRestart завершил процесс (os.Exit)
|
||||||
}
|
}
|
||||||
|
|
||||||
// ensureAgentsDir определяет каталог с агентами opencode и распаковывает
|
// ensureAgentsDir распаковывает встроенных агентов opencode (go:embed) в
|
||||||
// туда встроенных агентов (go:embed). Если config_dir не задан — использует
|
// project-каталог <worktree>/.opencode/agent/*.md. opencode при serve (cwd =
|
||||||
// ./agents рядом с бинарём. Встроенные агенты перезаписываются (всегда актуальны).
|
// worktree) обнаруживает .opencode поднятием вверх (paths.ts) и сам находит
|
||||||
|
// агентов. Глобальный конфиг opencode (модель/провайдеры) не трогаем и
|
||||||
|
// OPENCODE_CONFIG_DIR не выставляем.
|
||||||
//
|
//
|
||||||
// idempotent: вызывается только из New.
|
// idempotent: вызывается только из New.
|
||||||
func ensureAgentsDir(cfg *config.Config) error {
|
func ensureAgentsDir(cfg *config.Config) error {
|
||||||
if cfg.OpenCode.ConfigDir == "" {
|
if cfg.Paths.Worktree == "" {
|
||||||
exe, err := os.Executable()
|
return fmt.Errorf("paths.worktree не задан — некуда раскладывать агентов")
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("resolve executable: %w", err)
|
|
||||||
}
|
|
||||||
// ./agents рядом с бинарём
|
|
||||||
cfg.OpenCode.ConfigDir = filepath.Join(filepath.Dir(exe), "agents")
|
|
||||||
}
|
}
|
||||||
if err := agents.WriteTo(cfg.OpenCode.ConfigDir); err != nil {
|
base := filepath.Join(cfg.Paths.Worktree, ".opencode")
|
||||||
return fmt.Errorf("write agents to %s: %w", cfg.OpenCode.ConfigDir, err)
|
if err := agents.WriteTo(base); err != nil {
|
||||||
|
return fmt.Errorf("write agents to %s: %w", base, err)
|
||||||
}
|
}
|
||||||
log.Printf("app: agents ensured in %s", cfg.OpenCode.ConfigDir)
|
log.Printf("app: agents ensured in %s", base)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -146,54 +146,6 @@ log:
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestLoad_OpenCodeConfigDir(t *testing.T) {
|
|
||||||
t.Setenv("TG_TOKEN", "tok")
|
|
||||||
t.Setenv("TG_CHAT_ID", "42")
|
|
||||||
|
|
||||||
// без config_dir — пусто
|
|
||||||
yaml := `telegram:
|
|
||||||
token: "${TG_TOKEN}"
|
|
||||||
chat_id: "${TG_CHAT_ID}"
|
|
||||||
`
|
|
||||||
cfg, err := Load(writeCfg(t, yaml))
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("Load: %v", err)
|
|
||||||
}
|
|
||||||
if cfg.OpenCode.ConfigDir != "" {
|
|
||||||
t.Errorf("config_dir = %q, want пусто", cfg.OpenCode.ConfigDir)
|
|
||||||
}
|
|
||||||
|
|
||||||
// через env OPENCODE_CONFIG_DIR
|
|
||||||
t.Setenv("OPENCODE_CONFIG_DIR", "/opt/agents")
|
|
||||||
cfg, err = Load(writeCfg(t, yaml))
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("Load: %v", err)
|
|
||||||
}
|
|
||||||
if cfg.OpenCode.ConfigDir != "/opt/agents" {
|
|
||||||
t.Errorf("config_dir = %q, want /opt/agents", cfg.OpenCode.ConfigDir)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoad_EnvOverride(t *testing.T) {
|
|
||||||
t.Setenv("TG_TOKEN", "tok")
|
|
||||||
t.Setenv("TG_CHAT_ID", "42")
|
|
||||||
// env override для opencode.timeout
|
|
||||||
t.Setenv("OPENCODE_CONFIG", "/custom/opencode.json")
|
|
||||||
|
|
||||||
yaml := `telegram:
|
|
||||||
token: "${TG_TOKEN}"
|
|
||||||
chat_id: "${TG_CHAT_ID}"
|
|
||||||
`
|
|
||||||
path := writeCfg(t, yaml)
|
|
||||||
cfg, err := Load(path)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("Load: %v", err)
|
|
||||||
}
|
|
||||||
if cfg.OpenCode.Config != "/custom/opencode.json" {
|
|
||||||
t.Errorf("opencode config = %q", cfg.OpenCode.Config)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoad_MissingEnvVarUsesDefault(t *testing.T) {
|
func TestLoad_MissingEnvVarUsesDefault(t *testing.T) {
|
||||||
t.Setenv("TG_TOKEN", "tok")
|
t.Setenv("TG_TOKEN", "tok")
|
||||||
t.Setenv("TG_CHAT_ID", "42")
|
t.Setenv("TG_CHAT_ID", "42")
|
||||||
@@ -277,7 +229,7 @@ paths:
|
|||||||
}
|
}
|
||||||
|
|
||||||
func TestLoad_MissingRequired(t *testing.T) {
|
func TestLoad_MissingRequired(t *testing.T) {
|
||||||
// пустой конфиг — все обязательные поля отсутствуют
|
// пустой конфиг — все обязательные поля отсутствуют (telegram.enabled по дефолту true)
|
||||||
t.Setenv("TG_TOKEN", "")
|
t.Setenv("TG_TOKEN", "")
|
||||||
t.Setenv("TG_CHAT_ID", "")
|
t.Setenv("TG_CHAT_ID", "")
|
||||||
|
|
||||||
@@ -298,6 +250,41 @@ func TestLoad_MissingRequired(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestLoad_TelegramDisabledSkipsToken(t *testing.T) {
|
||||||
|
// enabled: false — token/chat_id не обязательны, конфиг валиден
|
||||||
|
t.Setenv("TG_TOKEN", "")
|
||||||
|
t.Setenv("TG_CHAT_ID", "")
|
||||||
|
|
||||||
|
yaml := `telegram:
|
||||||
|
enabled: false
|
||||||
|
token: ""
|
||||||
|
chat_id: ""
|
||||||
|
`
|
||||||
|
path := writeCfg(t, yaml)
|
||||||
|
cfg, err := Load(path)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Load: %v", err)
|
||||||
|
}
|
||||||
|
if cfg.Telegram.Enabled {
|
||||||
|
t.Error("telegram.enabled = true, want false")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestLoad_TelegramEnabledDefaultTrue(t *testing.T) {
|
||||||
|
// enabled не задан — дефолт true, token обязателен
|
||||||
|
t.Setenv("TG_TOKEN", "")
|
||||||
|
t.Setenv("TG_CHAT_ID", "42")
|
||||||
|
|
||||||
|
yaml := `telegram:
|
||||||
|
chat_id: "42"
|
||||||
|
`
|
||||||
|
path := writeCfg(t, yaml)
|
||||||
|
_, err := Load(path)
|
||||||
|
if err == nil || !strings.Contains(err.Error(), "telegram.token") {
|
||||||
|
t.Errorf("ожидалась ошибка telegram.token (enabled по дефолту true), got %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestLoad_InvalidDuration(t *testing.T) {
|
func TestLoad_InvalidDuration(t *testing.T) {
|
||||||
t.Setenv("TG_TOKEN", "x")
|
t.Setenv("TG_TOKEN", "x")
|
||||||
t.Setenv("TG_CHAT_ID", "x")
|
t.Setenv("TG_CHAT_ID", "x")
|
||||||
|
|||||||
@@ -35,6 +35,10 @@ func Load(path string) (*Config, error) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
cfg := &Config{}
|
cfg := &Config{}
|
||||||
|
// Telegram по умолчанию включён (если в YAML не указан enabled).
|
||||||
|
// Делаем до unmarshal: yaml.Unmarshal перезапишет поле, только если ключ
|
||||||
|
// есть в документе, поэтому отсутствующий enabled останется true.
|
||||||
|
cfg.Telegram.Enabled = true
|
||||||
if !noFile {
|
if !noFile {
|
||||||
expanded := os.Expand(string(raw), envLookup)
|
expanded := os.Expand(string(raw), envLookup)
|
||||||
if err := yaml.Unmarshal([]byte(expanded), cfg); err != nil {
|
if err := yaml.Unmarshal([]byte(expanded), cfg); err != nil {
|
||||||
|
|||||||
@@ -76,16 +76,18 @@ type UpdateCfg struct {
|
|||||||
CheckInterval Duration `yaml:"check_interval" default:"24h"`
|
CheckInterval Duration `yaml:"check_interval" default:"24h"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TelegramCfg — настройки Telegram-канала. Enabled=false отключает канал
|
||||||
|
// полностью: бот не опрашивается, сетевые вызовы к api.telegram.org не идут,
|
||||||
|
// token/chat_id не обязательны. По умолчанию включён (см. Load: presetTelegramEnabled).
|
||||||
type TelegramCfg struct {
|
type TelegramCfg struct {
|
||||||
Token string `yaml:"token" env:"TG_TOKEN"`
|
Enabled bool `yaml:"enabled"`
|
||||||
ChatID string `yaml:"chat_id" env:"TG_CHAT_ID"`
|
Token string `yaml:"token" env:"TG_TOKEN"`
|
||||||
|
ChatID string `yaml:"chat_id" env:"TG_CHAT_ID"`
|
||||||
}
|
}
|
||||||
|
|
||||||
type OpenCodeCfg struct {
|
type OpenCodeCfg struct {
|
||||||
Bin string `yaml:"bin" default:"opencode"`
|
Bin string `yaml:"bin" default:"opencode"`
|
||||||
DBPath string `yaml:"db_path" default:""`
|
DBPath string `yaml:"db_path" default:""`
|
||||||
Config string `yaml:"config" env:"OPENCODE_CONFIG"`
|
|
||||||
ConfigDir string `yaml:"config_dir" env:"OPENCODE_CONFIG_DIR"`
|
|
||||||
HardTimeout Duration `yaml:"hard_timeout" default:"20m"`
|
HardTimeout Duration `yaml:"hard_timeout" default:"20m"`
|
||||||
IdleTimeout Duration `yaml:"idle_timeout" default:"5m"`
|
IdleTimeout Duration `yaml:"idle_timeout" default:"5m"`
|
||||||
PollMs Duration `yaml:"poll_ms" default:"2s"`
|
PollMs Duration `yaml:"poll_ms" default:"2s"`
|
||||||
@@ -121,11 +123,13 @@ type PathsCfg struct {
|
|||||||
// (склеенную, все ошибки сразу) или C2 InvalidFormat.
|
// (склеенную, все ошибки сразу) или C2 InvalidFormat.
|
||||||
func (c *Config) Validate() error {
|
func (c *Config) Validate() error {
|
||||||
var errs []error
|
var errs []error
|
||||||
if c.Telegram.Token == "" {
|
if c.Telegram.Enabled {
|
||||||
errs = append(errs, fmt.Errorf("%w: telegram.token", ErrMissingField))
|
if c.Telegram.Token == "" {
|
||||||
}
|
errs = append(errs, fmt.Errorf("%w: telegram.token", ErrMissingField))
|
||||||
if c.Telegram.ChatID == "" {
|
}
|
||||||
errs = append(errs, fmt.Errorf("%w: telegram.chat_id", ErrMissingField))
|
if c.Telegram.ChatID == "" {
|
||||||
|
errs = append(errs, fmt.Errorf("%w: telegram.chat_id", ErrMissingField))
|
||||||
|
}
|
||||||
}
|
}
|
||||||
if !c.Log.Debug() && !strings.EqualFold(c.Log.Level, "info") {
|
if !c.Log.Debug() && !strings.EqualFold(c.Log.Level, "info") {
|
||||||
errs = append(errs, fmt.Errorf("%w: log.level (ожидается \"info\" или \"debug\")", ErrInvalidFormat))
|
errs = append(errs, fmt.Errorf("%w: log.level (ожидается \"info\" или \"debug\")", ErrInvalidFormat))
|
||||||
|
|||||||
@@ -228,6 +228,11 @@ func (c *Core) handleRetry(ctx context.Context, rest string) (Result, error) {
|
|||||||
if err := c.Store.ClearHistory(ctx, id); err != nil {
|
if err := c.Store.ClearHistory(ctx, id); err != nil {
|
||||||
return Result{}, err
|
return Result{}, err
|
||||||
}
|
}
|
||||||
|
// Чистый перезапуск: сбрасываем маркер постмортем-анализа от прошлого
|
||||||
|
// прогона, чтобы на новом failed/timeout постмортем запустился заново.
|
||||||
|
if err := c.Store.DeleteTracesByAgent(ctx, id, "postmortem"); err != nil {
|
||||||
|
return Result{}, err
|
||||||
|
}
|
||||||
return Result{
|
return Result{
|
||||||
Reply: "Задача перезапущена. Опишите, что меняем:",
|
Reply: "Задача перезапущена. Опишите, что меняем:",
|
||||||
TaskID: id,
|
TaskID: id,
|
||||||
|
|||||||
@@ -6,8 +6,8 @@ import (
|
|||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"log"
|
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -16,7 +16,7 @@ import (
|
|||||||
// Пути v2 начинаются с префикса /api (см. README, минимальная версия opencode):
|
// Пути v2 начинаются с префикса /api (см. README, минимальная версия opencode):
|
||||||
// - POST /api/session создать сессию {model:{...}} → {data: Session.Info}
|
// - POST /api/session создать сессию {model:{...}} → {data: Session.Info}
|
||||||
// - POST /api/session/{id}/prompt отправить промпт {prompt:{text}} →
|
// - POST /api/session/{id}/prompt отправить промпт {prompt:{text}} →
|
||||||
// НЕБЛОКИРУЮЩЕ (admit) → {data: Admitted}
|
// НЕБЛОКИРУЮЩЕ (admit) → {data: Admitted}
|
||||||
// - GET /api/session/{id}/message?order=desc → {data:[Message,...]}
|
// - GET /api/session/{id}/message?order=desc → {data:[Message,...]}
|
||||||
// - POST /api/session/{id}/interrupt прервать активный ответ (204)
|
// - POST /api/session/{id}/interrupt прервать активный ответ (204)
|
||||||
// - GET /api/session/active активные дренажи → {data:{sessionID:...}}
|
// - GET /api/session/active активные дренажи → {data:{sessionID:...}}
|
||||||
@@ -24,10 +24,18 @@ import (
|
|||||||
// Prompt не блокирует: вердикт собирается поллингом из content[].type=="text"
|
// Prompt не блокирует: вердикт собирается поллингом из content[].type=="text"
|
||||||
// новых assistant-сообщений (см. Runner.awaitVerdict).
|
// новых assistant-сообщений (см. Runner.awaitVerdict).
|
||||||
type Client struct {
|
type Client struct {
|
||||||
BaseURL string // http://host:port (без завершающего слеша)
|
BaseURL string // http://host:port (без завершающего слеша)
|
||||||
Password string // basic auth (username "opencode")
|
Password string // basic auth (username "opencode")
|
||||||
Debug bool // включать отладочные логи API-вызовов (log.level=debug)
|
Directory string // каталог сервера — идёт заголовком x-opencode-directory
|
||||||
http *http.Client // единый клиент: все операции быстрые (нет блокирующего Send)
|
Debug bool // включать отладочные логи API-вызовов (log.level=debug)
|
||||||
|
http *http.Client // единый клиент: все операции быстрые (нет блокирующего Send)
|
||||||
|
}
|
||||||
|
|
||||||
|
// SessionMessagesURL возвращает URL для просмотра сообщений сессии в браузере.
|
||||||
|
// Сортировка по времени: сначала новые (order=desc). Без credentials — пароль
|
||||||
|
// в лог не попадает.
|
||||||
|
func SessionMessagesURL(baseURL, sessionID string) string {
|
||||||
|
return baseURL + "/api/session/" + sessionID + "/message?order=desc"
|
||||||
}
|
}
|
||||||
|
|
||||||
// ClientErr — классы ошибок клиента.
|
// ClientErr — классы ошибок клиента.
|
||||||
@@ -56,14 +64,15 @@ func (c *Client) do(ctx context.Context, method, path, op string, body []byte) (
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, &ClientErr{Op: "connect", Err: err}
|
return nil, &ClientErr{Op: "connect", Err: err}
|
||||||
}
|
}
|
||||||
|
if body != nil {
|
||||||
|
req.ContentLength = int64(len(body))
|
||||||
|
}
|
||||||
if c.Password != "" {
|
if c.Password != "" {
|
||||||
req.SetBasicAuth("opencode", c.Password)
|
req.SetBasicAuth("opencode", c.Password)
|
||||||
}
|
}
|
||||||
if body != nil {
|
req.Header.Set("Content-Type", "application/json")
|
||||||
req.Header.Set("Content-Type", "application/json")
|
if c.Directory != "" {
|
||||||
}
|
req.Header.Set("x-opencode-directory", c.Directory)
|
||||||
if c.Debug {
|
|
||||||
log.Printf("opencode api debug: %s -> %s %s%s", op, method, c.BaseURL, path)
|
|
||||||
}
|
}
|
||||||
resp, err := c.http.Do(req)
|
resp, err := c.http.Do(req)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -75,14 +84,8 @@ func (c *Client) do(ctx context.Context, method, path, op string, body []byte) (
|
|||||||
return nil, &ClientErr{Op: "connect", Err: err}
|
return nil, &ClientErr{Op: "connect", Err: err}
|
||||||
}
|
}
|
||||||
if resp.StatusCode < 200 || resp.StatusCode > 299 {
|
if resp.StatusCode < 200 || resp.StatusCode > 299 {
|
||||||
if c.Debug {
|
|
||||||
log.Printf("opencode api debug: %s response: status %d", op, resp.StatusCode)
|
|
||||||
}
|
|
||||||
return nil, &ClientErr{Op: op, Err: fmt.Errorf("status %d: %s", resp.StatusCode, truncateStr(string(b), 300))}
|
return nil, &ClientErr{Op: op, Err: fmt.Errorf("status %d: %s", resp.StatusCode, truncateStr(string(b), 300))}
|
||||||
}
|
}
|
||||||
if c.Debug {
|
|
||||||
log.Printf("opencode api debug: %s response (%d bytes)", op, len(b))
|
|
||||||
}
|
|
||||||
return b, nil
|
return b, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -94,7 +97,7 @@ type ModelRef struct {
|
|||||||
Variant string `json:"variant,omitempty"`
|
Variant string `json:"variant,omitempty"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// String возвращает каноничное представление "provider/id[/variant]".
|
// String возвращает каноническое представление "provider/id[/variant]".
|
||||||
func (m *ModelRef) String() string {
|
func (m *ModelRef) String() string {
|
||||||
if m == nil {
|
if m == nil {
|
||||||
return ""
|
return ""
|
||||||
@@ -105,13 +108,14 @@ func (m *ModelRef) String() string {
|
|||||||
return m.ProviderID + "/" + m.ID
|
return m.ProviderID + "/" + m.ID
|
||||||
}
|
}
|
||||||
|
|
||||||
// CreateSession создаёт новую сессию и возвращает её id. model != nil —
|
// CreateSession создаёт новую сессию и возвращает её id. agent — имя агента
|
||||||
// хардпин модели (top-level "model" из конфига opencode), чтобы не зависеть
|
// (analyst/dev/reviewer/chat/postmortem) из встроенных определений ratatoskr.
|
||||||
// от fallback-логики выбора модели в самом opencode.
|
// Модель мы НЕ выбираем: opencode сам берёт модель по умолчанию из своего
|
||||||
func (c *Client) CreateSession(ctx context.Context, model *ModelRef) (string, error) {
|
// глобального конфига (ratatoskr про неё ничего не знает).
|
||||||
|
func (c *Client) CreateSession(ctx context.Context, agent string) (string, error) {
|
||||||
payload := map[string]any{}
|
payload := map[string]any{}
|
||||||
if model != nil {
|
if agent != "" {
|
||||||
payload["model"] = model
|
payload["agent"] = agent
|
||||||
}
|
}
|
||||||
body, _ := json.Marshal(payload)
|
body, _ := json.Marshal(payload)
|
||||||
raw, err := c.do(ctx, http.MethodPost, "/api/session", "create", body)
|
raw, err := c.do(ctx, http.MethodPost, "/api/session", "create", body)
|
||||||
@@ -129,6 +133,12 @@ func (c *Client) CreateSession(ctx context.Context, model *ModelRef) (string, er
|
|||||||
if out.Data.ID == "" {
|
if out.Data.ID == "" {
|
||||||
return "", &ClientErr{Op: "create", Err: fmt.Errorf("пустой id сессии")}
|
return "", &ClientErr{Op: "create", Err: fmt.Errorf("пустой id сессии")}
|
||||||
}
|
}
|
||||||
|
time.Sleep(2 * time.Second)
|
||||||
|
// Холостой вызов /api/model: прогревает сервер (модель/провайдеры),
|
||||||
|
// чтобы первый промпт не спотыкался о «тёплый» старт.
|
||||||
|
_, _ = c.do(ctx, http.MethodGet, "/api/model", "models-warmup", nil)
|
||||||
|
time.Sleep(2 * time.Second)
|
||||||
|
_, _ = c.do(ctx, http.MethodGet, "/api/model", "models-warmup", nil)
|
||||||
return out.Data.ID, nil
|
return out.Data.ID, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -340,6 +350,59 @@ func assistantVerdict(msgs []v2Message, since int64) (texts []string, usedReason
|
|||||||
return reasoning, len(reasoning) > 0
|
return reasoning, len(reasoning) > 0
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// transcriptRole возвращает человекочитаемую подпись роли сообщения сессии.
|
||||||
|
func transcriptRole(t string) string {
|
||||||
|
switch t {
|
||||||
|
case "user":
|
||||||
|
return "Пользователь"
|
||||||
|
case "assistant":
|
||||||
|
return "Ассистент"
|
||||||
|
case "tool":
|
||||||
|
return "Инструмент"
|
||||||
|
case "system":
|
||||||
|
return "Система"
|
||||||
|
default:
|
||||||
|
return t
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// finishLabel — подпись для finish-reason (пусто/: опускаем).
|
||||||
|
func finishLabel(f string) string {
|
||||||
|
if f == "" || f == "stop" {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return f
|
||||||
|
}
|
||||||
|
|
||||||
|
// renderTranscript форматирует сообщения сессии (API отдаёт новыми первыми)
|
||||||
|
// в хронологическом порядке: роль, текст (и tool-вызовы), с обрезкой.
|
||||||
|
func renderTranscript(msgs []v2Message) string {
|
||||||
|
var b strings.Builder
|
||||||
|
for i := len(msgs) - 1; i >= 0; i-- {
|
||||||
|
m := &msgs[i]
|
||||||
|
b.WriteString("\n== " + transcriptRole(m.Type))
|
||||||
|
if m.Type == "assistant" {
|
||||||
|
if m.Model != nil {
|
||||||
|
b.WriteString(" (" + m.Model.String() + ")")
|
||||||
|
}
|
||||||
|
if fl := finishLabel(m.Finish); fl != "" {
|
||||||
|
b.WriteString(" finish=" + fl)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
b.WriteString(" ==\n")
|
||||||
|
for _, p := range m.Content {
|
||||||
|
if p.Text == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
b.WriteString(p.Text)
|
||||||
|
if !strings.HasSuffix(p.Text, "\n") {
|
||||||
|
b.WriteString("\n")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return b.String()
|
||||||
|
}
|
||||||
|
|
||||||
func truncateStr(s string, n int) string {
|
func truncateStr(s string, n int) string {
|
||||||
if len(s) <= n {
|
if len(s) <= n {
|
||||||
return s
|
return s
|
||||||
|
|||||||
@@ -31,7 +31,7 @@ type fakeAPIServer struct {
|
|||||||
verdictReasoning string // завершённый ответ только с reasoning-партом (без text)
|
verdictReasoning string // завершённый ответ только с reasoning-партом (без text)
|
||||||
failCreate bool
|
failCreate bool
|
||||||
failMessages bool
|
failMessages bool
|
||||||
createdModel *ModelRef // модель, полученная на POST /api/session
|
createdAgent string // агент, полученный на POST /api/session
|
||||||
promptCalls int
|
promptCalls int
|
||||||
|
|
||||||
// streamGrow: стрим одного растущего парта — текст/reasoning растёт с
|
// streamGrow: стрим одного растущего парта — текст/reasoning растёт с
|
||||||
@@ -55,10 +55,10 @@ func (f *fakeAPIServer) handler() http.Handler {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
var in struct {
|
var in struct {
|
||||||
Model *ModelRef `json:"model"`
|
Agent string `json:"agent"`
|
||||||
}
|
}
|
||||||
_ = json.NewDecoder(r.Body).Decode(&in)
|
_ = json.NewDecoder(r.Body).Decode(&in)
|
||||||
f.createdModel = in.Model
|
f.createdAgent = in.Agent
|
||||||
f.sessionID = "sess-fake"
|
f.sessionID = "sess-fake"
|
||||||
f.created = true
|
f.created = true
|
||||||
writeJSON(w, map[string]any{"data": map[string]any{"id": "sess-fake"}})
|
writeJSON(w, map[string]any{"data": map[string]any{"id": "sess-fake"}})
|
||||||
@@ -186,33 +186,43 @@ func fakeClient(t *testing.T, f *fakeAPIServer) *Client {
|
|||||||
func TestClient_CreateSession(t *testing.T) {
|
func TestClient_CreateSession(t *testing.T) {
|
||||||
f := &fakeAPIServer{}
|
f := &fakeAPIServer{}
|
||||||
c := fakeClient(t, f)
|
c := fakeClient(t, f)
|
||||||
id, err := c.CreateSession(context.Background(), nil)
|
id, err := c.CreateSession(context.Background(), "dev")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("CreateSession err: %v", err)
|
t.Fatalf("CreateSession err: %v", err)
|
||||||
}
|
}
|
||||||
if id != "sess-fake" {
|
if id != "sess-fake" {
|
||||||
t.Errorf("id = %q, want sess-fake", id)
|
t.Errorf("id = %q, want sess-fake", id)
|
||||||
}
|
}
|
||||||
if f.createdModel != nil {
|
if f.createdAgent != "dev" {
|
||||||
t.Errorf("createdModel = %+v, want nil", f.createdModel)
|
t.Errorf("createdAgent = %q, want dev", f.createdAgent)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestClient_CreateSessionHardpinsModel(t *testing.T) {
|
func TestClient_CreateSessionPassesAgent(t *testing.T) {
|
||||||
want := &ModelRef{ProviderID: "tokentool", ID: "deepseek/deepseek-v4-flash-0731"}
|
|
||||||
f := &fakeAPIServer{}
|
f := &fakeAPIServer{}
|
||||||
c := fakeClient(t, f)
|
c := fakeClient(t, f)
|
||||||
if _, err := c.CreateSession(context.Background(), want); err != nil {
|
if _, err := c.CreateSession(context.Background(), "postmortem"); err != nil {
|
||||||
t.Fatalf("CreateSession err: %v", err)
|
t.Fatalf("CreateSession err: %v", err)
|
||||||
}
|
}
|
||||||
if f.createdModel == nil || f.createdModel.ProviderID != want.ProviderID || f.createdModel.ID != want.ID {
|
if f.createdAgent != "postmortem" {
|
||||||
t.Errorf("createdModel = %+v, want %+v", f.createdModel, want)
|
t.Errorf("createdAgent = %q, want postmortem", f.createdAgent)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestClient_CreateSessionNoAgent(t *testing.T) {
|
||||||
|
f := &fakeAPIServer{}
|
||||||
|
c := fakeClient(t, f)
|
||||||
|
if _, err := c.CreateSession(context.Background(), ""); err != nil {
|
||||||
|
t.Fatalf("CreateSession err: %v", err)
|
||||||
|
}
|
||||||
|
if f.createdAgent != "" {
|
||||||
|
t.Errorf("createdAgent = %q, want пусто", f.createdAgent)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestClient_CreateSessionFail(t *testing.T) {
|
func TestClient_CreateSessionFail(t *testing.T) {
|
||||||
c := fakeClient(t, &fakeAPIServer{failCreate: true})
|
c := fakeClient(t, &fakeAPIServer{failCreate: true})
|
||||||
if _, err := c.CreateSession(context.Background(), nil); err == nil {
|
if _, err := c.CreateSession(context.Background(), "dev"); err == nil {
|
||||||
t.Fatal("CreateSession должен упасть при 500, а не nil")
|
t.Fatal("CreateSession должен упасть при 500, а не nil")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -338,23 +348,19 @@ func Test_assistantVerdict(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func Test_parseModelString(t *testing.T) {
|
|
||||||
m := parseModelString("tokentool/deepseek/deepseek-v4-flash-0731")
|
|
||||||
if m == nil || m.ProviderID != "tokentool" || m.ID != "deepseek/deepseek-v4-flash-0731" {
|
|
||||||
t.Errorf("parse = %+v, want tokentool/deepseek-v4-flash-0731", m)
|
|
||||||
}
|
|
||||||
if parseModelString("onlyprovider") != nil {
|
|
||||||
t.Error("parse без '/' должен вернуть nil")
|
|
||||||
}
|
|
||||||
if parseModelString("") != nil {
|
|
||||||
t.Error("parse пустой должен вернуть nil")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestClientErr_Unwrap(t *testing.T) {
|
func TestClientErr_Unwrap(t *testing.T) {
|
||||||
ce := &ClientErr{Op: "prompt", Err: errors.New("boom")}
|
ce := &ClientErr{Op: "prompt", Err: errors.New("boom")}
|
||||||
var target *ClientErr
|
var target *ClientErr
|
||||||
if !errors.As(ce, &target) {
|
if !errors.As(ce, &target) {
|
||||||
t.Fatal("expected *ClientErr")
|
t.Fatal("expected *ClientErr")
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestSessionMessagesURL — ссылка на сообщения сессии, сортировка по времени (новые сверху).
|
||||||
|
func TestSessionMessagesURL(t *testing.T) {
|
||||||
|
got := SessionMessagesURL("http://127.0.0.1:4101", "sess-abc")
|
||||||
|
want := "http://127.0.0.1:4101/api/session/sess-abc/message?order=desc"
|
||||||
|
if got != want {
|
||||||
|
t.Errorf("SessionMessagesURL() = %q, want %q", got, want)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
@@ -1,176 +0,0 @@
|
|||||||
package opencode
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
|
||||||
"os"
|
|
||||||
"path/filepath"
|
|
||||||
"strings"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Чтение top-level "model" из эффективного конфига opencode.
|
|
||||||
//
|
|
||||||
// Зачем: ratatoskr хардпинит модель в сессии (CreateSession), чтобы не зависеть
|
|
||||||
// от fallback-логики opencode. Если в конфиге модель не задана (или конфиг
|
|
||||||
// написан по старой v1-схеме — npm/options, которые v2 молча игнорирует),
|
|
||||||
// opencode сам выберет «дефолтную» модельную запись, и это может оказаться не
|
|
||||||
// той моделью. Поэтому мы явно логируем предупреждение (класс O5 WARN).
|
|
||||||
|
|
||||||
// opencodeConfigPath определяет путь к конфигу opencode, который видит
|
|
||||||
// serve-процесс этого пула (см. README): (1) явный OPENCODE_CONFIG из Server
|
|
||||||
// или окружения процесса, (2) OPENCODE_CONFIG_DIR / глобальный каталог
|
|
||||||
// ~/.config/opencode. Возвращает "" если ничего не найдено.
|
|
||||||
func opencodeConfigPath(cfgFile, cfgDir string) string {
|
|
||||||
// (1) явный файл конфига — Server.Config или env OPENCODE_CONFIG.
|
|
||||||
p := cfgFile
|
|
||||||
if p == "" {
|
|
||||||
p = os.Getenv("OPENCODE_CONFIG")
|
|
||||||
}
|
|
||||||
if p != "" {
|
|
||||||
if st, err := os.Stat(p); err == nil && !st.IsDir() {
|
|
||||||
return p
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// (2) каталог конфигов.
|
|
||||||
dir := cfgDir
|
|
||||||
if dir == "" {
|
|
||||||
dir = os.Getenv("OPENCODE_CONFIG_DIR")
|
|
||||||
}
|
|
||||||
if dir == "" {
|
|
||||||
home, err := os.UserHomeDir()
|
|
||||||
if err != nil || home == "" {
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
dir = filepath.Join(home, ".config", "opencode")
|
|
||||||
if x := os.Getenv("XDG_CONFIG_HOME"); x != "" {
|
|
||||||
dir = filepath.Join(x, "opencode")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for _, name := range []string{"opencode.json", "opencode.jsonc"} {
|
|
||||||
cand := filepath.Join(dir, name)
|
|
||||||
if st, err := os.Stat(cand); err == nil && !st.IsDir() {
|
|
||||||
return cand
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
|
|
||||||
// ReadModelRef извлекает top-level "model" из конфига opencode и возвращает
|
|
||||||
// его как ModelRef. Модель не задана — вернёт (nil, nil); ошибка чтения/парсинга
|
|
||||||
// возвращается (вызывающий логирует warning и продолжает без хардпина).
|
|
||||||
func ReadModelRef(cfgFile, cfgDir string) (*ModelRef, error) {
|
|
||||||
path := opencodeConfigPath(cfgFile, cfgDir)
|
|
||||||
if path == "" {
|
|
||||||
return nil, nil
|
|
||||||
}
|
|
||||||
b, err := os.ReadFile(path)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("config: читать %s: %w", path, err)
|
|
||||||
}
|
|
||||||
doc := struct {
|
|
||||||
Model json.RawMessage `json:"model"`
|
|
||||||
}{}
|
|
||||||
if err := json.Unmarshal(stripJSONC(b), &doc); err != nil {
|
|
||||||
return nil, fmt.Errorf("config: парсить %s: %w", path, err)
|
|
||||||
}
|
|
||||||
if len(doc.Model) == 0 || strings.TrimSpace(string(doc.Model)) == "null" {
|
|
||||||
return nil, nil
|
|
||||||
}
|
|
||||||
// "model" может быть строкой "provider/id" или объектом {providerID, id}.
|
|
||||||
var s string
|
|
||||||
if err := json.Unmarshal(doc.Model, &s); err == nil {
|
|
||||||
ref := parseModelString(s)
|
|
||||||
if ref == nil {
|
|
||||||
return nil, fmt.Errorf("config: некорректная model %q в %s (ожидается provider/id)", s, path)
|
|
||||||
}
|
|
||||||
return ref, nil
|
|
||||||
}
|
|
||||||
var ref ModelRef
|
|
||||||
if err := json.Unmarshal(doc.Model, &ref); err != nil {
|
|
||||||
return nil, fmt.Errorf("config: некорректная model в %s", path)
|
|
||||||
}
|
|
||||||
if ref.ProviderID == "" || ref.ID == "" {
|
|
||||||
return nil, fmt.Errorf("config: model без providerID/id в %s", path)
|
|
||||||
}
|
|
||||||
return &ref, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// parseModelString разбирает "provider/id" (как ModelV2.parse: провайдер — всё
|
|
||||||
// до первого '/', id — остаток). Возвращает nil при пустой/некорректной строке.
|
|
||||||
func parseModelString(s string) *ModelRef {
|
|
||||||
s = strings.TrimSpace(s)
|
|
||||||
if s == "" {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
i := strings.IndexByte(s, '/')
|
|
||||||
if i <= 0 || i == len(s)-1 {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
return &ModelRef{ProviderID: s[:i], ID: s[i+1:]}
|
|
||||||
}
|
|
||||||
|
|
||||||
// stripJSONC удаляет // и /* */ комментарии (вне строк), сохраняя позиции
|
|
||||||
// переводов строк, чтобы json.Unmarshal не споткнулся о trailing-комма.
|
|
||||||
func stripJSONC(b []byte) []byte {
|
|
||||||
out := make([]byte, 0, len(b))
|
|
||||||
inStr := false
|
|
||||||
esc := false
|
|
||||||
i := 0
|
|
||||||
for i < len(b) {
|
|
||||||
c := b[i]
|
|
||||||
if inStr {
|
|
||||||
out = append(out, c)
|
|
||||||
if esc {
|
|
||||||
esc = false
|
|
||||||
} else if c == '\\' {
|
|
||||||
esc = true
|
|
||||||
} else if c == '"' {
|
|
||||||
inStr = false
|
|
||||||
}
|
|
||||||
i++
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
switch {
|
|
||||||
case c == '"':
|
|
||||||
inStr = true
|
|
||||||
out = append(out, c)
|
|
||||||
i++
|
|
||||||
case c == '/' && i+1 < len(b) && b[i+1] == '/':
|
|
||||||
for i < len(b) && b[i] != '\n' {
|
|
||||||
i++
|
|
||||||
}
|
|
||||||
if i < len(b) {
|
|
||||||
out = append(out, '\n')
|
|
||||||
i++
|
|
||||||
}
|
|
||||||
case c == '/' && i+1 < len(b) && b[i+1] == '*':
|
|
||||||
i += 2
|
|
||||||
for i+1 < len(b) && !(b[i] == '*' && b[i+1] == '/') {
|
|
||||||
i++
|
|
||||||
}
|
|
||||||
i += 2
|
|
||||||
default:
|
|
||||||
out = append(out, c)
|
|
||||||
i++
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return dropTrailingCommas(out)
|
|
||||||
}
|
|
||||||
|
|
||||||
// dropTrailingCommas убирает запятые перед '}' / ']' (допускаются в JSONC).
|
|
||||||
func dropTrailingCommas(b []byte) []byte {
|
|
||||||
out := make([]byte, 0, len(b))
|
|
||||||
for i := 0; i < len(b); i++ {
|
|
||||||
if b[i] == ',' {
|
|
||||||
j := i + 1
|
|
||||||
for j < len(b) && (b[j] == ' ' || b[j] == '\t' || b[j] == '\n' || b[j] == '\r') {
|
|
||||||
j++
|
|
||||||
}
|
|
||||||
if j < len(b) && (b[j] == '}' || b[j] == ']') {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
}
|
|
||||||
out = append(out, b[i])
|
|
||||||
}
|
|
||||||
return out
|
|
||||||
}
|
|
||||||
@@ -1,85 +0,0 @@
|
|||||||
package opencode
|
|
||||||
|
|
||||||
import (
|
|
||||||
"os"
|
|
||||||
"path/filepath"
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestReadModelRef_String(t *testing.T) {
|
|
||||||
dir := t.TempDir()
|
|
||||||
path := filepath.Join(dir, "opencode.jsonc")
|
|
||||||
// конфиг с комментариями и trailing-запятыми (JSONC).
|
|
||||||
src := `{
|
|
||||||
// комментарий
|
|
||||||
"model": "tokentool/deepseek/deepseek-v4-flash-0731", /* и блочный */
|
|
||||||
"provider": {
|
|
||||||
"tokentool": {"api": {"type": "aisdk", "package": "@ai-sdk/openai-compatible", "url": "https://x"}},
|
|
||||||
},
|
|
||||||
}`
|
|
||||||
if err := os.WriteFile(path, []byte(src), 0o644); err != nil {
|
|
||||||
t.Fatalf("write: %v", err)
|
|
||||||
}
|
|
||||||
m, err := ReadModelRef(path, "")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("ReadModelRef err: %v", err)
|
|
||||||
}
|
|
||||||
if m == nil || m.ProviderID != "tokentool" || m.ID != "deepseek/deepseek-v4-flash-0731" {
|
|
||||||
t.Errorf("model = %+v, want tokentool/deepseek-v4-flash-0731", m)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestReadModelRef_Object(t *testing.T) {
|
|
||||||
dir := t.TempDir()
|
|
||||||
path := filepath.Join(dir, "opencode.json")
|
|
||||||
src := `{"model": {"providerID": "tokentool", "id": "deepseek/deepseek-v4-flash-0731"}}`
|
|
||||||
if err := os.WriteFile(path, []byte(src), 0o644); err != nil {
|
|
||||||
t.Fatalf("write: %v", err)
|
|
||||||
}
|
|
||||||
m, err := ReadModelRef(path, "")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("ReadModelRef err: %v", err)
|
|
||||||
}
|
|
||||||
if m == nil || m.ID != "deepseek/deepseek-v4-flash-0731" {
|
|
||||||
t.Errorf("model = %+v, want object-форма", m)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestReadModelRef_Missing(t *testing.T) {
|
|
||||||
dir := t.TempDir()
|
|
||||||
path := filepath.Join(dir, "opencode.json")
|
|
||||||
src := `{"provider": {}}`
|
|
||||||
if err := os.WriteFile(path, []byte(src), 0o644); err != nil {
|
|
||||||
t.Fatalf("write: %v", err)
|
|
||||||
}
|
|
||||||
m, err := ReadModelRef(path, "")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("ReadModelRef err: %v", err)
|
|
||||||
}
|
|
||||||
if m != nil {
|
|
||||||
t.Errorf("model = %+v, want nil (model не задан)", m)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestReadModelRef_NoFile(t *testing.T) {
|
|
||||||
dir := t.TempDir()
|
|
||||||
m, err := ReadModelRef(filepath.Join(dir, "nope.json"), dir)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("ReadModelRef err: %v", err)
|
|
||||||
}
|
|
||||||
if m != nil {
|
|
||||||
t.Errorf("model = %+v, want nil", m)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestReadModelRef_Bad(t *testing.T) {
|
|
||||||
dir := t.TempDir()
|
|
||||||
path := filepath.Join(dir, "opencode.json")
|
|
||||||
src := `{"model": 12345}`
|
|
||||||
if err := os.WriteFile(path, []byte(src), 0o644); err != nil {
|
|
||||||
t.Fatalf("write: %v", err)
|
|
||||||
}
|
|
||||||
if _, err := ReadModelRef(path, ""); err == nil {
|
|
||||||
t.Error("ReadModelRef должен упасть на некорректной model")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -19,8 +19,6 @@ import (
|
|||||||
// директории → каждая сессия API привязана к правильному project-каталогу.
|
// директории → каждая сессия API привязана к правильному project-каталогу.
|
||||||
type Pool struct {
|
type Pool struct {
|
||||||
Bin string
|
Bin string
|
||||||
Config string
|
|
||||||
ConfigDir string
|
|
||||||
DBPath string
|
DBPath string
|
||||||
Host string
|
Host string
|
||||||
BasePort int
|
BasePort int
|
||||||
@@ -72,13 +70,11 @@ func (p *Pool) EnsureRoot(ctx context.Context) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
s := &Server{
|
s := &Server{
|
||||||
Bin: p.Bin,
|
Bin: p.Bin,
|
||||||
Config: p.Config,
|
DBPath: p.DBPath,
|
||||||
ConfigDir: p.ConfigDir,
|
Host: p.Host,
|
||||||
DBPath: p.DBPath,
|
Password: p.Password,
|
||||||
Host: p.Host,
|
Dir: p.rootDir,
|
||||||
Password: p.Password,
|
|
||||||
Dir: p.rootDir,
|
|
||||||
}
|
}
|
||||||
if err := p.assign(s); err != nil {
|
if err := p.assign(s); err != nil {
|
||||||
return err
|
return err
|
||||||
@@ -102,13 +98,11 @@ func (p *Pool) Ensure(ctx context.Context, dir string) (*Server, error) {
|
|||||||
}
|
}
|
||||||
abs := filepath.Clean(dir)
|
abs := filepath.Clean(dir)
|
||||||
s := &Server{
|
s := &Server{
|
||||||
Bin: p.Bin,
|
Bin: p.Bin,
|
||||||
Config: p.Config,
|
DBPath: p.DBPath,
|
||||||
ConfigDir: p.ConfigDir,
|
Host: p.Host,
|
||||||
DBPath: p.DBPath,
|
Password: p.Password,
|
||||||
Host: p.Host,
|
Dir: abs,
|
||||||
Password: p.Password,
|
|
||||||
Dir: abs,
|
|
||||||
}
|
}
|
||||||
if err := p.assign(s); err != nil {
|
if err := p.assign(s); err != nil {
|
||||||
p.mu.Unlock()
|
p.mu.Unlock()
|
||||||
|
|||||||
@@ -73,31 +73,45 @@ func (r *Runner) Run(ctx context.Context, prompt, cwd, agent, sessionID string)
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
c := &Client{BaseURL: srv.Addr(), Password: srv.Password, Debug: r.Debug}
|
c := &Client{BaseURL: srv.Addr(), Password: srv.Password, Directory: srv.Dir, Debug: r.Debug}
|
||||||
|
|
||||||
// Модель по умолчанию из конфига opencode — хардпиним её в сессии, чтобы
|
// Сессия: заданная (resume) или новая. Модель НЕ выбираем — opencode сам
|
||||||
// не зависеть от fallback-логики opencode (класс O5 WARN: если модель не
|
// берёт её по умолчанию из своего глобального конфига.
|
||||||
// считывается/не задана — предупреждаем и работаем без явного указания).
|
|
||||||
model, mErr := ReadModelRef(srv.Config, srv.ConfigDir)
|
|
||||||
if mErr != nil {
|
|
||||||
r.logf("WARN opencode: не удалось прочитать model из конфига: %v", mErr)
|
|
||||||
} else if model == nil {
|
|
||||||
r.logf("WARN opencode: в конфиге opencode не задан top-level model — модель не хардпинится (риск fallback)")
|
|
||||||
} else {
|
|
||||||
r.logf("opencode(%s) model=%s", agent, model)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Сессия: заданная (resume) или новая.
|
|
||||||
sid := sessionID
|
sid := sessionID
|
||||||
if sid == "" {
|
if sid == "" {
|
||||||
sid, err = c.CreateSession(ctx, model)
|
sid, err = c.CreateSession(ctx, agent)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("opencode: create session: %w", err)
|
return nil, fmt.Errorf("opencode: create session: %w", err)
|
||||||
}
|
}
|
||||||
r.logf("opencode(%s) session=%s на %s", agent, sid, srv.Addr())
|
r.logf("opencode(%s) session=%s на %s", agent, sid, srv.Addr())
|
||||||
|
r.logf("opencode(%s) сообщения сессии: %s", agent, SessionMessagesURL(srv.Addr(), sid))
|
||||||
}
|
}
|
||||||
|
|
||||||
return r.awaitVerdict(ctx, c, model, sid, agent, prompt)
|
return r.awaitVerdict(ctx, c, sid, agent, prompt)
|
||||||
|
}
|
||||||
|
|
||||||
|
// SessionMessages возвращает транскрипт сессии (шаги агента: сообщения,
|
||||||
|
// тексты, tool-вызовы) в хронологическом порядке для постмортем-анализа.
|
||||||
|
// sessionID пустой → пустая строка без ошибки. Ошибки чтения возвращаются
|
||||||
|
// как есть — вызывающий (постмортем) решает, логировать и продолжить.
|
||||||
|
func (r *Runner) SessionMessages(ctx context.Context, cwd, sessionID string) (string, error) {
|
||||||
|
r.defaults()
|
||||||
|
if r.Pool == nil {
|
||||||
|
return "", fmt.Errorf("opencode: Pool не задан (API-режим обязателен)")
|
||||||
|
}
|
||||||
|
if sessionID == "" {
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
srv, err := r.Pool.Ensure(ctx, cwd)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
c := &Client{BaseURL: srv.Addr(), Password: srv.Password, Directory: srv.Dir, Debug: r.Debug}
|
||||||
|
msgs, err := c.Messages(ctx, sessionID)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return renderTranscript(msgs), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// settlePolls — сколько подряд опросов должно подтвердить завершение ответа,
|
// settlePolls — сколько подряд опросов должно подтвердить завершение ответа,
|
||||||
@@ -109,7 +123,7 @@ const settlePolls = 2
|
|||||||
// контролируя idle/hard таймауты. Завершение: сессия ушла из активных дренажей
|
// контролируя idle/hard таймауты. Завершение: сессия ушла из активных дренажей
|
||||||
// И есть новое завершённое assistant-сообщение, стабильное в течение settlePolls
|
// И есть новое завершённое assistant-сообщение, стабильное в течение settlePolls
|
||||||
// опросов. Возвращает вердикт (текст text-партов), либо rc=-1 при таймауте.
|
// опросов. Возвращает вердикт (текст text-партов), либо rc=-1 при таймауте.
|
||||||
func (r *Runner) awaitVerdict(ctx context.Context, c *Client, model *ModelRef, sid, agent, prompt string) (*Result, error) {
|
func (r *Runner) awaitVerdict(ctx context.Context, c *Client, sid, agent, prompt string) (*Result, error) {
|
||||||
// admit промпта; граница «новых» сообщений — время создания user-сообщения.
|
// admit промпта; граница «новых» сообщений — время создания user-сообщения.
|
||||||
admittedAt := time.Now().UnixMilli()
|
admittedAt := time.Now().UnixMilli()
|
||||||
adm, err := c.Prompt(ctx, sid, prompt)
|
adm, err := c.Prompt(ctx, sid, prompt)
|
||||||
@@ -187,7 +201,7 @@ func (r *Runner) awaitVerdict(ctx context.Context, c *Client, model *ModelRef, s
|
|||||||
doneSeen++
|
doneSeen++
|
||||||
emptySeen = 0
|
emptySeen = 0
|
||||||
if doneSeen >= settlePolls {
|
if doneSeen >= settlePolls {
|
||||||
return r.verdict(model, cur, msgs, admittedAt, sid)
|
return r.verdict(cur, msgs, admittedAt, sid)
|
||||||
}
|
}
|
||||||
case !active && cur == nil:
|
case !active && cur == nil:
|
||||||
// сессия завершилась, но нового assistant-сообщения так и нет
|
// сессия завершилась, но нового assistant-сообщения так и нет
|
||||||
@@ -207,12 +221,7 @@ func (r *Runner) awaitVerdict(ctx context.Context, c *Client, model *ModelRef, s
|
|||||||
}
|
}
|
||||||
|
|
||||||
// verdict собирает финальный результат из новых assistant-сообщений.
|
// verdict собирает финальный результат из новых assistant-сообщений.
|
||||||
// Проверяет фактическую модель ответа и логирует warning при расхождении
|
func (r *Runner) verdict(cur *v2Message, msgs []v2Message, since int64, sid string) (*Result, error) {
|
||||||
// с ожидаемой (устойчивость к «не той» модели — класс O5 WARN).
|
|
||||||
func (r *Runner) verdict(model *ModelRef, cur *v2Message, msgs []v2Message, since int64, sid string) (*Result, error) {
|
|
||||||
if model != nil && cur.Model != nil && (model.ProviderID != cur.Model.ProviderID || model.ID != cur.Model.ID) {
|
|
||||||
r.logf("WARN opencode: сессия %s отвечала моделью %s, а не ожидаемой %s — проверь providers в конфиге (v2-схема: provider.api / request, а не npm/options)", sid, cur.Model, model)
|
|
||||||
}
|
|
||||||
if cur.Error != nil && cur.Error.Message != "" {
|
if cur.Error != nil && cur.Error.Message != "" {
|
||||||
return nil, &ClientErr{Op: "prompt", Err: errors.New(cur.Error.Message)}
|
return nil, &ClientErr{Op: "prompt", Err: errors.New(cur.Error.Message)}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -10,8 +10,6 @@ import (
|
|||||||
|
|
||||||
// fakePool создаёт Pool, в котором уже «живёт» сервер для каталога (без spawn):
|
// fakePool создаёт Pool, в котором уже «живёт» сервер для каталога (без spawn):
|
||||||
// Server{URL: fake.URL}, поэтому Runner ходит по HTTP на фейк-API.
|
// Server{URL: fake.URL}, поэтому Runner ходит по HTTP на фейк-API.
|
||||||
// XDG_CONFIG_HOME уводится во временный каталог, чтобы ReadModelRef не читал
|
|
||||||
// реальный пользовательский конфиг opencode (детерминизм тестов).
|
|
||||||
func fakePool(t *testing.T, f *fakeAPIServer, dir string) (*Pool, *Client) {
|
func fakePool(t *testing.T, f *fakeAPIServer, dir string) (*Pool, *Client) {
|
||||||
t.Helper()
|
t.Helper()
|
||||||
ts := httptestURL(t, f)
|
ts := httptestURL(t, f)
|
||||||
@@ -148,7 +146,7 @@ func TestRun_ContextCancel(t *testing.T) {
|
|||||||
done <- res
|
done <- res
|
||||||
errCh <- err
|
errCh <- err
|
||||||
}()
|
}()
|
||||||
time.Sleep(30 * time.Millisecond)
|
time.Sleep(6000 * time.Millisecond)
|
||||||
cancel()
|
cancel()
|
||||||
res := <-done
|
res := <-done
|
||||||
if err := <-errCh; err != nil {
|
if err := <-errCh; err != nil {
|
||||||
|
|||||||
@@ -8,6 +8,7 @@ import (
|
|||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
"os/exec"
|
"os/exec"
|
||||||
|
"path/filepath"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
@@ -24,8 +25,6 @@ import (
|
|||||||
// отдаёт URL, процессом не владеет.
|
// отдаёт URL, процессом не владеет.
|
||||||
type Server struct {
|
type Server struct {
|
||||||
Bin string // путь к opencode (по умолчанию "opencode")
|
Bin string // путь к opencode (по умолчанию "opencode")
|
||||||
Config string // OPENCODE_CONFIG
|
|
||||||
ConfigDir string // OPENCODE_CONFIG_DIR
|
|
||||||
DBPath string // рабочая БД сервера (передам env, если задана)
|
DBPath string // рабочая БД сервера (передам env, если задана)
|
||||||
|
|
||||||
Host string // hostname для прослушивания
|
Host string // hostname для прослушивания
|
||||||
@@ -33,6 +32,12 @@ type Server struct {
|
|||||||
Password string // basic auth (если непустой — сервер защищён)
|
Password string // basic auth (если непустой — сервер защищён)
|
||||||
Dir string // каталог, в котором запускается serve (project сервера)
|
Dir string // каталог, в котором запускается serve (project сервера)
|
||||||
|
|
||||||
|
// ConfigPath — путь к глобальному конфигу opencode (opencode.jsonc).
|
||||||
|
// Передаётся serve через env OPENCODE_CONFIG, чтобы гарантированно
|
||||||
|
// подхватить модель/провайдеров независимо от резолва глобального пути.
|
||||||
|
// Пусто — serve резолвит конфиг сам (см. defaults).
|
||||||
|
ConfigPath string
|
||||||
|
|
||||||
// URL задаёт внешний сервер. Пусто — супервайзер владеет процессом.
|
// URL задаёт внешний сервер. Пусто — супервайзер владеет процессом.
|
||||||
URL string
|
URL string
|
||||||
PollInterval time.Duration // как часто проверять /global/health
|
PollInterval time.Duration // как часто проверять /global/health
|
||||||
@@ -62,6 +67,20 @@ func (s *Server) defaults() {
|
|||||||
if s.Stdout == nil {
|
if s.Stdout == nil {
|
||||||
s.Stdout = os.Stderr
|
s.Stdout = os.Stderr
|
||||||
}
|
}
|
||||||
|
if s.ConfigPath == "" {
|
||||||
|
s.ConfigPath = defaultOpenCodeConfigPath()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// defaultOpenCodeConfigPath возвращает путь к глобальному конфигу opencode,
|
||||||
|
// совпадающий с тем, что opencode загружает по умолчанию: ~/.config/opencode/
|
||||||
|
// (в т.ч. на Windows — см. Global.Path.config в исходниках opencode).
|
||||||
|
func defaultOpenCodeConfigPath() string {
|
||||||
|
home, _ := os.UserHomeDir()
|
||||||
|
if home == "" {
|
||||||
|
home = "."
|
||||||
|
}
|
||||||
|
return filepath.Join(home, ".config", "opencode", "opencode.jsonc")
|
||||||
}
|
}
|
||||||
|
|
||||||
// baseURL собирает полный адрес сервера (http://host:port).
|
// baseURL собирает полный адрес сервера (http://host:port).
|
||||||
@@ -101,6 +120,8 @@ func (s *Server) Start(ctx context.Context) error {
|
|||||||
s.done = done
|
s.done = done
|
||||||
s.mu.Unlock()
|
s.mu.Unlock()
|
||||||
|
|
||||||
|
log.Printf("opencode serve: spawn dir=%s HOME=%q USERPROFILE=%q OPENCODE_CONFIG=%q (serve наследует окружение родителя)", cmd.Dir, os.Getenv("HOME"), os.Getenv("USERPROFILE"), s.ConfigPath)
|
||||||
|
|
||||||
if err := cmd.Start(); err != nil {
|
if err := cmd.Start(); err != nil {
|
||||||
return fmt.Errorf("opencode serve: start %v: %w", s.Bin, err)
|
return fmt.Errorf("opencode serve: start %v: %w", s.Bin, err)
|
||||||
}
|
}
|
||||||
@@ -124,22 +145,8 @@ func (s *Server) serveCmd(ctx context.Context) *exec.Cmd {
|
|||||||
// Своя process-group: чтобы killGroup (по pgid) убивал только сервер и его
|
// Своя process-group: чтобы killGroup (по pgid) убивал только сервер и его
|
||||||
// дочерние процессы, а не чужой процесс (например, тест-реннер).
|
// дочерние процессы, а не чужой процесс (например, тест-реннер).
|
||||||
setpgid(cmd)
|
setpgid(cmd)
|
||||||
env := append(os.Environ(),
|
// Env не задаём: serve наследует окружение родителя и сам резолвит свой
|
||||||
"OPENCODE_DISABLE_AUTOUPDATE=1",
|
// глобальный/локальный конфиг (модель/провайдеры, агентов, БД) как обычно.
|
||||||
"OPENCODE_DISABLE_MODELS_FETCH=1")
|
|
||||||
if s.Config != "" {
|
|
||||||
env = append(env, "OPENCODE_CONFIG="+s.Config)
|
|
||||||
}
|
|
||||||
if s.ConfigDir != "" {
|
|
||||||
env = append(env, "OPENCODE_CONFIG_DIR="+s.ConfigDir)
|
|
||||||
}
|
|
||||||
if s.DBPath != "" {
|
|
||||||
env = append(env, "OPENCODE_DB="+s.DBPath)
|
|
||||||
}
|
|
||||||
if s.Password != "" {
|
|
||||||
env = append(env, "OPENCODE_SERVER_PASSWORD="+s.Password)
|
|
||||||
}
|
|
||||||
cmd.Env = env
|
|
||||||
return cmd
|
return cmd
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -198,4 +198,28 @@ func atoiOrZero(s string) int {
|
|||||||
n = n*10 + int(c-'0')
|
n = n*10 + int(c-'0')
|
||||||
}
|
}
|
||||||
return n
|
return n
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestServeCmd_NoCustomEnv(t *testing.T) {
|
||||||
|
// serveCmd не должен задавать свой Env: serve наследует окружение родителя
|
||||||
|
// и сам резолвит конфиг/БД/пароль как обычно.
|
||||||
|
s := &Server{}
|
||||||
|
s.defaults()
|
||||||
|
cmd := s.serveCmd(context.Background())
|
||||||
|
if cmd.Env != nil {
|
||||||
|
for _, kv := range cmd.Env {
|
||||||
|
if strings.HasPrefix(kv, "OPENCODE_") {
|
||||||
|
t.Errorf("serveCmd: не ожидал env %q", kv)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestServerCmd_CustomOpenCodeConfig(t *testing.T) {
|
||||||
|
// явно заданный путь переопределяет дефолтный
|
||||||
|
s := &Server{ConfigPath: `C:\custom\opencode.jsonc`}
|
||||||
|
s.defaults()
|
||||||
|
if s.ConfigPath != `C:\custom\opencode.jsonc` {
|
||||||
|
t.Errorf("ConfigPath = %q, want explicit", s.ConfigPath)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
@@ -14,17 +14,17 @@ type Status = model.Status
|
|||||||
|
|
||||||
// Статусы задачи — re-export из model.
|
// Статусы задачи — re-export из model.
|
||||||
const (
|
const (
|
||||||
StatusDraft = model.StatusDraft
|
StatusDraft = model.StatusDraft
|
||||||
StatusCollecting = model.StatusCollecting
|
StatusCollecting = model.StatusCollecting
|
||||||
StatusReady = model.StatusReady
|
StatusReady = model.StatusReady
|
||||||
StatusApproved = model.StatusApproved
|
StatusApproved = model.StatusApproved
|
||||||
StatusRunning = model.StatusRunning
|
StatusRunning = model.StatusRunning
|
||||||
StatusSuccess = model.StatusSuccess
|
StatusSuccess = model.StatusSuccess
|
||||||
StatusFailed = model.StatusFailed
|
StatusFailed = model.StatusFailed
|
||||||
StatusTimeout = model.StatusTimeout
|
StatusTimeout = model.StatusTimeout
|
||||||
StatusCancelled = model.StatusCancelled
|
StatusCancelled = model.StatusCancelled
|
||||||
StatusAborted = model.StatusAborted
|
StatusAborted = model.StatusAborted
|
||||||
StatusClosed = model.StatusClosed
|
StatusClosed = model.StatusClosed
|
||||||
)
|
)
|
||||||
|
|
||||||
// AllStatuses — все возможные статусы для валидации.
|
// AllStatuses — все возможные статусы для валидации.
|
||||||
@@ -40,17 +40,24 @@ func IsTerminal(s Status) bool {
|
|||||||
return model.IsTerminal(s)
|
return model.IsTerminal(s)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Step — этап задачи с собственным критерием готовности.
|
||||||
|
type Step struct {
|
||||||
|
Title string `json:"title"`
|
||||||
|
AC string `json:"ac"` // acceptance criterion этапа
|
||||||
|
}
|
||||||
|
|
||||||
// Task — запись задачи в БД.
|
// Task — запись задачи в БД.
|
||||||
type Task struct {
|
type Task struct {
|
||||||
ID int64 `json:"id"`
|
ID int64 `json:"id"`
|
||||||
ChatID string `json:"chat_id"` // tg://<id>
|
ChatID string `json:"chat_id"` // tg://<id>
|
||||||
Title string `json:"title"`
|
Title string `json:"title"`
|
||||||
Goal string `json:"goal"`
|
Goal string `json:"goal"`
|
||||||
Repo string `json:"repo"` // обратная совместимость: одиночный репозиторий
|
Repo string `json:"repo"` // обратная совместимость: одиночный репозиторий
|
||||||
Repos []string `json:"repos"` // список репозиториев (основной)
|
Repos []string `json:"repos"` // список репозиториев (основной)
|
||||||
Why string `json:"why"`
|
Why string `json:"why"`
|
||||||
AC string `json:"ac"` // acceptance criteria
|
AC string `json:"ac"` // acceptance criteria
|
||||||
TaskTag string `json:"task_tag"` // UUID, стабильный на всю жизнь
|
Steps []Step `json:"steps"` // этапы задачи (опционально)
|
||||||
|
TaskTag string `json:"task_tag"` // UUID, стабильный на всю жизнь
|
||||||
Status Status `json:"status"`
|
Status Status `json:"status"`
|
||||||
CreatedAt SQLiteTime `json:"created_at"`
|
CreatedAt SQLiteTime `json:"created_at"`
|
||||||
UpdatedAt SQLiteTime `json:"updated_at"`
|
UpdatedAt SQLiteTime `json:"updated_at"`
|
||||||
@@ -88,6 +95,25 @@ func (t *Task) SetReposFromDB(repos string) {
|
|||||||
_ = json.Unmarshal([]byte(repos), &t.Repos)
|
_ = json.Unmarshal([]byte(repos), &t.Repos)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// StepsJoined возвращает steps как одну строку (JSON-массив) для хранения в БД.
|
||||||
|
// Пустой список → пустая строка.
|
||||||
|
func (t *Task) StepsJoined() string {
|
||||||
|
if len(t.Steps) == 0 {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
b, _ := json.Marshal(t.Steps)
|
||||||
|
return string(b)
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetStepsFromDB заполняет Steps из сохранённой строки (JSON).
|
||||||
|
func (t *Task) SetStepsFromDB(steps string) {
|
||||||
|
if steps == "" {
|
||||||
|
t.Steps = nil
|
||||||
|
return
|
||||||
|
}
|
||||||
|
_ = json.Unmarshal([]byte(steps), &t.Steps)
|
||||||
|
}
|
||||||
|
|
||||||
// TraceStatus — алиас доменного статуса трассировки.
|
// TraceStatus — алиас доменного статуса трассировки.
|
||||||
type TraceStatus = model.TraceStatus
|
type TraceStatus = model.TraceStatus
|
||||||
|
|
||||||
@@ -105,7 +131,7 @@ type Trace struct {
|
|||||||
Agent string `json:"agent"` // analyst | researcher | dev | reviewer
|
Agent string `json:"agent"` // analyst | researcher | dev | reviewer
|
||||||
SessionID string `json:"session_id"` // opencode session_id
|
SessionID string `json:"session_id"` // opencode session_id
|
||||||
Prompt string `json:"prompt"`
|
Prompt string `json:"prompt"`
|
||||||
Output string `json:"output"` // полный NDJSON или summary
|
Output string `json:"output"` // полный NDJSON или summary
|
||||||
Status TraceStatus `json:"status"`
|
Status TraceStatus `json:"status"`
|
||||||
StartedAt SQLiteTime `json:"started_at"`
|
StartedAt SQLiteTime `json:"started_at"`
|
||||||
FinishedAt NullSQLiteTime `json:"finished_at,omitempty"`
|
FinishedAt NullSQLiteTime `json:"finished_at,omitempty"`
|
||||||
@@ -113,8 +139,8 @@ type Trace struct {
|
|||||||
|
|
||||||
// TaskFilter — параметры фильтрации списка задач.
|
// TaskFilter — параметры фильтрации списка задач.
|
||||||
type TaskFilter struct {
|
type TaskFilter struct {
|
||||||
ChatID string
|
ChatID string
|
||||||
Status Status
|
Status Status
|
||||||
Limit int
|
Limit int
|
||||||
Offset int
|
Offset int
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -173,6 +173,13 @@ func (s *Storage) migrate(ctx context.Context) error {
|
|||||||
return fmt.Errorf("%w: migrate add repos: %w", ErrDB, err)
|
return fmt.Errorf("%w: migrate add repos: %w", ErrDB, err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
// Доп. колонка steps (этапы задачи). Idempotent.
|
||||||
|
if _, err := s.db.ExecContext(ctx,
|
||||||
|
`ALTER TABLE tasks ADD COLUMN steps TEXT NOT NULL DEFAULT ''`); err != nil {
|
||||||
|
if !isDuplicateColumn(err) {
|
||||||
|
return fmt.Errorf("%w: migrate add steps: %w", ErrDB, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -185,4 +192,4 @@ func isDuplicateColumn(err error) bool {
|
|||||||
// Now возвращает текущее время UTC как SQLiteTime.
|
// Now возвращает текущее время UTC как SQLiteTime.
|
||||||
func Now() SQLiteTime {
|
func Now() SQLiteTime {
|
||||||
return SQLiteTime(time.Now().UTC())
|
return SQLiteTime(time.Now().UTC())
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -150,6 +150,44 @@ func TestUpdateTaskNotFound(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestCreateAndGetTask_Steps(t *testing.T) {
|
||||||
|
s, ctx := setupTestDB(t)
|
||||||
|
task := &Task{
|
||||||
|
ChatID: "tg://steps",
|
||||||
|
Title: "Steps task",
|
||||||
|
TaskTag: "steps-1",
|
||||||
|
Steps: []Step{
|
||||||
|
{Title: "Реализовать модель", AC: "структура готова"},
|
||||||
|
{Title: "Добавить API", AC: "эндпоинт отвечает"},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
id, err := s.CreateTask(ctx, task)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("CreateTask: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
got, err := s.GetTask(ctx, id)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("GetTask: %v", err)
|
||||||
|
}
|
||||||
|
if len(got.Steps) != 2 {
|
||||||
|
t.Fatalf("steps len = %d, want 2", len(got.Steps))
|
||||||
|
}
|
||||||
|
if got.Steps[0].Title != "Реализовать модель" || got.Steps[0].AC != "структура готова" {
|
||||||
|
t.Fatalf("steps[0] = %q / %q, want модель / структура готова", got.Steps[0].Title, got.Steps[0].AC)
|
||||||
|
}
|
||||||
|
|
||||||
|
// апдейт этапов
|
||||||
|
got.Steps = append(got.Steps, Step{Title: "Ревью", AC: "пройден review"})
|
||||||
|
if err := s.UpdateTask(ctx, got); err != nil {
|
||||||
|
t.Fatalf("UpdateTask steps: %v", err)
|
||||||
|
}
|
||||||
|
got2, _ := s.GetTask(ctx, id)
|
||||||
|
if len(got2.Steps) != 3 {
|
||||||
|
t.Fatalf("steps len after update = %d, want 3", len(got2.Steps))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestListTasks(t *testing.T) {
|
func TestListTasks(t *testing.T) {
|
||||||
s, ctx := setupTestDB(t)
|
s, ctx := setupTestDB(t)
|
||||||
for i := 0; i < 5; i++ {
|
for i := 0; i < 5; i++ {
|
||||||
@@ -264,6 +302,26 @@ func TestUpdateTraceOutput(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TestDeleteTracesByAgent — удаление трасс по агенту (сброс маркера при /retry).
|
||||||
|
func TestDeleteTracesByAgent(t *testing.T) {
|
||||||
|
s, ctx := setupTestDB(t)
|
||||||
|
task := &Task{ChatID: "tg://del", TaskTag: "delete-by-agent"}
|
||||||
|
id, _ := s.CreateTask(ctx, task)
|
||||||
|
|
||||||
|
_, _ = s.AppendTrace(ctx, &Trace{TaskID: id, Agent: "dev"})
|
||||||
|
_, _ = s.AppendTrace(ctx, &Trace{TaskID: id, Agent: "postmortem"})
|
||||||
|
_, _ = s.AppendTrace(ctx, &Trace{TaskID: id, Agent: "postmortem"})
|
||||||
|
|
||||||
|
if err := s.DeleteTracesByAgent(ctx, id, "postmortem"); err != nil {
|
||||||
|
t.Fatalf("DeleteTracesByAgent: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
traces, _ := s.GetTraces(ctx, id)
|
||||||
|
if len(traces) != 1 || traces[0].Agent != "dev" {
|
||||||
|
t.Fatalf("traces = %d (agent %q), want только dev", len(traces), traces[0].Agent)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// хелперы для проверки классов ошибок
|
// хелперы для проверки классов ошибок
|
||||||
func IsNotFound(err error) bool {
|
func IsNotFound(err error) bool {
|
||||||
return errors.Is(err, ErrNotFound)
|
return errors.Is(err, ErrNotFound)
|
||||||
|
|||||||
@@ -30,10 +30,10 @@ func (s *Storage) CreateTask(ctx context.Context, t *Task) (int64, error) {
|
|||||||
}
|
}
|
||||||
now := Now()
|
now := Now()
|
||||||
res, err := s.db.ExecContext(ctx, `
|
res, err := s.db.ExecContext(ctx, `
|
||||||
INSERT INTO tasks (chat_id, title, goal, repo, repos, why, ac, task_tag, status, created_at, updated_at)
|
INSERT INTO tasks (chat_id, title, goal, repo, repos, why, ac, steps, task_tag, status, created_at, updated_at)
|
||||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
|
||||||
t.ChatID, t.Title, t.Goal, t.Repo, t.ReposJoined(), t.Why, t.AC,
|
t.ChatID, t.Title, t.Goal, t.Repo, t.ReposJoined(), t.Why, t.AC,
|
||||||
t.TaskTag, StatusDraft, now, now,
|
t.StepsJoined(), t.TaskTag, StatusDraft, now, now,
|
||||||
)
|
)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return 0, fmt.Errorf("%w: create task: %w", ErrDB, err)
|
return 0, fmt.Errorf("%w: create task: %w", ErrDB, err)
|
||||||
@@ -53,11 +53,12 @@ func (s *Storage) CreateTask(ctx context.Context, t *Task) (int64, error) {
|
|||||||
func (s *Storage) GetTask(ctx context.Context, id int64) (*Task, error) {
|
func (s *Storage) GetTask(ctx context.Context, id int64) (*Task, error) {
|
||||||
t := &Task{}
|
t := &Task{}
|
||||||
var reposStr string
|
var reposStr string
|
||||||
|
var stepsStr string
|
||||||
err := s.db.QueryRowContext(ctx, `
|
err := s.db.QueryRowContext(ctx, `
|
||||||
SELECT id, chat_id, title, goal, repo, repos, why, ac, task_tag, status, created_at, updated_at
|
SELECT id, chat_id, title, goal, repo, repos, why, ac, steps, task_tag, status, created_at, updated_at
|
||||||
FROM tasks WHERE id = ?`, id).Scan(
|
FROM tasks WHERE id = ?`, id).Scan(
|
||||||
&t.ID, &t.ChatID, &t.Title, &t.Goal, &t.Repo, &reposStr,
|
&t.ID, &t.ChatID, &t.Title, &t.Goal, &t.Repo, &reposStr,
|
||||||
&t.Why, &t.AC, &t.TaskTag, &t.Status, &t.CreatedAt, &t.UpdatedAt,
|
&t.Why, &t.AC, &stepsStr, &t.TaskTag, &t.Status, &t.CreatedAt, &t.UpdatedAt,
|
||||||
)
|
)
|
||||||
if err == sql.ErrNoRows {
|
if err == sql.ErrNoRows {
|
||||||
return nil, fmt.Errorf("%w: task %d", ErrNotFound, id)
|
return nil, fmt.Errorf("%w: task %d", ErrNotFound, id)
|
||||||
@@ -66,6 +67,7 @@ func (s *Storage) GetTask(ctx context.Context, id int64) (*Task, error) {
|
|||||||
return nil, fmt.Errorf("%w: get task %d: %w", ErrDB, id, err)
|
return nil, fmt.Errorf("%w: get task %d: %w", ErrDB, id, err)
|
||||||
}
|
}
|
||||||
t.SetReposFromDB(reposStr)
|
t.SetReposFromDB(reposStr)
|
||||||
|
t.SetStepsFromDB(stepsStr)
|
||||||
return t, nil
|
return t, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -94,9 +96,9 @@ func (s *Storage) UpdateTask(ctx context.Context, t *Task) error {
|
|||||||
now := Now()
|
now := Now()
|
||||||
res, err := s.db.ExecContext(ctx, `
|
res, err := s.db.ExecContext(ctx, `
|
||||||
UPDATE tasks
|
UPDATE tasks
|
||||||
SET title=?, goal=?, repo=?, repos=?, why=?, ac=?, status=?, updated_at=?
|
SET title=?, goal=?, repo=?, repos=?, why=?, ac=?, steps=?, status=?, updated_at=?
|
||||||
WHERE id=?`,
|
WHERE id=?`,
|
||||||
t.Title, t.Goal, t.Repo, t.ReposJoined(), t.Why, t.AC, t.Status, now, t.ID,
|
t.Title, t.Goal, t.Repo, t.ReposJoined(), t.Why, t.AC, t.StepsJoined(), t.Status, now, t.ID,
|
||||||
)
|
)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("%w: update task %d: %w", ErrDB, t.ID, err)
|
return fmt.Errorf("%w: update task %d: %w", ErrDB, t.ID, err)
|
||||||
@@ -114,13 +116,14 @@ func (s *Storage) UpdateTask(ctx context.Context, t *Task) error {
|
|||||||
func (s *Storage) GetActiveTaskByChatID(ctx context.Context, chatID string) (*Task, error) {
|
func (s *Storage) GetActiveTaskByChatID(ctx context.Context, chatID string) (*Task, error) {
|
||||||
t := &Task{}
|
t := &Task{}
|
||||||
var reposStr string
|
var reposStr string
|
||||||
|
var stepsStr string
|
||||||
err := s.db.QueryRowContext(ctx, `
|
err := s.db.QueryRowContext(ctx, `
|
||||||
SELECT id, chat_id, title, goal, repo, repos, why, ac, task_tag, status, created_at, updated_at
|
SELECT id, chat_id, title, goal, repo, repos, why, ac, steps, task_tag, status, created_at, updated_at
|
||||||
FROM tasks
|
FROM tasks
|
||||||
WHERE chat_id = ? AND status NOT IN ('success','cancelled','aborted','closed')
|
WHERE chat_id = ? AND status NOT IN ('success','cancelled','aborted','closed')
|
||||||
ORDER BY updated_at DESC LIMIT 1`, chatID).Scan(
|
ORDER BY updated_at DESC LIMIT 1`, chatID).Scan(
|
||||||
&t.ID, &t.ChatID, &t.Title, &t.Goal, &t.Repo, &reposStr,
|
&t.ID, &t.ChatID, &t.Title, &t.Goal, &t.Repo, &reposStr,
|
||||||
&t.Why, &t.AC, &t.TaskTag, &t.Status, &t.CreatedAt, &t.UpdatedAt)
|
&t.Why, &t.AC, &stepsStr, &t.TaskTag, &t.Status, &t.CreatedAt, &t.UpdatedAt)
|
||||||
if err == sql.ErrNoRows {
|
if err == sql.ErrNoRows {
|
||||||
return nil, fmt.Errorf("%w: no active task for chat %s", ErrNotFound, chatID)
|
return nil, fmt.Errorf("%w: no active task for chat %s", ErrNotFound, chatID)
|
||||||
}
|
}
|
||||||
@@ -128,6 +131,7 @@ func (s *Storage) GetActiveTaskByChatID(ctx context.Context, chatID string) (*Ta
|
|||||||
return nil, fmt.Errorf("%w: get active task %s: %w", ErrDB, chatID, err)
|
return nil, fmt.Errorf("%w: get active task %s: %w", ErrDB, chatID, err)
|
||||||
}
|
}
|
||||||
t.SetReposFromDB(reposStr)
|
t.SetReposFromDB(reposStr)
|
||||||
|
t.SetStepsFromDB(stepsStr)
|
||||||
return t, nil
|
return t, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -149,7 +153,7 @@ func (s *Storage) ListTasks(ctx context.Context, filter TaskFilter) ([]*Task, er
|
|||||||
args = append(args, filter.Limit, filter.Offset)
|
args = append(args, filter.Limit, filter.Offset)
|
||||||
|
|
||||||
rows, err := s.db.QueryContext(ctx, `
|
rows, err := s.db.QueryContext(ctx, `
|
||||||
SELECT id, chat_id, title, goal, repo, repos, why, ac, task_tag, status, created_at, updated_at
|
SELECT id, chat_id, title, goal, repo, repos, why, ac, steps, task_tag, status, created_at, updated_at
|
||||||
FROM tasks WHERE `+where+` ORDER BY updated_at DESC LIMIT ? OFFSET ?`, args...)
|
FROM tasks WHERE `+where+` ORDER BY updated_at DESC LIMIT ? OFFSET ?`, args...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("%w: list tasks: %w", ErrDB, err)
|
return nil, fmt.Errorf("%w: list tasks: %w", ErrDB, err)
|
||||||
@@ -160,11 +164,13 @@ func (s *Storage) ListTasks(ctx context.Context, filter TaskFilter) ([]*Task, er
|
|||||||
for rows.Next() {
|
for rows.Next() {
|
||||||
t := &Task{}
|
t := &Task{}
|
||||||
var reposStr string
|
var reposStr string
|
||||||
|
var stepsStr string
|
||||||
if err := rows.Scan(&t.ID, &t.ChatID, &t.Title, &t.Goal, &t.Repo, &reposStr,
|
if err := rows.Scan(&t.ID, &t.ChatID, &t.Title, &t.Goal, &t.Repo, &reposStr,
|
||||||
&t.Why, &t.AC, &t.TaskTag, &t.Status, &t.CreatedAt, &t.UpdatedAt); err != nil {
|
&t.Why, &t.AC, &stepsStr, &t.TaskTag, &t.Status, &t.CreatedAt, &t.UpdatedAt); err != nil {
|
||||||
return nil, fmt.Errorf("%w: scan task: %w", ErrDB, err)
|
return nil, fmt.Errorf("%w: scan task: %w", ErrDB, err)
|
||||||
}
|
}
|
||||||
t.SetReposFromDB(reposStr)
|
t.SetReposFromDB(reposStr)
|
||||||
|
t.SetStepsFromDB(stepsStr)
|
||||||
tasks = append(tasks, t)
|
tasks = append(tasks, t)
|
||||||
}
|
}
|
||||||
return tasks, rows.Err()
|
return tasks, rows.Err()
|
||||||
|
|||||||
@@ -107,6 +107,17 @@ func (s *Storage) GetLatestTrace(ctx context.Context, taskID int64, agent string
|
|||||||
return tr, nil
|
return tr, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// DeleteTracesByAgent удаляет все трассы задачи с заданным агентом
|
||||||
|
// (сброс маркера при /retry N: чистый перезапуск без наследия анализа).
|
||||||
|
func (s *Storage) DeleteTracesByAgent(ctx context.Context, taskID int64, agent string) error {
|
||||||
|
_, err := s.db.ExecContext(ctx,
|
||||||
|
`DELETE FROM traces WHERE task_id=? AND agent=?`, taskID, agent)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("%w: delete traces agent %s for task %d: %w", ErrDB, agent, taskID, err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
// DeleteTrace удаляет трассу. Только для тестов/админки.
|
// DeleteTrace удаляет трассу. Только для тестов/админки.
|
||||||
func (s *Storage) DeleteTrace(ctx context.Context, id int64) error {
|
func (s *Storage) DeleteTrace(ctx context.Context, id int64) error {
|
||||||
res, err := s.db.ExecContext(ctx, `DELETE FROM traces WHERE id = ?`, id)
|
res, err := s.db.ExecContext(ctx, `DELETE FROM traces WHERE id = ?`, id)
|
||||||
|
|||||||
@@ -18,6 +18,9 @@ const postMortemAgent = "postmortem"
|
|||||||
// не превращался в полные транскрипты и не переполнял контекст модели).
|
// не превращался в полные транскрипты и не переполнял контекст модели).
|
||||||
const traceOutputMax = 6000
|
const traceOutputMax = 6000
|
||||||
|
|
||||||
|
// sessionTranscriptMax — обрезка транскрипта сессии dev/reviewer в промпте.
|
||||||
|
const sessionTranscriptMax = 20000
|
||||||
|
|
||||||
// postMortemPromptTemplate — промпт для постмортем-агента после failed/timeout:
|
// postMortemPromptTemplate — промпт для постмортем-агента после failed/timeout:
|
||||||
// задача + сессии dev/reviewer. Ожидается резюме простым текстом на русском.
|
// задача + сессии dev/reviewer. Ожидается резюме простым текстом на русском.
|
||||||
var postMortemPromptTemplate = template.Must(template.New("postmortem").Parse(`Ты — постмортем-аналитик в конвейере Ratatoskr. Задача завершилась неудачей ({{.Status}}). Проанализируй сессии агентов 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
|
return buf.String(), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// postMortemsText форматирует сессии dev/reviewer в секцию промпта.
|
// postMortemsText форматирует сессии dev/reviewer в секцию промпта: поля
|
||||||
func postMortemsText(traces []storage.Trace) string {
|
// трассы (session_id, промпт, вывод) + транскрипт сессии (шаги агента).
|
||||||
|
// transcripts — sessionID → транскрипт (пустая строка — в сессии нет сообщений);
|
||||||
|
// отсутствие ключа — транскрипт недоступен (сбой чтения).
|
||||||
|
func postMortemsText(traces []storage.Trace, transcripts map[string]string) string {
|
||||||
var b strings.Builder
|
var b strings.Builder
|
||||||
for _, tr := range traces {
|
for _, tr := range traces {
|
||||||
b.WriteString("\n=== Агент: " + tr.Agent + " (статус " + string(tr.Status) + ") ===\n")
|
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(truncateTrace(tr.Output, traceOutputMax))
|
||||||
b.WriteString("\n")
|
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 {
|
if b.Len() == 0 {
|
||||||
return "(сессии dev/reviewer не найдены — вероятна инфраструктурная ошибка до запуска агентов)"
|
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{
|
prompt, pErr := RenderPostMortemPrompt(PostMortemPromptData{
|
||||||
Title: task.Title,
|
Title: task.Title,
|
||||||
Goal: task.Goal,
|
Goal: task.Goal,
|
||||||
@@ -151,7 +187,7 @@ func (w *Worker) postMortem(ctx context.Context, task *storage.Task) {
|
|||||||
Why: task.Why,
|
Why: task.Why,
|
||||||
AC: task.AC,
|
AC: task.AC,
|
||||||
Status: task.Status,
|
Status: task.Status,
|
||||||
Sessions: postMortemsText(sessions),
|
Sessions: postMortemsText(sessions, transcripts),
|
||||||
})
|
})
|
||||||
if pErr != nil {
|
if pErr != nil {
|
||||||
log.Printf("worker: task %d: постмортем: рендер промпта: %v", task.ID, pErr)
|
log.Printf("worker: task %d: постмортем: рендер промпта: %v", task.ID, pErr)
|
||||||
@@ -186,6 +222,7 @@ func (w *Worker) postMortem(ctx context.Context, task *storage.Task) {
|
|||||||
output = "(постмортем-агент не вернул текст)"
|
output = "(постмортем-агент не вернул текст)"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
log.Printf("worker: task %d: постмортем: %s", task.ID, truncateTrace(output, 2000))
|
||||||
w.finalizeTrace(ctx, traceID, status, output)
|
w.finalizeTrace(ctx, traceID, status, output)
|
||||||
|
|
||||||
if res.RC == 0 && output != "" {
|
if res.RC == 0 && output != "" {
|
||||||
|
|||||||
@@ -3,6 +3,8 @@ package worker
|
|||||||
import (
|
import (
|
||||||
"strings"
|
"strings"
|
||||||
"text/template"
|
"text/template"
|
||||||
|
|
||||||
|
"github.com/kamelion/ratatoskr-go/internal/storage"
|
||||||
)
|
)
|
||||||
|
|
||||||
// devPromptTemplate — промпт для dev-агента при запуске задачи.
|
// devPromptTemplate — промпт для dev-агента при запуске задачи.
|
||||||
@@ -20,6 +22,12 @@ var devPromptTemplate = template.Must(template.New("dev").Parse(`Ты — dev-а
|
|||||||
{{if .Why}}Зачем: {{.Why}}{{end}}
|
{{if .Why}}Зачем: {{.Why}}{{end}}
|
||||||
{{if .AC}}Критерии готовности:
|
{{if .AC}}Критерии готовности:
|
||||||
{{.AC}}{{end}}
|
{{.AC}}{{end}}
|
||||||
|
{{if .Steps}}
|
||||||
|
**Этапы (выполняй по порядку, у каждого свой критерий готовности):**
|
||||||
|
{{- range .Steps}}
|
||||||
|
{{.Title}}{{if .AC}} — готово, когда: {{.AC}}{{end}}
|
||||||
|
{{- end}}
|
||||||
|
{{end}}
|
||||||
|
|
||||||
**Инструкции:**
|
**Инструкции:**
|
||||||
1. Рабочий каталог — общий корень, в котором лежат все репозитории по именам.
|
1. Рабочий каталог — общий корень, в котором лежат все репозитории по именам.
|
||||||
@@ -38,6 +46,7 @@ type DevPromptData struct {
|
|||||||
Repos []string
|
Repos []string
|
||||||
Why string
|
Why string
|
||||||
AC string
|
AC string
|
||||||
|
Steps []storage.Step
|
||||||
|
|
||||||
// ReviewFeedback — замечания ревьюера при повторном прогоне dev
|
// ReviewFeedback — замечания ревьюера при повторном прогоне dev
|
||||||
// (не пусто → dev должен исправить именно это).
|
// (не пусто → dev должен исправить именно это).
|
||||||
@@ -124,4 +133,4 @@ func reviewFeedbackList(branch string, comments []string) []string {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
return []string{s}
|
return []string{s}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -4,6 +4,7 @@ import (
|
|||||||
"context"
|
"context"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"log"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
@@ -96,6 +97,10 @@ func (w *Worker) runReviewer(ctx context.Context, taskID int64, cwd, prompt stri
|
|||||||
}
|
}
|
||||||
w.finalizeTrace(ctx, traceID, ts, output)
|
w.finalizeTrace(ctx, traceID, ts, output)
|
||||||
|
|
||||||
|
if v != nil {
|
||||||
|
log.Printf("reviewer: вердикт: %s", formatReviewVerdict(v))
|
||||||
|
}
|
||||||
|
|
||||||
if v == nil {
|
if v == nil {
|
||||||
// Невалидный/пустой вывод или RC!=0: не парсится — сигнал на retry.
|
// Невалидный/пустой вывод или RC!=0: не парсится — сигнал на retry.
|
||||||
return nil, output, traceID, nil
|
return nil, output, traceID, nil
|
||||||
@@ -119,3 +124,35 @@ func parseReviewVerdict(out string) (*reviewVerdict, error) {
|
|||||||
}
|
}
|
||||||
return &v, nil
|
return &v, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// formatReviewVerdict собирает человекочитаемое однострочное описание вердикта
|
||||||
|
// ревьюера (без JSON-разметки) для панели «Логи».
|
||||||
|
func formatReviewVerdict(v *reviewVerdict) string {
|
||||||
|
var b strings.Builder
|
||||||
|
b.WriteString("passed=")
|
||||||
|
b.WriteString(boolElse(v.Passed, "yes", "no"))
|
||||||
|
if len(v.CriticalIssues) > 0 {
|
||||||
|
b.WriteString(", critical=[")
|
||||||
|
b.WriteString(strings.Join(v.CriticalIssues, " | "))
|
||||||
|
b.WriteString("]")
|
||||||
|
}
|
||||||
|
if len(v.SolidViolations) > 0 {
|
||||||
|
b.WriteString(", solid=[")
|
||||||
|
b.WriteString(strings.Join(v.SolidViolations, " | "))
|
||||||
|
b.WriteString("]")
|
||||||
|
}
|
||||||
|
if len(v.Comments) > 0 {
|
||||||
|
b.WriteString(", comments=[")
|
||||||
|
b.WriteString(strings.Join(v.Comments, " | "))
|
||||||
|
b.WriteString("]")
|
||||||
|
}
|
||||||
|
return b.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// boolElse возвращает строку для логического значения (без вызова тернарника).
|
||||||
|
func boolElse(cond bool, yes, no string) string {
|
||||||
|
if cond {
|
||||||
|
return yes
|
||||||
|
}
|
||||||
|
return no
|
||||||
|
}
|
||||||
|
|||||||
@@ -18,6 +18,9 @@ import (
|
|||||||
// OpenCodeRunner — интерфейс для opencode (подменяемый в тестах).
|
// OpenCodeRunner — интерфейс для opencode (подменяемый в тестах).
|
||||||
type OpenCodeRunner interface {
|
type OpenCodeRunner interface {
|
||||||
Run(ctx context.Context, prompt, cwd, agent, sessionID string) (*opencode.Result, error)
|
Run(ctx context.Context, prompt, cwd, agent, sessionID string) (*opencode.Result, error)
|
||||||
|
// SessionMessages возвращает транскрипт сессии (шаги агента) по sessionID
|
||||||
|
// для постмортем-анализа. cwd — каталог, где живёт сервер пула сессии.
|
||||||
|
SessionMessages(ctx context.Context, cwd, sessionID string) (string, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
// PollTaskFunc — callback для обработки готовой задачи (подменяемый в тестах).
|
// PollTaskFunc — callback для обработки готовой задачи (подменяемый в тестах).
|
||||||
@@ -252,6 +255,7 @@ func (w *Worker) runTask(ctx context.Context, task *storage.Task) (err error) {
|
|||||||
Repos: repos,
|
Repos: repos,
|
||||||
Why: task.Why,
|
Why: task.Why,
|
||||||
AC: task.AC,
|
AC: task.AC,
|
||||||
|
Steps: task.Steps,
|
||||||
Branch: branch,
|
Branch: branch,
|
||||||
ReviewFeedback: reviewFeedbackList(branch, feedback),
|
ReviewFeedback: reviewFeedbackList(branch, feedback),
|
||||||
}
|
}
|
||||||
@@ -281,6 +285,8 @@ func (w *Worker) runTask(ctx context.Context, task *storage.Task) (err error) {
|
|||||||
|
|
||||||
output := res.Stdout
|
output := res.Stdout
|
||||||
|
|
||||||
|
log.Printf("worker: task %d: dev вердикт (хвост): %s", task.ID, truncateTrace(output, 2000))
|
||||||
|
|
||||||
// 5b. dev не завершился успешно (RC!=0) → фиксируем без ревью.
|
// 5b. dev не завершился успешно (RC!=0) → фиксируем без ревью.
|
||||||
switch res.RC {
|
switch res.RC {
|
||||||
case 0:
|
case 0:
|
||||||
@@ -348,11 +354,11 @@ func (w *Worker) runTask(ctx context.Context, task *storage.Task) (err error) {
|
|||||||
return pErr
|
return pErr
|
||||||
}
|
}
|
||||||
if e := w.setStatus(ctx, task, storage.StatusSuccess); e != nil {
|
if e := w.setStatus(ctx, task, storage.StatusSuccess); e != nil {
|
||||||
return fmt.Errorf("%w: set %s: %v", ErrUpdate, task.Status, e)
|
return fmt.Errorf("%w: set %s: %v", ErrUpdate, task.Status, e)
|
||||||
|
}
|
||||||
|
w.notifyStatus(ctx, task, storage.StatusSuccess)
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
w.notifyStatus(ctx, task, storage.StatusSuccess)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Не пройдено: если есть итерации — dev дорабатывает.
|
// Не пройдено: если есть итерации — dev дорабатывает.
|
||||||
if iter+1 < maxReviewIterations {
|
if iter+1 < maxReviewIterations {
|
||||||
|
|||||||
@@ -59,6 +59,11 @@ type mockRunnerWorker struct {
|
|||||||
// резюме. postMortemCount — сколько раз постмортем вызывался.
|
// резюме. postMortemCount — сколько раз постмортем вызывался.
|
||||||
postMortemResult *opencode.Result
|
postMortemResult *opencode.Result
|
||||||
postMortemCount int
|
postMortemCount int
|
||||||
|
|
||||||
|
// sessionTranscripts — sessionID → транскрипт для постмортем-агента;
|
||||||
|
// sessionMsgsErr — ошибка чтения транскрипта.
|
||||||
|
sessionTranscripts map[string]string
|
||||||
|
sessionMsgsErr error
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *mockRunnerWorker) Run(_ context.Context, _, _, agent, _ string) (*opencode.Result, 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
|
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 с комментариями.
|
// reviewFailedRunner возвращает вердикт not-passed с комментариями.
|
||||||
func reviewFailedRunner() *opencode.Result {
|
func reviewFailedRunner() *opencode.Result {
|
||||||
return &opencode.Result{RC: 0, Stdout: `{"passed":false,"critical_issues":[],"solid_violations":["DIP: высокая связанность"],"comments":["исправь связанность"]}`}
|
return &opencode.Result{RC: 0, Stdout: `{"passed":false,"critical_issues":[],"solid_violations":["DIP: высокая связанность"],"comments":["исправь связанность"]}`}
|
||||||
@@ -713,6 +725,7 @@ func TestWorkerBadStatus(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TestWorkerPromptRendered — dev-промпт собирается из полей задачи.
|
||||||
func TestWorkerPromptRendered(t *testing.T) {
|
func TestWorkerPromptRendered(t *testing.T) {
|
||||||
s := setupWorkerDB(t)
|
s := setupWorkerDB(t)
|
||||||
task := createReadyTask(t, s, "prompt-test")
|
task := createReadyTask(t, s, "prompt-test")
|
||||||
@@ -746,6 +759,63 @@ func TestWorkerPromptRendered(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TestWorkerPromptIncludesSteps — этапы задачи попадают в dev-промпт по порядку
|
||||||
|
// с критериями готовности.
|
||||||
|
func TestWorkerPromptIncludesSteps(t *testing.T) {
|
||||||
|
s := setupWorkerDB(t)
|
||||||
|
task := createReadyTask(t, s, "steps-test")
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
task.Steps = []storage.Step{
|
||||||
|
{Title: "Модель", AC: "операции готовы"},
|
||||||
|
{Title: "UI"},
|
||||||
|
}
|
||||||
|
if err := s.UpdateTask(ctx, task); err != nil {
|
||||||
|
t.Fatalf("update steps: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
w := &Worker{
|
||||||
|
Store: s,
|
||||||
|
Runner: &mockRunnerWorker{result: &opencode.Result{RC: 0, Stdout: "ok", SessionID: "s"}},
|
||||||
|
Worktree: t.TempDir(),
|
||||||
|
}
|
||||||
|
seedFakeRepo(t, w.Worktree, "steps-test")
|
||||||
|
|
||||||
|
_ = w.runTask(ctx, task)
|
||||||
|
|
||||||
|
traces, err := s.GetTraces(ctx, task.ID)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("get traces: %v", err)
|
||||||
|
}
|
||||||
|
tr := traces[0]
|
||||||
|
for _, want := range []string{"Этапы", "Модель", "операции готовы", "UI"} {
|
||||||
|
if !strings.Contains(tr.Prompt, want) {
|
||||||
|
t.Errorf("dev-промпт не содержит %q", want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestRenderDevPromptSteps — прямой рендер dev-промпта с этапами.
|
||||||
|
func TestRenderDevPromptSteps(t *testing.T) {
|
||||||
|
prompt, err := RenderDevPrompt(DevPromptData{
|
||||||
|
Title: "Калькулятор",
|
||||||
|
AC: "работает",
|
||||||
|
Steps: []storage.Step{
|
||||||
|
{Title: "Модель", AC: "операции готовы"},
|
||||||
|
{Title: "UI"},
|
||||||
|
},
|
||||||
|
Branch: "feat/abc",
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("render: %v", err)
|
||||||
|
}
|
||||||
|
for _, want := range []string{"Этапы", "Модель — готово, когда: операции готовы", "UI"} {
|
||||||
|
if !strings.Contains(prompt, want) {
|
||||||
|
t.Errorf("prompt не содержит %q", want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestFeatureBranchName(t *testing.T) {
|
func TestFeatureBranchName(t *testing.T) {
|
||||||
cases := []struct {
|
cases := []struct {
|
||||||
tag string
|
tag string
|
||||||
@@ -987,7 +1057,7 @@ func TestWorkerPostMortemFailureDoesNotChangeStatus(t *testing.T) {
|
|||||||
|
|
||||||
runner := &mockRunnerWorker{
|
runner := &mockRunnerWorker{
|
||||||
// dev падает при спавне → failed; постмортем тоже падает.
|
// dev падает при спавне → failed; постмортем тоже падает.
|
||||||
err: errors.New("opencode not found"),
|
err: errors.New("opencode not found"),
|
||||||
postMortemResult: &opencode.Result{RC: 1, Stdout: ""},
|
postMortemResult: &opencode.Result{RC: 1, Stdout: ""},
|
||||||
}
|
}
|
||||||
w := &Worker{
|
w := &Worker{
|
||||||
@@ -1065,13 +1135,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) {
|
func TestRenderPostMortemPrompt(t *testing.T) {
|
||||||
tr := storage.Trace{
|
tr := storage.Trace{
|
||||||
Agent: "dev",
|
Agent: "dev",
|
||||||
Status: storage.TraceTimeout,
|
Status: storage.TraceTimeout,
|
||||||
Prompt: "промпт dev",
|
SessionID: "sess-dev-1",
|
||||||
Output: "вывод dev",
|
Prompt: "промпт dev",
|
||||||
|
Output: "вывод dev",
|
||||||
|
}
|
||||||
|
transcripts := map[string]string{
|
||||||
|
"sess-dev-1": "[Инструмент]\nпрочитал файл a.go\n[Ассистент]\nправлю код",
|
||||||
}
|
}
|
||||||
prompt, err := RenderPostMortemPrompt(PostMortemPromptData{
|
prompt, err := RenderPostMortemPrompt(PostMortemPromptData{
|
||||||
Title: "Таймаут-задача",
|
Title: "Таймаут-задача",
|
||||||
@@ -1079,14 +1202,80 @@ func TestRenderPostMortemPrompt(t *testing.T) {
|
|||||||
Repos: []string{"calc"},
|
Repos: []string{"calc"},
|
||||||
AC: "работает",
|
AC: "работает",
|
||||||
Status: storage.StatusTimeout,
|
Status: storage.StatusTimeout,
|
||||||
Sessions: postMortemsText([]storage.Trace{tr}),
|
Sessions: postMortemsText([]storage.Trace{tr}, transcripts),
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("render: %v", err)
|
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) {
|
if !strings.Contains(prompt, want) {
|
||||||
t.Errorf("промпт не содержит %q", 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 {
|
||||||
|
name string
|
||||||
|
v *reviewVerdict
|
||||||
|
want []string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "passed with everything",
|
||||||
|
v: &reviewVerdict{
|
||||||
|
Passed: true,
|
||||||
|
CriticalIssues: []string{"bug A"},
|
||||||
|
SolidViolations: []string{"v1", "v2"},
|
||||||
|
Comments: []string{"comment"},
|
||||||
|
},
|
||||||
|
want: []string{"passed=yes", "critical=[bug A]", "solid=[v1 | v2]", "comments=[comment]"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "failed with critical only",
|
||||||
|
v: &reviewVerdict{
|
||||||
|
Passed: false,
|
||||||
|
CriticalIssues: []string{"bug A", "bug B"},
|
||||||
|
},
|
||||||
|
want: []string{"passed=no", "critical=[bug A | bug B]"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "empty verdict",
|
||||||
|
v: &reviewVerdict{},
|
||||||
|
want: []string{"passed=no"},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
got := formatReviewVerdict(tt.v)
|
||||||
|
for _, w := range tt.want {
|
||||||
|
if !strings.Contains(got, w) {
|
||||||
|
t.Errorf("formatReviewVerdict = %q, want contain %q", got, w)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user