light-infer-chat commited on
Commit
a6a4880
·
1 Parent(s): cf85b28
app/services/sql_validator_service.py CHANGED
@@ -64,6 +64,9 @@ _READ_ONLY_TYPES: frozenset[QueryType] = frozenset(
64
 
65
  _AST_TYPE_MAP: list[tuple[type[exp.Expression], QueryType]] = [
66
  (exp.Select, QueryType.SELECT),
 
 
 
67
  (exp.Insert, QueryType.INSERT),
68
  (exp.Update, QueryType.UPDATE),
69
  (exp.Delete, QueryType.DELETE),
 
64
 
65
  _AST_TYPE_MAP: list[tuple[type[exp.Expression], QueryType]] = [
66
  (exp.Select, QueryType.SELECT),
67
+ (exp.Union, QueryType.SELECT),
68
+ (exp.Intersect, QueryType.SELECT),
69
+ (exp.Except, QueryType.SELECT),
70
  (exp.Insert, QueryType.INSERT),
71
  (exp.Update, QueryType.UPDATE),
72
  (exp.Delete, QueryType.DELETE),