logfire-api 5.1.0__tar.gz → 5.1.1__tar.gz

{logfire_api-5.1.0 → logfire_api-5.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: logfire-api
-Version: 5.1.0
+Version: 5.1.1
 Summary: Shim for the Logfire SDK which does nothing unless Logfire is installed
 Author-email: Pydantic Team <engineering@pydantic.dev>, Samuel Colvin <samuel@pydantic.dev>, Hasan Ramezani <hasan@pydantic.dev>, Adrian Garcia Badaracco <adrian@pydantic.dev>, David Montague <david@pydantic.dev>, Marcelo Trylesinski <marcelo@pydantic.dev>, David Hewitt <david.hewitt@pydantic.dev>, Alex Hall <alex@pydantic.dev>
 License-Expression: MIT

{logfire_api-5.1.0 → logfire_api-5.1.1}/logfire_api/_internal/client.pyi

@@ -1,5 +1,6 @@
 from .auth import UserToken as UserToken, UserTokenCollection as UserTokenCollection
 from .constants import HTTP_CONNECT_TIMEOUT as HTTP_CONNECT_TIMEOUT
+from .http_transport import install_connection_policy as install_connection_policy
 from .server_response import ServerResponseCallback as ServerResponseCallback, install_logfire_response_hook as install_logfire_response_hook
 from .utils import UnexpectedResponse as UnexpectedResponse
 from _typeshed import Incomplete

{logfire_api-5.1.0 → logfire_api-5.1.1}/logfire_api/_internal/config.pyi

@@ -15,6 +15,7 @@ from .exporters.quiet_metrics import QuietMetricExporter as QuietMetricExporter
 from .exporters.remove_pending import RemovePendingSpansExporter as RemovePendingSpansExporter
 from .exporters.test import TestExporter as TestExporter
 from .forwarding import OTLPForwardingManager as OTLPForwardingManager
+from .http_transport import install_connection_policy as install_connection_policy
 from .integrations.executors import instrument_executors as instrument_executors
 from .interactive import ask_or_default as ask_or_default, ask_required as ask_required, require_answer as require_answer
 from .logs import ProxyLoggerProvider as ProxyLoggerProvider

{logfire_api-5.1.0 → logfire_api-5.1.1}/logfire_api/_internal/exporters/otlp.pyi

@@ -1,6 +1,7 @@
 import atexit
 import requests
 from ..constants import ATTRIBUTES_MESSAGE_KEY as ATTRIBUTES_MESSAGE_KEY, ATTRIBUTES_SPAN_TYPE_KEY as ATTRIBUTES_SPAN_TYPE_KEY, HTTP_CONNECT_TIMEOUT as HTTP_CONNECT_TIMEOUT, OTLP_MAX_INT_SIZE as OTLP_MAX_INT_SIZE, log_level_attributes as log_level_attributes
+from ..http_transport import install_connection_policy as install_connection_policy
 from ..stack_info import STACK_INFO_KEYS as STACK_INFO_KEYS
 from ..utils import logger as logger, platform_is_emscripten as platform_is_emscripten, truncate_string as truncate_string
 from .wrapper import WrapperLogExporter as WrapperLogExporter, WrapperSpanExporter as WrapperSpanExporter
@@ -28,6 +29,7 @@ class BodySizeCheckingOTLPSpanExporter(OTLPSpanExporter):
 
 class OTLPExporterHttpSession(Session):
     """A requests.Session subclass that defers failed requests to a DiskRetryer."""
+    def __init__(self) -> None: ...
     def request(self, method: str, url: str, **kwargs: Any): ...
     def post(self, url: str, data: bytes, **kwargs: Any): ...
     @cached_property

logfire_api-5.1.1/logfire_api/_internal/http_transport.pyi

@@ -0,0 +1,47 @@
+from requests import Session
+from requests.adapters import HTTPAdapter
+from typing import Any
+from urllib3.connectionpool import HTTPConnectionPool
+
+IDLE_CONNECTION_RECYCLE_SECONDS: int
+TCP_KEEPALIVE_IDLE_SECONDS: int
+TCP_KEEPALIVE_INTERVAL_SECONDS: int
+TCP_KEEPALIVE_FAILED_PROBES: int
+
+def keepalive_socket_options() -> list[tuple[int, int, int | bytes]]:
+    """`urllib3`'s default socket options plus TCP keepalive.
+
+    Building on `HTTPConnection.default_socket_options` keeps `TCP_NODELAY`, which `urllib3` sets
+    and which we have no reason to drop. The class is read off the module at call time because
+    `urllib3` swaps it out under Pyodide, for an Emscripten connection that talks over `fetch()`
+    and so has neither a socket nor any default options.
+    """
+
+class _IdleRecyclingPoolMixin(HTTPConnectionPool):
+    """Closes a pooled connection that has sat idle longer than the recycle window.
+
+    The unit that goes stale is one connection, not the session: a session can be busy
+    continuously while an individual pooled connection sits unused. That is the normal shape for
+    Logfire, where one session carries traces, metrics and logs from different threads, so a
+    steady trace stream keeps the session active while the connection that last served a metric
+    export waits out the full export interval.
+
+    `urllib3` pools from a LIFO queue, which makes this sharper than steady-state staleness.
+    Sequential traffic reuses the connection on top of the stack, so a connection opened during
+    a brief overlap is used once and then left at the bottom, idle for as long as the workload
+    stays sequential. It is handed back out at the next overlap, which is exactly when a request
+    is least able to tolerate a dead socket. Recycling on the way out of the pool is what catches
+    that connection before it is used.
+
+    `urllib3` already drops a pooled connection it can see was closed. This extends that to the
+    case it cannot see, closing on the way out of the pool and letting `urllib3` reconnect
+    lazily, exactly as it does for a connection it detected as dropped.
+    """
+
+class LogfireHTTPAdapter(HTTPAdapter):
+    """A `requests` adapter that enables TCP keepalive and recycles idle pooled connections."""
+    def init_poolmanager(self, connections: int, maxsize: int, block: bool = ..., **pool_kwargs: Any) -> None: ...
+    def proxy_manager_for(self, proxy: str, **proxy_kwargs: Any) -> Any: ...
+
+def install_connection_policy(session: Session) -> None:
+    """Apply the keepalive and idle recycle policy to a session Logfire owns."""

{logfire_api-5.1.0 → logfire_api-5.1.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "logfire-api"
-version = "5.1.0"
+version = "5.1.1"
 description = "Shim for the Logfire SDK which does nothing unless Logfire is installed"
 authors = [
     { name = "Pydantic Team", email = "engineering@pydantic.dev" },