Remove cohere rerank from *sagemaker* cross-encoders

ystoneman · May 9, 2024 · 69cdc88 · 69cdc88
1 parent 57dc67c
commit 69cdc88
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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" || c.provider === "cohere")
+      .filter((c) => c.provider === "sagemaker")
       .map((c) => c.name);
 
     const model = new SageMakerModel(this, "Model", {