Merge pull request #86 from singnet/bugfix/change-type-hint

[das-query-engine#173] Change the type hint in query()
singnet · Mar 6, 2024 · 8b0c2e8 · 8b0c2e8
2 parents 60b0842 + 485f755
commit 8b0c2e8
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,
     )