From 02b22da3847f5151fcea8cab3acbcf852cad6cc3 Mon Sep 17 00:00:00 2001
From: Madarco <github@madarco.it>
Date: Tue, 19 Nov 2024 17:52:56 +0000
Subject: [PATCH] fix: supports vercel bundling (#1509)
Co-authored-by: Alex Yang <himself65@outlook.com>
---
.changeset/cuddly-geckos-clap.md | 5 +++++
packages/llamaindex/src/vector-store/PGVectorStore.ts | 7 ++++++-
2 files changed, 11 insertions(+), 1 deletion(-)
create mode 100644 .changeset/cuddly-geckos-clap.md
diff --git a/.changeset/cuddly-geckos-clap.md b/.changeset/cuddly-geckos-clap.md
new file mode 100644
index 000000000..739f90d03
--- /dev/null
+++ b/.changeset/cuddly-geckos-clap.md
@@ -0,0 +1,5 @@
+---
+"llamaindex": patch
+---
+
+fix: supports Vercel bundling
diff --git a/packages/llamaindex/src/vector-store/PGVectorStore.ts b/packages/llamaindex/src/vector-store/PGVectorStore.ts
index 9bf9d19f8..4524a3f3c 100644
--- a/packages/llamaindex/src/vector-store/PGVectorStore.ts
+++ b/packages/llamaindex/src/vector-store/PGVectorStore.ts
@@ -176,7 +176,12 @@ export class PGVectorStore extends BaseVectorStore {
if ("clientConfig" in config) {
this.clientConfig = config.clientConfig;
} else {
- if (config.client.constructor.name.includes("Vercel")) {
+ if (
+ config.client.constructor.name.includes("Vercel") ||
+ (!!(config.client as VercelPool).connect &&
+ !!(config.client as VercelPool).query &&
+ !(config.client as Sql).unsafe)
+ ) {
this.isDBConnected = true;
this.db = fromVercelPool(config.client as unknown as VercelPool);
} else if (typeof config.client === "function") {
--
GitLab