From 7dac4cc9cea6b268518affb76ce312ad4caf5d60 Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Tue, 3 Sep 2024 15:40:10 -0700 Subject: [PATCH] fixed regex --- vision_agent/tools/meta_tools.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/vision_agent/tools/meta_tools.py b/vision_agent/tools/meta_tools.py index 7129a94c..3670e600 100644 --- a/vision_agent/tools/meta_tools.py +++ b/vision_agent/tools/meta_tools.py @@ -501,12 +501,11 @@ def use_florence2_fine_tuning( code = artifacts[name] if task.lower() == "phrase_grounding": - pattern = r"florence2_phrase_grounding\(([^,]+),\s*([^\)]+)\)" + pattern = r"florence2_phrase_grounding\(\s*([^\)]+)\)" def replacer(match: re.Match) -> str: - arg1 = match.group(1) - arg2 = match.group(2) - return f'florence2_phrase_grounding({arg1}, {arg2}, "{fine_tune_id}")' + arg = match.group(1) # capture all initial arguments + return f'florence2_phrase_grounding({arg}, "{fine_tune_id}")' else: raise ValueError(f"Task {task} is not supported.")