diff --git a/src/andersen_ev/andersen.py b/src/andersen_ev/andersen.py index 03973e4..b9c4af8 100644 --- a/src/andersen_ev/andersen.py +++ b/src/andersen_ev/andersen.py @@ -6,7 +6,7 @@ from gql.transport.websockets import WebsocketsTransport # for file schema support -from graphql import DocumentNode, Source, parse +from graphql import DocumentNode, Source, parse, DefinitionNode from graphql.utilities import get_operation_ast import os @@ -43,8 +43,11 @@ def _execute_query(self, name, variable_values = None): query = self._get_query(name) opname = self._get_operation(query) + # Wrap the query in a DocumentNode for the gql library + query_document = DocumentNode(definitions=[query]) + result = self.gqclient.execute( - query, + query_document, variable_values = variable_values ) try: