fix(chinese-instruct.py): 修复输入为空时 build_prompt 报错问题

2025-07-21 13:09:54 +08:00
parent 1ef14a997e
commit 56b64ab8c3
1 changed files with 2 additions and 2 deletions
@@ -19,8 +19,8 @@ def build_prompt(query:str, response:str) -> str:


 def process_func(input_dict: dict):
-    query = input_dict["prompt"]
-    response = input_dict["response"]
+    query = input_dict["prompt"] if input_dict["prompt"] else ""
+    response = input_dict["response"] if input_dict["response"] else ""
    return {"text": build_prompt(query, response)}