From 9067a5031b7c54946c7264d78efad1399f196182 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 21 Nov 2024 12:48:17 -0800 Subject: [PATCH] fix langfuse import --- tests/local_testing/test_alangfuse.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/local_testing/test_alangfuse.py b/tests/local_testing/test_alangfuse.py index 78c9805da..2d32037c1 100644 --- a/tests/local_testing/test_alangfuse.py +++ b/tests/local_testing/test_alangfuse.py @@ -7,13 +7,12 @@ import sys from typing import Any from unittest.mock import MagicMock, patch -from litellm.cache import InMemoryCache - logging.basicConfig(level=logging.DEBUG) sys.path.insert(0, os.path.abspath("../..")) import litellm from litellm import completion +from litellm.cache import InMemoryCache litellm.num_retries = 3 litellm.success_callback = ["langfuse"]