-
Notifications
You must be signed in to change notification settings - Fork 140
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
enhance: make response_field customizable in MLModelTool #2007
enhance: make response_field customizable in MLModelTool #2007
Conversation
Signed-off-by: zhichao-aws <[email protected]>
Codecov ReportAll modified and coverable lines are covered by tests ✅
Additional details and impacted files@@ Coverage Diff @@
## main #2007 +/- ##
============================================
- Coverage 82.57% 82.56% -0.02%
+ Complexity 5577 5575 -2
============================================
Files 540 540
Lines 22453 22456 +3
Branches 2286 2286
============================================
Hits 18540 18540
- Misses 2981 2985 +4
+ Partials 932 931 -1
Flags with carried forward coverage won't be shown. Click here to find out more. ☔ View full report in Codecov by Sentry. |
@ylwu-amzn @dhrubo-os Do you think this PR is enough to fix this problem? Please help review thanks |
Signed-off-by: zhichao-aws <[email protected]>
Signed-off-by: zhichao-aws <[email protected]>
* enhance: make response_field customizable in MLModelTool Signed-off-by: zhichao-aws <[email protected]> * ut for malformed response field Signed-off-by: zhichao-aws <[email protected]> * remove import * in MLModelToolTests Signed-off-by: zhichao-aws <[email protected]> --------- Signed-off-by: zhichao-aws <[email protected]> (cherry picked from commit e1d21ba)
* enhance: make response_field customizable in MLModelTool Signed-off-by: zhichao-aws <[email protected]> * ut for malformed response field Signed-off-by: zhichao-aws <[email protected]> * remove import * in MLModelToolTests Signed-off-by: zhichao-aws <[email protected]> --------- Signed-off-by: zhichao-aws <[email protected]> (cherry picked from commit e1d21ba)
* enhance: make response_field customizable in MLModelTool Signed-off-by: zhichao-aws <[email protected]> * ut for malformed response field Signed-off-by: zhichao-aws <[email protected]> * remove import * in MLModelToolTests Signed-off-by: zhichao-aws <[email protected]> --------- Signed-off-by: zhichao-aws <[email protected]> (cherry picked from commit e1d21ba) Co-authored-by: zhichao-aws <[email protected]>
* enhance: make response_field customizable in MLModelTool Signed-off-by: zhichao-aws <[email protected]> * ut for malformed response field Signed-off-by: zhichao-aws <[email protected]> * remove import * in MLModelToolTests Signed-off-by: zhichao-aws <[email protected]> --------- Signed-off-by: zhichao-aws <[email protected]> (cherry picked from commit e1d21ba) Co-authored-by: zhichao-aws <[email protected]>
…project#2007) * enhance: make response_field customizable in MLModelTool Signed-off-by: zhichao-aws <[email protected]> * ut for malformed response field Signed-off-by: zhichao-aws <[email protected]> * remove import * in MLModelToolTests Signed-off-by: zhichao-aws <[email protected]> --------- Signed-off-by: zhichao-aws <[email protected]>
Description
Current MLModelTool hard code to fetch the
response
field from response body. However, in other non-LLM cases or non-Claude LLM services, we can not assume all of them useresponse
keyword.This PR makes response_field customizable in MLModelTool. If not provided, it will use
response
by default. This will make current API invokation behavior consistent.Issues Resolved
[List any issues this PR will resolve]
Check List
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.