def with_choice_ctx()

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

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

            def with_choice_ctx(choice_idx: int) -> ElementPath:
                return path.with_choice_ctx(
                    ChoiceContext(
                        index=choice_idx,
                        stage_index_mapper=self._get_stage_index_mapper(
                            choice_idx
                        ),
                    )
                )