buildai-cli 0.3.75__tar.gz → 0.3.76__tar.gz

{buildai_cli-0.3.75 → buildai_cli-0.3.76}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: buildai-cli
-Version: 0.3.75
+Version: 0.3.76
 Summary: Build AI CLI (Typer)
 Requires-Python: >=3.11
 Requires-Dist: httpx>=0.27.0

{buildai_cli-0.3.75 → buildai_cli-0.3.76}/cli/commands/db/query.py

@@ -65,6 +65,8 @@ def _is_write_query(sql: str) -> bool:
             return True
 
     normalized = _normalize_sql(sql)
+    if normalized.startswith("WITH ") and re.search(r"\b(INSERT|UPDATE|DELETE)\b", normalized):
+        return True
     for keyword in ("VACUUM", "REASSIGN"):
         if normalized.startswith(keyword + " ") or normalized == keyword:
             return True
@@ -98,14 +100,21 @@ def _is_ddl_query(sql: str) -> bool:
 def _is_select_query(sql: str) -> bool:
     """Return whether the SQL is a row-returning query."""
 
+    import sqlparse
+    from sqlparse.tokens import DML
+
     normalized = _normalize_sql(sql)
     if normalized.startswith("SELECT "):
         return True
     if normalized.startswith("WITH "):
-        for keyword in ("INSERT", "UPDATE", "DELETE"):
-            if f" {keyword} " in normalized:
-                return False
-        return True
+        statements = sqlparse.parse(sql)
+        if statements:
+            for token in statements[0].tokens:
+                if token.is_whitespace:
+                    continue
+                if token.ttype is DML:
+                    return token.value.upper() == "SELECT"
+        return re.search(r"\b(INSERT|UPDATE|DELETE)\b", normalized) is None
     return any(normalized.startswith(keyword) for keyword in ("SHOW ", "EXPLAIN ", "TABLE "))
 
 

{buildai_cli-0.3.75 → buildai_cli-0.3.76}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "buildai-cli"
-version = "0.3.75"
+version = "0.3.76"
 description = "Build AI CLI (Typer)"
 requires-python = ">=3.11"
 dependencies = [