types

type Action interface {
	Run(ctx context.Context, sharedState *AgentSharedState, action ActionParams) (ActionResult, error)
	Definition() ActionDefinition
}

type ActionContext struct {
	context.Context
	// contains filtered or unexported fields
}

type ActionCurrentState struct {
	Job       *Job
	Action    Action
	Params    ActionParams
	Reasoning string
}

type ActionDefinition struct {
	Properties  map[string]jsonschema.Definition
	Required    []string
	Name        ActionDefinitionName
	Description string
}

type ActionDefinitionName string

type ActionParams map[string]interface{}

type ActionRequest struct {
	Action Action
	Params *ActionParams
}

type ActionResult struct {
	Job               *Job
	Result            string
	ImageBase64Result string
	Metadata          map[string]interface{}
}

type ActionState struct {
	ActionCurrentState
	ActionResult
}

type Actions []Action

type AgentInternalState struct {
	NowDoing    string   `json:"doing_now"`
	DoingNext   string   `json:"doing_next"`
	DoneHistory []string `json:"done_history"`
	Memories    []string `json:"memories"`
	Goal        string   `json:"goal"`
}

type AgentSharedState struct {
	ConversationTracker *conversations.ConversationTracker[string] `json:"conversation_tracker"`
	Scheduler           TaskScheduler                              `json:"-"` // Not serialized, set at runtime
	AgentName           string                                     `json:"agent_name"`
}

type BaseAction struct{}

type Completion struct {
	Error                  string                         `json:"error,omitempty"`
	ChatCompletionResponse *openai.ChatCompletionResponse `json:"chat_completion_response,omitempty"`
	Conversation           []openai.ChatCompletionMessage `json:"conversation,omitempty"`
	ActionResult           string                         `json:"action_result,omitempty"`
	AgentState             *AgentInternalState            `json:"agent_state,omitempty"`
	FilterResult           *FilterResult                  `json:"filter_result,omitempty"`
}

type ConversationMessage struct {
	Message  openai.ChatCompletionMessage
	Metadata map[string]interface{}
}

type Creation struct {
	ChatCompletionMessage *openai.ChatCompletionMessage `json:"chat_completion_message,omitempty"`
	ChatCompletionRequest *openai.ChatCompletionRequest `json:"chat_completion_request,omitempty"`
	FunctionDefinition    *openai.FunctionDefinition    `json:"function_definition,omitempty"`
	FunctionParams        ActionParams                  `json:"function_params,omitempty"`
}

type FilterResult struct {
	HasTriggers bool   `json:"has_triggers"`
	TriggeredBy string `json:"triggered_by,omitempty"`
	FailedBy    string `json:"failed_by,omitempty"`
}

type Job struct {
	// The job is a request to the agent to do something
	// It can be a question, a command, or a request to do something
	// The agent will try to do it, and return a response
	Result              *JobResult
	ReasoningCallback   func(ActionCurrentState) bool
	ResultCallback      func(ActionState)
	ConversationHistory []openai.ChatCompletionMessage
	UUID                string
	Metadata            map[string]interface{}
	DoneFilter          bool

	// Tools available for this job
	BuiltinTools []ActionDefinition // Built-in tools like web search
	UserTools    []ActionDefinition // User-defined function tools
	ToolChoice   string

	Obs *Observable
	// contains filtered or unexported fields
}

type JobFilter interface {
	Name() string
	Apply(job *Job) (bool, error)
	IsTrigger() bool
}

type JobFilters []JobFilter

type JobOption func(*Job)

type JobResult struct {
	sync.Mutex
	// The result of a job
	State        []ActionState
	Plans        []cogito.PlanStatus
	Conversation []openai.ChatCompletionMessage

	Finalizers []func([]openai.ChatCompletionMessage)

	Response string
	Error    error
	// contains filtered or unexported fields
}

type Observable struct {
	ID       int32  `json:"id"`
	ParentID int32  `json:"parent_id,omitempty"`
	Agent    string `json:"agent"`
	Name     string `json:"name"`
	Icon     string `json:"icon"`

	Creation   *Creation   `json:"creation,omitempty"`
	Progress   []Progress  `json:"progress,omitempty"`
	Completion *Completion `json:"completion,omitempty"`
}

type OneTimeReminderParams struct {
	Message string `json:"message"`
	Delay   string `json:"delay"` // Go duration format with day support: "30m", "2h", "1d", "1d12h"
}

type Progress struct {
	Error                  string                         `json:"error,omitempty"`
	ChatCompletionResponse *openai.ChatCompletionResponse `json:"chat_completion_response,omitempty"`
	ActionResult           string                         `json:"action_result,omitempty"`
	AgentState             *AgentInternalState            `json:"agent_state"`
}

type PromptResult struct {
	Content     string
	ImageBase64 string
}

type RecurringReminderParams struct {
	Message  string `json:"message"`
	CronExpr string `json:"cron_expr"`
}

type ReminderActionResponse = RecurringReminderParams

type TaskScheduler interface {
	CreateTask(task *scheduler.Task) error
	GetAllTasks() ([]*scheduler.Task, error)
	GetTask(id string) (*scheduler.Task, error)
	DeleteTask(id string) error
	PauseTask(id string) error
	ResumeTask(id string) error
}

type UserDefinedAction struct {
	ActionDef *ActionDefinition
}

type UserDefinedChecker interface {
	IsUserDefined() bool
}