salvacybersec/strix
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: Strip custom_llm_provider before cost lookup for proxied models

Browse Source
This commit is contained in:
0xallam
2026-02-20 06:52:27 -08:00
parent 3b3576b024
commit bf8020fafb
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
strix/llm/llm.py
View File

@@ -259,6 +259,8 @@ class LLM:
if direct_cost is not None: if direct_cost is not None:
return float(direct_cost) return float(direct_cost)
try: 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 return completion_cost(response, model=self.config.canonical_model) or 0.0
except Exception: # noqa: BLE001 except Exception: # noqa: BLE001
return 0.0 return 0.0