Skip to content

Commit 6eb17c9

Browse files
reset all executor references
1 parent 62c32cb commit 6eb17c9

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/integrations/litellm/test_litellm.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,7 @@ async def __call__(self, *args, **kwargs):
3838
import litellm.utils as litellm_utils
3939
from litellm.litellm_core_utils import streaming_handler
4040
from litellm.litellm_core_utils import thread_pool_executor
41+
from litellm.litellm_core_utils import litellm_logging
4142

4243

4344
LITELLM_VERSION = package_version("litellm")
@@ -49,6 +50,7 @@ def reset_litellm_executor():
4950
thread_pool_executor.executor = ThreadPoolExecutor(max_workers=100)
5051
litellm_utils.executor = thread_pool_executor.executor
5152
streaming_handler.executor = thread_pool_executor.executor
53+
litellm_logging.executor = thread_pool_executor.executor
5254

5355

5456
@pytest.fixture

0 commit comments

Comments
 (0)