diff options
Diffstat (limited to 'llamachat/providers.py')
| -rw-r--r-- | llamachat/providers.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/llamachat/providers.py b/llamachat/providers.py index ce2d01d..a8b2350 100644 --- a/llamachat/providers.py +++ b/llamachat/providers.py @@ -46,6 +46,9 @@ class Provider: vision: bool | None = None price_in: float | None = None price_out: float | None = None + # Provider-specific reasoning token budget. Currently SiliconFlow only; + # ignored when unset so other endpoints do not receive an unknown key. + thinking_budget: int | None = None @property def is_local(self) -> bool: @@ -185,6 +188,7 @@ def parse(values: dict, warnings: list[str] | None = None) -> dict[str, Provider vision=None if vision is None else bool(vision), price_in=_number(entry.get("price_in"), float), price_out=_number(entry.get("price_out"), float), + thinking_budget=_number(entry.get("thinking_budget"), int), ) return out |