Skip to content

Commit 5557db4

Browse files
committed
chore: clean up a littl
Signed-off-by: Henry Schreiner <henryfs@princeton.edu>
1 parent b00a6b9 commit 5557db4

1 file changed

Lines changed: 17 additions & 20 deletions

File tree

src/sp_repo_review/ruff_checks/__main__.py

Lines changed: 17 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -40,27 +40,24 @@
4040
with RESOURCE_DIR.joinpath("ignore.json").open(encoding="utf-8") as f:
4141
IGNORE_INFO = json.load(f)
4242

43+
# Tool-specific agent variables
44+
# Based on https://github.com/agentsmd/agents.md/issues/136
45+
_AGENT_VARS = [
46+
"AGENT", # Pi, Goose, Amp
47+
"CLAUDECODE",
48+
"CURSOR_AGENT",
49+
"CLINE_ACTIVE",
50+
"GEMINI_CLI",
51+
"CODEX_SANDBOX",
52+
"AUGMENT_AGENT",
53+
"TRAE_AI_SHELL_ID",
54+
"OPENCODE_CLIENT",
55+
]
4356

44-
def _is_agent_environment() -> bool:
45-
"""Check if running from an AI coding agent.
46-
47-
Uses environment variables from https://github.com/agentsmd/agents.md/issues/136
48-
"""
49-
50-
# Tool-specific agent variables
51-
agent_vars = [
52-
"AGENT", # Pi, Goose, Amp
53-
"CLAUDECODE",
54-
"CURSOR_AGENT",
55-
"CLINE_ACTIVE",
56-
"GEMINI_CLI",
57-
"CODEX_SANDBOX",
58-
"AUGMENT_AGENT",
59-
"TRAE_AI_SHELL_ID",
60-
"OPENCODE_CLIENT",
61-
]
6257

63-
return any(os.environ.get(var) for var in agent_vars)
58+
def _is_agent_environment() -> bool:
59+
"""Check if running from an AI coding agent using env vars."""
60+
return any(os.environ.get(var) for var in _AGENT_VARS)
6461

6562

6663
def _resolve_format(format_arg: str) -> str:
@@ -82,7 +79,7 @@ def _print_each_plain(items: Mapping[str, str], indent: int = 2) -> Iterator[str
8279
"""Generate plain text formatted rule lines."""
8380
size = max(len(k) for k in items) if items else 0
8481
for k, v in items.items():
85-
yield f"{' ' * indent}\"{k}\",{' ' * (size - len(k))} # {v}"
82+
yield f'{" " * indent}"{k}",{" " * (size - len(k))} # {v}'
8683

8784

8885
def _print_each_rich(items: Mapping[str, str]) -> Iterator[str]:

0 commit comments

Comments
 (0)