Merge pull request #117 from Azure-Samples/pgdemo

Fix for vector extension order
Azure-Samples · Oct 23, 2024 · 8227c72 · 8227c72
2 parents cdb1f28 + 3f4674c
commit 8227c72
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/.vscode/settings.json b/.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,

diff --git a/src/backend/fastapi_app/postgres_engine.py b/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):