github/OpenHands
1
0
Fork 0
mirror of https://github.com/OpenHands/OpenHands.git synced 2026-03-22 13:47:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(llm): support draft editor retries by adding correct_num to LLMConfig (#11530)

Browse Source 
Co-authored-by: openhands <openhands@all-hands.dev>
Co-authored-by: Justin Coffi <jcoffi+github@gmail.com>
This commit is contained in:
Ray Myers
2025-10-27 20:02:50 -05:00
committed by GitHub
parent 4decd8b3e9
commit 450aa3b527
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
openhands/core/config/llm_config.py
View File

@@ -95,6 +95,8 @@ class LLMConfig(BaseModel):
description='Safety settings for models that support them (like Mistral AI and Gemini)',
)
for_routing: bool = Field(default=False)
# The number of correction attempts for the LLM draft editor
correct_num: int = Field(default=5)
completion_kwargs: dict[str, Any] | None = Field(
default=None,
description='Custom kwargs to pass to litellm.completion',

4
openhands/runtime/utils/edit.py
View File

@@ -400,8 +400,8 @@ class FileEditRuntimeMixin(FileEditRuntimeInterface):
ret_obs.llm_metrics = self.draft_editor_llm.metrics
return ret_obs
def check_retry_num(self, retry_num):
correct_num = self.draft_editor_llm.config.correct_num # type: ignore[attr-defined]
def check_retry_num(self, retry_num: int) -> bool:
correct_num = self.draft_editor_llm.config.correct_num
return correct_num < retry_num
def correct_edit(