jSrSSKJr SSKJrJr SSKJr SSKJ r J r J r J r SSK r SSKJrJr SSKJr \"\5R)S 5rSS jr\SS j5r\SS j5rSS jrg)aPostgres/pgvector connection helper for the ADE RAG playground. Reuses `quber.settings` for credentials (it loads `.env` from the current working directory) and registers the pgvector type adapters so `VECTOR` columns round-trip as Python lists / numpy arrays. ) annotations)asynccontextmanagercontextmanager)Path)AsyncGenerator Generator LiteralStringcastN)register_vectorregister_vector_async) get_settingsz schema.sqlc [5RnSURSURSURSURSUR 3 $)Nzhost=z port=z dbname=z user=z password=)r dbhostportuserpassword)rs 1/home/mande/repo/quber/src/quber/playground/db.pydsnrsL   B 277)6"''(255'y SUS^S^R_ ``c## [R"[5SS9n[U5 Uv SSS5 g!,(df  g=f7f)a.Yield a connection with pgvector adapters registered and autocommit on. For code off the event loop: ingest, the CLI, scripts. A request handler or anything else running under asyncio uses `connect_async`, because a blocking query there holds the loop and stalls every other request's in-flight work. T autocommitN)psycopgconnectrr conns rrrs1 4 0D  1 0 0sA 9 A  AA c# [RR[5SS9IShvN IShvN n[ U5IShvN U7v SSS5IShvN gN6N/NN !,IShvN (df  g=f7f)zThe same connection for code on the event loop: queries are awaited, so the loop keeps serving other requests while Postgres works.TrN)rAsyncConnectionrrr rs r connect_asyncr )s_,,44SUt4LLLLPT#D))) MLL)MLLLsf*BA#BA%BA+A' A+ BA)B%B'A+)B+B1A4 2 B>Bc[[[R55n[ 5nUR U5 SSS5 g!,(df  g=f)zKCreate the `ade_playground` schema from schema.sql (drops existing tables).N)r r SCHEMA_FILE read_textrexecute)sqlrs r apply_schemar&2s6 }k335 6C d S s A A)returnstr)r'zGenerator[psycopg.Connection])r'z'AsyncGenerator[psycopg.AsyncConnection])r'None)__doc__ __future__r contextlibrrpathlibrtypingrrr r rpgvector.psycopgr r quber.settingsr __file__ with_namer"rrr r&rrr4sk#:AAC'8n&&|4 a   r