Use kwargs instead of output_prefix

Signed-off-by: Future-Outlier <[email protected]> Co-authored-by: pingsutw <[email protected]>
flyteorg · Jun 4, 2024 · 5bab3d3 · 5bab3d3
1 parent e3ca097
commit 5bab3d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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})