@thoughtspot/visual-embed-sdk 1.33.0 → 1.33.1-dc.3

package/src/utils/graphql/nlsService/conversation-queries.ts

@@ -0,0 +1,310 @@
+export const createConversation = `
+mutation CreateConversation($params: Input_convassist_CreateConversationRequest) {
+  ConvAssist__createConversation(request: $params) {
+    convId
+    initialCtx {
+      type
+      tsAnsCtx {
+        sessionId
+        genNo
+        stateKey {
+          transactionId
+          generationNumber
+        }
+        worksheet {
+          worksheetId
+          worksheetName
+        }
+      }
+    }
+  }
+}
+`;
+
+export const sendMessage = `
+query SendMessage($params: Input_convassist_SendMessageRequest) {
+  ConvAssist__sendMessage(request: $params) {
+    responses {
+      timestamp
+      msgId
+      data {
+        asstRespData {
+          tool
+          asstRespText
+          nlsAnsData {
+            sageQuerySuggestions {
+              sageQueryTokens {
+                additions {
+                  phrase {
+                    isCompletePhrase
+                    numTokens
+                    phraseType
+                    startIndex
+                    __typename
+                  }
+                  tokens {
+                    token
+                    dataType
+                    typeEnum
+                    guid
+                    tokenMetadata {
+                      name
+                      __typename
+                    }
+                    __typename
+                  }
+                  __typename
+                }
+                phrases {
+                  isCompletePhrase
+                  numTokens
+                  phraseType
+                  startIndex
+                  __typename
+                }
+                removals {
+                  phrase {
+                    isCompletePhrase
+                    numTokens
+                    phraseType
+                    startIndex
+                    __typename
+                  }
+                  tokens {
+                    token
+                    dataType
+                    typeEnum
+                    guid
+                    tokenMetadata {
+                      name
+                      __typename
+                    }
+                    __typename
+                  }
+                  __typename
+                }
+                tokens {
+                  token
+                  dataType
+                  typeEnum
+                  guid
+                  tokenMetadata {
+                    name
+                    __typename
+                  }
+                  __typename
+                }
+                __typename
+              }
+              llmReasoning {
+                assumptions
+                clarifications
+                interpretation
+                __typename
+              }
+              tokens
+              tmlTokens
+              worksheetId
+              tokens
+              description
+              title
+              tmlTokens
+              cached
+              sqlQuery
+              sessionId
+              genNo
+              formulaInfo {
+                name
+                expression
+                __typename
+              }
+              tmlPhrases
+              ambiguousPhrases {
+                alternativePhrases {
+                  phraseType
+                  token {
+                    token
+                    dataType
+                    typeEnum
+                    guid
+                    tokenMetadata {
+                      name
+                      __typename
+                    }
+                    __typename
+                  }
+                  __typename
+                }
+                ambiguityType
+                token {
+                  token
+                  dataType
+                  typeEnum
+                  guid
+                  tokenMetadata {
+                    name
+                    __typename
+                  }
+                  __typename
+                }
+                __typename
+              }
+              ambiguousTokens {
+                alternativeTokens {
+                  token
+                  dataType
+                  typeEnum
+                  guid
+                  tokenMetadata {
+                    name
+                    deprecatedTableGuid
+                    deprecatedTableName
+                    isFormula
+                    rootTables {
+                      created
+                      description
+                      guid
+                      indexVersion
+                      modified
+                      name
+                      __typename
+                    }
+                    schemaTableUserDefinedName
+                    table {
+                      created
+                      description
+                      guid
+                      indexVersion
+                      modified
+                      name
+                      __typename
+                    }
+                    __typename
+                  }
+                  __typename
+                }
+                ambiguityType
+                token {
+                  token
+                  dataType
+                  typeEnum
+                  guid
+                  tokenMetadata {
+                    name
+                    __typename
+                  }
+                  __typename
+                }
+                __typename
+              }
+              stateKey {
+                transactionId
+                generationNumber
+                transactionId
+                __typename
+              }
+              subQueries {
+                tokens
+                cohortConfig {
+                  anchorColumnId
+                  cohortAnswerGuid
+                  cohortGroupingType
+                  cohortGuid
+                  cohortType
+                  combineNonGroupValues
+                  description
+                  groupExcludedQueryValues
+                  hideExcludedQueryValues
+                  isEditable
+                  name
+                  nullOutputValue
+                  returnColumnId
+                  __typename
+                }
+                formulas {
+                  name
+                  expression
+                  __typename
+                }
+                __typename
+              }
+              visualizationSuggestion {
+                chartType
+                displayMode
+                axisConfigs {
+                  category
+                  color
+                  hidden
+                  size
+                  sort
+                  x
+                  y
+                  __typename
+                }
+                usersVizIntentApplied
+                customChartConfigs {
+                  dimensions {
+                    columns
+                    key
+                    __typename
+                  }
+                  key
+                  __typename
+                }
+                customChartGuid
+                __typename
+              }
+              tableData {
+                columnDataLite {
+                  columnId
+                  columnDataType
+                  dataValue
+                  columnName
+                  __typename
+                }
+                __typename
+              }
+              warningType
+              cached
+              __typename
+            }
+            debugInfo {
+              fewShotExamples {
+                chartType
+                id
+                mappingId
+                nlQuery
+                nlQueryConcepts
+                sageQuery
+                scope
+                sql
+                tml
+                __typename
+              }
+              __typename
+            }
+            __typename
+          }
+          __typename
+        }
+        errorData {
+          tool
+          errCode
+          errTxt
+          toolErrCode
+          __typename
+        }
+        __typename
+      }
+      type
+      __typename
+    }
+    prevCtx {
+      genNo
+      sessionId
+      __typename
+    }
+    __typename
+  }
+}
+`;

