Skip to content

Commit 9fe9842

Browse files
authored
Merge pull request #789 from narengogi/chore/sagemaker/model-name
add optional model name header for sagemaker
2 parents a66c3a3 + e4c7b9b commit 9fe9842

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/handlers/handlerUtils.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -786,6 +786,8 @@ export function constructConfigFromRequestHeaders(
786786
requestHeaders[`x-${POWERED_BY}-amzn-sagemaker-inference-component`],
787787
amznSagemakerSessionId:
788788
requestHeaders[`x-${POWERED_BY}-amzn-sagemaker-session-id`],
789+
amznSagemakerModelName:
790+
requestHeaders[`x-${POWERED_BY}-amzn-sagemaker-model-name`],
789791
};
790792

791793
const workersAiConfig = {

0 commit comments

Comments
 (0)