drop bargein base class

Author: chenghao-mou

examples/bargein_server.py

@@ -47,10 +47,7 @@ async def lifespan(app: FastAPI) -> AsyncIterator[None]:
 class BargeinRequest(BaseModel):
     """Request model for bargein detection."""
 
-    jobId: str = Field(..., description="Job ID from LiveKit")
-    workerId: str = Field(..., description="Worker ID from LiveKit")
     waveform: str = Field(..., description="Base64-encoded audio waveform (float32)")
-    agentId: str | None = Field(None, description="Optional agent ID")
     threshold: float = Field(0.95, description="Threshold for bargein detection")
     min_frames: int = Field(2, description="Minimum number of frames for bargein detection")
     created_at: float = Field(..., description="Timestamp of the audio waveform")
@@ -144,9 +141,6 @@ async def detect_bargein(request: BargeinRequest) -> BargeinResponse:
     Returns:
         BargeinResponse with detection result
     """
-    logger.info(
-        f"Received bargein detection request for job={request.jobId}, worker={request.workerId}"
-    )
 
     if onnx_session is None:
         raise HTTPException(

examples/voice_agents/basic_agent.py

@@ -11,10 +11,10 @@
     MetricsCollectedEvent,
     RunContext,
     cli,
+    inference,
     metrics,
     room_io,
 )
-from livekit.agents.inference.bargein import BargeinDetector
 from livekit.agents.llm import function_tool
 from livekit.plugins import silero
 from livekit.plugins.turn_detector.multilingual import MultilingualModel
@@ -37,7 +37,7 @@ def __init__(self) -> None:
             "you will speak english to the user",
         )
 
-    async def on_enter(self):
+    async def on_enter(self) -> None:
         # when the agent is added to the session, it'll generate a reply
         # according to its instructions
         self.session.generate_reply()
@@ -47,7 +47,7 @@ async def on_enter(self):
     @function_tool
     async def lookup_weather(
         self, context: RunContext, location: str, latitude: str, longitude: str
-    ):
+    ) -> str:
         """Called when the user asks for weather related information.
         Ensure the user's location (city or region) is provided.
         When given a location, please estimate the latitude and longitude of the location and
@@ -67,20 +67,20 @@ async def lookup_weather(
 server = AgentServer()
 
 
-def prewarm(proc: JobProcess):
+def prewarm(proc: JobProcess) -> None:
     proc.userdata["vad"] = silero.VAD.load()
 
 
 server.setup_fnc = prewarm
 
 
 @server.rtc_session()
-async def entrypoint(ctx: JobContext):
+async def entrypoint(ctx: JobContext) -> None:
     # each log entry will include these fields
     ctx.log_context_fields = {
         "room": ctx.room.name,
     }
-    session = AgentSession(
+    session: AgentSession = AgentSession(
         # Speech-to-text (STT) is your agent's ears, turning the user's speech into text that the LLM can understand
         # See all available models at https://docs.livekit.io/agents/models/stt/
         stt="deepgram/nova-3",
@@ -94,7 +94,7 @@ async def entrypoint(ctx: JobContext):
         # See more at https://docs.livekit.io/agents/build/turns
         turn_detection=MultilingualModel(),
         vad=ctx.proc.userdata["vad"],
-        bargein_detector=BargeinDetector(),
+        bargein_detector=inference.BargeinDetector(),
         # allow the LLM to generate a response while waiting for the end of turn
         # See more at https://docs.livekit.io/agents/build/audio/#preemptive-generation
         preemptive_generation=True,
@@ -108,11 +108,11 @@ async def entrypoint(ctx: JobContext):
     usage_collector = metrics.UsageCollector()
 
     @session.on("metrics_collected")
-    def _on_metrics_collected(ev: MetricsCollectedEvent):
+    def _on_metrics_collected(ev: MetricsCollectedEvent) -> None:
         metrics.log_metrics(ev.metrics)
         usage_collector.collect(ev.metrics)
 
-    async def log_usage():
+    async def log_usage() -> None:
         summary = usage_collector.get_summary()
         logger.info(f"Usage: {summary}")
 

livekit-agents/livekit/agents/bargein.py

@@ -1,3 +1,10 @@
+from .bargein import (
+    BargeinDetector,
+    BargeinError,
+    BargeinEvent,
+    BargeinEventType,
+    BargeinStreamBase,
+)
 from .llm import LLM, LLMModels, LLMStream
 from .stt import STT, STTModels
 from .tts import TTS, TTSModels
@@ -10,4 +17,9 @@
     "STTModels",
     "TTSModels",
     "LLMModels",
+    "BargeinDetector",
+    "BargeinStreamBase",
+    "BargeinEvent",
+    "BargeinError",
+    "BargeinEventType",
 ]