From d9bbaf95f3f70e38d6144d1841d6351832f9a5b9 Mon Sep 17 00:00:00 2001
From: siquick <siquick@users.noreply.github.com>
Date: Tue, 4 Feb 2025 15:18:39 +1100
Subject: [PATCH] chore: updated docs and examples to use correct PGVectorStore
imports (#1611)
Co-authored-by: Marcus Schiesser <mail@marcusschiesser.de>
---
e2e/node/vector-store/pg-vector-store.e2e.ts | 3 ++-
examples/vector-store/pg/load-docs.ts | 2 +-
examples/vector-store/pg/neon.ts | 3 ++-
examples/vector-store/pg/query.ts | 3 ++-
examples/vector-store/pg/supabase.ts | 3 ++-
examples/vector-store/pg/vercel.ts | 3 ++-
6 files changed, 11 insertions(+), 6 deletions(-)
diff --git a/e2e/node/vector-store/pg-vector-store.e2e.ts b/e2e/node/vector-store/pg-vector-store.e2e.ts
index 8bcb25976..b1725a1a8 100644
--- a/e2e/node/vector-store/pg-vector-store.e2e.ts
+++ b/e2e/node/vector-store/pg-vector-store.e2e.ts
@@ -1,5 +1,6 @@
import { config } from "dotenv";
-import { Document, PGVectorStore, VectorStoreQueryMode } from "llamaindex";
+import { Document, VectorStoreQueryMode } from "llamaindex";
+import { PGVectorStore } from "llamaindex/vector-store/PGVectorStore";
import assert from "node:assert";
import { test } from "node:test";
import pg from "pg";
diff --git a/examples/vector-store/pg/load-docs.ts b/examples/vector-store/pg/load-docs.ts
index 2152ac0ad..1932a472c 100755
--- a/examples/vector-store/pg/load-docs.ts
+++ b/examples/vector-store/pg/load-docs.ts
@@ -1,10 +1,10 @@
// load-docs.ts
import {
- PGVectorStore,
SimpleDirectoryReader,
storageContextFromDefaults,
VectorStoreIndex,
} from "llamaindex";
+import { PGVectorStore } from "llamaindex/vector-store/PGVectorStore";
import fs from "node:fs/promises";
async function getSourceFilenames(sourceDir: string) {
diff --git a/examples/vector-store/pg/neon.ts b/examples/vector-store/pg/neon.ts
index 5baa79a6e..877ccc71f 100644
--- a/examples/vector-store/pg/neon.ts
+++ b/examples/vector-store/pg/neon.ts
@@ -1,5 +1,6 @@
import dotenv from "dotenv";
-import { Document, PGVectorStore, VectorStoreQueryMode } from "llamaindex";
+import { Document, VectorStoreQueryMode } from "llamaindex";
+import { PGVectorStore } from "llamaindex/vector-store/PGVectorStore";
import postgres from "postgres";
dotenv.config();
diff --git a/examples/vector-store/pg/query.ts b/examples/vector-store/pg/query.ts
index 4d88e1cbd..f8d73fc4a 100755
--- a/examples/vector-store/pg/query.ts
+++ b/examples/vector-store/pg/query.ts
@@ -1,4 +1,5 @@
-import { PGVectorStore, VectorStoreIndex } from "llamaindex";
+import { VectorStoreIndex } from "llamaindex";
+import { PGVectorStore } from "llamaindex/vector-store/PGVectorStore";
async function main() {
// eslint-disable-next-line @typescript-eslint/no-require-imports
diff --git a/examples/vector-store/pg/supabase.ts b/examples/vector-store/pg/supabase.ts
index 748b48362..1cbbb5354 100644
--- a/examples/vector-store/pg/supabase.ts
+++ b/examples/vector-store/pg/supabase.ts
@@ -1,11 +1,12 @@
import dotenv from "dotenv";
import {
- PGVectorStore,
SimpleDirectoryReader,
storageContextFromDefaults,
VectorStoreIndex,
} from "llamaindex";
+import { PGVectorStore } from "llamaindex/vector-store/PGVectorStore";
+
dotenv.config();
// Get direct connection string from Supabase and set it as POSTGRES_URL environment variable
diff --git a/examples/vector-store/pg/vercel.ts b/examples/vector-store/pg/vercel.ts
index 7f8cfd0c3..0e6a6bbda 100644
--- a/examples/vector-store/pg/vercel.ts
+++ b/examples/vector-store/pg/vercel.ts
@@ -1,7 +1,8 @@
// https://vercel.com/docs/storage/vercel-postgres/sdk
import { sql } from "@vercel/postgres";
import dotenv from "dotenv";
-import { Document, PGVectorStore, VectorStoreQueryMode } from "llamaindex";
+import { Document, VectorStoreQueryMode } from "llamaindex";
+import { PGVectorStore } from "llamaindex/vector-store/PGVectorStore";
dotenv.config();
--
GitLab