⚡️ Speed up method AstraDBVectorStoreComponent.reset_database_list by 42% in PR #6028 (PlaygroundPage)

[codeflash-ai]

⚡️ This pull request contains optimizations for PR #6028

If you approve this dependent PR, these changes will be merged into the original PR branch PlaygroundPage.

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

---

📄 42% (0.42x) speedup for AstraDBVectorStoreComponent.reset_database_list in src/backend/base/langflow/components/vectorstores/astradb.py

⏱️ Runtime : 1.64 millisecond → 1.16 millisecond (best of 19 runs)

📝 Explanation and details

Here’s the optimized version.

2. Memory Efficient Operations.

3. Reduced Overhead.

✅ Correctness verification report:

Test	Status
⚙️ Existing Unit Tests	🔘 None Found
🌀 Generated Regression Tests	✅ 17 Passed
⏪ Replay Tests	🔘 None Found
🔎 Concolic Coverage Tests	🔘 None Found
📊 Tests Coverage	undefined

🌀 Generated Regression Tests Details

from unittest.mock import patch

# imports
import pytest  # used for our unit tests
# function to test
from langflow.base.vectorstores.model import LCVectorStoreComponent
from langflow.components.vectorstores.astradb import \
    AstraDBVectorStoreComponent

# unit tests

# Helper function to mock get_database_list
def mock_get_database_list():
    return {
        "db1": {"collections": ["col1", "col2"], "api_endpoint": "endpoint1"},
        "db2": {"collections": ["col3"], "api_endpoint": "endpoint2"}
    }

# Test class for AstraDBVectorStoreComponent
class TestAstraDBVectorStoreComponent:
    @patch.object(AstraDBVectorStoreComponent, 'get_database_list', side_effect=mock_get_database_list)
    def test_basic_functionality(self, mock_get_db_list):
        component = AstraDBVectorStoreComponent()
        build_config = {"api_endpoint": {"options": [], "options_metadata": [], "value": ""}}
        expected_config = {
            "api_endpoint": {
                "options": ["db1", "db2"],
                "options_metadata": [
                    {"collections": ["col1", "col2"], "api_endpoint": "endpoint1"},
                    {"collections": ["col3"], "api_endpoint": "endpoint2"}
                ],
                "value": ""
            }
        }
        codeflash_output = component.reset_database_list(build_config)

    @patch.object(AstraDBVectorStoreComponent, 'get_database_list', side_effect=lambda: {})
    def test_empty_database_list(self, mock_get_db_list):
        component = AstraDBVectorStoreComponent()
        build_config = {"api_endpoint": {"options": [], "options_metadata": [], "value": ""}}
        expected_config = {"api_endpoint": {"options": [], "options_metadata": [], "value": ""}}
        codeflash_output = component.reset_database_list(build_config)

    

import pytest  # used for our unit tests
# function to test
from langflow.base.vectorstores.model import LCVectorStoreComponent
from langflow.components.vectorstores.astradb import \
    AstraDBVectorStoreComponent

# unit tests

@pytest.fixture
def mock_component():
    class MockAstraDBVectorStoreComponent(AstraDBVectorStoreComponent):
        def get_database_list(self):
            return self.mock_database_list
    return MockAstraDBVectorStoreComponent()

def test_reset_database_list_multiple_databases(mock_component):
    # Basic functionality: multiple databases
    mock_component.mock_database_list = {
        "db1": {"collections": ["col1"], "api_endpoint": "http://db1.endpoint"},
        "db2": {"collections": ["col2"], "api_endpoint": "http://db2.endpoint"}
    }
    build_config = {"api_endpoint": {"options": [], "options_metadata": [], "value": ""}}
    codeflash_output = mock_component.reset_database_list(build_config)

def test_reset_database_list_single_database(mock_component):
    # Basic functionality: single database
    mock_component.mock_database_list = {
        "db1": {"collections": ["col1"], "api_endpoint": "http://db1.endpoint"}
    }
    build_config = {"api_endpoint": {"options": [], "options_metadata": [], "value": ""}}
    codeflash_output = mock_component.reset_database_list(build_config)

def test_reset_database_list_empty_database_list(mock_component):
    # Edge case: empty database list
    mock_component.mock_database_list = {}
    build_config = {"api_endpoint": {"options": [], "options_metadata": [], "value": ""}}
    codeflash_output = mock_component.reset_database_list(build_config)


def test_reset_database_list_empty_build_config(mock_component):
    # Edge case: empty build_config
    mock_component.mock_database_list = {
        "db1": {"collections": ["col1"], "api_endpoint": "http://db1.endpoint"}
    }
    build_config = {}
    with pytest.raises(KeyError):
        mock_component.reset_database_list(build_config)

def test_reset_database_list_missing_api_endpoint_key(mock_component):
    # Edge case: missing api_endpoint key in build_config
    mock_component.mock_database_list = {
        "db1": {"collections": ["col1"], "api_endpoint": "http://db1.endpoint"}
    }
    build_config = {"some_other_key": {}}
    with pytest.raises(KeyError):
        mock_component.reset_database_list(build_config)


def test_reset_database_list_large_number_of_databases(mock_component):
    # Large scale test: large number of databases
    mock_component.mock_database_list = {
        f"db{i}": {"collections": [f"col{i}"], "api_endpoint": f"http://db{i}.endpoint"} for i in range(1000)
    }
    build_config = {"api_endpoint": {"options": [], "options_metadata": [], "value": ""}}
    codeflash_output = mock_component.reset_database_list(build_config)

def test_reset_database_list_nested_collections(mock_component):
    # Complex data structures: nested collections
    mock_component.mock_database_list = {
        "db1": {"collections": [{"sub_col": ["sub1", "sub2"]}], "api_endpoint": "http://db1.endpoint"}
    }
    build_config = {"api_endpoint": {"options": [], "options_metadata": [], "value": ""}}
    codeflash_output = mock_component.reset_database_list(build_config)

def test_reset_database_list_performance_under_load(mock_component):
    # Performance and scalability: performance under load
    mock_component.mock_database_list = {
        f"db{i}": {"collections": [f"col{i}"], "api_endpoint": f"http://db{i}.endpoint"} for i in range(1000)
    }
    build_config = {"api_endpoint": {"options": [], "options_metadata": [], "value": ""}}
    build_config.update({f"key{i}": f"value{i}" for i in range(1000)})
    codeflash_output = mock_component.reset_database_list(build_config)
# codeflash_output is used to check that the output of the original code is the same as that of the optimized code.