feat(plan): services/planning — start_planning aggregator

This commit is contained in:
2026-05-28 10:16:36 -04:00
parent 737467f996
commit e269ac9d5c
2 changed files with 109 additions and 0 deletions
+63
View File
@@ -0,0 +1,63 @@
"""Planning service — start_planning aggregates plan-task creation with the
project's applicable Rulebook rules and a little context, so planning happens
in Scribe and rules surface at the planning moment.
"""
from __future__ import annotations
from fabledassistant.services import notes as notes_svc
from fabledassistant.services import projects as projects_svc
from fabledassistant.services import rulebooks as rulebooks_svc
PLAN_TEMPLATE = """## Goal
## Approach
## Steps
- [ ]
## Verification
"""
async def start_planning(user_id: int, project_id: int, title: str) -> dict:
"""Create a plan-task seeded with the plan template and return it with the
project's applicable rules + brief context.
Returns:
{
"task": <task dict>,
"applicable_rules": [...],
"subscribed_rulebooks": [...],
"applicable_rules_truncated": bool,
"project_goal": str,
"open_task_count": int,
}
"""
project = await projects_svc.get_project(user_id, project_id)
if project is None:
raise ValueError(f"project {project_id} not found")
note = await notes_svc.create_note(
user_id,
title=title,
body=PLAN_TEMPLATE,
status="todo",
task_kind="plan",
project_id=project_id,
)
applicable = await rulebooks_svc.get_applicable_rules(
project_id=project_id, user_id=user_id,
)
_, open_count = await notes_svc.list_notes(
user_id, is_task=True, status="todo", project_id=project_id, limit=1,
)
return {
"task": note.to_dict(),
"applicable_rules": applicable["rules"],
"subscribed_rulebooks": applicable["subscribed_rulebooks"],
"applicable_rules_truncated": applicable["truncated"],
"project_goal": getattr(project, "goal", "") or "",
"open_task_count": open_count,
}
+46
View File
@@ -0,0 +1,46 @@
from unittest.mock import AsyncMock, MagicMock, patch
import pytest
@pytest.mark.asyncio
async def test_start_planning_creates_plan_task_and_returns_rules():
fake_note = MagicMock()
fake_note.to_dict.return_value = {"id": 5, "title": "Plan it", "task_kind": "plan"}
applicable = {
"rules": [{"id": 1, "title": "dev is home", "statement": "...",
"topic_title": "git-workflow", "rulebook_title": "FabledSword family"}],
"truncated": False,
"subscribed_rulebooks": [{"id": 2, "title": "FabledSword family"}],
}
with patch("fabledassistant.services.planning.notes_svc.create_note",
AsyncMock(return_value=fake_note)) as mock_create, \
patch("fabledassistant.services.planning.rulebooks_svc.get_applicable_rules",
AsyncMock(return_value=applicable)), \
patch("fabledassistant.services.planning.notes_svc.list_notes",
AsyncMock(return_value=([], 3))), \
patch("fabledassistant.services.planning.projects_svc.get_project",
AsyncMock(return_value=MagicMock(goal="ship it"))):
from fabledassistant.services.planning import start_planning
out = await start_planning(user_id=7, project_id=3, title="Plan it")
# Created a plan-task (status set => task, kind=plan)
kwargs = mock_create.call_args.kwargs
assert kwargs["task_kind"] == "plan"
assert kwargs["status"] == "todo"
assert kwargs["project_id"] == 3
assert "## Goal" in kwargs["body"] # seeded template
# Returned shape
assert out["task"]["id"] == 5
assert out["applicable_rules"][0]["title"] == "dev is home"
assert out["subscribed_rulebooks"] == [{"id": 2, "title": "FabledSword family"}]
assert out["open_task_count"] == 3
@pytest.mark.asyncio
async def test_start_planning_raises_when_project_not_found():
with patch("fabledassistant.services.planning.projects_svc.get_project",
AsyncMock(return_value=None)):
from fabledassistant.services.planning import start_planning
with pytest.raises(ValueError, match="project 999 not found"):
await start_planning(user_id=7, project_id=999, title="x")