diff --git a/python/dify_plugin/core/entities/plugin/request.py b/python/dify_plugin/core/entities/plugin/request.py index 34bb9a92..0ebe72af 100644 --- a/python/dify_plugin/core/entities/plugin/request.py +++ b/python/dify_plugin/core/entities/plugin/request.py @@ -203,6 +203,7 @@ class ModelInvokeTextEmbeddingRequest(PluginAccessModelRequest): action: ModelActions = ModelActions.InvokeTextEmbedding texts: list[str] + input_type: EmbeddingInputType = EmbeddingInputType.DOCUMENT class ModelInvokeMultimodalEmbeddingRequest(PluginAccessModelRequest): diff --git a/python/dify_plugin/core/plugin_executor.py b/python/dify_plugin/core/plugin_executor.py index 04a2d9ca..db78a39f 100644 --- a/python/dify_plugin/core/plugin_executor.py +++ b/python/dify_plugin/core/plugin_executor.py @@ -217,6 +217,7 @@ def invoke_text_embedding(self, session: Session, data: ModelInvokeTextEmbedding data.credentials, data.texts, data.user_id, + data.input_type, ) else: raise ValueError(f"Model `{data.model_type}` not found for provider `{data.provider}`")