Externalize MODEL_NAME since OS endpoint is slightly different

bo-lu · bo-lu · commit f54e6ca16796 · 2024-12-31T12:01:34.000-05:00
diff --git a/docs/cloudformation/geocore-semantic-search-with-opensearch.yml b/docs/cloudformation/geocore-semantic-search-with-opensearch.yml
@@ -228,6 +228,10 @@ Resources:
           OS_ENDPOINT:  !Ref OSEndpoint
           OS_SECRET_ID: !Ref OSSecretID
           SAGEMAKER_ENDPOINT: !Ref SagemakerEndpoint
+          MODEL_NAME: !If 
+            - IsProd
+            - minilm-knn
+            - minilm-pretrain-knn
       PackageType: Zip
       Role: !GetAtt InvokeSagemakerLambdaExecutionRole.Arn
       Events:
