feat: allow union as input types

[meskill]

Summary:
Briefly describe the changes made in this PR.

Issue Reference(s):
Fixes #1990

Build & Testing:

• I ran cargo test successfully.
• I have run ./lint.sh --mode=fix to fix all linting issues raised by ./lint.sh --mode=check.

Checklist:

• I have added relevant unit & integration tests.
• I have updated the documentation accordingly.
• I have performed a self-review of my code.
• PR follows the naming convention of <type>(<optional scope>): <title>

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 85.90%. Comparing base (8a7dd73) to head (68b790f).
Report is 2 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #2221      +/-   ##
==========================================
+ Coverage   85.74%   85.90%   +0.15%     
==========================================
  Files         219      223       +4     
  Lines       21022    21315     +293     
==========================================
+ Hits        18025    18310     +285     
- Misses       2997     3005       +8     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[github-actions]

Action required: PR inactive for 2 days.
Status update or closure in 5 days.