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

utsumi-fj · utsumi-fj · commit 3bf597841b93 · 2025-12-08T23:31:16.000Z
Signed-off-by: Yuichiro Utsumi &lt;utsumi.yuichiro@fujitsu.com&gt;
diff --git 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 | None = EmbeddingInputType.DOCUMENT
 
 
 class ModelInvokeMultimodalEmbeddingRequest(PluginAccessModelRequest):
diff --git 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}`")

Original file line number	Diff line number	Diff line change
`@@ -217,6 +217,7 @@ def invoke_text_embedding(self, session: Session, data: ModelInvokeTextEmbedding`
`217`	`217`	`data.credentials,`
`218`	`218`	`data.texts,`
`219`	`219`	`data.user_id,`
	`220`	`+ data.input_type,`
`220`	`221`	`)`
`221`	`222`	`else:`
`222`	`223`	raise ValueError(f"Model `{data.model_type}` not found for provider `{data.provider}`")