fix: re-run pre-commit

Author: harris-ranque

src/agents/__init__.py

@@ -9,9 +9,9 @@
 from src.agents.acknowledgment_agent import AcknowledgmentAgent
 from src.agents.base import AgentResult, BaseAgent
 from src.agents.engine_agent import RuleEngineAgent
+from src.agents.extractor_agent import RuleExtractorAgent
 from src.agents.factory import get_agent
 from src.agents.feasibility_agent import RuleFeasibilityAgent
-from src.agents.extractor_agent import RuleExtractorAgent
 from src.agents.repository_analysis_agent import RepositoryAnalysisAgent
 
 __all__ = [

src/agents/extractor_agent/agent.py

@@ -72,11 +72,25 @@ def _build_graph(self):
         async def extract_node(state: ExtractorState) -> dict:
             raw = (state.markdown_content or "").strip()
             if not raw:
-                return {"statements": [], "decision": "none", "confidence": 0.0, "reasoning": "Empty input", "recommendations": [], "strategy_used": ""}
+                return {
+                    "statements": [],
+                    "decision": "none",
+                    "confidence": 0.0,
+                    "reasoning": "Empty input",
+                    "recommendations": [],
+                    "strategy_used": "",
+                }
             # Centralized sanitization (see execute(): defense-in-depth with redact_and_cap at entry).
             content = redact_and_cap(raw)
             if not content:
-                return {"statements": [], "decision": "none", "confidence": 0.0, "reasoning": "Empty after sanitization", "recommendations": [], "strategy_used": ""}
+                return {
+                    "statements": [],
+                    "decision": "none",
+                    "confidence": 0.0,
+                    "reasoning": "Empty after sanitization",
+                    "recommendations": [],
+                    "strategy_used": "",
+                }
             prompt = EXTRACTOR_PROMPT.format(markdown_content=content)
             structured_llm = self.llm.with_structured_output(ExtractorOutput)
             result = await structured_llm.ainvoke(prompt)
@@ -198,7 +212,11 @@ async def execute(self, **kwargs: Any) -> AgentResult:
                     "recommendations": [],
                     "strategy_used": "",
                 },
-                metadata={"execution_time_ms": execution_time * 1000, "error_type": "timeout", "routing": "human_review"},
+                metadata={
+                    "execution_time_ms": execution_time * 1000,
+                    "error_type": "timeout",
+                    "routing": "human_review",
+                },
             )
         except APIConnectionError as e:
             execution_time = time.time() - start_time
@@ -238,5 +256,9 @@ async def execute(self, **kwargs: Any) -> AgentResult:
                     "recommendations": [],
                     "strategy_used": "",
                 },
-                metadata={"execution_time_ms": execution_time * 1000, "error_type": type(e).__name__, "routing": "human_review"},
+                metadata={
+                    "execution_time_ms": execution_time * 1000,
+                    "error_type": type(e).__name__,
+                    "routing": "human_review",
+                },
             )

src/agents/factory.py

@@ -11,8 +11,8 @@
 from src.agents.acknowledgment_agent import AcknowledgmentAgent
 from src.agents.base import BaseAgent
 from src.agents.engine_agent import RuleEngineAgent
-from src.agents.feasibility_agent import RuleFeasibilityAgent
 from src.agents.extractor_agent import RuleExtractorAgent
+from src.agents.feasibility_agent import RuleFeasibilityAgent
 from src.agents.repository_analysis_agent import RepositoryAnalysisAgent
 
 logger = logging.getLogger(__name__)

src/api/recommendations.py

@@ -2,6 +2,7 @@
 from typing import Any, TypedDict
 
 import structlog
+import yaml
 from fastapi import APIRouter, Depends, HTTPException, Request, status
 from giturlparse import parse  # type: ignore
 from pydantic import BaseModel, Field, HttpUrl
@@ -13,12 +14,10 @@
 # Internal: User model, auth assumed present—see core/api for details.
 from src.core.models import User
 from src.integrations.github.api import github_client
-
 from src.rules.ai_rules_scan import (
     scan_repo_for_ai_rule_files,
     translate_ai_rule_files_to_yaml,
 )
-import yaml
 
 logger = structlog.get_logger()
 
@@ -141,6 +140,7 @@ class MetricConfig(TypedDict):
     thresholds: dict[str, float]
     explanation: Callable[[float | int], str]
 
+
 class ScanAIFilesRequest(BaseModel):
     """
     Payload for scanning a repo for AI assistant rule files (Cursor, Claude, Copilot, etc.).
@@ -178,6 +178,7 @@ class ScanAIFilesResponse(BaseModel):
     )
     warnings: list[str] = Field(default_factory=list, description="Warnings (e.g. rate limit, partial results)")
 
+
 class TranslateAIFilesRequest(BaseModel):
     """Request for translating AI rule files into .watchflow rules YAML."""
 
@@ -205,7 +206,6 @@ class TranslateAIFilesResponse(BaseModel):
     warnings: list[str] = Field(default_factory=list)
 
 
-
 def _get_severity_label(value: float, thresholds: dict[str, float]) -> tuple[str, str]:
     """
     Determine severity label and color based on value and thresholds.
