Skip to content

⚡ Bolt: Optimize linear lookups in file extension checking#358

Open
bashandbone wants to merge 1 commit into
mainfrom
bolt/optimize-metadata-linear-lookups-3287533106860394289
Open

⚡ Bolt: Optimize linear lookups in file extension checking#358
bashandbone wants to merge 1 commit into
mainfrom
bolt/optimize-metadata-linear-lookups-3287533106860394289

Conversation

@bashandbone
Copy link
Copy Markdown
Contributor

@bashandbone bashandbone commented May 23, 2026

💡 What: Replaced the next() with generator comprehension pattern used in the is_doc and is_data property methods of ChunkKind (src/codeweaver/core/metadata.py) with standard for loops utilizing early returns. Added # noqa: SIM110 to preserve the optimization from Ruff's simplify rules.

🎯 Why: In Python, creating a generator comprehension inside next() (or any()) incurs overhead because it has to allocate a generator frame. For simple linear lookups over a collection (like iterating through DOC_FILES_EXTENSIONS), a standard for loop with an early return is measurably faster as it bypasses this generator allocation overhead entirely. Since these properties are likely evaluated frequently during file parsing and indexing, this micro-optimization provides a small but consistent speed boost.

📊 Impact: Reduces the execution time of is_doc and is_data extension checks by eliminating generator frame allocation overhead per call, leading to a small improvement in file parsing and metadata resolution speed.

🔬 Measurement: Verify by running the test suite (uv run pytest tests/unit/core/ --no-cov). The tests pass, ensuring no change in functionality. Performance can be benchmarked using timeit comparing the next(...) approach vs the for ... return approach over a sample array.


PR created automatically by Jules for task 3287533106860394289 started by @bashandbone

Summary by Sourcery

Enhancements:

  • Replace generator-based next() usage in ChunkKind.is_doc and ChunkKind.is_data with early-return for loops for faster linear extension lookups.

Replaces `next()` with a generator comprehension with a standard `for` loop and an early return in `is_doc` and `is_data` properties of `ChunkKind` in `src/codeweaver/core/metadata.py`. This optimization bypasses the overhead of generator frame allocation, resulting in faster linear lookups. Added `# noqa: SIM110` to prevent Ruff from reverting the loop back to an `any()` comprehension.

Co-authored-by: bashandbone <89049923+bashandbone@users.noreply.github.com>
@google-labs-jules
Copy link
Copy Markdown
Contributor

👋 Jules, reporting for duty! I'm here to lend a hand with this pull request.

When you start a review, I'll add a 👀 emoji to each comment to let you know I've read it. I'll focus on feedback directed at me and will do my best to stay out of conversations between you and other bots or reviewers to keep the noise down.

I'll push a commit with your requested changes shortly after. Please note there might be a delay between these steps, but rest assured I'm on the job!

For more direct control, you can switch me to Reactive Mode. When this mode is on, I will only act on comments where you specifically mention me with @jules. You can find this option in the Pull Request section of your global Jules UI settings. You can always switch back!

New to Jules? Learn more at jules.google/docs.


For security, I will only act on instructions from the user who triggered this task.

Copilot AI review requested due to automatic review settings May 23, 2026 12:30
@sourcery-ai
Copy link
Copy Markdown
Contributor

sourcery-ai Bot commented May 23, 2026

Reviewer's guide (collapsed on small PRs)

Reviewer's Guide

Optimizes the ChunkKind.is_doc and ChunkKind.is_data property methods by replacing generator-based next() lookups with explicit for-loops that short-circuit on match, and annotates them to avoid Ruff simplification so the micro-optimization is preserved.

Flow diagram for optimized ChunkKind.is_doc and ChunkKind.is_data lookups

flowchart TD
    subgraph is_doc
        A[Call ChunkKind.is_doc] --> B[Iterate DOC_FILES_EXTENSIONS]
        B --> C{doc_ext.ext == self.ext?}
        C -->|yes| D[return True]
        C -->|no| B
        B -->|no more items| E[return False]
    end

    subgraph is_data
        F[Call ChunkKind.is_data] --> G[Iterate DATA_FILES_EXTENSIONS]
        G --> H{data_ext.ext == self.ext?}
        H -->|yes| I[return True]
        H -->|no| G
        G -->|no more items| J[return False]
    end
Loading

File-Level Changes

Change Details Files
Optimize extension membership checks for documentation files.
  • Replace next() with generator comprehension pattern with an explicit for-loop that returns True on first matching doc_ext.ext and False otherwise
  • Add a brief code comment explaining the performance rationale for preferring the for-loop over the generator-based next()
  • Add a Ruff noqa directive (SIM110) on the for-loop to prevent automatic simplification that would negate the optimization
