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.

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]

sure seems similar to the above function, can it be deduplicated?

[brdskggs]

Yes, I'll try to clean it up -- the main difference is which direction to shorten.

[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.