Include LLM prompt in data extraction result (#51)

closes #49

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

schema.gql

@@ -49,6 +49,7 @@ type DataExtractionResult {
   expectedResponse: String!
   id: ID!
   inScope: Boolean!
+  prompt: String!
   question: String!
   watsonxResponse: String!
 }

src/graphql-types/data-extraction.graphql.ts

@@ -31,5 +31,7 @@ export class DataExtractionResult implements DataExtractionResultModel {
     expectedResponse: string;
     @Field()
     watsonxResponse: string;
+    @Field()
+    prompt: string;
 }
 

src/models/data-extraction.model.ts

@@ -5,6 +5,7 @@ export interface DataExtractionQuestionModel {
 }
 
 export interface DataExtractionResultModel extends DataExtractionQuestionModel {
+    prompt: string;
     expectedResponse: string;
     watsonxResponse: string;
 }

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

@@ -88,14 +88,15 @@ export class DataExtractionImpl extends DataExtractionCsv<WatsonBackends> implem
 
         const text = await this.queryDiscovery(customer, config, backends);
 
-        const watsonxResponse = await this.generateResponse(customer, config, text, backends);
+        const {watsonxResponse, prompt} = await this.generateResponse(customer, config, text, backends);
 
         return {
             id: config.id,
             question: config.question,
             inScope: config.inScope,
             expectedResponse: config.expectedResponse,
             watsonxResponse,
+            prompt,
         }
     }
 
@@ -140,7 +141,7 @@ export class DataExtractionImpl extends DataExtractionCsv<WatsonBackends> implem
             .join(' ')
     }
 
-    async generateResponse(customer: string, config: DataExtractionConfig, text: string, backends: WatsonBackends): Promise<string> {
+    async generateResponse(customer: string, config: DataExtractionConfig, text: string, backends: WatsonBackends): Promise<{watsonxResponse: string, prompt: string}> {
 
         const prompt = (config.prompt || `From below text find answer for ${config.question} ${customer}`).replace('#', customer);
         // const prompt = (`From below text find answer for ${config.question} ${customer}`).replace('#', customer);
@@ -160,7 +161,7 @@ export class DataExtractionImpl extends DataExtractionCsv<WatsonBackends> implem
 
         console.log('2. Text generated from watsonx.ai:', {prompt, generatedText: result.generatedText, input})
 
-        return result.generatedText;
+        return {watsonxResponse: result.generatedText, prompt: input};
     }
 
     async getBackends(): Promise<WatsonBackends> {