From b727d4a8b7934e8c8a5404cafbf127db653de449 Mon Sep 17 00:00:00 2001
From: lidia9 <lidia@aliasrobotics.com>
Date: Tue, 13 May 2025 09:28:23 +0200
Subject: [PATCH] Fix CAI_MAX_TURNS type error

---
 src/cai/sdk/agents/run.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/cai/sdk/agents/run.py b/src/cai/sdk/agents/run.py
index 3bc79a19..bdfe8721 100644
--- a/src/cai/sdk/agents/run.py
+++ b/src/cai/sdk/agents/run.py
@@ -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