agent

Documentation

Overview

Package agent is the core orchestration layer for Revcli AI agents.

It provides session-based AI agent functionality for managing conversations, tool execution, and message handling. It coordinates interactions between language models, messages, sessions, and tools while handling features like automatic summarization, queuing, and token management.

Index

• Constants
• Variables
• func InitializePrompt(cfg config.Config) (string, error)
• type AgentParams
• type Coordinator
  • func NewCoordinator(ctx context.Context, cfg *config.Config, sessions session.Service, ...) (Coordinator, error)
• type Model
• type SessionAgent
  • func NewSessionAgent(opts SessionAgentOptions) SessionAgent
• type SessionAgentCall
• type SessionAgentOptions

Constants

const (
	AgentToolName = "agent"
)

Variables

var (
	ErrRequestCancelled = errors.New("request canceled by user")
	ErrSessionBusy      = errors.New("session is currently processing another request")
	ErrEmptyPrompt      = errors.New("prompt is empty")
	ErrSessionMissing   = errors.New("session id is missing")
)

Functions

func InitializePrompt

func InitializePrompt(cfg config.Config) (string, error)

Types

type AgentParams

type AgentParams struct {
	Prompt string `json:"prompt" description:"The task for the agent to perform"`
}

type Coordinator

type Coordinator interface {
	// INFO: (kujtim) this is not used yet we will use this when we have multiple agents
	// SetMainAgent(string)
	Run(ctx context.Context, sessionID, prompt string, attachments ...message.Attachment) (*fantasy.AgentResult, error)
	Cancel(sessionID string)
	CancelAll()
	IsSessionBusy(sessionID string) bool
	IsBusy() bool
	QueuedPrompts(sessionID string) int
	QueuedPromptsList(sessionID string) []string
	ClearQueue(sessionID string)
	Summarize(context.Context, string) error
	Model() Model
	UpdateModels(ctx context.Context) error
}

func NewCoordinator

func NewCoordinator(
	ctx context.Context,
	cfg *config.Config,
	sessions session.Service,
	messages message.Service,
	permissions permission.Service,
	history history.Service,
	lspClients *csync.Map[string, *lsp.Client],
) (Coordinator, error)

type Model

type Model struct {
	Model      fantasy.LanguageModel
	CatwalkCfg catwalk.Model
	ModelCfg   config.SelectedModel
}

type SessionAgent

type SessionAgent interface {
	Run(context.Context, SessionAgentCall) (*fantasy.AgentResult, error)
	SetModels(large Model, small Model)
	SetTools(tools []fantasy.AgentTool)
	Cancel(sessionID string)
	CancelAll()
	IsSessionBusy(sessionID string) bool
	IsBusy() bool
	QueuedPrompts(sessionID string) int
	QueuedPromptsList(sessionID string) []string
	ClearQueue(sessionID string)
	Summarize(context.Context, string, fantasy.ProviderOptions) error
	Model() Model
}

func NewSessionAgent

func NewSessionAgent(
	opts SessionAgentOptions,
) SessionAgent

type SessionAgentCall

type SessionAgentCall struct {
	SessionID        string
	Prompt           string
	ProviderOptions  fantasy.ProviderOptions
	Attachments      []message.Attachment
	MaxOutputTokens  int64
	Temperature      *float64
	TopP             *float64
	TopK             *int64
	FrequencyPenalty *float64
	PresencePenalty  *float64
}

type SessionAgentOptions

type SessionAgentOptions struct {
	LargeModel           Model
	SmallModel           Model
	SystemPromptPrefix   string
	SystemPrompt         string
	IsSubAgent           bool
	DisableAutoSummarize bool
	IsYolo               bool
	Sessions             session.Service
	Messages             message.Service
	Tools                []fantasy.AgentTool
}