Skip to content

Commit 5a47237

Browse files
committed
Update aiservice.py
1 parent 745a3f7 commit 5a47237

File tree

1 file changed

+3
-8
lines changed

1 file changed

+3
-8
lines changed

codeflash/api/aiservice.py

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -439,16 +439,11 @@ def generate_regression_tests( # noqa: D417
439439
"test_index": test_index,
440440
"python_version": platform.python_version(),
441441
"codeflash_version": codeflash_version,
442-
"is_async": bool(getattr(function_to_optimize, "is_async", False)),
442+
"is_async": function_to_optimize.is_async,
443443
}
444-
445-
endpoint = "/testgen"
446-
logger.debug(
447-
f"Using unified test generation endpoint for function {function_to_optimize.function_name} (is_async={payload['is_async']})"
448-
)
449-
444+
450445
try:
451-
response = self.make_ai_service_request(endpoint, payload=payload, timeout=600)
446+
response = self.make_ai_service_request("/testgen", payload=payload, timeout=600)
452447
except requests.exceptions.RequestException as e:
453448
logger.exception(f"Error generating tests: {e}")
454449
ph("cli-testgen-error-caught", {"error": str(e)})

0 commit comments

Comments
 (0)