@@ -966,6 +966,7 @@ async def proceed_with_pr(
             detail="Failed to create pull request. Please try again.",
         ) from e
 
+
 @router.post(
     "/scan-ai-files",
     response_model=ScanAIFilesResponse,
@@ -981,7 +982,7 @@ async def scan_ai_rule_files(
     request: Request,
     payload: ScanAIFilesRequest,
     user: User | None = Depends(get_current_user_optional),
-    ) -> ScanAIFilesResponse:
+) -> ScanAIFilesResponse:
     """
     Scan a repository for AI assistant rule files (Cursor, Claude, Copilot, etc.).
 
@@ -1007,9 +1008,7 @@ async def scan_ai_rule_files(
         repo_full_name = parse_repo_from_url(repo_url_str)
     except ValueError as e:
         logger.warning("invalid_url_provided", url=repo_url_str, error=str(e))
-        raise HTTPException(
-            status_code=status.HTTP_422_UNPROCESSABLE_ENTITY, detail=str(e)
-        ) from e
+        raise HTTPException(status_code=status.HTTP_422_UNPROCESSABLE_ENTITY, detail=str(e)) from e
 
     # Resolve token (same as recommend_rules)
     github_token = None
@@ -1057,9 +1056,7 @@ async def scan_ai_rule_files(
 
     # Optional content fetcher for keyword scan (and optionally include in response)
     async def get_content(path: str):
-        return await github_client.get_file_content(
-            repo_full_name, path, installation_id, github_token
-        )
+        return await github_client.get_file_content(repo_full_name, path, installation_id, github_token)
 
     # Always fetch content so has_keywords is set; strip content in response unless include_content
     raw_candidates = await scan_repo_for_ai_rule_files(
@@ -1084,6 +1081,7 @@ async def get_content(path: str):
         warnings=[],
     )
 
+
 @router.post(
     "/translate-ai-files",
     response_model=TranslateAIFilesResponse,
@@ -1166,9 +1164,7 @@ async def translate_ai_rule_files(
     async def get_content(path: str):
         return await github_client.get_file_content(repo_full_name, path, installation_id, github_token)
 
-    raw_candidates = await scan_repo_for_ai_rule_files(
-        tree_entries, fetch_content=True, get_file_content=get_content
-    )
+    raw_candidates = await scan_repo_for_ai_rule_files(tree_entries, fetch_content=True, get_file_content=get_content)
     candidates_with_content = [c for c in raw_candidates if c.get("content")]
     if not candidates_with_content:
         return TranslateAIFilesResponse(
@@ -1197,12 +1193,7 @@ async def get_content(path: str):
         reason = item.get("reason", "") if isinstance(item, dict) else ""
         if not isinstance(reason, str):
             reason = str(reason)
-        if (
-            len(reason) > 200
-            or "Error" in reason
-            or "Exception" in reason
-            or "Traceback" in reason
-        ):
+        if len(reason) > 200 or "Error" in reason or "Exception" in reason or "Traceback" in reason:
             logger.debug(
                 "translate_ai_rule_files_ambiguous_reason_redacted",
                 repo_full_name=repo_full_name,
@@ -1226,4 +1217,4 @@ async def get_content(path: str):
         rules_count=rules_count,
         ambiguous=safe_ambiguous,
         warnings=[],
-    )
+    )

src/event_processors/pull_request/processor.py

@@ -110,7 +110,7 @@ async def process(self, task: Task) -> ProcessingResult:
                     )
                     if rules_count > 0:
                         suggested_rules_yaml = rules_yaml
-                except Exception as e:
+                except Exception:
                     latency_ms = int((time.time() - scan_start) * 1000)
                     logger.exception(
                         "Suggested rules scan failed",

src/event_processors/push.py

@@ -12,7 +12,6 @@
 from src.rules.ai_rules_scan import is_relevant_push
 from src.tasks.task_queue import Task
 
-
 logger = logging.getLogger(__name__)
 
 
@@ -294,9 +293,7 @@ async def _create_pr_with_suggested_rules(
                 base_ref = (pr.get("base") or {}).get("ref") or ""
                 head_ref = (pr.get("head") or {}).get("ref") or ""
                 title = pr.get("title") or ""
-                if base_ref == default_branch and (
-                    head_ref.startswith(branch_prefix) or title == pr_title
-                ):
+                if base_ref == default_branch and (head_ref.startswith(branch_prefix) or title == pr_title):
                     existing_pr = pr
                     break
             if existing_pr:

src/integrations/github/api.py

@@ -135,9 +135,7 @@ async def get_repository(
         Fetch repository metadata. Returns (repo_data, None) on success;
         (None, {"status": int, "message": str}) on failure for meaningful API responses.
         """
