X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/5da3ea038d1e9f9a2634d5ab137aef6989a3a789..eb05edcf5d575e8a3b1040debfb57423312e8320:/prompt.go diff --git a/prompt.go b/prompt.go index 373aab6..c6df057 100644 --- a/prompt.go +++ b/prompt.go @@ -9,17 +9,17 @@ 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 @@ -126,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, date, shell, +// 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 agentTools() []Tool { - return []Tool{ +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, }, @@ -150,8 +150,8 @@ 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, }, @@ -159,8 +159,8 @@ func agentTools() []Tool { 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."), + Parameters: promptObjectSchema(map[string]any{ + "command": promptStringSchema("The bash command."), }), Strict: true, }, @@ -168,8 +168,8 @@ func agentTools() []Tool { 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."), + Parameters: promptObjectSchema(map[string]any{ + "path": promptStringSchema("The file, absolute or relative to the working directory."), }), Strict: true, }, @@ -177,9 +177,9 @@ func agentTools() []Tool { 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."), + Parameters: promptObjectSchema(map[string]any{ + "path": promptStringSchema("The file, absolute or relative to the working directory."), + "text": promptStringSchema("The whole new content."), }), Strict: true, }, @@ -187,19 +187,19 @@ func agentTools() []Tool { 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."), + 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) @@ -215,16 +215,16 @@ 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, } } -// stringSchema returns the JSON schema of a string. -func stringSchema(description string) map[string]any { +// promptStringSchema returns the JSON schema of a string. +func promptStringSchema(description string) map[string]any { return map[string]any{ "type": "string", "description": description,