完成和大模型的通信

tests/test_language_rule.py

@@ -0,0 +1,32 @@
+"""语言规则生成函数测试。"""
+
+import sys
+from pathlib import Path
+
+sys.path.insert(0, str(Path(__file__).parent.parent / "src"))
+
+from prompt import language_rule
+
+
+class TestLanguageRule:
+    """测试语言规则生成函数。"""
+
+    def test_explicit_language(self) -> None:
+        """测试显式指定语言时，提示中包含该语言名称。"""
+        result = language_rule("test content", "Chinese")
+        assert "Chinese" in result
+
+    def test_auto_with_chinese_content(self) -> None:
+        """测试中文内容自动检测。"""
+        result = language_rule("这是一段中文内容", None)
+        assert "简体中文" in result
+
+    def test_auto_with_english_content(self) -> None:
+        """测试英文内容自动检测。"""
+        result = language_rule("This is English content", None)
+        assert "源文档" in result
+
+    def test_auto_detection_string(self) -> None:
+        """测试传入 "auto" 字符串时触发自动检测。"""
+        result = language_rule("test", "auto")
+        assert "源文档" in result