added cache_threshold flagsmith variable

src/modules/prompt-history/prompt-history.controller.ts

@@ -0,0 +1,15 @@
+import { Body, Controller, Post, Get, HttpException, HttpStatus, Param, Delete, NotFoundException } from "@nestjs/common";
+import { PromptHistoryService } from "./prompt-history.service";
+import { document as Document, Prisma } from "@prisma/client";
+import { SearchPromptHistoryDto } from "./prompt.dto";
+
+@Controller("history")
+export class PromptHistoryController {
+  constructor(private readonly promptHistoryService: PromptHistoryService) {}
+  @Post("/searchSimilar")
+  async findByCriteria(
+    @Body() searchQueryDto: SearchPromptHistoryDto
+  ): Promise<Document[]> {
+    return this.promptHistoryService.findByCriteria(searchQueryDto);
+  }
+}

src/modules/prompt-history/prompt.dto.ts

@@ -33,11 +33,6 @@ export class SearchPromptHistoryDto {
   @IsDefined({ message: "Query needs to be defined to search documents" })
   query: string;
 
-  @IsDefined({
-    message: "Unique Identifier for PDF",
-  })
-  pdfIds: Array<string>;
-
   @IsDefined({ message: "Similarity Threashold needs to be defined" })
   similarityThreshold: number;
 

src/xstate/prompt/prompt.service.ts

@@ -59,16 +59,16 @@ export const promptServices = {
     },
 
     findSimilarQuestion: async (context) => {
-        let pdfIds = context.prompt.input.pdfId ? 
-                        [context.prompt.input.pdfId] : 
-                        context.prompt.input.pdfIds && context.prompt.input.pdfIds.length ?
-                        context.prompt.input.pdfIds :
-                        JSON.parse(configService.get('DEFAULT_PDFS'))
+        const flags = await flagsmith.getIdentityFlags(context.prompt.input.userId);
+
+        let similarityThreshold = flags.getFeatureValue('cache_threshold');
+        if(similarityThreshold) similarityThreshold = parseFloat(similarityThreshold)
+        else similarityThreshold = 0.95
+
         const olderSimilarQuestion =
         await promptHistoryService.findByCriteria({
-            query: context.prompt.neuralCoreference,
-            pdfIds,
-            similarityThreshold: 0.97,
+            query: context.prompt.input.body,
+            similarityThreshold,
             matchCount: 1,
         });
         return olderSimilarQuestion