Chiến Lược Cắt Ngắn Chuỗi Quá Dài (Sequence Truncation: Head, Tail, Middle)

Mô tả bài toán

Khi tài liệu vượt quá giới hạn độ dài max_length, việc cắt ngắn (truncation) cần được thực hiện có chủ đích tùy theo đặc thù tác vụ:

• "tail" (cắt đuôi): Giữ lại phần đầu, cắt bỏ phần sau tokens[:max_length]. Đây là chiến lược mặc định phổ biến nhất.
• "head" (cắt đầu): Giữ lại phần đuôi, cắt bỏ phần trước tokens[-max_length:]. Thường dùng trong chat history khi các tin nhắn gần nhất quan trọng hơn tin nhắn cũ.
• "middle" (cắt khúc giữa): Giữ lại cả phần mở đầu và phần kết luận, cắt bỏ đoạn giữa tài liệu:
• Phần đầu giữ: lceil max_length / 2 rceil token đầu tiên (tính bằng (max_length + 1) // 2).
• Phần đuôi giữ: ⌊ max_length / 2 ⌋ token cuối cùng (tính bằng max_length // 2).
• Ghép lại: head_tokens + tail_tokens có tổng độ dài đúng bằng max_length.

Yêu cầu

Viết hàm:

def truncate_sequence(
    tokens: list[int],
    max_length: int,
    strategy: str = "tail"
) -> list[int]

• Kiểm tra tham số:
• max_length > 0. Nếu không, ném ValueError("max_length must be positive").
• strategy phải thuộc ("head", "tail", "middle"). Nếu không, ném ValueError(f"Unknown strategy: {strategy}").
• Nếu len(tokens) <= max_length: trả về bản sao danh sách gốc list(tokens).
• Nếu len(tokens) > max_length: thực hiện cắt theo chiến lược tương ứng và trả về list[int] có độ dài chính xác bằng max_length.

Input

• Hàm truncate_sequence(tokens, max_length, strategy): Các tham số đầu vào chứa dữ liệu Tensor/mảng NumPy hoặc giá trị siêu tham số tương ứng.

Output

• Hàm truncate_sequence: Trả về kết quả kiểu list[int] theo đúng đặc tả kỹ thuật và kích thước quy định.

Ràng buộc

• Thời gian chạy tối đa: 6000ms.
• Giới hạn bộ nhớ: 512MB.
• Dữ liệu đầu vào hợp lệ theo đúng kiểu dữ liệu và miền giá trị được mô tả.

Ví dụ 1