parse string env to int

firstbatchxyz · Jun 12, 2024 · bace81e · bace81e
1 parent 47fe496
commit bace81e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/dria_searching_agent/config/config.py b/src/dria_searching_agent/config/config.py
@@ -32,7 +32,11 @@ def AGENT_MODEL_NAME():
     return config.agent_model_name
 
 def AGENT_MAX_ITER():
-    return config.agent_max_iter
+    try:
+        int_val = int(config.agent_max_iter)
+        return int_val
+    except ValueError:
+        return 10
 
 def ANTHROPIC_KEY():
     return config.anthropic_key