@thoughtspot/visual-embed-sdk 1.33.0-alpha.1 → 1.33.0-dc.1

package/cjs/src/utils/graphql/conversationService/conversation-service.js

@@ -0,0 +1,89 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Conversation = void 0;
+const tslib_1 = require("tslib");
+const graphql_request_1 = require("../graphql-request");
+const queries = tslib_1.__importStar(require("./conversation-queries"));
+class Conversation {
+    constructor(thoughtSpotHost, worksheetId) {
+        this.thoughtSpotHost = thoughtSpotHost;
+        this.worksheetId = worksheetId;
+        this.inProgress = false;
+        this.init();
+    }
+    async init() {
+        try {
+            this.inProgress = true;
+            const { convId } = await this.createConversation();
+            this.conversationId = convId;
+        }
+        finally {
+            this.inProgress = false;
+        }
+    }
+    createConversation() {
+        return this.executeQuery(queries.createConversation, {
+            params: {
+                initialCtx: {
+                    tsWorksheetCtx: {
+                        worksheet: {
+                            worksheetId: this.worksheetId,
+                        },
+                    },
+                    type: 'TS_WORKSHEET',
+                },
+                userInfo: {
+                    tenantUrl: `${this.thoughtSpotHost}/prism`,
+                },
+            },
+        });
+    }
+    async sendMessage(userMessage) {
+        if (this.inProgress) {
+            console.warn('Previous message still in progress');
+            return { error: 'Previous message still in progress' };
+        }
+        try {
+            this.inProgress = true;
+            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',
+                    },
+                },
+            });
+            this.inProgress = false;
+            const data = responses[0].data;
+            return {
+                data: data.asstRespData.nlsAnsData.sageQuerySuggestions[0],
+                error: null,
+            };
+        }
+        catch (error) {
+            return { error };
+        }
+        finally {
+            this.inProgress = false;
+        }
+    }
+    async executeQuery(query, variables) {
+        return (0, graphql_request_1.graphqlQuery)({
+            query,
+            variables,
+            thoughtSpotHost: this.thoughtSpotHost,
+            isCompositeQuery: false,
+        });
+    }
+}
+exports.Conversation = Conversation;
+//# sourceMappingURL=conversation-service.js.map

package/cjs/src/utils/graphql/conversationService/conversation-service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"conversation-service.js","sourceRoot":"","sources":["../../../../../src/utils/graphql/conversationService/conversation-service.ts"],"names":[],"mappings":";;;;AAAA,wDAAkD;AAClD,wEAAkD;AAElD,MAAa,YAAY;IAKrB,YAAoB,eAAuB,EAAU,WAAmB;QAApD,oBAAe,GAAf,eAAe,CAAQ;QAAU,gBAAW,GAAX,WAAW,CAAQ;QAFhE,eAAU,GAAG,KAAK,CAAC;QAGvB,IAAI,CAAC,IAAI,EAAE,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,IAAI;QACN,IAAI;YACA,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;YACvB,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;YACnD,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC;SAChC;gBAAS;YACN,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;SAC3B;IACL,CAAC;IAED,kBAAkB;QACd,OAAO,IAAI,CAAC,YAAY,CACpB,OAAO,CAAC,kBAAkB,EAC1B;YACI,MAAM,EAAE;gBACJ,UAAU,EAAE;oBACR,cAAc,EAAE;wBACZ,SAAS,EAAE;4BACP,WAAW,EAAE,IAAI,CAAC,WAAW;yBAChC;qBACJ;oBACD,IAAI,EAAE,cAAc;iBACvB;gBACD,QAAQ,EAAE;oBACN,SAAS,EAAE,GAAG,IAAI,CAAC,eAAe,QAAQ;iBAC7C;aACJ;SACJ,CACJ,CAAC;IACN,CAAC;IAEM,KAAK,CAAC,WAAW,CAAC,WAAmB;QACxC,IAAI,IAAI,CAAC,UAAU,EAAE;YACjB,OAAO,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC;YACnD,OAAO,EAAE,KAAK,EAAE,oCAAoC,EAAE,CAAC;SAC1D;QAED,IAAI;YACA,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;YACvB,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,IAAI,CAAC,YAAY,CACzC,OAAO,CAAC,WAAW,EACnB;gBACI,MAAM,EAAE;oBACJ,MAAM,EAAE,IAAI,CAAC,cAAc;oBAC3B,GAAG,EAAE;wBACD,IAAI,EAAE;4BACF,WAAW,EAAE;gCACT,OAAO,EAAE,WAAW;gCACpB,OAAO,EAAE;oCACL,WAAW,EAAE,IAAI,CAAC,WAAW;iCAChC;6BACJ;yBACJ;wBACD,KAAK,EAAE,MAAM,CAAC,UAAU,EAAE;wBAC1B,IAAI,EAAE,cAAc;qBACvB;iBACJ;aACJ,CACJ,CAAC;YACF,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;YACxB,MAAM,IAAI,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC/B,OAAO;gBACH,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,oBAAoB,CAAC,CAAC,CAAC;gBAC1D,KAAK,EAAE,IAAI;aACd,CAAC;SACL;QAAC,OAAO,KAAK,EAAE;YACZ,OAAO,EAAE,KAAK,EAAE,CAAC;SACpB;gBAAS;YACN,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;SAC3B;IACL,CAAC;IAEM,KAAK,CAAC,YAAY,CAAC,KAAa,EAAE,SAAc;QACnD,OAAO,IAAA,8BAAY,EAAC;YAChB,KAAK;YACL,SAAS;YACT,eAAe,EAAE,IAAI,CAAC,eAAe;YACrC,gBAAgB,EAAE,KAAK;SAC1B,CAAC,CAAC;IACP,CAAC;CACJ;AAzFD,oCAyFC"}

