tools

Documentation

Index

• Variables
• func Invoke(call Call) string
• func NewValidationError(fieldsMissing []string) error
• type AiTool
• type Call
  • func (c Call) Json() string
• type CatTool
  • func (c CatTool) Call(input Input) (string, error)
  • func (c CatTool) UserFunction() UserFunction
• type FileTreeTool
  • func (f FileTreeTool) Call(input Input) (string, error)
  • func (f FileTreeTool) UserFunction() UserFunction
• type FileTypeTool
  • func (f FileTypeTool) Call(input Input) (string, error)
  • func (f FileTypeTool) UserFunction() UserFunction
• type FindTool
  • func (f FindTool) Call(input Input) (string, error)
  • func (f FindTool) UserFunction() UserFunction
• type GoTool
  • func (g GoTool) Call(input Input) (string, error)
  • func (g GoTool) UserFunction() UserFunction
• type Input
• type InputSchema
• type LsTool
  • func (f LsTool) Call(input Input) (string, error)
  • func (f LsTool) UserFunction() UserFunction
• type ParameterObject
• type RipGrepTool
  • func (r RipGrepTool) Call(input Input) (string, error)
  • func (r RipGrepTool) UserFunction() UserFunction
• type UserFunction
  • func UserFunctionFromName(name string) UserFunction
• type ValidationError
  • func (v ValidationError) Error() string
• type WebsiteTextTool
  • func (w WebsiteTextTool) Call(input Input) (string, error)
  • func (w WebsiteTextTool) UserFunction() UserFunction
• type WriteFileTool
  • func (w WriteFileTool) Call(input Input) (string, error)
  • func (w WriteFileTool) UserFunction() UserFunction

Variables

var Cat = CatTool{
	Name:        "cat",
	Description: "Display the contents of a file. Uses the linux command 'cat'.",
	Inputs: &InputSchema{
		Type: "object",
		Properties: map[string]ParameterObject{
			"file": {
				Type:        "string",
				Description: "The file to display the contents of.",
			},
			"number": {
				Type:        "boolean",
				Description: "Number all output lines.",
			},
			"showEnds": {
				Type:        "boolean",
				Description: "Display $ at end of each line.",
			},
			"squeezeBlank": {
				Type:        "boolean",
				Description: "Suppress repeated empty output lines.",
			},
		},
		Required: []string{"file"},
	},
}

var FileTree = FileTreeTool{
	Name:        "file_tree",
	Description: "List the filetree of some directory. Uses linux command 'tree'.",
	Inputs: &InputSchema{
		Type: "object",
		Properties: map[string]ParameterObject{
			"directory": {
				Type:        "string",
				Description: "The directory to list the filetree of.",
			},
			"level": {
				Type:        "integer",
				Description: "The depth of the tree to display.",
			},
		},
		Required: []string{"directory"},
	},
}

var FileType = FileTypeTool{
	Name:        "file_type",
	Description: "Determine the file type of a given file. Uses the linux command 'file'.",
	Inputs: &InputSchema{
		Type: "object",
		Properties: map[string]ParameterObject{
			"file_path": {
				Type:        "string",
				Description: "The path to the file to analyze.",
			},
			"mime_type": {
				Type:        "boolean",
				Description: "Whether to display the MIME type of the file.",
			},
		},
		Required: []string{"file_path"},
	},
}

var Find = FindTool{
	Name:        "find",
	Description: "Search for files in a directory hierarchy. Uses linux command 'find'.",
	Inputs: &InputSchema{
		Type: "object",
		Properties: map[string]ParameterObject{
			"directory": {
				Type:        "string",
				Description: "The directory to start the search from.",
			},
			"name": {
				Type:        "string",
				Description: "The name pattern to search for.",
			},
			"type": {
				Type:        "string",
				Description: "The file type to search for (f: regular file, d: directory).",
			},
			"maxdepth": {
				Type:        "integer",
				Description: "The maximum depth of directories to search.",
			},
		},
		Required: []string{"directory"},
	},
}

var Go = GoTool{
	Name:        "go",
	Description: "Run Go commands like 'go test' and 'go run' to compile, test, and run Go programs. Run 'go help' to get details of this tool.",
	Inputs: &InputSchema{
		Type: "object",
		Properties: map[string]ParameterObject{
			"command": {
				Type:        "string",
				Description: "The Go command to run (e.g., 'run', 'test', 'build').",
			},
			"args": {
				Type:        "string",
				Description: "Additional arguments for the Go command (e.g., file names, flags).",
			},
			"dir": {
				Type:        "string",
				Description: "The directory to run the command in (optional, defaults to current directory).",
			},
		},
		Required: []string{"command"},
	},
}

var LS = LsTool{
	Name:        "ls",
	Description: "List the files in a directory. Uses the Linux command 'ls'.",
	Inputs: &InputSchema{
		Type: "object",
		Properties: map[string]ParameterObject{
			"directory": {
				Type:        "string",
				Description: "The directory to list the files of.",
			},
			"all": {
				Type:        "boolean",
				Description: "Show all files, including hidden files.",
			},
			"long": {
				Type:        "boolean",
				Description: "Use a long listing format.",
			},
		},
		Required: []string{"directory"},
	},
}

