@@ -149,15 +149,15 @@ public class RestMLRAGSearchProcessorIT extends RestMLRemoteInferenceIT {
149
149
? BEDROCK_CONNECTOR_BLUEPRINT2
150
150
: BEDROCK_CONNECTOR_BLUEPRINT1 ;
151
151
152
- private static final String COHERE_API_KEY = System .getenv ("COHERE_API_KEY " );
152
+ private static final String COHERE_KEY = System .getenv ("COHERE_KEY " );
153
153
private static final String COHERE_CONNECTOR_BLUEPRINT = "{\n "
154
154
+ " \" name\" : \" Cohere Chat Model\" ,\n "
155
155
+ " \" description\" : \" The connector to Cohere's public chat API\" ,\n "
156
156
+ " \" version\" : \" 1\" ,\n "
157
157
+ " \" protocol\" : \" http\" ,\n "
158
158
+ " \" credential\" : {\n "
159
159
+ " \" cohere_key\" : \" "
160
- + COHERE_API_KEY
160
+ + COHERE_KEY
161
161
+ "\" \n "
162
162
+ " },\n "
163
163
+ " \" parameters\" : {\n "
@@ -521,7 +521,7 @@ public void testBM25WithBedrockWithConversation() throws Exception {
521
521
522
522
public void testBM25WithCohere () throws Exception {
523
523
// Skip test if key is null
524
- if (COHERE_API_KEY == null ) {
524
+ if (COHERE_KEY == null ) {
525
525
return ;
526
526
}
527
527
Response response = createConnector (COHERE_CONNECTOR_BLUEPRINT );
@@ -573,7 +573,7 @@ public void testBM25WithCohere() throws Exception {
573
573
574
574
public void testBM25WithCohereUsingLlmResponseField () throws Exception {
575
575
// Skip test if key is null
576
- if (COHERE_API_KEY == null ) {
576
+ if (COHERE_KEY == null ) {
577
577
return ;
578
578
}
579
579
Response response = createConnector (COHERE_CONNECTOR_BLUEPRINT );
@@ -660,8 +660,6 @@ private Response performSearch(String indexName, String pipeline, int size, Sear
660
660
requestParameters .match ,
661
661
requestParameters .llmModel ,
662
662
requestParameters .llmQuestion ,
663
- requestParameters .systemPrompt ,
664
- requestParameters .userInstructions ,
665
663
requestParameters .contextSize ,
666
664
requestParameters .interactionSize ,
667
665
requestParameters .timeout ,
0 commit comments