package/cjs/src/utils/graphql/graphql-request.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"graphql-request.d.ts","sourceRoot":"","sources":["../../../../src/utils/graphql/graphql-request.ts"],"names":[],"mappings":"AAGA;;;;;;;GAOG;AACH,wBAAsB,YAAY,CAAC,EAC/B,KAAK,EACL,SAAS,EACT,eAAe,EACf,gBAAwB,GAC3B,EAAE;IACC,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,GAAG,CAAC;IACf,eAAe,EAAE,MAAM,CAAC;IACxB,gBAAgB,CAAC,EAAE,OAAO,CAAA;CAC7B,gBAwBA"}
+{"version":3,"file":"graphql-request.d.ts","sourceRoot":"","sources":["../../../../src/utils/graphql/graphql-request.ts"],"names":[],"mappings":"AAGA;;;;;;;GAOG;AACH,wBAAsB,YAAY,CAAC,EAC/B,KAAK,EACL,SAAS,EACT,eAAe,EACf,gBAAwB,GAC3B,EAAE;IACC,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,GAAG,CAAC;IACf,eAAe,EAAE,MAAM,CAAC;IACxB,gBAAgB,CAAC,EAAE,OAAO,CAAA;CAC7B,GAAG,OAAO,CAAC,GAAG,CAAC,CAwBf"}

package/cjs/src/utils/graphql/graphql-request.spec.d.ts

@@ -0,0 +1,2 @@
+import 'jest-fetch-mock';
+//# sourceMappingURL=graphql-request.spec.d.ts.map

package/cjs/src/utils/graphql/graphql-request.spec.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"graphql-request.spec.d.ts","sourceRoot":"","sources":["../../../../src/utils/graphql/graphql-request.spec.ts"],"names":[],"mappings":"AAAA,OAAO,iBAAiB,CAAC"}

package/cjs/src/utils/graphql/graphql-request.spec.js

@@ -0,0 +1,48 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const tslib_1 = require("tslib");
+require("jest-fetch-mock");
+const tokenizedFetchUtil = tslib_1.__importStar(require("../../tokenizedFetch"));
+const graphql_request_1 = require("./graphql-request");
+const getSourceDetailQuery = `
+    query GetSourceDetail($ids: [GUID!]!) {
+        getSourceDetailById(ids: $ids, type: LOGICAL_TABLE) {
+            id
+            name
+        }
+    }  
+`;
+const thoughtSpotHost = 'TSHOST';
+describe('graphQl tests', () => {
+    test('should call tokenizedFetch with correct parameters when graphqlQuery is called', async () => {
+        jest.spyOn(tokenizedFetchUtil, 'tokenizedFetch').mockResolvedValue({
+            json: jest.fn().mockResolvedValue({
+                data: {},
+            }),
+        });
+        const details = await (0, graphql_request_1.graphqlQuery)({
+            query: getSourceDetailQuery,
+            variables: {
+                ids: [2],
+            },
+            thoughtSpotHost,
+        });
+        expect(tokenizedFetchUtil.tokenizedFetch).toBeCalledWith('TSHOST/prism/?op=GetSourceDetail', {
+            body: '{"operationName":"GetSourceDetail","query":"\\n    query GetSourceDetail($ids: [GUID!]!) {\\n        getSourceDetailById(ids: $ids, type: LOGICAL_TABLE) {\\n            id\\n            name\\n        }\\n    }  \\n","variables":{"ids":[2]}}',
+            credentials: 'include',
+            headers: {
+                accept: '*/*', 'accept-language': 'en-us', 'content-type': 'application/json;charset=UTF-8', 'x-requested-by': 'ThoughtSpot',
+            },
+            method: 'POST',
+        });
+        const details2 = await (0, graphql_request_1.graphqlQuery)({
+            query: getSourceDetailQuery,
+            variables: {
+                ids: [2],
+            },
+            thoughtSpotHost,
+            isCompositeQuery: true,
+        });
+    });
+});
+//# sourceMappingURL=graphql-request.spec.js.map

