updating cohere blueprint doc (#1213)

* updating cohere blueprint doc Signed-off-by: Dhrubo Saha <dhrubo@amazon.com> * fixing the blueprint Signed-off-by: Dhrubo Saha <dhrubo@amazon.com> --------- Signed-off-by: Dhrubo Saha <dhrubo@amazon.com>
opensearch-project · Sep 1, 2023 · a72b204 · a72b204
1 parent 1bd9d80
commit a72b204
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 105 deletions.
diff --git a/docs/remote_inference_blueprints/cohere_connector_embedding_blueprint.md b/docs/remote_inference_blueprints/cohere_connector_embedding_blueprint.md
@@ -12,6 +12,10 @@ POST /_plugins/_ml/connectors/_create
   "credential": {
     "cohere_key": "<PLEASE ADD YOUR Cohere API KEY HERE>"
   },
+  "parameters": {
+    "model": "embed-english-v2.0",
+    "truncate": "END"
+  },
   "actions": [
     {
       "action_type": "predict",
@@ -20,7 +24,7 @@ POST /_plugins/_ml/connectors/_create
       "headers": {
         "Authorization": "Bearer ${credential.cohere_key}"
       },
-      "request_body": "{ \"texts\": ${parameters.prompt}, \"truncate\": \"END\" }"
+      "request_body": "{ \"texts\": ${parameters.texts}, \"truncate\": \"${parameters.truncate}\", \"model\": \"${parameters.model}\" }"
     }
   ]
 }
@@ -39,7 +43,7 @@ POST /_plugins/_ml/connectors/_create
 POST /_plugins/_ml/models/<ENTER MODEL ID HERE>/_predict
 {
   "parameters": {
-    "prompt": ["Say this is a test"]
+    "texts": ["Say this is a test"]
   }
 }
 ```

diff --git a/docs/remote_inference_blueprints/cohere_connector_rerank_blueprint.md b/docs/remote_inference_blueprints/cohere_connector_rerank_blueprint.md