gaard-connectors 0.1.0__py3-none-any.whl

gaard_connectors/sqlalchemy/executor.py

@@ -0,0 +1,51 @@
+import re
+from collections.abc import Mapping
+from typing import Any, cast
+
+from sqlalchemy import create_engine, text
+from sqlalchemy.engine import Engine, RowMapping
+from sqlalchemy.exc import SQLAlchemyError
+
+from gaard_core.errors import QueryExecutionError
+from gaard_core.json_utils import to_jsonable
+from gaard_core.query_pipeline.models import QueryResult
+
+
+class SQLAlchemyQueryExecutor:
+    def __init__(self, database_url: str, max_rows: int = 100) -> None:
+        self.database_url = database_url
+        self.max_rows = max_rows
+        self.engine: Engine = create_engine(database_url)
+
+    def execute(self, sql: str) -> QueryResult:
+        limited_sql = self._apply_limit(sql)
+
+        try:
+            with self.engine.connect() as connection:
+                result = connection.execute(text(limited_sql))
+                rows = result.mappings().fetchall()
+        except SQLAlchemyError as exc:
+            raise QueryExecutionError(
+                f"Query execution failed. SQL: {limited_sql}. Error: {exc}",
+                sql=limited_sql,
+                error_detail=str(exc),
+            ) from exc
+
+        normalized_rows = [self._normalize_row(row) for row in rows]
+        columns = list(normalized_rows[0].keys()) if normalized_rows else []
+
+        return QueryResult(
+            columns=columns,
+            rows=normalized_rows,
+        )
+
+    def _apply_limit(self, sql: str) -> str:
+        normalized = sql.strip().rstrip(";")
+
+        if re.search(r"\blimit\s+\d+\b", normalized, flags=re.IGNORECASE):
+            return normalized
+
+        return f"{normalized} LIMIT {self.max_rows}"
+
+    def _normalize_row(self, row: Mapping[str, Any] | RowMapping) -> dict[str, Any]:
+        return cast(dict[str, Any], to_jsonable(dict(row)))

gaard_connectors/sqlalchemy/introspector.py

@@ -0,0 +1,62 @@
+from typing import Literal
+
+from sqlalchemy import create_engine, inspect
+from sqlalchemy.engine import Engine
+from sqlalchemy.engine.reflection import Inspector
+
+from gaard_core.schema.models import ColumnInfo, DatabaseSchema, ForeignKeyInfo, TableInfo
+
+
+class SQLAlchemySchemaIntrospector:
+    def __init__(self, database_url: str) -> None:
+        self.database_url = database_url
+        self.engine: Engine = create_engine(database_url)
+
+    def introspect(self) -> DatabaseSchema:
+        inspector = inspect(self.engine)
+
+        tables: list[TableInfo] = []
+
+        for table_name in inspector.get_table_names():
+            tables.append(self._introspect_table(inspector, table_name, "table"))
+
+        for view_name in inspector.get_view_names():
+            tables.append(self._introspect_table(inspector, view_name, "view"))
+
+        return DatabaseSchema(tables=tables)
+
+    def _introspect_table(
+        self,
+        inspector: Inspector,
+        name: str,
+        object_type: Literal["table", "view"],
+    ) -> TableInfo:
+        columns = [
+            ColumnInfo(
+                name=column["name"],
+                type=str(column["type"]),
+                nullable=bool(column.get("nullable", True)),
+                primary_key=bool(column.get("primary_key", False)),
+            )
+            for column in inspector.get_columns(name)
+        ]
+
+        foreign_keys = []
+
+        if object_type == "table":
+            foreign_keys = [
+                ForeignKeyInfo(
+                    constrained_columns=list(foreign_key.get("constrained_columns") or []),
+                    referred_table=str(foreign_key.get("referred_table")),
+                    referred_columns=list(foreign_key.get("referred_columns") or []),
+                )
+                for foreign_key in inspector.get_foreign_keys(name)
+                if foreign_key.get("referred_table")
+            ]
+
+        return TableInfo(
+            name=name,
+            object_type=object_type,
+            columns=columns,
+            foreign_keys=foreign_keys,
+        )

gaard_connectors-0.1.0.dist-info/METADATA

@@ -0,0 +1,31 @@
+Metadata-Version: 2.4
+Name: gaard-connectors
+Version: 0.1.0
+Summary: Database connectors for GAARD based on SQLAlchemy
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+Requires-Dist: gaard-core==0.1.0
+Requires-Dist: sqlalchemy>=2.0.0
+Provides-Extra: postgres
+Requires-Dist: psycopg[binary]>=3.2.0; extra == "postgres"
+Provides-Extra: mysql
+Requires-Dist: pymysql>=1.1.0; extra == "mysql"
+Provides-Extra: mssql
+Requires-Dist: pyodbc>=5.1.0; extra == "mssql"
+Provides-Extra: oracle
+Requires-Dist: oracledb>=2.0.0; extra == "oracle"
+Provides-Extra: dev
+Requires-Dist: pytest>=8.0.0; extra == "dev"
+Requires-Dist: ruff>=0.5.0; extra == "dev"
+Requires-Dist: mypy>=1.10.0; extra == "dev"
+
+# GAARD - Governed AI Access to Relational Data
+
+GAARD is a self-hosted AI SQL Gateway for governed natural-language access to relational data.
+
+GAARD allows applications and users to ask questions about relational databases using natural language while keeping SQL generation, validation, execution, prompts, connectors, and auditability under control.
+
+For more informacion see https://github.com/pkroliszewski/gaard
+
+# This package
+Package gaard-connectors extends gaard functionality by adding support for various database connection schemes

gaard_connectors-0.1.0.dist-info/RECORD

@@ -0,0 +1,13 @@
+gaard_connectors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+gaard_connectors/mssql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+gaard_connectors/mysql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+gaard_connectors/oracle/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+gaard_connectors/postgres/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+gaard_connectors/sqlalchemy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+gaard_connectors/sqlalchemy/executor.py,sha256=CwAYDCk5nFk9CjcRZEGlpCyVidXqOH6BErawYQMPNow,1775
+gaard_connectors/sqlalchemy/introspector.py,sha256=RDmxqKdCxXqs3gyQBNRaLJ6uqTk-BSx0hFSZnm2wKGw,2067
+gaard_connectors/sqlite/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+gaard_connectors-0.1.0.dist-info/METADATA,sha256=A0uMaGaf__qt5tJCUrCAsHvzlP5nWLC6rC6npHdq8Vg,1244
+gaard_connectors-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+gaard_connectors-0.1.0.dist-info/top_level.txt,sha256=kih-wrFnx5FwDucrAcPgxDJe5N4lZwYDoxnR9PhBA64,17
+gaard_connectors-0.1.0.dist-info/RECORD,,

gaard_connectors-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

gaard_connectors-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+gaard_connectors