X-Git-Url: http://git.giacomolli.uk/onechat/blobdiff_plain/4a605fa39078354654ae648e489c117c7a445197..97274864799ac2eea84bf80c9727a810b695db55:/prompt.go

diff --git a/prompt.go b/prompt.go
index e48c5a6..373aab6 100644
--- a/prompt.go
+++ b/prompt.go
@@ -26,7 +26,8 @@ const (
 // word. The reference design names its own agent, so promptText has the name of
 // the agent and the kind reply in its place. It describes only what OneChat can
 // do, so it leaves out the spec text on images, subagents, computers, and
-// background tasks. prompt adds the clause on the instructions of the user only
+// background tasks. Its paragraph on persistent mode tells the model that a
+// final answer ends the turn. prompt adds the clause on the instructions of the user only
 // when there are instructions. It holds no date and no state, so turns and
 // compactions share its prompt cache.
 const promptText = `You are {name}, an AI agent that works for one user in a single chat that never
@@ -72,6 +73,13 @@ say in your reply what you learned that will matter later.
 
 Messages the user sends while you work reach you between tool calls.
 
+You are in persistent mode: a reply with no tool call is your final answer, and
+it ends your turn. You act again only when the user writes. So keep going until
+the user's task is done, or until only the user can unblock it. Answer a
+question asked in the middle of a task, then go on with the task in the same
+turn. When the user says "continue", resume the latest unfinished task from the
+view.
+
 # Compactions
 
 You write {name}'s memory: one step of the tree, compressing one message into a
@@ -123,9 +131,9 @@ func prompt(name, instructions string) string {
 	return text + instructionsHeading + instructions
 }
 
-// agentTools returns the function tools of every call: zoom and date. A
-// compaction sends them too and never calls them, because a change of the
-// tools ends the shared prompt cache.
+// agentTools returns the function tools of every call: zoom, date, shell,
+// read, write, and edit. A compaction sends them too and never calls them,
+// because a change of the tools ends the shared prompt cache.
 func agentTools() []Tool {
 	return []Tool{
 		{
@@ -147,6 +155,45 @@ func agentTools() []Tool {
 			}),
 			Strict: true,
 		},
+		{
+			Type:        "function",
+			Name:        "shell",
+			Description: "Run a command with bash, as the user, in the working directory, with no input. Gives its output and error output, and its exit status when it fails.",
+			Parameters: objectSchema(map[string]any{
+				"command": stringSchema("The bash command."),
+			}),
+			Strict: true,
+		},
+		{
+			Type:        "function",
+			Name:        "read",
+			Description: "Read a text file.",
+			Parameters: objectSchema(map[string]any{
+				"path": stringSchema("The file, absolute or relative to the working directory."),
+			}),
+			Strict: true,
+		},
+		{
+			Type:        "function",
+			Name:        "write",
+			Description: "Write a text file whole, replacing it when it exists. Makes the missing directories.",
+			Parameters: objectSchema(map[string]any{
+				"path": stringSchema("The file, absolute or relative to the working directory."),
+				"text": stringSchema("The whole new content."),
+			}),
+			Strict: true,
+		},
+		{
+			Type:        "function",
+			Name:        "edit",
+			Description: "Replace the one occurrence of old in a text file with new.",
+			Parameters: objectSchema(map[string]any{
+				"path": stringSchema("The file, absolute or relative to the working directory."),
+				"old":  stringSchema("The exact text to replace. It must occur once in the file."),
+				"new":  stringSchema("The text that replaces old."),
+			}),
+			Strict: true,
+		},
 	}
 }
 
@@ -175,3 +222,11 @@ func integerSchema(description string) map[string]any {
 		"description": description,
 	}
 }
+
+// stringSchema returns the JSON schema of a string.
+func stringSchema(description string) map[string]any {
+	return map[string]any{
+		"type":        "string",
+		"description": description,
+	}
+}