Skip to content

Commit cb66cd9

Browse files
committed
feat: allow custom Cohere rerank model names
Change the Model Name field from a hardcoded dropdown to a free-text input with a default of rerank-v3.5. This allows users with custom fine-tuned Cohere rerank models to enter their model name directly. Fixes #4086 Signed-off-by: Vedant Madane <6527493+VedantMadane@users.noreply.github.com>
1 parent 02813aa commit cb66cd9

1 file changed

Lines changed: 4 additions & 15 deletions

File tree

packages/components/nodes/retrievers/CohereRerankRetriever/CohereRerankRetriever.ts

Lines changed: 4 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -43,21 +43,10 @@ class CohereRerankRetriever_Retrievers implements INode {
4343
{
4444
label: 'Model Name',
4545
name: 'model',
46-
type: 'options',
47-
options: [
48-
{
49-
label: 'rerank-v3.5',
50-
name: 'rerank-v3.5'
51-
},
52-
{
53-
label: 'rerank-english-v3.0',
54-
name: 'rerank-english-v3.0'
55-
},
56-
{
57-
label: 'rerank-multilingual-v3.0',
58-
name: 'rerank-multilingual-v3.0'
59-
}
60-
],
46+
type: 'string',
47+
placeholder: 'rerank-v3.5',
48+
description:
49+
'Name of the Cohere rerank model. Defaults to rerank-v3.5. Supports custom fine-tuned rerank models.',
6150
default: 'rerank-v3.5',
6251
optional: true
6352
},

0 commit comments

Comments
 (0)