Skip to content

Commit 7e17f79

Browse files
committed
saurabh suggested fixes
1 parent 49cab09 commit 7e17f79

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

codeflash/api/aiservice.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -328,15 +328,15 @@ def get_new_explanation( # noqa: D417
328328
logger.info("Generating explanation")
329329
console.rule()
330330
try:
331-
response = self.make_ai_service_request("/explain", payload=payload, timeout=600)
331+
response = self.make_ai_service_request("/explain", payload=payload, timeout=60)
332332
except requests.exceptions.RequestException as e:
333333
logger.exception(f"Error generating explanations: {e}")
334334
ph("cli-optimize-error-caught", {"error": str(e)})
335335
return ""
336336

337337
if response.status_code == 200:
338338
explanation: str = response.json()["explanation"]
339-
logger.info(f"New Explanation: {explanation}")
339+
logger.debug(f"New Explanation: {explanation}")
340340
console.rule()
341341
return explanation
342342
try:

codeflash/optimization/function_optimizer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1108,7 +1108,7 @@ def find_and_process_best_optimization(
11081108
generated_tests = add_runtime_comments_to_generated_tests(
11091109
generated_tests, original_runtime_by_test, optimized_runtime_by_test
11101110
)
1111-
generated_tests_str = "\n\n".join(
1111+
generated_tests_str = "\n#------------------------------------------------\n".join(
11121112
[test.generated_original_test_source for test in generated_tests.generated_tests]
11131113
)
11141114
existing_tests = existing_tests_source_for(
@@ -1119,7 +1119,7 @@ def find_and_process_best_optimization(
11191119
optimized_runtimes_all=optimized_runtime_by_test,
11201120
)
11211121
if concolic_test_str:
1122-
generated_tests_str += "\n\n" + concolic_test_str
1122+
generated_tests_str += "\n#------------------------------------------------\n" + concolic_test_str
11231123
new_explanation_raw_str = self.aiservice_client.get_new_explanation(
11241124
source_code=code_context.read_writable_code,
11251125
dependency_code=code_context.read_only_context_code,

0 commit comments

Comments
 (0)