zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: xfembedding model

Browse Source
This commit is contained in:
wxg0103 2025-06-27 16:34:22 +08:00
parent db58ed8aba
commit a7cbba161c
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
apps/models_provider/impl/xf_model_provider/model/embedding.py
View File

@ -22,6 +22,7 @@ class XFEmbedding(MaxKBBaseModel, SparkLLMTextEmbeddings):
@staticmethod @staticmethod
def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs): def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
return XFEmbedding( return XFEmbedding(
base_url=model_credential.get('base_url'),
spark_app_id=model_credential.get('spark_app_id'), spark_app_id=model_credential.get('spark_app_id'),
spark_api_key=model_credential.get('spark_api_key'), spark_api_key=model_credential.get('spark_api_key'),
spark_api_secret=model_credential.get('spark_api_secret') spark_api_secret=model_credential.get('spark_api_secret')