Feat: tool write_append

tools.go

@@ -71,8 +71,13 @@ Your current tools:
 },
 {
 "name":"file_write",
-"args": ["path", "content", "mode"],
-"when_to_use": "when asked to write content to a file; mode is optional (overwrite or append, default: overwrite)"
+"args": ["path", "content"],
+"when_to_use": "when needed to overwrite content to a file"
+},
+{
+"name":"file_write_append",
+"args": ["path", "content"],
+"when_to_use": "when asked to append content to a file; use sed to edit content"
 },
 {
 "name":"file_delete",
@@ -359,36 +364,35 @@ func fileWrite(args map[string]string) []byte {
 		logger.Error(msg)
 		return []byte(msg)
 	}
-
 	content, ok := args["content"]
 	if !ok {
 		content = ""
 	}
-
-	mode, ok := args["mode"]
-	if !ok || mode == "" {
-		mode = "overwrite"
-	}
-
-	switch mode {
-	case "overwrite":
 	if err := writeStringToFile(path, content); err != nil {
 		msg := "failed to write to file; error: " + err.Error()
 		logger.Error(msg)
 		return []byte(msg)
 	}
-	case "append":
+	msg := "file written successfully at " + path
+	return []byte(msg)
+}
+
+func fileWriteAppend(args map[string]string) []byte {
+	path, ok := args["path"]
+	if !ok || path == "" {
+		msg := "path not provided to file_write_append tool"
+		logger.Error(msg)
+		return []byte(msg)
+	}
+	content, ok := args["content"]
+	if !ok {
+		content = ""
+	}
 	if err := appendStringToFile(path, content); err != nil {
 		msg := "failed to append to file; error: " + err.Error()
 		logger.Error(msg)
 		return []byte(msg)
 	}
-	default:
-		msg := "invalid mode; use 'overwrite' or 'append'"
-		logger.Error(msg)
-		return []byte(msg)
-	}
-
 	msg := "file written successfully at " + path
 	return []byte(msg)
 }
@@ -887,6 +891,7 @@ var fnMap = map[string]fnSig{
 	"file_create":       fileCreate,
 	"file_read":         fileRead,
 	"file_write":        fileWrite,
+	"file_write_append": fileWriteAppend,
 	"file_delete":       fileDelete,
 	"file_move":         fileMove,
 	"file_copy":         fileCopy,
@@ -1056,7 +1061,7 @@ var baseTools = []models.Tool{
 		Type: "function",
 		Function: models.ToolFunc{
 			Name:        "file_write",
-			Description: "Write content to a file. Use when you want to create or modify a file (overwrite or append).",
+			Description: "Write content to a file. Will overwrite any content present.",
 			Parameters: models.ToolFuncParams{
 				Type:     "object",
 				Required: []string{"path", "content"},
@@ -1069,9 +1074,28 @@ var baseTools = []models.Tool{
 						Type:        "string",
 						Description: "content to write to the file",
 					},
-					"mode": models.ToolArgProps{
+				},
+			},
+		},
+	},
+
+	// file_write_append
+	models.Tool{
+		Type: "function",
+		Function: models.ToolFunc{
+			Name:        "file_write_append",
+			Description: "Append content to a file.",
+			Parameters: models.ToolFuncParams{
+				Type:     "object",
+				Required: []string{"path", "content"},
+				Properties: map[string]models.ToolArgProps{
+					"path": models.ToolArgProps{
 						Type:        "string",
-						Description: "write mode: 'overwrite' to replace entire file content, 'append' to add to the end (defaults to 'overwrite')",
+						Description: "path of the file to write to",
+					},
+					"content": models.ToolArgProps{
+						Type:        "string",
+						Description: "content to write to the file",
 					},
 				},
 			},

tui.go

@@ -1259,6 +1259,7 @@ func init() {
 		// cannot send msg in editMode or botRespMode
 		if event.Key() == tcell.KeyEscape && !editMode && !botRespMode {
 			msgText := textArea.GetText()
+			// TODO: add shellmode command -> output to the chat history, or at least have an option
 			if shellMode && msgText != "" {
 				// In shell mode, execute command instead of sending to LLM
 				executeCommandAndDisplay(msgText)