autoforge-ai 0.1.20 → 0.1.22

package/server/routers/projects.py

@@ -66,10 +66,12 @@ def _get_registry_functions():
         sys.path.insert(0, str(root))
 
     from registry import (
+        get_project_auto_improve,
         get_project_concurrency,
         get_project_path,
         list_registered_projects,
         register_project,
+        set_project_auto_improve,
         set_project_concurrency,
         unregister_project,
         validate_project_path,
@@ -82,6 +84,8 @@ def _get_registry_functions():
         validate_project_path,
         get_project_concurrency,
         set_project_concurrency,
+        get_project_auto_improve,
+        set_project_auto_improve,
     )
 
 
@@ -118,7 +122,7 @@ async def list_projects():
     _init_imports()
     assert _check_spec_exists is not None  # guaranteed by _init_imports()
     (_, _, _, list_registered_projects, validate_project_path,
-     get_project_concurrency, _) = _get_registry_functions()
+     get_project_concurrency, _, _, _) = _get_registry_functions()
 
     projects = list_registered_projects()
     result = []
@@ -140,6 +144,10 @@ async def list_projects():
             has_spec=has_spec,
             stats=stats,
             default_concurrency=info.get("default_concurrency", 3),
+            auto_improve_enabled=bool(info.get("auto_improve_enabled", False)),
+            auto_improve_interval_minutes=int(
+                info.get("auto_improve_interval_minutes", 10) or 10
+            ),
         ))
 
     return result
@@ -151,7 +159,7 @@ async def create_project(project: ProjectCreate):
     _init_imports()
     assert _scaffold_project_prompts is not None  # guaranteed by _init_imports()
     (register_project, _, get_project_path, list_registered_projects,
-     _, _, _) = _get_registry_functions()
+     _, _, _, _, _) = _get_registry_functions()
 
     name = validate_project_name(project.name)
     project_path = Path(project.path).resolve()
@@ -232,7 +240,8 @@ async def get_project(name: str):
     _init_imports()
     assert _check_spec_exists is not None  # guaranteed by _init_imports()
     assert _get_project_prompts_dir is not None  # guaranteed by _init_imports()
-    (_, _, get_project_path, _, _, get_project_concurrency, _) = _get_registry_functions()
+    (_, _, get_project_path, _, _, get_project_concurrency, _,
+     get_project_auto_improve, _) = _get_registry_functions()
 
     name = validate_project_name(name)
     project_dir = get_project_path(name)
@@ -246,6 +255,7 @@ async def get_project(name: str):
     has_spec = _check_spec_exists(project_dir)
     stats = get_project_stats(project_dir)
     prompts_dir = _get_project_prompts_dir(project_dir)
+    ai_enabled, ai_interval = get_project_auto_improve(name)
 
     return ProjectDetail(
         name=name,
@@ -254,6 +264,8 @@ async def get_project(name: str):
         stats=stats,
         prompts_dir=str(prompts_dir),
         default_concurrency=get_project_concurrency(name),
+        auto_improve_enabled=ai_enabled,
+        auto_improve_interval_minutes=ai_interval,
     )
 
 
@@ -267,7 +279,7 @@ async def delete_project(name: str, delete_files: bool = False):
         delete_files: If True, also delete the project directory and files
     """
     _init_imports()
-    (_, unregister_project, get_project_path, _, _, _, _) = _get_registry_functions()
+    (_, unregister_project, get_project_path, _, _, _, _, _, _) = _get_registry_functions()
 
     name = validate_project_name(name)
     project_dir = get_project_path(name)
@@ -304,7 +316,7 @@ async def get_project_prompts(name: str):
     """Get the content of project prompt files."""
     _init_imports()
     assert _get_project_prompts_dir is not None  # guaranteed by _init_imports()
-    (_, _, get_project_path, _, _, _, _) = _get_registry_functions()
+    (_, _, get_project_path, _, _, _, _, _, _) = _get_registry_functions()
 
     name = validate_project_name(name)
     project_dir = get_project_path(name)
@@ -338,7 +350,7 @@ async def update_project_prompts(name: str, prompts: ProjectPromptsUpdate):
     """Update project prompt files."""
     _init_imports()
     assert _get_project_prompts_dir is not None  # guaranteed by _init_imports()
-    (_, _, get_project_path, _, _, _, _) = _get_registry_functions()
+    (_, _, get_project_path, _, _, _, _, _, _) = _get_registry_functions()
 
     name = validate_project_name(name)
     project_dir = get_project_path(name)
@@ -368,7 +380,7 @@ async def update_project_prompts(name: str, prompts: ProjectPromptsUpdate):
 async def get_project_stats_endpoint(name: str):
     """Get current progress statistics for a project."""
     _init_imports()
-    (_, _, get_project_path, _, _, _, _) = _get_registry_functions()
+    (_, _, get_project_path, _, _, _, _, _, _) = _get_registry_functions()
 
     name = validate_project_name(name)
     project_dir = get_project_path(name)
@@ -395,7 +407,7 @@ async def reset_project(name: str, full_reset: bool = False):
         Dictionary with list of deleted files and reset type
     """
     _init_imports()
