Add customizable system prompt

agent/agent.go

@@ -207,6 +207,16 @@ func (a *Agent) consumeJob(job *Job, role string) {
 		})
 	}
 
+	if a.options.systemPrompt != "" {
+		if !Messages(a.currentConversation).Exist(a.options.systemPrompt) {
+			a.currentConversation = append([]openai.ChatCompletionMessage{
+				{
+					Role:    "system",
+					Content: a.options.systemPrompt,
+				}}, a.currentConversation...)
+		}
+	}
+
 	// RAG
 	if memory {
 		// Walk conversation from bottom to top, and find the first message of the user

agent/options.go

@@ -30,6 +30,8 @@ type options struct {
 	kbResults                                         int
 	ragdb                                             RAGDB
 
+	systemPrompt string
+
 	// callbacks
 	reasoningCallback func(ActionCurrentState) bool
 	resultCallback    func(ActionState)
@@ -110,6 +112,13 @@ func WithRAGDB(db RAGDB) Option {
 	}
 }
 
+func WithSystemPrompt(prompt string) Option {
+	return func(o *options) error {
+		o.systemPrompt = prompt
+		return nil
+	}
+}
+
 func WithLLMAPIURL(url string) Option {
 	return func(o *options) error {
 		o.LLMAPI.APIURL = url