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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: logfire-api
-Version: 5.0.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.0.0 → logfire_api-5.1.1}/logfire_api/__init__.py

@@ -167,6 +167,9 @@ except ImportError:
 
             def instrument_starlette(self, *args, **kwargs) -> None: ...
 
+            def instrument_litestar(self, app, *args, **kwargs):
+                return app
+
             def instrument_django(self, *args, **kwargs) -> None: ...
 
             def instrument_psycopg(self, *args, **kwargs) -> None: ...
@@ -256,6 +259,7 @@ except ImportError:
         instrument_django = DEFAULT_LOGFIRE_INSTANCE.instrument_django
         instrument_flask = DEFAULT_LOGFIRE_INSTANCE.instrument_flask
         instrument_starlette = DEFAULT_LOGFIRE_INSTANCE.instrument_starlette
+        instrument_litestar = DEFAULT_LOGFIRE_INSTANCE.instrument_litestar
         instrument_aiohttp_client = DEFAULT_LOGFIRE_INSTANCE.instrument_aiohttp_client
         instrument_aiohttp_server = DEFAULT_LOGFIRE_INSTANCE.instrument_aiohttp_server
         instrument_sqlalchemy = DEFAULT_LOGFIRE_INSTANCE.instrument_sqlalchemy

{logfire_api-5.0.0 → logfire_api-5.1.1}/logfire_api/__init__.pyi

@@ -16,7 +16,7 @@ from logfire.propagate import attach_context as attach_context, get_context as g
 from logfire.sampling import SamplingOptions as SamplingOptions
 from typing import Any
 
-__all__ = ['Logfire', 'LogfireSpan', 'LevelName', 'AdvancedOptions', 'ConsoleOptions', 'CodeSource', 'PydanticPlugin', 'configure', 'span', 'instrument', 'log', 'trace', 'debug', 'notice', 'info', 'warn', 'warning', 'error', 'exception', 'fatal', 'force_flush', 'log_slow_async_callbacks', 'install_auto_tracing', 'instrument_asgi', 'instrument_wsgi', 'instrument_pydantic', 'instrument_pydantic_ai', 'instrument_fastapi', 'instrument_openai', 'instrument_openai_agents', 'instrument_anthropic', 'instrument_google_genai', 'instrument_litellm', 'instrument_dspy', 'instrument_print', 'instrument_asyncpg', 'instrument_httpx', 'instrument_celery', 'instrument_requests', 'instrument_psycopg', 'instrument_django', 'instrument_flask', 'instrument_starlette', 'instrument_aiohttp_client', 'instrument_aiohttp_server', 'instrument_sqlalchemy', 'instrument_sqlite3', 'instrument_aws_lambda', 'instrument_redis', 'instrument_pymongo', 'instrument_mysql', 'instrument_surrealdb', 'instrument_system_metrics', 'instrument_mcp', 'instrument_claude_agent_sdk', 'AutoTraceModule', 'with_tags', 'with_settings', 'suppress_scopes', 'shutdown', 'no_auto_trace', 'ScrubMatch', 'ScrubbingOptions', 'VERSION', 'add_non_user_code_prefix', 'suppress_instrumentation', 'StructlogProcessor', 'LogfireLoggingHandler', 'loguru_handler', 'SamplingOptions', 'MetricsOptions', 'VariablesOptions', 'LocalVariablesOptions', 'variables', 'var', 'template_var', 'variables_get', 'variables_push', 'variables_push_types', 'variables_validate', 'variables_push_config', 'variables_pull_config', 'variables_build_config', 'logfire_info', 'get_baggage', 'set_baggage', 'get_context', 'attach_context', 'url_from_eval', 'forward_export_request', 'forward_export_request_starlette']
+__all__ = ['Logfire', 'LogfireSpan', 'LevelName', 'AdvancedOptions', 'ConsoleOptions', 'CodeSource', 'PydanticPlugin', 'configure', 'span', 'instrument', 'log', 'trace', 'debug', 'notice', 'info', 'warn', 'warning', 'error', 'exception', 'fatal', 'force_flush', 'log_slow_async_callbacks', 'install_auto_tracing', 'instrument_asgi', 'instrument_wsgi', 'instrument_pydantic', 'instrument_pydantic_ai', 'instrument_fastapi', 'instrument_openai', 'instrument_openai_agents', 'instrument_anthropic', 'instrument_google_genai', 'instrument_litellm', 'instrument_dspy', 'instrument_print', 'instrument_asyncpg', 'instrument_httpx', 'instrument_celery', 'instrument_requests', 'instrument_psycopg', 'instrument_django', 'instrument_flask', 'instrument_starlette', 'instrument_litestar', 'instrument_aiohttp_client', 'instrument_aiohttp_server', 'instrument_sqlalchemy', 'instrument_sqlite3', 'instrument_aws_lambda', 'instrument_redis', 'instrument_pymongo', 'instrument_mysql', 'instrument_surrealdb', 'instrument_system_metrics', 'instrument_mcp', 'instrument_claude_agent_sdk', 'AutoTraceModule', 'with_tags', 'with_settings', 'suppress_scopes', 'shutdown', 'no_auto_trace', 'ScrubMatch', 'ScrubbingOptions', 'VERSION', 'add_non_user_code_prefix', 'suppress_instrumentation', 'StructlogProcessor', 'LogfireLoggingHandler', 'loguru_handler', 'SamplingOptions', 'MetricsOptions', 'VariablesOptions', 'LocalVariablesOptions', 'variables', 'var', 'template_var', 'variables_get', 'variables_push', 'variables_push_types', 'variables_validate', 'variables_push_config', 'variables_pull_config', 'variables_build_config', 'logfire_info', 'get_baggage', 'set_baggage', 'get_context', 'attach_context', 'url_from_eval', 'forward_export_request', 'forward_export_request_starlette']
 
 DEFAULT_LOGFIRE_INSTANCE = Logfire()
 span = DEFAULT_LOGFIRE_INSTANCE.span
