Function executeDifyCompletionMessages

api/controller/relay/relay_workflow.go:1002–1153

executeDifyCompletionMessages 执行 DIFY Completion Messages

(c *gin.Context, workflowRequest *WorkflowRunRequest, agent *model.Agent, channel *model.Channel, modelName string)

// executeDifyCompletionMessages 执行 DIFY Completion Messages
func executeDifyCompletionMessages(c *gin.Context, workflowRequest *WorkflowRunRequest, agent *model.Agent, channel *model.Channel, modelName string) (*custom.WorkflowResponseData, error) {
	// 获取元数据
	meta := GetByContext(c)
	meta.APIType = model.GetApiType(channel.Type)
	meta.OriginModelName = modelName
	meta.ChannelId = int(channel.ChannelID)
	if channel.BaseURL != nil {
		meta.BaseURL = *channel.BaseURL
	}
	meta.APIKey = channel.Key

	// 应用模型映射
	mappedModel, _ := getMappedModelName(modelName, meta.ModelMapping)
	meta.ActualModelName = mappedModel

	logger.SysLogf("DIFY补全消息执行 - 模型映射，OriginModel: %s, ActualModel: %s",
		meta.OriginModelName, meta.ActualModelName)

	// 构建 completion request
	workflowID := extractWorkflowID(agent.Model, agent.CustomConfig)
	if workflowID == "" {
		return nil, fmt.Errorf("无法提取工作流ID")
	}

	// 构建请求体，符合 DIFY /completion-messages API 规范
	var inputs map[string]interface{}

	// 检查参数中是否包含 query，如果没有则默认添加
	queryExists := false
	for key := range workflowRequest.Parameters {
		if key == "query" {
			queryExists = true
			break
		}
	}

	if !queryExists {
		// 如果没有query参数，从参数中提取第一个值作为query，或者使用默认值
		if len(workflowRequest.Parameters) > 0 {
			for _, value := range workflowRequest.Parameters {
				if strValue, ok := value.(string); ok && strValue != "" {
					workflowRequest.Parameters["query"] = strValue
					queryExists = true
					break
				}
			}
		}

		if !queryExists {
			workflowRequest.Parameters["query"] = "请处理请求"
		}
	}

	inputs = workflowRequest.Parameters

	// 创建 completion request 对象
	completionRequest := map[string]interface{}{
		"inputs":        inputs,