-    (_, _, get_project_path, _, _, _, _) = _get_registry_functions()
+    (_, _, get_project_path, _, _, _, _, _, _) = _get_registry_functions()
 
     name = validate_project_name(name)
     project_dir = get_project_path(name)
@@ -487,12 +499,13 @@ async def reset_project(name: str, full_reset: bool = False):
 
 @router.patch("/{name}/settings", response_model=ProjectDetail)
 async def update_project_settings(name: str, settings: ProjectSettingsUpdate):
-    """Update project-level settings (concurrency, etc.)."""
+    """Update project-level settings (concurrency, auto-improve, etc.)."""
     _init_imports()
     assert _check_spec_exists is not None  # guaranteed by _init_imports()
     assert _get_project_prompts_dir is not None  # guaranteed by _init_imports()
     (_, _, get_project_path, _, _, get_project_concurrency,
-     set_project_concurrency) = _get_registry_functions()
+     set_project_concurrency, get_project_auto_improve,
+     set_project_auto_improve) = _get_registry_functions()
 
     name = validate_project_name(name)
     project_dir = get_project_path(name)
@@ -509,10 +522,43 @@ async def update_project_settings(name: str, settings: ProjectSettingsUpdate):
         if not success:
             raise HTTPException(status_code=500, detail="Failed to update concurrency")
 
+    # Update auto-improve config if either field was provided
+    auto_improve_touched = (
+        settings.auto_improve_enabled is not None
+        or settings.auto_improve_interval_minutes is not None
+    )
+    if auto_improve_touched:
+        try:
+            success = set_project_auto_improve(
+                name,
+                enabled=settings.auto_improve_enabled,
+                interval_minutes=settings.auto_improve_interval_minutes,
+            )
+        except ValueError as e:
+            raise HTTPException(status_code=400, detail=str(e))
+        if not success:
+            raise HTTPException(status_code=500, detail="Failed to update auto-improve settings")
+
+        # Sync the scheduler with the new state.
+        from ..services.scheduler_service import get_scheduler
+        scheduler = get_scheduler()
+        ai_enabled, ai_interval = get_project_auto_improve(name)
+        if ai_enabled:
+            try:
+                await scheduler.register_auto_improve(name, project_dir, ai_interval)
+            except Exception as e:
+                raise HTTPException(
+                    status_code=500,
+                    detail=f"Failed to register auto-improve schedule: {e}",
+                )
+        else:
+            scheduler.remove_auto_improve(name)
+
     # Return updated project details
     has_spec = _check_spec_exists(project_dir)
     stats = get_project_stats(project_dir)
     prompts_dir = _get_project_prompts_dir(project_dir)
+    ai_enabled, ai_interval = get_project_auto_improve(name)
 
     return ProjectDetail(
         name=name,
@@ -521,4 +567,6 @@ async def update_project_settings(name: str, settings: ProjectSettingsUpdate):
         stats=stats,
         prompts_dir=str(prompts_dir),
         default_concurrency=get_project_concurrency(name),
+        auto_improve_enabled=ai_enabled,
+        auto_improve_interval_minutes=ai_interval,
     )

package/server/routers/settings.py

@@ -26,6 +26,7 @@ from registry import (
     AVAILABLE_MODELS,
     DEFAULT_MODEL,
     get_all_settings,
+    get_effort_setting,
     get_setting,
     set_setting,
 )
@@ -95,6 +96,8 @@ def _parse_bool(value: str | None, default: bool = False) -> bool:
     return value.lower() == "true"
 
 
+
+
 @router.get("", response_model=SettingsResponse)
 async def get_settings():
     """Get current global settings."""
@@ -114,6 +117,7 @@ async def get_settings():
         playwright_headless=True,  # Always headless - embedded browser view replaces desktop windows
         batch_size=_parse_int(all_settings.get("batch_size"), 3),
         testing_batch_size=_parse_int(all_settings.get("testing_batch_size"), 3),
+        effort=get_effort_setting(),
         api_provider=api_provider,
         api_base_url=all_settings.get("api_base_url"),
         api_has_auth_token=bool(all_settings.get("api_auth_token")),
@@ -142,6 +146,9 @@ async def update_settings(update: SettingsUpdate):
     if update.testing_batch_size is not None:
         set_setting("testing_batch_size", str(update.testing_batch_size))
 
+    if update.effort is not None:
+        set_setting("effort", update.effort)
+
     # API provider settings
     if update.api_provider is not None:
         old_provider = get_setting("api_provider", "claude")
@@ -182,6 +189,7 @@ async def update_settings(update: SettingsUpdate):
         playwright_headless=True,  # Always headless - embedded browser view replaces desktop windows
         batch_size=_parse_int(all_settings.get("batch_size"), 3),
         testing_batch_size=_parse_int(all_settings.get("testing_batch_size"), 3),
+        effort=get_effort_setting(),
         api_provider=api_provider,
         api_base_url=all_settings.get("api_base_url"),
         api_has_auth_token=bool(all_settings.get("api_auth_token")),

package/server/schemas.py

@@ -18,7 +18,7 @@ _root = Path(__file__).parent.parent
 if str(_root) not in sys.path:
     sys.path.insert(0, str(_root))
 
-from registry import DEFAULT_MODEL, VALID_MODELS
+from registry import DEFAULT_MODEL, LEGACY_MODEL_MAP, VALID_MODELS
 
 # ============================================================================
 # Project Schemas
@@ -46,6 +46,8 @@ class ProjectSummary(BaseModel):
     has_spec: bool
     stats: ProjectStats
     default_concurrency: int = 3
+    auto_improve_enabled: bool = False
+    auto_improve_interval_minutes: int = 10
 
 
 class ProjectDetail(BaseModel):
@@ -56,6 +58,8 @@ class ProjectDetail(BaseModel):
     stats: ProjectStats
     prompts_dir: str
     default_concurrency: int = 3
+    auto_improve_enabled: bool = False
+    auto_improve_interval_minutes: int = 10
 
 
 class ProjectPrompts(BaseModel):
@@ -75,6 +79,8 @@ class ProjectPromptsUpdate(BaseModel):
 class ProjectSettingsUpdate(BaseModel):
     """Request schema for updating project-level settings."""
     default_concurrency: int | None = None
+    auto_improve_enabled: bool | None = None
+    auto_improve_interval_minutes: int | None = None
 
     @field_validator('default_concurrency')
     @classmethod
@@ -83,6 +89,13 @@ class ProjectSettingsUpdate(BaseModel):
             raise ValueError("default_concurrency must be between 1 and 5")
         return v
 
+    @field_validator('auto_improve_interval_minutes')
+    @classmethod
+    def validate_auto_improve_interval(cls, v: int | None) -> int | None:
+        if v is not None and (v < 1 or v > 1440):
+            raise ValueError("auto_improve_interval_minutes must be between 1 and 1440")
+        return v
+
 
 # ============================================================================
 # Feature Schemas
@@ -471,6 +484,7 @@ class SettingsResponse(BaseModel):
     playwright_headless: bool = True
     batch_size: int = 3  # Features per coding agent batch (1-15)
     testing_batch_size: int = 3  # Features per testing agent batch (1-15)
+    effort: Literal["low", "medium", "high", "xhigh", "max"] = "xhigh"
     api_provider: str = "claude"
     api_base_url: str | None = None
     api_has_auth_token: bool = False  # Never expose actual token
@@ -491,6 +505,7 @@ class SettingsUpdate(BaseModel):
     playwright_headless: bool | None = None
     batch_size: int | None = None  # Features per agent batch (1-15)
     testing_batch_size: int | None = None  # Features per testing agent batch (1-15)
+    effort: Literal["low", "medium", "high", "xhigh", "max"] | None = None
     api_provider: str | None = None
     api_base_url: str | None = Field(None, max_length=500)
     api_auth_token: str | None = Field(None, max_length=500)  # Write-only, never returned
@@ -507,12 +522,16 @@ class SettingsUpdate(BaseModel):
 
     @field_validator('model')
     @classmethod
-    def validate_model(cls, v: str | None, info) -> str | None:  # type: ignore[override]
+    def validate_model(cls, v: str | None, info) -> str | None:
         if v is not None:
             # Skip VALID_MODELS check when using an alternative API provider
             api_provider = info.data.get("api_provider")
             if api_provider and api_provider != "claude":
                 return v
+            # Transparently accept legacy IDs so in-flight clients don't 422
+            # during an upgrade window; LEGACY_MODEL_MAP already covers migration.
+            if v in LEGACY_MODEL_MAP:
+                v = LEGACY_MODEL_MAP[v]
             if v not in VALID_MODELS:
                 raise ValueError(f"Invalid model. Must be one of: {VALID_MODELS}")
         return v

package/server/services/assistant_chat_session.py

@@ -270,8 +270,9 @@ class AssistantChatSession:
         system_cli = shutil.which("claude")
 
         # Build environment overrides for API configuration
-        from registry import DEFAULT_MODEL, get_effective_sdk_env
+        from registry import DEFAULT_MODEL, get_effective_sdk_env, get_effort_setting
         sdk_env = get_effective_sdk_env()
+        effort = get_effort_setting()
 
         # Determine model from SDK env (provider-aware) or fallback to env/default
         model = sdk_env.get("ANTHROPIC_DEFAULT_OPUS_MODEL") or os.getenv("ANTHROPIC_DEFAULT_OPUS_MODEL", DEFAULT_MODEL)
@@ -281,6 +282,7 @@ class AssistantChatSession:
             self.client = ClaudeSDKClient(
                 options=ClaudeAgentOptions(
                     model=model,
+                    effort=effort,  # type: ignore[arg-type]  # SDK 0.1.61 Literal omits "xhigh"
                     cli_path=system_cli,
                     # System prompt loaded from CLAUDE.md via setting_sources
                     # This avoids Windows command line length limit (~8191 chars)