def _get_stage_index_mapper()

in aidial_interceptors_sdk/chat_completion/response_handler.py [0:0]

• 4 lines of code
• 2 McCabe index (conditional complexity)

    def _get_stage_index_mapper(self, choice_idx: int) -> IndexMapper[int]:
        if choice_idx not in self._stage_indices:
            self._stage_indices[choice_idx] = IndexMapper()
        return self._stage_indices[choice_idx]