fix: Strip custom_llm_provider before cost lookup for proxied models

This commit is contained in:
0xallam
2026-02-20 06:52:27 -08:00
parent 3b3576b024
commit bf8020fafb

View File

@@ -259,6 +259,8 @@ class LLM:
if direct_cost is not None:
return float(direct_cost)
try:
if hasattr(response, "_hidden_params"):
response._hidden_params.pop("custom_llm_provider", None)
return completion_cost(response, model=self.config.canonical_model) or 0.0
except Exception: # noqa: BLE001
return 0.0