test: more robust arg parsing

Signed-off-by: vsoch <vsoch@users.noreply.github.com>

Author: vsoch

resource_secretary/agents/secretary.py

@@ -168,7 +168,7 @@ async def deliberate(
             obs = "OBSERVATIONS:\n"
             for call in calls:
                 p_name, f_name, args_str = call
-                args = utils.parse_args_string(args_str)
+                args = utils.parse_args(args_str)
                 result = self.execute_call(p_name, f_name, args)
 
                 # Results are usually json, but not always

resource_secretary/utils/text.py

@@ -16,26 +16,6 @@ def format_calls(calls_block):
         return calls
 
 
-import ast
-import re
-
-
-def parse_args_string(args_str):
-    flattened = args_str.replace("\n", " ").replace("\r", " ")
-    flattened = re.sub(r"\s+", " ", flattened).strip()
-    try:
-        tree = ast.parse(f"f({flattened})")
-
-        kwargs = {}
-        for kw in tree.body[0].value.keywords:
-            kwargs[kw.arg] = ast.literal_eval(kw.value)
-
-        return kwargs
-    except SyntaxError as e:
-        print(f"Syntax error in agent output: {e}")
-        return {}
-
-
 def ensure_command(command):
     """
     Ensure the command if provided as string is split into list.
@@ -45,6 +25,23 @@ def ensure_command(command):
     return command
 
 
+import ast
+import re
+
+
+def parse_args(args_str):
+    pattern = r"(\w+)\s*=\s*(\{.*?\}|'[^']*'|\"[^\"]*\"|[^,]+)"
+    matches = re.findall(pattern, args_str)
+    result = {}
+    for key, value in matches:
+        value = value.strip()
+        try:
+            result[key] = ast.literal_eval(value)
+        except:
+            result[key] = value.strip("'\"")
+    return result
+
+
 def from_string_arg(val):
     """
     When we parse a call (from string) we need to convert into Python types.