External/securelens-backend
1
0
Fork 0
mirror of https://github.com/Rarebuffalo/securelens-backend.git synced 2026-06-19 07:00:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

make securelens show usage when run with no subcommand and pass api_base in scan triage and analysis

Browse Source
This commit is contained in:
rarebuffalo
2026-06-15 02:50:07 +05:30
parent 6f83412d6f
commit b1fa5890e8
4 changed files with 49 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
cli/securelens/ai/__init__.py
View File

@@ -70,9 +70,10 @@ async def call_ai_json(
api_key: str,
model: str,
temperature: float = 0.2,
api_base: Optional[str] = None,
) -> Optional[dict]:
"""Convenience wrapper — calls AI in JSON mode and parses the result."""
raw = await call_ai(prompt, api_key, model, temperature=temperature, json_mode=True)
raw = await call_ai(prompt, api_key, model, temperature=temperature, json_mode=True, api_base=api_base)
if not raw:
return None
try:

6
cli/securelens/cli.py
View File

@@ -52,7 +52,7 @@ def main(ctx):
Scan codebases, URLs and get instant security reports.
"""
if ctx.invoked_subcommand is None:
ctx.invoke(scan, path=".", model=None, output=None, max_files=None, ci=False, fail_on=None, no_ai=False, sync=False)
click.echo(ctx.get_help())
# ── configure ─────────────────────────────────────────────────────────────────
@@ -269,7 +269,7 @@ async def _scan_async(path, model, output, max_files, ci, fail_on, no_ai, sync):
for v in vulnerabilities
]
prompt = summary_prompt(str(root), _json.dumps(issues_data, indent=2))
ai_summary = await call_ai(prompt, cfg.api_key, cfg.default_model, temperature=0.4)
ai_summary = await call_ai(prompt, cfg.api_key, cfg.default_model, temperature=0.4, api_base=cfg.api_base)
progress.update(task_summary, completed=100, total=100, detail="Done")
# ── Build result ─────────────────────────────────────────────────────────
@@ -389,7 +389,7 @@ async def _web_async(url, model, output, ci, fail_on, no_ai):
]
prompt = web_summary_prompt(url, _json.dumps(issues_data, indent=2),
result.score, result.grade)
result.ai_summary = await call_ai(prompt, cfg.api_key, cfg.default_model, temperature=0.4)
result.ai_summary = await call_ai(prompt, cfg.api_key, cfg.default_model, temperature=0.4, api_base=cfg.api_base)
progress.update(task2, completed=100, total=100, detail="Done")
fmt = cfg.output_format

4
cli/securelens/scanners/__init__.py
View File

@@ -195,7 +195,7 @@ async def triage_files(
if rel_paths and remaining_budget > 0 and cfg.api_key:
file_list_str = "\n".join(rel_paths[:300]) # cap to ~300 paths for token budget
prompt = triage_prompt(file_list_str, remaining_budget)
result = await call_ai_json(prompt, cfg.api_key, cfg.default_model, temperature=0.1)
result = await call_ai_json(prompt, cfg.api_key, cfg.default_model, temperature=0.1, api_base=cfg.api_base)
if result and "critical_files" in result:
for rel in result["critical_files"]:
abs_path = root / rel
@@ -233,7 +233,7 @@ async def analyze_file(
content = content[:30_000] + "\n... (truncated)"
prompt = analysis_prompt(rel, content)
result = await call_ai_json(prompt, cfg.api_key, cfg.default_model, temperature=0.2)
result = await call_ai_json(prompt, cfg.api_key, cfg.default_model, temperature=0.2, api_base=cfg.api_base)
if not result:
return []