I have a small PR to improve agent-service plugin. Please take a look and let me know what you think. Thank you. https://github.com/flyteorg/flyteplugins/pull/368