tools

package
v0.1.2 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Feb 17, 2026 License: MIT Imports: 22 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func ToolToSchema 

func ToolToSchema(tool Tool) map[string]interface{}

Types

type AppendFileTool 

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

func NewAppendFileTool 

func NewAppendFileTool(workspace string, restrict bool) *AppendFileTool

func (*AppendFileTool) Description 

func (t *AppendFileTool) Description() string

func (*AppendFileTool) Execute 

func (t *AppendFileTool) Execute(ctx context.Context, args map[string]interface{}) *ToolResult

func (*AppendFileTool) Name 

func (t *AppendFileTool) Name() string

func (*AppendFileTool) Parameters 

func (t *AppendFileTool) Parameters() map[string]interface{}

type AsyncCallback added in v0.1.2 

type AsyncCallback func(ctx context.Context, result *ToolResult)

AsyncCallback is a function type that async tools use to notify completion. When an async tool finishes its work, it calls this callback with the result.

The ctx parameter allows the callback to be canceled if the agent is shutting down. The result parameter contains the tool's execution result.

Example usage in an async tool: 

func (t *MyAsyncTool) Execute(ctx context.Context, args map[string]interface{}) *ToolResult {
    // Start async work in background
    go func() {
        result := doAsyncWork()
        if t.callback != nil {
            t.callback(ctx, result)
        }
    }()
    return AsyncResult("Async task started")
}

type AsyncTool added in v0.1.2 

type AsyncTool interface {
	Tool
	// SetCallback registers a callback function to be invoked when the async operation completes.
	// The callback will be called from a goroutine and should handle thread-safety if needed.
	SetCallback(cb AsyncCallback)
}

AsyncTool is an optional interface that tools can implement to support asynchronous execution with completion callbacks.

Async tools return immediately with an AsyncResult, then notify completion via the callback set by SetCallback.

This is useful for: - Long-running operations that shouldn't block the agent loop - Subagent spawns that complete independently - Background tasks that need to report results later

Example: 

type SpawnTool struct {
    callback AsyncCallback
}

func (t *SpawnTool) SetCallback(cb AsyncCallback) {
    t.callback = cb
}

func (t *SpawnTool) Execute(ctx context.Context, args map[string]interface{}) *ToolResult {
    go t.runSubagent(ctx, args)
    return AsyncResult("Subagent spawned, will report back")
}

type BraveSearchProvider added in v0.1.2 

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

func (*BraveSearchProvider) Search added in v0.1.2 

func (p *BraveSearchProvider) Search(ctx context.Context, query string, count int) (string, error)

type ContextualTool added in v0.1.1 

type ContextualTool interface {
	Tool
	SetContext(channel, chatID string)
}

ContextualTool is an optional interface that tools can implement to receive the current message context (channel, chatID)

type CronTool added in v0.1.1 

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

CronTool provides scheduling capabilities for the agent

func NewCronTool added in v0.1.1 

func NewCronTool(cronService *cron.CronService, executor JobExecutor, msgBus *bus.MessageBus, workspace string, restrict bool) *CronTool

NewCronTool creates a new CronTool

func (*CronTool) Description added in v0.1.1 

func (t *CronTool) Description() string

Description returns the tool description

func (*CronTool) Execute added in v0.1.1 

func (t *CronTool) Execute(ctx context.Context, args map[string]interface{}) *ToolResult

Execute runs the tool with the given arguments

func (*CronTool) ExecuteJob added in v0.1.1 

func (t *CronTool) ExecuteJob(ctx context.Context, job *cron.CronJob) string

ExecuteJob executes a cron job through the agent

func (*CronTool) Name added in v0.1.1 

func (t *CronTool) Name() string

Name returns the tool name

func (*CronTool) Parameters added in v0.1.1 

func (t *CronTool) Parameters() map[string]interface{}

Parameters returns the tool parameters schema

func (*CronTool) SetContext added in v0.1.1 

func (t *CronTool) SetContext(channel, chatID string)

SetContext sets the current session context for job creation

type DuckDuckGoSearchProvider added in v0.1.2 

type DuckDuckGoSearchProvider struct{}

func (*DuckDuckGoSearchProvider) Search added in v0.1.2 

func (p *DuckDuckGoSearchProvider) Search(ctx context.Context, query string, count int) (string, error)

type EditFileTool 

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

