⚡️ Speed up function build_valid_menu by 332% in src/backend/base/langflow/custom/directory_reader/utils.py

[codeflash-ai]

📄 build_valid_menu in src/backend/base/langflow/custom/directory_reader/utils.py

✨ Performance Summary:

• Speed Increase: 📈 332% (3.32x faster)
• Runtime Reduction: ⏱️ From 793 milliseconds down to 184 milliseconds (best of 5 runs)

---

📝 Explanation and details

Here's the optimized version of the program.

These optimizations should help the program to run faster, especially for larger datasets.

---

✅ Correctness verification

The new optimized code was tested for correctness. The results are listed below:

Test	Status	Details
⚙️ Existing Unit Tests	🔘 None Found
🌀 Generated Regression Tests	✅ 27 Passed	See below
⏪ Replay Tests	🔘 None Found
🔎 Concolic Coverage Tests	🔘 None Found
📊 Coverage	86.7%

🌀 Generated Regression Tests Details

Click to view details

import pytest  # used for our unit tests
from langflow.custom.directory_reader.utils import build_valid_menu
from loguru import logger
from loguru._logger import Core


# unit tests
def test_single_menu_item_one_component():
    # Single menu item with one component
    valid_components = {"menu": [{"name": "Menu1", "components": [("Component1", "Template1", {"output_types": "type1"})]}]}
    expected_output = {"Menu1": {"Component1": "Template1"}}
    codeflash_output = build_valid_menu(valid_components)

def test_multiple_menu_items_one_component_each():
    # Multiple menu items with one component each
    valid_components = {"menu": [{"name": "Menu1", "components": [("Component1", "Template1", {"output_types": "type1"})]}, {"name": "Menu2", "components": [("Component2", "Template2", {"output_types": "type2"})]}]}
    expected_output = {"Menu1": {"Component1": "Template1"}, "Menu2": {"Component2": "Template2"}}
    codeflash_output = build_valid_menu(valid_components)

def test_single_menu_item_multiple_components():
    # Single menu item with multiple components
    valid_components = {"menu": [{"name": "Menu1", "components": [("Component1", "Template1", {"output_types": "type1"}), ("Component2", "Template2", {"output_types": "type2"})]}]}
    expected_output = {"Menu1": {"Component1": "Template1", "Component2": "Template2"}}
    codeflash_output = build_valid_menu(valid_components)

def test_empty_components_list():
    # Empty components list
    valid_components = {"menu": [{"name": "Menu1", "components": []}]}
    expected_output = {"Menu1": {}}
    codeflash_output = build_valid_menu(valid_components)

def test_empty_menu_list():
    # Empty menu list
    valid_components = {"menu": []}
    expected_output = {}
    codeflash_output = build_valid_menu(valid_components)

def test_no_menu_key():
    # No menu key in valid_components
    valid_components = {}
    with pytest.raises(KeyError):
        build_valid_menu(valid_components)

def test_missing_name_key():
    # Missing 'name' key in menu item
    valid_components = {"menu": [{"components": [("Component1", "Template1", {"output_types": "type1"})]}]}
    with pytest.raises(KeyError):
        build_valid_menu(valid_components)

def test_missing_components_key():
    # Missing 'components' key in menu item
    valid_components = {"menu": [{"name": "Menu1"}]}
    with pytest.raises(KeyError):
        build_valid_menu(valid_components)

def test_incorrect_structure_in_components():
    # Incorrect structure in components list
    valid_components = {"menu": [{"name": "Menu1", "components": ["Component1", "Template1", {"output_types": "type1"}]}]}
    with pytest.raises(ValueError):
        build_valid_menu(valid_components)

def test_non_dict_valid_components():
    # Non-dictionary valid_components
    valid_components = ["menu"]
    with pytest.raises(TypeError):
        build_valid_menu(valid_components)

def test_non_list_menu_key():
    # Non-list 'menu' key
    valid_components = {"menu": "Menu1"}
    with pytest.raises(TypeError):
        build_valid_menu(valid_components)


def test_large_number_of_menu_items():
    # Large number of menu items
    valid_components = {"menu": [{"name": f"Menu{i}", "components": [(f"Component{i}", f"Template{i}", {"output_types": f"type{i}"})]} for i in range(1000)]}
    expected_output = {f"Menu{i}": {f"Component{i}": f"Template{i}"} for i in range(1000)}
    codeflash_output = build_valid_menu(valid_components)

def test_large_number_of_components_in_single_menu_item():
    # Large number of components in a single menu item
    valid_components = {"menu": [{"name": "Menu1", "components": [(f"Component{i}", f"Template{i}", {"output_types": f"type{i}"}) for i in range(1000)]}]}
    expected_output = {"Menu1": {f"Component{i}": f"Template{i}" for i in range(1000)}}
    codeflash_output = build_valid_menu(valid_components)

