added EMBEDDING_DIM if available or undefined to fallback to default …

…config (#487)
run-llama · Jan 22, 2025 · 18c8d25 · 18c8d25
1 parent d4b4338
commit 18c8d25
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/templates/components/vectordbs/typescript/pg/index.ts b/templates/components/vectordbs/typescript/pg/index.ts
@@ -14,6 +14,9 @@ export async function getDataSource(params?: any) {
     },
     schemaName: PGVECTOR_SCHEMA,
     tableName: PGVECTOR_TABLE,
+    dimensions: process.env.EMBEDDING_DIM
+      ? parseInt(process.env.EMBEDDING_DIM)
+      : undefined,
   });
   return await VectorStoreIndex.fromVectorStore(pgvs);
 }