Fix and Refactor Test with Builder Pattern for Issue #2825

[junweid62]

Description

Ensured compatibility with the new parameter added in opensearch-project/ml-commons#2825.

Related Issues

Resolves #858

Should fix compile errors causing #856.

Check List

• [x] New functionality includes testing. (Fixes a test!)
• [x] Commits are signed per the DCO using --signoff.

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license. For more information on following Developer Certificate of Origin and signing off your commits, please check here.

[codecov[bot]]

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Project coverage is 77.55%. Comparing base (bbd089b) to head (0c7c77a). Report is 1 commits behind head on main.

Additional details and impacted files

```diff @@ Coverage Diff @@ ## main #859 +/- ## ========================================= Coverage 77.55% 77.55% Complexity 966 966 ========================================= Files 97 97 Lines 4531 4531 Branches 422 422 ========================================= Hits 3514 3514 Misses 835 835 Partials 182 182 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.