From 7c2a3f69a772080e1eb43375007aa24d992f5625 Mon Sep 17 00:00:00 2001
From: Thuc Pham <51660321+thucpn@users.noreply.github.com>
Date: Wed, 25 Sep 2024 16:24:14 +0700
Subject: [PATCH] fix: postgres import (#322)
---
.changeset/grumpy-knives-brake.md | 5 +++++
helpers/typescript.ts | 1 +
.../components/loaders/typescript/llama_parse/loader.ts | 2 +-
templates/components/vectordbs/typescript/pg/generate.ts | 8 ++------
templates/components/vectordbs/typescript/pg/index.ts | 4 ++--
5 files changed, 11 insertions(+), 9 deletions(-)
create mode 100644 .changeset/grumpy-knives-brake.md
diff --git a/.changeset/grumpy-knives-brake.md b/.changeset/grumpy-knives-brake.md
new file mode 100644
index 00000000..809e8f38
--- /dev/null
+++ b/.changeset/grumpy-knives-brake.md
@@ -0,0 +1,5 @@
+---
+"create-llama": patch
+---
+
+fix: postgres import
diff --git a/helpers/typescript.ts b/helpers/typescript.ts
index 9b4c9ffc..6dc3d6b7 100644
--- a/helpers/typescript.ts
+++ b/helpers/typescript.ts
@@ -261,6 +261,7 @@ async function updatePackageJson({
packageJson.dependencies = {
...packageJson.dependencies,
pg: "^8.12.0",
+ pgvector: "^0.2.0",
};
}
diff --git a/templates/components/loaders/typescript/llama_parse/loader.ts b/templates/components/loaders/typescript/llama_parse/loader.ts
index d5d7c08f..916c5867 100644
--- a/templates/components/loaders/typescript/llama_parse/loader.ts
+++ b/templates/components/loaders/typescript/llama_parse/loader.ts
@@ -1,4 +1,4 @@
-import { LlamaParseReader } from "llamaindex/readers/LlamaParseReader";
+import { LlamaParseReader } from "llamaindex";
import {
FILE_EXT_TO_READER,
SimpleDirectoryReader,
diff --git a/templates/components/vectordbs/typescript/pg/generate.ts b/templates/components/vectordbs/typescript/pg/generate.ts
index f5664b6f..2af9630e 100644
--- a/templates/components/vectordbs/typescript/pg/generate.ts
+++ b/templates/components/vectordbs/typescript/pg/generate.ts
@@ -1,10 +1,6 @@
-/* eslint-disable turbo/no-undeclared-env-vars */
import * as dotenv from "dotenv";
-import {
- PGVectorStore,
- VectorStoreIndex,
- storageContextFromDefaults,
-} from "llamaindex";
+import { VectorStoreIndex, storageContextFromDefaults } from "llamaindex";
+import { PGVectorStore } from "llamaindex/vector-store/PGVectorStore";
import { getDocuments } from "./loader";
import { initSettings } from "./settings";
import {
diff --git a/templates/components/vectordbs/typescript/pg/index.ts b/templates/components/vectordbs/typescript/pg/index.ts
index 6a4f1370..0ca1dc46 100644
--- a/templates/components/vectordbs/typescript/pg/index.ts
+++ b/templates/components/vectordbs/typescript/pg/index.ts
@@ -1,5 +1,5 @@
-/* eslint-disable turbo/no-undeclared-env-vars */
-import { PGVectorStore, VectorStoreIndex } from "llamaindex";
+import { VectorStoreIndex } from "llamaindex";
+import { PGVectorStore } from "llamaindex/vector-store/PGVectorStore";
import {
PGVECTOR_SCHEMA,
PGVECTOR_TABLE,
--
GitLab