From 753119374a952cac55a0d87c13e5ae081e09de4b Mon Sep 17 00:00:00 2001 From: HsiangNianian Date: Mon, 5 Jan 2026 14:33:37 +0800 Subject: feat: add max_length validation in TRPGParser to enforce input constraints --- src/base_model_trpgner/inference/__init__.py | 3 +++ 1 file changed, 3 insertions(+) (limited to 'src/base_model_trpgner/inference/__init__.py') diff --git a/src/base_model_trpgner/inference/__init__.py b/src/base_model_trpgner/inference/__init__.py index 41f3504..5824666 100644 --- a/src/base_model_trpgner/inference/__init__.py +++ b/src/base_model_trpgner/inference/__init__.py @@ -232,6 +232,9 @@ class TRPGParser: >>> result['metadata']['speaker'] '风雨' """ + if max_length > 512: + raise ValueError("max_length 不能超过 512") + # Tokenize inputs = self.tokenizer( text, -- cgit v1.2.3-70-g09d2