dify

type AlertAnalyzeResponse struct {
	// contains filtered or unexported fields
}

type AlertCheckConfig struct {
	FlowId        string
	FlowName      string
	APIKey        string
	Authorization string
	AnalyzeAuth   string

	User string

	Sampling       string
	CacheMinutes   int
	MaxConcurrency int

	Prom prometheus.Repo
}

type AlertCheckResponse struct {
	// contains filtered or unexported fields
}

type ChunkCompletionResponse struct{}

type CompletionResponse struct {
	WorkflowRunID string                 `json:"workflow_run_id"`
	TaskID        string                 `json:"task_id"`
	Data          CompletionResponseData `json:"data"`
}

type CompletionResponseData struct {
	ID         string          `json:"id"`
	WorkflowID string          `json:"workload_id"`
	Status     string          `json:"status"`
	Outputs    json.RawMessage `json:"outputs"`

	CreatedAt int64 `json:"created_at"`
}

type DifyClient struct {
	*http.Client

	BaseURL string
}

type DifyRepo interface {
	AddUser(username string, password string, role string) (*DifyResponse, error)
	UpdatePassword(username string, oldPassword string, newPassword string) (*DifyResponse, error)
	RemoveUser(username string) (*DifyResponse, error)
	ResetPassword(username string, newPassword string) (*DifyResponse, error)

	PrepareAsyncAlertCheckWorkflow(cfg *AlertCheckConfig, logger *zap.Logger) (records <-chan *model.WorkflowRecord, err error)
	SubmitAlertEvents(events []alert.AlertEvent)
	GetCacheMinutes() int
	GetAlertCheckFlowID() string
	GetAlertAnalyzeFlowID() string

	WorkflowsRun(req *WorkflowRequest, authorization string) (*CompletionResponse, error)
}

type DifyResponse struct {
	Result  string `json:"result"`
	Message string `json:"message"`
}

type DifyUser struct {
	Password    string `json:"password"`
	NewPassword string `json:"new_password"`
	Role        string `json:"role"`
	Username    string `json:"username"`
}

type Handle func(ctx core.Context, record *model.WorkflowRecord) error

type WorkflowRequest struct {
	Inputs       json.RawMessage `json:"inputs"`
	ResponseMode string          `json:"response_mode"`
	User         string          `json:"user"`
}

type WorkflowResponse interface {
	// contains filtered or unexported methods
}