From 3f4674cfd57377dc86a89b6b412e7876aaecec11 Mon Sep 17 00:00:00 2001 From: Pamela Fox Date: Wed, 23 Oct 2024 05:01:23 +0000 Subject: [PATCH] Fix for vector extension order --- .vscode/settings.json | 1 + src/backend/fastapi_app/postgres_engine.py | 5 ++++- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/.vscode/settings.json b/.vscode/settings.json index 6c6ce906..c9eb00cc 100644 --- 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 index 621dda1d..8de2324b 100644 --- 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):