Add getRagSettings function to rag-settings.ts

n4ze3m · Apr 21, 2024 · 15b38c9 · 15b38c9
1 parent 3f9ebbf
commit 15b38c9
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/server/src/utils/rag-settings.ts b/server/src/utils/rag-settings.ts
@@ -0,0 +1,15 @@
+import { PrismaClient } from "@prisma/client";
+
+export const getRagSettings = async (prisma: PrismaClient) => {
+  const data = await prisma.dialoqbaseSettings.findFirst({
+    select: {
+      defaultChunkSize: true,
+      defaultChunkOverlap: true,
+    },
+  });
+
+  return {
+    chunkSize: data?.defaultChunkSize || 1000,
+    chunkOverlap: data?.defaultChunkOverlap || 200,
+  };
+};