revert worker changes

Signed-off-by: Uri Shaket <ushaket@redhat.com>

Author: ushaket

src/guidellm/scheduler/worker.py

@@ -370,6 +370,7 @@ async def _process_next_request(self, target_start: float):
                 if request_info is None:
                     raise RuntimeError("Received invalid request info from backend")
 
+            # Complete the request
             request_info.timings.resolve_end = time.time()
             self._send_update("completed", response, request, request_info)
 

tests/unit/scheduler/test_worker.py

@@ -18,10 +18,7 @@
     SynchronousStrategy,
     WorkerProcess,
 )
-from guidellm.schemas import (
-    RequestInfo,
-    RequestTimings,
-)
+from guidellm.schemas import RequestInfo, RequestTimings
 from guidellm.utils import InterProcessMessagingQueue
 from tests.unit.testing_utils import async_timeout
 
@@ -53,15 +50,11 @@ def __init__(
         resolve_delay: float = 0.0,
         should_fail: bool = False,
         request_error_rate: float = 0.0,
-        should_yield_response: bool = True,
-        response_payload: Any | None = None,
     ):
         self.lifecycle_delay = lifecycle_delay
         self.resolve_delay = resolve_delay
         self.should_fail = should_fail
         self.request_error_rate = request_error_rate
-        self.should_yield_response = should_yield_response
-        self.response_payload = response_payload
         self.process_startup_called = False
         self.validate_called = False
         self.process_shutdown_called = False
@@ -106,13 +99,7 @@ async def resolve(self, request, request_info, request_history):
             raise RuntimeError("Mock resolve failed")
         if self.request_error_rate > 0.0 and random.random() < self.request_error_rate:
             raise RuntimeError("Mock resolve failed")
-        if self.should_yield_response:
-            payload = (
-                self.response_payload
-                if self.response_payload is not None
-                else f"response_for_{request}"
-            )
-            yield payload, request_info
+        yield f"response_for_{request}", request_info
 
 
 class TestWorkerProcess: