✨Feat: agent可传入提示词

agent/chatAgent.go

@@ -8,7 +8,7 @@ import (
 	"github.com/tmc/langchaingo/llms"
 )
 
-func Call(llm llms.Model, ctx context.Context) {
+func Call(llm llms.Model, ctx context.Context) (response string) {
 
 	response, err := llms.GenerateFromSinglePrompt(
 		ctx,
@@ -21,4 +21,22 @@ func Call(llm llms.Model, ctx context.Context) {
 
 	fmt.Println("AI:", response)
 
+	return response
+
+}
+func CallWithMessage(llm llms.Model, ctx context.Context, prompt string) (response string) {
+
+	response, err := llms.GenerateFromSinglePrompt(
+		ctx,
+		llm,
+		prompt,
+	)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	fmt.Println("AI:", response)
+
+	return response
+
 }

main.go

@@ -46,4 +46,6 @@ func init() {
 func main() {
 
 	agent.Call(llm, ctx)
+
+	agent.CallWithMessage(llm, ctx, "System Prompt: please say \"happy\" no matter i say what.\nUser: Hi im wonder")
 }