Release v3.16.1 (#1091)

Author: alexmojaki

CHANGELOG.md

@@ -1,5 +1,10 @@
 # Release Notes
 
+## [v3.16.1] (2025-05-26)
+
+* Infer base URL from read token in query client by @Viicos in [#1088](https://github.com/pydantic/logfire/pull/1088)
+* Add `include_binary_content` ([#1090](https://github.com/pydantic/logfire/pull/1090)) and `**kwargs` ([#1078](https://github.com/pydantic/logfire/pull/1078)) to `instrument_pydantic_ai` by @alexmojaki
+
 ## [v3.16.0] (2025-05-14)
 
 * Make OpenAI spans show token usage in logfire UI by @alexmojaki in [#1076](https://github.com/pydantic/logfire/pull/1076)
@@ -700,3 +705,4 @@ First release from new repo!
 [v3.15.0]: https://github.com/pydantic/logfire/compare/v3.14.1...v3.15.0
 [v3.15.1]: https://github.com/pydantic/logfire/compare/v3.15.0...v3.15.1
 [v3.16.0]: https://github.com/pydantic/logfire/compare/v3.15.1...v3.16.0
+[v3.16.1]: https://github.com/pydantic/logfire/compare/v3.16.0...v3.16.1

logfire-api/logfire_api/_internal/integrations/pydantic_ai.pyi

@@ -2,6 +2,6 @@ from logfire import Logfire as Logfire
 from pydantic_ai import Agent
 from pydantic_ai.models import Model
 from pydantic_ai.models.instrumented import InstrumentedModel
-from typing import Literal
+from typing import Any, Literal
 
-def instrument_pydantic_ai(logfire_instance: Logfire, obj: Agent | Model | None, event_mode: Literal['attributes', 'logs'] | None) -> None | InstrumentedModel: ...
+def instrument_pydantic_ai(logfire_instance: Logfire, obj: Agent | Model | None, event_mode: Literal['attributes', 'logs'] | None, **kwargs: Any) -> None | InstrumentedModel: ...

logfire-api/logfire_api/_internal/main.pyi

@@ -422,9 +422,9 @@ class Logfire:
                 Exclude specific modules from instrumentation.
         """
     @overload
-    def instrument_pydantic_ai(self, obj: pydantic_ai.Agent | None = None, /, *, event_mode: Literal['attributes', 'logs'] = 'attributes') -> None: ...
+    def instrument_pydantic_ai(self, obj: pydantic_ai.Agent | None = None, /, *, event_mode: Literal['attributes', 'logs'] = 'attributes', include_binary_content: bool | None = None, **kwargs: Any) -> None: ...
     @overload
-    def instrument_pydantic_ai(self, obj: pydantic_ai.models.Model, /, *, event_mode: Literal['attributes', 'logs'] = 'attributes') -> pydantic_ai.models.Model: ...
+    def instrument_pydantic_ai(self, obj: pydantic_ai.models.Model, /, *, event_mode: Literal['attributes', 'logs'] = 'attributes', include_binary_content: bool | None = None, **kwargs: Any) -> pydantic_ai.models.Model: ...
     def instrument_fastapi(self, app: FastAPI, *, capture_headers: bool = False, request_attributes_mapper: Callable[[Request | WebSocket, dict[str, Any]], dict[str, Any] | None] | None = None, excluded_urls: str | Iterable[str] | None = None, record_send_receive: bool = False, **opentelemetry_kwargs: Any) -> ContextManager[None]:
         """Instrument a FastAPI app so that spans and logs are automatically created for each request.
 

logfire-api/logfire_api/experimental/query_client.pyi

@@ -2,9 +2,11 @@ from _typeshed import Incomplete
 from datetime import datetime
 from httpx import AsyncClient, Client, Response, Timeout
 from httpx._client import BaseClient
+from logfire._internal.config import get_base_url_from_token as get_base_url_from_token
 from pyarrow import Table
 from types import TracebackType
 from typing import Any, Generic, TypeVar, TypedDict
+from typing_extensions import Self
 
 DEFAULT_TIMEOUT: Incomplete
 
@@ -32,8 +34,6 @@ class RowQueryResults(TypedDict):
     columns: list[ColumnDetails]
     rows: list[dict[str, Any]]
 T = TypeVar('T', bound=BaseClient)
-S = TypeVar('S', bound='LogfireQueryClient')
-R = TypeVar('R', bound='AsyncLogfireQueryClient')
 
 class _BaseLogfireQueryClient(Generic[T]):
     base_url: Incomplete
@@ -46,8 +46,8 @@ class _BaseLogfireQueryClient(Generic[T]):
 
 class LogfireQueryClient(_BaseLogfireQueryClient[Client]):
     """A synchronous client for querying Logfire data."""
-    def __init__(self, read_token: str, base_url: str = 'https://logfire-api.pydantic.dev/', timeout: Timeout = ..., **client_kwargs: Any) -> None: ...
-    def __enter__(self) -> S: ...
+    def __init__(self, read_token: str, base_url: str | None = None, timeout: Timeout = ..., **client_kwargs: Any) -> None: ...
+    def __enter__(self) -> Self: ...
     def __exit__(self, exc_type: type[BaseException] | None = None, exc_value: BaseException | None = None, traceback: TracebackType | None = None) -> None: ...
     def query_json(self, sql: str, min_timestamp: datetime | None = None, max_timestamp: datetime | None = None, limit: int | None = None) -> QueryResults:
         """Query Logfire data and return the results as a column-oriented dictionary."""
@@ -68,8 +68,8 @@ class LogfireQueryClient(_BaseLogfireQueryClient[Client]):
 
 class AsyncLogfireQueryClient(_BaseLogfireQueryClient[AsyncClient]):
     """An asynchronous client for querying Logfire data."""
-    def __init__(self, read_token: str, base_url: str = 'https://logfire-api.pydantic.dev/', timeout: Timeout = ..., **async_client_kwargs: Any) -> None: ...
-    async def __aenter__(self) -> R: ...
+    def __init__(self, read_token: str, base_url: str | None = None, timeout: Timeout = ..., **async_client_kwargs: Any) -> None: ...
+    async def __aenter__(self) -> Self: ...
     async def __aexit__(self, exc_type: type[BaseException] | None = None, exc_value: BaseException | None = None, traceback: TracebackType | None = None) -> None: ...
     async def query_json(self, sql: str, min_timestamp: datetime | None = None, max_timestamp: datetime | None = None, limit: int | None = None) -> QueryResults:
         """Query Logfire data and return the results as a column-oriented dictionary."""

logfire-api/pyproject.toml

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

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "logfire"
-version = "3.16.0"
+version = "3.16.1"
 description = "The best Python observability tool! 🪵🔥"
 requires-python = ">=3.8"
 authors = [