add trunk_config to WarmTransferTask for SIP endpoint transfers

examples/warm-transfer/warm_transfer.py

@@ -59,7 +59,7 @@ async def transfer_to_human(self) -> None:
             assert SUPERVISOR_PHONE_NUMBER is not None
 
             result = await WarmTransferTask(
-                target_phone_number=SUPERVISOR_PHONE_NUMBER,
+                sip_call_to=SUPERVISOR_PHONE_NUMBER,
                 sip_trunk_id=SIP_TRUNK_ID,
                 sip_number=SIP_NUMBER,
                 chat_ctx=self.chat_ctx,

livekit-agents/livekit/agents/beta/workflows/warm_transfer.py

@@ -66,10 +66,11 @@ class WarmTransferResult:
 class WarmTransferTask(AgentTask[WarmTransferResult]):
     def __init__(
         self,
-        target_phone_number: str,
+        sip_call_to: NotGivenOr[str] = NOT_GIVEN,
         *,
         hold_audio: NotGivenOr[AudioSource | AudioConfig | list[AudioConfig] | None] = NOT_GIVEN,
-        sip_trunk_id: NotGivenOr[str] = NOT_GIVEN,
+        sip_trunk_id: NotGivenOr[str | None] = NOT_GIVEN,
+        sip_connection: NotGivenOr[api.SIPOutboundConfig] = NOT_GIVEN,
         sip_number: NotGivenOr[str] = NOT_GIVEN,
         sip_headers: NotGivenOr[dict[str, str]] = NOT_GIVEN,
         extra_instructions: str = "",
@@ -81,6 +82,8 @@ def __init__(
         llm: NotGivenOr[llm.LLM | llm.RealtimeModel | None] = NOT_GIVEN,
         tts: NotGivenOr[tts.TTS | None] = NOT_GIVEN,
         allow_interruptions: NotGivenOr[bool] = NOT_GIVEN,
+        # deprecated
+        target_phone_number: NotGivenOr[str] = NOT_GIVEN,
     ) -> None:
         super().__init__(
             instructions=self.get_instructions(
@@ -101,13 +104,25 @@ def __init__(
         self._human_agent_failed_fut: asyncio.Future[None] = asyncio.Future()
         self._human_agent_identity = "human-agent-sip"
 
-        self._target_phone_number = target_phone_number
+        if target_phone_number:
+            logger.warning("`target_phone_number` is deprecated, use `sip_call_to` instead")
+            if not sip_call_to:
+                sip_call_to = target_phone_number
+
+        if not sip_call_to:
+            raise ValueError("`sip_call_to` must be set")
+
+        self._sip_call_to = sip_call_to
+        self._sip_connection = sip_connection if is_given(sip_connection) else None
         self._sip_trunk_id = (
-            sip_trunk_id if is_given(sip_trunk_id) else os.getenv("LIVEKIT_SIP_OUTBOUND_TRUNK", "")
+            sip_trunk_id
+            if is_given(sip_trunk_id)
+            else os.getenv("LIVEKIT_SIP_OUTBOUND_TRUNK", None)
         )
-        if not self._sip_trunk_id:
+        if self._sip_trunk_id is None and self._sip_connection is None:
             raise ValueError(
-                "`LIVEKIT_SIP_OUTBOUND_TRUNK` environment variable or `sip_trunk_id` argument must be set"
+                "`LIVEKIT_SIP_OUTBOUND_TRUNK` environment variable, `sip_trunk_id`,"
+                " or `sip_connection` must be set"
             )
 
         self._sip_number = (
@@ -301,17 +316,18 @@ async def _dial_human_agent(self) -> AgentSession:
         )
 
         # dial the human agent
-        await job_ctx.api.sip.create_sip_participant(
-            api.CreateSIPParticipantRequest(
-                sip_trunk_id=self._sip_trunk_id,
-                sip_call_to=self._target_phone_number,
-                room_name=human_agent_room_name,
-                participant_identity=self._human_agent_identity,
-                wait_until_answered=True,
-                sip_number=self._sip_number or None,
-                headers=self._sip_headers,
-            )
+        sip_request = api.CreateSIPParticipantRequest(
+            sip_trunk_id=self._sip_trunk_id,
+            sip_call_to=self._sip_call_to,
+            room_name=human_agent_room_name,
+            participant_identity=self._human_agent_identity,
+            wait_until_answered=True,
+            sip_number=self._sip_number or None,
+            headers=self._sip_headers,
         )
+        if self._sip_connection is not None:
+            sip_request.trunk.CopyFrom(self._sip_connection)
+        await job_ctx.api.sip.create_sip_participant(sip_request)
 
         return human_agent_sess