From 2bb176427a41d1558592444ca2376625e342c2ec Mon Sep 17 00:00:00 2001 From: Tim Date: Tue, 11 Feb 2025 18:12:45 +0800 Subject: [PATCH] avoid length exception (#580) --- rdagent/components/coder/data_science/model/__init__.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/rdagent/components/coder/data_science/model/__init__.py b/rdagent/components/coder/data_science/model/__init__.py index 812a71c6..e6da42fe 100644 --- a/rdagent/components/coder/data_science/model/__init__.py +++ b/rdagent/components/coder/data_science/model/__init__.py @@ -100,12 +100,14 @@ class ModelMultiProcessEvolvingStrategy(MultiProcessEvolvingStrategy): for key, value in batch_edit.items() } + user_prompt = user_prompt + "\nPlease avoid generating same code to former code!" + if batch_edit and max(len(i.encode("utf-8")) for i in batch_edit.keys()) > 255: + continue + if batch_edit[f"{target_task.name}.py"] != "__DEL__" and batch_edit[ f"{target_task.name}.py" ] != workspace.file_dict.get(f"{target_task.name}.py"): break - else: - user_prompt = user_prompt + "\nPlease avoid generating same code to former code!" else: raise CoderError("Failed to generate a new model code.")