schema

type Agent interface {
	Run(ctx context.Context, messages []Message, opts ...llm.GenerateOption) (*Generation, error)

	Name() string

	Description() string

	WithEnv(env Environment)

	Env() Environment

	Tools() []tool.Tool
}

type Document struct {
	PageContent string
	Metadata    map[string]any
	Score       float32
}

type Environment interface {
	// Produce produce msg
	Produce(ctx context.Context, msgs ...Message) error
	// Consume consume msg
	Consume(ctx context.Context) *Message

	// SOP task SOP
	SOP() string
	// GetTeam all team members
	GetTeam() []Agent
	// GetTeamLeader team Leader
	GetTeamLeader() Agent

	// LoadMemory get Agent's historical msg
	LoadMemory(ctx context.Context, receiver Agent) []Message

	GetSubscribeAgents(_ context.Context, subscribed Agent) []Agent
}

type Generation struct {
	// Generated text output.
	Messages []Message
	// Raw generation info response from the provider.
	// May include things like reason for finishing (e.g. in OpenAI).
	TotalTokens int
}

type Memory interface {
	Load(ctx context.Context, filter func(index, consumption int, message Message) bool) []Message

	// LoadNext load next msg，filter check next msg，if not passed，then do not return
	LoadNext(ctx context.Context, filter func(message Message) bool) *Message

	Save(ctx context.Context, msg Message) error
	// Clear memory contents.
	Clear(ctx context.Context) error
}

type Message struct {
	Type      string `json:"cate"`
	Thought   string `json:"thought"`
	Content   string `json:"content"`
	Sender    string `json:"sender"`
	Receiver  string `json:"receiver"`
	Condition string `json:"condition"`
	Token     int    `json:"token"`
	Log       string
	// control msg, to remove and update Agent
	MngInfo     *MngInfo
	AllReceiver []string
}

type MngInfo struct {
	Create []struct {
		Name        string   `json:"name"`
		Description string   `json:"description"`
		Tools       []string `json:"tools"`
		Prompt      string   `json:"prompt"`
	} `json:"create"`
	Select []string `json:"select"`
	Remove []string `json:"remove"`
}

type StepAction struct {
	Id          string `json:"id"`
	Action      string `json:"action"`
	Thought     string `json:"thought"`
	Input       string `json:"input"`
	Feedback    string `json:"feedback"`
	Log         string `json:"log"`
	Observation string `json:"observation"`
}

type StepActionInput struct {
	Input any `json:"input"`
}

type StepFeedback struct {
	Feedback string `json:"feedback"`
	Log      string
}

type StepType struct {
	Type string `json:"type"`
}

type Subscribe struct {
	Subscribed Agent
	Subscriber Agent
	Condition  string
}