1717import dataclasses
1818import json # type: ignore
1919import re
20- from typing import Callable , Dict , List , Optional , Sequence , Tuple , Union
20+ from typing import Any , Callable , Dict , List , Optional , Sequence , Tuple , Union
2121import warnings
2222
2323from google .api_core import (
@@ -112,7 +112,7 @@ def pre_get_location(
112112 self ,
113113 request : locations_pb2 .GetLocationRequest ,
114114 metadata : Sequence [Tuple [str , str ]],
115- ) -> locations_pb2 .Location :
115+ ) -> Tuple [ locations_pb2 .GetLocationRequest , Sequence [ Tuple [ str , str ]]] :
116116 """Pre-rpc interceptor for get_location
117117
118118 Override in a subclass to manipulate the request or metadata
@@ -121,7 +121,7 @@ def pre_get_location(
121121 return request , metadata
122122
123123 def post_get_location (
124- self , response : locations_pb2 .GetLocationRequest
124+ self , response : locations_pb2 .Location
125125 ) -> locations_pb2 .Location :
126126 """Post-rpc interceptor for get_location
127127
@@ -135,7 +135,7 @@ def pre_list_locations(
135135 self ,
136136 request : locations_pb2 .ListLocationsRequest ,
137137 metadata : Sequence [Tuple [str , str ]],
138- ) -> locations_pb2 .ListLocationsResponse :
138+ ) -> Tuple [ locations_pb2 .ListLocationsRequest , Sequence [ Tuple [ str , str ]]] :
139139 """Pre-rpc interceptor for list_locations
140140
141141 Override in a subclass to manipulate the request or metadata
@@ -144,7 +144,7 @@ def pre_list_locations(
144144 return request , metadata
145145
146146 def post_list_locations (
147- self , response : locations_pb2 .ListLocationsRequest
147+ self , response : locations_pb2 .ListLocationsResponse
148148 ) -> locations_pb2 .ListLocationsResponse :
149149 """Post-rpc interceptor for list_locations
150150
@@ -158,17 +158,15 @@ def pre_cancel_operation(
158158 self ,
159159 request : operations_pb2 .CancelOperationRequest ,
160160 metadata : Sequence [Tuple [str , str ]],
161- ) -> None :
161+ ) -> Tuple [ operations_pb2 . CancelOperationRequest , Sequence [ Tuple [ str , str ]]] :
162162 """Pre-rpc interceptor for cancel_operation
163163
164164 Override in a subclass to manipulate the request or metadata
165165 before they are sent to the WorkflowsServiceV2Beta server.
166166 """
167167 return request , metadata
168168
169- def post_cancel_operation (
170- self , response : operations_pb2 .CancelOperationRequest
171- ) -> None :
169+ def post_cancel_operation (self , response : None ) -> None :
172170 """Post-rpc interceptor for cancel_operation
173171
174172 Override in a subclass to manipulate the response
@@ -181,7 +179,7 @@ def pre_get_operation(
181179 self ,
182180 request : operations_pb2 .GetOperationRequest ,
183181 metadata : Sequence [Tuple [str , str ]],
184- ) -> operations_pb2 .Operation :
182+ ) -> Tuple [ operations_pb2 .GetOperationRequest , Sequence [ Tuple [ str , str ]]] :
185183 """Pre-rpc interceptor for get_operation
186184
187185 Override in a subclass to manipulate the request or metadata
@@ -190,7 +188,7 @@ def pre_get_operation(
190188 return request , metadata
191189
192190 def post_get_operation (
193- self , response : operations_pb2 .GetOperationRequest
191+ self , response : operations_pb2 .Operation
194192 ) -> operations_pb2 .Operation :
195193 """Post-rpc interceptor for get_operation
196194
@@ -204,7 +202,7 @@ def pre_list_operations(
204202 self ,
205203 request : operations_pb2 .ListOperationsRequest ,
206204 metadata : Sequence [Tuple [str , str ]],
207- ) -> operations_pb2 .ListOperationsResponse :
205+ ) -> Tuple [ operations_pb2 .ListOperationsRequest , Sequence [ Tuple [ str , str ]]] :
208206 """Pre-rpc interceptor for list_operations
209207
210208 Override in a subclass to manipulate the request or metadata
@@ -213,7 +211,7 @@ def pre_list_operations(
213211 return request , metadata
214212
215213 def post_list_operations (
216- self , response : operations_pb2 .ListOperationsRequest
214+ self , response : operations_pb2 .ListOperationsResponse
217215 ) -> operations_pb2 .ListOperationsResponse :
218216 """Post-rpc interceptor for list_operations
219217
@@ -373,7 +371,7 @@ class _RunPipeline(WorkflowsServiceV2BetaRestStub):
373371 def __hash__ (self ):
374372 return hash ("RunPipeline" )
375373
376- __REQUIRED_FIELDS_DEFAULT_VALUES : Dict [str , str ] = {}
374+ __REQUIRED_FIELDS_DEFAULT_VALUES : Dict [str , Any ] = {}
377375
378376 @classmethod
379377 def _get_unset_required_fields (cls , message_dict ):
0 commit comments