This commit is contained in:
2023-05-21 01:59:17 +02:00
parent 4d3aca532f
commit cd6f28cdbc

View File

@@ -14,6 +14,8 @@ class FastAPIREtryMiddleware(SimpleRetryMiddleware):
) -> dict[str, Any]: ) -> dict[str, Any]:
sig = signature(task_func) sig = signature(task_func)
keys_to_remove = []
for key in message_kwargs.keys(): for key in message_kwargs.keys():
param = sig.parameters.get(key, None) param = sig.parameters.get(key, None)
@@ -21,6 +23,9 @@ class FastAPIREtryMiddleware(SimpleRetryMiddleware):
continue continue
if isinstance(param.default, Dependency): if isinstance(param.default, Dependency):
keys_to_remove.append(key)
for key in keys_to_remove:
message_kwargs.pop(key) message_kwargs.pop(key)
return message_kwargs return message_kwargs