⚡️ Speed up function _top_level by 13% in PR #9539 (docs-openai-api-endpoint)

[codeflash-ai]

⚡️ This pull request contains optimizations for PR #9539

If you approve this dependent PR, these changes will be merged into the original PR branch docs-openai-api-endpoint.

This PR will be automatically closed if the original PR is merged.

---

📄 13% (0.13x) speedup for _top_level in langflow/custom/dependency_analyzer.py

⏱️ Runtime : 298 microseconds → 263 microseconds (best of 137 runs)

📝 Explanation and details

The optimization replaces pkg.split(".", 1)[0] with pkg.partition('.')[0]. This change delivers a 13% speedup because:

Key Optimization:

• str.partition() is specifically designed for splitting a string at the first occurrence of a separator, returning a 3-tuple (before, separator, after)
• str.split(".", 1) creates an intermediate list object before indexing, while partition() directly returns a tuple
• Tuple access is faster than list access in CPython, and no intermediate list allocation occurs

Performance Impact:

• Line profiler shows per-hit time improved from 681.5ns to 613.3ns (10% per-call improvement)
• The optimization is most effective for high-frequency calls, as evidenced by the 1060 hits in the profiler

Test Case Performance:

• Works equally well across all test scenarios - basic package names, edge cases with dots/spaces, and large-scale tests with long strings
• Maintains identical behavior for all edge cases (empty strings, leading dots, unicode characters)
• The optimization benefits any workload that processes many package names, regardless of string length or complexity

This is a classic example of choosing the right string method for the task - partition() is purpose-built for "split at first occurrence" operations and avoids the overhead of general-purpose split().

✅ Correctness verification report:

Test	Status
⏪ Replay Tests	🔘 None Found
⚙️ Existing Unit Tests	✅ 24 Passed
🔎 Concolic Coverage Tests	🔘 None Found
🌀 Generated Regression Tests	✅ 1057 Passed
📊 Tests Coverage	100.0%

⚙️ Existing Unit Tests and Runtime

🌀 Generated Regression Tests and Runtime

from __future__ import annotations

# imports
import pytest  # used for our unit tests
from langflow.custom.dependency_analyzer import _top_level

# unit tests

# -------------------------
# Basic Test Cases
# -------------------------

def test_basic_single_name():
    # Single package name, no dot
    codeflash_output = _top_level("requests")

def test_basic_simple_dotted_name():
    # Dotted package name, should return first part
    codeflash_output = _top_level("os.path")

def test_basic_multiple_dots():
    # Multiple dots, should return first part
    codeflash_output = _top_level("a.b.c.d")

def test_basic_leading_space():
    # Leading space should be included as part of the name
    codeflash_output = _top_level("  foo.bar")

def test_basic_trailing_space():
    # Trailing space after first part should not affect result
    codeflash_output = _top_level("foo .bar")

def test_basic_empty_string():
    # Empty string, should return empty string
    codeflash_output = _top_level("")

# -------------------------
# Edge Test Cases
# -------------------------

def test_edge_leading_dot():
    # Leading dot, should return empty string before first dot
    codeflash_output = _top_level(".foo.bar")

def test_edge_trailing_dot():
    # Trailing dot, should return all before first dot
    codeflash_output = _top_level("foo.")

def test_edge_only_dot():
    # String is just a dot, should return empty string before first dot
    codeflash_output = _top_level(".")

def test_edge_multiple_consecutive_dots():
    # Multiple consecutive dots, should split at first dot
    codeflash_output = _top_level("foo..bar")

def test_edge_dot_at_start_and_end():
    # Dot at start and end, should return empty string before first dot
    codeflash_output = _top_level(".foo.")

def test_edge_double_dot_at_start():
    # Double dot at start, should return empty string before first dot
    codeflash_output = _top_level("..foo")

def test_edge_unicode_characters():
    # Unicode in package name
    codeflash_output = _top_level("naïve.module")
    codeflash_output = _top_level("模块.子模块")

def test_edge_numeric_package_name():
    # Numeric package name, should return as string
    codeflash_output = _top_level("123.abc")

def test_edge_special_characters():
    # Special characters in package name
    codeflash_output = _top_level("foo-bar.baz")
    codeflash_output = _top_level("foo_bar.baz")
    codeflash_output = _top_level("foo$bar.baz")

