fix(text_embedding): pass input_type to distinguish between document and query

Signed-off-by: Yuichiro Utsumi <utsumi.yuichiro@fujitsu.com>

Author: utsumi-fj

python/dify_plugin/core/entities/plugin/request.py

@@ -9,7 +9,10 @@
     OnlineDriveBrowseFilesRequest,
     OnlineDriveDownloadFileRequest,
 )
-from dify_plugin.entities.model import ModelType
+from dify_plugin.entities.model import (
+    EmbeddingInputType,
+    ModelType,
+)
 from dify_plugin.entities.model.message import (
     AssistantPromptMessage,
     PromptMessage,
@@ -200,6 +203,7 @@ class ModelInvokeTextEmbeddingRequest(PluginAccessModelRequest):
     action: ModelActions = ModelActions.InvokeTextEmbedding
 
     texts: list[str]
+    input_type: EmbeddingInputType | None = EmbeddingInputType.DOCUMENT
 
 
 class ModelGetTextEmbeddingNumTokens(PluginAccessModelRequest):

python/dify_plugin/core/plugin_executor.py

@@ -215,6 +215,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}`")