From cdd7e55a88ef0e8cf62099df8d1fd08582cc62ca Mon Sep 17 00:00:00 2001 From: CXwudi Date: Mon, 30 Sep 2024 21:14:21 -0400 Subject: [PATCH] chore: add missing models from Voyage (#8950) --- .../model_providers/voyage/rerank/_position.yaml | 4 ++++ .../model_providers/voyage/rerank/rerank-2.yaml | 4 ++++ .../model_providers/voyage/rerank/rerank-lite-2.yaml | 4 ++++ .../model_providers/voyage/text_embedding/_position.yaml | 6 ++++++ .../voyage/text_embedding/voyage-code-2.yaml | 8 ++++++++ .../voyage/text_embedding/voyage-finance-2.yaml | 8 ++++++++ .../voyage/text_embedding/voyage-law-2.yaml | 8 ++++++++ .../voyage/text_embedding/voyage-multilingual-2.yaml | 8 ++++++++ 8 files changed, 50 insertions(+) create mode 100644 api/core/model_runtime/model_providers/voyage/rerank/_position.yaml create mode 100644 api/core/model_runtime/model_providers/voyage/rerank/rerank-2.yaml create mode 100644 api/core/model_runtime/model_providers/voyage/rerank/rerank-lite-2.yaml create mode 100644 api/core/model_runtime/model_providers/voyage/text_embedding/_position.yaml create mode 100644 api/core/model_runtime/model_providers/voyage/text_embedding/voyage-code-2.yaml create mode 100644 api/core/model_runtime/model_providers/voyage/text_embedding/voyage-finance-2.yaml create mode 100644 api/core/model_runtime/model_providers/voyage/text_embedding/voyage-law-2.yaml create mode 100644 api/core/model_runtime/model_providers/voyage/text_embedding/voyage-multilingual-2.yaml diff --git a/api/core/model_runtime/model_providers/voyage/rerank/_position.yaml b/api/core/model_runtime/model_providers/voyage/rerank/_position.yaml new file mode 100644 index 000000000..32afefbe0 --- /dev/null +++ b/api/core/model_runtime/model_providers/voyage/rerank/_position.yaml @@ -0,0 +1,4 @@ +- rerank-2 +- rerank-lite-2 +- rerank-1 +- rerank-lite-1 diff --git a/api/core/model_runtime/model_providers/voyage/rerank/rerank-2.yaml b/api/core/model_runtime/model_providers/voyage/rerank/rerank-2.yaml new file mode 100644 index 000000000..b760d3c41 --- /dev/null +++ b/api/core/model_runtime/model_providers/voyage/rerank/rerank-2.yaml @@ -0,0 +1,4 @@ +model: rerank-2 +model_type: rerank +model_properties: + context_size: 16000 diff --git a/api/core/model_runtime/model_providers/voyage/rerank/rerank-lite-2.yaml b/api/core/model_runtime/model_providers/voyage/rerank/rerank-lite-2.yaml new file mode 100644 index 000000000..b6fa37a25 --- /dev/null +++ b/api/core/model_runtime/model_providers/voyage/rerank/rerank-lite-2.yaml @@ -0,0 +1,4 @@ +model: rerank-lite-2 +model_type: rerank +model_properties: + context_size: 8000 diff --git a/api/core/model_runtime/model_providers/voyage/text_embedding/_position.yaml b/api/core/model_runtime/model_providers/voyage/text_embedding/_position.yaml new file mode 100644 index 000000000..595663990 --- /dev/null +++ b/api/core/model_runtime/model_providers/voyage/text_embedding/_position.yaml @@ -0,0 +1,6 @@ +- voyage-3 +- voyage-3-lite +- voyage-finance-2 +- voyage-multilingual-2 +- voyage-law-2 +- voyage-code-2 diff --git a/api/core/model_runtime/model_providers/voyage/text_embedding/voyage-code-2.yaml b/api/core/model_runtime/model_providers/voyage/text_embedding/voyage-code-2.yaml new file mode 100644 index 000000000..693669c82 --- /dev/null +++ b/api/core/model_runtime/model_providers/voyage/text_embedding/voyage-code-2.yaml @@ -0,0 +1,8 @@ +model: voyage-code-2 +model_type: text-embedding +model_properties: + context_size: 16000 +pricing: + input: '0.00012' + unit: '0.001' + currency: USD diff --git a/api/core/model_runtime/model_providers/voyage/text_embedding/voyage-finance-2.yaml b/api/core/model_runtime/model_providers/voyage/text_embedding/voyage-finance-2.yaml new file mode 100644 index 000000000..555e11002 --- /dev/null +++ b/api/core/model_runtime/model_providers/voyage/text_embedding/voyage-finance-2.yaml @@ -0,0 +1,8 @@ +model: voyage-finance-2 +model_type: text-embedding +model_properties: + context_size: 32000 +pricing: + input: '0.00012' + unit: '0.001' + currency: USD diff --git a/api/core/model_runtime/model_providers/voyage/text_embedding/voyage-law-2.yaml b/api/core/model_runtime/model_providers/voyage/text_embedding/voyage-law-2.yaml new file mode 100644 index 000000000..032693286 --- /dev/null +++ b/api/core/model_runtime/model_providers/voyage/text_embedding/voyage-law-2.yaml @@ -0,0 +1,8 @@ +model: voyage-law-2 +model_type: text-embedding +model_properties: + context_size: 16000 +pricing: + input: '0.00012' + unit: '0.001' + currency: USD diff --git a/api/core/model_runtime/model_providers/voyage/text_embedding/voyage-multilingual-2.yaml b/api/core/model_runtime/model_providers/voyage/text_embedding/voyage-multilingual-2.yaml new file mode 100644 index 000000000..9ecf4d500 --- /dev/null +++ b/api/core/model_runtime/model_providers/voyage/text_embedding/voyage-multilingual-2.yaml @@ -0,0 +1,8 @@ +model: voyage-multilingual-2 +model_type: text-embedding +model_properties: + context_size: 32000 +pricing: + input: '0.00012' + unit: '0.001' + currency: USD