context api implementation

[gaudyb]

Description

Separate context logic into different methods.

Checklist

• I have tested these changes locally.
• I have reviewed the code changes.
• I have updated the documentation (if necessary).
• I have added appropriate unit tests (if applicable).

Additional Notes

[Add any additional notes or context that may be helpful for the reviewer(s).]

[gaudyb]

We will discuss again this implementation in the future