Skip to content

Commit 254d13a

Browse files
committed
Fixed neo4j pruning process
1 parent 5365704 commit 254d13a

File tree

1 file changed

+20
-16
lines changed

1 file changed

+20
-16
lines changed

ai-assistant/src/endpoints/purgeDB.ts

Lines changed: 20 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -27,27 +27,31 @@ export class PurgeDBEndpoint extends ApiEndpoint {
2727
async setupIndices(db: IDB) {
2828
const query = [
2929
// Drop existing indices
30-
"DROP INDEX `nameEmbeddings`;",
31-
"DROP INDEX `codeEmbeddings`;",
30+
["DROP INDEX `nameEmbeddings`;"],
31+
["DROP INDEX `codeEmbeddings`;"],
3232

3333
// Create indices for name embeddings
34-
"CREATE VECTOR INDEX `nameEmbeddings`",
35-
"FOR (n: Node) ON (n.nameEmbeddings)",
36-
"OPTIONS {indexConfig: {",
37-
" `vector.dimensions`: 384,",
38-
" `vector.similarity_function`: 'COSINE'",
39-
"}};",
34+
[
35+
"CREATE VECTOR INDEX `nameEmbeddings`",
36+
"FOR (n: Node) ON (n.nameEmbeddings)",
37+
"OPTIONS {indexConfig: {",
38+
" `vector.dimensions`: 384,",
39+
" `vector.similarity_function`: 'COSINE'",
40+
"}};",
41+
],
4042

4143
// Create indices for code embeddings
42-
"CREATE VECTOR INDEX `codeEmbeddings`",
43-
"FOR (n: Node) ON (n.codeEmbeddings)",
44-
"OPTIONS {indexConfig: {",
45-
" `vector.dimensions`: 384,",
46-
" `vector.similarity_function`: 'COSINE'",
47-
"}};",
48-
].join("\n");
44+
[
45+
"CREATE VECTOR INDEX `codeEmbeddings`",
46+
"FOR (n: Node) ON (n.codeEmbeddings)",
47+
"OPTIONS {indexConfig: {",
48+
" `vector.dimensions`: 384,",
49+
" `vector.similarity_function`: 'COSINE'",
50+
"}};",
51+
],
52+
].map((q) => q.join("\n"));
4953

50-
await db.run(query);
54+
for (const q of query) await db.run(q);
5155
}
5256

5357
public async post(

0 commit comments

Comments
 (0)