feat: support ppl data grid type in save to notebook function (#93)

* feat: support ppl data grid type in save to notebook function Signed-off-by: tygao <tygao@amazon.com> * update Signed-off-by: tygao <tygao@amazon.com> --------- Signed-off-by: tygao <tygao@amazon.com>
opensearch-project · Jan 8, 2024 · e278a47 · e278a47
1 parent 4c0386b
commit e278a47
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/common/types/chat_saved_object_attributes.ts b/common/types/chat_saved_object_attributes.ts
@@ -43,7 +43,8 @@ export interface IOutput {
   type: 'output';
   traceId?: string; // used for tracing agent calls
   toolsUsed?: string[];
-  contentType: 'error' | 'markdown' | 'visualization' | 'ppl_visualization';
+  // TODO: ppl_visualization type may need to be removed in the PR which replaces ppl query render from visualization to data grid. @suzhou
+  contentType: 'error' | 'markdown' | 'visualization' | 'ppl_visualization' | 'ppl_data_grid';
   content: string;
   suggestedActions?: ISuggestedAction[];
   messageId?: string;

diff --git a/public/utils/notebook.ts b/public/utils/notebook.ts
@@ -92,6 +92,20 @@ export const convertMessagesToParagraphs = (messages: IMessage[], username: stri
         });
         break;
 
+      case 'ppl_data_grid':
+        const queryText = message.content;
+        Object.assign(paragraph, {
+          input: { inputText: `%ppl\n${queryText}`, inputType: 'MARKDOWN' },
+          output: [
+            {
+              result: `\n${queryText}`,
+              outputType: 'QUERY',
+              execution_time: '0 ms',
+            },
+          ],
+        });
+        break;
+
       case 'visualization':
         const visualizationObjectId = message.content;
         const inputText = JSON.stringify(createDashboardVizObject(visualizationObjectId));