Litestar compatibility

[ddahan]

Description

Hi there, I was wondering is there is an easy way to use Logfire with a Litestar app?
I can't see the framework in the integrations list:

Thanks!

[alexmojaki]

import uvicorn
from litestar import Litestar, get
from litestar.contrib.opentelemetry import OpenTelemetryConfig, OpenTelemetryPlugin

import logfire
from logfire._internal.integrations.asgi import tweak_asgi_spans_tracer_provider

logfire_instance = logfire.configure()


@get('/')
async def hello_world() -> str:
    return 'Hello, world!'


app = Litestar(
    [hello_world],
    plugins=[
        OpenTelemetryPlugin(
            OpenTelemetryConfig(
                # This isn't strictly necessary, but gets rid of some annoying extra spans.
                # The underlying opentelemetry.instrumentation.asgi.OpenTelemetryMiddleware
                # has a config exclude_spans which has the same effect,
                # but Litestar's OpenTelemetryConfig doesn't expose that option, at least not easily.
                # `app = logfire.instrument_asgi(app)` would also work instead of this whole plugin,
                # but you wouldn't get the `http.route` attribute or useful span names.
                tracer_provider=tweak_asgi_spans_tracer_provider(logfire_instance, record_send_receive=False)
            )
        )
    ],
)

uvicorn.run(app)

[alexmojaki]

See https://docs.litestar.dev/2/usage/metrics/open-telemetry.html and https://logfire.pydantic.dev/docs/integrations/web-frameworks/asgi/

[ddahan]

Thanks a lot for pointing right documentation links.
What's the con's of not having a dedicated integration in that case? Do we lose some interesting features?

[alexmojaki]

In this case, not really, but it would still be good to have a more user-friendly method.