Merge pull request #614 from narengogi/chore/support-additional-param…

…s-cohere-embed Chore/support additional params cohere embed
Portkey-AI · Sep 16, 2024 · b12ce10 · b12ce10
2 parents 699a63f + 8ca2491
commit b12ce10
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/providers/cohere/embed.ts b/src/providers/cohere/embed.ts
@@ -19,6 +19,18 @@ export const CohereEmbedConfig: ProviderConfig = {
     param: 'model',
     default: 'embed-english-light-v2.0',
   },
+  input_type: {
+    param: 'input_type',
+    required: false,
+  },
+  embedding_types: {
+    param: 'embedding_types',
+    required: false,
+  },
+  truncate: {
+    param: 'truncate',
+    required: false,
+  },
 };
 
 /**