def reserve()

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

• 7 lines of code
• 3 McCabe index (conditional complexity)

    def reserve(self, index: int | None = None) -> int:
        if index is None:
            return self._get_fresh_index()

        if index in self.used_indices:
            raise ValueError(f"Index {index} is already taken")

        self.used_indices.add(index)
        return index