kaizen-loop 0.1.0__tar.gz → 0.2.1__tar.gz

{kaizen_loop-0.1.0 → kaizen_loop-0.2.1}/.github/workflows/ci.yml

@@ -6,6 +6,9 @@ on:
   pull_request:
     branches: [main]
 
+env:
+  FORCE_JAVASCRIPT_ACTIONS_TO_NODE24: true
+
 jobs:
   lint:
     runs-on: ubuntu-latest

{kaizen_loop-0.1.0 → kaizen_loop-0.2.1}/.github/workflows/release.yml

@@ -4,6 +4,9 @@ on:
   push:
     tags: ["v*"]
 
+env:
+  FORCE_JAVASCRIPT_ACTIONS_TO_NODE24: true
+
 jobs:
   build:
     runs-on: ubuntu-latest

{kaizen_loop-0.1.0 → kaizen_loop-0.2.1}/PKG-INFO

@@ -1,10 +1,11 @@
 Metadata-Version: 2.4
 Name: kaizen-loop
-Version: 0.1.0
+Version: 0.2.1
 Summary: Continuous code improvement: autonomous work → review → fix → ship
 License-Expression: MIT
 License-File: LICENSE
 Requires-Python: >=3.10
+Requires-Dist: pydantic>=2.0
 Provides-Extra: dev
 Requires-Dist: pytest; extra == 'dev'
 Requires-Dist: ruff; extra == 'dev'

{kaizen_loop-0.1.0 → kaizen_loop-0.2.1}/pyproject.toml

@@ -4,10 +4,11 @@ build-backend = "hatchling.build"
 
 [project]
 name = "kaizen-loop"
-version = "0.1.0"
+version = "0.2.1"
 description = "Continuous code improvement: autonomous work → review → fix → ship"
 requires-python = ">=3.10"
 license = "MIT"
+dependencies = ["pydantic>=2.0"]
 
 [project.optional-dependencies]
 dev = ["pytest", "ruff"]

kaizen_loop-0.2.1/src/kaizen/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.2.1"

kaizen_loop-0.2.1/src/kaizen/findings.py

@@ -0,0 +1,44 @@
+from __future__ import annotations
+
+from functools import cached_property
+from typing import Literal
+
+from pydantic import BaseModel, Field
+
+Action = Literal["no-op", "auto-fix"]
+Severity = Literal["info", "warning", "error"]
+RiskLevel = Literal["low", "medium", "high"]
+
+
+class Finding(BaseModel):
+    id: str
+    severity: Severity
+    file: str = ""
+    line: int = 0
+    description: str = ""
+    action: Action = "no-op"
+
+
+class FindingsResult(BaseModel):
+    items: list[Finding] = Field(default_factory=list)
+    summary: str = ""
+    risk_level: RiskLevel = "low"
+    risk_rationale: str = ""
+
+    @cached_property
+    def has_auto_fix(self) -> bool:
+        return any(f.action == "auto-fix" for f in self.items)
+
+    @cached_property
+    def auto_fix_items(self) -> list[Finding]:
+        return [f for f in self.items if f.action == "auto-fix"]
+
+
+def parse_findings(data: dict) -> FindingsResult:
+    mapped = {
+        "items": data.get("findings", []),
+        "summary": data.get("summary", ""),
+        "risk_level": data.get("risk_level", "low"),
+        "risk_rationale": data.get("risk_rationale", ""),
+    }
+    return FindingsResult.model_validate(mapped)

{kaizen_loop-0.1.0 → kaizen_loop-0.2.1}/src/kaizen/loop.py

@@ -23,7 +23,7 @@ from kaizen.git import (
     slugify_prompt,
 )
 from kaizen.orchestrator import Orchestrator
-from kaizen.review_prompt import build_fix_prompt
+from kaizen.review_prompt import FIX_SCHEMA, build_fix_prompt
 from kaizen.run import RunInfo, setup_run, update_run_head, update_run_pr_url, update_run_status
 from kaizen.steps.pr import PRStep
 from kaizen.steps.push import PushStep
