[workflows][Feature] Retain function type info when used as a workflow step

[tekumara]

Search before asking

• [X] I had searched in the issues and found no similar feature requirement.

Description

When a function is annotated with @workflow.step and then later used, the types of function parameters are lost and return types become Any.

Use case

Improves developer experience eg: autocomplete in IDEs and type-checking

Related issues

No response

Are you willing to submit a PR?

• [ ] Yes I am willing to submit a PR!

[stale[bot]]

Hi, I'm a bot from the Ray team :)

To help human contributors to focus on more relevant issues, I will automatically add the stale label to issues that have had no activity for more than 4 months.

If there is no further activity in the 14 days, the issue will be closed!

• If you'd like to keep the issue open, just leave any comment, and the stale label will be removed!
• If you'd like to get more attention to the issue, please tag one of Ray's contributors.

You can always ask for help on our discussion forum or Ray's public slack channel.

[tekumara]

bump!

[stale[bot]]

Hi again! The issue will be closed because there has been no more activity in the 14 days since the last message.

Please feel free to reopen or open a new issue if you'd still like it to be addressed.

Again, you can always ask for help on our discussion forum or Ray's public slack channel.

Thanks again for opening the issue!