def test_edge_whitespace_only():
    # String with only whitespace, should return as is
    codeflash_output = _top_level("   ")

def test_edge_dot_in_middle_of_spaces():
    # Dot surrounded by spaces
    codeflash_output = _top_level("  .  ")

def test_edge_long_package_name_no_dot():
    # Long package name, no dot
    name = "a" * 100
    codeflash_output = _top_level(name)

def test_edge_long_package_name_with_dot():
    # Long package name with dot
    name = "a" * 100 + ".b"
    codeflash_output = _top_level(name)

# -------------------------
# Large Scale Test Cases
# -------------------------

def test_large_scale_many_dots():
    # Very long package name with many dots
    pkg = ".".join(["pkg"] + [f"sub{i}" for i in range(999)])
    codeflash_output = _top_level(pkg)

def test_large_scale_long_first_part():
    # Very long first part, then dot
    first = "x" * 999
    pkg = first + ".submodule"
    codeflash_output = _top_level(pkg)

def test_large_scale_long_second_part():
    # Short first part, very long second part
    second = "y" * 999
    pkg = "foo." + second
    codeflash_output = _top_level(pkg)

def test_large_scale_all_dots():
    # String of only dots, should return empty string before first dot
    pkg = "." * 999
    codeflash_output = _top_level(pkg)

def test_large_scale_many_packages():
    # Test many different package names in a loop (under 1000)
    for i in range(1, 1001):
        pkg = f"pkg{i}.sub{i}"
        codeflash_output = _top_level(pkg)

def test_large_scale_no_dot_large_name():
    # Large name without dot
    pkg = "a" * 1000
    codeflash_output = _top_level(pkg)

def test_large_scale_dot_at_end_of_large_name():
    # Large name with dot at end
    pkg = "a" * 999 + "."
    codeflash_output = _top_level(pkg)
# codeflash_output is used to check that the output of the original code is the same as that of the optimized code.
#------------------------------------------------
from __future__ import annotations

# imports
import pytest  # used for our unit tests
from langflow.custom.dependency_analyzer import _top_level

# unit tests

# 1. Basic Test Cases

def test_basic_single_name():
    # Test with a single package name, no dots
    codeflash_output = _top_level("requests")

def test_basic_simple_module():
    # Test with a simple module path
    codeflash_output = _top_level("os.path")

def test_basic_deep_module():
    # Test with a deeper module path
    codeflash_output = _top_level("numpy.linalg.eig")

def test_basic_leading_and_trailing_spaces():
    # Test with leading and trailing spaces (should not strip)
    codeflash_output = _top_level("  pandas.dataframe")

def test_basic_dot_in_middle():
    # Test with a dot in the middle
    codeflash_output = _top_level("abc.def")

# 2. Edge Test Cases

def test_edge_empty_string():
    # Test with empty string input
    codeflash_output = _top_level("")

def test_edge_only_dot():
    # Test with only a dot
    codeflash_output = _top_level(".")

def test_edge_starts_with_dot():
    # Test with string starting with a dot
    codeflash_output = _top_level(".module")

def test_edge_ends_with_dot():
    # Test with string ending with a dot
    codeflash_output = _top_level("module.")

def test_edge_multiple_dots():
    # Test with multiple consecutive dots
    codeflash_output = _top_level("a..b.c")

def test_edge_only_dots():
    # Test with string of only dots
    codeflash_output = _top_level("...")

def test_edge_unicode_characters():
    # Test with unicode characters in package name
    codeflash_output = _top_level("naïve.module")

def test_edge_numeric_package():
    # Test with numeric package name
    codeflash_output = _top_level("123.abc")

def test_edge_dot_and_space():
    # Test with dot and space in the string
    codeflash_output = _top_level("abc. def")

def test_edge_leading_spaces():
    # Test with leading spaces before package name
    codeflash_output = _top_level("   abc.def")

def test_edge_trailing_spaces():
    # Test with trailing spaces after package name
    codeflash_output = _top_level("abc   .def")

def test_edge_long_package_name():
    # Test with a very long package name
    long_name = "a" * 100 + ".b.c"
    codeflash_output = _top_level(long_name)

