Function executeDifyWorkflowByMode

api/controller/relay/relay_workflow.go:951–999

executeDifyWorkflowByMode 根据mode选择适当的DIFY执行方式

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

// executeDifyWorkflowByMode 根据mode选择适当的DIFY执行方式
func executeDifyWorkflowByMode(c *gin.Context, workflowRequest *WorkflowRunRequest, agent *model.Agent, channel *model.Channel, modelName string) (*custom.WorkflowResponseData, error) {
	// 检查模型名称是否包含 "workflow-"
	if !strings.Contains(modelName, "workflow-") {
		return executeDifyWorkflow(c, workflowRequest, agent, channel, modelName)
	}

	// 解析配置获取 mode
	var config map[string]interface{}
	if channel.Config != "" {
		if err := json.Unmarshal([]byte(channel.Config), &config); err != nil {
			logger.SysLogf("解析渠道配置失败: %v", err)
			// 如果解析失败，默认使用工作流模式
			return executeDifyWorkflow(c, workflowRequest, agent, channel, modelName)
		}
	} else {
		config = make(map[string]interface{})
	}

	// 检查 mode 值是否存在
	mode, exists := config["mode"]
	if !exists {
		// 如果不存在mode字段，则动态获取
		fetchedMode, err := fetchDifyAppMode(channel)
		if err != nil {
			logger.SysLogf("获取DIFY应用mode失败: %v，使用默认工作流模式", err)
			return executeDifyWorkflow(c, workflowRequest, agent, channel, modelName)
		}

		// 将获取到的mode信息更新到数据库中，以便后续请求使用
		config["mode"] = fetchedMode
		if err := model.UpdateChannelConfigOnly(channel.ChannelID, config); err != nil {
			logger.SysLogf("更新渠道配置失败: %v", err)
			// 即使更新失败，仍然使用当前获取到的mode值
		}

		// 直接使用获取到的mode值
		if fetchedMode == "completion" {
			return executeDifyCompletionMessages(c, workflowRequest, agent, channel, modelName)
		}
		return executeDifyWorkflow(c, workflowRequest, agent, channel, modelName)
	}

	// 检查 mode 值
	if mode == "completion" {
		return executeDifyCompletionMessages(c, workflowRequest, agent, channel, modelName)
	}

	return executeDifyWorkflow(c, workflowRequest, agent, channel, modelName)
}

// executeDifyCompletionMessages 执行 DIFY Completion Messages
func executeDifyCompletionMessages(c *gin.Context, workflowRequest *WorkflowRunRequest, agent *model.Agent, channel *model.Channel, modelName string) (*custom.WorkflowResponseData, error) {