def with_stage_idx()

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

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

    def with_stage_idx(self, stage_idx: int) -> Self:
        return self.copy(update={"stage_idx": stage_idx})