diff --git a/plugins/flytekit-openai/flytekitplugins/openai/chatgpt/agent.py b/plugins/flytekit-openai/flytekitplugins/openai/chatgpt/agent.py
index 1bf0f6a485..e4f24baa5a 100644
--- a/plugins/flytekit-openai/flytekitplugins/openai/chatgpt/agent.py
+++ b/plugins/flytekit-openai/flytekitplugins/openai/chatgpt/agent.py
@@ -27,7 +27,7 @@ async def do(
         self,
         task_template: TaskTemplate,
         inputs: Optional[LiteralMap] = None,
-        output_prefix: Optional[str] = None,
+        **kwargs,
     ) -> Resource:
         ctx = FlyteContextManager.current_context()
         input_python_value = TypeEngine.literal_map_to_kwargs(ctx, inputs, {"message": str})