I have done it trough stages, so first stages emits in natural language in the format of "context: ... and question: ...." and then the second stage collect it

I have done it trough stages, so first stages emits in natural language in the format of "context: ... and question: ...." and then the second stage collect it as json, but then wait time doubles.