Fixes bugs in the Cohere Blueprint (opensearch-project#1351)

* Fixes blueprint bugs found in opensearch-project#1344 Signed-off-by: David Tippett <[email protected]> * fixing indentation and adding bin files rule to git ignore Signed-off-by: David Tippett <[email protected]> * reverting change from prompts -> texts Signed-off-by: David Tippett <[email protected]> --------- Signed-off-by: David Tippett <[email protected]> Signed-off-by: TrungBui59 <[email protected]>
TrungBui59 · Nov 21, 2023 · de44ff5 · de44ff5
1 parent cc61c88
commit de44ff5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -6,3 +6,4 @@ common/build/
 ml-algorithms/build/
 plugin/build/
 .DS_Store
+*/bin/
diff --git a/docs/remote_inference_blueprints/cohere_connector_embedding_blueprint.md b/docs/remote_inference_blueprints/cohere_connector_embedding_blueprint.md
@@ -24,7 +24,9 @@ POST /_plugins/_ml/connectors/_create
       "headers": {
         "Authorization": "Bearer ${credential.cohere_key}"
       },
-      "request_body": "{ \"texts\": ${parameters.texts}, \"truncate\": \"${parameters.truncate}\", \"model\": \"${parameters.model}\" }"
+      "request_body": "{ \"texts\": ${parameters.texts}, \"truncate\": \"${parameters.truncate}\", \"model\": \"${parameters.model}\" }", 
+      "pre_process_function": "connector.pre_process.cohere.embedding",
+      "post_process_function": "connector.post_process.cohere.embedding"
     }
   ]
 }