check if a function is async and add to FTO

[KRRT7]

PR Type

Enhancement, Bug fix, Tests

---

Description

• Add support for async function detection

• Remove has_any_async_functions usage

• Fix coverage utils empty database return

• Update tests for async discovery and code validation

---

Diagram Walkthrough

flowchart LR
  A["Code parsing"] -- "detect `AsyncFunctionDef`" --> B["Function discovery"]
  B -- "flag `is_async`" --> C["FunctionToOptimize"]
  C -- "optimize without blocking on async" --> D["FunctionOptimizer"]
  D -- "generate coverage" --> E["CoverageUtils"]

Loading

File Walkthrough

	Relevant files
Enhancement	4 files code_utils.py Remove unused async detection function                                      +0/-8      static_analysis.py Add AsyncFunctionDef support to lookup functions                  +9/-3      functions_to_optimize.py Enable async function discovery and flagging                          +30/-2    function_optimizer.py Remove async restriction and support async AST                      +1/-7
Bug fix	1 files coverage_utils.py Fix empty DB check order and return `CoverageData`              +3/-3
Tests	3 files test_async_function_discovery.py Add comprehensive async discovery tests                                    +286/-0  test_code_context_extractor.py Resolve symlink paths in code context tests                            +9/-4      test_code_utils.py Replace async detection tests with code validation              +34/-18

---

[github-actions]

PR Reviewer Guide 🔍

Here are some key observations to aid the review process:

⏱️ Estimated effort to review: 4 🔵🔵🔵🔵⚪

🧪 PR contains tests

🔒 No security concerns identified

⚡ Recommended focus areas for review Coverage Empty DB Ensure the fallback to create_empty uses the correct CoverageData.create_empty method and that CoverageData is properly imported. Confirm the exists/stat check avoids errors when the database file is missing. if not database_path.exists() or not database_path.stat().st_size: logger.debug(f"Coverage database {database_path} is empty or does not exist") sentry_sdk.capture_message(f"Coverage database {database_path} is empty or does not exist") return CoverageData.create_empty(source_code_path, function_name, code_context) Async metadata consistency The AsyncFunctionDef visitor appends functions without setting starting_line and ending_line, and uses a different parent list than the sync FunctionDef visitor. Ensure metadata consistency for async functions. def visit_AsyncFunctionDef(self, node: AsyncFunctionDef) -> None: # Check if the async function has a return statement and add it to the list if function_has_return_statement(node) and not function_is_a_property(node): self.functions.append( FunctionToOptimize( function_name=node.name, file_path=self.file_path, parents=self.ast_path[:], is_async=True ) ) AST return type get_first_top_level_function_or_method_ast now returns both FunctionDef and AsyncFunctionDef. Verify downstream callers and type hints handle the async case correctly. ) -> ast.FunctionDef | ast.AsyncFunctionDef | None: if not parents: result = get_first_top_level_object_def_ast(function_name, ast.FunctionDef, node) if result is not None: return result return get_first_top_level_object_def_ast(function_name, ast.AsyncFunctionDef, node)

[github-actions]

PR Code Suggestions ✨

No code suggestions found for the PR.

[codeflash-ai]

⚡️ Codeflash found optimizations for this PR

📄 17% (0.17x) speedup for get_first_top_level_function_or_method_ast in codeflash/code_utils/static_analysis.py

⏱️ Runtime : 390 microseconds → 334 microseconds (best of 59 runs)

I created a new dependent PR with the suggested changes. Please review:

• ⚡️ Speed up function get_first_top_level_function_or_method_ast by 17% in PR #678 (standalone-fto-async) #702

If you approve, it will be merged into this PR (branch standalone-fto-async).