mirror of
https://github.com/tiennm99/litellm.git
synced 2026-06-17 18:48:36 +00:00
Harshit Jain
13130ea3e1
* fix: support multi-project keys and fix trace leakage * fix: Langfuse otel handle * fix lint errors mypy * passing all test case --------- Co-authored-by: Krish Dholakia <krrishdholakia@gmail.com>
53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
import sys
|
|
import os
|
|
|
|
sys.path.insert(0, os.path.abspath("../.."))
|
|
|
|
from litellm.litellm_core_utils.initialize_dynamic_callback_params import (
|
|
initialize_standard_callback_dynamic_params,
|
|
)
|
|
|
|
|
|
def test_dynamic_key_extraction_from_metadata():
|
|
"""
|
|
Test extraction of langfuse keys from metadata in kwargs.
|
|
This simulates a Proxy request where keys are passed in metadata.
|
|
"""
|
|
kwargs = {
|
|
"metadata": {
|
|
"langfuse_public_key": "pk-test",
|
|
"langfuse_secret_key": "sk-test",
|
|
"langfuse_host": "https://test.langfuse.com",
|
|
}
|
|
}
|
|
|
|
params = initialize_standard_callback_dynamic_params(kwargs)
|
|
|
|
assert params.get("langfuse_public_key") == "pk-test"
|
|
assert params.get("langfuse_secret_key") == "sk-test"
|
|
assert params.get("langfuse_host") == "https://test.langfuse.com"
|
|
|
|
|
|
def test_dynamic_key_extraction_from_litellm_params_metadata():
|
|
"""
|
|
Test extraction of langfuse keys from litellm_params.metadata.
|
|
"""
|
|
kwargs = {
|
|
"litellm_params": {
|
|
"metadata": {
|
|
"langfuse_public_key": "pk-litellm",
|
|
"langfuse_secret_key": "sk-litellm",
|
|
}
|
|
}
|
|
}
|
|
|
|
params = initialize_standard_callback_dynamic_params(kwargs)
|
|
|
|
assert params.get("langfuse_public_key") == "pk-litellm"
|
|
assert params.get("langfuse_secret_key") == "sk-litellm"
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_dynamic_key_extraction_from_metadata()
|
|
test_dynamic_key_extraction_from_litellm_params_metadata()
|