def reserve_stage_index()

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

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

    def reserve_stage_index(self, choice_idx: int) -> int:
        return self._get_stage_index_mapper(choice_idx).reserve()