EditFileTool edits a file by replacing old_text with new_text. The old_text must exist exactly in the file.

func NewEditFileTool 

func NewEditFileTool(allowedDir string, restrict bool) *EditFileTool

NewEditFileTool creates a new EditFileTool with optional directory restriction.

func (*EditFileTool) Description 

func (t *EditFileTool) Description() string

func (*EditFileTool) Execute 

func (t *EditFileTool) Execute(ctx context.Context, args map[string]interface{}) *ToolResult

func (*EditFileTool) Name 

func (t *EditFileTool) Name() string

func (*EditFileTool) Parameters 

func (t *EditFileTool) Parameters() map[string]interface{}

type ExecTool 

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

func NewExecTool 

func NewExecTool(workingDir string, restrict bool) *ExecTool

func (*ExecTool) Description 

func (t *ExecTool) Description() string

func (*ExecTool) Execute 

func (t *ExecTool) Execute(ctx context.Context, args map[string]interface{}) *ToolResult

func (*ExecTool) Name 

func (t *ExecTool) Name() string

func (*ExecTool) Parameters 

func (t *ExecTool) Parameters() map[string]interface{}

func (*ExecTool) SetAllowPatterns 

func (t *ExecTool) SetAllowPatterns(patterns []string) error

func (*ExecTool) SetRestrictToWorkspace 

func (t *ExecTool) SetRestrictToWorkspace(restrict bool)

func (*ExecTool) SetTimeout 

func (t *ExecTool) SetTimeout(timeout time.Duration)

type FunctionCall 

type FunctionCall struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments"`
}

type I2CTool added in v0.1.2 

type I2CTool struct{}

I2CTool provides I2C bus interaction for reading sensors and controlling peripherals.

func NewI2CTool added in v0.1.2 

func NewI2CTool() *I2CTool

func (*I2CTool) Description added in v0.1.2 

func (t *I2CTool) Description() string

func (*I2CTool) Execute added in v0.1.2 

func (t *I2CTool) Execute(ctx context.Context, args map[string]interface{}) *ToolResult

func (*I2CTool) Name added in v0.1.2 

func (t *I2CTool) Name() string

func (*I2CTool) Parameters added in v0.1.2 

func (t *I2CTool) Parameters() map[string]interface{}

type JobExecutor added in v0.1.1 

type JobExecutor interface {
	ProcessDirectWithChannel(ctx context.Context, content, sessionKey, channel, chatID string) (string, error)
}

JobExecutor is the interface for executing cron jobs through the agent

type LLMProvider 

type LLMProvider interface {
	Chat(ctx context.Context, messages []Message, tools []ToolDefinition, model string, options map[string]interface{}) (*LLMResponse, error)
	GetDefaultModel() string
}

type LLMResponse 

type LLMResponse struct {
	Content      string     `json:"content"`
	ToolCalls    []ToolCall `json:"tool_calls,omitempty"`
	FinishReason string     `json:"finish_reason"`
	Usage        *UsageInfo `json:"usage,omitempty"`
}

type ListDirTool 

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

func NewListDirTool added in v0.1.1 

func NewListDirTool(workspace string, restrict bool) *ListDirTool

func (*ListDirTool) Description 

func (t *ListDirTool) Description() string

func (*ListDirTool) Execute 

func (t *ListDirTool) Execute(ctx context.Context, args map[string]interface{}) *ToolResult

func (*ListDirTool) Name 

func (t *ListDirTool) Name() string

func (*ListDirTool) Parameters 

func (t *ListDirTool) Parameters() map[string]interface{}

type Message 

type Message struct {
	Role       string     `json:"role"`
	Content    string     `json:"content"`
	ToolCalls  []ToolCall `json:"tool_calls,omitempty"`
	ToolCallID string     `json:"tool_call_id,omitempty"`
}

type MessageTool 

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

func NewMessageTool 

func NewMessageTool() *MessageTool

func (*MessageTool) Description 

func (t *MessageTool) Description() string

func (*MessageTool) Execute 

func (t *MessageTool) Execute(ctx context.Context, args map[string]interface{}) *ToolResult

func (*MessageTool) HasSentInRound added in v0.1.2 

func (t *MessageTool) HasSentInRound() bool

HasSentInRound returns true if the message tool sent a message during the current round.

func (*MessageTool) Name 

func (t *MessageTool) Name() string

func (*MessageTool) Parameters 

