diff --git a/adbnx_adapter/adapter.py b/adbnx_adapter/adapter.py index 58b4cef..4fc75f8 100644 --- a/adbnx_adapter/adapter.py +++ b/adbnx_adapter/adapter.py @@ -446,13 +446,9 @@ def __fetch_adb_docs( """ aql_return_value = "doc" if explicit_metagraph: - edge_keys = "_from: doc._from, _to: doc._to" if is_edge else "" - aql_return_value = f""" - MERGE( - KEEP(doc, {list(attributes)}), - {{"_id": doc._id, {edge_keys}}} - ) - """ + default_keys = ["_id", "_key"] + default_keys += ["_from", "_to"] if is_edge else [] + aql_return_value = f"KEEP(doc, {list(attributes) + default_keys})" col_size: int = self.__db.collection(col).count()