X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/4a605fa39078354654ae648e489c117c7a445197..6b21fcf78372995c6d8130c63f2f8aa2609bf7fe:/prompt.go?ds=sidebyside diff --git a/prompt.go b/prompt.go index e48c5a6..c6df057 100644 --- a/prompt.go +++ b/prompt.go @@ -9,24 +9,25 @@ const ( // promptName stands for the name of the agent in promptText. promptName = "{name}" - // taskSentence is the sentence of promptText that gets instructionsClause - // when the user has instructions. - taskSentence = "Do the user's tasks yourself, with your tools." + // promptTaskSentence is the sentence of promptText that gets + // promptInstructionsClause when the user has instructions. + promptTaskSentence = "Do the user's tasks yourself, with your tools." - // instructionsClause points to the instructions of the user at the end - // of the prompt. - instructionsClause = ", following the user's instructions\nat the end of this prompt: who they are, how their files are organized and how\nthey want work done" + // promptInstructionsClause points to the instructions of the user at + // the end of the prompt. + promptInstructionsClause = ", following the user's instructions\nat the end of this prompt: who they are, how their files are organized and how\nthey want work done" - // instructionsHeading starts the instructions of the user after + // promptInstructionsHeading starts the instructions of the user after // promptText. - instructionsHeading = "\n\n# The user's instructions\n\n" + promptInstructionsHeading = "\n\n# The user's instructions\n\n" ) // promptText is the system prompt of the reference design, nearly word for // 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 @@ -118,23 +126,23 @@ func prompt(name, instructions string) string { return text } - text = strings.Replace(text, taskSentence, strings.TrimSuffix(taskSentence, ".")+instructionsClause+".", 1) + text = strings.Replace(text, promptTaskSentence, strings.TrimSuffix(promptTaskSentence, ".")+promptInstructionsClause+".", 1) - return text + instructionsHeading + instructions + return text + promptInstructionsHeading + 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. -func agentTools() []Tool { - return []Tool{ +// promptTools 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 promptTools() []ModelTool { + return []ModelTool{ { Type: "function", Name: "zoom", Description: "Open the line id+n of the view into the two lines of n/2 under it; n = 1 gives the message whole.", - Parameters: objectSchema(map[string]any{ - "id": integerSchema("The first message of the line: a multiple of n."), - "n": integerSchema("The number of messages of the line: a power of 2."), + Parameters: promptObjectSchema(map[string]any{ + "id": promptIntegerSchema("The first message of the line: a multiple of n."), + "n": promptIntegerSchema("The number of messages of the line: a power of 2."), }), Strict: true, }, @@ -142,17 +150,56 @@ func agentTools() []Tool { Type: "function", Name: "date", Description: "The date and time of message id.", - Parameters: objectSchema(map[string]any{ - "id": integerSchema("The message."), + Parameters: promptObjectSchema(map[string]any{ + "id": promptIntegerSchema("The message."), + }), + 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: promptObjectSchema(map[string]any{ + "command": promptStringSchema("The bash command."), + }), + Strict: true, + }, + { + Type: "function", + Name: "read", + Description: "Read a text file.", + Parameters: promptObjectSchema(map[string]any{ + "path": promptStringSchema("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: promptObjectSchema(map[string]any{ + "path": promptStringSchema("The file, absolute or relative to the working directory."), + "text": promptStringSchema("The whole new content."), + }), + Strict: true, + }, + { + Type: "function", + Name: "edit", + Description: "Replace the one occurrence of old in a text file with new.", + Parameters: promptObjectSchema(map[string]any{ + "path": promptStringSchema("The file, absolute or relative to the working directory."), + "old": promptStringSchema("The exact text to replace. It must occur once in the file."), + "new": promptStringSchema("The text that replaces old."), }), Strict: true, }, } } -// objectSchema returns a strict JSON schema of an object whose properties are -// all required. -func objectSchema(properties map[string]any) map[string]any { +// promptObjectSchema returns a strict JSON schema of an object whose properties +// are all required. +func promptObjectSchema(properties map[string]any) map[string]any { required := make([]string, 0, len(properties)) for name := range properties { required = append(required, name) @@ -168,10 +215,18 @@ func objectSchema(properties map[string]any) map[string]any { } } -// integerSchema returns the JSON schema of an integer. -func integerSchema(description string) map[string]any { +// promptIntegerSchema returns the JSON schema of an integer. +func promptIntegerSchema(description string) map[string]any { return map[string]any{ "type": "integer", "description": description, } } + +// promptStringSchema returns the JSON schema of a string. +func promptStringSchema(description string) map[string]any { + return map[string]any{ + "type": "string", + "description": description, + } +}