From 4b1d193f4dd2d2e179ab98841207b2829ca9d262 Mon Sep 17 00:00:00 2001 From: Yann Stoneman Date: Sun, 21 Apr 2024 18:33:00 -0400 Subject: [PATCH] crossencodermodels providers for config props --- lib/rag-engines/sagemaker-rag-models/index.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/rag-engines/sagemaker-rag-models/index.ts b/lib/rag-engines/sagemaker-rag-models/index.ts index bfc185777..a7bec0c9a 100644 --- a/lib/rag-engines/sagemaker-rag-models/index.ts +++ b/lib/rag-engines/sagemaker-rag-models/index.ts @@ -22,7 +22,7 @@ export class SageMakerRagModels extends Construct { .map((c) => c.name); const sageMakerCrossEncoderModelIds = props.config.rag.crossEncoderModels - .filter((c) => c.provider === "sagemaker") + .filter((c) => c.provider === "sagemaker" || c.provider === "cohere") .map((c) => c.name); const model = new SageMakerModel(this, "Model", {