From 660eed93bb882db66d5d07dd12e948fc4e3cdc28 Mon Sep 17 00:00:00 2001 From: "jinli.yl" Date: Fri, 21 Nov 2025 10:37:59 +0800 Subject: [PATCH] feat(file_tool): add save_answer option to BatchWriteFileOp --- reme_ai/config/default.yaml | 1 - reme_ai/context/file_tool/batch_write_file_op.py | 6 +++++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/reme_ai/config/default.yaml b/reme_ai/config/default.yaml index 52330320..f17449ac 100644 --- a/reme_ai/config/default.yaml +++ b/reme_ai/config/default.yaml @@ -243,7 +243,6 @@ llm: params: temperature: 0.6 token_count: # Optional - model_name: Qwen/Qwen3-30B-A3B-Instruct-2507 backend: base qwen3_30b_instruct: diff --git a/reme_ai/context/file_tool/batch_write_file_op.py b/reme_ai/context/file_tool/batch_write_file_op.py index ae2b2a5c..149a111e 100644 --- a/reme_ai/context/file_tool/batch_write_file_op.py +++ b/reme_ai/context/file_tool/batch_write_file_op.py @@ -20,6 +20,10 @@ class BatchWriteFileOp(BaseAsyncOp): WriteFileOp. Returns a combined result of all write operations. """ + def __init__(self, save_answer: bool = False, **kwargs): + super().__init__(**kwargs) + self.save_answer: bool = save_answer + async def async_execute(self): """Execute the batch write file operation. @@ -37,7 +41,7 @@ class BatchWriteFileOp(BaseAsyncOp): # Process each file in the dictionary result = [] for file_path, content in write_file_dict.items(): - write_op = WriteFileOp() + write_op = WriteFileOp(save_answer=self.save_answer) await write_op.async_call(file_path=file_path, content=content) result.append(write_op.output)