From 9c37f963b0a9f0cd8c60a1beccea3ccb061da508 Mon Sep 17 00:00:00 2001 From: HeshamHM28 Date: Fri, 15 Aug 2025 21:16:14 +0300 Subject: [PATCH 1/3] log the staging link --- codeflash/optimization/function_optimizer.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/codeflash/optimization/function_optimizer.py b/codeflash/optimization/function_optimizer.py index 320fc1d77..7b8fb28b9 100644 --- a/codeflash/optimization/function_optimizer.py +++ b/codeflash/optimization/function_optimizer.py @@ -1247,7 +1247,13 @@ def process_review( data["git_remote"] = self.args.git_remote check_create_pr(**data) elif self.args.staging_review: - create_staging(**data) + response = create_staging(**data) + if response.status_code == 200: + staging_url = f"https://app.codeflash.ai/review-optimizations/{self.function_trace_id[:-4] + exp_type if self.experiment_id else self.function_trace_id}" + console.print(f"[bold green]✅ Staging created:[/bold green] [link={staging_url}]{staging_url}[/link]") + else: + console.print(f"[bold red]❌ Failed to create staging [/bold red] — status {response.status_code}") + else: # Mark optimization success since no PR will be created mark_optimization_success( From 2b486dae42f7c33626389cec0ff1cbda61f1880d Mon Sep 17 00:00:00 2001 From: HeshamHM28 Date: Mon, 18 Aug 2025 18:55:06 +0300 Subject: [PATCH 2/3] add panel --- codeflash/optimization/function_optimizer.py | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/codeflash/optimization/function_optimizer.py b/codeflash/optimization/function_optimizer.py index 7b8fb28b9..74222dfdf 100644 --- a/codeflash/optimization/function_optimizer.py +++ b/codeflash/optimization/function_optimizer.py @@ -1250,9 +1250,21 @@ def process_review( response = create_staging(**data) if response.status_code == 200: staging_url = f"https://app.codeflash.ai/review-optimizations/{self.function_trace_id[:-4] + exp_type if self.experiment_id else self.function_trace_id}" - console.print(f"[bold green]✅ Staging created:[/bold green] [link={staging_url}]{staging_url}[/link]") + console.print( + Panel( + f"[bold green]✅ Staging created:[/bold green]\n[link={staging_url}]{staging_url}[/link]", + title="Staging Link", + border_style="green" + ) + ) else: - console.print(f"[bold red]❌ Failed to create staging [/bold red] — status {response.status_code}") + console.print( + Panel( + f"[bold red]❌ Failed to create staging[/bold red]\nStatus: {response.status_code}", + title="Staging Error", + border_style="red" + ) + ) else: # Mark optimization success since no PR will be created From e06405993be66481aa94f1f945c9856d4c315ddf Mon Sep 17 00:00:00 2001 From: HeshamHM28 Date: Mon, 18 Aug 2025 19:01:16 +0300 Subject: [PATCH 3/3] formate the file --- codeflash/optimization/function_optimizer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/codeflash/optimization/function_optimizer.py b/codeflash/optimization/function_optimizer.py index b3898d26c..8aebbac64 100644 --- a/codeflash/optimization/function_optimizer.py +++ b/codeflash/optimization/function_optimizer.py @@ -1260,7 +1260,7 @@ def process_review( Panel( f"[bold green]✅ Staging created:[/bold green]\n[link={staging_url}]{staging_url}[/link]", title="Staging Link", - border_style="green" + border_style="green", ) ) else: @@ -1268,7 +1268,7 @@ def process_review( Panel( f"[bold red]❌ Failed to create staging[/bold red]\nStatus: {response.status_code}", title="Staging Error", - border_style="red" + border_style="red", ) )