From 568a221293057a8e82bd8019511f4c85193cee40 Mon Sep 17 00:00:00 2001
From: Philip Meier <github.pmeier@posteo.de>
Date: Tue, 19 May 2026 17:03:20 +0200
Subject: [PATCH 1/2] improve austion permits agent

---
 backend/pyproject.toml                        |  1 +
 backend/src/ravnar_nebari_chat/__init__.py    |  3 +-
 ...tin_permit_agent.py => _austin_permits.py} | 50 ++++++-------------
 3 files changed, 17 insertions(+), 37 deletions(-)
 rename backend/src/ravnar_nebari_chat/{austin_permit_agent.py => _austin_permits.py} (80%)

diff --git a/backend/pyproject.toml b/backend/pyproject.toml
index 4cae1c6..6ed9960 100644
--- a/backend/pyproject.toml
+++ b/backend/pyproject.toml
@@ -20,6 +20,7 @@ dependencies = [
     "httpx>=0.28.1",
     "openai>=2.35.1,<3",
     "pyjwt>=2.12.1,<3",
+    "sqlalchemy[asyncio]>=2.0,<3",
     "psycopg[binary,pool]>=3.2.9,<4",
     "ravnar[serve,pydantic-ai]==0.0.8",
 ]
diff --git a/backend/src/ravnar_nebari_chat/__init__.py b/backend/src/ravnar_nebari_chat/__init__.py
index 4412083..47be723 100644
--- a/backend/src/ravnar_nebari_chat/__init__.py
+++ b/backend/src/ravnar_nebari_chat/__init__.py
@@ -1,5 +1,6 @@
-__all__ = ["__version__", "keycloak_authenticator"]
+__all__ = ["__version__", "keycloak_authenticator", "make_austin_permits_agent"]
 
+from ._austin_permits import make_austin_permits_agent
 from ._authenticators import keycloak_authenticator
 
 try:
diff --git a/backend/src/ravnar_nebari_chat/austin_permit_agent.py b/backend/src/ravnar_nebari_chat/_austin_permits.py
similarity index 80%
rename from backend/src/ravnar_nebari_chat/austin_permit_agent.py
rename to backend/src/ravnar_nebari_chat/_austin_permits.py
index aff3d8b..b7ff279 100644
--- a/backend/src/ravnar_nebari_chat/austin_permit_agent.py
+++ b/backend/src/ravnar_nebari_chat/_austin_permits.py
@@ -1,48 +1,26 @@
 import uuid
-from typing import Any
 
-import psycopg
 from ag_ui.core import ActivitySnapshotEvent
 from pydantic_ai import Agent, ToolReturn
+from sqlalchemy import text
+from sqlalchemy.ext.asyncio import create_async_engine
 
 from ravnar.agents import PydanticAiAgentWrapper
 
 
-class PostgresDatabase:
-    def __init__(
-        self,
-        *,
-        host: str,
-        port: int = 5432,
-        name: str,
-        user: str,
-        password: str,
-    ) -> None:
-        self._host = host
-        self._port = port
-        self._name = name
-        self._user = user
-        self._password = password
-
-    async def _connect(self) -> psycopg.AsyncConnection[Any]:
-        return await psycopg.AsyncConnection.connect(
-            dbname=self._name,
-            user=self._user,
-            password=self._password,
-            host=self._host,
-            port=self._port,
-        )
-
-    async def execute(self, query: str) -> list[tuple]:
-        async with await self._connect() as conn, conn.cursor() as cur:
-            await cur.execute(query)
-            return list(await cur.fetchall())
-
 
-def create_agent(
+def make_austin_permits_agent(
     agent: Agent,
-    database: PostgresDatabase,
+    *,
+    database_url: str,
 ) -> PydanticAiAgentWrapper:
+    engine = create_async_engine(database_url)
+
+    async def execute(query: str) -> list[tuple]:
+        async with engine.connect() as conn:
+            result = await conn.execute(text(query))
+            return [tuple(row) for row in result.fetchall()]
+
     @agent.system_prompt
     def _system_prompt() -> str:
         return """
@@ -76,7 +54,7 @@ async def get_db_schema() -> list[tuple]:
             WHERE t.table_schema = 'public' AND t.table_type = 'BASE TABLE'
             ORDER BY t.table_name, c.ordinal_position;
         """
-        return await database.execute(query)
+        return await execute(query)
 
     @agent.tool_plain
     async def execute_query(query: str) -> list[tuple]:
@@ -91,7 +69,7 @@ async def execute_query(query: str) -> list[tuple]:
         SQL syntax.
 
         """
-        return await database.execute(query)
+        return await execute(query)
 
     @agent.tool_plain
     async def create_chart(option: dict) -> ToolReturn:

From 340d1ece2b9dce32e8fc74f215dd0d2c8c6ca5a2 Mon Sep 17 00:00:00 2001
From: Philip Meier <github.pmeier@posteo.de>
Date: Tue, 19 May 2026 17:05:37 +0200
Subject: [PATCH 2/2] lock

---
 backend/uv.lock | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/backend/uv.lock b/backend/uv.lock
index 3d09013..249ac18 100644
--- a/backend/uv.lock
+++ b/backend/uv.lock
@@ -1633,6 +1633,7 @@ dependencies = [
     { name = "psycopg", extra = ["binary", "pool"] },
     { name = "pyjwt" },
     { name = "ravnar", extra = ["pydantic-ai", "serve"] },
+    { name = "sqlalchemy", extra = ["asyncio"] },
 ]
 
 [package.dev-dependencies]
@@ -1657,6 +1658,7 @@ requires-dist = [
     { name = "psycopg", extras = ["binary", "pool"], specifier = ">=3.2.9,<4" },
     { name = "pyjwt", specifier = ">=2.12.1,<3" },
     { name = "ravnar", extras = ["pydantic-ai", "serve"], specifier = "==0.0.8" },
+    { name = "sqlalchemy", extras = ["asyncio"], specifier = ">=2.0,<3" },
 ]
 
 [package.metadata.requires-dev]