From 5dd30c472da7a6eaa19071c5a49a510693e03f19 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Arturo=20Filast=C3=B2?= <arturo@filasto.net>
Date: Wed, 22 Nov 2023 11:38:47 +0100
Subject: [PATCH] Add optional extra keys

---
 oonidata/fastapi/routers/aggregation.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/oonidata/fastapi/routers/aggregation.py b/oonidata/fastapi/routers/aggregation.py
index ab87442e..fd3c9555 100644
--- a/oonidata/fastapi/routers/aggregation.py
+++ b/oonidata/fastapi/routers/aggregation.py
@@ -56,6 +56,11 @@ class AggregationEntry(BaseModel):
     # loni_ok_map: Dict[str, float]
     loni_ok_value: float
 
+    domain: Optional[str] = None
+    probe_cc: Optional[str] = None
+    probe_asn: Optional[int] = None
+    test_name: Optional[str] = None
+
 
 class AggregationResponse(BaseModel):
     # TODO(arturo): these keys are inconsistent with the other APIs