fix(amazonq): Previous and subsequent cells are used as context for completion in a Jupyter notebook

[brdskggs]

Problem

VS Code treats each cell in a notebook as a separate editor. As a result, when building the left- and right-contexts for the completion from the current editor, we were limited to just the current cell, which might be very small and/or reference variables and functions defined in other cells. That meant that completions never used the context of other cells when making suggestions, and were often very generic. #7031

Solution

The extractContextForCodeWhisperer function now checks if it is being called in a cell in a Jupyter notebook. If so, it collects the surrounding cells to use as context, respecting the maximum context length. During this process, Markdown cells have each line prefixed with a language-specific comment character.

---

• Treat all work as PUBLIC. Private feature/x branches will not be squash-merged at release time.
• Your code changes must meet the guidelines in CONTRIBUTING.md.
• License: I confirm that my contribution is made under the terms of the Apache 2.0 license.

[github-actions]

• This pull request modifies code in src/* but no tests were added/updated.
  • Confirm whether tests should be added or ensure the PR description explains why tests are not required.

[justinmk3]

This is helpful, thanks! Tests for this function are here:

aws-toolkit-vscode/packages/amazonq/test/unit/codewhisperer/util/editorContext.test.ts

Line 32 in d457022

describe('extractContextForCodeWhisperer', function () {

You can run that test file by opening it in vscode then choose Extension tests (current file) (amazon) in the run/debug menu:

[brdskggs]

Will consolidate the function doing the suffix and prefix context fetching.

[brdskggs]

Both code cells and Markdown cells are used for context now; Markdown cells have lines prefixed with a language-specific comment based on the language of the cell the completion is triggered from.

[justinmk3]

lint failure

Error:   112:13  error  'cellText' is never reassigned. Use 'const' instead  prefer-const

[justinmk3]

I just noticed this. It's not sustainable for us to introduce yet another language-specific handling just for notebooks. Why isn't the notebook logic using the existing logic used by other Q functionality ?

[brdskggs]

Good question! I'm new to this codebase, so I'll have to research what existing logic you are referring to -- any quick pointers of where to start looking are appreciated.

[brdskggs]

Never mind, Q to the rescue -- figured out the changes I needed to make to runtimeLanguageContext.