engine

type AgentEngine struct {
	ConvID    int64
	MessageID int64

	BotEngine
	// contains filtered or unexported fields
}

type BotEngine struct {
	SessionID string
	MessageID int64

	HistoryMessages []llm.Message

	Settings *Settings
	Hooks    []Hook
	// contains filtered or unexported fields
}

type ChatEngine struct {
	*BotEngine
}

type Engine interface {
	Execute(ctx context.Context, input Input) *EventStream
}

type Event struct {
	Data     any
	Name     EventName
	Error    error
	Source   string
	SourceID string
}

type EventName string

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

type Extra struct {
	Responder   string
	ResponderID string
}

type Hook interface {
	OnStart(context.Context) context.Context
	OnFinish(context.Context, error)

	BeforeRetrieveStart(context.Context)
	AfterRetrieveFinish(context.Context)

	BeforeToolStart(context.Context, int64, map[string]any)
	AfterToolFinish(context.Context, int64, map[string]any, error)

	OnReasonStart(context.Context)
	OnReasonFinish(context.Context)

	BeforeLLMStart(context.Context) context.Context
	AfterLLMFinish(context.Context, error)
}

type Input struct {
	Inputs    map[string]any `json:"inputs"`
	Query     string         `json:"query"`
	TurnID    string         `json:"turn_id"`
	SceneID   string         `json:"scene_id"`
	Turnround int            `json:"turnround"`
	ChunkID   string         `json:"chunk_id"`
}

type Memory interface {
	FetchMemories(context.Context, int64) ([]llm.Message, error)
	SaveMemory(context.Context, int64, string, *Extra) error
}

type MemoryOption func(*MemoryOptions)

type MemoryOptions struct {
	Limit int
}

type Option func(o *Options)

type Options struct {
	Rag            *retrival.Retrival
	Memory         Memory
	ConversationID int64
	MessageID      int64
	Hooks          []Hook
	// contains filtered or unexported fields
}

type ProxyEngine struct {
	*BotEngine
}

type Settings struct {
	PresetPrompt string        `json:"preset_prompt"`
	Preamble     string        `json:"preamble"` // 机器人开场白
	Knowledge    []rag.Options `json:"knowledge"`
	LLMModel     struct {
		Provider string `json:"provider"`
		Name     string `json:"name"`
	} `json:"llm_model"`
	Agent struct {
		Mode         string `json:"mode"`
		MaxIteration int    `json:"max_iteration"`
	}
	Workflow *workflow.Workflow `json:"workflow"`
	Tools    []string           `json:"tools"`
	Stream   bool               `json:"stream"`
}

type WorkflowEngine struct {
	BotEngine
}