From 0d2e84f58e2ed79d3534add4adf9dafd70d571aa Mon Sep 17 00:00:00 2001
From: Boris Arzentar <borisarzentar@gmail.com>
Date: Wed, 3 Dec 2025 10:59:17 +0100
Subject: [PATCH] test: test_strip_quotes_from_strings

---
 .../infrastructure/llm/test_llm_config.py     | 46 +++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 cognee/tests/unit/infrastructure/llm/test_llm_config.py

diff --git a/cognee/tests/unit/infrastructure/llm/test_llm_config.py b/cognee/tests/unit/infrastructure/llm/test_llm_config.py
new file mode 100644
index 000000000..0c024db2c
--- /dev/null
+++ b/cognee/tests/unit/infrastructure/llm/test_llm_config.py
@@ -0,0 +1,46 @@
+import pytest
+
+from cognee.infrastructure.llm.config import LLMConfig
+
+
+def test_strip_quotes_from_strings():
+    """
+    Test if the LLMConfig.strip_quotes_from_strings model validator behaves as expected.
+    """
+    config = LLMConfig(
+        # Strings with surrounding double quotes ("value" → value)
+        llm_api_key='"double_value"',
+        # Strings with surrounding single quotes ('value' → value)
+        llm_endpoint="'single_value'",
+        # Strings without quotes (value → value)
+        llm_api_version="no_quotes_value",
+        # Empty quoted strings ("" → empty string)
+        fallback_model='""',
+        # None values (should remain None)
+        baml_llm_api_key=None,
+        # Mixed quotes ("value' → unchanged)
+        fallback_endpoint="\"mixed_quote'",
+        # Strings with internal quotes ("internal\"quotes" → internal"quotes")
+        baml_llm_model='"internal"quotes"',
+    )
+
+    # Strings with surrounding double quotes ("value" → value)
+    assert config.llm_api_key == "double_value"
+
+    # Strings with surrounding single quotes ('value' → value)
+    assert config.llm_endpoint == "single_value"
+
+    # Strings without quotes (value → value)
+    assert config.llm_api_version == "no_quotes_value"
+
+    # Empty quoted strings ("" → empty string)
+    assert config.fallback_model == ""
+
+    # None values (should remain None)
+    assert config.baml_llm_api_key is None
+
+    # Mixed quotes ("value' → unchanged)
+    assert config.fallback_endpoint == "\"mixed_quote'"
+
+    # Strings with internal quotes ("internal\"quotes" → internal"quotes")
+    assert config.baml_llm_model == 'internal"quotes'