scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 89

fix

Browse source
This commit is contained in:
Ishaan Jaffer 2025-11-14 18:56:25 -08:00
parent 9ced18b695
commit 39c1a970a1
1 changed files with 8 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
tests/test_litellm/integrations/test_langfuse.py
View file

@ -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()