@kusto/monaco-kusto 14.0.1 → 14.2.2

package/README.md

@@ -56,6 +56,24 @@ There are 2 APIs to set a Kusto schema:
    interface `Result` in `schema.ts`), so when this method is used, it also
    requires a cluster URI and the name of the database in context.
 
+### Schema Parsing Flow
+
+The schema parsing process in Monaco-Kusto involves several components, starting from the Kusto command and ending with language features in the editor. Here is how the flow works:
+
+Below is a diagram illustrating the schema parsing flow:
+
+![img_1.png](img_1.png)
+
+## Getting Started: Running the Local Demo App
+
+To quickly try Monaco-Kusto locally, run the following command from the `package` folder:
+
+```
+yarn start
+```
+
+This will start a local development server and open the demo app at http://localhost:7777.
+
 ## Contributing
 
 Every PR should come with a test that checks it.
@@ -89,8 +107,83 @@ Every PR should come with a test that checks it.
 > Before running `yarn test:it` or `yarn test:it:watch`, first run `yarn test:it:serve`.
 > These scripts (`test:it` and `test:it:watch`) do **not** automatically rebuild the project, so running the server ensures your latest code is tested.
 
+## Running Monaco-Kusto Locally in Azure-Kusto-WebUX
+
+To run monaco-kusto locally inside the Azure-Kusto-WebUX project (which should exist outside this project folder with the original project name `Azure-Kusto-WebUX`), you can use the provided script:
+
+```sh
+./debug-monaco-kusto.sh
+```
+
+### Script Modes: KustoWeb and Fabric
+
+There are two ways to run the script:
+
+-   **KustoWeb (default):** The script is currently set up to run the `kustoweb` app inside Azure-Kusto-WebUX. This is the default behavior.
+-   **Fabric:** To run the Fabric app instead, simply comment out the Step 4 section for KustoWeb and uncomment the Step 4 section for Fabric in `debug-monaco-kusto.sh`.
+
+This allows you to easily switch between running KustoWeb and Fabric for local development.
+
+### Project Structure Requirements
+
+For the debug-monaco-kusto.sh script to work, ensure that your project structure resembles the following, with both `monaco-kusto` and `Azure-Kusto-WebUX` as sibling folders:
+
+```
+/your-workspace-root
+|-- monaco-kusto
+|   |-- ...
+|   |-- debug-monaco-kusto.sh
+|   |-- ...
+|
+|-- Azure-Kusto-WebUX
+    `-- ...
+```
+
+The important part is that the `Azure-Kusto-WebUX` folder is a sibling to `monaco-kusto`, and that its `node_modules` folder and `package.json` file are at its root. The script relies on this structure to copy files and run the correct build steps.
+
+## Architecture Overview
+
+This section provides a high-level overview of the main files and their responsibilities in the project.
+
+![img.png](img.png)
+
+-   **`monaco.contribution`**  
+    Declares and exports the Kusto language as a Monaco Editor contribution, making it available for registration and use externally.
+
+-   **`kustoMode`** Sets up and registers the Kusto language in Monaco Editor, wiring together language features, workers, and configuration.
+
+-   **`workerManager`** Manages the lifecycle and communication with web workers that run language services in the background.
+
+-   **`kustoWorker`** Implements the actual worker logic, handling requests for language features from the main thread.
+
+-   **`kustoLanguageService`**  
+    Implements the core logic for Kusto language features such as parsing, validation, and providing language intelligence (completion, diagnostics, etc.).  
+    Uses the `language-service-next` library, which was originally created in C# and migrated to TypeScript using bridgejs.
+
+-   **`languageFeatures`**  
+    Contains adapters and implementations for Monaco Editor language features (completion, hover, formatting, folding, etc.) specific to Kusto.
+
+-   **`monacoInstance`**  
+    Represents the Monaco Editor instance itself. It is responsible for editor creation, configuration, and interaction with the registered Kusto language features.
+
 ## Changelog
 
+### 14.2.2
+
+-   docs: add external references to semver-classification skill
+
+### 14.2.1
+
+-   chore: bump CI Node.js version to 22
+
+### 14.2.0
+
+-   feat: add `onSchemaUpdateComplete` event emitter so consumers can react to schema refresh completion
+
+### 14.1.0
+
+-   feat: support graph schema
+
 ### 14.0.1
 
 -   chore: Update @kusto/language-service-next upgrade to 12.2.0
@@ -1014,17 +1107,3 @@ Every PR should come with a test that checks it.
 #### Bug fixes
 
 -   setSchema does not update syntax highlighting
-
-# Contributing
-
-This project welcomes contributions and suggestions. Most contributions require you to agree to a
-Contributor License Agreement (CLA) declaring that you have the right to, and actually do, grant us
-the rights to use your contribution. For details, visit https://cla.microsoft.com.
-
-When you submit a pull request, a CLA-bot will automatically determine whether you need to provide
-a CLA and decorate the PR appropriately (e.g., label, comment). Simply follow the instructions
-provided by the bot. You will only need to do this once across all repos using our CLA.
-
-This project has adopted the [Microsoft Open Source Code of Conduct](https://opensource.microsoft.com/codeofconduct/).
-For more information see the [Code of Conduct FAQ](https://opensource.microsoft.com/codeofconduct/faq/) or
-contact [opencode@microsoft.com](mailto:opencode@microsoft.com) with any additional questions or comments.

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@kusto/monaco-kusto",
-    "version": "14.0.1",
+    "version": "14.2.2",
     "description": "CSL, KQL plugin for the Monaco Editor",
     "author": {
         "name": "Microsoft"

package/release/dev/kustoMode.js

@@ -1,11 +1,11 @@
 /*!-----------------------------------------------------------------------------
  * Copyright (c) Microsoft Corporation. All rights reserved.
- * monaco-kusto version: 14.0.1(d58ade77c070c9eb832c5ae067820b5be7eb3003)
+ * monaco-kusto version: 14.2.2(22fb70324f78ec316d3733a3c97919f80215e03d)
  * Released under the MIT license
  * https://https://github.com/Azure/monaco-kusto/blob/master/README.md
  *-----------------------------------------------------------------------------*/
 
-define('vs/language/kusto/kustoMode', ['exports', 'vs/editor/editor.main', './main-6dbedcd3', './types-dde70929'], (function (exports, monaco, main, types$1) { 'use strict';
+define('vs/language/kusto/kustoMode', ['exports', 'vs/editor/editor.main', './main-2d6da563', './types-fb9b493a'], (function (exports, monaco, main, types$1) { 'use strict';
 
     function _interopNamespaceDefault(e) {
         var n = Object.create(null);
@@ -1556,9 +1556,10 @@ define('vs/language/kusto/kustoMode', ['exports', 'vs/editor/editor.main', './ma
     function _toPropertyKey$1(t) { var i = _toPrimitive$1(t, "string"); return "symbol" == _typeof$2(i) ? i : i + ""; }
     function _toPrimitive$1(t, r) { if ("object" != _typeof$2(t) || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != _typeof$2(i)) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); }
     var SemanticTokensProvider = /*#__PURE__*/function () {
-      function SemanticTokensProvider(classificationsGetter) {
+      function SemanticTokensProvider(classificationsGetter, onTokensProvided) {
         _classCallCheck(this, SemanticTokensProvider);
         this.classificationsGetter = classificationsGetter;
+        this.onTokensProvided = onTokensProvided;
       }
       return _createClass(SemanticTokensProvider, [{
         key: "getLegend",
@@ -1572,7 +1573,8 @@ define('vs/language/kusto/kustoMode', ['exports', 'vs/editor/editor.main', './ma
         key: "provideDocumentSemanticTokens",
         value: function () {
           var _provideDocumentSemanticTokens = _asyncToGenerator$2( /*#__PURE__*/_regeneratorRuntime$2().mark(function _callee(model) {
-            var resource, classifications, tokens, prevLine, prevChar, _iterator, _step, classification, parts, _i, _parts, part, _part, absLine, absChar, length, kind, modifiers, deltaLine, deltaChar;
+            var _this$onTokensProvide;
+            var resource, classifications, tokens, prevLine, prevChar, _iterator, _step, classification, parts, _i, _parts, part, _part, absLine, absChar, length, kind, modifiers, deltaLine, deltaChar, result;
             return _regeneratorRuntime$2().wrap(function _callee$(_context) {
               while (1) switch (_context.prev = _context.next) {
                 case 0:
@@ -1604,11 +1606,13 @@ define('vs/language/kusto/kustoMode', ['exports', 'vs/editor/editor.main', './ma
                   } finally {
                     _iterator.f();
                   }
-                  return _context.abrupt("return", {
+                  result = {
                     data: new Uint32Array(tokens.flat(2)),
                     resultId: model.getVersionId().toString()
-                  });
-                case 10:
+                  };
+                  (_this$onTokensProvide = this.onTokensProvided) === null || _this$onTokensProvide === void 0 || _this$onTokensProvide.call(this, resource);
+                  return _context.abrupt("return", result);
+                case 12:
                 case "end":
                   return _context.stop();
               }
@@ -1651,10 +1655,10 @@ define('vs/language/kusto/kustoMode', ['exports', 'vs/editor/editor.main', './ma
     function _asyncToGenerator$1(n) { return function () { var t = this, e = arguments; return new Promise(function (r, o) { var a = n.apply(t, e); function _next(n) { asyncGeneratorStep$1(a, r, o, _next, _throw, "next", n); } function _throw(n) { asyncGeneratorStep$1(a, r, o, _next, _throw, "throw", n); } _next(void 0); }); }; }
     // Registers semantic token provider that utilizes the language service
     // for more context-relevant syntax highlighting.
-    function semanticTokensProviderRegistrarCreator() {
+    function semanticTokensProviderRegistrarCreator(onTokensProvided) {
       var semanticTokensProviderRegistrar = semanticTokensProviderRegistrarCreatorForTest();
       return function (monacoInstance, workerAccessor) {
-        var semanticTokensProvider = semanticTokensProviderMaker(workerAccessor);
+        var semanticTokensProvider = semanticTokensProviderMaker(workerAccessor, onTokensProvided);
         semanticTokensProviderRegistrar(monacoInstance, semanticTokensProvider);
       };
     }
@@ -1667,7 +1671,7 @@ define('vs/language/kusto/kustoMode', ['exports', 'vs/editor/editor.main', './ma
         semanticTokensDisposable = monacoInstance.languages.registerDocumentSemanticTokensProvider(types$1.LANGUAGE_ID, semanticTokensProvider);
       };
     }
-    function semanticTokensProviderMaker(workerAccessor) {
+    function semanticTokensProviderMaker(workerAccessor, onTokensProvided) {
       var classificationsGetter = /*#__PURE__*/function () {
         var _ref = _asyncToGenerator$1( /*#__PURE__*/_regeneratorRuntime$1().mark(function _callee(resource) {
           var worker;
@@ -1689,7 +1693,7 @@ define('vs/language/kusto/kustoMode', ['exports', 'vs/editor/editor.main', './ma
           return _ref.apply(this, arguments);
         };
       }();
-      return new SemanticTokensProvider(classificationsGetter);
+      return new SemanticTokensProvider(classificationsGetter, onTokensProvided);
     }
 
     function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); }
@@ -1717,10 +1721,16 @@ define('vs/language/kusto/kustoMode', ['exports', 'vs/editor/editor.main', './ma
     /**
      * Called when Kusto language is first needed (a model has the language set)
      * @param defaults
+     * @param monacoInstance
+     * @param onSchemaUpdateCompleteEmitter Optional,fired when the language service has finished all schema-related work
      */
-    function setupMode(defaults, monacoInstance) {
+    function setupMode(defaults, monacoInstance, onSchemaUpdateCompleteEmitter) {
       var onSchemaChange = new monaco__namespace.Emitter();
-      var semanticTokensProviderRegistrar = semanticTokensProviderRegistrarCreator();
+      var semanticTokensProviderRegistrar = semanticTokensProviderRegistrarCreator(function (uri) {
+        onSchemaUpdateCompleteEmitter === null || onSchemaUpdateCompleteEmitter === void 0 || onSchemaUpdateCompleteEmitter.fire({
+          uri: uri
+        });
+      });
       var client = new WorkerManager(monacoInstance, defaults);
       var workerAccessor = function workerAccessor(first) {
         var augmentedSetSchema = /*#__PURE__*/function () {

package/release/dev/kustoWorker.js

@@ -1,11 +1,11 @@
 /*!-----------------------------------------------------------------------------
  * Copyright (c) Microsoft Corporation. All rights reserved.
- * monaco-kusto version: 14.0.1(d58ade77c070c9eb832c5ae067820b5be7eb3003)
+ * monaco-kusto version: 14.2.2(22fb70324f78ec316d3733a3c97919f80215e03d)
  * Released under the MIT license
  * https://https://github.com/Azure/monaco-kusto/blob/master/README.md
  *-----------------------------------------------------------------------------*/
 
-define('vs/language/kusto/kustoWorker', ['exports', './main-6dbedcd3', './schema-954f3274'], (function (exports, main, schema) { 'use strict';
+define('vs/language/kusto/kustoWorker', ['exports', './main-2d6da563', './schema-c2628e1d'], (function (exports, main, schema) { 'use strict';
 
     function _slicedToArray$2(r, e) { return _arrayWithHoles$2(r) || _iterableToArrayLimit$2(r, e) || _unsupportedIterableToArray$3(r, e) || _nonIterableRest$2(); }
     function _nonIterableRest$2() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
@@ -4229,7 +4229,8 @@ define('vs/language/kusto/kustoWorker', ['exports', './main-6dbedcd3', './schema
               _ref6$EntityGroups = _ref6.EntityGroups,
               EntityGroups = _ref6$EntityGroups === void 0 ? {} : _ref6$EntityGroups,
               MinorVersion = _ref6.MinorVersion,
-              MajorVersion = _ref6.MajorVersion;
+              MajorVersion = _ref6.MajorVersion,
+              Graphs = _ref6.Graphs;
             return {
               name: Name,
               alternateName: databaseInContextAlternateName,
@@ -4303,7 +4304,17 @@ define('vs/language/kusto/kustoWorker', ['exports', './main-6dbedcd3', './schema
                     };
                   })
                 };
-              })
+              }),
+              graphs: Object.values(Graphs).reduce(function (graphArray, graph) {
+                var graphEntity = {
+                  name: graph.Name,
+                  entityType: 'Graph',
+                  edges: graph.Edges,
+                  nodes: graph.Nodes,
+                  snapshots: graph.Snapshots
+                };
+                return [].concat(_toConsumableArray(graphArray), [graphEntity]);
+              }, []) // this is a temporary workaround as graphs are not included in the .show schema as json command output
             };
           });
           return {
@@ -5323,7 +5334,8 @@ define('vs/language/kusto/kustoWorker', ['exports', './main-6dbedcd3', './schema
           var tableSymbols = (db.tables || []).map(this.createTableSymbol);
           var functionSymbols = (db.functions || []).map(this.createFunctionSymbol);
           var entityGroupsSymbols = (db.entityGroups || []).map(this.createEntityGroupSymbol);
-          var databaseSymbol = new sym.DatabaseSymbol.$ctor2(db.name, db.alternateName || null, tableSymbols.concat(functionSymbols).concat(entityGroupsSymbols));
+          var graphModelSymbols = (db.graphs || []).map(this.createGraphModelSymbol);
+          var databaseSymbol = new sym.DatabaseSymbol.$ctor2(db.name, db.alternateName || null, tableSymbols.concat(functionSymbols).concat(entityGroupsSymbols).concat(graphModelSymbols));
           return databaseSymbol;
         }
       }, {
@@ -5381,6 +5393,14 @@ define('vs/language/kusto/kustoWorker', ['exports', './main-6dbedcd3', './schema
         value: function createEntityGroupSymbol(entityGroup) {
           return new sym.EntityGroupSymbol.$ctor3(entityGroup.name, entityGroup.members.join(), null);
         }
+      }, {
+        key: "createGraphModelSymbol",
+        value: function createGraphModelSymbol(graph) {
+          var edges = new Bridge.ArrayEnumerable(graph.edges || []);
+          var nodes = new Bridge.ArrayEnumerable(graph.nodes || []);
+          var snapshots = new Bridge.ArrayEnumerable(graph.snapshots || []);
+          return new sym.GraphModelSymbol.$ctor1(graph.name, edges, nodes, snapshots);
+        }
       }]);
     }();
     var languageService = new KustoLanguageService(KustoLanguageService.dummySchema, {

package/release/dev/{main-6dbedcd3.js → main-2d6da563.js}

@@ -1,11 +1,11 @@
 /*!-----------------------------------------------------------------------------
  * Copyright (c) Microsoft Corporation. All rights reserved.
- * monaco-kusto version: 14.0.1(d58ade77c070c9eb832c5ae067820b5be7eb3003)
+ * monaco-kusto version: 14.2.2(22fb70324f78ec316d3733a3c97919f80215e03d)
  * Released under the MIT license
  * https://https://github.com/Azure/monaco-kusto/blob/master/README.md
  *-----------------------------------------------------------------------------*/
 
-define('vs/language/kusto/main-6dbedcd3', ['exports'], (function (exports) { 'use strict';
+define('vs/language/kusto/main-2d6da563', ['exports'], (function (exports) { 'use strict';
 
     /* --------------------------------------------------------------------------------------------
      * Copyright (c) Microsoft Corporation. All rights reserved.

package/release/dev/monaco.contribution.js

@@ -1,11 +1,11 @@
 /*!-----------------------------------------------------------------------------
  * Copyright (c) Microsoft Corporation. All rights reserved.
- * monaco-kusto version: 14.0.1(d58ade77c070c9eb832c5ae067820b5be7eb3003)
+ * monaco-kusto version: 14.2.2(22fb70324f78ec316d3733a3c97919f80215e03d)
  * Released under the MIT license
  * https://https://github.com/Azure/monaco-kusto/blob/master/README.md
  *-----------------------------------------------------------------------------*/
 
-define('vs/language/kusto/monaco.contribution', ['require', 'exports', 'vs/editor/editor.main', './types-dde70929', './schema-954f3274'], (function (require, exports, monaco, types, schema) { 'use strict';
+define('vs/language/kusto/monaco.contribution', ['require', 'exports', 'vs/editor/editor.main', './types-fb9b493a', './schema-c2628e1d'], (function (require, exports, monaco, types, schema) { 'use strict';
 
     function _interopNamespaceDefault(e) {
         var n = Object.create(null);
@@ -594,6 +594,10 @@ define('vs/language/kusto/monaco.contribution', ['require', 'exports', 'vs/edito
         });
       });
     }
+
+    // Fires after semantic tokens are (re)computed following a schema update; payload carries the model URI.
+    var onSchemaUpdateCompleteEmitter = new monaco__namespace.Emitter();
+    var onSchemaUpdateComplete = onSchemaUpdateCompleteEmitter.event;
     function withMode(callback) {
       return new Promise(function (resolve, reject) { require(['./kustoMode'], resolve, reject); }).then(callback);
     }
@@ -604,7 +608,7 @@ define('vs/language/kusto/monaco.contribution', ['require', 'exports', 'vs/edito
           case 0:
             _context.next = 2;
             return withMode(function (mode) {
-              return mode.setupMode(kustoDefaults, monaco__namespace);
+              return mode.setupMode(kustoDefaults, monaco__namespace, onSchemaUpdateCompleteEmitter);
             });
           case 2:
           case "end":
@@ -675,7 +679,8 @@ define('vs/language/kusto/monaco.contribution', ['require', 'exports', 'vs/edito
       getEntityDataTypeFromCslType: schema.getEntityDataTypeFromCslType,
       kustoDefaults: kustoDefaults,
       getKustoWorker: getKustoWorker,
-      getRangeHtml: getRangeHtml
+      getRangeHtml: getRangeHtml,
+      onSchemaUpdateComplete: onSchemaUpdateComplete
     };
     monaco__namespace.languages.kusto = globalApi;
 
@@ -688,5 +693,6 @@ define('vs/language/kusto/monaco.contribution', ['require', 'exports', 'vs/edito
     exports.getKustoWorker = getKustoWorker;
     exports.getRangeHtml = getRangeHtml;
     exports.kustoDefaults = kustoDefaults;
+    exports.onSchemaUpdateComplete = onSchemaUpdateComplete;
 
 }));

package/release/dev/{schema-954f3274.js → schema-c2628e1d.js}

@@ -1,11 +1,11 @@
 /*!-----------------------------------------------------------------------------
  * Copyright (c) Microsoft Corporation. All rights reserved.
- * monaco-kusto version: 14.0.1(d58ade77c070c9eb832c5ae067820b5be7eb3003)
+ * monaco-kusto version: 14.2.2(22fb70324f78ec316d3733a3c97919f80215e03d)
  * Released under the MIT license
  * https://https://github.com/Azure/monaco-kusto/blob/master/README.md
  *-----------------------------------------------------------------------------*/
 
-define('vs/language/kusto/schema-954f3274', ['exports'], (function (exports) { 'use strict';
+define('vs/language/kusto/schema-c2628e1d', ['exports'], (function (exports) { 'use strict';
 
     // Definition of schema object in the context of language services. This model is exposed to consumers of this library.
 

package/release/dev/{types-dde70929.js → types-fb9b493a.js}

@@ -1,11 +1,11 @@
 /*!-----------------------------------------------------------------------------
  * Copyright (c) Microsoft Corporation. All rights reserved.
- * monaco-kusto version: 14.0.1(d58ade77c070c9eb832c5ae067820b5be7eb3003)
+ * monaco-kusto version: 14.2.2(22fb70324f78ec316d3733a3c97919f80215e03d)
  * Released under the MIT license
  * https://https://github.com/Azure/monaco-kusto/blob/master/README.md
  *-----------------------------------------------------------------------------*/
 
-define('vs/language/kusto/types-dde70929', ['exports'], (function (exports) { 'use strict';
+define('vs/language/kusto/types-fb9b493a', ['exports'], (function (exports) { 'use strict';
 
     var LANGUAGE_ID = 'kusto';
 

package/release/esm/{globals-5d7a5fa1.js → globals-2482dd04.js}

@@ -1,6 +1,6 @@
 /*!-----------------------------------------------------------------------------
  * Copyright (c) Microsoft Corporation. All rights reserved.
- * monaco-kusto version: 14.0.1(d58ade77c070c9eb832c5ae067820b5be7eb3003)
+ * monaco-kusto version: 14.2.2(22fb70324f78ec316d3733a3c97919f80215e03d)
  * Released under the MIT license
  * https://https://github.com/Azure/monaco-kusto/blob/master/README.md
  *-----------------------------------------------------------------------------*/

package/release/esm/kusto.worker.js

@@ -1,6 +1,6 @@
 /*!-----------------------------------------------------------------------------
  * Copyright (c) Microsoft Corporation. All rights reserved.
- * monaco-kusto version: 14.0.1(d58ade77c070c9eb832c5ae067820b5be7eb3003)
+ * monaco-kusto version: 14.2.2(22fb70324f78ec316d3733a3c97919f80215e03d)
  * Released under the MIT license
  * https://https://github.com/Azure/monaco-kusto/blob/master/README.md
  *-----------------------------------------------------------------------------*/
@@ -8,7 +8,7 @@
 import * as worker from 'monaco-editor/esm/vs/editor/editor.worker';
 import * as ls from 'vscode-languageserver-types';
 import XRegExp from 'xregexp';
-import { d as getEntityDataTypeFromCslType, a as getCallName, b as getExpression, g as getCslTypeNameFromClrType } from './schema-2df06016.js';
+import { d as getEntityDataTypeFromCslType, a as getCallName, b as getExpression, g as getCslTypeNameFromClrType } from './schema-6e15bdaa.js';
 import '@kusto/language-service/bridge.min';
 import '@kusto/language-service/Kusto.JavaScript.Client.min';
 import '@kusto/language-service/newtonsoft.json.min';
@@ -699,7 +699,8 @@ class KustoLanguageService {
       Functions,
       EntityGroups = {},
       MinorVersion,
-      MajorVersion
+      MajorVersion,
+      Graphs
     }) => ({
       name: Name,
       alternateName: databaseInContextAlternateName,
@@ -750,7 +751,17 @@ class KustoLanguageService {
             cslType: col.CslType
           })) : inputParam.Columns
         }))
-      }))
+      })),
+      graphs: Object.values(Graphs).reduce((graphArray, graph) => {
+        const graphEntity = {
+          name: graph.Name,
+          entityType: 'Graph',
+          edges: graph.Edges,
+          nodes: graph.Nodes,
+          snapshots: graph.Snapshots
+        };
+        return [...graphArray, graphEntity];
+      }, []) // this is a temporary workaround as graphs are not included in the .show schema as json command output
     }));
     return {
       clusterType: 'Engine',
@@ -1355,7 +1366,8 @@ class KustoLanguageService {
     const tableSymbols = (db.tables || []).map(this.createTableSymbol);
     const functionSymbols = (db.functions || []).map(this.createFunctionSymbol);
     const entityGroupsSymbols = (db.entityGroups || []).map(this.createEntityGroupSymbol);
-    const databaseSymbol = new sym.DatabaseSymbol.$ctor2(db.name, db.alternateName || null, tableSymbols.concat(functionSymbols).concat(entityGroupsSymbols));
+    const graphModelSymbols = (db.graphs || []).map(this.createGraphModelSymbol);
+    const databaseSymbol = new sym.DatabaseSymbol.$ctor2(db.name, db.alternateName || null, tableSymbols.concat(functionSymbols).concat(entityGroupsSymbols).concat(graphModelSymbols));
     return databaseSymbol;
   }
   convertToKustoJsSchemaV2(schema) {
@@ -1719,6 +1731,12 @@ class KustoLanguageService {
   static createEntityGroupSymbol(entityGroup) {
     return new sym.EntityGroupSymbol.$ctor3(entityGroup.name, entityGroup.members.join(), null);
   }
+  static createGraphModelSymbol(graph) {
+    const edges = new Bridge.ArrayEnumerable(graph.edges || []);
+    const nodes = new Bridge.ArrayEnumerable(graph.nodes || []);
+    const snapshots = new Bridge.ArrayEnumerable(graph.snapshots || []);
+    return new sym.GraphModelSymbol.$ctor1(graph.name, edges, nodes, snapshots);
+  }
 }
 let languageService = new KustoLanguageService(KustoLanguageService.dummySchema, {
   includeControlCommands: true,

package/release/esm/kustoMode.d.ts

@@ -9,6 +9,10 @@ export interface AugmentedWorkerAccessor {
 /**
  * Called when Kusto language is first needed (a model has the language set)
  * @param defaults
+ * @param monacoInstance
+ * @param onSchemaUpdateCompleteEmitter Optional,fired when the language service has finished all schema-related work
  */
-export declare function setupMode(defaults: LanguageServiceDefaults, monacoInstance: typeof globalThis.monaco): void;
+export declare function setupMode(defaults: LanguageServiceDefaults, monacoInstance: typeof monaco, onSchemaUpdateCompleteEmitter?: monaco.Emitter<{
+    uri: monaco.Uri;
+}>): void;
 export declare function getKustoWorker(): Promise<AugmentedWorkerAccessor>;

package/release/esm/kustoMode.js

@@ -1,6 +1,6 @@
 /*!-----------------------------------------------------------------------------
  * Copyright (c) Microsoft Corporation. All rights reserved.
- * monaco-kusto version: 14.0.1(d58ade77c070c9eb832c5ae067820b5be7eb3003)
+ * monaco-kusto version: 14.2.2(22fb70324f78ec316d3733a3c97919f80215e03d)
  * Released under the MIT license
  * https://https://github.com/Azure/monaco-kusto/blob/master/README.md
  *-----------------------------------------------------------------------------*/
@@ -9,7 +9,7 @@ import * as monaco from 'monaco-editor/esm/vs/editor/editor.api';
 import { languages } from 'monaco-editor/esm/vs/editor/editor.api';
 import * as ls from 'vscode-languageserver-types';
 import debounce from 'lodash-es/debounce';
-import { L as LANGUAGE_ID, T as Token, t as tokenTypes } from './globals-5d7a5fa1.js';
+import { L as LANGUAGE_ID, T as Token, t as tokenTypes } from './globals-2482dd04.js';
 
 class WorkerManager {
   constructor(_monacoInstance, defaults) {
@@ -792,8 +792,9 @@ const kustoLanguageDefinition = {
 };
 
 class SemanticTokensProvider {
-  constructor(classificationsGetter) {
+  constructor(classificationsGetter, onTokensProvided) {
     this.classificationsGetter = classificationsGetter;
+    this.onTokensProvided = onTokensProvided;
   }
   getLegend() {
     return {
@@ -818,10 +819,12 @@ class SemanticTokensProvider {
         prevChar = absChar;
       }
     }
-    return {
+    const result = {
       data: new Uint32Array(tokens.flat(2)),
       resultId: model.getVersionId().toString()
     };
+    this.onTokensProvided?.(resource);
+    return result;
   }
   releaseDocumentSemanticTokens() {}
 }
@@ -850,10 +853,10 @@ function toSemanticTokens(classification, model) {
 
 // Registers semantic token provider that utilizes the language service
 // for more context-relevant syntax highlighting.
-function semanticTokensProviderRegistrarCreator() {
+function semanticTokensProviderRegistrarCreator(onTokensProvided) {
   const semanticTokensProviderRegistrar = semanticTokensProviderRegistrarCreatorForTest();
   return (monacoInstance, workerAccessor) => {
-    const semanticTokensProvider = semanticTokensProviderMaker(workerAccessor);
+    const semanticTokensProvider = semanticTokensProviderMaker(workerAccessor, onTokensProvided);
     semanticTokensProviderRegistrar(monacoInstance, semanticTokensProvider);
   };
 }
@@ -866,12 +869,12 @@ function semanticTokensProviderRegistrarCreatorForTest() {
     semanticTokensDisposable = monacoInstance.languages.registerDocumentSemanticTokensProvider(LANGUAGE_ID, semanticTokensProvider);
   };
 }
-function semanticTokensProviderMaker(workerAccessor) {
+function semanticTokensProviderMaker(workerAccessor, onTokensProvided) {
   const classificationsGetter = async resource => {
     const worker = await workerAccessor(resource);
     return worker.getClassifications(resource.toString());
   };
-  return new SemanticTokensProvider(classificationsGetter);
+  return new SemanticTokensProvider(classificationsGetter, onTokensProvided);
 }
 
 languages.LanguageConfiguration;
@@ -884,10 +887,16 @@ let workerPromise = new Promise((resolve, reject) => {
 /**
  * Called when Kusto language is first needed (a model has the language set)
  * @param defaults
+ * @param monacoInstance
+ * @param onSchemaUpdateCompleteEmitter Optional,fired when the language service has finished all schema-related work
  */
-function setupMode(defaults, monacoInstance) {
+function setupMode(defaults, monacoInstance, onSchemaUpdateCompleteEmitter) {
   const onSchemaChange = new monaco.Emitter();
-  const semanticTokensProviderRegistrar = semanticTokensProviderRegistrarCreator();
+  const semanticTokensProviderRegistrar = semanticTokensProviderRegistrarCreator(uri => {
+    onSchemaUpdateCompleteEmitter?.fire({
+      uri
+    });
+  });
   const client = new WorkerManager(monacoInstance, defaults);
   const workerAccessor = (first, ...more) => {
     const augmentedSetSchema = async (schema, worker) => {

package/release/esm/languageServiceManager/kustoLanguageService.js

@@ -814,7 +814,7 @@ var KustoLanguageService = /** @class */ (function () {
         var databases = Object.keys(schema.Databases)
             .map(function (key) { return schema.Databases[key]; })
             .map(function (_a) {
-            var Name = _a.Name, Tables = _a.Tables, ExternalTables = _a.ExternalTables, MaterializedViews = _a.MaterializedViews, Functions = _a.Functions, _b = _a.EntityGroups, EntityGroups = _b === void 0 ? {} : _b, MinorVersion = _a.MinorVersion, MajorVersion = _a.MajorVersion;
+            var Name = _a.Name, Tables = _a.Tables, ExternalTables = _a.ExternalTables, MaterializedViews = _a.MaterializedViews, Functions = _a.Functions, _b = _a.EntityGroups, EntityGroups = _b === void 0 ? {} : _b, MinorVersion = _a.MinorVersion, MajorVersion = _a.MajorVersion, Graphs = _a.Graphs;
             return ({
                 name: Name,
                 alternateName: databaseInContextAlternateName,
@@ -879,6 +879,16 @@ var KustoLanguageService = /** @class */ (function () {
                         }); }),
                     });
                 }),
+                graphs: Object.values(Graphs).reduce(function (graphArray, graph) {
+                    var graphEntity = {
+                        name: graph.Name,
+                        entityType: 'Graph',
+                        edges: graph.Edges,
+                        nodes: graph.Nodes,
+                        snapshots: graph.Snapshots,
+                    };
+                    return __spreadArray(__spreadArray([], graphArray, true), [graphEntity], false);
+                }, []), // this is a temporary workaround as graphs are not included in the .show schema as json command output
             });
         });
         return {
@@ -1467,7 +1477,8 @@ var KustoLanguageService = /** @class */ (function () {
         var tableSymbols = (db.tables || []).map(this.createTableSymbol);
         var functionSymbols = (db.functions || []).map(this.createFunctionSymbol);
         var entityGroupsSymbols = (db.entityGroups || []).map(this.createEntityGroupSymbol);
-        var databaseSymbol = new sym.DatabaseSymbol.$ctor2(db.name, db.alternateName || null, tableSymbols.concat(functionSymbols).concat(entityGroupsSymbols));
+        var graphModelSymbols = (db.graphs || []).map(this.createGraphModelSymbol);
+        var databaseSymbol = new sym.DatabaseSymbol.$ctor2(db.name, db.alternateName || null, tableSymbols.concat(functionSymbols).concat(entityGroupsSymbols).concat(graphModelSymbols));
         return databaseSymbol;
     };
     KustoLanguageService.prototype.convertToKustoJsSchemaV2 = function (schema) {
@@ -1737,6 +1748,12 @@ var KustoLanguageService = /** @class */ (function () {
     KustoLanguageService.createEntityGroupSymbol = function (entityGroup) {
         return new sym.EntityGroupSymbol.$ctor3(entityGroup.name, entityGroup.members.join(), null);
     };
+    KustoLanguageService.createGraphModelSymbol = function (graph) {
+        var edges = new Bridge.ArrayEnumerable(graph.edges || []);
+        var nodes = new Bridge.ArrayEnumerable(graph.nodes || []);
+        var snapshots = new Bridge.ArrayEnumerable(graph.snapshots || []);
+        return new sym.GraphModelSymbol.$ctor1(graph.name, edges, nodes, snapshots);
+    };
     return KustoLanguageService;
 }());
 var languageService = new KustoLanguageService(KustoLanguageService.dummySchema, {