Give each agent a config and user instructions

[onechat] / prompt.go

diff --git a/prompt.go b/prompt.go
index dc0fb06dcf8138cd3161574b1a92e2757db31d9c..e48c5a61b714350b61dc3bcca68b025c40b05383 100644 (file)
--- a/prompt.go
+++ b/prompt.go
@@ -5,15 +5,30 @@ import (
        "strings"
 )
 
-// promptName stands for the name of the agent in promptText.
-const promptName = "{name}"
+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."
+
+       // 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"
+
+       // instructionsHeading starts the instructions of the user after
+       // promptText.
+       instructionsHeading = "\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,
-// background tasks, and the instructions of the user. It holds no date and no
-// state, so turns and compactions share its prompt cache.
+// 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
+// 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
 ends. Each call to you is a turn or a compaction: the view below is followed by
 the user's new message, or by a task starting "Compaction:".
@@ -92,11 +107,20 @@ exactly. Tag each item with its kind ("user: ...; echo: ..."), and credit quoted
 text to its real author. Never make anything look further along than it was. If
 told the line is too long, shorten it. Non-ASCII characters cost 2-4 bytes.`
 
-// prompt returns the system prompt of the agent name. Turns and compactions
-// send the same prompt and tools, so compactions read the prompt cache of the
-// turns.
-func prompt(name string) string {
-       return strings.ReplaceAll(promptText, promptName, name)
+// prompt returns the system prompt of the agent name with the instructions of
+// the user at its end. Without instructions it has neither them nor the clause
+// that points to them. Turns and compactions send the same prompt and tools, so
+// compactions read the prompt cache of the turns.
+func prompt(name, instructions string) string {
+       text := strings.ReplaceAll(promptText, promptName, name)
+       instructions = strings.TrimSpace(instructions)
+       if instructions == "" {
+               return text
+       }
+
+       text = strings.Replace(text, taskSentence, strings.TrimSuffix(taskSentence, ".")+instructionsClause+".", 1)
+
+       return text + instructionsHeading + instructions
 }
 
 // agentTools returns the function tools of every call: zoom and date. A