The cleanup method can't verify if sessions created via op_sb_ai_ort_init_session() are still in use.
What is the new behavior?
This PR fixes this issue, by ensure that sessions created from op_sb_ai_ort_init_session() will have its reference stored in Worker's OpState. This guarantees that cleanup method will don't drop Sessions attached to an active worker.
Also was introduced a better error handling when it can't found a session.
Additional context
Solved:
Trying to retrieve a session that was dropped while the worker was still consuming it
What kind of change does this PR introduce?
Bug fix
What is the current behavior?
The
cleanup
method can't verify if sessions created viaop_sb_ai_ort_init_session()
are still in use.What is the new behavior?
This PR fixes this issue, by ensure that sessions created from
op_sb_ai_ort_init_session()
will have its reference stored in Worker'sOpState
. This guarantees thatcleanup
method will don't dropSessions
attached to an active worker.Also was introduced a better error handling when it can't found a session.
Additional context
Solved: