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 36

fix: req changes by greptile

Browse source
This commit is contained in:
Harshit28j 2026-03-10 18:33:17 +05:30
parent 8c3d6db482
commit 558523fd75
2 changed files with 39 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/utils.py
View file

@ -1149,8 +1149,8 @@ def function_setup( # noqa: PLR0915
# of "metadata" (to avoid conflicting with provider API metadata fields),
# populate litellm_params["metadata"] so callbacks (e.g. Langfuse) that
# read API key info from litellm_params["metadata"] see the fields.
if litellm_params.get("metadata") is None:
litellm_params["metadata"] = kwargs["litellm_metadata"]
if not litellm_params.get("metadata"):
litellm_params["metadata"] = kwargs["litellm_metadata"].copy()
logging_obj.update_environment_variables(
model=model,

37
tests/test_litellm/litellm_core_utils/test_litellm_logging.py
View file

@ -1756,6 +1756,11 @@ def test_function_setup_litellm_metadata_populates_metadata():
assert litellm_metadata is not None
assert litellm_metadata.get("user_api_key_hash") == test_api_key_hash
# metadata should be a COPY, not an alias — mutating one must not affect the other
assert (
metadata is not litellm_metadata
), "litellm_params['metadata'] should be a copy, not the same object"
def test_function_setup_metadata_takes_precedence_over_litellm_metadata():
"""
@ -1796,3 +1801,35 @@ def test_function_setup_metadata_takes_precedence_over_litellm_metadata():
litellm_metadata = litellm_params.get("litellm_metadata")
assert litellm_metadata is not None
assert litellm_metadata.get("user_api_key_hash") == "sk-hashed-xyz"
def test_function_setup_empty_metadata_falls_back_to_litellm_metadata():
"""
Test that when metadata is explicitly set to {} (empty dict), litellm_metadata
is still used to populate litellm_params["metadata"] so API key fields are visible.
"""
import litellm
kwargs = {
"model": "claude-3-5-sonnet",
"messages": [{"role": "user", "content": "hello"}],
"litellm_call_id": "test-call-id-789",
"metadata": {},
"litellm_metadata": {
"user_api_key_hash": "sk-hashed-empty-test",
"user_api_key_team_id": "team-empty-test",
},
}
logging_obj, _ = litellm.utils.function_setup(
original_function="anthropic_messages",
rules_obj=litellm.utils.Rules(),
start_time=time.time(),
**kwargs,
)
litellm_params = logging_obj.model_call_details.get("litellm_params", {})
metadata = litellm_params.get("metadata")
assert metadata is not None
assert metadata.get("user_api_key_hash") == "sk-hashed-empty-test"
assert metadata.get("user_api_key_team_id") == "team-empty-test"