pull/9711/head
Nuno Campos 1 year ago
parent 4eecf90f33
commit 85088dc5df

@ -65,10 +65,7 @@ class RunnableRetry(RunnableBinding[Input, Output]):
return self._patch_config(config, retry_state) return self._patch_config(config, retry_state)
def invoke( def invoke(
self, self, input: Input, config: Optional[RunnableConfig] = None, **kwargs: Any
input: Input,
config: Optional[RunnableConfig] = None,
**kwargs: Any | None
) -> Output: ) -> Output:
for attempt in self._sync_retrying(): for attempt in self._sync_retrying():
with attempt: with attempt:
@ -80,7 +77,7 @@ class RunnableRetry(RunnableBinding[Input, Output]):
return result return result
async def ainvoke( async def ainvoke(
self, input: Input, config: RunnableConfig | None = None, **kwargs: Any | None self, input: Input, config: Optional[RunnableConfig] = None, **kwargs: Any
) -> Output: ) -> Output:
async for attempt in self._async_retrying(): async for attempt in self._async_retrying():
with attempt: with attempt:

Loading…
Cancel
Save