@@ -44,6 +44,7 @@ instrument_psycopg = DEFAULT_LOGFIRE_INSTANCE.instrument_psycopg
 instrument_django = DEFAULT_LOGFIRE_INSTANCE.instrument_django
 instrument_flask = DEFAULT_LOGFIRE_INSTANCE.instrument_flask
 instrument_starlette = DEFAULT_LOGFIRE_INSTANCE.instrument_starlette
+instrument_litestar = DEFAULT_LOGFIRE_INSTANCE.instrument_litestar
 instrument_aiohttp_client = DEFAULT_LOGFIRE_INSTANCE.instrument_aiohttp_client
 instrument_aiohttp_server = DEFAULT_LOGFIRE_INSTANCE.instrument_aiohttp_server
 instrument_sqlalchemy = DEFAULT_LOGFIRE_INSTANCE.instrument_sqlalchemy

{logfire_api-5.0.0 → logfire_api-5.1.1}/logfire_api/_internal/cli/run.pyi

@@ -14,6 +14,7 @@ TARGET_PACKAGE_ALIASES: Incomplete
 INSTRUMENTATION_CALL_ARGUMENTS: Incomplete
 MINIMUM_INSTRUMENTATION_VERSIONS: Incomplete
 OTEL_INSTRUMENTATION_MAP: Incomplete
+INSTRUMENTATION_TO_EXTRA: dict[str, str]
 
 @dataclass(frozen=True, order=True)
 class InstrumentationRecommendation:

