profit/llm-team-ui
1
0
Fork 0
Code Issues Pull Requests 2 Actions Packages Projects Releases Wiki Activity

Fix enrichment: run web-check before AI analysis so data is available

Browse Source 
Web-check (ports, DNS, blocklists) now runs as step 3, AI analysis
as step 4. AI prompt includes open ports and blocklist status for
richer threat verdicts.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
root 2026-03-26 03:42:23 -05:00
parent e816e81820
commit 51ffd2b82c
1 changed files with 13 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
llm_team_ui.py
View File

@ -4056,7 +4056,19 @@ def admin_enrich_ip():
except Exception: except Exception:
pass pass
# Step 3: AI threat analysis with full context (including web-check data) # Step 3: Web-Check deep scan (ports, DNS, blocklists)
WEB_CHECK_BASE = "http://localhost:3000/api"
webcheck = {}
for endpoint in ["ports", "dns", "block-lists"]:
try:
wc_resp = requests.get(f"{WEB_CHECK_BASE}/{endpoint}?url={ip}", timeout=15)
if wc_resp.status_code == 200:
webcheck[endpoint.replace("-", "_")] = wc_resp.json()
except Exception:
pass
result["webcheck"] = webcheck
# Step 4: AI threat analysis with full context (including web-check data)
try: try:
geo_ctx = "" geo_ctx = ""
if result["geo"] and not result["geo"].get("error"): if result["geo"] and not result["geo"].get("error"):
@ -4117,17 +4129,6 @@ def admin_enrich_ip():
except Exception as e: except Exception as e:
result["ai_analysis"] = {"error": str(e)} result["ai_analysis"] = {"error": str(e)}
# Step 4: Web-Check deep scan (ports, DNS, blocklists)
WEB_CHECK_BASE = "http://localhost:3000/api"
webcheck = {}
for endpoint in ["ports", "dns", "block-lists"]:
try:
wc_resp = requests.get(f"{WEB_CHECK_BASE}/{endpoint}?url={ip}", timeout=15)
if wc_resp.status_code == 200:
webcheck[endpoint.replace("-", "_")] = wc_resp.json()
except Exception:
pass
result["webcheck"] = webcheck
result["log_count"] = len(ip_logs) result["log_count"] = len(ip_logs)
return jsonify(result) return jsonify(result)