Spaces:

obversarystudios
/

agent-threat-map

Runtime error

agent-threat-map / agent_threat_map /loader.py

Threat-map metrics + observable geometry (embed/cluster/MI)

6c3043e verified 6 months ago

572 Bytes

from __future__ import annotations

import json
from pathlib import Path

from agent_threat_map.schema import Probe


def load_probes(path: str | Path) -> list[Probe]:
    path = Path(path)
    probes: list[Probe] = []
    with path.open(encoding="utf-8") as f:
        for line in f:
            line = line.strip()
            if not line:
                continue
            probes.append(Probe.from_dict(json.loads(line)))
    return probes


def load_categories(path: str | Path) -> dict:
    with Path(path).open(encoding="utf-8") as f:
        return json.load(f)

	from __future__ import annotations

	import json
	from pathlib import Path

	from agent_threat_map.schema import Probe


	def load_probes(path: str \| Path) -> list[Probe]:
	path = Path(path)
	probes: list[Probe] = []
	with path.open(encoding="utf-8") as f:
	for line in f:
	line = line.strip()
	if not line:
	continue
	probes.append(Probe.from_dict(json.loads(line)))
	return probes


	def load_categories(path: str \| Path) -> dict:
	with Path(path).open(encoding="utf-8") as f:
	return json.load(f)