def choice_stage_index_mapper()

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

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

    def choice_stage_index_mapper(self) -> IndexMapper[int] | None:
        return self.choice_ctx.stage_index_mapper if self.choice_ctx else None