KQL syntax stopped working

[digitarald]

Environment data

• Language Server version: 2021.6.2-pre.1
• OS and version: linux x64
• Python version (and distribution if applicable, e.g. Anaconda):
• python.analysis.indexing: undefined
• python.analysis.typeCheckingMode: off

Expected behaviour

Can use %kql

Actual behaviour

Pylance puts errors everywhere.

Code Snippet / Additional information

%%kql RetainedByCohort <<
let cohortWindow = 7d;

[judej]

The code snippet is not Python. Not sure how Pylance is involved. @DonJayamanne

[jakebailey]

This seems like a notebooks bug to me; non-python cells should not be treated as python code and given to language servers.

[jakebailey]

Related is microsoft/pylance-release#894, but the code snippet given in the issue isn't Python.

The screenshot seems sort of like a magic command in a Python cell, maybe? That, I'm not sure without more context.

[digitarald]

Forgot to add: This works with Jedi.

[jakebailey]

Jedi isn't a language server (at least not yet), so the notebook APIs don't try and use it. If you use JediLSP, I'm guessing you'll see it break as well.

[DonJayamanne]

FYI Installing https://marketplace.visualstudio.com/items?itemName=donjayamanne.kusto will give you code completions for KQL in Jupyter notebooks.

[digitarald]

Thanks @DonJayamanne, I have it installed and am an avid fan.

[claudiaregio]

@jakebailey Does this need to be transferred to vscode-jupyter? Sounds like the recommendation above of not marking these cells as python cells and passing them to the language server would be work on jupyter?

[jakebailey]

Yes, I'll transfer it for you.

[DonJayamanne]

This isn't something we did in the past, hence not sure what you (@digitarald ) mean by stopped working?
@digitarald Pleaes could you provide the following info:

• Version of VS Code
• Version of Jupyter extension

[jakebailey]

I think the confusing thing is why the KQL cell is being sent to Pylance at all...

[digitarald]

This isn't something we did in the past,

When I said it worked in the past I meant it worked before Pylance rolled out. Using Jedi works.

[DonJayamanne]

think the confusing thing is why the KQL cell is being sent to Pylance at all...

thats because its a Python cell, %kql is a Python magic, meaning this code is executed in the Python kernel.
Thats why my extension chagnes the language of the cell from Python to Kql so that we dont get Python language server getting messages for this cell.

[DonJayamanne]

TODO:

• When opening a notebook, ensure we look for cell magics & set language appropriately
• When typing look for magics & change language accordingly

[rchiodo]

Dupe of #4588