Allow internal prompt to be skipped

Author: ibuildthecloud

pkg/engine/engine.go

@@ -180,7 +180,7 @@ func (e *Engine) Start(ctx Context, input string) (*Return, error) {
 		Temperature:  tool.Temperature,
 	}
 
-	if InternalSystemPrompt != "" {
+	if InternalSystemPrompt != "" && (tool.InternalPrompt == nil || *tool.InternalPrompt) {
 		completion.Messages = append(completion.Messages, types.CompletionMessage{
 			Role:    types.CompletionMessageRoleTypeSystem,
 			Content: types.Text(InternalSystemPrompt),

pkg/monitor/display.go

@@ -68,7 +68,7 @@ func (l *livePrinter) end() {
 		return
 	}
 	if l.needsNewline {
-		fmt.Println()
+		_, _ = fmt.Fprintln(os.Stderr)
 	}
 	l.needsNewline = false
 }
@@ -83,7 +83,7 @@ func (l *livePrinter) print(event runner.Event, c call) {
 
 	last := l.lastLines[c.ID]
 	line := strings.TrimPrefix(event.Content, last)
-	fmt.Print(line)
+	_, _ = fmt.Fprint(os.Stderr, line)
 	l.needsNewline = !strings.HasSuffix(line, "\n")
 	l.lastLines[c.ID] = event.Content
 }

pkg/parser/parser.go

@@ -79,6 +79,12 @@ func isParam(line string, tool *types.Tool) (_ bool, err error) {
 		tool.ModelName = value
 	case "description":
 		tool.Description = value
+	case "internalprompt":
+		v, err := toBool(value)
+		if err != nil {
+			return false, err
+		}
+		tool.InternalPrompt = &v
 	case "tools":
 		tool.Tools = append(tool.Tools, csv(strings.ToLower(value))...)
 	case "args":

pkg/types/tool.go

@@ -23,15 +23,16 @@ type Program struct {
 type BuiltinFunc func(ctx context.Context, env []string, input string) (string, error)
 
 type Tool struct {
-	ID           string            `json:"id,omitempty"`
-	Name         string            `json:"name,omitempty"`
-	Description  string            `json:"description,omitempty"`
-	Arguments    *JSONSchema       `json:"arguments,omitempty"`
-	Instructions string            `json:"instructions,omitempty"`
-	Tools        []string          `json:"tools,omitempty"`
-	ToolMapping  map[string]string `json:"toolMapping,omitempty"`
-	LocalTools   map[string]string `json:"localTools,omitempty"`
-	BuiltinFunc  BuiltinFunc       `json:"-"`
+	ID             string            `json:"id,omitempty"`
+	Name           string            `json:"name,omitempty"`
+	Description    string            `json:"description,omitempty"`
+	InternalPrompt *bool             `json:"internalPrompt"`
+	Arguments      *JSONSchema       `json:"arguments,omitempty"`
+	Instructions   string            `json:"instructions,omitempty"`
+	Tools          []string          `json:"tools,omitempty"`
+	ToolMapping    map[string]string `json:"toolMapping,omitempty"`
+	LocalTools     map[string]string `json:"localTools,omitempty"`
+	BuiltinFunc    BuiltinFunc       `json:"-"`
 
 	Vision       bool     `json:"vision,omitempty"`
 	MaxTokens    int      `json:"maxTokens,omitempty"`