Commit a49c605f authored by myhloli's avatar myhloli
Browse files

fix: update request handling to pass state from send_one_request to wait_one_response

parent 55cb4c90
...@@ -183,8 +183,8 @@ async def _one_request( ...@@ -183,8 +183,8 @@ async def _one_request(
created_time: Optional[float], created_time: Optional[float],
): ):
tokenized_obj = await self._tokenize_one_request(obj) tokenized_obj = await self._tokenize_one_request(obj)
self._send_one_request(obj, tokenized_obj, created_time) state = self._send_one_request(obj, tokenized_obj, created_time)
async for out in self._wait_one_response(obj, request): async for out in self._wait_one_response(obj, state, request):
yield out yield out
...@@ -256,8 +256,8 @@ async def _generate_request( ...@@ -256,8 +256,8 @@ async def _generate_request(
is_single = obj.is_single is_single = obj.is_single
if is_single: if is_single:
tokenized_obj = await self._tokenize_one_request(obj) tokenized_obj = await self._tokenize_one_request(obj)
self._send_one_request(obj, tokenized_obj, created_time) state = self._send_one_request(obj, tokenized_obj, created_time)
async for response in self._wait_one_response(obj, request): async for response in self._wait_one_response(obj, state, request):
yield response yield response
else: else:
async for response in _handle_batch_request(self, obj, request, created_time): async for response in _handle_batch_request(self, obj, request, created_time):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment