def close()

in aidial_sdk/chat_completion/choice.py [0:0]

• 10 lines of code
• 5 McCabe index (conditional complexity)

    def close(self, finish_reason: Optional[FinishReason] = None) -> None:
        if not self._opened:
            raise runtime_error("Trying to close an unopened choice")
        if self._closed:
            raise runtime_error(
                "Trying to close a choice which is already closed"
            )

        reason = finish_reason or self._last_finish_reason or FinishReason.STOP

        self._closed = True
        self.send_chunk(EndChoiceChunk(reason, self._index))