Skip to content

Commit 7eab405

Browse files
committed
Minor prompt and comment changes
1 parent 1016109 commit 7eab405

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

agent/repairer.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,8 +39,8 @@ def _initial_prompt(self, results: list[Result]) -> Prompt:
3939
)
4040
prompt = prompt_builder.build(example_pair=[])
4141
self.protocol = (
42-
'Use the run_bash_command_or_submit_revised_fuzz_target_and_build_script tool to inspect code or the submit the fuzz target and '
43-
'build script')
42+
'Use the inspect_code_or_conclude_fuzz_target tool to inspect code or '
43+
'the submit the fuzz target and build script')
4444
return prompt
4545

4646
def _update_fuzz_target_and_build_script(self, cur_round: int,
@@ -232,6 +232,7 @@ def invalid_function_usage(self, cur_round: int, call: FunctionCall) -> Part:
232232
def call_function(self, cur_round: int, call: FunctionCall,
233233
build_result: BuildResult,
234234
results: list[Part]) -> Optional[Part]:
235+
"""Calls tool functions based on LLM response."""
235236
args = dict(call.args)
236237
if call.name != 'inspect_code_or_conclude_fuzz_target':
237238
return Part.from_function_response(

0 commit comments

Comments
 (0)