def test_edge_dot_at_end():
    # Test with dot at end of package name
    codeflash_output = _top_level("abc.")

def test_edge_dot_at_start():
    # Test with dot at start of package name
    codeflash_output = _top_level(".abc")

def test_edge_multiple_leading_dots():
    # Test with multiple leading dots
    codeflash_output = _top_level("..abc.def")

def test_edge_multiple_trailing_dots():
    # Test with multiple trailing dots
    codeflash_output = _top_level("abc..def")

def test_edge_only_one_char():
    # Test with single character package name
    codeflash_output = _top_level("a")

def test_edge_dot_and_empty():
    # Test with dot and empty after dot
    codeflash_output = _top_level("abc.")

# 3. Large Scale Test Cases

def test_large_scale_long_module_path():
    # Test with a very long dotted module path (999 submodules)
    long_path = "pkg" + ".".join(str(i) for i in range(1, 1000))
    codeflash_output = _top_level(long_path)  # because "pkg1.2.3..." is the result

def test_large_scale_long_top_level_name():
    # Test with a very long top-level package name
    top_name = "a" * 999
    path = top_name + ".submodule"
    codeflash_output = _top_level(path)

def test_large_scale_many_dots():
    # Test with a string of many dots (999)
    dots = "." * 999
    codeflash_output = _top_level(dots)

def test_large_scale_repeated_pattern():
    # Test with a repeated pattern in top-level name
    pattern = "xy" * 499  # 998 chars
    path = pattern + ".mod"
    codeflash_output = _top_level(path)

def test_large_scale_package_name_with_numbers():
    # Test with large package name containing numbers
    pkg = "pkg1234567890" * 90  # 990 chars
    path = pkg + ".mod"
    codeflash_output = _top_level(path)

def test_large_scale_long_module_path_with_spaces():
    # Test with long module path including spaces
    path = "pkg " + ".".join(["mod" for _ in range(999)])
    codeflash_output = _top_level(path)
# codeflash_output is used to check that the output of the original code is the same as that of the optimized code.

To edit these changes git checkout codeflash/optimize-pr9539-2025-08-26T16.40.35 and push.

[coderabbitai]

Important

Review skipped

Bot user detected.

To trigger a single review, invoke the @coderabbitai review command.

You can disable this status message by setting the reviews.review_status to false in the CodeRabbit configuration file.

---

🪧 Tips

Chat

There are 3 ways to chat with CodeRabbit:

• Review comments: Directly reply to a review comment made by CodeRabbit. Example:
  • I pushed a fix in commit <commit_id>, please review it.
  • Open a follow-up GitHub issue for this discussion.
• Files and specific lines of code (under the "Files changed" tab): Tag @coderabbitai in a new review comment at the desired location with your query.
• PR comments: Tag @coderabbitai in a new PR comment to ask questions about the PR branch. For the best results, please provide a very specific query, as very limited context is provided in this mode. Examples:
  • @coderabbitai gather interesting stats about this repository and render them as a table. Additionally, render a pie chart showing the language distribution in the codebase.
  • @coderabbitai read the files in the src/scheduler package and generate a class diagram using mermaid and a README in the markdown format.

Support

Need help? Join our Discord community for assistance with any issues or questions.

CodeRabbit Commands (Invoked using PR/Issue comments)

Type @coderabbitai help to get the list of available commands.

Other keywords and placeholders

• Add @coderabbitai ignore anywhere in the PR description to prevent this PR from being reviewed.
• Add @coderabbitai summary to generate the high-level summary at a specific location in the PR description.
• Add @coderabbitai anywhere in the PR title to generate the title automatically.

Status, Documentation and Community

• Visit our Status Page to check the current availability of CodeRabbit.
• Visit our Documentation for detailed information on how to use CodeRabbit.
• Join our Discord Community to get help, request features, and share feedback.
• Follow us on X/Twitter for updates and announcements.

[sonarqubecloud]

Quality Gate passed

Issues
0 New issues
0 Accepted issues

Measures
0 Security Hotspots
0.0% Coverage on New Code
0.0% Duplication on New Code

See analysis details on SonarQube Cloud

[codeflash-ai]

This PR has been automatically closed because the original PR #9539 by mendonk was closed.