change only model codes in model coder (microsoft#707)

XianBW · web-flow · commit 9e8ff17575c9 · 2025-03-19T19:37:26.000+08:00
diff --git a/rdagent/components/coder/data_science/model/__init__.py b/rdagent/components/coder/data_science/model/__init__.py
@@ -87,6 +87,10 @@ def implement_one_task(
                 )
             )
 
+            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 = {

Original file line number	Diff line number	Diff line change
`@@ -87,6 +87,10 @@ def implement_one_task(`
`87`	`87`	`)`
`88`	`88`	`)`
`89`	`89`
	`90`	`+ if not all(i.startswith("model_") for i in batch_edit.keys()):`
	`91`	`+ user_prompt += "\nYou should only update model codes!"`
	`92`	`+ continue`
	`93`	`+`
`90`	`94`	`# 3. post process to align file name to the task name`
`91`	`95`	`# we assumpt batch_edit only contains one model file update.`
`92`	`96`	`batch_edit = {`