72 lines
1.7 KiB
Go
72 lines
1.7 KiB
Go
package tools
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"git.mlow.ca/mlow/lmcli/pkg/lmcli/model"
|
|
toolutil "git.mlow.ca/mlow/lmcli/pkg/lmcli/tools/util"
|
|
)
|
|
|
|
const READ_FILE_DESCRIPTION = `Read the contents of a text file relative to the current working directory.
|
|
|
|
Each line of the returned content is prefixed with its line number and a tab (\t).
|
|
|
|
Example result:
|
|
{
|
|
"message": "success",
|
|
"result": "1\tthe contents\n2\tof the file\n"
|
|
}`
|
|
|
|
var ReadFileTool = model.Tool{
|
|
Name: "read_file",
|
|
Description: READ_FILE_DESCRIPTION,
|
|
Parameters: []model.ToolParameter{
|
|
{
|
|
Name: "path",
|
|
Type: "string",
|
|
Description: "Path to a file within the current working directory to read.",
|
|
Required: true,
|
|
},
|
|
},
|
|
|
|
Impl: func(tool *model.Tool, args map[string]interface{}) (string, error) {
|
|
tmp, ok := args["path"]
|
|
if !ok {
|
|
return "", fmt.Errorf("Path parameter to read_file was not included.")
|
|
}
|
|
path, ok := tmp.(string)
|
|
if !ok {
|
|
return "", fmt.Errorf("Invalid path in function arguments: %v", tmp)
|
|
}
|
|
result := readFile(path)
|
|
ret, err := result.ToJson()
|
|
if err != nil {
|
|
return "", fmt.Errorf("Could not serialize result: %v", err)
|
|
}
|
|
return ret, nil
|
|
},
|
|
}
|
|
|
|
func readFile(path string) model.CallResult {
|
|
ok, reason := toolutil.IsPathWithinCWD(path)
|
|
if !ok {
|
|
return model.CallResult{Message: reason}
|
|
}
|
|
data, err := os.ReadFile(path)
|
|
if err != nil {
|
|
return model.CallResult{Message: fmt.Sprintf("Could not read path: %s", err.Error())}
|
|
}
|
|
|
|
lines := strings.Split(string(data), "\n")
|
|
content := strings.Builder{}
|
|
for i, line := range lines {
|
|
content.WriteString(fmt.Sprintf("%d\t%s\n", i+1, line))
|
|
}
|
|
|
|
return model.CallResult{
|
|
Result: content.String(),
|
|
}
|
|
}
|