fix(cloud): retain all execution records for activity counts

This commit is contained in:
RockChinQ
2026-10-03 05:03:02 +00:00
parent b205c4000f
commit dcc7e17d77
2 changed files with 40 additions and 0 deletions
@@ -0,0 +1,34 @@
from types import SimpleNamespace
from langbot.pkg.telemetry.execution import ExecutionCounters
from langbot.pkg.utils import constants
def test_cloud_counts_are_not_sampled(monkeypatch):
monkeypatch.setattr(constants, 'edition', 'cloud')
counters = ExecutionCounters(SimpleNamespace(telemetry_config={'execution_trace': 'sampled'}))
assert counters.trace_mode() == 'all'
state = SimpleNamespace(failure_reason='', stages=[], synthetic=False, trace_id='00000001')
assert counters._trace_emitted(state)
def test_community_retains_sampling(monkeypatch):
monkeypatch.setattr(constants, 'edition', 'community')
counters = ExecutionCounters(SimpleNamespace(telemetry_config={'execution_trace': 'sampled'}))
assert counters.trace_mode() == 'sampled'
def test_dns_survives_saturated_executor():
import socket
import threading
from langbot.pkg.utils.bounded_executor import BoundedThreadPoolExecutor
gate = threading.Event()
executor = BoundedThreadPoolExecutor(max_workers=1, max_pending=0)
try:
blocked = executor.submit(gate.wait)
assert executor.submit(socket.getaddrinfo, '127.0.0.1', 443).result(timeout=2)
finally:
gate.set()
blocked.result(timeout=2)
executor.shutdown()