diff --git a/tests/unit/test_prompt.py b/tests/unit/test_prompt.py index eadfb8221..598deaec5 100644 --- a/tests/unit/test_prompt.py +++ b/tests/unit/test_prompt.py @@ -127,15 +127,13 @@ def test_compile_with_placeholders( @pytest.fixture def langfuse(): - from langfuse._client.resource_manager import LangfuseResourceManager - - langfuse_instance = Langfuse() + langfuse_instance = Langfuse( + public_key="pk-lf-test", + secret_key="sk-lf-test", + base_url="http://localhost:3000", + ) langfuse_instance.api = Mock() - if langfuse_instance._resources is None: - langfuse_instance._resources = Mock(spec=LangfuseResourceManager) - langfuse_instance._resources.prompt_cache = PromptCache() - return langfuse_instance