Fix for optional References in token trees (#2164)

Signed-off-by: ammar <[email protected]> Co-authored-by: Jeroen Dekkers <[email protected]>
minvws · Dec 19, 2023 · 18ee8e6 · 18ee8e6
1 parent cec75e9
commit 18ee8e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/octopoes/octopoes/models/__init__.py b/octopoes/octopoes/models/__init__.py
@@ -265,7 +265,7 @@ def build_token_tree(ooi_class: Type[OOI]) -> Dict:
         field = ooi_class.model_fields[attribute]
         value = ""
 
-        if field.annotation == Reference:
+        if field.annotation in (Reference, Optional[Reference]):
             from octopoes.models.types import related_object_type
 
             related_class = related_object_type(field)