Skip to content

Commit 329e7b6

Browse files
committed
feat: send coverage info to test review endpoint
Include coverage percentage in the review request so the AI reviewer can flag tests with low coverage that don't exercise the function.
1 parent 49b2623 commit 329e7b6

2 files changed

Lines changed: 9 additions & 1 deletion

File tree

codeflash/api/aiservice.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -775,9 +775,10 @@ def review_generated_tests(
775775
function_source_code: str,
776776
function_name: str,
777777
trace_id: str,
778+
coverage_summary: str = "",
778779
language: str = "python",
779780
) -> list[TestFileReview]:
780-
payload = {
781+
payload: dict[str, Any] = {
781782
"tests": tests,
782783
"function_source_code": function_source_code,
783784
"function_name": function_name,
@@ -786,6 +787,8 @@ def review_generated_tests(
786787
"codeflash_version": codeflash_version,
787788
"call_sequence": self.get_next_sequence(),
788789
}
790+
if coverage_summary:
791+
payload["coverage_summary"] = coverage_summary
789792
try:
790793
response = self.make_ai_service_request("/testgen_review", payload=payload, timeout=self.timeout)
791794
except requests.exceptions.RequestException as e:

codeflash/optimization/function_optimizer.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2030,13 +2030,18 @@ def review_and_repair_tests(
20302030
}
20312031
)
20322032

2033+
coverage_summary = ""
2034+
if coverage_results and coverage_results.coverage is not None:
2035+
coverage_summary = f"{coverage_results.coverage:.1f}%"
2036+
20332037
console.rule()
20342038
with progress_bar("Reviewing generated tests for quality issues..."):
20352039
review_results = self.aiservice_client.review_generated_tests(
20362040
tests=tests_for_review,
20372041
function_source_code=self.function_to_optimize_source_code,
20382042
function_name=self.function_to_optimize.function_name,
20392043
trace_id=self.function_trace_id,
2044+
coverage_summary=coverage_summary,
20402045
language=self.function_to_optimize.language,
20412046
)
20422047

0 commit comments

Comments
 (0)