package/src/utils/graphql/nlsService/conversation-service.ts

@@ -0,0 +1,81 @@
+import { graphqlQuery } from '../graphql-request';
+import * as queries from './conversation-queries';
+
+export class Conversation {
+    private conversationId: string;
+
+    private inProgress: Promise<any> | null = null;
+
+    constructor(private thoughtSpotHost: string, private worksheetId: string) {
+        this.inProgress = this.init();
+    }
+
+    async init() {
+        const { convId } = await this.createConversation();
+        this.conversationId = convId;
+    }
+
+    createConversation(): Promise<any> {
+        return this.executeQuery(
+            queries.createConversation,
+            {
+                params: {
+                    initialCtx: {
+                        tsWorksheetCtx: {
+                            worksheet: {
+                                worksheetId: this.worksheetId,
+                            },
+                        },
+                        type: 'TS_WORKSHEET',
+                    },
+                    userInfo: {
+                        tenantUrl: `${this.thoughtSpotHost}/prism`,
+                    },
+                },
+            },
+        );
+    }
+
+    public async sendMessage(userMessage: string): Promise<any> {
+        await this.inProgress;
+
+        try {
+            const { responses } = await this.executeQuery(
+                queries.sendMessage,
+                {
+                    params: {
+                        convId: this.conversationId,
+                        msg: {
+                            data: {
+                                userCmdData: {
+                                    cmdText: userMessage,
+                                    nlsData: {
+                                        worksheetId: this.worksheetId,
+                                    },
+                                },
+                            },
+                            msgId: crypto.randomUUID(),
+                            type: 'USER_COMMAND',
+                        },
+                    },
+                },
+            );
+            const data = responses[0].data;
+            return {
+                data: data.asstRespData.nlsAnsData.sageQuerySuggestions[0],
+                error: null,
+            };
+        } catch (error) {
+            return { error };
+        }
+    }
+
+    public async executeQuery(query: string, variables: any) {
+        return graphqlQuery({
+            query,
+            variables,
+            thoughtSpotHost: this.thoughtSpotHost,
+            isCompositeQuery: false,
+        });
+    }
+}