@@ -133,32 +133,33 @@ def _resolve(
133133 evaluation_context : EvaluationContext ,
134134 ):
135135 context = self ._convert_context (evaluation_context )
136+ call_args = {"timeout" : self .timeout }
136137 try :
137138 if flag_type == FlagType .BOOLEAN :
138139 request = schema_pb2 .ResolveBooleanRequest (
139140 flag_key = flag_key , context = context
140141 )
141- response = self .stub .ResolveBoolean (request )
142+ response = self .stub .ResolveBoolean (request , ** call_args )
142143 elif flag_type == FlagType .STRING :
143144 request = schema_pb2 .ResolveStringRequest (
144145 flag_key = flag_key , context = context
145146 )
146- response = self .stub .ResolveString (request )
147+ response = self .stub .ResolveString (request , ** call_args )
147148 elif flag_type == FlagType .OBJECT :
148149 request = schema_pb2 .ResolveObjectRequest (
149150 flag_key = flag_key , context = context
150151 )
151- response = self .stub .ResolveObject (request )
152+ response = self .stub .ResolveObject (request , ** call_args )
152153 elif flag_type == FlagType .FLOAT :
153154 request = schema_pb2 .ResolveFloatRequest (
154155 flag_key = flag_key , context = context
155156 )
156- response = self .stub .ResolveFloat (request )
157+ response = self .stub .ResolveFloat (request , ** call_args )
157158 elif flag_type == FlagType .INTEGER :
158159 request = schema_pb2 .ResolveIntRequest (
159160 flag_key = flag_key , context = context
160161 )
161- response = self .stub .ResolveInt (request )
162+ response = self .stub .ResolveInt (request , ** call_args )
162163 else :
163164 raise ValueError (f"Unknown flag type: { flag_type } " )
164165
0 commit comments