langchain/libs/community/tests/unit_tests/utilities/test_tavily.py
William FH e5cf1e2414
Community[patch]use secret str in Tavily and HuggingFaceInferenceEmbeddings (#16109)
So the api keys don't show up in repr's 

Still need to do tests
2024-01-17 00:30:07 -08:00

8 lines
312 B
Python

from langchain_community.utilities.tavily_search import TavilySearchAPIWrapper
def test_api_wrapper_api_key_not_visible() -> None:
"""Test that an exception is raised if the API key is not present."""
wrapper = TavilySearchAPIWrapper(tavily_api_key="abcd123")
assert "abcd123" not in repr(wrapper)