From 60b66aa6c9c837817631281ed6198d61dac34bd3 Mon Sep 17 00:00:00 2001
From: XianBW <36835909+XianBW@users.noreply.github.com>
Date: Wed, 19 Mar 2025 19:37:26 +0800
Subject: [PATCH] change only model codes in model coder (#707)

---
 rdagent/components/coder/data_science/model/__init__.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/rdagent/components/coder/data_science/model/__init__.py b/rdagent/components/coder/data_science/model/__init__.py
index ea592cec..da960f8f 100644
--- a/rdagent/components/coder/data_science/model/__init__.py
+++ b/rdagent/components/coder/data_science/model/__init__.py
@@ -87,6 +87,10 @@ class ModelMultiProcessEvolvingStrategy(MultiProcessEvolvingStrategy):
                 )
             )
 
+            if not all(i.startswith("model_") for i in batch_edit.keys()):
+                user_prompt += "\nYou should only update model codes!"
+                continue
+
             # 3. post process to align file name to the task name
             # we assumpt batch_edit only contains one model file update.
             batch_edit = {