AIOHTTP Client

AIOHTTP is an asynchronous HTTP client/server framework for asyncio and Python.

The logfire.instrument_aiohttp_client() method will create a span for every request made by your AIOHTTP clients.

For AIOHTTP server instrumentation, see here.

Installation

Install logfire with the aiohttp-client extra:

pipuv

pip install 'logfire[aiohttp-client]'

uv add 'logfire[aiohttp-client]'

Usage

Let's see a minimal example below. You can run it with python main.py:

main.py

import logfire
import aiohttp


logfire.configure()
logfire.instrument_aiohttp_client()


async def main():
    async with aiohttp.ClientSession() as session:
        await session.get("https://httpbin.org/get")


if __name__ == "__main__":
    import asyncio

    asyncio.run(main())

The keyword arguments of logfire.instrument_aiohttp_client() are passed to the AioHttpClientInstrumentor().instrument() method of the OpenTelemetry aiohttp client Instrumentation package, read more about it here.

Configuration

The logfire.instrument_aiohttp_client() method accepts various parameters to configure the instrumentation.

Note

The aiohttp client instrumentation captures request and response headers, and response bodies. Request bodies are not captured.

Capture HTTP Headers

By default, Logfire doesn't capture HTTP headers. You can enable capturing both request and response headers by setting the capture_headers parameter to True.

import aiohttp
import logfire

logfire.configure()
logfire.instrument_aiohttp_client(capture_headers=True)

async def main():
    async with aiohttp.ClientSession() as session:
        await session.get("https://httpbin.org/get")

if __name__ == "__main__":
    import asyncio
    asyncio.run(main())

Capture Only Request Headers

Instead of capturing both request and response headers, you can create a request hook to capture only the request headers:

import aiohttp
import logfire
from aiohttp.tracing import TraceRequestStartParams
from opentelemetry.trace import Span


def capture_request_headers(span: Span, request: TraceRequestStartParams):
    headers = request.headers
    span.set_attributes(
        {
            f'http.request.header.{header_name}': headers.getall(header_name)
            for header_name in headers.keys()
        }
    )


logfire.configure()
logfire.instrument_aiohttp_client(request_hook=capture_request_headers)

async def main():
    async with aiohttp.ClientSession() as session:
        await session.get("https://httpbin.org/get")

if __name__ == "__main__":
    import asyncio
    asyncio.run(main())

Capture Only Response Headers

Similarly, you can create a response hook to capture only the response headers:

import aiohttp
import logfire
from aiohttp.tracing import TraceRequestEndParams, TraceRequestExceptionParams
from opentelemetry.trace import Span
from typing import Union


def capture_response_headers(span: Span, response: Union[TraceRequestEndParams, TraceRequestExceptionParams]):
    if hasattr(response, 'response') and response.response:
        headers = response.response.headers
        span.set_attributes(
            {f'http.response.header.{header_name}': headers.getall(header_name)
             for header_name in headers.keys()}
        )


logfire.configure()
logfire.instrument_aiohttp_client(response_hook=capture_response_headers)

async def main():
    async with aiohttp.ClientSession() as session:
        await session.get('https://httpbin.org/get')

if __name__ == "__main__":
    import asyncio
    asyncio.run(main())

You can also use the hooks to filter headers or modify them before capturing them.

Capture HTTP Response Bodies

By default, Logfire doesn't capture HTTP response bodies.

To capture response bodies, you can set the capture_response_body parameter to True.

import aiohttp
import logfire

logfire.configure()
logfire.instrument_aiohttp_client(capture_response_body=True)

async def main():
    async with aiohttp.ClientSession() as session:
        response = await session.get("https://httpbin.org/get")
        await response.text()

if __name__ == "__main__":
    import asyncio
    asyncio.run(main())

Hiding sensitive URL parameters

The url_filter keyword argument can be used to modify the URL that's recorded in spans. Here's an example of how to use this to redact query parameters:

from yarl import URL

def mask_url(url: URL) -> str:
    sensitive_keys = {"username", "password", "token", "api_key", "api_secret", "apikey"}
    masked_query = {key: "*****" if key in sensitive_keys else value for key, value in url.query.items()}
    return str(url.with_query(masked_query))

logfire.instrument_aiohttp_client(url_filter=mask_url)