diff --git a/src/services/flows_service.py b/src/services/flows_service.py index db4d9992..a64b75e0 100644 --- a/src/services/flows_service.py +++ b/src/services/flows_service.py @@ -452,16 +452,24 @@ 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, ) + + # Replace text LLM ID references (if applicable) + 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)