feat: Add Number of Retrieved Documents on RAG

2024-08-20 18:32:01 +05:30
parent 564b137e66
commit 35b8579028
14 changed files with 133 additions and 31 deletions
--- a/src/services/app.ts
+++ b/src/services/app.ts
@@ -108,4 +108,13 @@ export const getTotalFilePerKB = async (): Promise<number> => {

 export const setTotalFilePerKB = async (totalFilePerKB: number): Promise<void> => {
  await storage.set("totalFilePerKB", totalFilePerKB);
-};
+};
+
+export const getNoOfRetrievedDocs = async (): Promise<number> => {
+  const noOfRetrievedDocs = await storage.get<number>("noOfRetrievedDocs");
+  return noOfRetrievedDocs || 4 
+}
+
+export const setNoOfRetrievedDocs = async (noOfRetrievedDocs: number): Promise<void> => {
+  await storage.set("noOfRetrievedDocs", noOfRetrievedDocs);
+}