flowengine

type Answer struct {
	ID    string `json:"id"`
	Label string `json:"label"`
}

type Candidate struct {
	NodeID   string `json:"node_id"`
	NodeType string `json:"node_type"`
	Content  string `json:"content"`
}

type CompactData struct {
	Model          string `json:"model"`
	Prompt         string `json:"prompt"`
	ThresholdWords int    `json:"threshold_words"`
}

type Document struct {
	Filename string
	Content  string
}

type ExecuteConfig struct {
	LLM                LLMClient
	BraveAPIKey        string
	CompactModel       string
	Messages           []Message
	Documents          []Document
	OnLLMCost          func(cost float64) error
	OnSearchCharged    func(ctx context.Context, endpoint string, resultCount int)
	OnCompactInflation func(
		ctx context.Context,
		model string,
		originalWords, compactedWords int,
	)
	LoadFlow   func(flowID string) (FlowDefinition, error)
	RootFlowID string
	Streaming  bool
	// contains filtered or unexported fields
}

type ExecuteResult struct {
	// TotalCost is the billed cost (raw groq × markup) for
	// the entire execution. Sourced from cfg.LLM at the
	// end of Execute since all per-call accounting lives
	// in the billedllm Recorder.
	TotalCost   float64
	FinalOutput string
	Candidates  []Candidate
}

type Executor func(
	ctx context.Context, n NodeCtx,
) (NodeResult, error)

type FilesData struct{}

type FlowDefinition struct {
	Nodes []FlowNode `json:"nodes"`
	Edges []FlowEdge `json:"edges"`
}

type FlowEdge struct {
	ID           string `json:"id"`
	Source       string `json:"source"`
	Target       string `json:"target"`
	SourceHandle string `json:"sourceHandle,omitempty"`
}

type FlowNode struct {
	ID       string          `json:"id"`
	Type     NodeType        `json:"type"`
	Position Position        `json:"position"`
	Data     json.RawMessage `json:"data"`
}

type LLMClient interface {
	Complete(
		ctx context.Context,
		model string,
		messages []llmclient.Message,
		maxTokens int,
	) (string, error)

	CompleteRaw(
		ctx context.Context,
		model string,
		messages []llmclient.Message,
		tools []llmclient.ToolDef,
		maxTokens int,
	) (llmclient.Message, error)

	Stream(
		ctx context.Context,
		model string,
		messages []llmclient.Message,
		maxTokens int,
	) (<-chan string, <-chan error)

	TotalBilled() float64
}

type LLMData struct {
	Model       string  `json:"model"`
	MaxTokens   int     `json:"max_tokens"`
	Temperature float64 `json:"temperature"`
}

type Message struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type MsgHistoryData struct {
	MaxMessages int `json:"max_messages"`
}

type NodeCtx struct {
	Node      *FlowNode
	NodeID    string
	Inputs    []Message
	UserInput string
	Streaming bool
	Config    ExecuteConfig
	Result    *ExecuteResult
	Emit      func(StreamEvent)
}

type NodeResult struct {
	Produced []Message
	Decision string
}

type NodeType string

type Position struct {
	X float64 `json:"x"`
	Y float64 `json:"y"`
}

type QuestionData struct {
	Question      string   `json:"question"`
	Answers       []Answer `json:"answers"`
	Model         string   `json:"model"`
	MaxIterations int      `json:"max_iterations"`
}

type StreamEvent struct {
	Type              string      `json:"type"`
	NodeID            string      `json:"node_id,omitempty"`
	NodeType          string      `json:"node_type,omitempty"`
	Delta             string      `json:"delta,omitempty"`
	Output            string      `json:"output,omitempty"`
	OutputPreview     string      `json:"output_preview,omitempty"`
	ShowOutput        bool        `json:"show_output,omitempty"`
	Message           string      `json:"message,omitempty"`
	UserError         bool        `json:"user_error,omitempty"`
	MessageID         string      `json:"message_id,omitempty"`
	Usage             any         `json:"usage,omitempty"`
	TotalCost         string      `json:"total_cost,omitempty"`
	RetrySeconds      int         `json:"retry_seconds,omitempty"`
	Source            string      `json:"source,omitempty"`
	Content           string      `json:"content,omitempty"`
	Branch            string      `json:"branch,omitempty"`
	Title             string      `json:"title,omitempty"`
	Candidates        []Candidate `json:"candidates,omitempty"`
	GuestLLMCallsUsed *int32      `json:"guest_llm_calls_used,omitempty"`
	GuestLimitReached bool        `json:"guest_limit_reached,omitempty"`
}

type SubflowData struct {
	FlowID string `json:"flow_id"`
	Name   string `json:"name,omitempty"`
}

type SystemPromptData struct {
	Text string `json:"text"`
}

type ValidationError struct {
	NodeID  string
	Message string
}

type WebSearchData struct {
	ResultCount int `json:"result_count"`
}