⚡️ Speed up function set_tag by 14%

[codeflash-ai]

📄 14% (0.14x) speedup for set_tag in sentry_sdk/api.py

⏱️ Runtime : 88.2 microseconds → 77.3 microseconds (best of 18 runs)

📝 Explanation and details

The optimization eliminates redundant attribute lookups by caching the Scope.get_isolation_scope method reference in a module-level variable _get_isolation_scope.

Key changes:

• Added _get_isolation_scope = Scope.get_isolation_scope at module level to cache the method reference
• Modified get_isolation_scope() to call _get_isolation_scope() instead of Scope.get_isolation_scope()
• Modified set_tag() to call _get_isolation_scope() directly instead of going through the get_isolation_scope() wrapper function

Why this is faster:

1. Reduced attribute lookups: Each call to Scope.get_isolation_scope() requires Python to traverse the attribute chain (lookup Scope, then lookup get_isolation_scope on that class). The cached reference eliminates this lookup overhead.
2. Eliminated function call overhead: In set_tag(), the optimization bypasses the intermediate get_isolation_scope() function call entirely, reducing call stack depth.

The line profiler shows the impact - set_tag() execution time dropped from 3275.3ns per hit to 1511.3ns per hit (54% improvement per call). This optimization is particularly effective for workloads with frequent tagging operations, as evidenced by the performance test cases in the suite that make hundreds of set_tag() calls in tight loops.

✅ Correctness verification report:

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

⚙️ Existing Unit Tests and Runtime

Test File::Test Function	Original ⏱️	Optimized ⏱️	Speedup
integrations/launchdarkly/test_launchdarkly.py::test_launchdarkly_integration_asyncio	1.12μs	898ns	24.9%✅
integrations/launchdarkly/test_launchdarkly.py::test_launchdarkly_integration_threaded	1.52μs	1.19μs	27.7%✅
integrations/openfeature/test_openfeature.py::test_openfeature_integration_asyncio	1.11μs	930ns	19.5%✅
integrations/openfeature/test_openfeature.py::test_openfeature_integration_threaded	1.45μs	1.22μs	19.3%✅
integrations/statsig/test_statsig.py::test_check_gate_asyncio	1.18μs	927ns	27.1%✅
integrations/statsig/test_statsig.py::test_check_gate_threaded	1.55μs	1.18μs	31.9%✅
integrations/threading/test_threading.py::test_scope_data_not_leaked_in_threads	27.3μs	26.1μs	4.78%✅
integrations/unleash/test_unleash.py::test_is_enabled_asyncio	1.22μs	963ns	26.2%✅
integrations/unleash/test_unleash.py::test_is_enabled_threaded	1.47μs	1.11μs	32.6%✅
new_scopes_compat/test_new_scopes_compat.py::test_configure_scope_sdk1	3.47μs	2.71μs	27.9%✅
new_scopes_compat/test_new_scopes_compat.py::test_push_scope_sdk1	3.48μs	2.82μs	23.3%✅
new_scopes_compat/test_new_scopes_compat.py::test_with_cloned_hub_configure_scope_sdk1	4.89μs	4.04μs	21.1%✅
new_scopes_compat/test_new_scopes_compat.py::test_with_cloned_hub_push_scope_sdk1	4.63μs	3.76μs	23.3%✅
new_scopes_compat/test_new_scopes_compat.py::test_with_cloned_hub_sdk1	3.43μs	2.83μs	20.9%✅
new_scopes_compat/test_new_scopes_compat.py::test_with_hub_configure_scope_sdk1	4.74μs	4.02μs	17.9%✅
new_scopes_compat/test_new_scopes_compat.py::test_with_hub_push_scope_sdk1	4.66μs	3.86μs	20.9%✅
new_scopes_compat/test_new_scopes_compat.py::test_with_hub_sdk1	3.40μs	2.66μs	27.8%✅
test_client.py::test_nan	13.9μs	13.0μs	6.28%✅
test_crons.py::test_scope_data_in_checkin	1.12μs	1.00μs	11.9%✅
test_feature_flags.py::test_featureflags_integration_asyncio	1.14μs	904ns	25.9%✅
test_feature_flags.py::test_featureflags_integration_threaded	1.43μs	1.16μs	22.5%✅

🌀 Generated Regression Tests and Runtime

from copy import copy, deepcopy

# imports
import pytest  # used for our unit tests
from sentry_sdk.api import set_tag


def scopemethod(func):
    # Dummy decorator for testing
    return func
from sentry_sdk.api import set_tag

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






def test_set_tag_overwrite_existing_tag():
    # Test overwriting an existing tag value
    set_tag("env", "staging")
    set_tag("env", "production")
    scope = get_isolation_scope()

def test_set_tag_multiple_keys():
    # Test setting multiple distinct tags
    set_tag("env", "dev")
    set_tag("user", "alice")
    set_tag("region", "us-west")
    scope = get_isolation_scope()

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







def test_set_tag_key_case_sensitivity():
    # Test that keys are case-sensitive
    set_tag("Key", "upper")
    set_tag("key", "lower")
    scope = get_isolation_scope()










def test_set_tag_large_object_values():
    # Test setting tags with large dict/list values
    large_dict = {str(i): i for i in range(500)}
    large_list = list(range(500))
    set_tag("large_dict", large_dict)
    set_tag("large_list", large_list)
    scope = get_isolation_scope()

def test_set_tag_performance_under_load():
    # Test performance: setting many tags quickly
    import time
    start = time.time()
    for i in range(800):
        set_tag(f"perf{i}", i)
    duration = time.time() - start
    scope = get_isolation_scope()

def test_set_tag_memory_usage():
    # Test memory usage does not explode with many tags
    import sys
    for i in range(900):
        set_tag(f"mem{i}", "x" * 10)
    scope = get_isolation_scope()
    size = sys.getsizeof(scope._tags)
# codeflash_output is used to check that the output of the original code is the same as that of the optimized code.
#------------------------------------------------
from copy import copy, deepcopy

# imports
import pytest  # used for our unit tests
from sentry_sdk.api import set_tag


def get_tag(key):
    return get_isolation_scope().get_tag(key)

def clear_tags():
    return get_isolation_scope().clear_tags()

# ------------------- UNIT TESTS -------------------

# BASIC TEST CASES

To edit these changes git checkout codeflash/optimize-set_tag-mg95cj8s and push.