agentloop

type AddArtifactArgs struct {
	Path     string            `json:"path"`
	Metadata map[string]string `json:"metadata,omitempty"`
}

type AddTodoArgs struct {
	Todos []TodoItemInput `json:"todos"`
}

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

type AgentConfig struct {
	// Name is an optional identifier for the agent used in logs.
	// If empty, defaults to "AgentLoop".
	Name string

	// Model is the LLM model to use.
	Model model.ToolCallingChatModel

	// MaxRunSteps limits the maximum number of ReAct rounds (tool-call cycles) the agent may execute.
	// Each round corresponds to one tool-calling iteration; the value is multiplied by 5 internally
	// to derive the actual Eino graph step budget (1 round ≈ 4 graph steps, ×5 includes a safety margin).
	// If 0 or negative, the step budget is determined automatically by Eino (node count + 10).
	MaxRunSteps int

	// Language specifies the language for agent responses.
	// If empty, defaults to "English".
	Language string

	// LogOnStart controls whether the agent logs when it starts processing.
	// If nil, defaults to true.
	LogOnStart *bool

	// LogOnEnd controls whether the agent logs token stats when it finishes.
	// If nil, defaults to true.
	LogOnEnd *bool

	// LogInputs controls whether the agent logs input messages to the model.
	// If nil, defaults to false.
	LogInputs *bool

	// LogOutputs controls whether the agent logs model output content.
	// If nil, defaults to true.
	LogOutputs *bool

	// Skills is a list of skill paths to load.
	// Skills are loaded from the backend and injected into the system prompt.
	Skills []string

	// PreloadedSkills is a map of file paths to content that will be written to the backend
	// before loading skills. This is useful for predefining skills when using TmpFileStore.
	// Example: {"/skills/web-research/SKILL.md": "# Web Research\n\n..."}
	//
	// See [BuildPreloadedSkills], [BuildPreloadedSkillsWithFilter]
	PreloadedSkills map[string]string

	// Tools is a list of custom tools to add to the built-in tools.
	// Built-in tools are always registered; this field adds additional tools.
	//
	// Create custom tools using helper functions:
	//   - [NewToolFunc] - for simple tools with map-based arguments
	//   - [NewCtxAwareToolFunc] - for tools needing AgentContext (Store, Todos)
	//   - [NewTypedToolFunc] - for tools with typed struct arguments
	//   - [NewTypedCtxAwareToolFunc] - for tools with typed arguments and AgentContext
	Tools []Tool

	// TaskPrompt is the main task prompt for the agent.
	TaskPrompt string

	// SystemPrompt is an optional custom system prompt.
	// If provided, it will be prepended to the base system prompt.
	SystemPrompt string

	// BackendFactory is a factory function that creates a fresh FileStore for each execution.
	// If nil, a new TmpFileStore will be created for each execution.
	// This allows stateful backends to be created fresh per execution.
	BackendFactory func() FileStore

	// Timezone is the timezone offset in hours for time display (default: 0, UTC).
	Timezone float64

	// MaxTokens is the maximum number of tokens allowed in the conversation history.
	// When exceeded, old messages will be pruned (except system messages).
	// If 0 or negative, no token limit is enforced.
	// Default: 0 (no limit)
	MaxTokens int

	// TokenizerModelName is the name of the model used for token counting.
	// This is used to select the appropriate tiktoken encoding (e.g., cl100k_base for gpt-3.5-turbo, o200k_base for gpt-4o).
	// If empty, defaults to "gpt-3.5-turbo".
	// Common values: "gpt-3.5-turbo", "gpt-4", "gpt-4o", "qwen-plus", "deepseek-chat"
	TokenizerModelName string

	// EvictToolResultsThreshold is the maximum token count for tool results before eviction.
	// Tool results exceeding this threshold are evicted to the filesystem and replaced with a reference.
	// If 0 or negative, no eviction is performed.
	// Default: 0 (no eviction)
	// Recommended: 1000-2000 tokens for most use cases
	EvictToolResultsThreshold int

	// EvictToolResultsKeepPreview is the number of tokens to keep as a preview in the reference message.
	// Allows the agent to see context without loading the full content.
	// If 0, no preview is kept (only metadata).
	// Default: 0 (no preview)
	EvictToolResultsKeepPreview int

	// EnableFileTool enables the built-in file tools: read_file, write_file, edit_file,
	// list_directory, glob, and add_artifact. When false, these tools are not registered.
	// Default: false
	EnableFileTool bool
}

type AgentContext struct {
	Store     FileStore
	Todos     *TodoManager
	Artifacts *ArtifactManager
}

type AgentRequest struct {
	UserInput           string
	PreloadBackendFiles func(store FileStore) error                       // Optional callback to preload files into the backend before execution
	ArtifactCallback    func(store FileStore, artifacts []Artifact) error // Optional callback for artifacts
}

type Artifact struct {
	Path        string            // Backend file path
	SizeInBytes int64             // File size in bytes
	Meta        map[string]string // Additional metadata (title, url, etc.)
}

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

type BuiltinToolsOption struct {
	// EnableFileTool enables the file-related tools: read_file, write_file, edit_file,
	// list_directory, glob, and add_artifact. Default: false.
	EnableFileTool bool
}

type DeleteTodoArgs struct {
	IDs []string `json:"ids"`
}

type EditFileArgs struct {
	Path       string `json:"path"`
	OldString  string `json:"old_string"`
	NewString  string `json:"new_string"`
	ReplaceAll bool   `json:"replace_all,omitempty"`
}

type FileInfo struct {
	Path       string    `json:"path"`
	IsDir      bool      `json:"is_dir"`
	Size       int64     `json:"size"`
	ModifiedAt time.Time `json:"modified_at"`
}

type FileStore interface {
	// ReadFile reads a file from the backend.
	ReadFile(ctx context.Context, path string) ([]byte, error)

	// WriteFile writes content to a file in the backend.
	WriteFile(ctx context.Context, path string, content []byte) error

	// ListDirectory lists files and directories in a path.
	ListDirectory(ctx context.Context, path string) ([]FileInfo, error)

	// FileExists checks if a file exists.
	FileExists(ctx context.Context, path string) (bool, error)

	// DeleteFile deletes a file.
	DeleteFile(ctx context.Context, path string) error
}

type GlobArgs struct {
	Pattern string `json:"pattern"`
}

type ListDirectoryArgs struct {
	Path string `json:"path"`
}

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

type ReadFileArgs struct {
	Path   string `json:"path"`
	Offset int    `json:"offset,omitempty"`
	Limit  int    `json:"limit,omitempty"`
}

type SelfInvokeTool interface {
	ExecuteJson(ctx context.Context, jsonArg string) (string, error)
}

type SessionAware interface {
	// OnSessionStart is called when an agent session begins, before any file operations.
	// Implementations should perform any initialisation here (e.g. creating a tmp directory).
	OnSessionStart(rail flow.Rail) error

	// OnSessionEnd is called when an agent session ends.
	// Implementations should release resources here (e.g. removing the tmp directory).
	OnSessionEnd(rail flow.Rail) error
}

type Skill struct {
	Metadata SkillMetadata
	Path     string
	Content  string // The markdown content (excluding YAML frontmatter)
}

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

type SkillMetadata struct {
	Name         string   `yaml:"name"`
	Description  string   `yaml:"description"`
	License      string   `yaml:"license,omitempty"`
	Compatible   string   `yaml:"compatibility,omitempty"`
	Metadata     string   `yaml:"metadata,omitempty"`
	AllowedTools []string `yaml:"allowed_tools,omitempty"`
}

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

type SkillsMap map[string]*Skill

type TaskOutput struct {
	Response  string     // Main response (research report)
	Artifacts []Artifact // Artifacts collected during execution
}

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

type TodoItem struct {
	ID          string `json:"id"`
	Task        string `json:"task"`
	Status      string `json:"status"` // "pending", "completed"
	Description string `json:"description,omitempty"`
}

type TodoItemInput struct {
	Task        string `json:"task"`
	Description string `json:"description,omitempty"`
}

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

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

type Tool interface {
	// Name returns the name of the tool.
	Name() string

	// Description returns a description of the tool.
	Description() string

	// Parameters returns the JSON schema for the tool parameters.
	Parameters() map[string]*schema.ParameterInfo

	// Execute executes the tool with the given arguments.
	Execute(ctx context.Context, args map[string]interface{}) (string, error)
}

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

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

type TypedCtxAwareToolFunc[T any] struct {
	// contains filtered or unexported fields
}

type TypedToolFunc[T any] struct {
	// contains filtered or unexported fields
}

type UpdateTodoArgs struct {
	ID     string `json:"id,omitempty"`
	Status string `json:"status,omitempty"`
}

type WriteFileArgs struct {
	Path    string `json:"path"`
	Content string `json:"content"`
}