RomanNum3ral
/
cai
mirror of https://github.com/aliasrobotics/cai.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

FIX CAI LIMIT PRICE update with /config

This commit is contained in:
lidia9 2025-05-19 11:08:37 +02:00
parent 7aabfe856f
commit d744d8f7bf
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/cai/util.py
View File

@ -232,12 +232,17 @@ class CostTracker:
def check_price_limit(self, new_cost: float) -> None:
"""Check if adding the new cost would exceed the price limit."""
from cai.sdk.agents.exceptions import PriceLimitExceeded
from cai.sdk.agents.run import DEFAULT_PRICE_LIMIT
if DEFAULT_PRICE_LIMIT != float("inf"):
import os
price_limit_env = os.getenv("CAI_PRICE_LIMIT")
try:
price_limit = float(price_limit_env) if price_limit_env is not None else float("inf")
except ValueError:
price_limit = float("inf")
if price_limit != float("inf"):
total_cost = self.session_total_cost + new_cost
if total_cost > DEFAULT_PRICE_LIMIT:
raise PriceLimitExceeded(total_cost, DEFAULT_PRICE_LIMIT)
if total_cost > price_limit:
raise PriceLimitExceeded(total_cost, price_limit)
def update_session_cost(self, new_cost: float) -> None:
"""Add cost to session total and log the update"""