mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-12 23:01:41 +00:00
fix req changes by greptile
This commit is contained in:
parent
94d4a4dce0
commit
5abfddf47f
2 changed files with 15 additions and 2 deletions
|
|
@ -601,7 +601,7 @@ class Logging(LiteLLMLoggingBaseClass):
|
|||
if "litellm_metadata" in kwargs and isinstance(
|
||||
kwargs["litellm_metadata"], dict
|
||||
):
|
||||
if "metadata" not in base_litellm_params:
|
||||
if not base_litellm_params.get("metadata"):
|
||||
base_litellm_params["metadata"] = dict(kwargs["litellm_metadata"])
|
||||
else:
|
||||
base_litellm_params["metadata"] = dict(base_litellm_params["metadata"])
|
||||
|
|
|
|||
|
|
@ -271,6 +271,19 @@ class TestUpdateFromKwargs:
|
|||
assert result_meta["from_caller"] is True
|
||||
assert result_meta["user_api_key_hash"] == "hashed-xyz"
|
||||
|
||||
def test_merge_handles_metadata_none(self, logging_obj):
|
||||
"""When metadata=None is explicitly in kwargs alongside litellm_metadata,
|
||||
the merge should not crash (dict(None) would TypeError)."""
|
||||
kwargs = {
|
||||
"metadata": None,
|
||||
"litellm_metadata": {"user_api_key_hash": "hashed-abc"},
|
||||
}
|
||||
|
||||
logging_obj.update_from_kwargs(kwargs=kwargs)
|
||||
|
||||
result_meta = logging_obj.litellm_params["metadata"]
|
||||
assert result_meta["user_api_key_hash"] == "hashed-abc"
|
||||
|
||||
def test_caller_litellm_params_win_over_kwargs(self, logging_obj):
|
||||
"""Explicit litellm_params from the caller should override auto-extracted values."""
|
||||
kwargs = {"metadata": {"from_kwargs": True}}
|
||||
|
|
@ -1937,7 +1950,7 @@ def test_function_setup_litellm_metadata_populates_metadata():
|
|||
), "litellm_params['metadata'] should be a copy, not the same object"
|
||||
|
||||
|
||||
def test_function_setup_metadata_takes_precedence_over_litellm_metadata():
|
||||
def test_function_setup_merges_litellm_metadata_without_overwriting():
|
||||
"""
|
||||
Test that when BOTH metadata and litellm_metadata are present (e.g., user sets
|
||||
Anthropic API metadata AND proxy adds litellm_metadata), existing metadata keys
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue