shared

type ABaseDBSaving struct {
	Args
	// DBQuery is a function that executes a query on the database.
	DBQuery func(ctx context.Context) error
}

type ACheckingMenuRefs struct {
	Args
	// RetStr returns dereferenced user prompts based on the list of offer.
	RetOfferRef chan<- *OfferRef
	// Perform additional checks via LLM
	CheckLLM bool `log:"check_llm"`
	// Prompt is a prompt to be checked.
	Prompt string `log:"prompt"`
	// List of choices (optional)
	Choices []string
}

type AConfigUpdate struct {
	Args
	ConfigAI *ConfigAI
}

type AIClient struct {
	OpenAI *OpenAIClient
	Gemini *GeminiClient
}

type AInterrupted struct {
	Args
	IntByTimeout bool `log:"int_by_timeout"`
}

type APrompt struct {
	Args
	// Prompt is a prompt to be sent to LLM.
	Prompt string `log:"prompt"`
}

type ARequestingAI struct {
	Args
	Length string `log:"length"`
	Source string `log:"source"`
}

type ARequestingTool struct {
	Args
	Name   string `log:"name"`
	Source string `log:"source"`
	Target string `log:"target"`
}

type ASSHConn struct {
	Args
	SSHSess ssh.Session
	// Done is a buffered channel to be closed by the receiver
	Done chan<- struct{}
}

type ASSHDisconn struct {
	Args
	Addr string `log:"addr"`
	// Id of the disconnected TUI.
	Id string `log:"id"`
}

type ASSHReady struct {
	Args
	SSHServer *ssh.Server
}

type AStoryAction struct {
	Args
	// ID of the action.
	ID string `log:"id"`
}

type AStoryChanged struct {
	Args
	StatesList   []string `log:"states_list"`
	ActivateList []bool   `log:"activate_list"`
}

type AUIMsg struct {
	Args
	// Msg is a single message with an author and text.
	Msg *Msg `log:"msg"`
}

type AUIRenderClock struct {
	Args
	ClockDiff [][]int
}

type AUIRenderStories struct {
	Args
	Actions []ActionInfo `log:"actions"`
	Stories []StoryInfo  `log:"stories"`
}

type Action struct {
	// random ID
	ID string

	// Current value.
	Value func() int
	// Maximum value.
	ValueEnd func() int

	Label  string
	Desc   string
	Action func()

	StateAdd    string
	StateRemove string

	VisibleAgent amhelp.Cond
	VisibleMem   amhelp.Cond
	IsDisabled   func() bool
	LabelEnd     string

	Pos         int
	PosInferred bool
}

type ActionInfo struct {
	// random ID
	ID string

	// Current value.
	Value int
	// Maximum value.
	ValueEnd int

	Label  string
	Desc   string
	Action bool

	StateAdd    string
	StateRemove string

	VisibleAgent bool
	VisibleMem   bool
	IsDisabled   bool
	LabelEnd     string

	Pos         int
	PosInferred bool
}

type AgentAPI interface {
	AgentBaseAPI

	Msgs() []*Msg
	Splash() string
	MachSchema() (am.Schema, am.S)
	Actions() []ActionInfo
	Stories() []StoryInfo
	Story(state string) *Story
	DBAgent() *sql.DB
	MachMem() *am.Machine
	// Moves return a list of moves and their descriptions for orienting.
	OrientingMoves() map[string]string
	// HistoryStates returns a list of states to track in the history.
	HistoryStates() am.S
}

type AgentBaseAPI interface {
	// AgentImpl returns the top-level agent implementation.
	AgentImpl() AgentAPI
	// Output outputs text to the user.
	Output(txt string, from From) am.Result

	Mach() *am.Machine
	DBG() *debugger.Debugger
	Hist() (history.MemoryApi, error)

	SetConfigBase(*Config) error
	ConfigBase() *Config
	AIs() []*AIClient

	Start() am.Result
	Stop(disposeCtx context.Context)
	Log(msg string, args ...any)
	LogTrace(msg string, args ...any)
	LogDebug(msg string, args ...any)
	LogWarn(msg string, err error, args ...any)
	LogErr(msg string, err error, args ...any)

	LogCtx(ctx context.Context, txt string, args ...any)
	LogCtxDebug(ctx context.Context, txt string, args ...any)
	LogCtxTrace(ctx context.Context, txt string, args ...any)
	LogCtxWarn(ctx context.Context, txt string, err error, args ...any)
	LogCtxErr(ctx context.Context, msg string, err error, args ...any)

	LogEv(e *am.Event, txt string, args ...any)
	LogEvDebug(e *am.Event, txt string, args ...any)
	LogEvTrace(e *am.Event, txt string, args ...any)
	LogEvWarn(e *am.Event, txt string, err error, args ...any)
	LogEvErr(e *am.Event, msg string, err error, args ...any)

	Logger() *slog.Logger
	Store() *AgentStore

	QueriesBase() *sqlc.Queries

	DBBase() *sql.DB
	DBHistory() *sql.DB
}

