From 8be7d8a63586682c23d9d3bfe1943656093b5280 Mon Sep 17 00:00:00 2001 From: Chenhe Gu Date: Fri, 26 Jan 2024 04:48:20 +0800 Subject: [PATCH] Add new OpenAI embedding models (#2217) --- .../openai/text_embedding/text-embedding-3-large.yaml | 9 +++++++++ .../openai/text_embedding/text-embedding-3-small.yaml | 9 +++++++++ 2 files changed, 18 insertions(+) create mode 100644 api/core/model_runtime/model_providers/openai/text_embedding/text-embedding-3-large.yaml create mode 100644 api/core/model_runtime/model_providers/openai/text_embedding/text-embedding-3-small.yaml diff --git a/api/core/model_runtime/model_providers/openai/text_embedding/text-embedding-3-large.yaml b/api/core/model_runtime/model_providers/openai/text_embedding/text-embedding-3-large.yaml new file mode 100644 index 0000000000..9489170fde --- /dev/null +++ b/api/core/model_runtime/model_providers/openai/text_embedding/text-embedding-3-large.yaml @@ -0,0 +1,9 @@ +model: text-embedding-3-large +model_type: text-embedding +model_properties: + context_size: 8191 + max_chunks: 32 +pricing: + input: '0.00013' + unit: '0.001' + currency: USD diff --git a/api/core/model_runtime/model_providers/openai/text_embedding/text-embedding-3-small.yaml b/api/core/model_runtime/model_providers/openai/text_embedding/text-embedding-3-small.yaml new file mode 100644 index 0000000000..586ba2b28f --- /dev/null +++ b/api/core/model_runtime/model_providers/openai/text_embedding/text-embedding-3-small.yaml @@ -0,0 +1,9 @@ +model: text-embedding-3-small +model_type: text-embedding +model_properties: + context_size: 8191 + max_chunks: 32 +pricing: + input: '0.00002' + unit: '0.001' + currency: USD