diff --git a/atlasbot/engine/answerer.py b/atlasbot/engine/answerer.py index 64ab4b8..790f82c 100644 --- a/atlasbot/engine/answerer.py +++ b/atlasbot/engine/answerer.py @@ -26,6 +26,8 @@ GENERIC_METRIC_TOKENS = {"atlas", "cluster", "kubernetes", "k8s", "titan", "lab" NS_ENTRY_MIN_LEN = 2 DEDUP_MIN_SENTENCES = 3 RUNBOOK_SIMILARITY_THRESHOLD = 0.4 +BYTES_KB = 1024 +BYTES_MB = 1024 * 1024 class LLMLimitReached(RuntimeError): @@ -1945,10 +1947,10 @@ def _format_metric_value(value: Any) -> str: num = float(value) except (TypeError, ValueError): return str(value) - if num >= 1024 * 1024: - return f"{num / (1024 * 1024):.2f} MB/s" - if num >= 1024: - return f"{num / 1024:.2f} KB/s" + if num >= BYTES_MB: + return f"{num / BYTES_MB:.2f} MB/s" + if num >= BYTES_KB: + return f"{num / BYTES_KB:.2f} KB/s" if num >= 1: return f"{num:.2f}" return f"{num:.4f}"