diff --git a/parser/query.go b/parser/query.go
index 7f31b0f4..f653bed5 100644
--- a/parser/query.go
+++ b/parser/query.go
@@ -9,7 +9,15 @@ import (
 func ParseQuery(source *Source) (*QueryDocument, error) {
 	p := parser{
 		lexer:         lexer.New(source),
-		maxTokenLimit: 0, // 0 is the default value
+		maxTokenLimit: 0, // 0 means unlimited
+	}
+	return p.parseQueryDocument(), p.err
+}
+
+func ParseQueryWithTokenLimit(source *Source, maxTokenLimit int) (*QueryDocument, error) {
+	p := parser{
+		lexer:         lexer.New(source),
+		maxTokenLimit: maxTokenLimit,
 	}
 	return p.parseQueryDocument(), p.err
 }