func (t *MessageTool) Parameters() map[string]interface{}

func (*MessageTool) SetContext 

func (t *MessageTool) SetContext(channel, chatID string)

func (*MessageTool) SetSendCallback 

func (t *MessageTool) SetSendCallback(callback SendCallback)

type ReadFileTool 

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

func NewReadFileTool added in v0.1.1 

func NewReadFileTool(workspace string, restrict bool) *ReadFileTool

func (*ReadFileTool) Description 

func (t *ReadFileTool) Description() string

func (*ReadFileTool) Execute 

func (t *ReadFileTool) Execute(ctx context.Context, args map[string]interface{}) *ToolResult

func (*ReadFileTool) Name 

func (t *ReadFileTool) Name() string

func (*ReadFileTool) Parameters 

func (t *ReadFileTool) Parameters() map[string]interface{}

type SPITool added in v0.1.2 

type SPITool struct{}

SPITool provides SPI bus interaction for high-speed peripheral communication.

func NewSPITool added in v0.1.2 

func NewSPITool() *SPITool

func (*SPITool) Description added in v0.1.2 

func (t *SPITool) Description() string

func (*SPITool) Execute added in v0.1.2 

func (t *SPITool) Execute(ctx context.Context, args map[string]interface{}) *ToolResult

func (*SPITool) Name added in v0.1.2 

func (t *SPITool) Name() string

func (*SPITool) Parameters added in v0.1.2 

func (t *SPITool) Parameters() map[string]interface{}

type SearchProvider added in v0.1.2 

type SearchProvider interface {
	Search(ctx context.Context, query string, count int) (string, error)
}

type SendCallback 

type SendCallback func(channel, chatID, content string) error

type SpawnTool 

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

func NewSpawnTool 

func NewSpawnTool(manager *SubagentManager) *SpawnTool

func (*SpawnTool) Description 

func (t *SpawnTool) Description() string

func (*SpawnTool) Execute 

func (t *SpawnTool) Execute(ctx context.Context, args map[string]interface{}) *ToolResult

func (*SpawnTool) Name 

func (t *SpawnTool) Name() string

func (*SpawnTool) Parameters 

func (t *SpawnTool) Parameters() map[string]interface{}

func (*SpawnTool) SetCallback added in v0.1.2 

func (t *SpawnTool) SetCallback(cb AsyncCallback)

SetCallback implements AsyncTool interface for async completion notification

func (*SpawnTool) SetContext 

func (t *SpawnTool) SetContext(channel, chatID string)

type SubagentManager 

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

func NewSubagentManager 

func NewSubagentManager(provider providers.LLMProvider, defaultModel, workspace string, bus *bus.MessageBus) *SubagentManager

func (*SubagentManager) GetTask 

func (sm *SubagentManager) GetTask(taskID string) (*SubagentTask, bool)

func (*SubagentManager) ListTasks 

func (sm *SubagentManager) ListTasks() []*SubagentTask

func (*SubagentManager) RegisterTool added in v0.1.2 

func (sm *SubagentManager) RegisterTool(tool Tool)

RegisterTool registers a tool for subagent execution.

func (*SubagentManager) SetTools added in v0.1.2 

func (sm *SubagentManager) SetTools(tools *ToolRegistry)

SetTools sets the tool registry for subagent execution. If not set, subagent will have access to the provided tools.

func (*SubagentManager) Spawn 

func (sm *SubagentManager) Spawn(ctx context.Context, task, label, originChannel, originChatID string, callback AsyncCallback) (string, error)

type SubagentTask 

type SubagentTask struct {
	ID            string
	Task          string
	Label         string
	OriginChannel string
	OriginChatID  string
	Status        string
	Result        string
	Created       int64
}

type SubagentTool added in v0.1.2 

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

SubagentTool executes a subagent task synchronously and returns the result. Unlike SpawnTool which runs tasks asynchronously, SubagentTool waits for completion and returns the result directly in the ToolResult.

func NewSubagentTool added in v0.1.2 

func NewSubagentTool(manager *SubagentManager) *SubagentTool

func (*SubagentTool) Description added in v0.1.2 

func (t *SubagentTool) Description() string

func (*SubagentTool) Execute added in v0.1.2 

func (t *SubagentTool) Execute(ctx context.Context, args map[string]interface{}) *ToolResult

func (*SubagentTool) Name added in v0.1.2 

func (t *SubagentTool) Name() string

func (*SubagentTool) Parameters added in v0.1.2 

func (t *SubagentTool) Parameters() map[string]interface{}

func (*SubagentTool) SetContext added in v0.1.2 

func (t *SubagentTool) SetContext(channel, chatID string)

type Tool 

type Tool interface {
	Name() string
	Description() string
	Parameters() map[string]interface{}
	Execute(ctx context.Context, args map[string]interface{}) *ToolResult
}

Tool is the interface that all tools must implement.

type ToolCall 

type ToolCall struct {
	ID        string                 `json:"id"`
	Type      string                 `json:"type"`
	Function  *FunctionCall          `json:"function,omitempty"`
	Name      string                 `json:"name,omitempty"`
	Arguments map[string]interface{} `json:"arguments,omitempty"`
}

type ToolDefinition 

type ToolDefinition struct {
	Type     string                 `json:"type"`
	Function ToolFunctionDefinition `json:"function"`
}

type ToolFunctionDefinition 

type ToolFunctionDefinition struct {
	Name        string                 `json:"name"`
	Description string                 `json:"description"`
	Parameters  map[string]interface{} `json:"parameters"`
}

type ToolLoopConfig added in v0.1.2 

type ToolLoopConfig struct {
	Provider      providers.LLMProvider
	Model         string
	Tools         *ToolRegistry
	MaxIterations int
	LLMOptions    map[string]any
}

ToolLoopConfig configures the tool execution loop.

type ToolLoopResult added in v0.1.2 

type ToolLoopResult struct {
	Content    string
	Iterations int
}

ToolLoopResult contains the result of running the tool loop.

func RunToolLoop added in v0.1.2 

func RunToolLoop(ctx context.Context, config ToolLoopConfig, messages []providers.Message, channel, chatID string) (*ToolLoopResult, error)

RunToolLoop executes the LLM + tool call iteration loop. This is the core agent logic that can be reused by both main agent and subagents.

type ToolRegistry 

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

func NewToolRegistry 

func NewToolRegistry() *ToolRegistry

func (*ToolRegistry) Count added in v0.1.1 

func (r *ToolRegistry) Count() int

Count returns the number of registered tools.

func (*ToolRegistry) Execute 

func (r *ToolRegistry) Execute(ctx context.Context, name string, args map[string]interface{}) *ToolResult

func (*ToolRegistry) ExecuteWithContext added in v0.1.1 

func (r *ToolRegistry) ExecuteWithContext(ctx context.Context, name string, args map[string]interface{}, channel, chatID string, asyncCallback AsyncCallback) *ToolResult

ExecuteWithContext executes a tool with channel/chatID context and optional async callback. If the tool implements AsyncTool and a non-nil callback is provided, the callback will be set on the tool before execution.

func (*ToolRegistry) Get 

func (r *ToolRegistry) Get(name string) (Tool, bool)

func (*ToolRegistry) GetDefinitions 

func (r *ToolRegistry) GetDefinitions() []map[string]interface{}

func (*ToolRegistry) GetSummaries added in v0.1.1 

func (r *ToolRegistry) GetSummaries() []string

GetSummaries returns human-readable summaries of all registered tools. Returns a slice of "name - description" strings.

func (*ToolRegistry) List added in v0.1.1 

func (r *ToolRegistry) List() []string

List returns a list of all registered tool names.

func (*ToolRegistry) Register 

func (r *ToolRegistry) Register(tool Tool)

func (*ToolRegistry) ToProviderDefs added in v0.1.2 

func (r *ToolRegistry) ToProviderDefs() []providers.ToolDefinition

ToProviderDefs converts tool definitions to provider-compatible format. This is the format expected by LLM provider APIs.

type ToolResult added in v0.1.2 

type ToolResult struct {
	// ForLLM is the content sent to the LLM for context.
	// Required for all results.
	ForLLM string `json:"for_llm"`

	// ForUser is the content sent directly to the user.
	// If empty, no user message is sent.
	// Silent=true overrides this field.
	ForUser string `json:"for_user,omitempty"`

	// Silent suppresses sending any message to the user.
	// When true, ForUser is ignored even if set.
	Silent bool `json:"silent"`

	// IsError indicates whether the tool execution failed.
	// When true, the result should be treated as an error.
	IsError bool `json:"is_error"`

	// Async indicates whether the tool is running asynchronously.
	// When true, the tool will complete later and notify via callback.
	Async bool `json:"async"`

	// Err is the underlying error (not JSON serialized).
	// Used for internal error handling and logging.
	Err error `json:"-"`
}

ToolResult represents the structured return value from tool execution. It provides clear semantics for different types of results and supports async operations, user-facing messages, and error handling.

func AsyncResult added in v0.1.2 

func AsyncResult(forLLM string) *ToolResult

AsyncResult creates a ToolResult for async operations. The task will run in the background and complete later.

Use this for long-running operations like: - Subagent spawns - Background processing - External API calls with callbacks

Example: 

result := AsyncResult("Subagent spawned, will report back")

func ErrorResult added in v0.1.2 

func ErrorResult(message string) *ToolResult

ErrorResult creates a ToolResult representing an error. Sets IsError=true and includes the error message.

Example: 

result := ErrorResult("Failed to connect to database: connection refused")

func NewToolResult added in v0.1.2 

func NewToolResult(forLLM string) *ToolResult

NewToolResult creates a basic ToolResult with content for the LLM. Use this when you need a simple result with default behavior.

Example: 

result := NewToolResult("File updated successfully")

func SilentResult added in v0.1.2 

func SilentResult(forLLM string) *ToolResult

SilentResult creates a ToolResult that is silent (no user message). The content is only sent to the LLM for context.

Use this for operations that should not spam the user, such as: - File reads/writes - Status updates - Background operations

Example: 

result := SilentResult("Config file saved")

func UserResult added in v0.1.2 

func UserResult(content string) *ToolResult

UserResult creates a ToolResult with content for both LLM and user. Both ForLLM and ForUser are set to the same content.

Use this when the user needs to see the result directly: - Command execution output - Fetched web content - Query results

Example: 

result := UserResult("Total files found: 42")

func (*ToolResult) MarshalJSON added in v0.1.2 

func (tr *ToolResult) MarshalJSON() ([]byte, error)

MarshalJSON implements custom JSON serialization. The Err field is excluded from JSON output via the json:"-" tag.

func (*ToolResult) WithError added in v0.1.2 

func (tr *ToolResult) WithError(err error) *ToolResult

WithError sets the Err field and returns the result for chaining. This preserves the error for logging while keeping it out of JSON.

Example: 

result := ErrorResult("Operation failed").WithError(err)

type UsageInfo 

type UsageInfo struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}

type WebFetchTool 

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

func NewWebFetchTool 

func NewWebFetchTool(maxChars int) *WebFetchTool

func (*WebFetchTool) Description 

func (t *WebFetchTool) Description() string

func (*WebFetchTool) Execute 

func (t *WebFetchTool) Execute(ctx context.Context, args map[string]interface{}) *ToolResult

func (*WebFetchTool) Name 

func (t *WebFetchTool) Name() string

func (*WebFetchTool) Parameters 

func (t *WebFetchTool) Parameters() map[string]interface{}

type WebSearchTool 

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

func NewWebSearchTool 

func NewWebSearchTool(opts WebSearchToolOptions) *WebSearchTool

func (*WebSearchTool) Description 

func (t *WebSearchTool) Description() string

func (*WebSearchTool) Execute 

func (t *WebSearchTool) Execute(ctx context.Context, args map[string]interface{}) *ToolResult

func (*WebSearchTool) Name 

func (t *WebSearchTool) Name() string

func (*WebSearchTool) Parameters 

func (t *WebSearchTool) Parameters() map[string]interface{}

type WebSearchToolOptions added in v0.1.2 

type WebSearchToolOptions struct {
	BraveAPIKey          string
	BraveMaxResults      int
	BraveEnabled         bool
	DuckDuckGoMaxResults int
	DuckDuckGoEnabled    bool
}

type WriteFileTool 

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

func NewWriteFileTool added in v0.1.1 

func NewWriteFileTool(workspace string, restrict bool) *WriteFileTool

func (*WriteFileTool) Description 

func (t *WriteFileTool) Description() string

func (*WriteFileTool) Execute 

func (t *WriteFileTool) Execute(ctx context.Context, args map[string]interface{}) *ToolResult

func (*WriteFileTool) Name 

func (t *WriteFileTool) Name() string

func (*WriteFileTool) Parameters 

func (t *WriteFileTool) Parameters() map[string]interface{}

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.