chat worked

backend/src/services/__tests__/test_llm_prompt_templates.py

@@ -0,0 +1,62 @@
+# [DEF:backend.src.services.__tests__.test_llm_prompt_templates:Module]
+# @TIER: STANDARD
+# @SEMANTICS: tests, llm, prompts, templates, settings
+# @PURPOSE: Validate normalization and rendering behavior for configurable LLM prompt templates.
+# @LAYER: Domain Tests
+# @RELATION: DEPENDS_ON -> backend.src.services.llm_prompt_templates
+# @INVARIANT: All required prompt keys remain available after normalization.
+
+from src.services.llm_prompt_templates import (
+    DEFAULT_LLM_PROMPTS,
+    normalize_llm_settings,
+    render_prompt,
+)
+
+
+# [DEF:test_normalize_llm_settings_adds_default_prompts:Function]
+# @TIER: STANDARD
+# @PURPOSE: Ensure legacy/partial llm settings are expanded with all prompt defaults.
+# @PRE: Input llm settings do not contain complete prompts object.
+# @POST: Returned structure includes required prompt templates with fallback defaults.
+def test_normalize_llm_settings_adds_default_prompts():
+    normalized = normalize_llm_settings({"default_provider": "x"})
+
+    assert "prompts" in normalized
+    assert normalized["default_provider"] == "x"
+    for key in DEFAULT_LLM_PROMPTS:
+        assert key in normalized["prompts"]
+        assert isinstance(normalized["prompts"][key], str)
+# [/DEF:test_normalize_llm_settings_adds_default_prompts:Function]
+
+
+# [DEF:test_normalize_llm_settings_keeps_custom_prompt_values:Function]
+# @TIER: STANDARD
+# @PURPOSE: Ensure user-customized prompt values are preserved during normalization.
+# @PRE: Input llm settings contain custom prompt override.
+# @POST: Custom prompt value remains unchanged in normalized output.
+def test_normalize_llm_settings_keeps_custom_prompt_values():
+    custom = "Doc for {dataset_name} using {columns_json}"
+    normalized = normalize_llm_settings(
+        {"prompts": {"documentation_prompt": custom}}
+    )
+
+    assert normalized["prompts"]["documentation_prompt"] == custom
+# [/DEF:test_normalize_llm_settings_keeps_custom_prompt_values:Function]
+
+
+# [DEF:test_render_prompt_replaces_known_placeholders:Function]
+# @TIER: STANDARD
+# @PURPOSE: Ensure template placeholders are deterministically replaced.
+# @PRE: Template contains placeholders matching provided variables.
+# @POST: Rendered prompt string contains substituted values.
+def test_render_prompt_replaces_known_placeholders():
+    rendered = render_prompt(
+        "Hello {name}, diff={diff}",
+        {"name": "bot", "diff": "A->B"},
+    )
+
+    assert rendered == "Hello bot, diff=A->B"
+# [/DEF:test_render_prompt_replaces_known_placeholders:Function]
+
+
+# [/DEF:backend.src.services.__tests__.test_llm_prompt_templates:Module]