diff --git a/codeflash/api/cfapi.py b/codeflash/api/cfapi.py index d410f75dd..085806023 100644 --- a/codeflash/api/cfapi.py +++ b/codeflash/api/cfapi.py @@ -28,10 +28,12 @@ if os.environ.get("CODEFLASH_CFAPI_SERVER", "prod").lower() == "local": CFAPI_BASE_URL = "http://localhost:3001" + CFWEBAPP_BASE_URL = "http://localhost:3000" logger.info(f"Using local CF API at {CFAPI_BASE_URL}.") console.rule() else: CFAPI_BASE_URL = "https://app.codeflash.ai" + CFWEBAPP_BASE_URL = "https://app.codeflash.ai" def make_cfapi_request( diff --git a/codeflash/optimization/function_optimizer.py b/codeflash/optimization/function_optimizer.py index eb61a689b..27763ff23 100644 --- a/codeflash/optimization/function_optimizer.py +++ b/codeflash/optimization/function_optimizer.py @@ -19,7 +19,7 @@ from rich.tree import Tree from codeflash.api.aiservice import AiServiceClient, AIServiceRefinerRequest, LocalAiServiceClient -from codeflash.api.cfapi import CFAPI_BASE_URL, add_code_context_hash, create_staging, mark_optimization_success +from codeflash.api.cfapi import CFWEBAPP_BASE_URL, add_code_context_hash, create_staging, mark_optimization_success from codeflash.benchmarking.utils import process_benchmark_data from codeflash.cli_cmds.console import code_print, console, logger, lsp_log, progress_bar from codeflash.code_utils import env_utils @@ -1482,7 +1482,8 @@ def process_review( elif staging_review: response = create_staging(**data) if response.status_code == 200: - staging_url = f"{CFAPI_BASE_URL}/review-optimizations/{self.function_trace_id[:-4] + exp_type if self.experiment_id else self.function_trace_id}" + trace_id = self.function_trace_id[:-4] + exp_type if self.experiment_id else self.function_trace_id + staging_url = f"{CFWEBAPP_BASE_URL}/review-optimizations/{trace_id}" console.print( Panel( f"[bold green]✅ Staging created:[/bold green]\n[link={staging_url}]{staging_url}[/link]",