Guard against missing creators/collections in aggregation

datalab-org · Jul 26, 2023 · 84132d1 · 84132d1
1 parent 012d150
commit 84132d1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pydatalab/pydatalab/routes/v0_1/graphs.py b/pydatalab/pydatalab/routes/v0_1/graphs.py
@@ -45,7 +45,7 @@ def get_graph_cy_format(item_id: Optional[str] = None, collection_id: Optional[s
         )
 
         node_ids = {document["item_id"] for document in all_documents} | {
-            relationship["item_id"]
+            relationship.get("item_id")
             for document in all_documents
             for relationship in document.get("relationships", [])
         }

diff --git a/pydatalab/pydatalab/routes/v0_1/items.py b/pydatalab/pydatalab/routes/v0_1/items.py
@@ -179,7 +179,7 @@ def creators_lookup() -> Dict:
             {
                 "$match": {
                     "$expr": {
-                        "$in": ["$_id", "$$creator_ids"],
+                        "$in": ["$_id", {"$ifNull": ["$$creator_ids", []]}],
                     },
                 }
             },
@@ -211,7 +211,7 @@ def collections_lookup() -> Dict:
             {
                 "$match": {
                     "$expr": {
-                        "$in": ["$_id", "$$collection_ids"],
+                        "$in": ["$_id", {"$ifNull": ["$$collection_ids", []]}],
                     },
                     "type": "collections",
                 }