Skip to content

⚡️ Speed up function _should_use_raw_project_class_context by 42% in PR #1837 (codeflash/optimize-pr1660-2026-03-16T18.33.59) #1840

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
KRRT7 merged 2 commits into from
Mar 16, 2026
Merged

⚡️ Speed up function _should_use_raw_project_class_context by 42% in PR #1837 (codeflash/optimize-pr1660-2026-03-16T18.33.59) #1840

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
63ff575
Optimize _should_use_raw_project_class_context
codeflash-ai[bot] Mar 16, 2026
f7d0e79
style: fix trailing whitespace on blank lines (W293)
github-actions[bot] Mar 16, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
36 changes: 22 additions & 14 deletions codeflash/languages/python/context/code_context_extractor.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -918,29 +918,37 @@ def _has_descriptor_like_class_fields(class_node: ast.ClassDef) -> bool:


def _should_use_raw_project_class_context(class_node: ast.ClassDef, import_aliases: dict[str, str]) -> bool:
# Check decorator presence first - cheapest check that can short-circuit
if class_node.decorator_list:
return True

# Check for namedtuple/dataclass early - these are common patterns that avoid body scanning
if _is_namedtuple_class(class_node, import_aliases):
return True
is_dataclass, _, _ = _get_dataclass_config(class_node, import_aliases)
if is_dataclass:
return True

# Calculate size metrics once
start_line = _get_class_start_line(class_node)
assert class_node.end_lineno is not None
class_line_count = class_node.end_lineno - start_line + 1
is_small = (
class_line_count <= MAX_RAW_PROJECT_CLASS_LINES and len(class_node.body) <= MAX_RAW_PROJECT_CLASS_BODY_ITEMS
)

if is_small and _class_has_explicit_init(class_node):
return True
if _is_namedtuple_class(class_node, import_aliases):
return True
is_dataclass, _, _ = _get_dataclass_config(class_node, import_aliases)
if is_dataclass:
return True
if class_node.decorator_list:
return True
if _has_descriptor_like_class_fields(class_node):
return True
# Single-pass body scan with early returns
has_explicit_init = False

for item in class_node.body:
if isinstance(item, (ast.FunctionDef, ast.AsyncFunctionDef)) and _has_non_property_method_decorator(
item, import_aliases
):
if isinstance(item, (ast.FunctionDef, ast.AsyncFunctionDef)):
if item.name == "__init__":
has_explicit_init = True
if is_small:
return True
if _has_non_property_method_decorator(item, import_aliases):
return True
elif isinstance(item, (ast.Assign, ast.AnnAssign)) and isinstance(item.value, ast.Call):
return True

return False
Expand Down
Loading