Completion suggestion with contexts

Versions (relevant - OpenSearch/Dashboard/Server OS/Browser):

Describe the issue:
Basic completion suggestion works well but in our use case, we need to suggest using a subset of documents to do completions. For example, how do I index suggestions with additional context with each suggestion and only use those contexts during query time to filter out suggestion completion.

Index time:
<> key=abc
<> key=def

Query time: I will specify key=abc which means I get suggestions1 completions, and so on.

Appreciate your advice on this.

For reference, see here -Context suggester inspiration

Configuration:

Latest opensearch

Relevant Logs or Screenshots: