Expose findRelevantPassages as REST service (#135)

Signed-off-by: Sean Sundberg <[email protected]>

src/controllers/data-extraction/data-extraction.controller.ts

@@ -1,12 +1,13 @@
-import {Controller, Get, Query} from "@nestjs/common";
+import {Controller, Get, Post, Query} from "@nestjs/common";
 
 import {DataExtractionApi} from "../../services/data-extraction";
 import {ApiTags} from "@nestjs/swagger";
 
 @ApiTags('data-extraction')
 @Controller('data-extraction')
 export class DataExtractionController {
-    constructor(private readonly service: DataExtractionApi) {}
+    constructor(private readonly service: DataExtractionApi) {
+    }
 
     @Get('questions')
     listQuestions() {
@@ -36,4 +37,14 @@ export class DataExtractionController {
         return this.service.extractData(customer, questionIds.map(id => ({id})))
             .catch(err => console.error(err))
     }
+
+    @Get('findRelevantPassages')
+    async findRelevantPassages(
+        @Query('question') question: string,
+        @Query('passages') passages: string[] = []
+    ): Promise<string> {
+        const passageVals = Array.isArray(passages) ? passages : [passages]
+
+        return this.service.findRelevantPassages(question, passageVals)
+    }
 }

src/services/data-extraction/data-extraction.api.ts

@@ -10,4 +10,6 @@ export abstract class DataExtractionApi {
     abstract extractDataForQuestion(customer: string, question: {id: string}): Promise<DataExtractionResultModel>;
 
     abstract extractDataObservable(customer: string, questions: Array<{id: string}>): Observable<DataExtractionResultModel[]>;
+
+    abstract findRelevantPassages(question: string, passages: string[]): Promise<string>;
 }

src/services/data-extraction/data-extraction.impl.ts

@@ -200,6 +200,10 @@ export class DataExtractionImpl extends DataExtractionCsv<WatsonBackends, Contex
     async findRelevantPassages(question: string, passages: string[]): Promise<string> {
         const url = process.env.RELEVANT_PASSAGES_URL || 'https://similarity-check.18xu6cedovu0.us-south.codeengine.appdomain.cloud/api/find_relevant_passage'
 
+        if (passages.length === 1) {
+            return passages[0]
+        }
+
         return await queue
             .add(async () => {
                 console.log('Getting relevant passage')

src/services/data-extraction/data-extraction.mock.ts

@@ -28,4 +28,8 @@ export class DataExtractionMock extends DataExtractionCsv<{}, {}> implements Dat
         return {};
     }
 
+    async findRelevantPassages(question: string, passages: string[]): Promise<string> {
+        return ''
+    }
+
 }