gaard-core 0.2.0__tar.gz → 0.2.2__tar.gz

{gaard_core-0.2.0 → gaard_core-0.2.2}/PKG-INFO

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: gaard-core
-Version: 0.2.0
+Version: 0.2.2
 Summary: Core GAARD query pipeline, prompt compiler, policies and SQL validation
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
-Requires-Dist: gaard-plugin-api<0.3.0,>=0.2.0
+Requires-Dist: gaard-plugin-api<0.3.0,>=0.2.2
 Requires-Dist: pydantic>=2.7.0
 Requires-Dist: sqlglot>=25.0.0
 Provides-Extra: dev

{gaard_core-0.2.0 → gaard_core-0.2.2}/pyproject.toml

@@ -4,12 +4,12 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "gaard-core"
-version = "0.2.0"
+version = "0.2.2"
 description = "Core GAARD query pipeline, prompt compiler, policies and SQL validation"
 readme = "README.md"
 requires-python = ">=3.11"
 dependencies = [
-    "gaard-plugin-api>=0.2.0,<0.3.0",
+    "gaard-plugin-api>=0.2.2,<0.3.0",
     "pydantic>=2.7.0",
     "sqlglot>=25.0.0",
 ]

{gaard_core-0.2.0 → gaard_core-0.2.2}/src/gaard_core/query_intent/llm_classifier.py

@@ -1,5 +1,5 @@
 import json
-from typing import Any, Protocol
+from typing import Any, Protocol, cast
 
 from gaard_core.llm_output import remove_thinking_blocks
 from gaard_core.prompt_compiler.intent_classification_prompt import (
@@ -105,7 +105,7 @@ def parse_query_intent_decision(value: object) -> QueryIntentDecision:
 
 def parse_confidence(value: object) -> float:
     try:
-        confidence = float(value)
+        confidence = float(cast(Any, value))
     except (TypeError, ValueError):
         return 0.0
 

{gaard_core-0.2.0 → gaard_core-0.2.2}/src/gaard_core/query_pipeline/models.py

@@ -29,6 +29,7 @@ class QueryIntentClassification(BaseModel):
 class QueryRequest(BaseModel):
     question: str = Field(min_length=1)
     datasource_id: str = "default"
+    datasource_ids: list[str] = Field(default_factory=list)
     user_id: str = "local-admin"
     interpret: bool = True
 

{gaard_core-0.2.0 → gaard_core-0.2.2}/src/gaard_core/sql_validator/select_only.py

@@ -19,7 +19,7 @@ class SelectOnlySqlValidator:
 
         statement = statements[0]
 
-        if not isinstance(statement, exp.Select):
+        if not isinstance(statement, (exp.Select, exp.SetOperation)):
             raise SqlValidationError(f"Only SELECT queries are allowed. {sql}")
 
         forbidden_expressions = (

{gaard_core-0.2.0 → gaard_core-0.2.2}/src/gaard_core.egg-info/PKG-INFO

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: gaard-core
-Version: 0.2.0
+Version: 0.2.2
 Summary: Core GAARD query pipeline, prompt compiler, policies and SQL validation
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
-Requires-Dist: gaard-plugin-api<0.3.0,>=0.2.0
+Requires-Dist: gaard-plugin-api<0.3.0,>=0.2.2
 Requires-Dist: pydantic>=2.7.0
 Requires-Dist: sqlglot>=25.0.0
 Provides-Extra: dev

{gaard_core-0.2.0 → gaard_core-0.2.2}/src/gaard_core.egg-info/SOURCES.txt

@@ -4,6 +4,7 @@ src/gaard_core/__init__.py
 src/gaard_core/errors.py
 src/gaard_core/json_utils.py
 src/gaard_core/llm_output.py
+src/gaard_core/py.typed
 src/gaard_core.egg-info/PKG-INFO
 src/gaard_core.egg-info/SOURCES.txt
 src/gaard_core.egg-info/dependency_links.txt

{gaard_core-0.2.0 → gaard_core-0.2.2}/src/gaard_core.egg-info/requires.txt

@@ -1,4 +1,4 @@
-gaard-plugin-api<0.3.0,>=0.2.0
+gaard-plugin-api<0.3.0,>=0.2.2
 pydantic>=2.7.0
 sqlglot>=25.0.0
 

{gaard_core-0.2.0 → gaard_core-0.2.2}/tests/test_sql_validator.py

@@ -22,6 +22,12 @@ def test_validator_accepts_mysql_quoted_identifier() -> None:
     validator.validate("SELECT COUNT(*) AS total_leads FROM `lead`")
 
 
+def test_validator_accepts_union_of_literal_selects() -> None:
+    validator = SelectOnlySqlValidator()
+
+    validator.validate("SELECT 'form_id' AS column_name UNION ALL SELECT 'status'")
+
+
 def test_validator_rejects_delete() -> None:
     validator = SelectOnlySqlValidator()