-        headers = await self._get_auth_headers(
-            installation_id=installation_id, user_token=user_token
-        ) or {}
+        headers = await self._get_auth_headers(installation_id=installation_id, user_token=user_token) or {}
         url = f"{config.github.api_base_url}/repos/{repo_full_name}"
         session = await self._get_session()
         async with session.get(url, headers=headers) as response:
@@ -160,17 +158,18 @@ async def get_repository(
             if response.status == 401:
                 return (
                     None,
-                    {"status": 401, "message": gh_message or "Invalid or expired token. Check github_token or installation_id."},
+                    {
+                        "status": 401,
+                        "message": gh_message or "Invalid or expired token. Check github_token or installation_id.",
+                    },
                 )
             return None, {"status": response.status, "message": gh_message or f"GitHub API returned {response.status}."}
 
     async def list_directory_any_auth(
         self, repo_full_name: str, path: str, installation_id: int | None = None, user_token: str | None = None
     ) -> list[dict[str, Any]]:
         """List directory contents using installation or user token (auth required)."""
-        headers = await self._get_auth_headers(
-            installation_id=installation_id, user_token=user_token
-        ) or {}
+        headers = await self._get_auth_headers(installation_id=installation_id, user_token=user_token) or {}
         url = f"{config.github.api_base_url}/repos/{repo_full_name}/contents/{path}"
         session = await self._get_session()
         async with session.get(url, headers=headers) as response:
@@ -183,29 +182,36 @@ async def list_directory_any_auth(
             response.raise_for_status()
             return []
 
-
     async def get_repository_tree(
-    self,
-    repo_full_name: str,
-    ref: str | None = None,
-    installation_id: int | None = None,
-    user_token: str | None = None,
-    recursive: bool = True,
+        self,
+        repo_full_name: str,
+        ref: str | None = None,
+        installation_id: int | None = None,
+        user_token: str | None = None,
+        recursive: bool = True,
     ) -> list[dict[str, Any]]:
         """Get the tree of a repository. Requires authentication (github_token or installation_id)."""
         start = time.monotonic()
-        headers = await self._get_auth_headers(
-            installation_id=installation_id,
-            user_token=user_token,
-        ) or {}
+        headers = (
+            await self._get_auth_headers(
+                installation_id=installation_id,
+                user_token=user_token,
+            )
+            or {}
+        )
         ref = ref or "main"
         tree_sha = await self._resolve_tree_sha(repo_full_name, ref, headers)
         if not tree_sha:
             latency_ms = int((time.monotonic() - start) * 1000)
             logger.info(
                 "get_repository_tree",
                 operation="get_repository_tree",
-                subject_ids={"repo": repo_full_name, "installation_id": installation_id, "user_token_present": bool(user_token), "ref": ref},
+                subject_ids={
+                    "repo": repo_full_name,
+                    "installation_id": installation_id,
+                    "user_token_present": bool(user_token),
+                    "ref": ref,
+                },
                 decision="ref_resolution_failed",
                 latency_ms=latency_ms,
             )
@@ -228,7 +234,6 @@ async def get_repository_tree(
             data = await response.json()
             return cast("list[dict[str, Any]]", data.get("tree", []))
 
-        
     async def _resolve_tree_sha(self, repo_full_name: str, ref: str, headers: dict[str, str]) -> str | None:
         """Resolve the tree SHA for the given ref (branch, tag, or commit SHA) via the commits API."""
         session = await self._get_session()
@@ -254,11 +259,14 @@ async def get_file_content(
         Fetches the content of a file from a repository. Requires authentication (github_token or installation_id).
         When ref is provided (branch name, tag, or commit SHA), returns content at that ref; otherwise uses default branch.
         """
-        headers = await self._get_auth_headers(
-            installation_id=installation_id,
-            user_token=user_token,
-            accept="application/vnd.github.raw",
-        ) or {}
+        headers = (
+            await self._get_auth_headers(
+                installation_id=installation_id,
+                user_token=user_token,
+                accept="application/vnd.github.raw",
+            )
+            or {}
+        )
         url = f"{config.github.api_base_url}/repos/{repo_full_name}/contents/{file_path}"
         params = {"ref": ref} if ref else None
 
@@ -1350,9 +1358,7 @@ async def execute_graphql(
         payload = {"query": query, "variables": variables}
 
         # Get appropriate headers (auth required: user_token or installation_id)
-        headers = await self._get_auth_headers(
-            user_token=user_token, installation_id=installation_id
-        )
+        headers = await self._get_auth_headers(user_token=user_token, installation_id=installation_id)
         if not headers:
             # Fallback or error? GraphQL usually demands auth.
             # If we have no headers, we likely can't query GraphQL successfully for many fields.