diff --git a/apps/knowledge/task/embedding.py b/apps/knowledge/task/embedding.py index c0949574a..79ce9ed01 100644 --- a/apps/knowledge/task/embedding.py +++ b/apps/knowledge/task/embedding.py @@ -27,7 +27,9 @@ def get_embedding_model(model_id, exception_handler=lambda e: maxkb_logger.error ))): try: model = QuerySet(Model).filter(id=model_id).first() - embedding_model = ModelManage.get_model(model_id, lambda _id: get_model(model)) + + s = {p.get('field'): p.get('default_value') for p in model.model_params_form if p.get('default_value') is not None} + embedding_model = ModelManage.get_model(model_id, lambda _id: get_model(model, **{**s})) except Exception as e: exception_handler(e) raise e @@ -77,6 +79,7 @@ def embedding_by_document(document_id, model_id, state_list=None): )) embedding_model = get_embedding_model(model_id, exception_handler) + # ListenerManagement.embedding_by_document(document_id, embedding_model, state_list) diff --git a/apps/models_provider/impl/aliyun_bai_lian_model_provider/model/embedding.py b/apps/models_provider/impl/aliyun_bai_lian_model_provider/model/embedding.py index 786469e02..c02f5dc52 100644 --- a/apps/models_provider/impl/aliyun_bai_lian_model_provider/model/embedding.py +++ b/apps/models_provider/impl/aliyun_bai_lian_model_provider/model/embedding.py @@ -22,6 +22,9 @@ class AliyunBaiLianEmbedding(MaxKBBaseModel): self.model_name = model_name self.optional_params = optional_params + def is_cache_model(self): + return False + @staticmethod def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs): optional_params = MaxKBBaseModel.filter_optional_params(model_kwargs) diff --git a/apps/models_provider/impl/openai_model_provider/model/embedding.py b/apps/models_provider/impl/openai_model_provider/model/embedding.py index 7362e8fcd..9b5a1c417 100644 --- a/apps/models_provider/impl/openai_model_provider/model/embedding.py +++ b/apps/models_provider/impl/openai_model_provider/model/embedding.py @@ -22,6 +22,9 @@ class OpenAIEmbeddingModel(MaxKBBaseModel): self.model_name = model_name self.optional_params = optional_params + def is_cache_model(self): + return False + @staticmethod def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs): optional_params = MaxKBBaseModel.filter_optional_params(model_kwargs)