From 39c1a970a13e11bf477dea286dafe363bfa5b99b Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Fri, 14 Nov 2025 18:56:25 -0800 Subject: [PATCH] fix --- tests/test_litellm/integrations/test_langfuse.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/tests/test_litellm/integrations/test_langfuse.py b/tests/test_litellm/integrations/test_langfuse.py index b7a2ed5095..b9009d2005 100644 --- a/tests/test_litellm/integrations/test_langfuse.py +++ b/tests/test_litellm/integrations/test_langfuse.py @@ -1,12 +1,14 @@ -import unittest import asyncio -from unittest.mock import patch, MagicMock -from typing import Optional -import sys -import os import datetime import json +import os +import sys +import unittest +from typing import Optional +from unittest.mock import MagicMock, patch + import pytest + import litellm from litellm.integrations.langfuse import langfuse as langfuse_module from litellm.integrations.langfuse.langfuse import LangFuseLogger @@ -255,6 +257,7 @@ class TestLangfuseUsageDetails(unittest.TestCase): with patch( "litellm.integrations.langfuse.langfuse._add_prompt_to_generation_params", side_effect=lambda generation_params, **kwargs: generation_params, + create=True, ) as mock_add_prompt_params: # Create a mock response object with usage information containing None values response_obj = MagicMock()