github/autogen
1
1
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-04-20 03:02:16 -04:00
Code Issues Packages Projects Releases Wiki Activity

Update _worker_runtime_host_servicer.py (#557)

Browse Source 
Updated _process_request to add to _pending_requests waiting for a response from the target, rather than the source of the request.
This commit is contained in:
Taylor Rockey
2024-09-18 11:19:17 -07:00
committed by GitHub
parent 4998ab5398
commit a479a5e692
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/packages/autogen-core/src/autogen_core/application/_worker_runtime_host_servicer.py
View File

@@ -135,9 +135,9 @@ class WorkerAgentRuntimeHostServicer(agent_worker_pb2_grpc.AgentRpcServicer):
return
await target_send_queue.put(agent_worker_pb2.Message(request=request))
# Create a future to wait for the response.
# Create a future to wait for the response from the target.
future = asyncio.get_event_loop().create_future()
self._pending_requests.setdefault(client_id, {})[request.request_id] = future
self._pending_requests.setdefault(target_client_id, {})[request.request_id] = future
# Create a task to wait for the response and send it back to the client.
send_response_task = asyncio.create_task(self._wait_and_send_response(future, client_id))