mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-08 03:08:45 +00:00
Merge ca1329b572 into eb0e3f8c18
This commit is contained in:
commit
55a1fb36e2
2 changed files with 31 additions and 2 deletions
|
|
@ -23,7 +23,7 @@ else:
|
|||
class LangFuseHandler:
|
||||
@staticmethod
|
||||
def get_langfuse_logger_for_request(
|
||||
standard_callback_dynamic_params: StandardCallbackDynamicParams,
|
||||
standard_callback_dynamic_params: StandardCallbackDynamicParams | None,
|
||||
in_memory_dynamic_logger_cache: DynamicLoggingCache,
|
||||
globalLangfuseLogger: LangFuseLogger | None = None,
|
||||
) -> LangFuseLogger:
|
||||
|
|
@ -163,7 +163,7 @@ class LangFuseHandler:
|
|||
|
||||
@staticmethod
|
||||
def _dynamic_langfuse_credentials_are_passed(
|
||||
standard_callback_dynamic_params: StandardCallbackDynamicParams,
|
||||
standard_callback_dynamic_params: StandardCallbackDynamicParams | None,
|
||||
) -> bool:
|
||||
"""
|
||||
This function is used to check if the dynamic langfuse credentials are passed in standard_callback_dynamic_params
|
||||
|
|
@ -172,6 +172,9 @@ class LangFuseHandler:
|
|||
bool: True if the dynamic langfuse credentials are passed, False otherwise
|
||||
"""
|
||||
|
||||
if standard_callback_dynamic_params is None:
|
||||
return False
|
||||
|
||||
if (
|
||||
standard_callback_dynamic_params.get("langfuse_host") is not None
|
||||
or standard_callback_dynamic_params.get("langfuse_public_key") is not None
|
||||
|
|
|
|||
26
tests/test_litellm/test_langfuse_handler.py
Normal file
26
tests/test_litellm/test_langfuse_handler.py
Normal file
|
|
@ -0,0 +1,26 @@
|
|||
from unittest.mock import MagicMock
|
||||
|
||||
from litellm.integrations.langfuse.langfuse_handler import LangFuseHandler
|
||||
|
||||
|
||||
def test_missing_dynamic_params_returns_global_logger() -> None:
|
||||
global_logger = MagicMock()
|
||||
dynamic_logger_cache = MagicMock()
|
||||
|
||||
logger = LangFuseHandler.get_langfuse_logger_for_request(
|
||||
standard_callback_dynamic_params=None,
|
||||
in_memory_dynamic_logger_cache=dynamic_logger_cache,
|
||||
globalLangfuseLogger=global_logger,
|
||||
)
|
||||
|
||||
assert logger is global_logger
|
||||
dynamic_logger_cache.get_cache.assert_not_called()
|
||||
|
||||
|
||||
def test_missing_dynamic_params_are_not_dynamic_credentials() -> None:
|
||||
assert (
|
||||
LangFuseHandler._dynamic_langfuse_credentials_are_passed(
|
||||
standard_callback_dynamic_params=None
|
||||
)
|
||||
is False
|
||||
)
|
||||
Loading…
Add table
Reference in a new issue