refactor: support volcanic engine embeddings

Author: liuruibin

apps/setting/models_provider/impl/volcanic_engine_model_provider/credential/embedding.py

@@ -15,7 +15,7 @@
 from django.utils.translation import gettext_lazy as _
 
 
-class OpenAIEmbeddingCredential(BaseForm, BaseModelCredential):
+class VolcanicEmbeddingCredential(BaseForm, BaseModelCredential):
     def is_valid(self, model_type: str, model_name, model_credential: Dict[str, object], model_params, provider,
                  raise_exception=True):
         model_type_list = provider.get_model_type_list()

apps/setting/models_provider/impl/volcanic_engine_model_provider/model/embedding.py

@@ -1,15 +1,16 @@
 from typing import Dict
 
-from langchain_community.embeddings import VolcanoEmbeddings
+from langchain_openai import OpenAIEmbeddings
 
 from setting.models_provider.base_model_provider import MaxKBBaseModel
 
 
-class VolcanicEngineEmbeddingModel(MaxKBBaseModel, VolcanoEmbeddings):
+class VolcanicEngineEmbeddingModel(MaxKBBaseModel, OpenAIEmbeddings):
     @staticmethod
     def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
         return VolcanicEngineEmbeddingModel(
-            api_key=model_credential.get('api_key'),
+            openai_api_key=model_credential.get('api_key'),
             model=model_name,
             openai_api_base=model_credential.get('api_base'),
+            check_embedding_ctx_length=False,
         )

apps/setting/models_provider/impl/volcanic_engine_model_provider/volcanic_engine_model_provider.py

@@ -14,10 +14,12 @@
 from setting.models_provider.impl.openai_model_provider.credential.embedding import OpenAIEmbeddingCredential
 from setting.models_provider.impl.openai_model_provider.credential.llm import OpenAILLMModelCredential
 from setting.models_provider.impl.openai_model_provider.model.embedding import OpenAIEmbeddingModel
+from setting.models_provider.impl.volcanic_engine_model_provider.credential.embedding import VolcanicEmbeddingCredential
 from setting.models_provider.impl.volcanic_engine_model_provider.credential.image import \
     VolcanicEngineImageModelCredential
 from setting.models_provider.impl.volcanic_engine_model_provider.credential.tti import VolcanicEngineTTIModelCredential
 from setting.models_provider.impl.volcanic_engine_model_provider.credential.tts import VolcanicEngineTTSModelCredential
+from setting.models_provider.impl.volcanic_engine_model_provider.model.embedding import VolcanicEngineEmbeddingModel
 from setting.models_provider.impl.volcanic_engine_model_provider.model.image import VolcanicEngineImage
 from setting.models_provider.impl.volcanic_engine_model_provider.model.llm import VolcanicEngineChatModel
 from setting.models_provider.impl.volcanic_engine_model_provider.credential.stt import VolcanicEngineSTTModelCredential
@@ -82,12 +84,13 @@
               ),
 ]
 
-open_ai_embedding_credential = OpenAIEmbeddingCredential()
+open_ai_embedding_credential = VolcanicEmbeddingCredential()
 model_info_embedding_list = [
     ModelInfo('ep-xxxxxxxxxx-yyyy',
               _('The user goes to the model inference page of Volcano Ark to create an inference access point. Here, you need to enter ep-xxxxxxxxxx-yyyy to call it.'),
               ModelTypeConst.EMBEDDING, open_ai_embedding_credential,
-              OpenAIEmbeddingModel)]
+              VolcanicEngineEmbeddingModel)
+]
 
 model_info_manage = (
     ModelInfoManage.builder()
@@ -97,6 +100,8 @@
     .append_default_model_info(model_info_list[2])
     .append_default_model_info(model_info_list[3])
     .append_default_model_info(model_info_list[4])
+    .append_model_info_list(model_info_embedding_list)
+    .append_default_model_info(model_info_embedding_list[0])
     .build()
 )