diff --git a/src/services/flows_service.py b/src/services/flows_service.py index b33fa0fb..4c3872ca 100644 --- a/src/services/flows_service.py +++ b/src/services/flows_service.py @@ -361,22 +361,17 @@ class FlowsService: flow_json_str = re.sub( re.escape(old_llm_id), new_llm_id, flow_json_str ) + if old_llm_text_id: + flow_json_str = re.sub( + re.escape(old_llm_text_id), new_llm_text_id, flow_json_str + ) + flow_json_str = re.sub( re.escape(old_llm_id.split("-")[0]), new_llm_id.split("-")[0], flow_json_str, ) - if old_llm_text_id: - flow_json_str = re.sub( - re.escape(old_llm_text_id), new_llm_text_id, flow_json_str - ) - flow_json_str = re.sub( - re.escape(old_llm_text_id.split("-")[0]), - new_llm_text_id.split("-")[0], - flow_json_str, - ) - # Convert back to JSON flow_data = json.loads(flow_json_str)