var RipGrep = RipGrepTool{
	Name:        "rg",
	Description: "Search for a pattern in files using ripgrep.",
	Inputs: &InputSchema{
		Type: "object",
		Properties: map[string]ParameterObject{
			"pattern": {
				Type:        "string",
				Description: "The pattern to search for.",
			},
			"path": {
				Type:        "string",
				Description: "The path to search in.",
			},
			"case_sensitive": {
				Type:        "boolean",
				Description: "Whether the search should be case sensitive.",
			},
			"line_number": {
				Type:        "boolean",
				Description: "Whether to show line numbers.",
			},
			"hidden": {
				Type:        "boolean",
				Description: "Whether to search hidden files and directories.",
			},
		},
		Required: []string{"pattern"},
	},
}

var Tools = map[string]AiTool{
	"file_tree":    FileTree,
	"cat":          Cat,
	"find":         Find,
	"file_type":    FileType,
	"ls":           LS,
	"website_text": WebsiteText,
	"rg":           RipGrep,
	"go":           Go,
	"write_file":   WriteFile,
}

var WebsiteText = WebsiteTextTool{
	Name:        "website_text",
	Description: "Get the text content of a website by stripping all non-text tags and trimming whitespace.",
	Inputs: &InputSchema{
		Type: "object",
		Properties: map[string]ParameterObject{
			"url": {
				Type:        "string",
				Description: "The URL of the website to retrieve the text content from.",
			},
		},
		Required: []string{"url"},
	},
}

var WriteFile = WriteFileTool{
	Name:        "write_file",
	Description: "Write content to a file. Creates the file if it doesn't exist, or overwrites it if it does.",
	Inputs: &InputSchema{
		Type: "object",
		Properties: map[string]ParameterObject{
			"file_path": {
				Type:        "string",
				Description: "The path to the file to write to.",
			},
			"content": {
				Type:        "string",
				Description: "The content to write to the file.",
			},
			"append": {
				Type:        "boolean",
				Description: "If true, append to the file instead of overwriting it.",
			},
		},
		Required: []string{"file_path", "content"},
	},
}

Functions

func Invoke

func Invoke(call Call) string

Invoke the call, and gather both error and output in the same string

func NewValidationError

func NewValidationError(fieldsMissing []string) error

Types

type AiTool

type AiTool interface {
	// Call the AI tool with the given Input. Returns output from the tool or an error
	// if the call returned an error-like. An error-like is either exit code non-zero or
	// restful response non 2xx.
	Call(Input) (string, error)

	// Return the UserFunction, later on used
	// by text queriers to send to their respective
	// models
	UserFunction() UserFunction
}

type Call

type Call struct {
	ID       string       `json:"id,omitempty"`
	Name     string       `json:"name,omitempty"`
	Type     string       `json:"type,omitempty"`
	Inputs   Input        `json:"inputs,omitempty"`
	Function UserFunction `json:"function,omitempty"`
}

func (Call) Json

func (c Call) Json() string

type CatTool

type CatTool UserFunction

func (CatTool) Call

func (c CatTool) Call(input Input) (string, error)

func (CatTool) UserFunction

func (c CatTool) UserFunction() UserFunction

type FileTreeTool

type FileTreeTool UserFunction

func (FileTreeTool) Call

func (f FileTreeTool) Call(input Input) (string, error)

func (FileTreeTool) UserFunction

func (f FileTreeTool) UserFunction() UserFunction

type FileTypeTool

type FileTypeTool UserFunction

func (FileTypeTool) Call

func (f FileTypeTool) Call(input Input) (string, error)

func (FileTypeTool) UserFunction

func (f FileTypeTool) UserFunction() UserFunction

type FindTool

type FindTool UserFunction

func (FindTool) Call

func (f FindTool) Call(input Input) (string, error)

func (FindTool) UserFunction

func (f FindTool) UserFunction() UserFunction

type GoTool

type GoTool UserFunction

func (GoTool) Call

func (g GoTool) Call(input Input) (string, error)

func (GoTool) UserFunction

func (g GoTool) UserFunction() UserFunction

type Input

type Input map[string]any

type InputSchema

type InputSchema struct {
	Type       string                     `json:"type"`
	Required   []string                   `json:"required"`
	Properties map[string]ParameterObject `json:"properties"`
}

type LsTool

type LsTool UserFunction

func (LsTool) Call

func (f LsTool) Call(input Input) (string, error)

func (LsTool) UserFunction

func (f LsTool) UserFunction() UserFunction

type ParameterObject

type ParameterObject struct {
	Type        string   `json:"type"`
	Description string   `json:"description"`
	Enum        []string `json:"enum,omitempty"`
}

type RipGrepTool

type RipGrepTool UserFunction

func (RipGrepTool) Call

func (r RipGrepTool) Call(input Input) (string, error)

func (RipGrepTool) UserFunction

func (r RipGrepTool) UserFunction() UserFunction

type UserFunction

type UserFunction struct {
	Name        string `json:"name"`
	Description string `json:"description,omitempty"`
	// Format is the same, but name of the field different. So this way, each
	// vendor can set their own field name
	Inputs *InputSchema `json:"input_schema,omitempty"`
	// Chatgpt wants this
	Arguments string `json:"arguments,omitempty"`
}

func UserFunctionFromName

func UserFunctionFromName(name string) UserFunction

type ValidationError

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

func (ValidationError) Error

func (v ValidationError) Error() string

type WebsiteTextTool

type WebsiteTextTool UserFunction

func (WebsiteTextTool) Call

func (w WebsiteTextTool) Call(input Input) (string, error)

func (WebsiteTextTool) UserFunction

func (w WebsiteTextTool) UserFunction() UserFunction

type WriteFileTool

type WriteFileTool UserFunction

func (WriteFileTool) Call

func (w WriteFileTool) Call(input Input) (string, error)

func (WriteFileTool) UserFunction

func (w WriteFileTool) UserFunction() UserFunction