fix(agents-api): strip expression before checking compatibility

julep-ai · Feb 12, 2025 · 3c3de64 · 3c3de64
1 parent 29cb81d
commit 3c3de64
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/agents-api/agents_api/activities/task_steps/base_evaluate.py b/agents-api/agents_api/activities/task_steps/base_evaluate.py
@@ -18,6 +18,8 @@
 
 
 def backwards_compatibility(expr: str) -> str:
+    expr = expr.strip()
+
     if expr.startswith("$ "):
         return expr
 

diff --git a/agents-api/tests/test_base_evaluate.py b/agents-api/tests/test_base_evaluate.py
@@ -227,3 +227,8 @@ async def _():
     exprs = "hello world"
     result = backwards_compatibility(exprs)
     assert result == "hello world"
+
+    # Test spaces at the beginning and end
+    exprs = "  _[0]  "
+    result = backwards_compatibility(exprs)
+    assert result == "$ _[0]"