src/codeweaver/core/metadata.py
Optimize extension membership checks for data files.
  • Replace next() with generator comprehension pattern with an explicit for-loop that returns True on first matching data_ext.ext and False otherwise
  • Add a brief code comment explaining the performance rationale for preferring the for-loop over the generator-based next()
  • Add a Ruff noqa directive (SIM110) on the for-loop to prevent automatic simplification that would negate the optimization
src/codeweaver/core/metadata.py

Tips and commands

Interacting with Sourcery

  • Trigger a new review: Comment @sourcery-ai review on the pull request.
  • Continue discussions: Reply directly to Sourcery's review comments.
  • Generate a GitHub issue from a review comment: Ask Sourcery to create an
    issue from a review comment by replying to it. You can also reply to a
    review comment with @sourcery-ai issue to create an issue from it.
  • Generate a pull request title: Write @sourcery-ai anywhere in the pull
    request title to generate a title at any time. You can also comment
    @sourcery-ai title on the pull request to (re-)generate the title at any time.
  • Generate a pull request summary: Write @sourcery-ai summary anywhere in
    the pull request body to generate a PR summary at any time exactly where you
    want it. You can also comment @sourcery-ai summary on the pull request to
    (re-)generate the summary at any time.
  • Generate reviewer's guide: Comment @sourcery-ai guide on the pull
    request to (re-)generate the reviewer's guide at any time.
  • Resolve all Sourcery comments: Comment @sourcery-ai resolve on the
    pull request to resolve all Sourcery comments. Useful if you've already
    addressed all the comments and don't want to see them anymore.
  • Dismiss all Sourcery reviews: Comment @sourcery-ai dismiss on the pull
    request to dismiss all existing Sourcery reviews. Especially useful if you
    want to start fresh with a new review - don't forget to comment
    @sourcery-ai review to trigger a new review!

Customizing Your Experience

Access your dashboard to:

  • Enable or disable review features such as the Sourcery-generated pull request
    summary, the reviewer's guide, and others.
  • Change the review language.
  • Add, remove or edit custom review instructions.
  • Adjust other review settings.

Getting Help

@github-actions
Copy link
Copy Markdown
Contributor

🤖 Hi @bashandbone, I've received your request, and I'm working on it now! You can track my progress in the logs for more details.

@github-actions
Copy link
Copy Markdown
Contributor

🤖 I'm sorry @bashandbone, but I was unable to process your request. Please see the logs for more details.

Copy link
Copy Markdown
Contributor

@sourcery-ai sourcery-ai Bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Hey - I've left some high level feedback:

  • Given that this is a micro-optimization on relatively small collections, consider whether the added complexity (inline performance comments and noqa override) is justified here versus keeping the simpler next(...) or any(...) form for readability and maintainability.
  • If these lookups are truly hot paths, you might get a clearer and more substantial speedup by normalizing the extensions into a set (or using a precomputed lookup structure) rather than repeatedly doing linear scans over the *_FILES_EXTENSIONS sequences.
  • The optimization commentary in the method bodies is quite verbose; consider shortening it or relying on the commit/PR description so the code itself remains focused on intent rather than micro-benchmark details.
Prompt for AI Agents
Please address the comments from this code review:

## Overall Comments
- Given that this is a micro-optimization on relatively small collections, consider whether the added complexity (inline performance comments and noqa override) is justified here versus keeping the simpler `next(...)` or `any(...)` form for readability and maintainability.
- If these lookups are truly hot paths, you might get a clearer and more substantial speedup by normalizing the extensions into a `set` (or using a precomputed lookup structure) rather than repeatedly doing linear scans over the `*_FILES_EXTENSIONS` sequences.
- The optimization commentary in the method bodies is quite verbose; consider shortening it or relying on the commit/PR description so the code itself remains focused on intent rather than micro-benchmark details.

Sourcery is free for open source - if you like our reviews please consider sharing them ✨
Help me be more useful! Please click 👍 or 👎 on each comment and I'll use the feedback to improve your reviews.

Copy link
Copy Markdown
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

This PR optimizes file-extension category checks in ExtLangPair by replacing generator-based next(...) lookups with explicit for loops that early-return, aiming to reduce per-call overhead during frequent metadata resolution.

Changes:

  • Replaced next((True for ...), False) patterns in ExtLangPair.is_doc and ExtLangPair.is_data with early-return for loops.
  • Added inline # noqa: SIM110 to prevent Ruff simplify from rewriting the optimized loops back into generator-based forms.
  • Added explanatory comments documenting the rationale for the loop-based implementation.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants