agent

package

v0.0.0-...-32dd017 Latest Latest Go to latest Published: Apr 25, 2026 License: MIT Imports: 21 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

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

Agent orchestrates LLM tool-use conversations.

func New(
	log logger.Logger,
	llmProvider provider.Provider,
	services *baseservices.BaseServices,
	badgerDB *badgerdb.DB,
) *Agent

New creates a new Agent.

func (a *Agent) ClearConversation(ctx context.Context, userID int64) error

ClearConversation clears the conversation history for a user.

func (a *Agent) Process(ctx context.Context, userID int64, text string) (string, error)

Process handles a user message: loads conversation history, runs the agent loop, and returns the final text response.

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

ConversationStore manages per-user conversation history in BadgerDB.

func NewConversationStore(db *badgerdb.DB) *ConversationStore

NewConversationStore creates a new ConversationStore.

func (s *ConversationStore) Clear(ctx context.Context, userID int64) error

Clear removes conversation history for a user.

func (s *ConversationStore) Load(ctx context.Context, userID int64) ([]provider.ChatMessage, error)

Load retrieves the conversation history for a user. Returns empty slice if no history exists.

func (s *ConversationStore) Save(ctx context.Context, userID int64, messages []provider.ChatMessage) error

Save stores conversation history, keeping only the last N messages.

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

Executor dispatches tool calls to the appropriate service methods.

func NewExecutor(services *baseservices.BaseServices) *Executor

NewExecutor creates a new Executor.

func (e *Executor) Execute(ctx context.Context, userID int64, call provider.ToolCall) (string, error)

Execute runs a tool call and returns the JSON result string.