@@ -178,7 +178,7 @@ def run_loop(
                 print(f"\n  auto-fixing {len(auto_fix_items)} issues...")
                 fix_prompt = build_fix_prompt([_finding_to_dict(f) for f in auto_fix_items])
                 try:
-                    agent.run(fix_prompt, ctx.work_dir, repo_dir=cwd)
+                    agent.run(fix_prompt, ctx.work_dir, schema=FIX_SCHEMA, repo_dir=cwd)
                     commit_all(f"kaizen: fix {len(auto_fix_items)} review findings", ctx.work_dir)
                     current_head = head_commit(ctx.work_dir)
                     update_run_head(ctx.run_info.run_dir, current_head)

{kaizen_loop-0.1.0 → kaizen_loop-0.2.1}/src/kaizen/orchestrator.py

@@ -1,5 +1,7 @@
 import time
 
+from pydantic import BaseModel, Field
+
 from kaizen.agent import OpenCodeAgent
 from kaizen.config import load_config
 from kaizen.git import (
@@ -25,6 +27,14 @@ WORK_SCHEMA = {
 }
 
 
+class WorkOutput(BaseModel):
+    success: bool
+    summary: str
+    key_changes_made: list[str] = Field(default_factory=list)
+    key_learnings: list[str] = Field(default_factory=list)
+    should_fully_stop: bool = False
+
+
 class Orchestrator:
     def __init__(
         self,
@@ -93,17 +103,13 @@ class Orchestrator:
                             break
                         continue
 
-                    success = bool(result.output.get("success"))
-                    summary = str(result.output.get("summary", ""))
-                    changes = result.output.get("key_changes_made") or []
-                    learnings = result.output.get("key_learnings") or []
-                    should_stop = bool(result.output.get("should_fully_stop", False))
+                    work = WorkOutput.model_validate(result.output)
 
                     self.total_input_tokens += result.input_tokens
                     self.total_output_tokens += result.output_tokens
 
-                    if success:
-                        commit_msg = f"kaizen {self.iteration}: {summary}"
+                    if work.success:
+                        commit_msg = f"kaizen {self.iteration}: {work.summary}"
                         try:
                             commit_all(commit_msg, self.cwd)
                         except RuntimeError as e:
@@ -117,9 +123,9 @@ class Orchestrator:
                         self.consecutive_failures = 0
                         append_notes(
                             self.run_info.run_dir + "/notes.md",
-                            self.iteration, summary, changes, learnings,
+                            self.iteration, work.summary, work.key_changes_made, work.key_learnings,
                         )
-                        print(f"  committed: {summary}")
+                        print(f"  committed: {work.summary}")
 
                         if self.push_remote:
                             try:
@@ -133,11 +139,11 @@ class Orchestrator:
                         reset_hard(self.cwd)
                         append_notes(
                             self.run_info.run_dir + "/notes.md",
-                            self.iteration, f"[FAIL] {summary}", [], learnings,
+                            self.iteration, f"[FAIL] {work.summary}", [], work.key_learnings,
                         )
-                        print(f"  failed: {summary}")
+                        print(f"  failed: {work.summary}")
 
-                    if self.stop_when and should_stop:
+                    if self.stop_when and work.should_fully_stop:
                         print(f"  stop condition met: {self.stop_when}")
                         status = "stopped"
                         break

{kaizen_loop-0.1.0 → kaizen_loop-0.2.1}/src/kaizen/review_prompt.py

@@ -1,3 +1,7 @@
+from __future__ import annotations
+
+from pydantic import BaseModel, Field
+
 REVIEW_SCHEMA = {
     "type": "object",
     "additionalProperties": False,
@@ -25,6 +29,21 @@ REVIEW_SCHEMA = {
     "required": ["findings", "summary", "risk_level"],
 }
 
+FIX_SCHEMA = {
+    "type": "object",
+    "additionalProperties": False,
+    "properties": {
+        "changes_made": {"type": "array", "items": {"type": "string"}},
+        "summary": {"type": "string"},
+    },
+    "required": ["changes_made", "summary"],
+}
+
+
+class FixOutput(BaseModel):
+    changes_made: list[str] = Field(default_factory=list)
+    summary: str = ""
+
 
 def build_review_prompt(diff: str, intent: str = "") -> str:
     intent_section = ""
@@ -62,5 +81,8 @@ def build_fix_prompt(findings_items: list[dict]) -> str:
         lines.append(f"1. [{f['severity']}]{loc} — {f['description']} ({f['action']})")
 
     lines.append("\nAfter fixing, run any available tests or linters to verify.")
+    lines.append("\n## Output\n\nReturn structured output with:")
+    lines.append("- changes_made: array of descriptions of fixes applied")
+    lines.append("- summary: one-sentence summary of fixes")
 
     return "\n".join(lines)

{kaizen_loop-0.1.0 → kaizen_loop-0.2.1}/src/kaizen/steps/__init__.py

@@ -1,10 +1,9 @@
-from dataclasses import dataclass
+from pydantic import BaseModel
 
 from kaizen.findings import FindingsResult
 
 
-@dataclass
-class StepOutcome:
+class StepOutcome(BaseModel):
     findings: FindingsResult | None = None
     skipped: bool = False
     skip_remaining: bool = False

kaizen_loop-0.1.0/src/kaizen/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.1.0"

kaizen_loop-0.1.0/src/kaizen/findings.py

@@ -1,53 +0,0 @@
-from dataclasses import dataclass, field
-
-ACTION_NOOP = "no-op"
-ACTION_AUTO_FIX = "auto-fix"
-
-SEVERITY_INFO = "info"
-SEVERITY_WARNING = "warning"
-SEVERITY_ERROR = "error"
-
-
-@dataclass
-class Finding:
-    id: str
-    severity: str
-    file: str = ""
-    line: int = 0
-    description: str = ""
-    action: str = ACTION_NOOP
-
-
-@dataclass
-class FindingsResult:
-    items: list[Finding] = field(default_factory=list)
-    summary: str = ""
-    risk_level: str = "low"
-    risk_rationale: str = ""
-
-    @property
-    def has_auto_fix(self) -> bool:
-        return any(f.action == ACTION_AUTO_FIX for f in self.items)
-
-    @property
-    def auto_fix_items(self) -> list[Finding]:
-        return [f for f in self.items if f.action == ACTION_AUTO_FIX]
-
-
-def parse_findings(data: dict) -> FindingsResult:
-    items = []
-    for i, f in enumerate(data.get("findings", [])):
-        items.append(Finding(
-            id=f.get("id", f"f{i + 1}"),
-            severity=f.get("severity", SEVERITY_INFO),
-            file=f.get("file", ""),
-            line=f.get("line", 0),
-            description=f.get("description", ""),
-            action=f.get("action", ACTION_NOOP),
-        ))
-    return FindingsResult(
-        items=items,
-        summary=data.get("summary", ""),
-        risk_level=data.get("risk_level", "low"),
-        risk_rationale=data.get("risk_rationale", ""),
-    )