change type in query

singnet · Mar 6, 2024 · 485f755 · 485f755
1 parent 60b0842
commit 485f755
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/das-query-engine/actions.py b/das-query-engine/actions.py
@@ -114,7 +114,7 @@ def get_incoming_links(
     @remove_none_args
     def query(
         self,
-        query: Dict[str, Any],
+        query: List[Dict[str, Any]] | Dict[str, Any],
         parameters: Dict[str, Any] = {"no_iterator": True},
     ) -> List[Dict[str, Any]]:
         return self.distributed_atom_space.query(query, parameters)

diff --git a/das-query-engine/validators/actions.py b/das-query-engine/validators/actions.py
@@ -46,7 +46,7 @@ class QueryValidator(PayloadValidator):
     strict = True
 
     query = datatypes.Function(
-        lambda value, *args, **kwargs: isinstance(value, dict),
+        lambda value, *args, **kwargs: isinstance(value, dict) or all(isinstance(item, dict) for item in value),
         required=True,
     )