From 71899ec792183bb180ee24016f40516dc3a8841c Mon Sep 17 00:00:00 2001 From: Bartosz Sypytkowski Date: Fri, 28 Jun 2024 08:59:53 +0200 Subject: [PATCH 1/2] chore: disabling indexing on workspace prunes collab embeddings --- libs/database/src/workspace.rs | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/libs/database/src/workspace.rs b/libs/database/src/workspace.rs index 1fecebcd..926e3c02 100644 --- a/libs/database/src/workspace.rs +++ b/libs/database/src/workspace.rs @@ -814,6 +814,20 @@ pub async fn upsert_workspace_settings( .execute(tx.deref_mut()) .await?; + if settings.disable_search_indexing { + sqlx::query!( + r#"DELETE FROM af_collab_embeddings e WHERE e.oid in ( + SELECT c.oid + FROM af_collab c + WHERE c.partition_key = e.partition_key + AND c.oid = e.oid + AND c.workspace_id = $1)"#, + workspace_id + ) + .execute(tx.deref_mut()) + .await?; + } + Ok(()) } From bf5b122557473df818466d86fee785e68fd4f12f Mon Sep 17 00:00:00 2001 From: Bartosz Sypytkowski Date: Fri, 28 Jun 2024 09:26:19 +0200 Subject: [PATCH 2/2] chore: generate sqlx offline query definitions --- ...1d1b603f63e80e4f05a45a0790dd0bde892429ef14.json | 14 ++++++++++++++ 1 file changed, 14 insertions(+) create mode 100644 .sqlx/query-e5d51236e5c96ac91c70cd1d1b603f63e80e4f05a45a0790dd0bde892429ef14.json diff --git a/.sqlx/query-e5d51236e5c96ac91c70cd1d1b603f63e80e4f05a45a0790dd0bde892429ef14.json b/.sqlx/query-e5d51236e5c96ac91c70cd1d1b603f63e80e4f05a45a0790dd0bde892429ef14.json new file mode 100644 index 00000000..256dd3e3 --- /dev/null +++ b/.sqlx/query-e5d51236e5c96ac91c70cd1d1b603f63e80e4f05a45a0790dd0bde892429ef14.json @@ -0,0 +1,14 @@ +{ + "db_name": "PostgreSQL", + "query": "DELETE FROM af_collab_embeddings e WHERE e.oid in (\n SELECT c.oid\n FROM af_collab c\n WHERE c.partition_key = e.partition_key\n AND c.oid = e.oid\n AND c.workspace_id = $1)", + "describe": { + "columns": [], + "parameters": { + "Left": [ + "Uuid" + ] + }, + "nullable": [] + }, + "hash": "e5d51236e5c96ac91c70cd1d1b603f63e80e4f05a45a0790dd0bde892429ef14" +}