feat: enhance model handling with default parameters and cache model check

Author: liuruibin

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 exception_handler(e):
             ))
 
     embedding_model = get_embedding_model(model_id, exception_handler)
+    #
     ListenerManagement.embedding_by_document(document_id, embedding_model, state_list)
 
 

apps/models_provider/impl/aliyun_bai_lian_model_provider/model/embedding.py

@@ -22,6 +22,9 @@ def __init__(self, api_key, model_name: str, optional_params: dict):
         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)

apps/models_provider/impl/openai_model_provider/model/embedding.py

@@ -22,6 +22,9 @@ def __init__(self, api_key, base_url, model_name: str, optional_params: dict):
         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)