agents

type Agent interface {
	AnalysisSession

	Profile() *Profile
	Log() thoughtstream.Branch
	History() []*thoughtstream.Thought
	WorldState() WorldState

	AttachTo(r Router)
	ForkSession() (AnalysisSession, error)

	Step(ctx context.Context, options ...StepOption) error
}

type AgentBase struct {
	psi.NodeBase
	// contains filtered or unexported fields
}

type AgentContext interface {
	Context() context.Context
	Profile() *Profile
	Agent() Agent
	Branch() thoughtstream.Branch
	Stream() thoughtstream.Stream
	WorldState() WorldState
}

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

type AgentPrompt interface {
	Render(ctx AgentContext) (chat.Message, error)
}

type AgentPromptFunc func(ctx AgentContext) (chat.Message, error)

type AnalysisSession interface {
	psi.Node

	Introspect(ctx context.Context, prompt AgentPrompt, options ...StepOption) (*thoughtstream.Thought, error)

	ReceiveMessage(ctx context.Context, msg *thoughtstream.Thought) error
}

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

type Colab struct {
	psi.NodeBase
	// contains filtered or unexported fields
}

type PostStepHook func(ctx AgentContext, thought *thoughtstream.Thought) error

type Profile struct {
	psi.NodeBase

	Name        string
	Description any

	BaselineSystemPrompt string

	Rank     float64
	Priority float64

	Provides []string
	Requires []string

	PostStep PostStepHook
}

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

type Router interface {
	RegisterAgent(agent Agent)

	ReceiveIncomingMessage(ctx context.Context, msg *thoughtstream.Thought)
	RouteIncomingMessages(ctx context.Context) error

	RouteMessage(ctx context.Context, msg *thoughtstream.Thought) error

	OutgoingMessages() []*thoughtstream.Thought
	ResetOutbox()
}

type Scheduler interface {
	NextSpeaker(ctx AgentContext, candidates ...Agent) (Agent, error)
}

type StepOption func(*StepOptions) error

type StepOptions struct {
	Base   thoughtstream.Branch
	Head   thoughtstream.Pointer
	Stream thoughtstream.Stream
}

type WorldState interface {
	psi.Node

	Get(key string) any
	Set(key string, value any)
}

type WorldStateKey[T any] string