feat: reward verify is ready

feat: 增加打赏核销功能
feat: 集成微信收款服务
2025-11-01 14:53:44 +08:00 · 2023-07-21 22:29:14 +08:00 · 2023-07-21 18:26:51 +08:00 · 2023-07-20 17:46:32 +08:00 · 2023-07-20 16:36:12 +08:00 · 2023-07-16 16:05:11 +08:00
32 changed files with 1070 additions and 306 deletions
--- a/api/.gitignore
+++ b/api/.gitignore
@@ -16,4 +16,5 @@ tmp
 bin
 data
 config.toml
-static/upload
+static/upload 
+storage.json
--- a/api/core/app_server.go
+++ b/api/core/app_server.go
@@ -2,6 +2,7 @@ package core

 import (
 	"chatplus/core/types"
+	"chatplus/service/function"
 	"chatplus/store/model"
 	"chatplus/utils"
 	"chatplus/utils/resp"
@@ -22,27 +23,37 @@ type AppServer struct {
 	Debug        bool
 	Config       *types.AppConfig
 	Engine       *gin.Engine
-	ChatContexts *types.LMap[string, []types.Message] // 聊天上下文 Map [chatId] => []Message
-	ChatConfig   *types.ChatConfig                    // 聊天配置
+	ChatContexts *types.LMap[string, []interface{}] // 聊天上下文 Map [chatId] => []Message
+	ChatConfig   *types.ChatConfig                  // 聊天配置

 	// 保存 Websocket 会话 UserId, 每个 UserId 只能连接一次
 	// 防止第三方直接连接 socket 调用 OpenAI API
 	ChatSession   *types.LMap[string, types.ChatSession]  //map[sessionId]UserId
 	ChatClients   *types.LMap[string, *types.WsClient]    // map[sessionId]Websocket 连接集合
 	ReqCancelFunc *types.LMap[string, context.CancelFunc] // HttpClient 请求取消 handle function
+	Functions     map[string]function.Function
 }

-func NewServer(appConfig *types.AppConfig) *AppServer {
+func NewServer(
+	appConfig *types.AppConfig,
+	funZaoBao function.FuncZaoBao,
+	funZhiHu function.FuncHeadlines,
+	funWeibo function.FuncWeiboHot) *AppServer {
 	gin.SetMode(gin.ReleaseMode)
 	gin.DefaultWriter = io.Discard
 	return &AppServer{
 		Debug:         false,
 		Config:        appConfig,
 		Engine:        gin.Default(),
-		ChatContexts:  types.NewLMap[string, []types.Message](),
+		ChatContexts:  types.NewLMap[string, []interface{}](),
 		ChatSession:   types.NewLMap[string, types.ChatSession](),
 		ChatClients:   types.NewLMap[string, *types.WsClient](),
 		ReqCancelFunc: types.NewLMap[string, context.CancelFunc](),
+		Functions: map[string]function.Function{
+			types.FuncZaoBao:   funZaoBao,
+			types.FuncWeibo:    funWeibo,
+			types.FuncHeadLine: funZhiHu,
+		},
 	}
 }

--- a/api/core/types/chat.go
+++ b/api/core/types/chat.go
@@ -2,17 +2,17 @@ package types

 // ApiRequest API 请求实体
 type ApiRequest struct {
-	Model       string    `json:"model"`
-	Temperature float32   `json:"temperature"`
-	MaxTokens   int       `json:"max_tokens"`
-	Stream      bool      `json:"stream"`
-	Messages    []Message `json:"messages"`
+	Model       string        `json:"model"`
+	Temperature float32       `json:"temperature"`
+	MaxTokens   int           `json:"max_tokens"`
+	Stream      bool          `json:"stream"`
+	Messages    []interface{} `json:"messages"`
+	Functions   []Function    `json:"functions"`
 }

 type Message struct {
-	Role         string       `json:"role"`
-	Content      string       `json:"content"`
-	FunctionCall FunctionCall `json:"function_call"`
+	Role    string `json:"role"`
+	Content string `json:"content"`
 }

 type ApiResponse struct {
@@ -21,8 +21,15 @@ type ApiResponse struct {

 // ChoiceItem API 响应实体
 type ChoiceItem struct {
-	Delta        Message `json:"delta"`
-	FinishReason string  `json:"finish_reason"`
+	Delta        Delta  `json:"delta"`
+	FinishReason string `json:"finish_reason"`
+}
+
+type Delta struct {
+	Role         string       `json:"role"`
+	Name         string       `json:"name"`
+	Content      interface{}  `json:"content"`
+	FunctionCall FunctionCall `json:"function_call,omitempty"`
 }

 // ChatSession 聊天会话对象
--- a/api/core/types/config.go
+++ b/api/core/types/config.go
@@ -72,7 +72,9 @@ type ChatConfig struct {
 	MaxTokens     int     `json:"max_tokens"`
 	EnableContext bool    `json:"enable_context"` // 是否开启聊天上下文
 	EnableHistory bool    `json:"enable_history"` // 是否允许保存聊天记录
-	ApiKey        string  `json:"api_key"`        // OpenAI  API key
+	ApiKey        string  `json:"api_key"`
+	ContextDeep   int     `json:"context_deep"` // 上下文深度
+
 }

 type SystemConfig struct {
--- a/api/core/types/function.go
+++ b/api/core/types/function.go
@@ -6,13 +6,71 @@ type FunctionCall struct {
 }

 type Function struct {
-	Name        string
-	Description string
-	Parameters  []Parameter
+	Name        string     `json:"name"`
+	Description string     `json:"description"`
+	Parameters  Parameters `json:"parameters"`
 }

-type Parameter struct {
-	Type       string
-	Required   []string
-	Properties map[string]interface{}
+type Parameters struct {
+	Type       string              `json:"type"`
+	Required   []string            `json:"required"`
+	Properties map[string]Property `json:"properties"`
+}
+
+type Property struct {
+	Type        string `json:"type"`
+	Description string `json:"description"`
+}
+
+const (
+	FuncZaoBao   = "zao_bao"   // 每日早报
+	FuncHeadLine = "headline"  // 今日头条
+	FuncWeibo    = "weibo_hot" // 微博热搜
+)
+
+var InnerFunctions = []Function{
+	{
+		Name:        FuncZaoBao,
+		Description: "每日早报，获取当天全球的热门新闻事件列表",
+		Parameters: Parameters{
+
+			Type: "object",
+			Properties: map[string]Property{
+				"text": {
+					Type:        "string",
+					Description: "",
+				},
+			},
+			Required: []string{},
+		},
+	},
+	{
+		Name:        FuncWeibo,
+		Description: "新浪微博热搜榜，微博当日热搜榜单",
+		Parameters: Parameters{
+			Type: "object",
+			Properties: map[string]Property{
+				"text": {
+					Type:        "string",
+					Description: "",
+				},
+			},
+			Required: []string{},
+		},
+	},
+
+	{
+		Name:        FuncHeadLine,
+		Description: "今日头条，给用户推荐当天的头条新闻，周榜热文",
+		Parameters: Parameters{
+			Type: "object",
+			Properties: map[string]Property{
+				"text": {
+					Type:        "string",
+					Description: "",
+				},
+			},
+			Required: []string{},
+		},
+	},
 }
--- a/api/core/types/locked_map.go
+++ b/api/core/types/locked_map.go
@@ -9,7 +9,7 @@ type MKey interface {
 	string | int
 }
 type MValue interface {
-	*WsClient | ChatSession | []Message | context.CancelFunc
+	*WsClient | ChatSession | context.CancelFunc | []interface{}
 }
 type LMap[K MKey, T MValue] struct {
 	lock sync.RWMutex
--- a/api/go.mod
+++ b/api/go.mod
@@ -5,37 +5,41 @@ go 1.19
 require (
 	github.com/BurntSushi/toml v1.1.0
 	github.com/aliyun/alibaba-cloud-sdk-go v1.62.405
+	github.com/eatmoreapple/openwechat v1.2.1
 	github.com/gin-contrib/sessions v0.0.5
-	github.com/gin-gonic/gin v1.9.0
+	github.com/gin-gonic/gin v1.9.1
 	github.com/gorilla/websocket v1.5.0
 	github.com/lionsoul2014/ip2region/binding/golang v0.0.0-20230415042440-a5e3d8259ae0
 	github.com/pkoukk/tiktoken-go v0.1.1-0.20230418101013-cae809389480
+	github.com/skip2/go-qrcode v0.0.0-20200617195104-da1b6568686e
 	github.com/syndtr/goleveldb v1.0.0
 	go.uber.org/zap v1.23.0
+	gopkg.in/natefinch/lumberjack.v2 v2.2.1
 	gorm.io/driver/mysql v1.4.7
 )

 require (
 	github.com/boj/redistore v0.0.0-20180917114910-cd5dcc76aeff // indirect
-	github.com/bytedance/sonic v1.8.0 // indirect
+	github.com/bytedance/sonic v1.9.1 // indirect
 	github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 // indirect
 	github.com/dlclark/regexp2 v1.8.1 // indirect
+	github.com/gabriel-vasile/mimetype v1.4.2 // indirect
 	github.com/go-sql-driver/mysql v1.7.0 // indirect
-	github.com/goccy/go-json v0.10.0 // indirect
+	github.com/goccy/go-json v0.10.2 // indirect
 	github.com/gomodule/redigo v2.0.0+incompatible // indirect
 	github.com/jinzhu/inflection v1.0.0 // indirect
 	github.com/jinzhu/now v1.1.5 // indirect
 	github.com/jmespath/go-jmespath v0.0.0-20180206201540-c2b33e8439af // indirect
-	github.com/klauspost/cpuid/v2 v2.0.9 // indirect
+	github.com/klauspost/cpuid/v2 v2.2.4 // indirect
 	github.com/opentracing/opentracing-go v1.2.1-0.20220228012449-10b1cf09e00b // indirect
-	github.com/pelletier/go-toml/v2 v2.0.6 // indirect
+	github.com/pelletier/go-toml/v2 v2.0.8 // indirect
 	github.com/quasoft/memstore v0.0.0-20191010062613-2bce066d2b0b // indirect
 	github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
 	go.uber.org/dig v1.16.1 // indirect
-	golang.org/x/arch v0.0.0-20210923205945-b76863e36670 // indirect
-	golang.org/x/net v0.9.0 // indirect
+	golang.org/x/arch v0.3.0 // indirect
+	golang.org/x/net v0.10.0 // indirect
 	golang.org/x/text v0.9.0 // indirect
-	google.golang.org/protobuf v1.28.1 // indirect
+	google.golang.org/protobuf v1.30.0 // indirect
 	gopkg.in/ini.v1 v1.66.2 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect
 )
@@ -44,21 +48,21 @@ require (
 	github.com/gin-contrib/sse v0.1.0 // indirect
 	github.com/go-playground/locales v0.14.1 // indirect
 	github.com/go-playground/universal-translator v0.18.1 // indirect
-	github.com/go-playground/validator/v10 v10.11.2 // indirect
+	github.com/go-playground/validator/v10 v10.14.0 // indirect
 	github.com/golang/snappy v0.0.1 // indirect
 	github.com/gorilla/context v1.1.1 // indirect
 	github.com/gorilla/securecookie v1.1.1 // indirect
 	github.com/gorilla/sessions v1.2.1 // indirect
 	github.com/json-iterator/go v1.1.12 // indirect
-	github.com/leodido/go-urn v1.2.1 // indirect
-	github.com/mattn/go-isatty v0.0.17 // indirect
+	github.com/leodido/go-urn v1.2.4 // indirect
+	github.com/mattn/go-isatty v0.0.19 // indirect
 	github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
 	github.com/modern-go/reflect2 v1.0.2 // indirect
-	github.com/ugorji/go/codec v1.2.9 // indirect
+	github.com/ugorji/go/codec v1.2.11 // indirect
 	go.uber.org/atomic v1.9.0 // indirect
 	go.uber.org/fx v1.19.3
 	go.uber.org/multierr v1.6.0 // indirect
-	golang.org/x/crypto v0.6.0
-	golang.org/x/sys v0.7.0 // indirect
+	golang.org/x/crypto v0.9.0
+	golang.org/x/sys v0.8.0 // indirect
 	gorm.io/gorm v1.25.1
 )
--- a/api/go.sum
+++ b/api/go.sum
@@ -6,8 +6,8 @@ github.com/benbjohnson/clock v1.3.0 h1:ip6w0uFQkncKQ979AypyG0ER7mqUSBdKLOgAle/AT
 github.com/boj/redistore v0.0.0-20180917114910-cd5dcc76aeff h1:RmdPFa+slIr4SCBg4st/l/vZWVe9QJKMXGO60Bxbe04=
 github.com/boj/redistore v0.0.0-20180917114910-cd5dcc76aeff/go.mod h1:+RTT1BOk5P97fT2CiHkbFQwkK3mjsFAP6zCYV2aXtjw=
 github.com/bytedance/sonic v1.5.0/go.mod h1:ED5hyg4y6t3/9Ku1R6dU/4KyJ48DZ4jPhfY1O2AihPM=
-github.com/bytedance/sonic v1.8.0 h1:ea0Xadu+sHlu7x5O3gKhRpQ1IKiMrSiHttPF0ybECuA=
-github.com/bytedance/sonic v1.8.0/go.mod h1:i736AoUSYt75HyZLoJW9ERYxcy6eaN6h4BZXU064P/U=
+github.com/bytedance/sonic v1.9.1 h1:6iJ6NqdoxCDr6mbY8h18oSO+cShGSMRGCEo7F2h0x8s=
+github.com/bytedance/sonic v1.9.1/go.mod h1:i736AoUSYt75HyZLoJW9ERYxcy6eaN6h4BZXU064P/U=
 github.com/chenzhuoyu/base64x v0.0.0-20211019084208-fb5309c8db06/go.mod h1:DH46F32mSOjUmXrMHnKwZdA8wcEefY7UVqBKYGjpdQY=
 github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 h1:qSGYFH7+jGhDF8vLC+iwCD4WpbV1EBDSzWkJODFLams=
 github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311/go.mod h1:b583jCggY9gE99b6G5LEC39OIiVsWj+R97kbl5odCEk=
@@ -16,24 +16,28 @@ github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/dlclark/regexp2 v1.8.1 h1:6Lcdwya6GjPUNsBct8Lg/yRPwMhABj269AAzdGSiR+0=
 github.com/dlclark/regexp2 v1.8.1/go.mod h1:DHkYz0B9wPfa6wondMfaivmHpzrQ3v9q8cnmRbL6yW8=
+github.com/eatmoreapple/openwechat v1.2.1 h1:ez4oqF/Y2NSEX/DbPV8lvj7JlfkYqvieeo4awx5lzfU=
+github.com/eatmoreapple/openwechat v1.2.1/go.mod h1:61HOzTyvLobGdgWhL68jfGNwTJEv0mhQ1miCXQrvWU8=
 github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
+github.com/gabriel-vasile/mimetype v1.4.2 h1:w5qFW6JKBz9Y393Y4q372O9A7cUSequkh1Q7OhCmWKU=
+github.com/gabriel-vasile/mimetype v1.4.2/go.mod h1:zApsH/mKG4w07erKIaJPFiX0Tsq9BFQgN3qGY5GnNgA=
 github.com/gin-contrib/sessions v0.0.5 h1:CATtfHmLMQrMNpJRgzjWXD7worTh7g7ritsQfmF+0jE=
 github.com/gin-contrib/sessions v0.0.5/go.mod h1:vYAuaUPqie3WUSsft6HUlCjlwwoJQs97miaG2+7neKY=
 github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=
 github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
-github.com/gin-gonic/gin v1.9.0 h1:OjyFBKICoexlu99ctXNR2gg+c5pKrKMuyjgARg9qeY8=
-github.com/gin-gonic/gin v1.9.0/go.mod h1:W1Me9+hsUSyj3CePGrd1/QrKJMSJ1Tu/0hFEH89961k=
+github.com/gin-gonic/gin v1.9.1 h1:4idEAncQnU5cB7BeOkPtxjfCSye0AAm1R0RVIqJ+Jmg=
+github.com/gin-gonic/gin v1.9.1/go.mod h1:hPrL7YrpYKXt5YId3A/Tnip5kqbEAP+KLuI3SUcPTeU=
 github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s=
 github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA=
 github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY=
 github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY=
 github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
-github.com/go-playground/validator/v10 v10.11.2 h1:q3SHpufmypg+erIExEKUmsgmhDTyhcJ38oeKGACXohU=
-github.com/go-playground/validator/v10 v10.11.2/go.mod h1:NieE624vt4SCTJtD87arVLvdmjPAeV8BQlHtMnw9D7s=
+github.com/go-playground/validator/v10 v10.14.0 h1:vgvQWe3XCz3gIeFDm/HnTIbj6UGmg/+t63MyGU2n5js=
+github.com/go-playground/validator/v10 v10.14.0/go.mod h1:9iXMNT7sEkjXb0I+enO7QXmzG6QCsPWY4zveKFVRSyU=
 github.com/go-sql-driver/mysql v1.7.0 h1:ueSltNNllEqE3qcWBTD0iQd3IpL/6U+mJxLkazJ7YPc=
 github.com/go-sql-driver/mysql v1.7.0/go.mod h1:OXbVy3sEdcQ2Doequ6Z5BW6fXNQTmx+9S1MCJN5yJMI=
-github.com/goccy/go-json v0.10.0 h1:mXKd9Qw4NuzShiRlOXKews24ufknHO7gx30lsDyokKA=
-github.com/goccy/go-json v0.10.0/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I=
+github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU=
+github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I=
 github.com/goji/httpauth v0.0.0-20160601135302-2da839ab0f4d/go.mod h1:nnjvkQ9ptGaCkuDUx6wNykzzlUixGxvkme+H/lnzb+A=
 github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=
 github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
@@ -66,19 +70,20 @@ github.com/jmespath/go-jmespath v0.0.0-20180206201540-c2b33e8439af/go.mod h1:Nht
 github.com/json-iterator/go v1.1.5/go.mod h1:+SdeFBvtyEkXs7REEP0seUULqWtbJapLOCVDaaPEHmU=
 github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=
 github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=
-github.com/klauspost/cpuid/v2 v2.0.9 h1:lgaqFMSdTdQYdZ04uHyN2d/eKdOMyi2YLSvlQIBFYa4=
 github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
+github.com/klauspost/cpuid/v2 v2.2.4 h1:acbojRNwl3o09bUq+yDCtZFc1aiwaAAxtcn8YkZXnvk=
+github.com/klauspost/cpuid/v2 v2.2.4/go.mod h1:RVVoqg1df56z8g3pUjL/3lE5UfnlrJX8tyFgg4nqhuY=
+github.com/kr/pretty v0.1.0 h1:L/CwN0zerZDmRFUapSPitk6f+Q3+0za1rQkzVuMiMFI=
 github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
-github.com/kr/pretty v0.3.0 h1:WgNl7dwNpEZ6jJ9k1snq4pZsg7DOEN8hP9Xw0Tsjwk0=
 github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
+github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE=
 github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
-github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
-github.com/leodido/go-urn v1.2.1 h1:BqpAaACuzVSgi/VLzGZIobT2z4v53pjosyNd9Yv6n/w=
-github.com/leodido/go-urn v1.2.1/go.mod h1:zt4jvISO2HfUBqxjfIshjdMTYS56ZS/qv49ictyFfxY=
+github.com/leodido/go-urn v1.2.4 h1:XlAE/cm/ms7TE/VMVoduSpNBoyc2dOxHs5MZSwAN63Q=
+github.com/leodido/go-urn v1.2.4/go.mod h1:7ZrI8mTSeBSHl/UaRyKQW1qZeMgak41ANeCNaVckg+4=
 github.com/lionsoul2014/ip2region/binding/golang v0.0.0-20230415042440-a5e3d8259ae0 h1:LgmjED/yQILqmUED4GaXjrINWe7YJh4HM6z2EvEINPs=
 github.com/lionsoul2014/ip2region/binding/golang v0.0.0-20230415042440-a5e3d8259ae0/go.mod h1:C5LA5UO2ZXJrLaPLYtE1wUJMiyd/nwWaCO5cw/2pSHs=
-github.com/mattn/go-isatty v0.0.17 h1:BTarxUcIeDqL27Mc+vyvdWYSL28zpIhv3RoTdsLMPng=
-github.com/mattn/go-isatty v0.0.17/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
+github.com/mattn/go-isatty v0.0.19 h1:JITubQf0MOLdlGRuRq+jtsDlekdYPia9ZFsB8h/APPA=
+github.com/mattn/go-isatty v0.0.19/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
 github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
 github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg=
 github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
@@ -92,8 +97,8 @@ github.com/onsi/gomega v1.4.3 h1:RE1xgDvH7imwFD45h+u2SgIfERHlS2yNG4DObb5BSKU=
 github.com/onsi/gomega v1.4.3/go.mod h1:ex+gbHU/CVuBBDIJjb2X0qEXbFg53c61hWP/1CpauHY=
 github.com/opentracing/opentracing-go v1.2.1-0.20220228012449-10b1cf09e00b h1:FfH+VrHHk6Lxt9HdVS0PXzSXFyS2NbZKXv33FYPol0A=
 github.com/opentracing/opentracing-go v1.2.1-0.20220228012449-10b1cf09e00b/go.mod h1:AC62GU6hc0BrNm+9RK9VSiwa/EUe1bkIeFORAMcHvJU=
-github.com/pelletier/go-toml/v2 v2.0.6 h1:nrzqCb7j9cDFj2coyLNLaZuJTLjWjlaz6nvTvIwycIU=
-github.com/pelletier/go-toml/v2 v2.0.6/go.mod h1:eumQOmlWiOPt5WriQQqoM5y18pDHwha2N+QD+EUNTek=
+github.com/pelletier/go-toml/v2 v2.0.8 h1:0ctb6s9mE31h0/lhu+J6OPmVeDxJn+kYnJc2jZR9tGQ=
+github.com/pelletier/go-toml/v2 v2.0.8/go.mod h1:vuYfssBdrU2XDZ9bYydBu6t+6a6PYNcZljzZR9VXg+4=
 github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
 github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
 github.com/pkoukk/tiktoken-go v0.1.1-0.20230418101013-cae809389480 h1:IFhPCcB0/HtnEN+ZoUGDT55YgFCymbFJ15kXqs3nv5w=
@@ -102,17 +107,19 @@ github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZb
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/quasoft/memstore v0.0.0-20191010062613-2bce066d2b0b h1:aUNXCGgukb4gtY99imuIeoh8Vr0GSwAlYxPAhqZrpFc=
 github.com/quasoft/memstore v0.0.0-20191010062613-2bce066d2b0b/go.mod h1:wTPjTepVu7uJBYgZ0SdWHQlIas582j6cn2jgk4DDdlg=
-github.com/rogpeppe/go-internal v1.8.0 h1:FCbCCtXNOY3UtUuHUYaghJg4y7Fd14rXifAYUAtL9R8=
+github.com/skip2/go-qrcode v0.0.0-20200617195104-da1b6568686e h1:MRM5ITcdelLK2j1vwZ3Je0FKVCfqOLp5zO6trqMLYs0=
+github.com/skip2/go-qrcode v0.0.0-20200617195104-da1b6568686e/go.mod h1:XV66xRDqSt+GTGFMVlhk3ULuV0y9ZmzeVGR4mloJI3M=
 github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
 github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
 github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
 github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
-github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
 github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
 github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
 github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
 github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
-github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8=
+github.com/stretchr/testify v1.8.2/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
+github.com/stretchr/testify v1.8.3 h1:RP3t2pwF7cMEbC1dqtB6poj3niw/9gnV4Cjg5oW5gtY=
+github.com/stretchr/testify v1.8.3/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
 github.com/syndtr/goleveldb v1.0.0 h1:fBdIW9lB4Iz0n9khmH8w27SJ3QEJ7+IgjPEwGSZiFdE=
 github.com/syndtr/goleveldb v1.0.0/go.mod h1:ZVVdQEZoIme9iO1Ch2Jdy24qqXrMMOU6lpPAyBWyWuQ=
 github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=
@@ -121,8 +128,8 @@ github.com/uber/jaeger-client-go v2.30.0+incompatible h1:D6wyKGCecFaSRUpo8lCVbaO
 github.com/uber/jaeger-client-go v2.30.0+incompatible/go.mod h1:WVhlPFC8FDjOFMMWRy2pZqQJSXxYSwNYOkTr/Z6d3Kk=
 github.com/uber/jaeger-lib v2.4.1+incompatible h1:td4jdvLcExb4cBISKIpHuGoVXh+dVKhn2Um6rjCsSsg=
 github.com/uber/jaeger-lib v2.4.1+incompatible/go.mod h1:ComeNDZlWwrWnDv8aPp0Ba6+uUTzImX/AauajbLI56U=
-github.com/ugorji/go/codec v1.2.9 h1:rmenucSohSTiyL09Y+l2OCk+FrMxGMzho2+tjr5ticU=
-github.com/ugorji/go/codec v1.2.9/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg=
+github.com/ugorji/go/codec v1.2.11 h1:BMaWp1Bb6fHwEtbplGBGJ498wD+LKlNSl25MjdZY4dU=
+github.com/ugorji/go/codec v1.2.11/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg=
 go.uber.org/atomic v1.7.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=
 go.uber.org/atomic v1.9.0 h1:ECmE8Bn/WFTYwEW/bpKD3M8VtR/zQVbavAoalC1PYyE=
 go.uber.org/atomic v1.9.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=
@@ -135,33 +142,37 @@ go.uber.org/multierr v1.6.0 h1:y6IPFStTAIT5Ytl7/XYmHvzXQ7S3g/IeZW9hyZ5thw4=
 go.uber.org/multierr v1.6.0/go.mod h1:cdWPpRnG4AhwMwsgIHip0KRBQjJy5kYEpYjJxpXp9iU=
 go.uber.org/zap v1.23.0 h1:OjGQ5KQDEUawVHxNwQgPpiypGHOxo2mNZsOqTak4fFY=
 go.uber.org/zap v1.23.0/go.mod h1:D+nX8jyLsMHMYrln8A0rJjFt/T/9/bGgIhAqxv5URuY=
-golang.org/x/arch v0.0.0-20210923205945-b76863e36670 h1:18EFjUmQOcUvxNYSkA6jO9VAiXCnxFY6NyDX0bHDmkU=
 golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
-golang.org/x/crypto v0.6.0 h1:qfktjS5LUO+fFKeJXZ+ikTRijMmljikvG68fpMMruSc=
-golang.org/x/crypto v0.6.0/go.mod h1:OFC/31mSvZgRz0V1QTNCzfAI1aIRzbiufJtkMIlEp58=
+golang.org/x/arch v0.3.0 h1:02VY4/ZcO/gBOH6PUaoiptASxtXU10jazRCP865E97k=
+golang.org/x/arch v0.3.0/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
+golang.org/x/crypto v0.9.0 h1:LF6fAI+IutBocDJ2OT0Q1g8plpYljMZ4+lty+dsqw3g=
+golang.org/x/crypto v0.9.0/go.mod h1:yrmDGqONDYtNj3tH8X9dzUun2m2lzPa9ngI6/RUPGR0=
 golang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
-golang.org/x/net v0.9.0 h1:aWJ/m6xSmxWBx+V0XRHTlrYrPG56jKsLdTFmsSsCzOM=
-golang.org/x/net v0.9.0/go.mod h1:d48xBJpPfHeWQsugry2m+kC02ZBRGRgulfHnEXEuWns=
+golang.org/x/net v0.10.0 h1:X2//UzNDwYmtCLn7To6G58Wr6f5ahEAQgKNzv9Y951M=
+golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg=
 golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sys v0.0.0-20180909124046-d0be0721c37e/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
-golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
-golang.org/x/sys v0.7.0 h1:3jlCCIQZPdOYu1h8BkNvLz8Kgwtae2cagcG/VamtZRU=
-golang.org/x/sys v0.7.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.0.0-20220704084225-05e143d24a9e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.8.0 h1:EBmGv8NaZBZTWvrbjNoL6HVt+IVy3QDQpJs7VRIw3tU=
+golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
 golang.org/x/text v0.9.0 h1:2sjJmO8cDvYveuX97RDLsxlyUxLl+GHoLxBiRdHllBE=
 golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8=
 golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4=
 golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
 google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
-google.golang.org/protobuf v1.28.1 h1:d0NfwRgPtno5B1Wa6L2DAG+KivqkdutMf1UhdNx175w=
-google.golang.org/protobuf v1.28.1/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=
+google.golang.org/protobuf v1.30.0 h1:kPPoIgf3TsEvrm0PFe15JQ+570QVxYzEvvHqChK+cng=
+google.golang.org/protobuf v1.30.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 h1:qIbj1fsPNlZgppZ+VLlY7N33q108Sa+fhmuc+sWQYwY=
 gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
-gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
 gopkg.in/fsnotify.v1 v1.4.7 h1:xOHLXZwVvI9hhs+cLKq5+I5onOuwQLhQwiu63xxlHs4=
 gopkg.in/fsnotify.v1 v1.4.7/go.mod h1:Tz8NjZHkW78fSQdbUxIjBTcgA1z1m8ZHf0WmKUhAMys=
 gopkg.in/ini.v1 v1.66.2 h1:XfR1dOYubytKy4Shzc2LHrrGhU0lDCfDGG1yLPmpgsI=
 gopkg.in/ini.v1 v1.66.2/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k=
+gopkg.in/natefinch/lumberjack.v2 v2.2.1 h1:bBRl1b0OH9s/DuPhuXpNl+VtCaJXFZ5/uEFST95x9zc=
+gopkg.in/natefinch/lumberjack.v2 v2.2.1/go.mod h1:YD8tP3GAjkrDg1eZH7EGmyESg/lsYskCTPBJVb9jqSc=
 gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 h1:uRGJdciOHaEIrze2W8Q3AKkepLTh2hOroT7a+7czfdQ=
 gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw=
 gopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
--- a/api/handler/chat_handler.go
+++ b/api/handler/chat_handler.go
@@ -5,7 +5,6 @@ import (
 	"bytes"
 	"chatplus/core"
 	"chatplus/core/types"
-	"chatplus/service/function"
 	"chatplus/store/model"
 	"chatplus/store/vo"
 	"chatplus/utils"
@@ -30,16 +29,17 @@ const ErrorMsg = "抱歉，AI 助手开小差了，请稍后再试。"

 type ChatHandler struct {
 	BaseHandler
-	db         *gorm.DB
-	funcZaoBao *function.FuncZaoBao
+	db *gorm.DB
 }

-func NewChatHandler(app *core.AppServer, db *gorm.DB, zaoBao *function.FuncZaoBao) *ChatHandler {
-	handler := ChatHandler{db: db, funcZaoBao: zaoBao}
+func NewChatHandler(app *core.AppServer, db *gorm.DB) *ChatHandler {
+	handler := ChatHandler{db: db}
 	handler.App = app
 	return &handler
 }

+var chatConfig types.ChatConfig
+
 // ChatHandle 处理聊天 WebSocket 请求
 func (h *ChatHandler) ChatHandle(c *gin.Context) {
 	ws, err := (&websocket.Upgrader{CheckOrigin: func(r *http.Request) bool { return true }}).Upgrade(c.Writer, c.Request, nil)
@@ -84,7 +84,17 @@ func (h *ChatHandler) ChatHandle(c *gin.Context) {
 	var chatRole model.ChatRole
 	res = h.db.First(&chatRole, roleId)
 	if res.Error != nil || !chatRole.Enable {
-		replyMessage(client, "当前聊天角色不存在或者未启用！！！")
+		replyMessage(client, "当前聊天角色不存在或者未启用，连接已关闭！！！")
+		c.Abort()
+		return
+	}
+
+	// 初始化聊天配置
+	var config model.Config
+	h.db.Where("marker", "chat").First(&config)
+	err = utils.JsonDecode(config.Config, &chatConfig)
+	if err != nil {
+		replyMessage(client, "加载系统配置失败，连接已关闭！！！")
 		c.Abort()
 		return
 	}
@@ -141,8 +151,8 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session types.ChatSession
 		return nil
 	}

-	if userVo.Calls <= 0 {
-		replyMessage(ws, "您的对话次数已经用尽，请联系管理员充值！")
+	if userVo.Calls <= 0 && userVo.ChatConfig.ApiKey == "" {
+		replyMessage(ws, "您的对话次数已经用尽，请联系管理员或者点击左下角菜单加入众筹获得100次对话！")
 		replyMessage(ws, "![](/images/wx.png)")
 		return nil
 	}
@@ -157,10 +167,11 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session types.ChatSession
 		Temperature: userVo.ChatConfig.Temperature,
 		MaxTokens:   userVo.ChatConfig.MaxTokens,
 		Stream:      true,
+		Functions:   types.InnerFunctions,
 	}

 	// 加载聊天上下文
-	var chatCtx []types.Message
+	var chatCtx []interface{}
 	if userVo.ChatConfig.EnableContext {
 		if h.App.ChatContexts.Has(session.ChatId) {
 			chatCtx = h.App.ChatContexts.Get(session.ChatId)
@@ -169,18 +180,23 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session types.ChatSession
 			var messages []types.Message
 			err := utils.JsonDecode(role.Context, &messages)
 			if err == nil {
-				chatCtx = messages
+				for _, v := range messages {
+					chatCtx = append(chatCtx, v)
+				}
 			}
-			// TODO: 这里默认加载最近 4 条聊天记录作为上下文，后期应该做成可配置的
-			var historyMessages []model.HistoryMessage
-			res := h.db.Where("chat_id = ?", session.ChatId).Limit(4).Order("created_at desc").Find(&historyMessages)
-			if res.Error == nil {
-				for _, msg := range historyMessages {
-					ms := types.Message{Role: "user", Content: msg.Content}
-					if msg.Type == types.ReplyMsg {
-						ms.Role = "assistant"
+
+			// 加载最近的聊天记录作为聊天上下文
+			if chatConfig.ContextDeep > 0 {
+				var historyMessages []model.HistoryMessage
+				res := h.db.Where("chat_id = ? and use_context = 1", session.ChatId).Limit(2).Order("created_at desc").Find(&historyMessages)
+				if res.Error == nil {
+					for _, msg := range historyMessages {
+						ms := types.Message{Role: "user", Content: msg.Content}
+						if msg.Type == types.ReplyMsg {
+							ms.Role = "assistant"
+						}
+						chatCtx = append(chatCtx, ms)
 					}
-					chatCtx = append(chatCtx, ms)
 				}
 			}
 		}
@@ -189,12 +205,17 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session types.ChatSession
 			logger.Info("聊天上下文：", chatCtx)
 		}
 	}
-	req.Messages = append(chatCtx, types.Message{
-		Role:    "user",
-		Content: prompt,
+	reqMgs := make([]interface{}, 0)
+	for _, m := range chatCtx {
+		reqMgs = append(reqMgs, m)
+	}
+
+	req.Messages = append(reqMgs, map[string]interface{}{
+		"role":    "user",
+		"content": prompt,
 	})
 	var apiKey string
-	response, err := h.fakeRequest(ctx, userVo, &apiKey, req)
+	response, err := h.doRequest(ctx, userVo, &apiKey, req)
 	if err != nil {
 		if strings.Contains(err.Error(), "context canceled") {
 			logger.Info("用户取消了请求：", prompt)
@@ -213,176 +234,203 @@ func (h *ChatHandler) sendMessage(ctx context.Context, session types.ChatSession
 		defer response.Body.Close()
 	}

-	//contentType := response.Header.Get("Content-Type")
-	//if strings.Contains(contentType, "text/event-stream") || true {
-	if true {
-		replyCreatedAt := time.Now()
-		// 循环读取 Chunk 消息
-		var message = types.Message{}
-		var contents = make([]string, 0)
-		var functionCall = false
-		var functionName string
-		var arguments = make([]string, 0)
-		reader := bufio.NewReader(response.Body)
-		for {
-			line, err := reader.ReadString('\n')
-			if err != nil {
-				if strings.Contains(err.Error(), "context canceled") {
-					logger.Info("用户取消了请求：", prompt)
-				} else {
-					logger.Error(err)
-				}
-				break
-			}
-			if !strings.Contains(line, "data:") || len(line) < 30 {
-				continue
-			}
-
-			var responseBody = types.ApiResponse{}
-			err = json.Unmarshal([]byte(line[6:]), &responseBody)
-			if err != nil || len(responseBody.Choices) == 0 { // 数据解析出错
-				logger.Error(err, line)
-				replyMessage(ws, ErrorMsg)
-				replyMessage(ws, "![](/images/wx.png)")
-				break
-			}
-
-			fun := responseBody.Choices[0].Delta.FunctionCall
-			if functionCall && fun.Name == "" {
-				arguments = append(arguments, fun.Arguments)
-				continue
-			}
-
-			if !utils.IsEmptyValue(fun) {
-				functionCall = true
-				functionName = fun.Name
-				replyChunkMessage(ws, types.WsMessage{Type: types.WsStart})
-				replyChunkMessage(ws, types.WsMessage{Type: types.WsMiddle, Content: fmt.Sprintf("正在调用函数 %s 作答 ...\n\n", functionName)})
-				continue
-			}
-
-			if responseBody.Choices[0].FinishReason == "function_call" { // 函数调用完毕
-				break
-			}
-
-			// 初始化 role
-			if responseBody.Choices[0].Delta.Role != "" && message.Role == "" {
-				message.Role = responseBody.Choices[0].Delta.Role
-				replyChunkMessage(ws, types.WsMessage{Type: types.WsStart})
-				continue
-			} else if responseBody.Choices[0].FinishReason != "" {
-				break // 输出完成或者输出中断了
-			} else {
-				content := responseBody.Choices[0].Delta.Content
-				contents = append(contents, content)
-				replyChunkMessage(ws, types.WsMessage{
-					Type:    types.WsMiddle,
-					Content: responseBody.Choices[0].Delta.Content,
-				})
-			}
-		} // end for
-
-		if functionCall { // 调用函数完成任务
-			// TODO 调用函数完成任务
-			data, err := h.funcZaoBao.Fetch()
-			if err != nil {
-				replyChunkMessage(ws, types.WsMessage{
-					Type:    types.WsMiddle,
-					Content: "调用函数出错",
-				})
-			} else {
-				replyChunkMessage(ws, types.WsMessage{
-					Type:    types.WsMiddle,
-					Content: data,
-				})
-			}
-			contents = append(contents, data)
-		}
-
-		// 消息发送成功
-		if len(contents) > 0 {
-			// 更新用户的对话次数
-			res := h.db.Model(&user).UpdateColumn("calls", gorm.Expr("calls - ?", 1))
-			if res.Error != nil {
-				return res.Error
-			}
-
-			if message.Role == "" {
-				message.Role = "assistant"
-			}
-			message.Content = strings.Join(contents, "")
-			useMsg := types.Message{Role: "user", Content: prompt}
-
-			// 更新上下文消息，如果是调用函数则不需要更新上下文
-			if userVo.ChatConfig.EnableContext && functionCall == false {
-				chatCtx = append(chatCtx, useMsg)  // 提问消息
-				chatCtx = append(chatCtx, message) // 回复消息
-				h.App.ChatContexts.Put(session.ChatId, chatCtx)
-			}
-
-			// 追加聊天记录
-			if userVo.ChatConfig.EnableHistory {
-				// for prompt
-				token, err := utils.CalcTokens(prompt, req.Model)
+	contentType := response.Header.Get("Content-Type")
+	if strings.Contains(contentType, "text/event-stream") {
+		if true {
+			replyCreatedAt := time.Now()
+			// 循环读取 Chunk 消息
+			var message = types.Message{}
+			var contents = make([]string, 0)
+			var functionCall = false
+			var functionName string
+			var arguments = make([]string, 0)
+			reader := bufio.NewReader(response.Body)
+			for {
+				line, err := reader.ReadString('\n')
 				if err != nil {
-					logger.Error(err)
+					if strings.Contains(err.Error(), "context canceled") {
+						logger.Info("用户取消了请求：", prompt)
+					} else if err != io.EOF {
+						logger.Error(err)
+					}
+					break
 				}
-				historyUserMsg := model.HistoryMessage{
-					UserId:  userVo.Id,
-					ChatId:  session.ChatId,
-					RoleId:  role.Id,
-					Type:    types.PromptMsg,
-					Icon:    user.Avatar,
-					Content: prompt,
-					Tokens:  token,
-				}
-				historyUserMsg.CreatedAt = promptCreatedAt
-				historyUserMsg.UpdatedAt = promptCreatedAt
-				res := h.db.Save(&historyUserMsg)
-				if res.Error != nil {
-					logger.Error("failed to save prompt history message: ", res.Error)
+				if !strings.Contains(line, "data:") || len(line) < 30 {
+					continue
 				}

-				// for reply
-				token, err = utils.CalcTokens(message.Content, req.Model)
+				var responseBody = types.ApiResponse{}
+				err = json.Unmarshal([]byte(line[6:]), &responseBody)
+				if err != nil || len(responseBody.Choices) == 0 { // 数据解析出错
+					logger.Error(err, line)
+					replyMessage(ws, ErrorMsg)
+					replyMessage(ws, "![](/images/wx.png)")
+					break
+				}
+
+				fun := responseBody.Choices[0].Delta.FunctionCall
+				if functionCall && fun.Name == "" {
+					arguments = append(arguments, fun.Arguments)
+					continue
+				}
+
+				if !utils.IsEmptyValue(fun) {
+					functionCall = true
+					functionName = fun.Name
+					f := h.App.Functions[functionName]
+					replyChunkMessage(ws, types.WsMessage{Type: types.WsStart})
+					replyChunkMessage(ws, types.WsMessage{Type: types.WsMiddle, Content: fmt.Sprintf("正在调用函数 `%s` 作答 ...\n\n", f.Name())})
+					continue
+				}
+
+				if responseBody.Choices[0].FinishReason == "function_call" { // 函数调用完毕
+					break
+				}
+
+				// 初始化 role
+				if responseBody.Choices[0].Delta.Role != "" && message.Role == "" {
+					message.Role = responseBody.Choices[0].Delta.Role
+					replyChunkMessage(ws, types.WsMessage{Type: types.WsStart})
+					continue
+				} else if responseBody.Choices[0].FinishReason != "" {
+					break // 输出完成或者输出中断了
+				} else {
+					content := responseBody.Choices[0].Delta.Content
+					contents = append(contents, utils.InterfaceToString(content))
+					replyChunkMessage(ws, types.WsMessage{
+						Type:    types.WsMiddle,
+						Content: utils.InterfaceToString(responseBody.Choices[0].Delta.Content),
+					})
+				}
+			} // end for
+
+			if functionCall { // 调用函数完成任务
+				logger.Info(functionName)
+				logger.Info(arguments)
+				f := h.App.Functions[functionName]
+				// TODO 调用函数完成任务
+				data, err := f.Invoke(arguments)
 				if err != nil {
-					logger.Error(err)
+					replyChunkMessage(ws, types.WsMessage{
+						Type:    types.WsMiddle,
+						Content: "调用函数出错",
+					})
+				} else {
+					replyChunkMessage(ws, types.WsMessage{
+						Type:    types.WsMiddle,
+						Content: data,
+					})
 				}
-				historyReplyMsg := model.HistoryMessage{
-					UserId:  userVo.Id,
-					ChatId:  session.ChatId,
-					RoleId:  role.Id,
-					Type:    types.ReplyMsg,
-					Icon:    role.Icon,
-					Content: message.Content,
-					Tokens:  token,
-				}
-				historyReplyMsg.CreatedAt = replyCreatedAt
-				historyReplyMsg.UpdatedAt = replyCreatedAt
-				res = h.db.Create(&historyReplyMsg)
-				if res.Error != nil {
-					logger.Error("failed to save reply history message: ", res.Error)
-				}
-
-				// 统计用户 token 数量
-				h.db.Model(&user).UpdateColumn("tokens", gorm.Expr("tokens + ?",
-					historyUserMsg.Tokens+historyReplyMsg.Tokens))
+				contents = append(contents, data)
 			}

-			// 保存当前会话
-			var chatItem model.ChatItem
-			res = h.db.Where("chat_id = ?", session.ChatId).First(&chatItem)
-			if res.Error != nil {
-				chatItem.ChatId = session.ChatId
-				chatItem.UserId = session.UserId
-				chatItem.RoleId = role.Id
-				chatItem.Model = session.Model
-				if utf8.RuneCountInString(prompt) > 30 {
-					chatItem.Title = string([]rune(prompt)[:30]) + "..."
-				} else {
-					chatItem.Title = prompt
+			// 消息发送成功
+			if len(contents) > 0 {
+				// 更新用户的对话次数
+				if userVo.ChatConfig.ApiKey == "" { // 如果用户使用的是自己绑定的 API KEY 则不扣减对话次数
+					res := h.db.Model(&user).UpdateColumn("calls", gorm.Expr("calls - ?", 1))
+					if res.Error != nil {
+						return res.Error
+					}
+				}
+
+				if message.Role == "" {
+					message.Role = "assistant"
+				}
+				message.Content = strings.Join(contents, "")
+				useMsg := types.Message{Role: "user", Content: prompt}
+
+				// 更新上下文消息，如果是调用函数则不需要更新上下文
+				if userVo.ChatConfig.EnableContext && functionCall == false {
+					chatCtx = append(chatCtx, useMsg)  // 提问消息
+					chatCtx = append(chatCtx, message) // 回复消息
+					h.App.ChatContexts.Put(session.ChatId, chatCtx)
+				}
+
+				// 追加聊天记录
+				if userVo.ChatConfig.EnableHistory {
+					useContext := true
+					if functionCall {
+						useContext = false
+					}
+
+					// for prompt
+					token, err := utils.CalcTokens(prompt, req.Model)
+					if err != nil {
+						logger.Error(err)
+					}
+					historyUserMsg := model.HistoryMessage{
+						UserId:     userVo.Id,
+						ChatId:     session.ChatId,
+						RoleId:     role.Id,
+						Type:       types.PromptMsg,
+						Icon:       user.Avatar,
+						Content:    prompt,
+						Tokens:     token,
+						UseContext: useContext,
+					}
+					historyUserMsg.CreatedAt = promptCreatedAt
+					historyUserMsg.UpdatedAt = promptCreatedAt
+					res := h.db.Save(&historyUserMsg)
+					if res.Error != nil {
+						logger.Error("failed to save prompt history message: ", res.Error)
+					}
+
+					// for reply
+					token, err = utils.CalcTokens(message.Content, req.Model)
+					if err != nil {
+						logger.Error(err)
+					}
+					historyReplyMsg := model.HistoryMessage{
+						UserId:     userVo.Id,
+						ChatId:     session.ChatId,
+						RoleId:     role.Id,
+						Type:       types.ReplyMsg,
+						Icon:       role.Icon,
+						Content:    message.Content,
+						Tokens:     token,
+						UseContext: useContext,
+					}
+					historyReplyMsg.CreatedAt = replyCreatedAt
+					historyReplyMsg.UpdatedAt = replyCreatedAt
+					res = h.db.Create(&historyReplyMsg)
+					if res.Error != nil {
+						logger.Error("failed to save reply history message: ", res.Error)
+					}
+
+					// 计算本次对话消耗的总 token 数量
+					var totalTokens = 0
+					if functionCall { // 函数名 + 参数 token
+						tokens, _ := utils.CalcTokens(functionName, req.Model)
+						totalTokens += tokens
+						tokens, _ = utils.CalcTokens(utils.InterfaceToString(arguments), req.Model)
+						totalTokens += tokens
+					} else {
+						req.Messages = append(req.Messages, message)
+						totalTokens += getTotalTokens(req)
+					}
+					//replyChunkMessage(ws, types.WsMessage{Type: types.WsMiddle, Content: fmt.Sprintf("\n\n `本轮对话共消耗 Token 数量: %d`", totalTokens+11)})
+					if userVo.ChatConfig.ApiKey != "" { // 调用自己的 API KEY 不计算 token 消耗
+						h.db.Model(&user).UpdateColumn("tokens", gorm.Expr("tokens + ?",
+							totalTokens))
+					}
+				}
+
+				// 保存当前会话
+				var chatItem model.ChatItem
+				res = h.db.Where("chat_id = ?", session.ChatId).First(&chatItem)
+				if res.Error != nil {
+					chatItem.ChatId = session.ChatId
+					chatItem.UserId = session.UserId
+					chatItem.RoleId = role.Id
+					chatItem.Model = session.Model
+					if utf8.RuneCountInString(prompt) > 30 {
+						chatItem.Title = string([]rune(prompt)[:30]) + "..."
+					} else {
+						chatItem.Title = prompt
+					}
+					h.db.Create(&chatItem)
 				}
-				h.db.Create(&chatItem)
 			}
 		}
 	} else {
@@ -469,13 +517,6 @@ func (h *ChatHandler) doRequest(ctx context.Context, user vo.User, apiKey *strin
 	return client.Do(request)
 }

-func (h *ChatHandler) fakeRequest(ctx context.Context, user vo.User, apiKey *string, req types.ApiRequest) (*http.Response, error) {
-	link := "https://img.r9it.com/chatgpt/response"
-	client := &http.Client{}
-	request, _ := http.NewRequest(http.MethodGet, link, nil)
-	return client.Do(request)
-}
-
 // 回复客户片段端消息
 func replyChunkMessage(client types.Client, message types.WsMessage) {
 	msg, err := json.Marshal(message)
@@ -509,6 +550,26 @@ func (h *ChatHandler) Tokens(c *gin.Context) {
 	resp.SUCCESS(c, tokens)
 }

+func getTotalTokens(req types.ApiRequest) int {
+	encode := utils.JsonEncode(req.Messages)
+	var items []map[string]interface{}
+	err := utils.JsonDecode(encode, &items)
+	if err != nil {
+		return 0
+	}
+	tokens := 0
+	for _, item := range items {
+		content, ok := item["content"]
+		if ok && !utils.IsEmptyValue(content) {
+			t, err := utils.CalcTokens(utils.InterfaceToString(content), req.Model)
+			if err == nil {
+				tokens += t
+			}
+		}
+	}
+	return tokens
+}
+
 // StopGenerate 停止生成
 func (h *ChatHandler) StopGenerate(c *gin.Context) {
 	sessionId := c.Query("session_id")
--- a/api/handler/reward_handler.go
+++ b/api/handler/reward_handler.go
@@ -0,0 +1,72 @@
+package handler
+
+import (
+	"chatplus/core"
+	"chatplus/core/types"
+	"chatplus/store/model"
+	"chatplus/utils"
+	"chatplus/utils/resp"
+	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
+)
+
+type RewardHandler struct {
+	BaseHandler
+	db *gorm.DB
+}
+
+func NewRewardHandler(server *core.AppServer, db *gorm.DB) *RewardHandler {
+	h := RewardHandler{db: db}
+	h.App = server
+	return &h
+}
+
+// Verify 打赏码核销
+func (h *RewardHandler) Verify(c *gin.Context) {
+	var data struct {
+		TxId string `json:"tx_id"`
+	}
+	if err := c.ShouldBindJSON(&data); err != nil {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+
+	var item model.Reward
+	res := h.db.Where("tx_id = ?", data.TxId).First(&item)
+	if res.Error != nil {
+		resp.ERROR(c, "无效的众筹交易流水号！")
+		return
+	}
+
+	if item.Status {
+		resp.ERROR(c, "当前众筹交易流水号已经被核销，请不要重复核销！")
+		return
+	}
+
+	user, err := utils.GetLoginUser(c, h.db)
+	if err != nil {
+		resp.HACKER(c)
+		return
+	}
+
+	tx := h.db.Begin()
+	calls := (item.Amount + 0.1) * 10
+	res = h.db.Model(&user).UpdateColumn("calls", gorm.Expr("calls + ?", calls))
+	if res.Error != nil {
+		resp.ERROR(c, "更新数据库失败！")
+		return
+	}
+
+	// 更新核销状态
+	item.Status = true
+	res = h.db.Updates(&item)
+	if res.Error != nil {
+		tx.Rollback()
+		resp.ERROR(c, "更新数据库失败！")
+		return
+	}
+
+	tx.Commit()
+	resp.SUCCESS(c)
+
+}
--- a/api/handler/verify_handler.go
+++ b/api/handler/verify_handler.go
@@ -13,7 +13,7 @@ import (
 	"github.com/gin-gonic/gin"
 )

-// 生成验证的控制器
+// 短信验证码控制器

 type VerifyHandler struct {
 	BaseHandler
--- a/api/logger/logger.go
+++ b/api/logger/logger.go
@@ -2,25 +2,59 @@ package logger

 import (
 	"go.uber.org/zap"
+	"go.uber.org/zap/zapcore"
+	"gopkg.in/natefinch/lumberjack.v2"
+	"os"
 )

-var logger *zap.SugaredLogger
+var logger *zap.Logger
+var sugarLogger *zap.SugaredLogger

 func GetLogger() *zap.SugaredLogger {
-	if logger != nil {
-		return logger
+	if sugarLogger != nil {
+		return sugarLogger
 	}

-	logLevel := zap.NewAtomicLevel()
-	logLevel.SetLevel(zap.InfoLevel)
-	log, _ := zap.Config{
-		Level:            logLevel,
-		Development:      false,
-		Encoding:         "console",
-		EncoderConfig:    zap.NewDevelopmentEncoderConfig(),
-		OutputPaths:      []string{"stderr"},
-		ErrorOutputPaths: []string{"stderr"},
-	}.Build()
-	logger = log.Sugar()
-	return logger
+	logLevel := zap.NewAtomicLevelAt(zapcore.InfoLevel)
+	encoder := getEncoder()
+	writerSyncer := getLogWriter()
+	fileCore := zapcore.NewCore(encoder, writerSyncer, logLevel)
+	consoleOutput := zapcore.Lock(os.Stdout)
+	consoleCore := zapcore.NewCore(
+		encoder,
+		consoleOutput,
+		logLevel,
+	)
+	core := zapcore.NewTee(fileCore, consoleCore)
+	logger = zap.New(core, zap.AddCaller())
+	sugarLogger = logger.Sugar()
+	return sugarLogger
+}
+
+// core 三个参数之  编码
+func getEncoder() zapcore.Encoder {
+	encoderConfig := zapcore.EncoderConfig{
+		TimeKey:        "time",
+		LevelKey:       "level",
+		NameKey:        "logger",
+		CallerKey:      "caller",
+		MessageKey:     "msg",
+		StacktraceKey:  "stacktrace",
+		EncodeTime:     zapcore.ISO8601TimeEncoder,
+		EncodeDuration: zapcore.SecondsDurationEncoder,
+		EncodeCaller:   zapcore.ShortCallerEncoder,
+		EncodeLevel:    zapcore.CapitalLevelEncoder,
+	}
+	return zapcore.NewConsoleEncoder(encoderConfig)
+}
+
+func getLogWriter() zapcore.WriteSyncer {
+	lumberJackLogger := &lumberjack.Logger{
+		Filename:   "logs/app.log",
+		MaxSize:    10,
+		MaxBackups: 5,
+		MaxAge:     30,
+		Compress:   false,
+	}
+	return zapcore.AddSync(lumberJackLogger)
 }
--- a/api/main.go
+++ b/api/main.go
@@ -6,11 +6,13 @@ import (
 	"chatplus/handler"
 	"chatplus/handler/admin"
 	logger2 "chatplus/logger"
+	"chatplus/modules/wexin"
 	"chatplus/service"
 	"chatplus/service/function"
 	"chatplus/store"
 	"context"
 	"embed"
+	"errors"
 	"io"
 	"log"
 	"os"
@@ -100,10 +102,38 @@ func main() {
 			return xdb.NewWithBuffer(cBuff)
 		}),

+		// 创建微信机器人
+		fx.Provide(wexin.NewWeChatBot),
+		fx.Invoke(func(bot *wexin.WeChatBot) {
+			go func() {
+				err := bot.Login()
+				if err != nil {
+					log.Fatal(err)
+				}
+			}()
+		}),
+
 		// 创建函数
-		fx.Provide(func() *function.FuncZaoBao {
-			token := os.Getenv("AL_API_TOKEN")
-			return function.NewZaoBao(token)
+		fx.Provide(func() (function.FuncZaoBao, error) {
+			apiToken := os.Getenv("AL_API_TOKEN")
+			if apiToken == "" {
+				return function.FuncZaoBao{}, errors.New("invalid AL api token")
+			}
+			return function.NewZaoBao(apiToken), nil
+		}),
+		fx.Provide(func() (function.FuncWeiboHot, error) {
+			apiToken := os.Getenv("AL_API_TOKEN")
+			if apiToken == "" {
+				return function.FuncWeiboHot{}, errors.New("invalid AL api token")
+			}
+			return function.NewWeiboHot(apiToken), nil
+		}),
+		fx.Provide(func() (function.FuncHeadlines, error) {
+			apiToken := os.Getenv("AL_API_TOKEN")
+			if apiToken == "" {
+				return function.FuncHeadlines{}, errors.New("invalid AL api token")
+			}
+			return function.NewHeadLines(apiToken), nil
 		}),

 		// 创建控制器
@@ -112,6 +142,7 @@ func main() {
 		fx.Provide(handler.NewChatHandler),
 		fx.Provide(handler.NewUploadHandler),
 		fx.Provide(handler.NewVerifyHandler),
+		fx.Provide(handler.NewRewardHandler),

 		fx.Provide(admin.NewConfigHandler),
 		fx.Provide(admin.NewAdminHandler),
@@ -157,6 +188,10 @@ func main() {
 			group.GET("token", h.Token)
 			group.POST("sms", h.SendMsg)
 		}),
+		fx.Invoke(func(s *core.AppServer, h *handler.RewardHandler) {
+			group := s.Engine.Group("/api/reward/")
+			group.POST("verify", h.Verify)
+		}),

 		// 管理后台控制器
 		fx.Invoke(func(s *core.AppServer, h *admin.ConfigHandler) {
--- a/api/modules/wexin/handler.go
+++ b/api/modules/wexin/handler.go
@@ -0,0 +1,54 @@
+package wexin
+
+import (
+	"chatplus/store/model"
+	"github.com/eatmoreapple/openwechat"
+	"github.com/skip2/go-qrcode"
+	"gorm.io/gorm"
+)
+
+// MessageHandler 消息处理
+func MessageHandler(msg *openwechat.Message, db *gorm.DB) {
+	sender, err := msg.Sender()
+	if err != nil {
+		return
+	}
+	
+	// 只处理微信支付的推送消息
+	if sender.NickName == "微信支付" ||
+		msg.MsgType == openwechat.MsgTypeApp ||
+		msg.AppMsgType == openwechat.AppMsgTypeUrl {
+		// 解析支付金额
+		message, err := parseTransactionMessage(msg.Content)
+		if err == nil {
+			transaction := extractTransaction(message)
+			logger.Infof("解析到收款信息：%+v", transaction)
+			if transaction.Amount <= 0 {
+				return
+			}
+			var item model.Reward
+			res := db.Where("tx_id = ?", transaction.TransId).First(&item)
+			if res.Error == nil {
+				logger.Infof("当前交易 ID %s 己经存在！", transaction.TransId)
+				return
+			}
+
+			res = db.Create(&model.Reward{
+				TxId:   transaction.TransId,
+				Amount: transaction.Amount,
+				Remark: transaction.Remark,
+				Status: false,
+			})
+			if res.Error != nil {
+				logger.Errorf("交易保存失败，ID: %s", transaction.TransId)
+			}
+		}
+	}
+}
+
+// QrCodeCallBack 登录扫码回调，
+func QrCodeCallBack(uuid string) {
+	logger.Info("请使用微信扫描下面二维码登录")
+	q, _ := qrcode.New("https://login.weixin.qq.com/l/"+uuid, qrcode.Medium)
+	logger.Info(q.ToString(true))
+}
--- a/api/modules/wexin/tranaction.go
+++ b/api/modules/wexin/tranaction.go
@@ -0,0 +1,68 @@
+package wexin
+
+import (
+	"encoding/xml"
+	"strconv"
+	"strings"
+)
+
+// Message 转账消息
+type Message struct {
+	XMLName xml.Name `xml:"msg"`
+	AppMsg  struct {
+		Des string `xml:"des"`
+		Url string `xml:"url"`
+	} `xml:"appmsg"`
+}
+
+// Transaction 解析后的交易信息
+type Transaction struct {
+	TransId string  `json:"trans_id"` // 微信转账交易 ID
+	Amount  float64 `json:"amount"`   // 微信转账交易金额
+	Remark  string  `json:"remark"`   // 转账备注
+}
+
+// 解析微信转账消息
+func parseTransactionMessage(xmlData string) (*Message, error) {
+	var msg Message
+	if err := xml.Unmarshal([]byte(xmlData), &msg); err != nil {
+		return nil, err
+	}
+
+	return &msg, nil
+}
+
+// 导出交易信息
+func extractTransaction(message *Message) Transaction {
+	var tx = Transaction{}
+	// 导出交易金额和备注
+	lines := strings.Split(message.AppMsg.Des, "\n")
+	for _, line := range lines {
+		line = strings.TrimSpace(line)
+		if len(line) == 0 {
+			continue
+		}
+		// 解析收款金额
+		prefix := "收款金额￥"
+		if strings.HasPrefix(line, prefix) {
+			if value, err := strconv.ParseFloat(line[len(prefix):], 64); err == nil {
+				tx.Amount = value
+				continue
+			}
+		}
+		// 解析收款备注
+		prefix = "付款方备注"
+		if strings.HasPrefix(line, prefix) {
+			tx.Remark = line[len(prefix):]
+			break
+		}
+	}
+
+	// 解析交易 ID
+	index := strings.Index(message.AppMsg.Url, "trans_id=")
+	if index != -1 {
+		end := strings.LastIndex(message.AppMsg.Url, "&")
+		tx.TransId = strings.TrimSpace(message.AppMsg.Url[index+9 : end])
+	}
+	return tx
+}
--- a/api/modules/wexin/wechat_bot.go
+++ b/api/modules/wexin/wechat_bot.go
@@ -0,0 +1,42 @@
+package wexin
+
+import (
+	logger2 "chatplus/logger"
+	"github.com/eatmoreapple/openwechat"
+	"gorm.io/gorm"
+)
+
+// 微信收款机器人服务
+var logger = logger2.GetLogger()
+
+type WeChatBot struct {
+	bot *openwechat.Bot
+	db  *gorm.DB
+}
+
+func NewWeChatBot(db *gorm.DB) *WeChatBot {
+	bot := openwechat.DefaultBot(openwechat.Desktop)
+	// 注册消息处理函数
+	bot.MessageHandler = func(msg *openwechat.Message) {
+		MessageHandler(msg, db)
+	}
+	// 注册登陆二维码回调
+	bot.UUIDCallback = QrCodeCallBack
+	return &WeChatBot{
+		bot: bot,
+		db:  db,
+	}
+}
+
+func (b *WeChatBot) Login() error {
+	// 创建热存储容器对象
+	reloadStorage := openwechat.NewJsonFileHotReloadStorage("storage.json")
+	// 执行热登录
+	err := b.bot.HotLogin(reloadStorage)
+	if err != nil {
+		logger.Error("login error: %v", err)
+		return b.bot.Login()
+	}
+	logger.Info("微信登录成功！")
+	return nil
+}
--- a/api/service/function/function.go
+++ b/api/service/function/function.go
@@ -0,0 +1,11 @@
+package function
+
+type Function interface {
+	Invoke(...interface{}) (string, error)
+	Name() string
+}
+
+type resVo struct {
+	Code int    `json:"code"`
+	Msg  string `json:"msg"`
+}
--- a/api/service/function/tou_tiao.go
+++ b/api/service/function/tou_tiao.go
@@ -0,0 +1,60 @@
+package function
+
+import (
+	"chatplus/utils"
+	"fmt"
+	"strings"
+)
+
+// 今日头条函数实现
+
+type FuncHeadlines struct {
+	name   string
+	apiURL string
+	token  string
+}
+
+func NewHeadLines(token string) FuncHeadlines {
+	return FuncHeadlines{name: "今日头条", apiURL: "https://v2.alapi.cn/api/tophub/get", token: token}
+}
+
+type HeadLineVo struct {
+	resVo
+	Data struct {
+		Name       string `json:"name"`
+		LastUpdate string `json:"last_update"`
+		List       []struct {
+			Title string `json:"title"`
+			Link  string `json:"link"`
+			Other string `json:"other"`
+		} `json:"list"`
+	} `json:"data"`
+}
+
+func (f FuncHeadlines) Invoke(...interface{}) (string, error) {
+
+	url := fmt.Sprintf("%s?type=toutiao&token=%s", f.apiURL, f.token)
+	bytes, err := utils.HttpGet(url, "")
+	if err != nil {
+		return "", err
+	}
+	var res HeadLineVo
+	err = utils.JsonDecode(string(bytes), &res)
+	if err != nil {
+		return "", err
+	}
+
+	if res.Code != 200 {
+		return "", fmt.Errorf("call api fail: %s", res.Msg)
+	}
+	builder := make([]string, 0)
+	builder = append(builder, fmt.Sprintf("**%s**，最新更新：%s", res.Data.Name, res.Data.LastUpdate))
+	for i, v := range res.Data.List {
+		builder = append(builder, fmt.Sprintf("%d、 [%s](%s) [%s]", i+1, v.Title, v.Link, v.Other))
+	}
+	return strings.Join(builder, "\n\n"), nil
+}
+
+func (f FuncHeadlines) Name() string {
+	return f.name
+}
--- a/api/service/function/weibo_hot.go
+++ b/api/service/function/weibo_hot.go
@@ -0,0 +1,56 @@
+package function
+
+import (
+	"chatplus/utils"
+	"fmt"
+	"strings"
+)
+
+// 微博热搜函数实现
+
+type FuncWeiboHot struct {
+	name   string
+	apiURL string
+	token  string
+}
+
+func NewWeiboHot(token string) FuncWeiboHot {
+	return FuncWeiboHot{name: "微博热搜", apiURL: "https://v2.alapi.cn/api/new/wbtop", token: token}
+}
+
+type WeiBoVo struct {
+	resVo
+	Data []struct {
+		HotWord    string `json:"hot_word"`
+		HotWordNum int    `json:"hot_word_num"`
+		Url        string `json:"url"`
+	} `json:"data"`
+}
+
+func (f FuncWeiboHot) Invoke(...interface{}) (string, error) {
+
+	url := fmt.Sprintf("%s?num=10&token=%s", f.apiURL, f.token)
+	bytes, err := utils.HttpGet(url, "")
+	if err != nil {
+		return "", err
+	}
+	var res WeiBoVo
+	err = utils.JsonDecode(string(bytes), &res)
+	if err != nil {
+		return "", err
+	}
+
+	if res.Code != 200 {
+		return "", fmt.Errorf("call api fail: %s", res.Msg)
+	}
+	builder := make([]string, 0)
+	builder = append(builder, "**新浪微博今日热搜：**")
+	for i, v := range res.Data {
+		builder = append(builder, fmt.Sprintf("%d、 [%s](%s) [热度：%d]", i+1, v.HotWord, v.Url, v.HotWordNum))
+	}
+	return strings.Join(builder, "\n\n"), nil
+}
+
+func (f FuncWeiboHot) Name() string {
+	return f.name
+}
--- a/api/service/function/zao_bao.go
+++ b/api/service/function/zao_bao.go
@@ -9,17 +9,17 @@ import (
 // 每日早报函数实现

 type FuncZaoBao struct {
+	name   string
 	apiURL string
 	token  string
 }

-func NewZaoBao(token string) *FuncZaoBao {
-	return &FuncZaoBao{apiURL: "https://v2.alapi.cn/api/zaobao", token: token}
+func NewZaoBao(token string) FuncZaoBao {
+	return FuncZaoBao{name: "每日早报", apiURL: "https://v2.alapi.cn/api/zaobao", token: token}
 }

-type resVo struct {
-	Code int    `json:"code"`
-	Msg  string `json:"msg"`
+type ZaoBaoVo struct {
+	resVo
 	Data struct {
 		Date  string   `json:"date"`
 		News  []string `json:"news"`
@@ -27,14 +27,14 @@ type resVo struct {
 	} `json:"data"`
 }

-func (f *FuncZaoBao) Fetch() (string, error) {
+func (f FuncZaoBao) Invoke(...interface{}) (string, error) {

 	url := fmt.Sprintf("%s?format=json&token=%s", f.apiURL, f.token)
 	bytes, err := utils.HttpGet(url, "")
 	if err != nil {
 		return "", err
 	}
-	var res resVo
+	var res ZaoBaoVo
 	err = utils.JsonDecode(string(bytes), &res)
 	if err != nil {
 		return "", err
@@ -49,3 +49,7 @@ func (f *FuncZaoBao) Fetch() (string, error) {
 	builder = append(builder, fmt.Sprintf("%s", res.Data.WeiYu))
 	return strings.Join(builder, "\n\n"), nil
 }
+
+func (f FuncZaoBao) Name() string {
+	return f.name
+}
--- a/api/store/model/chat_history.go
+++ b/api/store/model/chat_history.go
@@ -2,13 +2,14 @@ package model

 type HistoryMessage struct {
 	BaseModel
-	ChatId  string // 会话 ID
-	UserId  uint   // 用户 ID
-	RoleId  uint   // 角色 ID
-	Type    string
-	Icon    string
-	Tokens  int
-	Content string
+	ChatId     string // 会话 ID
+	UserId     uint   // 用户 ID
+	RoleId     uint   // 角色 ID
+	Type       string
+	Icon       string
+	Tokens     int
+	Content    string
+	UseContext bool // 是否可以作为聊天上下文
 }

 func (HistoryMessage) TableName() string {
--- a/api/store/model/reward.go
+++ b/api/store/model/reward.go
@@ -0,0 +1,11 @@
+package model
+
+// 用户打赏
+
+type Reward struct {
+	BaseModel
+	TxId   string  // 交易ID
+	Amount float64 // 打赏金额
+	Remark string  // 打赏备注
+	Status bool    // 核销状态
+}
--- a/api/store/vo/chat_history.go
+++ b/api/store/vo/chat_history.go
@@ -2,13 +2,14 @@ package vo

 type HistoryMessage struct {
 	BaseVo
-	ChatId  string `json:"chat_id"`
-	UserId  uint   `json:"user_id"`
-	RoleId  uint   `json:"role_id"`
-	Type    string `json:"type"`
-	Icon    string `json:"icon"`
-	Tokens  int    `json:"tokens"`
-	Content string `json:"content"`
+	ChatId     string `json:"chat_id"`
+	UserId     uint   `json:"user_id"`
+	RoleId     uint   `json:"role_id"`
+	Type       string `json:"type"`
+	Icon       string `json:"icon"`
+	Tokens     int    `json:"tokens"`
+	Content    string `json:"content"`
+	UseContext bool   `json:"use_context"`
 }

 func (HistoryMessage) TableName() string {
--- a/api/utils/strings.go
+++ b/api/utils/strings.go
@@ -81,3 +81,10 @@ func JsonEncode(value interface{}) string {
 func JsonDecode(src string, dest interface{}) error {
 	return json.Unmarshal([]byte(src), dest)
 }
+
+func InterfaceToString(value interface{}) string {
+	if str, ok := value.(string); ok {
+		return str
+	}
+	return JsonEncode(value)
+}
--- a/database/plugins.sql
+++ b/database/plugins.sql
@@ -0,0 +1 @@
+ALTER TABLE `chatgpt_chat_history` ADD `use_context` TINYINT(1) NOT NULL COMMENT '是否允许作为上下文语料' AFTER `tokens`;
--- a/database/reward.sql
+++ b/database/reward.sql
@@ -0,0 +1,19 @@
+CREATE TABLE `chatgpt_rewards` (
+                                  `id` int NOT NULL,
+                                  `tx_id` char(36) NOT NULL COMMENT '交易 ID',
+                                  `amount` decimal(10,2) NOT NULL COMMENT '打赏金额',
+                                  `remark` varchar(80) NOT NULL COMMENT '备注',
+                                  `status` tinyint(1) NOT NULL COMMENT '核销状态，0：未核销，1：已核销',
+                                  `created_at` datetime NOT NULL,
+                                  `updated_at` datetime NOT NULL
+) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci COMMENT='用户打赏';
+
+ALTER TABLE `chatgpt_rewards`
+    ADD PRIMARY KEY (`id`),
+  ADD UNIQUE KEY `tx_id` (`tx_id`);
+
+
+ALTER TABLE `chatgpt_rewards`
+    MODIFY `id` int NOT NULL AUTO_INCREMENT;
+
+update chatgpt_users set calls=0
--- a/docker/build.sh
+++ b/docker/build.sh
@@ -2,11 +2,11 @@

 version=$1
 # build go api
-cd ../api/go
+cd ../api
 make clean linux

 # build web app
-cd ../../web
+cd ../web
 npm run build

 cd ../docker
--- a/docker/dockerfile-api-go
+++ b/docker/dockerfile-api-go
@@ -4,7 +4,7 @@ FROM registry.cn-hangzhou.aliyuncs.com/geekmaster/ubuntu-ca:22.04
 MAINTAINER yangjian<yangjian102621@163.com>

 WORKDIR /var/www/app
-COPY ./api/go/bin/chatgpt-v3-amd64-linux /var/www/app
+COPY ./api/bin/chatgpt-v3-amd64-linux /var/www/app

 EXPOSE 5678

--- a/web/public/images/reward.png
+++ b/web/public/images/reward.png
--- a/web/src/components/RewardVerify.vue
+++ b/web/src/components/RewardVerify.vue
@@ -0,0 +1,75 @@
+<template>
+  <el-dialog
+      v-model="showDialog"
+      :close-on-click-modal="false"
+      :show-close="mobile !== ''"
+      :before-close="close"
+      :title="title"
+  >
+    <div class="form" id="bind-mobile-form">
+      <el-alert v-if="mobile !== ''" type="info" show-icon :closable="false" style="margin-bottom: 20px;">
+        <p>请输入您参与众筹的微信支付转账单号兑换相应的对话次数。</p>
+      </el-alert>
+
+      <el-form :model="form" label-width="120px">
+        <el-form-item label="转账单号">
+          <el-input v-model="form.tx_id"/>
+        </el-form-item>
+      </el-form>
+    </div>
+
+    <template #footer>
+      <span class="dialog-footer">
+        <el-button type="primary" @click="save">
+          确认核销
+        </el-button>
+      </span>
+    </template>
+  </el-dialog>
+</template>
+
+<script setup>
+import {computed, ref} from "vue";
+import {ElMessage} from "element-plus";
+import {httpPost} from "@/utils/http";
+
+const props = defineProps({
+  show: Boolean,
+  mobile: String
+});
+
+const showDialog = computed(() => {
+  return props.show
+})
+
+const title = ref('众筹码核销')
+const form = ref({
+  tx_id: '',
+})
+
+const emits = defineEmits(['hide']);
+
+const save = () => {
+  if (form.value.tx_id === '') {
+    return ElMessage.error({message: "请输入微信支付转账单号"});
+  }
+
+  httpPost('/api/reward/verify', form.value).then(() => {
+    ElMessage.success({
+      message: '核销成功',
+      duration: 1000,
+      onClose: () => emits('hide', false)
+    })
+  }).catch(e => {
+    ElMessage.error({message: "核销失败：" + e.message});
+  })
+}
+
+const close = function () {
+  emits('hide', false);
+}
+</script>
+
+<style scoped>
+
+</style>
--- a/web/src/views/ChatPlus.vue
+++ b/web/src/views/ChatPlus.vue
@@ -67,6 +67,20 @@
                  <span>绑定手机号</span>
                </el-dropdown-item>

+                <el-dropdown-item @click="showRewardDialog = true">
+                  <el-icon>
+                    <Present/>
+                  </el-icon>
+                  <span>加入众筹</span>
+                </el-dropdown-item>
+
+                <el-dropdown-item @click="showRewardVerifyDialog = true">
+                  <el-icon>
+                    <Checked/>
+                  </el-icon>
+                  <span>众筹核销</span>
+                </el-dropdown-item>
+
                <el-dropdown-item @click="clearAllChats">
                  <el-icon>
                    <Delete/>
@@ -94,6 +108,7 @@
      <el-main v-loading="loading" element-loading-background="rgba(122, 122, 122, 0.3)">
        <div class="chat-head">
          <div class="chat-config">
+            <span class="role-select-label">聊天角色：</span>
            <el-select v-model="roleId" filterable placeholder="角色" class="role-select">
              <el-option
                  v-for="item in roles"
@@ -197,6 +212,26 @@

    <bind-mobile v-if="isLogin" :show="showBindMobileDialog" :mobile="loginUser.mobile"
                 @hide="showBindMobileDialog = false"/>
+
+    <reward-verify v-if="isLogin" :show="showRewardVerifyDialog" @hide="showRewardVerifyDialog = false"/>
+
+    <el-dialog
+        v-model="showRewardDialog"
+        :show-close="true"
+        custom-class="donate-dialog"
+        width="400px"
+        top="5vh"
+        title="参与众筹"
+    >
+      <el-alert type="info" :closable="false">
+        <p>您好，ChatGPT-Plus 项目目前已经运行了快半年了，一直免费给大家使用的。然而免费服务始终难以维持，服务器即将到期，免费的
+          API KEY 也全部用完了，因此我们准备开启众筹模式，只需要打赏9.9元，就可以兑换 100 次对话，以此来覆盖我们的 OpenAI
+          账单和服务器的费用。</p>
+      </el-alert>
+      <p>
+        <el-image :src="rewardImg"/>
+      </p>
+    </el-dialog>
  </div>


@@ -208,10 +243,13 @@ import ChatReply from "@/components/ChatReply.vue";
 import {
  ArrowDown,
  Check,
+  Checked,
  Close,
  Delete,
-  Edit, Iphone,
+  Edit,
+  Iphone,
  Plus,
+  Present,
  Promotion,
  RefreshRight,
  Search,
@@ -230,9 +268,11 @@ import ConfigDialog from "@/components/ConfigDialog.vue";
 import PasswordDialog from "@/components/PasswordDialog.vue";
 import {checkSession} from "@/action/session";
 import BindMobile from "@/components/BindMobile.vue";
+import RewardVerify from "@/components/RewardVerify.vue";

 const title = ref('ChatGPT-智能助手');
 const logo = 'images/logo.png';
+const rewardImg = ref('images/reward.png')
 const models = ref([])
 const model = ref('gpt-3.5-turbo')
 const chatData = ref([]);
@@ -251,6 +291,8 @@ const router = useRouter();
 const showConfigDialog = ref(false);
 const showPasswordDialog = ref(false);
 const showBindMobileDialog = ref(false);
+const showRewardDialog = ref(false);
+const showRewardVerifyDialog = ref(false);
 const isLogin = ref(false)

 if (isMobile()) {
@@ -920,6 +962,10 @@ $borderColor = #4676d0;
          justify-content center;
          padding-top 10px;

+          .role-select-label {
+            color #ffffff
+          }
+
          .el-select {
            //max-width 150px;
            margin-right 10px;
--- a/web/src/views/admin/SysConfig.vue
+++ b/web/src/views/admin/SysConfig.vue
@@ -65,6 +65,14 @@
          <el-switch v-model="chat['enable_history']"/>
        </el-form-item>

+        <el-alert type="info" show-icon :closable="false">
+          <p>会话上下文深度：在老会话中继续会话，默认加载多少条聊天记录作为上下文。如果设置为 0
+            则不加载聊天记录，仅仅使用当前角色的上下文。该配置参数最好设置为 2 的整数倍。</p>
+        </el-alert>
+        <el-form-item label="会话上下文深度">
+          <el-input-number v-model="chat['context_deep']" :min="0" :max="10"/>
+        </el-form-item>
+
        <el-form-item style="text-align: right">
          <el-button type="primary" @click="save('chat')">保存</el-button>
        </el-form-item>
@@ -178,6 +186,10 @@ const addModel = function () {
            font-size 16px;
          }
        }
+
+        .tip-text {
+          padding-left 10px;
+        }
      }
    }
Author	SHA1	Message	Date
RockYang	e02badf7bb	feat: reward verify is ready	2023-07-21 22:29:14 +08:00
RockYang	dd88622c64	feat: 增加打赏核销功能	2023-07-21 18:26:51 +08:00
RockYang	c4d7126c4d	feat: 集成微信收款服务	2023-07-20 17:46:32 +08:00
RockYang	86bc063941	put logs in log file	2023-07-20 16:36:12 +08:00
RockYang	dce85eb519	fix: 修复阿里云短信发送失败的 bug	2023-07-16 16:05:11 +08:00
RockYang	4ab879d697	fix: 用户使用自己绑定的 API 调用，则不计算 calls 和 token 消耗	2023-07-16 11:55:45 +08:00
RockYang	681e52df50	Merge branch 'main' into prod	2023-07-16 09:57:27 +08:00
RockYang	fb554c0315	chore: 更新项目打包脚本	2023-07-16 09:55:00 +08:00
RockYang	accf8eeb77	feat: plugin function is ready	2023-07-15 21:52:30 +08:00
RockYang	3e41edd3b5	feat: 支持上下文深度配置，计算每轮对话消耗的总 token 数量	2023-07-15 18:37:25 +08:00
RockYang	9126cfff20	refactor: 更改 OpenAI 请求 Body 数据结构，兼容函数调用请求	2023-07-15 18:00:40 +08:00
				`@@ -0,0 +1 @@`
				ALTER TABLE `chatgpt_chat_history` ADD `use_context` TINYINT(1) NOT NULL COMMENT '是否允许作为上下文语料' AFTER `tokens`;