liumaolin commited on
Commit
1816130
·
1 Parent(s): 310c0ba

在测试文件中添加对中文和英文系统提示的导入检查,以确保在未定义时能够正确加载配置。

Browse files
Files changed (1) hide show
  1. tests/test_llm_dialogue.py +6 -0
tests/test_llm_dialogue.py CHANGED
@@ -31,6 +31,12 @@ ENGLISH_SYSTEM_PROMPT = (
31
  "/no_think"
32
  )
33
 
 
 
 
 
 
 
34
 
35
  class TestLLMDialogue(unittest.TestCase):
36
 
 
31
  "/no_think"
32
  )
33
 
34
+ if not CHINESE_SYSTEM_PROMPT:
35
+ from voice_dialogue.config.llm_config import CHINESE_SYSTEM_PROMPT
36
+
37
+ if not ENGLISH_SYSTEM_PROMPT:
38
+ from voice_dialogue.config.llm_config import ENGLISH_SYSTEM_PROMPT
39
+
40
 
41
  class TestLLMDialogue(unittest.TestCase):
42