From e278a47b93154286a17a5b3fa4c0ec37464ad56c Mon Sep 17 00:00:00 2001 From: tygao Date: Mon, 8 Jan 2024 17:20:28 +0800 Subject: [PATCH] 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 * update Signed-off-by: tygao --------- Signed-off-by: tygao --- common/types/chat_saved_object_attributes.ts | 3 ++- public/utils/notebook.ts | 14 ++++++++++++++ 2 files changed, 16 insertions(+), 1 deletion(-) diff --git a/common/types/chat_saved_object_attributes.ts b/common/types/chat_saved_object_attributes.ts index 1516eac6..608f83ff 100644 --- 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 index b9ead03d..f67d092d 100644 --- 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));