Merge pull request #117 from Azure-Samples/pgdemo

Fix for vector extension order

Author: pamelafox

.vscode/settings.json

@@ -28,6 +28,7 @@
     ],
     "python.testing.unittestEnabled": false,
     "python.testing.pytestEnabled": true,
+    "python.envFile": "",
     "files.exclude": {
         ".ruff_cache": true,
         ".pytest_cache": true,

src/backend/fastapi_app/postgres_engine.py

@@ -37,7 +37,10 @@ def get_password_from_azure_credential():
     @event.listens_for(engine.sync_engine, "connect")
     def register_custom_types(dbapi_connection: AdaptedConnection, *args):
         logger.info("Registering pgvector extension...")
-        dbapi_connection.run_async(register_vector)
+        try:
+            dbapi_connection.run_async(register_vector)
+        except ValueError:
+            logger.warning("Could not register pgvector data type yet as vector extension has not been CREATEd")
 
     @event.listens_for(engine.sync_engine, "do_connect")
     def update_password_token(dialect, conn_rec, cargs, cparams):