RAG pipeline

Author: kyawzinthant-coding

dump.rdb

@@ -1,7 +1,7 @@
 import express from "express";
 import { pdfUploadFilesController } from "../controllers/pdf/pdf-upload";
 import { upload } from "../middleware./uploadfiles";
-import { pdfChatController } from "../controllers/pdf/pdg-chat";
+import { pdfChatController } from "../controllers/pdf/pdf-chat";
 const router = express.Router();
 
 router.post("/uploads", upload.array("files", 10), pdfUploadFilesController);

src/controllers/pdf/pdg-chat.ts renamed to src/controllers/pdf/pdf-chat.ts

@@ -3,7 +3,9 @@ import { GoogleGenerativeAI } from "@google/generative-ai";
 
 const genAI = new GoogleGenerativeAI(process.env.GEMINI_API_KEY!);
 
-const model = genAI.getGenerativeModel({ model: "gemini-pro" });
+const model = genAI.getGenerativeModel({
+  model: "gemini-2.5-flash-preview-05-20",
+});
 
 export async function generateChatResponse(
   question: string,

src/routes/PDF-routes.ts

@@ -1,7 +1,10 @@
 import { ChromaClient, EmbeddingFunction } from "chromadb";
 import { generateEmbedding } from "./embedding-service";
 
-const client = new ChromaClient();
+const client = new ChromaClient({
+  host: "127.0.0.1",
+  port: 8000,
+});
 
 class GoogleAIEmbeddingFunction implements EmbeddingFunction {
   public async generate(texts: string[]): Promise<number[][]> {
@@ -24,15 +27,17 @@ export async function getKnowledgeCollection() {
   return collection;
 }
 
-export async function queryKnowledgeBase(question: string): Promise<string[]> {
-  const collection = getKnowledgeCollection();
+export async function queryKnowledgeBase(question: string) {
+  const collection = await getKnowledgeCollection();
 
-  const results = (await collection).query({
+  const results = await collection.query({
     queryTexts: [question],
     nResults: 5,
   });
 
-  return (await results).documents[0];
+  console.log(results);
+
+  return results.documents[0];
 }
 
 export async function cleanUpPreviousData() {

src/service/ai-response-service.ts

@@ -0,0 +1,29 @@
+import { ChromaClient } from "chromadb";
+
+async function main() {
+  console.log("--- Running Isolated ChromaDB Connection Test ---");
+
+  console.log("Attempting to create client with host: 127.0.0.1, port: 8000");
+
+  const client = new ChromaClient({
+    host: "127.0.0.1",
+    port: 8000,
+  });
+
+  try {
+    console.log("Client created. Attempting to get heartbeat...");
+    const heartbeat = await client.heartbeat();
+    console.log("✅✅✅ SUCCESS: ChromaDB heartbeat received:", heartbeat);
+    console.log(
+      "\nThis confirms the connection from Node.js is working correctly."
+    );
+  } catch (error) {
+    console.error("🔴🔴🔴 FAILED: The isolated connection test failed.");
+    console.error(
+      "This means the issue is fundamental to the Node.js client or your environment."
+    );
+    console.error("The full error is:", error);
+  }
+}
+
+main();