package/cjs/src/utils/graphql/graphql-request.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"graphql-request.spec.js","sourceRoot":"","sources":["../../../../src/utils/graphql/graphql-request.spec.ts"],"names":[],"mappings":";;;AAAA,2BAAyB;AAEzB,iFAA2D;AAC3D,uDAAiD;AAEjD,MAAM,oBAAoB,GAAG;;;;;;;CAO5B,CAAC;AAEF,MAAM,eAAe,GAAG,QAAQ,CAAC;AAEjC,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC3B,IAAI,CAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;QAC9F,IAAI,CAAC,KAAK,CAAC,kBAAkB,EAAE,gBAAgB,CAAC,CAAC,iBAAiB,CAAC;YAC/D,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBAC9B,IAAI,EAAE,EAAE;aACX,CAAC;SACL,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,MAAM,IAAA,8BAAY,EAAC;YAC/B,KAAK,EAAE,oBAAoB;YAC3B,SAAS,EAAE;gBACP,GAAG,EAAE,CAAC,CAAC,CAAC;aACX;YACD,eAAe;SAClB,CAAC,CAAC;QAEH,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC,cAAc,CAAC,kCAAkC,EAAE;YACzF,IAAI,EAAE,mPAAmP;YACzP,WAAW,EAAE,SAAS;YACtB,OAAO,EAAE;gBACL,MAAM,EAAE,KAAK,EAAE,iBAAiB,EAAE,OAAO,EAAE,cAAc,EAAE,gCAAgC,EAAE,gBAAgB,EAAE,aAAa;aAC/H;YACD,MAAM,EAAE,MAAM;SACjB,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,IAAA,8BAAY,EAAC;YAChC,KAAK,EAAE,oBAAoB;YAC3B,SAAS,EAAE;gBACP,GAAG,EAAE,CAAC,CAAC,CAAC;aACX;YACD,eAAe;YACf,gBAAgB,EAAE,IAAI;SACzB,CAAC,CAAC;IACP,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC"}

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

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

package/cjs/src/utils/graphql/nlsService/conversation-queries.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"conversation-queries.d.ts","sourceRoot":"","sources":["../../../../../src/utils/graphql/nlsService/conversation-queries.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,kBAAkB,2aAqB9B,CAAC;AAEF,eAAO,MAAM,WAAW,i/OA8RvB,CAAC"}

package/cjs/src/utils/graphql/nlsService/conversation-queries.js

