diff --git a/internal/core/dify_invocation/types.go b/internal/core/dify_invocation/types.go
index 7ec8532..677f4e9 100644
--- a/internal/core/dify_invocation/types.go
+++ b/internal/core/dify_invocation/types.go
@@ -2,6 +2,7 @@ package dify_invocation
 
 import (
 	"github.com/go-playground/validator/v10"
+	"github.com/langgenius/dify-plugin-daemon/internal/types/entities/model_entities"
 	"github.com/langgenius/dify-plugin-daemon/internal/types/entities/plugin_entities"
 	"github.com/langgenius/dify-plugin-daemon/internal/types/entities/requests"
 	"github.com/langgenius/dify-plugin-daemon/internal/types/validators"
@@ -32,10 +33,20 @@ const (
 	INVOKE_TYPE_UPLOAD_FILE              InvokeType = "upload_file"
 )
 
+type InvokeLLMSchema struct {
+	CompletionParams map[string]any                     `json:"completion_params"  validate:"omitempty"`
+	PromptMessages   []model_entities.PromptMessage     `json:"prompt_messages"  validate:"omitempty"`
+	Tools            []model_entities.PromptMessageTool `json:"tools" validate:"omitempty,dive"`
+	Stop             []string                           `json:"stop" validate:"omitempty"`
+	Stream           bool                               `json:"stream"`
+}
+
 type InvokeLLMRequest struct {
 	BaseInvokeDifyRequest
 	requests.BaseRequestInvokeModel
-	requests.InvokeLLMSchema
+	// requests.InvokeLLMSchema,
+	// TODO: as completion_params in requests.InvokeLLMSchema is "model_parameters", we declare another one here
+	InvokeLLMSchema
 }
 
 type InvokeTextEmbeddingRequest struct {