Display prompt information with data extraction

Signed-off-by: Sean Sundberg <[email protected]>
ibm-ecosystem-engineering · Sep 24, 2023 · 347c3f3 · 347c3f3
1 parent d13a762
commit 347c3f3
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 0 deletions.
diff --git a/src/models/data-extraction.model.ts b/src/models/data-extraction.model.ts
@@ -7,4 +7,5 @@ export interface DataExtractionQuestionModel {
 export interface DataExtractionResultModel extends DataExtractionQuestionModel {
     expectedResponse: string;
     watsonxResponse: string;
+    prompt: string;
 }
diff --git a/src/services/data-extraction/data-extraction.graphql.ts b/src/services/data-extraction/data-extraction.graphql.ts
@@ -44,6 +44,7 @@ const DATA_EXTRACTION_QUESTIONS = gql`
             inScope
             expectedResponse
             watsonxResponse
+            prompt
         }
     }
 `
@@ -59,6 +60,7 @@ const DATA_EXTRACTION_QUESTION = gql`
             inScope
             expectedResponse
             watsonxResponse
+            prompt
         }
     }
 `

diff --git a/src/services/schema.gql b/src/services/schema.gql
@@ -49,6 +49,7 @@ type DataExtractionResult {
   expectedResponse: String!
   id: ID!
   inScope: Boolean!
+  prompt: String!
   question: String!
   watsonxResponse: String!
 }

diff --git a/src/views/DataExtraction/DataExtraction.tsx b/src/views/DataExtraction/DataExtraction.tsx
@@ -188,6 +188,7 @@ const DataExtractionResults = () => {
 
     const headerData: DataTableHeader[] = [
         {header: 'Question', key: 'question'},
+        {header: 'Prompt', key: 'prompt'},
         {header: 'WatsonX Response', key: 'watsonxResponse'}
     ]
-Original file line number
+Diff line change
@@ Expand Up / @@ -44,6 +44,7 @@ const DATA_EXTRACTION_QUESTIONS = gql` @@
                 inScope
                 expectedResponse
                 watsonxResponse
+                prompt
             }
         }
     `
@@ Expand All / @@ -59,6 +60,7 @@ const DATA_EXTRACTION_QUESTION = gql` @@
                 inScope
                 expectedResponse
                 watsonxResponse
+                prompt
             }
         }
     `
@@ Expand Down @@