add terminate to JobRequest (livekit#2395)

Co-authored-by: Théo Monnom <theo.8bits@gmail.com>

Author: 2 people

livekit-agents/livekit/agents/job.py

@@ -494,7 +494,7 @@ def __init__(
         self,
         *,
         job: agent.Job,
-        on_reject: Callable[[], Coroutine[None, None, None]],
+        on_reject: Callable[[bool], Coroutine[None, None, None]],
         on_accept: Callable[[JobAcceptArguments], Coroutine[None, None, None]],
     ) -> None:
         self._job = job
@@ -523,8 +523,8 @@ def agent_name(self) -> str:
         return self._job.agent_name
 
     async def reject(self) -> None:
-        """Reject the job request. The job may be assigned to another worker"""
-        await self._on_reject()
+        """Reject the job request. The job will not be assigned to another worker"""
+        await self._on_reject(True)
 
     async def accept(
         self,

livekit-agents/livekit/agents/worker.py

@@ -820,13 +820,14 @@ async def _answer_availability(self, msg: agent.AvailabilityRequest) -> None:
 
         answered = False
 
-        async def _on_reject() -> None:
+        async def _on_reject(terminate: bool = False) -> None:
             nonlocal answered
             answered = True
 
             availability_resp = agent.WorkerMessage()
             availability_resp.availability.job_id = msg.job.id
             availability_resp.availability.available = False
+            availability_resp.availability.terminate = terminate
             await self._queue_msg(availability_resp)
 
         async def _on_accept(args: JobAcceptArguments) -> None: