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

Fix CAI_MAX_TURNS type error

This commit is contained in:
lidia9 2025-05-13 09:28:23 +02:00
parent 91e9bf63bc
commit b727d4a8b7
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/cai/sdk/agents/run.py
View File

@ -45,7 +45,18 @@ from .usage import Usage
from .util import _coro, _error_tracing
import os
DEFAULT_MAX_TURNS = os.getenv("CAI_MAX_TURNS", float("inf"))
# CAI_MAX_TURNS must be converted to an int to avoid type mismatch error when comparing.
max_turns_env = os.getenv("CAI_MAX_TURNS")
if max_turns_env is not None:
try:
DEFAULT_MAX_TURNS = int(max_turns_env)
except ValueError:
try:
DEFAULT_MAX_TURNS = float(max_turns_env)
except ValueError:
DEFAULT_MAX_TURNS = float("inf")
else:
DEFAULT_MAX_TURNS = float("inf")
@dataclass