@@ -0,0 +1,313 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.sendMessage = exports.createConversation = void 0;
+exports.createConversation = `
+mutation CreateConversation($params: Input_convassist_CreateConversationRequest) {
+  ConvAssist__createConversation(request: $params) {
+    convId
+    initialCtx {
+      type
+      tsAnsCtx {
+        sessionId
+        genNo
+        stateKey {
+          transactionId
+          generationNumber
+        }
+        worksheet {
+          worksheetId
+          worksheetName
+        }
+      }
+    }
+  }
+}
+`;
+exports.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
+  }
+}
+`;
+//# sourceMappingURL=conversation-queries.js.map

package/cjs/src/utils/graphql/nlsService/conversation-queries.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"conversation-queries.js","sourceRoot":"","sources":["../../../../../src/utils/graphql/nlsService/conversation-queries.ts"],"names":[],"mappings":";;;AAAa,QAAA,kBAAkB,GAAG;;;;;;;;;;;;;;;;;;;;;CAqBjC,CAAC;AAEW,QAAA,WAAW,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA8R1B,CAAC"}

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

@@ -0,0 +1,12 @@
+export declare class Conversation {
+    private thoughtSpotHost;
+    private worksheetId;
+    private conversationId;
+    private inProgress;
+    constructor(thoughtSpotHost: string, worksheetId: string);
+    init(): Promise<void>;
+    createConversation(): Promise<any>;
+    sendMessage(userMessage: string): Promise<any>;
+    executeQuery(query: string, variables: any): Promise<any>;
+}
+//# sourceMappingURL=conversation-service.d.ts.map

package/cjs/src/utils/graphql/nlsService/conversation-service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"conversation-service.d.ts","sourceRoot":"","sources":["../../../../../src/utils/graphql/nlsService/conversation-service.ts"],"names":[],"mappings":"AAGA,qBAAa,YAAY;IAKT,OAAO,CAAC,eAAe;IAAU,OAAO,CAAC,WAAW;IAJhE,OAAO,CAAC,cAAc,CAAS;IAE/B,OAAO,CAAC,UAAU,CAA6B;gBAE3B,eAAe,EAAE,MAAM,EAAU,WAAW,EAAE,MAAM;IAIlE,IAAI;IAKV,kBAAkB,IAAI,OAAO,CAAC,GAAG,CAAC;IAqBrB,WAAW,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC;IAkC9C,YAAY,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG;CAQ1D"}

package/cjs/src/utils/graphql/nlsService/conversation-service.js

@@ -0,0 +1,75 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Conversation = void 0;
+const tslib_1 = require("tslib");
+const graphql_request_1 = require("../graphql-request");
+const queries = tslib_1.__importStar(require("./conversation-queries"));
+class Conversation {
+    constructor(thoughtSpotHost, worksheetId) {
+        this.thoughtSpotHost = thoughtSpotHost;
+        this.worksheetId = worksheetId;
+        this.inProgress = null;
+        this.inProgress = this.init();
+    }
+    async init() {
+        const { convId } = await this.createConversation();
+        this.conversationId = convId;
+    }
+    createConversation() {
+        return this.executeQuery(queries.createConversation, {
+            params: {
+                initialCtx: {
+                    tsWorksheetCtx: {
+                        worksheet: {
+                            worksheetId: this.worksheetId,
+                        },
+                    },
+                    type: 'TS_WORKSHEET',
+                },
+                userInfo: {
+                    tenantUrl: `${this.thoughtSpotHost}/prism`,
+                },
+            },
+        });
+    }
+    async sendMessage(userMessage) {
+        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 };
+        }
+    }
+    async executeQuery(query, variables) {
+        return (0, graphql_request_1.graphqlQuery)({
+            query,
+            variables,
+            thoughtSpotHost: this.thoughtSpotHost,
+            isCompositeQuery: false,
+        });
+    }
+}
+exports.Conversation = Conversation;
+//# sourceMappingURL=conversation-service.js.map

package/cjs/src/utils/graphql/nlsService/conversation-service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"conversation-service.js","sourceRoot":"","sources":["../../../../../src/utils/graphql/nlsService/conversation-service.ts"],"names":[],"mappings":";;;;AAAA,wDAAkD;AAClD,wEAAkD;AAElD,MAAa,YAAY;IAKrB,YAAoB,eAAuB,EAAU,WAAmB;QAApD,oBAAe,GAAf,eAAe,CAAQ;QAAU,gBAAW,GAAX,WAAW,CAAQ;QAFhE,eAAU,GAAwB,IAAI,CAAC;QAG3C,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,IAAI;QACN,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACnD,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC;IACjC,CAAC;IAED,kBAAkB;QACd,OAAO,IAAI,CAAC,YAAY,CACpB,OAAO,CAAC,kBAAkB,EAC1B;YACI,MAAM,EAAE;gBACJ,UAAU,EAAE;oBACR,cAAc,EAAE;wBACZ,SAAS,EAAE;4BACP,WAAW,EAAE,IAAI,CAAC,WAAW;yBAChC;qBACJ;oBACD,IAAI,EAAE,cAAc;iBACvB;gBACD,QAAQ,EAAE;oBACN,SAAS,EAAE,GAAG,IAAI,CAAC,eAAe,QAAQ;iBAC7C;aACJ;SACJ,CACJ,CAAC;IACN,CAAC;IAEM,KAAK,CAAC,WAAW,CAAC,WAAmB;QACxC,MAAM,IAAI,CAAC,UAAU,CAAC;QAEtB,IAAI;YACA,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,IAAI,CAAC,YAAY,CACzC,OAAO,CAAC,WAAW,EACnB;gBACI,MAAM,EAAE;oBACJ,MAAM,EAAE,IAAI,CAAC,cAAc;oBAC3B,GAAG,EAAE;wBACD,IAAI,EAAE;4BACF,WAAW,EAAE;gCACT,OAAO,EAAE,WAAW;gCACpB,OAAO,EAAE;oCACL,WAAW,EAAE,IAAI,CAAC,WAAW;iCAChC;6BACJ;yBACJ;wBACD,KAAK,EAAE,MAAM,CAAC,UAAU,EAAE;wBAC1B,IAAI,EAAE,cAAc;qBACvB;iBACJ;aACJ,CACJ,CAAC;YACF,MAAM,IAAI,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC/B,OAAO;gBACH,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,oBAAoB,CAAC,CAAC,CAAC;gBAC1D,KAAK,EAAE,IAAI;aACd,CAAC;SACL;QAAC,OAAO,KAAK,EAAE;YACZ,OAAO,EAAE,KAAK,EAAE,CAAC;SACpB;IACL,CAAC;IAEM,KAAK,CAAC,YAAY,CAAC,KAAa,EAAE,SAAc;QACnD,OAAO,IAAA,8BAAY,EAAC;YAChB,KAAK;YACL,SAAS;YACT,eAAe,EAAE,IAAI,CAAC,eAAe;YACrC,gBAAgB,EAAE,KAAK;SAC1B,CAAC,CAAC;IACP,CAAC;CACJ;AA7ED,oCA6EC"}

package/cjs/src/utils/graphql/nlsService/nls-answer-queries.d.ts

@@ -0,0 +1,2 @@
+export declare const getAnswerSessionFromQuery = "\nquery GetEurekaResults($params: Input_eureka_SearchRequest) {\n        queryRequest(request: $params) {\n            ...eurekaResults\n            __typename\n        }\n    }\n\n        fragment eurekaResults on eureka_SearchResponse {\n        facets {\n            facetType\n            facetValue\n            facetValues {\n            id\n            resultCount\n            name\n            __typename\n            }\n            __typename\n        }\n        requestIdentifiers {\n            apiRequestId\n            appActivityId\n            __typename\n        }\n        sageQuerySuggestions {\n            llmReasoning {\n                assumptions\n                clarifications\n                interpretation\n                __typename\n            }\n            tokens\n            tmlTokens\n            worksheetId\n            description\n            title\n            tmlTokens\n            formulaInfo {\n            name\n            expression\n            __typename\n            }\n            tmlPhrases\n            ambiguousPhrases {\n            alternativePhrases {\n                phraseType\n                token {\n                token\n                dataType\n                typeEnum\n                guid\n                tokenMetadata {\n                    name\n                    __typename\n                }\n                __typename\n                }\n                __typename\n            }\n            ambiguityType\n            token {\n                token\n                dataType\n                typeEnum\n                guid\n                tokenMetadata {\n                name\n                __typename\n                }\n                __typename\n            }\n            __typename\n            }\n            ambiguousTokens {\n            alternativeTokens {\n                token\n                dataType\n                typeEnum\n                guid\n                tokenMetadata {\n                name\n                deprecatedTableGuid\n                deprecatedTableName\n                isFormula\n                rootTables {\n                    created\n                    description\n                    guid\n                    indexVersion\n                    modified\n                    name\n                    __typename\n                }\n                schemaTableUserDefinedName\n                table {\n                    created\n                    description\n                    guid\n                    indexVersion\n                    modified\n                    name\n                    __typename\n                }\n                __typename\n                }\n                __typename\n            }\n            ambiguityType\n            token {\n                token\n                dataType\n                typeEnum\n                guid\n                tokenMetadata {\n                name\n                __typename\n                }\n                __typename\n            }\n            __typename\n            }\n            sessionId\n            genNo\n            stateKey {\n                generationNumber\n                transactionId\n            __typename\n            }\n            subQueries {\n            tokens\n            cohortConfig {\n                anchorColumnId\n                cohortAnswerGuid\n                cohortGroupingType\n                cohortGuid\n                cohortType\n                combineNonGroupValues\n                description\n                groupExcludedQueryValues\n                hideExcludedQueryValues\n                isEditable\n                name\n                nullOutputValue\n                returnColumnId\n                __typename\n            }\n            formulas {\n                name\n                expression\n                __typename\n            }\n            __typename\n            }\n            visualizationSuggestion {\n            chartType\n            displayMode\n            axisConfigs {\n                category\n                color\n                hidden\n                size\n                sort\n                x\n                y\n                __typename\n            }\n            usersVizIntentApplied\n            customChartConfigs {\n                dimensions {\n                columns\n                key\n                __typename\n                }\n                key\n                __typename\n            }\n            customChartGuid\n            __typename\n            }\n            tableData {\n            columnDataLite {\n                columnId\n                columnDataType\n                dataValue\n                columnName\n                __typename\n            }\n            __typename\n            }\n            warningType\n            cached\n            warningDetails {\n            warningType\n            __typename\n            }\n            __typename\n        }\n        results {\n            objectSecurityInfo {\n            objectType\n            objectId\n            objectIdForDeletionCheck\n            objectTypeForDeletionCheck\n            isD13ySourced\n            offset\n            __typename\n            }\n            searchAnswer {\n            ...eurekaAnswer\n            __typename\n            }\n            searchPinboardViz {\n            answer {\n                ...eurekaAnswer\n                __typename\n            }\n            pinboardHeader {\n                id\n                title\n                __typename\n            }\n            __typename\n            }\n            searchPinboard {\n            header {\n                ...header\n                __typename\n            }\n            usageInfo {\n                ...usageInfo\n                __typename\n            }\n            answers {\n                ...eurekaAnswer\n                __typename\n            }\n            vizCount {\n                charts\n                metrics\n                tables\n                __typename\n            }\n            __typename\n            }\n            snippetInfo {\n            titleSnippet {\n                snippetString\n                highlights {\n                start\n                end\n                __typename\n                }\n                __typename\n            }\n            descriptionSnippet {\n                snippetString\n                highlights {\n                start\n                end\n                __typename\n                }\n                __typename\n            }\n            sageQuerySnippet {\n                phrase {\n                isCompletePhrase\n                numTokens\n                phraseType\n                startIndex\n                __typename\n                }\n                token {\n                token\n                dataType\n                typeEnum\n                __typename\n                }\n                __typename\n            }\n            sageQuerySnippetWithHighlights {\n                highlights {\n                start\n                end\n                __typename\n                }\n                phraseType\n                phraseValue\n                __typename\n            }\n            __typename\n            }\n            score\n            debugInfo\n            resultType\n            sageQuery\n            __typename\n        }\n        version\n        nextPageOffset\n        batchSizeRequired\n        isFinalPage\n        totalResults\n        totalFacetResultCount\n        errorCode\n        debugInfo {\n            fewShotExamples {\n            chartType\n            formulas {\n                name\n                expression\n                __typename\n            }\n            id\n            mappingId\n            nlQuery\n            nlQueryConcepts\n            sageQuery\n            scope\n            sql\n            tml\n            feedbackType\n            __typename\n            }\n            __typename\n        }\n        __typename\n        }\n\n        fragment eurekaAnswer on eureka_AnswerResult {\n        header {\n            ...header\n            __typename\n        }\n        usageInfo {\n            ...usageInfo\n            __typename\n        }\n        preferredViz {\n            ...visualizationMetadata\n            __typename\n        }\n        worksheetInfo {\n            ...worksheetInfo\n            __typename\n        }\n        formatted {\n            phrase {\n            isCompletePhrase\n            numTokens\n            phraseType\n            startIndex\n            __typename\n            }\n            token {\n            token\n            typeEnum\n            __typename\n            }\n            __typename\n        }\n        __typename\n        }\n\n        fragment header on eureka_Header {\n        id\n        title\n        description\n        authorGuid\n        authorName\n        createdOn\n        tagIds\n        __typename\n        }\n\n        fragment usageInfo on eureka_UsageInfo {\n        favouriteCount\n        viewCount\n        __typename\n        }\n\n        fragment visualizationMetadata on eureka_VisualizationMetadata {\n        vizType\n        chartType\n        vizSnapshotRequestData {\n            parentReportbookGuid\n            parentType\n            version\n            vizGuid\n            __typename\n        }\n        __typename\n        }\n\n        fragment worksheetInfo on eureka_WorksheetInfo {\n        id\n        name\n        __typename\n        }";
+//# sourceMappingURL=nls-answer-queries.d.ts.map

package/cjs/src/utils/graphql/nlsService/nls-answer-queries.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"nls-answer-queries.d.ts","sourceRoot":"","sources":["../../../../../src/utils/graphql/nlsService/nls-answer-queries.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,yBAAyB,m6SA8Y5B,CAAC"}