Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 0 additions & 3 deletions py/src/braintrust/integrations/livekit_agents/__init__.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,8 @@
"""Braintrust integration for LiveKit Agents."""

from .integration import LiveKitAgentsIntegration
from .patchers import wrap_livekit_agents


def setup_livekit_agents() -> bool:
"""Register the Braintrust LiveKit Agents OpenTelemetry span processor."""
return LiveKitAgentsIntegration.setup()


Expand Down
2 changes: 0 additions & 2 deletions py/src/braintrust/integrations/livekit_agents/integration.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,3 @@
"""LiveKit Agents integration."""

from braintrust.integrations.base import BaseIntegration

from .patchers import AgentSessionPatcher, MetricsEmitterPatcher
Expand Down
8 changes: 0 additions & 8 deletions py/src/braintrust/integrations/livekit_agents/patchers.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,3 @@
"""Direct LiveKit Agents patchers."""

from typing import Any

from braintrust.integrations.base import (
Expand Down Expand Up @@ -211,12 +209,6 @@ class MetricsEmitterPatcher(CompositeFunctionWrapperPatcher):


def wrap_livekit_agents(target: Any) -> Any:
"""Instrument a LiveKit Agents class or instance directly.

This helper is useful when global setup is not desired. It applies any
LiveKit Agents patchers whose target method exists on ``target`` and returns
``target`` for convenient chaining.
"""
AgentSessionPatcher.wrap_target(target)
MetricsEmitterPatcher.wrap_target(target)
return target
Loading