def test_component_raises_exception():
    # Component raises exception
    valid_components = {"menu": [{"name": "Menu1", "components": [("Component1", "Template1", {"output_types": "type1"}), ("Component2", None, {"output_types": "type2"})]}]}
    expected_output = {"Menu1": {"Component1": "Template1"}}
    codeflash_output = build_valid_menu(valid_components)

def test_invalid_component_template():
    # Invalid component template
    valid_components = {"menu": [{"name": "Menu1", "components": [("Component1", None, {"output_types": "type1"})]}]}
    expected_output = {"Menu1": {}}
    codeflash_output = build_valid_menu(valid_components)
# codeflash_output is used to check that the output of the original code is the same as that of the optimized code.
from typing import Any, Dict

# imports
import pytest  # used for our unit tests
from langflow.custom.directory_reader.utils import build_valid_menu
from loguru import logger


# unit tests
def test_basic_single_menu_item_single_component():
    valid_components = {"menu": [{"name": "Breakfast", "components": [("Pancakes", "pancake_template", {})]}]}
    expected_output = {"Breakfast": {"Pancakes": "pancake_template"}}
    codeflash_output = build_valid_menu(valid_components)

def test_basic_multiple_menu_items_single_component_each():
    valid_components = {"menu": [{"name": "Breakfast", "components": [("Pancakes", "pancake_template", {})]}, {"name": "Lunch", "components": [("Burger", "burger_template", {})]}]}
    expected_output = {"Breakfast": {"Pancakes": "pancake_template"}, "Lunch": {"Burger": "burger_template"}}
    codeflash_output = build_valid_menu(valid_components)

def test_basic_single_menu_item_multiple_components():
    valid_components = {"menu": [{"name": "Breakfast", "components": [("Pancakes", "pancake_template", {}), ("Coffee", "coffee_template", {})]}]}
    expected_output = {"Breakfast": {"Pancakes": "pancake_template", "Coffee": "coffee_template"}}
    codeflash_output = build_valid_menu(valid_components)

def test_edge_empty_valid_components():
    valid_components = {"menu": []}
    expected_output = {}
    codeflash_output = build_valid_menu(valid_components)

def test_edge_menu_item_with_no_components():
    valid_components = {"menu": [{"name": "Breakfast", "components": []}]}
    expected_output = {"Breakfast": {}}
    codeflash_output = build_valid_menu(valid_components)

def test_edge_component_with_missing_fields():
    valid_components = {"menu": [{"name": "Breakfast", "components": [("Pancakes", "pancake_template")]}]}
    with pytest.raises(ValueError):
        build_valid_menu(valid_components)

def test_invalid_data_type_for_valid_components():
    valid_components = "invalid_type"
    with pytest.raises(TypeError):
        build_valid_menu(valid_components)

def test_invalid_missing_menu_key_in_valid_components():
    valid_components = {}
    with pytest.raises(KeyError):
        build_valid_menu(valid_components)

def test_large_scale_large_number_of_menu_items_and_components():
    valid_components = {"menu": [{"name": f"Menu_{i}", "components": [(f"Component_{j}", f"template_{j}", {}) for j in range(100)]} for i in range(1000)]}
    codeflash_output = build_valid_menu(valid_components)
    for i in range(1000):
        pass

def test_large_scale_deeply_nested_components():
    valid_components = {"menu": [{"name": "Breakfast", "components": [("Pancakes", "pancake_template", {"output_types": [{"type": "nested", "value": {"type": "deeply_nested"}}]})]}]}
    expected_output = {"Breakfast": {"Pancakes": "pancake_template"}}
    codeflash_output = build_valid_menu(valid_components)

def test_error_handling_component_that_raises_exception():
    valid_components = {"menu": [{"name": "Breakfast", "components": [("Pancakes", "pancake_template", {"output_types": "invalid"})]}]}
    expected_output = {"Breakfast": {"Pancakes": "pancake_template"}}
    codeflash_output = build_valid_menu(valid_components)




def test_mixed_valid_and_invalid_data_some_valid_and_some_invalid_components():
    valid_components = {"menu": [{"name": "Breakfast", "components": [("Pancakes", "pancake_template", {}), ("InvalidComponent", None, {})]}]}
    expected_output = {"Breakfast": {"Pancakes": "pancake_template"}}
    codeflash_output = build_valid_menu(valid_components)

---

📣 **Feedback**

If you have any feedback or need assistance, feel free to join our Discord community: