From df94eb6c125279a9c32bc85de8633371d50afbed Mon Sep 17 00:00:00 2001
From: HsiangNianian <i@jyunko.cn>
Date: Mon, 5 Jan 2026 14:33:10 +0800
Subject: feat: update max_length parameter for TRPGParser and onnx_infer to
 improve text parsing capabilities

---
 src/base_model_trpgner/inference/__init__.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(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 3d1d720..41f3504 100644
--- a/src/base_model_trpgner/inference/__init__.py
+++ b/src/base_model_trpgner/inference/__init__.py
@@ -214,13 +214,13 @@ class TRPGParser:
                 9: "B-timestamp", 10: "I-timestamp",
             }
 
-    def parse(self, text: str) -> Dict[str, Any]:
+    def parse(self, text: str, max_length: int = 512) -> Dict[str, Any]:
         """
         解析单条 TRPG 日志
 
         Args:
             text: 待解析的日志文本
-
+            max_length: 最大序列长度，大小512以内
         Returns:
             包含 metadata 和 content 的字典
             - metadata: speaker, timestamp
@@ -239,7 +239,7 @@ class TRPGParser:
             return_offsets_mapping=True,
             padding="max_length",
             truncation=True,
-            max_length=128,
+            max_length=max_length or 512,
         )
 
         # 推理
-- 
cgit v1.2.3-70-g09d2