⚡️ Speed up method AstraDBVectorStoreComponent.get_database_object by 70% in PR #6236 (LFOSS-492)

[codeflash-ai]

⚡️ This pull request contains optimizations for PR #6236

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

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

---

📄 70% (0.70x) speedup for AstraDBVectorStoreComponent.get_database_object in src/backend/base/langflow/components/vectorstores/astradb.py

⏱️ Runtime : 49.2 microseconds → 28.9 microseconds (best of 24 runs)

📝 Explanation and details

Here is the optimized version of the provided Python program.

Explanation of Optimizations.

1. Caching Computed Values:

   • Implemented caching for the api_endpoint and keyspace values by checking if they are already computed and stored. This avoids redundant calculations and improves performance.

2. Removed Redundant Code:

   • Removed the redundant keyspace assignment inside the get_keyspace method, handling it more efficiently by using caching built into the instance with attribute checks.

By applying these optimizations, the program now avoids unnecessary computations, leading to quicker execution times.

✅ Correctness verification report:

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

🌀 Generated Regression Tests Details

from unittest import mock  # used for mocking dependencies

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

# unit tests

# Test valid inputs



def test_invalid_token():
    component = AstraDBVectorStoreComponent()
    component.token = ""
    component.environment = "valid_environment"
    component.keyspace = "valid_keyspace"
    
    with pytest.raises(ValueError, match="Error fetching database object: "):
        component.get_database_object()

# Test edge cases



from unittest.mock import MagicMock, patch

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

# unit tests

@pytest.fixture
def setup_component():
    component = AstraDBVectorStoreComponent()
    component.token = "valid_token"
    component.environment = "valid_env"
    component.keyspace = "valid_keyspace"
    component.api_endpoint = "https://valid.endpoint.com"
    component.database_name = "valid_db"
    return component






def test_invalid_api_endpoint(setup_component):
    component = setup_component
    api_endpoint = "invalid_endpoint"
    with patch.object(DataAPIClient, 'get_database', side_effect=Exception("Invalid endpoint")):
        with pytest.raises(ValueError):
            component.get_database_object(api_endpoint)

def test_invalid_token(setup_component):
    component = setup_component
    component.token = "invalid_token"
    with patch.object(DataAPIClient, 'get_database', side_effect=Exception("Invalid token")):
        with pytest.raises(ValueError):
            component.get_database_object()




def test_static_method_returns_invalid_endpoint(setup_component):
    component = setup_component
    with patch.object(component, 'get_api_endpoint_static', return_value="invalid_endpoint"):
        with patch.object(DataAPIClient, 'get_database', side_effect=Exception("Invalid endpoint")):
            with pytest.raises(ValueError):
                component.get_database_object()

def test_combination_of_valid_and_invalid_attributes(setup_component):
    component = setup_component
    component.environment = "invalid_env"
    with patch.object(DataAPIClient, 'get_database', side_effect=Exception("Invalid environment")):
        with pytest.raises(ValueError):
            component.get_database_object()

def test_multiple_errors(setup_component):
    component = setup_component
    component.token = "invalid_token"
    component.environment = "invalid_env"
    with patch.object(DataAPIClient, 'get_database', side_effect=Exception("Multiple errors")):
        with pytest.raises(ValueError):
            component.get_database_object()