{logfire_api-5.0.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.0.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.0.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.1/logfire_api/_internal/integrations/litestar.pyi

@@ -0,0 +1,7 @@
+from .asgi import ASGIApp as ASGIApp, ASGIInstrumentKwargs as ASGIInstrumentKwargs, instrument_asgi as instrument_asgi
+from litestar import Litestar
+from logfire import Logfire as Logfire
+from typing_extensions import Unpack
+
+def instrument_litestar(logfire_instance: Logfire, app: Litestar, *, record_send_receive: bool = False, capture_headers: bool = False, **kwargs: Unpack[ASGIInstrumentKwargs]) -> ASGIApp:
+    """Wrap a Litestar app with OpenTelemetry middleware configured for Logfire."""

{logfire_api-5.0.0 → logfire_api-5.1.1}/logfire_api/_internal/logs.pyi

@@ -3,7 +3,7 @@ from dataclasses import dataclass
 from logfire._internal.constants import LEVEL_NUMBERS as LEVEL_NUMBERS
 from opentelemetry._logs import LogRecord, Logger, LoggerProvider
 from opentelemetry.util.types import _ExtendedAttributes
-from threading import Lock
+from threading import RLock
 from typing import Any, overload
 from weakref import WeakSet
 
@@ -12,7 +12,7 @@ class ProxyLoggerProvider(LoggerProvider):
     """A logger provider that wraps another internal logger provider allowing it to be re-assigned."""
     provider: LoggerProvider
     loggers: WeakSet[ProxyLogger] = dataclasses.field(default_factory=WeakSet['ProxyLogger'])
-    lock: Lock = dataclasses.field(default_factory=Lock)
+    lock: RLock = dataclasses.field(default_factory=RLock)
     suppressed_scopes: set[str] = dataclasses.field(default_factory=set[str])
     min_level: int = ...
     def get_logger(self, name: str, version: str | None = None, schema_url: str | None = None, attributes: _ExtendedAttributes | None = None) -> Logger: ...

{logfire_api-5.0.0 → logfire_api-5.1.1}/logfire_api/_internal/main.pyi

@@ -42,6 +42,7 @@ from contextlib import AbstractContextManager
 from django.http import HttpRequest as HttpRequest, HttpResponse as HttpResponse
 from fastapi import FastAPI
 from flask.app import Flask
+from litestar import Litestar
 from opentelemetry.context import Context as Context
 from opentelemetry.instrumentation.asgi.types import ClientRequestHook, ClientResponseHook, ServerRequestHook
 from opentelemetry.metrics import CallbackT as CallbackT, Counter, Histogram, UpDownCounter, _Gauge as Gauge
@@ -800,6 +801,29 @@ class Logfire:
             response_hook: A function called right before a span is finished for the response.
             **kwargs: Additional keyword arguments to pass to the OpenTelemetry Flask instrumentation.
         """
+    def instrument_litestar(self, app: Litestar, *, capture_headers: bool = False, record_send_receive: bool = False, **kwargs: Unpack[ASGIInstrumentKwargs]) -> ASGIApp:
+        """Instrument a Litestar app to record requests with canonical route templates.
+
+        Uses `instrument_asgi()` with a Litestar route extractor, so requests to
+        `/users/1` and `/users/2` share the route `/users/{user_id}`.
+
+        Warning:
+            This method returns an ASGI wrapper instead of modifying the app in place.
+            Pass the returned app to your server. Access Litestar-specific attributes
+            through the original app.
+
+        Args:
+            app: The Litestar app to instrument.
+            capture_headers: Set to `True` to capture all request and response headers.
+            record_send_receive: Set to `True` to record low-level ASGI send and receive spans.
+                These are disabled by default to reduce noise.
+            **kwargs: Additional options for the OpenTelemetry ASGI middleware, including
+                `server_request_hook`, `client_request_hook`, `client_response_hook`,
+                `excluded_urls`, and `default_span_details`.
+
+        Returns:
+            The instrumented ASGI application.
+        """
     def instrument_starlette(self, app: Starlette, *, capture_headers: bool = False, record_send_receive: bool = False, server_request_hook: ServerRequestHook | None = None, client_request_hook: ClientRequestHook | None = None, client_response_hook: ClientResponseHook | None = None, **kwargs: Any) -> None:
         """Instrument `app` so that spans are automatically created for each request.
 

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

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "logfire-api"
-version = "5.0.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" },