Files
roboco/tests/unit/services/test_memory_distiller.py
5612375cba Feat/v0.13.0 (#270)
* feat(release): add release-manager feature flag (default off)

* feat(release): change classification + semver-bump derivation

* feat(release): readiness audit (changelog/version-ref/docs/migration/gate)

* feat(release): release-manager engine proposes a gated release

* feat(release): fail-closed release executor (bump, gate, publish)

* feat(release): CEO approve/reject release-proposal surface

* docs(release): document the gated release manager

* feat(memory): add org-memory feature flags (default off)

* feat(memory): add playbooks table + status enum + migration

* feat(memory): playbook service with auditor curation transitions

* feat(memory): playbooks RAG index plugin

* feat(memory): index a playbook into RAG on approval

* feat(memory): distill a high-signal lesson at task completion

* feat(memory): keep private journal reflections out of the shared RAG corpus

* feat(memory): draft_playbook verb + auditor curation verbs

* fix(ci): resolve mypy tests/ errors blocking the gate (UUID casts, annotations)

* feat(memory): auto-inject similar lessons/playbooks into the briefing

* feat(memory): auditor playbook review queue (api + panel)

* docs(memory): document the org-memory loop + playbook verbs

* fix(provisioning): idempotent pitch provisioning (reuse product/project by slug on re-approval)

* fix(memory): add chunks_playbooks to the chunk schema + isolate release route tests

- Migration 030's CHUNK_TABLES was missing chunks_playbooks, breaking the
  IndexType<->migration parity guard once the PLAYBOOKS index landed. The
  upgrade is ALTER ... IF EXISTS so adding it is safe on any DB shape.
- The release-route fixture's approve/reject paths call db.commit() (real
  behavior), so a held proposal outlived the per-test rollback and leaked
  into engine tests that read the global list_open_release_proposals().
  Tear down source=release_manager rows after each test.
- Make the gather_snapshot real-repo smoke version-agnostic (semver match)
  so it stops pinning the literal repo version.

* chore(release): 0.13.0

* ++

---------

Co-authored-by: Renn F <rennf93@users.noreply.github.com>
2026-06-26 01:43:08 +02:00

60 lines
2.0 KiB
Python

"""MemoryDistiller — a local-LLM distilled completion lesson (best-effort)."""
from __future__ import annotations
from unittest.mock import AsyncMock
import pytest
from roboco.services.memory_distiller import LessonInput, MemoryDistiller
def _input() -> LessonInput:
return LessonInput(
title="Add a retry to the flaky pg fixture",
acceptance_criteria=["The pg test passes 100 times in a row"],
dev_notes="Wrapped the connect in a 3x retry with backoff.",
qa_notes="Confirmed stable across 200 runs.",
commit_messages=["fix: retry pg connect", "test: stress the fixture"],
)
@pytest.mark.asyncio
async def test_distill_returns_lesson(monkeypatch: pytest.MonkeyPatch) -> None:
monkeypatch.setattr(
"roboco.services.memory_distiller._chat",
AsyncMock(
return_value="Problem: flaky pg. Approach: retry+backoff. Gotcha: reset."
),
)
out = await MemoryDistiller().distill(_input())
assert out is not None
assert "Gotcha" in out
@pytest.mark.asyncio
async def test_distill_none_on_error(monkeypatch: pytest.MonkeyPatch) -> None:
monkeypatch.setattr(
"roboco.services.memory_distiller._chat", AsyncMock(side_effect=RuntimeError)
)
assert await MemoryDistiller().distill(_input()) is None
@pytest.mark.asyncio
async def test_distill_none_on_empty_response(monkeypatch: pytest.MonkeyPatch) -> None:
monkeypatch.setattr(
"roboco.services.memory_distiller._chat", AsyncMock(return_value=" ")
)
assert await MemoryDistiller().distill(_input()) is None
@pytest.mark.asyncio
async def test_distill_caps_at_120_words(monkeypatch: pytest.MonkeyPatch) -> None:
long_lesson = " ".join(f"word{i}" for i in range(300))
monkeypatch.setattr(
"roboco.services.memory_distiller._chat",
AsyncMock(return_value=long_lesson),
)
out = await MemoryDistiller().distill(_input())
assert out is not None
assert len(out.split()) <= 120 # noqa: PLR2004 - the documented word budget