sql-data-guard 0.1.6__tar.gz → 0.1.7__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (27) hide show
  1. {sql_data_guard-0.1.6/src/sql_data_guard.egg-info → sql_data_guard-0.1.7}/PKG-INFO +1 -1
  2. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/pyproject.toml +1 -1
  3. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/src/sql_data_guard/sql_data_guard.py +1 -1
  4. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7/src/sql_data_guard.egg-info}/PKG-INFO +1 -1
  5. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/test/test_sql_guard_unit.py +2 -3
  6. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/LICENSE.md +0 -0
  7. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/README.md +0 -0
  8. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/setup.cfg +0 -0
  9. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/src/sql_data_guard/__init__.py +0 -0
  10. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/src/sql_data_guard/rest/__init__.py +0 -0
  11. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/src/sql_data_guard/rest/sql_data_guard_rest.py +0 -0
  12. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/src/sql_data_guard/restriction_validation.py +0 -0
  13. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/src/sql_data_guard/restriction_verification.py +0 -0
  14. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/src/sql_data_guard/verification_context.py +0 -0
  15. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/src/sql_data_guard/verification_utils.py +0 -0
  16. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/src/sql_data_guard.egg-info/SOURCES.txt +0 -0
  17. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/src/sql_data_guard.egg-info/dependency_links.txt +0 -0
  18. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/src/sql_data_guard.egg-info/requires.txt +0 -0
  19. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/src/sql_data_guard.egg-info/top_level.txt +0 -0
  20. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/test/test_duckdb_unit.py +0 -0
  21. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/test/test_rest_api_unit.py +0 -0
  22. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/test/test_sql_guard_curr_unit.py +0 -0
  23. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/test/test_sql_guard_joins_unit.py +0 -0
  24. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/test/test_sql_guard_llm.py +0 -0
  25. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/test/test_sql_guard_updates_unit.py +0 -0
  26. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/test/test_sql_guard_validation_unit.py +0 -0
  27. {sql_data_guard-0.1.6 → sql_data_guard-0.1.7}/test/test_utils.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: sql-data-guard
3
- Version: 0.1.6
3
+ Version: 0.1.7
4
4
  Summary: Safety Layer for LLM Database Interactions
5
5
  Author-email: Imperva - Threat Reseach Infra <ww.dis.imperva.threat-research-infra@thalesgroup.com>
6
6
  License: MIT License
@@ -9,7 +9,7 @@ package-dir = {"" = "src"}
9
9
  extra-index-url = ["https://pypi.org/simple"]
10
10
  [project]
11
11
  name = "sql-data-guard"
12
- version = "0.1.6"
12
+ version = "0.1.7"
13
13
  dependencies = [
14
14
  "sqlglot"
15
15
  ]
@@ -63,7 +63,7 @@ def verify_sql(sql: str, config: dict, dialect: str = None) -> dict:
63
63
  else:
64
64
  result.add_error("Could not find a query statement", False, 0.7)
65
65
  if result.can_fix and len(result.errors) > 0:
66
- result.fixed = parsed.sql()
66
+ result.fixed = parsed.sql(dialect=dialect)
67
67
  return {
68
68
  "allowed": len(result.errors) == 0,
69
69
  "errors": result.errors,
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: sql-data-guard
3
- Version: 0.1.6
3
+ Version: 0.1.7
4
4
  Summary: Safety Layer for LLM Database Interactions
5
5
  Author-email: Imperva - Threat Reseach Infra <ww.dis.imperva.threat-research-infra@thalesgroup.com>
6
6
  License: MIT License
@@ -369,14 +369,13 @@ class TestTrinoWithRestrictions:
369
369
  ]
370
370
  }
371
371
 
372
- @pytest.mark.skip(reason="Bug in sqlglot. Waiting for fix")
373
372
  def test_date_add(self, config):
374
373
  verify_sql_test(
375
- "SELECT id FROM accounts WHERE DATE(day) >= DATE_ADD('day', -7, CURRENT_DATE)",
374
+ "SELECT id FROM accounts WHERE DATE(day) >= DATE_ADD('DAY', -7, CURRENT_DATE)",
376
375
  config,
377
376
  dialect="trino",
378
377
  errors={"Missing restriction for table: accounts column: id value: 123"},
379
- fix="SELECT id FROM accounts WHERE DATE(day) >= DATE_ADD('day', -7, CURRENT_DATE) AND id = 123",
378
+ fix="SELECT id FROM accounts WHERE (DATE(day) >= DATE_ADD('DAY', -7, CURRENT_DATE)) AND id = 123",
380
379
  )
381
380
 
382
381
 
File without changes
File without changes