added separate write route, changed return format for api route

Author: dasarchan

codeflash/api/cfapi.py

@@ -209,3 +209,17 @@ def is_function_being_optimized_again(owner: str, repo: str, pr_number: int, cod
     )
     response.raise_for_status()
     return response.json()
+
+def add_code_context_hash(owner: str, repo: str, pr_number: int, code_context_hash: str) -> Response:
+    """Add code context to the DB cache"""
+    response = make_cfapi_request(
+        "/add-code-hash",
+        "POST",
+        {
+            "owner": owner,
+            "repo": repo,
+            "pr_number": pr_number,
+            "code_context_hash": code_context_hash
+        }
+    )
+    return response

codeflash/discovery/functions_to_optimize.py

@@ -532,6 +532,17 @@ def filter_functions(
     disable_logs: bool = False,  # noqa: FBT001, FBT002
 ) -> tuple[dict[Path, list[FunctionToOptimize]], int]:
     blocklist_funcs = get_blocklisted_functions()
+    already_optimized_count = 0
+    path_based_functions = {Path(k): v for k, v in modified_functions.items() if v}
+    try:
+        repository = git.Repo(Path.cwd(), search_parent_directories=True)
+        owner, repo = get_repo_owner_and_name(repository)
+    except git.exc.InvalidGitRepositoryError:
+        logger.warning("No git repository found")
+        owner, repo = None, None
+    pr_number = get_pr_number()
+    if owner and repo and pr_number is not None:
+        path_based_functions, functions_count = check_optimization_status(path_based_functions, owner, repo, pr_number)
     logger.debug(f"Blocklisted functions: {blocklist_funcs}")
     # Remove any function that we don't want to optimize
 
@@ -606,19 +617,10 @@ def filter_functions(
         functions_count += len(_functions)
 
     # Convert to Path keys for optimization check
-    path_based_functions = {Path(k): v for k, v in filtered_modified_functions.items() if v}
+
 
     # Check optimization status if repository info is provided
-    already_optimized_count = 0
-    try:
-        repository = git.Repo(Path.cwd(), search_parent_directories=True)
-        owner, repo = get_repo_owner_and_name(repository)
-    except git.exc.InvalidGitRepositoryError:
-        logger.warning("No git repository found")
-        owner, repo = None, None
-    pr_number = get_pr_number()
-    if owner and repo and pr_number is not None:
-        path_based_functions, functions_count = check_optimization_status(path_based_functions, owner, repo, pr_number)
+
         initial_count = sum(len(funcs) for funcs in filtered_modified_functions.values())
         already_optimized_count = initial_count - functions_count
 

codeflash/optimization/function_optimizer.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import ast
+import git
 import concurrent.futures
 import os
 import subprocess
@@ -18,6 +19,7 @@
 from rich.tree import Tree
 
 from codeflash.api.aiservice import AiServiceClient, LocalAiServiceClient
+from codeflash.api.cfapi import add_code_context_hash
 from codeflash.benchmarking.utils import process_benchmark_data
 from codeflash.cli_cmds.console import code_print, console, logger, progress_bar
 from codeflash.code_utils import env_utils
@@ -50,6 +52,8 @@
 from codeflash.code_utils.line_profile_utils import add_decorator_imports
 from codeflash.code_utils.static_analysis import get_first_top_level_function_or_method_ast
 from codeflash.code_utils.time_utils import humanize_runtime
+from codeflash.code_utils.env_utils import get_pr_number
+from codeflash.code_utils.git_utils import get_repo_owner_and_name
 from codeflash.context import code_context_extractor
 from codeflash.context.unused_definition_remover import detect_unused_helper_functions, revert_unused_helper_functions
 from codeflash.either import Failure, Success, is_successful
@@ -370,6 +374,19 @@ def optimize_function(self) -> Result[BestOptimization, str]:  # noqa: PLR0911
                         )
                 self.log_successful_optimization(explanation, generated_tests, exp_type)
 
+        # Add function to code context hash if in gh actions
+        try:
+            repository = git.Repo(Path.cwd(), search_parent_directories=True)
+            owner, repo = get_repo_owner_and_name(repository)
+        except git.exc.InvalidGitRepositoryError:
+            logger.warning("No git repository found")
+            owner, repo = None, None
+        pr_number = get_pr_number()
+
+        if owner and repo and pr_number is not None:
+            code_context_hash = self.function_to_optimize.get_code_context_hash()
+            add_code_context_hash(owner, repo, pr_number, code_context_hash)
+
         if self.args.override_fixtures:
             restore_conftest(original_conftest_content)
         if not best_optimization: