Skip to content

Commit e0ba10a

Browse files
committed
Presubmit fixes
1 parent 8cf9129 commit e0ba10a

File tree

2 files changed

+4
-6
lines changed

2 files changed

+4
-6
lines changed

agent/coverage_analyzer.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -49,11 +49,9 @@ def _initial_prompt(self, results: list[Result]) -> Prompt:
4949
function_requirements = self.get_function_requirements()
5050

5151
builder = CoverageAnalyzerTemplateBuilder(self.llm, benchmark, last_result)
52+
tool_guides = f'{self.inspect_tool.tutorial()}\n\n{self.coverage_tool.tutorial()}'
5253
prompt = builder.build(example_pair=[],
53-
tool_guides=[
54-
self.inspect_tool.tutorial(),
55-
self.coverage_tool.tutorial()
56-
],
54+
tool_guides=tool_guides,
5755
project_dir=self.inspect_tool.project_dir,
5856
function_requirements=function_requirements)
5957
# TODO: A different file name/dir.

llm_toolkit/prompt_builder.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -691,7 +691,7 @@ def build(self,
691691
example_pair: list[list[str]],
692692
project_example_content: Optional[list[list[str]]] = None,
693693
project_context_content: Optional[dict] = None,
694-
tool_guides: list[str] = [],
694+
tool_guides: str = '',
695695
project_dir: str = '',
696696
function_requirements: str = '') -> prompts.Prompt:
697697
"""Constructs a prompt using the templates in |self| and saves it."""
@@ -707,7 +707,7 @@ def build(self,
707707
prompt = prompt.replace('{FUNCTION_SIGNATURE}',
708708
self.benchmark.function_signature)
709709
prompt = prompt.replace('{FUZZ_TARGET}', self.run_result.fuzz_target_source)
710-
prompt = prompt.replace('{TOOL_GUIDES}', '\n'.join(tool_guides))
710+
prompt = prompt.replace('{TOOL_GUIDES}', tool_guides)
711711
prompt = prompt.replace('{FUZZING_LOG}', self.run_result.run_log)
712712
prompt = prompt.replace('{FUNCTION_REQUIREMENTS}', function_requirements)
713713

0 commit comments

Comments
 (0)