baseAgent

type AgentSession struct {
	Agent          *agent.Agent
	SessionManager *SessionManager
}

type AuthStore interface {
	APIKey(provider ai.Provider) (string, bool)
}

type BranchSummaryEntry struct {
	SessionEntryBase
	FromID   string `json:"fromId"`
	Summary  string `json:"summary"`
	Details  any    `json:"details,omitempty"`
	FromHook *bool  `json:"fromHook,omitempty"`
}

type BranchSummaryMessage struct {
	Role      string `json:"role"`
	Summary   string `json:"summary"`
	FromID    string `json:"fromId"`
	Timestamp int64  `json:"timestamp"`
}

type CompactionEntry struct {
	SessionEntryBase
	Summary          string `json:"summary"`
	FirstKeptEntryID string `json:"firstKeptEntryId"`
	TokensBefore     int64  `json:"tokensBefore"`
	Details          any    `json:"details,omitempty"`
	FromHook         *bool  `json:"fromHook,omitempty"`
}

type CompactionSummaryMessage struct {
	Role         string `json:"role"`
	Summary      string `json:"summary"`
	TokensBefore int64  `json:"tokensBefore"`
	Timestamp    int64  `json:"timestamp"`
}

type CreateAgentSessionOptions struct {
	Model          ai.Model[ai.API]
	ThinkingLevel  ai.ModelThinkingLevel
	SessionManager *SessionManager
	ModelRegistry  *ModelRegistry
	AuthStore      AuthStore

	SystemPrompt      *string
	Tools             []agent.AgentTool[any, any]
	ToolExecutionMode *agent.ToolExecutionMode

	/* Optional func to call before each tool call is executed. */
	BeforeToolCall func(
		ctx context.Context,
		call agent.BeforeToolCallContext,
	) (*agent.BeforeToolCallResult, error)

	/* Optional func to call after each tool call is executed. */
	AfterToolCall func(
		ctx context.Context,
		call agent.AfterToolCallContext[any],
	) (*agent.AfterToolCallResult, error)

	/* Optional func to call before each payload is sent to the provider. */
	OnPayload func(
		payload any,
		model ai.Model[ai.API],
	) (any, error)

	/* Optional func to call after each response is received from the provider. */
	OnResponse func(
		response ai.ProviderResponse,
		model ai.Model[ai.API],
	) error
}

type CreateAgentSessionResult struct {
	Session              *AgentSession
	ModelFallbackMessage *string
}

type CustomEntry struct {
	SessionEntryBase
	CustomType string `json:"customType"`
	Data       any    `json:"data,omitempty"`
}

type CustomMessage struct {
	Role       string           `json:"role"`
	CustomType string           `json:"customType"`
	Content    []ai.UserContent `json:"content"`
	Display    bool             `json:"display"`
	Details    any              `json:"details,omitempty"`
	Timestamp  int64            `json:"timestamp"`
}

type CustomMessageEntry struct {
	SessionEntryBase
	CustomType string           `json:"customType"`
	Content    []ai.UserContent `json:"content"`
	Details    any              `json:"details,omitempty"`
	Display    bool             `json:"display"`
}

type EnvAuthStore struct{}

type LabelEntry struct {
	SessionEntryBase
	TargetID string  `json:"targetId"`
	Label    *string `json:"label,omitempty"`
}

type ModelChangeEntry struct {
	SessionEntryBase
	Provider string `json:"provider"`
	ModelID  string `json:"modelId"`
}

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

type PromptOptions struct {
	Images []ai.ImageContent
	/** When streaming, how to queue the message: "steer" (interrupt) or "followUp" (wait). Required if streaming. */
	StreamingBehavior StreamingBehavior
}

type RequestAuth struct {
	APIKey  *string
	Headers map[string]string
}

type SessionContext struct {
	Messages      []agent.AgentMessage
	ThinkingLevel string
	Model         *SessionModel
}

type SessionEntry interface{ GetBase() *SessionEntryBase }

type SessionEntryBase struct {
	Type      string  `json:"type"`
	ID        string  `json:"id"`
	ParentID  *string `json:"parentId"`
	Timestamp string  `json:"timestamp"`
}

type SessionHeader struct {
	Type          string         `json:"type"`
	Version       int            `json:"version,omitempty"`
	ID            string         `json:"id"`
	Timestamp     string         `json:"timestamp"`
	ParentSession *string        `json:"parentSession,omitempty"`
	Name          *string        `json:"name,omitempty"`
	MessageCount  int64          `json:"messageCount,omitempty"`
	FirstMessage  *string        `json:"firstMessage,omitempty"`
	Metadata      map[string]any `json:"metadata,omitempty"`
}

type SessionInfo struct {
	ID                      string
	Name, ParentSessionPath *string
	Created, Modified       time.Time
	MessageCount            int
	FirstMessage            string
}

type SessionInfoEntry struct {
	SessionEntryBase
	Name *string `json:"name,omitempty"`
}

type SessionListProgress func(loaded, total int)

type SessionManager struct {
	Header  *SessionHeader
	Entries []SessionEntry

	Store SessionStore
	// contains filtered or unexported fields
}

type SessionMessageEntry struct {
	SessionEntryBase
	Message agent.AgentMessage `json:"message"`
}

type SessionModel struct{ Provider, ModelID string }

type SessionStore interface {
	CreateSession(ctx context.Context, header *SessionHeader) error
	LoadSession(ctx context.Context, sessionID string) (*SessionHeader, []SessionEntry, error)
	AppendEntry(ctx context.Context, header *SessionHeader, entry SessionEntry) error
	ListSessions(ctx context.Context, metadata map[string]any) ([]SessionInfo, error)
	FindMostRecentSession(ctx context.Context, metadata map[string]any) (*SessionInfo, error)
	CreateBranch(ctx context.Context, fromLeafID string, newHeader SessionHeader) error
}

type SessionTreeNode struct {
	Entry          SessionEntry       `json:"entry"`
	Children       []*SessionTreeNode `json:"children"`
	Label          *string            `json:"label,omitempty"`
	LabelTimestamp *string            `json:"labelTimestamp,omitempty"`
}

type StreamingBehavior string

type ThinkingLevelChangeEntry struct {
	SessionEntryBase
	ThinkingLevel string `json:"thinkingLevel"`
}