44
55from pathlib import Path
66
7+ from pydantic import SecretStr
78
89from models .config import (
910 ModelContextProtocolServer ,
@@ -47,7 +48,7 @@ def test_dump_configuration(tmp_path) -> None:
4748 llama_stack = LlamaStackConfiguration (
4849 use_as_library_client = True ,
4950 library_client_config_path = "tests/configuration/run.yaml" ,
50- api_key = "whatever" ,
51+ api_key = SecretStr ( "whatever" ) ,
5152 ),
5253 user_data_collection = UserDataCollection (
5354 feedback_enabled = False , feedback_storage = None
@@ -57,7 +58,7 @@ def test_dump_configuration(tmp_path) -> None:
5758 postgres = PostgreSQLDatabaseConfiguration (
5859 db = "lightspeed_stack" ,
5960 user = "ls_user" ,
60- password = "ls_password" ,
61+ password = SecretStr ( "ls_password" ) ,
6162 port = 5432 ,
6263 ca_cert_path = None ,
6364 ssl_mode = "require" ,
@@ -325,7 +326,7 @@ def test_dump_configuration_with_quota_limiters(tmp_path) -> None:
325326 llama_stack = LlamaStackConfiguration (
326327 use_as_library_client = True ,
327328 library_client_config_path = "tests/configuration/run.yaml" ,
328- api_key = "whatever" ,
329+ api_key = SecretStr ( "whatever" ) ,
329330 ),
330331 user_data_collection = UserDataCollection (
331332 feedback_enabled = False , feedback_storage = None
@@ -335,7 +336,7 @@ def test_dump_configuration_with_quota_limiters(tmp_path) -> None:
335336 postgres = PostgreSQLDatabaseConfiguration (
336337 db = "lightspeed_stack" ,
337338 user = "ls_user" ,
338- password = "ls_password" ,
339+ password = SecretStr ( "ls_password" ) ,
339340 port = 5432 ,
340341 ca_cert_path = None ,
341342 ssl_mode = "require" ,
0 commit comments