File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed
google/cloud/aiplatform_v1/services/reasoning_engine_execution_service Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -410,7 +410,7 @@ async def sample_query_reasoning_engine():
410
410
# Done; return the response.
411
411
return response
412
412
413
- def stream_query_reasoning_engine (
413
+ async def stream_query_reasoning_engine (
414
414
self ,
415
415
request : Optional [
416
416
Union [
@@ -543,7 +543,7 @@ async def sample_stream_query_reasoning_engine():
543
543
self ._client ._validate_universe_domain ()
544
544
545
545
# Send the request.
546
- response = rpc (
546
+ response = await rpc (
547
547
request ,
548
548
retry = retry ,
549
549
timeout = timeout ,
Original file line number Diff line number Diff line change @@ -1533,14 +1533,14 @@ def _wrap_async_stream_query_operation(
1533
1533
"""
1534
1534
1535
1535
async def _method (self , ** kwargs ) -> AsyncIterable [Any ]:
1536
- response = self .execution_api_client .stream_query_reasoning_engine (
1536
+ response = await self .execution_async_client .stream_query_reasoning_engine (
1537
1537
request = aip_types .StreamQueryReasoningEngineRequest (
1538
1538
name = self .resource_name ,
1539
1539
input = kwargs ,
1540
1540
class_method = method_name ,
1541
1541
),
1542
1542
)
1543
- for chunk in response :
1543
+ async for chunk in response :
1544
1544
for parsed_json in _utils .yield_parsed_json (chunk ):
1545
1545
if parsed_json is not None :
1546
1546
yield parsed_json
You can’t perform that action at this time.
0 commit comments