Merge pull request #635 from nkwangleiGIT/main

fix potential issue in the context of prompt

pkg/appruntime/chain/llmchain.go

@@ -89,7 +89,7 @@ func (l *LLMChain) Run(ctx context.Context, cli dynamic.Interface, args map[stri
 	// Add the answer to the context if it's not empty
 	if args["_answer"] != nil {
 		klog.Infoln("get answer from upstream:", args["_answer"])
-		args["context"] = args["_answer"]
+		args["context"] = fmt.Sprintf("%s\n%s", args["context"], args["_answer"])
 	}
 	args = runTools(ctx, args, instance.Spec.Tools)
 	chain := chains.NewLLMChain(llm, prompt)

pkg/appruntime/prompt/prompt.go

@@ -19,6 +19,7 @@ package prompt
 import (
 	"context"
 	"fmt"
+	"strings"
 
 	"github.com/tmc/langchaingo/prompts"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
@@ -30,6 +31,10 @@ import (
 	"github.com/kubeagi/arcadia/pkg/appruntime/base"
 )
 
+const (
+	promptContextPlaceholder = "{{.context}}"
+)
+
 type Prompt struct {
 	base.BaseNode
 	prompts.ChatPromptTemplate
@@ -58,9 +63,11 @@ func (p *Prompt) Run(ctx context.Context, cli dynamic.Interface, args map[string
 		ps = append(ps, prompts.NewSystemMessagePromptTemplate(instance.Spec.SystemMessage, []string{}))
 	}
 	if instance.Spec.UserMessage != "" {
-		// Add the context by default, and leave it empty
-		// so we can add more contexts as needed in all agents/chains
-		instance.Spec.UserMessage = fmt.Sprintf("{{.context}}\n%s", instance.Spec.UserMessage)
+		if !strings.Contains(instance.Spec.UserMessage, promptContextPlaceholder) {
+			// Add the context by default if it does not exist, and leave it empty
+			// so we can add more contexts as needed in all agents/chains
+			instance.Spec.UserMessage = fmt.Sprintf("%s\n%s", promptContextPlaceholder, instance.Spec.UserMessage)
+		}
 		ps = append(ps, prompts.NewHumanMessagePromptTemplate(instance.Spec.UserMessage, []string{"question"}))
 	}
 	template := prompts.NewChatPromptTemplate(ps)