Refactor: use SecretStr for llm_rails embeddings (#15090)

libs/community/tests/unit_tests/embeddings/test_imports.py

@@ -14,6 +14,7 @@ EXPECTED_ALL = [
     "GradientEmbeddings",
     "JinaEmbeddings",
     "LlamaCppEmbeddings",
+    "LLMRailsEmbeddings",
     "HuggingFaceHubEmbeddings",
     "MlflowAIGatewayEmbeddings",
     "MlflowEmbeddings",

libs/community/tests/unit_tests/embeddings/test_llm_rails.py

@@ -0,0 +1,21 @@
+"""Test LLMRailsEmbeddings embeddings"""
+
+from langchain_core.pydantic_v1 import SecretStr
+from pytest import CaptureFixture
+
+from langchain_community.embeddings import LLMRailsEmbeddings
+
+
+def test_api_key_is_string() -> None:
+    llm = LLMRailsEmbeddings(api_key="secret-api-key")
+    assert isinstance(llm.api_key, SecretStr)
+
+
+def test_api_key_masked_when_passed_via_constructor(
+    capsys: CaptureFixture,
+) -> None:
+    llm = LLMRailsEmbeddings(api_key="secret-api-key")
+    print(llm.api_key, end="")
+    captured = capsys.readouterr()
+
+    assert captured.out == "**********"