gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-08-01 09:27:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: add check when retrying gen model codes (#699)

Browse Source 
* add check when retrying gen model codes

* fix CI
This commit is contained in:
XianBW
2025-03-18 15:59:34 +08:00
committed by GitHub
parent 7b3473cf30
commit d66c045115
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/components/coder/data_science/model/__init__.py
+5
View File
@@ -90,12 +90,17 @@ class ModelMultiProcessEvolvingStrategy(MultiProcessEvolvingStrategy):
)
# 3. post process to align file name to the task name
# we assumpt batch_edit only contains one model file update.
batch_edit = {
(f"{target_task.name}.py" if value != "__DEL__" and key != f"{target_task.name}.py" else key): value
for key, value in batch_edit.items()
}
user_prompt = user_prompt + "\nPlease avoid generating same code to former code!"
# TODO: besides same code problem, we should also consider other problems lead to retry.
if f"{target_task.name}.py" not in batch_edit:
continue
if batch_edit and max(len(i.encode("utf-8")) for i in batch_edit.keys()) > 255:
continue