|
|
@@ -591,6 +591,36 @@ class WrapperGeneratorTests(unittest.TestCase):
|
|
|
profile_with_mmproj["capabilities"]["tools"],
|
|
|
)
|
|
|
|
|
|
+<<<<<<< HEAD
|
|
|
+=======
|
|
|
+ def test_prompt_template_removes_lmstudio_unsupported_safe_filter(self):
|
|
|
+ metadata = {
|
|
|
+ "general.architecture": "qwen35moe",
|
|
|
+ "general.name": "Safe Filter Model",
|
|
|
+ "general.basename": "Safe Filter Model",
|
|
|
+ "tokenizer.chat_template": (
|
|
|
+ "{{ '<think>\\n'|safe }}"
|
|
|
+ "{{- message.content | safe }}"
|
|
|
+ "{{ tool_call.arguments|tojson|safe }}"
|
|
|
+ ),
|
|
|
+ "qwen2.context_length": "262144",
|
|
|
+ }
|
|
|
+
|
|
|
+ profile = gen.build_virtual_model_profile(metadata)
|
|
|
+ files = gen.build_virtual_model_files(
|
|
|
+ publisher="publisher-e",
|
|
|
+ slug="safe-filter-model",
|
|
|
+ model_key="publisher-e/Safe-Filter-Model-GGUF/model.gguf",
|
|
|
+ profile=profile,
|
|
|
+ )
|
|
|
+
|
|
|
+ self.assertNotIn("|safe", files["model.yaml"])
|
|
|
+ self.assertNotIn("| safe", files["model.yaml"])
|
|
|
+ self.assertIn("{{ '<think>\\n' }}", files["model.yaml"])
|
|
|
+ self.assertIn("{{- message.content }}", files["model.yaml"])
|
|
|
+ self.assertIn("{{ tool_call.arguments|tojson }}", files["model.yaml"])
|
|
|
+
|
|
|
+>>>>>>> cd0a3e8 (feat: 更正修复Jinja prompt template)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
unittest.main()
|