scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-11 03:38:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 84

test(otel): keep the guardrail charge on the guardrail span for unpriced calls

Browse source
This commit is contained in:
HardMax71 2026-10-03 12:07:58 +02:00
parent 7c8673f8cc
commit 43b96d653d
1 changed files with 21 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
tests/unit/integrations/otel/test_otel_v2_emitter.py
View file

@ -123,6 +123,27 @@ def test_llm_call_span_tells_unpriced_call_from_free_call(cost_fields, expected_
assert langfuse_cost == expected_cost
def test_unpriced_call_keeps_the_guardrail_charge_on_the_guardrail_span():
engine, exporter = _engine()
payload = _payload(
response_cost=0.002,
response_cost_failure_debug_info=_PRICE_LOOKUP_FAILED,
cost_breakdown={"guardrail_cost": 0.002, "total_cost": 0.002},
)
engine.emit(SpanRole.LLM_CALL, LLMCallSpanData.from_standard_logging_payload(payload))
engine.emit(
SpanRole.GUARDRAIL,
GuardrailSpanData.from_logging_entry(
{"guardrail_name": "bedrock", "guardrail_status": "success", "guardrail_cost": 0.002}
),
)
llm_span, guardrail_span = exporter.get_finished_spans()
assert f"{LiteLLM.COST_PREFIX}total" not in llm_span.attributes
assert guardrail_span.attributes[LiteLLM.GUARDRAIL_COST] == 0.002
def test_tracer_scope_carries_litellm_version():
from litellm._version import version as litellm_version