type AgentInit interface {
	Init(
		agentImpl AgentAPI, cfg *Config, logArgs am.LogArgsMapperFn, groups any, states am.States, args []am.ArgsApi,
	) error
}

type AgentQueries[TQueries any] interface {
	Queries() *TQueries
}

type AgentStore struct {
	M         map[string]any
	ClockDiff [][]int
	Web       fs.FS
}

type Args struct {
	am.ArgsBase
}

type Config struct {
	AI    ConfigAI
	Agent ConfigAgent
	Web   ConfigWeb
	TUI   ConfigTUI
	Tools ConfigTools
	Debug ConfigDebug

	// File is the path to the loaded config file.
	File      string `kdl:"-"`
	ProdBuild bool   `kdl:"-"`
	// ServerMode doesnt autorun the GUI and listens on 0.0.0.0
	ServerMode bool `kdl:"-"`
	DockerMode bool `kdl:"-"`
	// random instance ID
	SessionID string `kdl:"-"`
}

type ConfigAI struct {
	OpenAI []ConfigAIOpenAI `kdl:"OpenAI,multiple"`
	Gemini []ConfigAIGemini `kdl:"Gemini,multiple"`
	// Max LLM requests per session.
	ReqLimit int
}

type ConfigAICommon struct {
	Model    string
	Key      string
	Disabled bool

	Name string `kdl:",omitempty"`
	// Def: 3
	Retries  int      `kdl:",omitempty"`
	Tags     []string `kdl:",omitempty"`
	Priority int      `kdl:",omitempty"`
	// For how long blacklist this provider on error (def 1h).
	FallbackTimeout time.Duration `kdl:",omitempty"`
	// Number of concurrent req for this provider TODO
	Concurrency int    `kdl:",omitempty"`
	URL         string `kdl:",omitempty"`

	// temporarily disabled (eg offline)
	DisabledUntil time.Time `kdl:"-"`
	// provider name
	API string `kdl:"-"`
	// num of times this backend was called without errs
	Calls int `kdl:"-"`
}

type ConfigAIGemini struct {
	ConfigAICommon

	// NoEnforceSchema skips `instr.WithMode(instr.ModeJSONSchema)` for local models
	NoEnforceSchema bool
}

type ConfigAIOpenAI struct {
	ConfigAICommon

	// NoEnforceSchema skips `instr.WithMode(instr.ModeJSONSchema)` for local models
	NoEnforceSchema bool
}

type ConfigAgent struct {
	// bot ID
	ID    string
	Label string
	// // dir for tmp files, defaults to CWD
	Dir       string
	Intro     string
	IntroDash string
	Footer    string
	Log       ConfigAgentLog
	History   ConfigAgentHistory
}

type ConfigAgentHistory struct {
	Backend string
	// TODO BackendParsed enum
	Max int
}

type ConfigAgentLog struct {
	// path to the log file
	File string
	// log prompts as log level debug
	Prompts bool
	// mark log errs as ErrLog multi state
	FwdErrState bool
	// mark log warning as WarnLog multi state
	FwdWarnState bool
	// mach log level 0-5
	MachLevel am.LogLevel
	// print machine log
	MachPrint bool
	// agent log level
	Level slog.Level
	// enable instructor-go logging for AI requests
	AI bool
}

type ConfigDebug struct {
	Root *Config `kdl:"-" json:"-"`

	// Display extra info about these stories in the machine log
	Story []string
	// Run the mock scenario
	Mock bool
	// Start pprof on addr
	ProfilerAddr string
	// Enable misc debugging modes (am-relay, browser RPC, TRACE logs)
	Verbose bool
	// Enable SQL query logging
	VerboseSQL bool
	// Create value files for inspection
	ValFiles bool
	// Enable REPL for agent, mem, and tools
	REPL bool
	// Connect and send dbg info to am-dbg
	DBGAddr string

	// Start an embedded debugger on localhost:{DBGAddr}
	DBGEmbed bool
	// Expose the embedded debugger on http://localhost:{DBGEmbedWeb}
	DBGEmbedWeb int
	// Start a web REPL on http://localhost:{REPLWeb}
	REPLWeb int
}

type ConfigSearXNG struct {
	// Port to start a local instance on
	Port string
	// URL of an existing instance (disables Port).
	URL string
}

type ConfigTUI struct {
	Root *Config `kdl:"-" json:"-"`

	// TODO Addr
	// TODO WebAddr
	// SSH host
	Host string
	// SSH port
	PortSSH int
	// Web host
	WebHost string
	// Web port
	PortWeb int
	// Number of transitions to show on the clock
	ClockRange int
}

type ConfigTools struct {
	SearXNG ConfigSearXNG
}

type ConfigWeb struct {
	Root *Config `kdl:"-" json:"-"`

	// Base address for HTTP services:
	// - main HTTP server
	// - +1 WebSocket addr of the agent's RPC server (dashboard)
	// - +2 WebSocket addr of the agent's RPC server (agent UI)
	Addr string
	// Start a DBPort web UI on http://localhost:{DBPort[0-2]}
	DBPort int
	// Start a log web UI on http://localhost:{LogPort}
	LogPort int
}

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

type FeedItem struct {
	Title  string
	Author string
	Link   string
	Date   string
	Desc   string
}

type From enum.Member[string]

type GeminiClient struct {
	Cfg *ConfigAIGemini
	C   *instructor.InstructorGoogle
}

type HasPromptUsage interface {
	GetPromptUsage() *PromptUsage
}

type MCPCallToolParams struct {
	Name      string          `json:"name" jsonschema:"description=The name of the target MCP tool"`
	Arguments json.RawMessage `json:"arguments" jsonschema:"description=JSON object of arguments matching the tool schema"`
}

type MCPCallToolRequest struct {
	JSONRPC string            `json:"jsonrpc" jsonschema:"const=2.0"`
	ID      int               `json:"id" jsonschema:"description=Request identifier"`
	Method  string            `json:"method" jsonschema:"const=tools/call"`
	Params  MCPCallToolParams `json:"params"`
}

type Msg struct {
	From      From
	Text      string
	CreatedAt time.Time
}

type OfferRef struct {
	// Index from 0
	Index int
	Text  string
}

type OpenAIClient struct {
	Cfg *ConfigAIOpenAI
	C   *instructor.InstructorOpenAI
}

type Prompt[P any, R any] struct {
	Conditions string
	Steps      string
	Result     string
	State      string
	A          AgentBaseAPI
	// number of previous messages to include (excluding system msg)
	HistoryMsgLen int
	// Msg holds msgs for this prompt and is used to build the Conversation instance (on-demand)
	Msgs      []*PromptMsg
	Thread    string
	Reasoning string
	LastModel string
	Godoc     map[string]string
	// contains filtered or unexported fields
}

type PromptApi interface {
	AddTool(tool ToolApi)
	AddDoc(doc *Document)

	GenSysPrompt() string
	Conversation() (*core.Conversation, string)
	HistClean()
}

type PromptConfig struct {
	ModelTag     string
	ModelSortTag string
	Reasoning    string
	MaxTries     int
	Concurrency  int
}

type PromptMsg struct {
	From    core.Role
	Content string
}

type PromptOpts struct {

	// required provider tags
	RequiredTags []string
	// disallowed provider tags
	NoTags []string
	// skip checking if state is active
	SkipState bool
	Reasoning string
}

type PromptRaw struct {
	Conditions string
	Steps      string
	Result     string
	// TODO use better schema pkg
	ResultSchema *oaischema.Definition
	State        string
	A            AgentBaseAPI
	// number of previous messages to include
	HistoryMsgLen int
	Msgs          []*PromptRawMsg
	// contains filtered or unexported fields
}

type PromptRawMsg struct {
	Role    PromptRawRole
	Content string
}

type PromptRawRole string

type PromptSchemaless = Prompt[any, any]

type PromptUsage struct {
	Model              string
	API                string
	URL                string
	PromptTokens       int
	PromptTokensCached int
	CompletionTokens   int
	TotalTokens        int
	PromptChars        int
	HistoryChars       int
	Provider           string
}

type RawAIClient struct {
	OpenAI *RawOpenAIClient
	Gemini *RawGeminiClient
}

type RawGeminiClient struct {
	Cfg *ConfigAIGemini
	C   *genai.Client
}

type RawOpenAIClient struct {
	Cfg *ConfigAIOpenAI
	C   *openai.Client
}

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

type Story struct {
	StoryInfo

	Agent  StoryActor
	Memory StoryActor

	// If is an optional function used to confirm that this story can activate. It has access to the whole story struct,
	// so all the involved state machines and their historical snapshots (relative to activation and deactivation of
	// this story).
	CanActivate func(instance *Story) bool
	Actions     []Action
}

type StoryActionsByIdx []Action

type StoryActor struct {
	Mach *am.Machine

	// actor's time when the story activated
	TimeActivated   am.Time
	TimeDeactivated am.Time

	// When these conditions are met, the story will activate itself.
	Trigger amhelp.Cond
}

type StoryInfo struct {
	// Name of the bound state (eg StoryFoo).
	State string
	// Tick is the current tick of the bound state.
	Tick uint64
	// Epoch is the sum of all previous memories, before a replacement.
	Epoch uint64

	// Title of this story.
	Title string
	// Description of this story.
	Desc string

	// The story was last deactivated at this human time.
	DeactivatedAt time.Time
	// The story was last active for this many ticks of the AgentLLM machine.
	LastActiveTicks uint64
}

type ToolApi interface {
	Mach() *am.Machine
	SetMach(*am.Machine)
	Document() *Document
}

type ValEnc enum.Member[string]

type ValFileEnc = ValEnc