Merge pull request #88 from renxiawang/master

Add a default system prompt env var

Author: pwh-pwh

chat/chat.go

@@ -3,12 +3,13 @@ package chat
 import (
 	_ "errors"
 	"fmt"
-	"github.com/pwh-pwh/aiwechat-vercel/client"
 	"os"
 	"strconv"
 	"strings"
 	"time"
 
+	"github.com/pwh-pwh/aiwechat-vercel/client"
+
 	"github.com/google/generative-ai-go/genai"
 
 	"github.com/pwh-pwh/aiwechat-vercel/db"
@@ -301,7 +302,11 @@ func GetMsgListWithDb[T ChatMsg](botType, userId string, msg T, f func(msg T) db
 	isSupportPrompt := config.IsSupportPrompt(botType)
 	if isSupportPrompt {
 		prompt, err := db.GetPrompt(userId, botType)
-		if err == nil && prompt != "" {
+		if err != nil || prompt == "" {
+			prompt = config.GetDefaultSystemPrompt()
+		}
+
+		if prompt != "" {
 			dbList = append(dbList, db.Msg{
 				Role: "system",
 				Msg:  prompt,

conf/.env.sample

@@ -1,5 +1,6 @@
 # common config
 botType=** 机器人类型 目前支持(gpt,spark,echo,qwen,gemini)例如botType=gpt
+defaultSystemPrompt=你是AI机器人。你会为用户提供安全，有帮助，准确的回答。
 
 # wx config
 WX_TOKEN=*** 微信公众号开发平台设置的token

config/config.go

@@ -5,8 +5,8 @@ import (
 	"fmt"
 	"os"
 	"slices"
-	"sync"
 	"strconv"
+	"sync"
 
 	"github.com/pwh-pwh/aiwechat-vercel/db"
 )
@@ -129,4 +129,8 @@ func GetMaxTokens() int {
 		return 0
 	}
 	return maxTokens
-}
+}
+
+func GetDefaultSystemPrompt() string {
+	return os.Getenv("defaultSystemPrompt")
+}