liumaolin
commited on
Commit
·
1816130
1
Parent(s):
310c0ba
在测试文件中添加对中文和英文系统提示的导入检查,以确保在未定义时能够正确加载配置。
Browse files
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 |
|