merge v4.2.2

2026-02-11 08:54:30 +08:00 · 2025-12-02 11:18:30 +08:00
parent 172807d326 a1acca6f7a
commit c84754fc0c
69 changed files with 19149 additions and 2793 deletions
--- a/api/core/app_server.go
+++ b/api/core/app_server.go
@@ -73,6 +73,23 @@ func (s *AppServer) Run(db *gorm.DB) error {
 	if err != nil {
 		return fmt.Errorf("failed to decode system config: %v", err)
 	}
+	// 统计安装信息
+	go func() {
+		info, err := host.Info()
+		if err == nil {
+			apiURL := fmt.Sprintf("%s/%s", s.Config.ApiConfig.ApiURL, "api/installs/push")
+			timestamp := time.Now().Unix()
+			product := "geekai-plus"
+			signStr := fmt.Sprintf("%s#%s#%d", product, info.HostID, timestamp)
+			sign := utils.Sha256(signStr)
+			resp, err := req.C().R().SetBody(map[string]interface{}{"product": product, "device_id": info.HostID, "timestamp": timestamp, "sign": sign}).Post(apiURL)
+			if err != nil {
+				logger.Errorf("register install info failed: %v", err)
+			} else {
+				logger.Debugf("register install info success: %v", resp.String())
+			}
+		}
+	}()
 	logger.Infof("http://%s", s.Config.Listen)

 	// 统计安装信息
@@ -115,19 +132,23 @@ func corsMiddleware() gin.HandlerFunc {
 	return func(c *gin.Context) {
 		method := c.Request.Method
 		origin := c.Request.Header.Get("Origin")
+		
+		// 设置允许的请求源
 		if origin != "" {
-			// 设置允许的请求源
 			c.Header("Access-Control-Allow-Origin", origin)
-			c.Header("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE, UPDATE")
-			//允许跨域设置可以返回其他子段，可以自定义字段
-			c.Header("Access-Control-Allow-Headers", "Authorization, Body-Length, Body-Type, Admin-Authorization,content-type")
-			// 允许浏览器（客户端）可以解析的头部 （重要）
-			c.Header("Access-Control-Expose-Headers", "Body-Length, Access-Control-Allow-Origin, Access-Control-Allow-Headers")
-			//设置缓存时间
-			c.Header("Access-Control-Max-Age", "172800")
-			//允许客户端传递校验信息比如 cookie (重要)
-			c.Header("Access-Control-Allow-Credentials", "true")
+		} else {
+			c.Header("Access-Control-Allow-Origin", "*")
 		}
+		
+		c.Header("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE, UPDATE")
+		//允许跨域设置可以返回其他子段，可以自定义字段
+		c.Header("Access-Control-Allow-Headers", "Authorization, Body-Length, Body-Type, Admin-Authorization,content-type")
+		// 允许浏览器（客户端）可以解析的头部 （重要）
+		c.Header("Access-Control-Expose-Headers", "Body-Length, Access-Control-Allow-Origin, Access-Control-Allow-Headers")
+		//设置缓存时间
+		c.Header("Access-Control-Max-Age", "172800")
+		//允许客户端传递校验信息比如 cookie (重要)
+		c.Header("Access-Control-Allow-Credentials", "true")

 		if method == http.MethodOptions {
 			c.JSON(http.StatusOK, "ok!")
--- a/api/core/types/chat.go
+++ b/api/core/types/chat.go
@@ -9,20 +9,20 @@ package types

 // ApiRequest API 请求实体
 type ApiRequest struct {
-	Model               string        `json:"model,omitempty"`
-	Temperature         float32       `json:"temperature"`
-	MaxTokens           int           `json:"max_tokens,omitempty"`
-	MaxCompletionTokens int           `json:"max_completion_tokens,omitempty"` // 兼容GPT O1 模型
-	Stream              bool          `json:"stream,omitempty"`
-	Messages            []interface{} `json:"messages,omitempty"`
-	Tools               []Tool        `json:"tools,omitempty"`
-	Functions           []interface{} `json:"functions,omitempty"`       // 兼容中转平台
-	ResponseFormat      interface{}   `json:"response_format,omitempty"` // 响应格式
+	Model               string  `json:"model,omitempty"`
+	Temperature         float32 `json:"temperature"`
+	MaxTokens           int     `json:"max_tokens,omitempty"`
+	MaxCompletionTokens int     `json:"max_completion_tokens,omitempty"` // 兼容GPT O1 模型
+	Stream              bool    `json:"stream,omitempty"`
+	Messages            []any   `json:"messages,omitempty"`
+	Tools               []Tool  `json:"tools,omitempty"`
+	Functions           []any   `json:"functions,omitempty"`       // 兼容中转平台
+	ResponseFormat      any     `json:"response_format,omitempty"` // 响应格式

 	ToolChoice string `json:"tool_choice,omitempty"`

-	Input      map[string]interface{} `json:"input,omitempty"`      //兼容阿里通义千问
-	Parameters map[string]interface{} `json:"parameters,omitempty"` //兼容阿里通义千问
+	Input      map[string]any `json:"input,omitempty"`      //兼容阿里通义千问
+	Parameters map[string]any `json:"parameters,omitempty"` //兼容阿里通义千问
 }

 type Message struct {
@@ -41,11 +41,12 @@ type ChoiceItem struct {
 }

 type Delta struct {
-	Role         string      `json:"role"`
-	Name         string      `json:"name"`
-	Content      interface{} `json:"content"`
-	ToolCalls    []ToolCall  `json:"tool_calls,omitempty"`
-	FunctionCall struct {
+	Role             string     `json:"role"`
+	Name             string     `json:"name"`
+	Content          any        `json:"content"`
+	ReasoningContent string     `json:"reasoning_content,omitempty"`
+	ToolCalls        []ToolCall `json:"tool_calls,omitempty"`
+	FunctionCall     struct {
 		Name      string `json:"name,omitempty"`
 		Arguments string `json:"arguments,omitempty"`
 	} `json:"function_call,omitempty"`
--- a/api/core/types/locked_map.go
+++ b/api/core/types/locked_map.go
@@ -16,7 +16,7 @@ type MKey interface {
 	string | int | uint
 }
 type MValue interface {
-	*WsClient | *ChatSession | context.CancelFunc | []interface{}
+	*WsClient | *ChatSession | context.CancelFunc | []any
 }
 type LMap[K MKey, T MValue] struct {
 	lock sync.RWMutex
--- a/api/go.mod
+++ b/api/go.mod
@@ -27,8 +27,10 @@ require github.com/xxl-job/xxl-job-executor-go v1.2.0

 require (
 	github.com/go-pay/gopay v1.5.101
+	github.com/go-rod/rod v0.116.2
 	github.com/google/go-tika v0.3.1
 	github.com/microcosm-cc/bluemonday v1.0.26
+	github.com/sashabaranov/go-openai v1.38.1
 	github.com/shirou/gopsutil v3.21.11+incompatible
 	github.com/shopspring/decimal v1.3.1
 	github.com/syndtr/goleveldb v1.0.0
@@ -45,13 +47,13 @@ require (
 	github.com/go-pay/xtime v0.0.2 // indirect
 	github.com/golang/snappy v0.0.0-20180518054509-2e65f85255db // indirect
 	github.com/gorilla/css v1.0.0 // indirect
-	github.com/gravityblast/fresh v0.0.0-20240621171608-8d1fef547a99 // indirect
-	github.com/howeyc/fsnotify v0.9.0 // indirect
-	github.com/mattn/go-colorable v0.1.13 // indirect
-	github.com/pilu/config v0.0.0-20131214182432-3eb99e6c0b9a // indirect
-	github.com/pilu/fresh v0.0.0-20240621171608-8d1fef547a99 // indirect
 	github.com/tklauser/go-sysconf v0.3.13 // indirect
 	github.com/tklauser/numcpus v0.7.0 // indirect
+	github.com/ysmood/fetchup v0.3.0 // indirect
+	github.com/ysmood/goob v0.4.0 // indirect
+	github.com/ysmood/got v0.40.0 // indirect
+	github.com/ysmood/gson v0.7.3 // indirect
+	github.com/ysmood/leakless v0.9.0 // indirect
 	github.com/yusufpapurcu/wmi v1.2.4 // indirect
 	go.uber.org/mock v0.4.0 // indirect
 )
--- a/api/go.sum
+++ b/api/go.sum
@@ -73,6 +73,8 @@ github.com/go-playground/validator/v10 v10.14.0 h1:vgvQWe3XCz3gIeFDm/HnTIbj6UGmg
 github.com/go-playground/validator/v10 v10.14.0/go.mod h1:9iXMNT7sEkjXb0I+enO7QXmzG6QCsPWY4zveKFVRSyU=
 github.com/go-redis/redis/v8 v8.11.5 h1:AcZZR7igkdvfVmQTPnu9WE37LRrO/YrBH5zWyjDC0oI=
 github.com/go-redis/redis/v8 v8.11.5/go.mod h1:gREzHqY1hg6oD9ngVRbLStwAWKhA0FEgq8Jd4h5lpwo=
+github.com/go-rod/rod v0.116.2 h1:A5t2Ky2A+5eD/ZJQr1EfsQSe5rms5Xof/qj296e+ZqA=
+github.com/go-rod/rod v0.116.2/go.mod h1:H+CMO9SCNc2TJ2WfrG+pKhITz57uGNYU43qYHh438Mg=
 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/go-task/slim-sprig v0.0.0-20230315185526-52ccab3ef572 h1:tfuBGBXKqDEevZMzYi5KSi8KkcZtzBcTgAUUtapy0OI=
@@ -100,15 +102,11 @@ github.com/gorilla/css v1.0.0 h1:BQqNyPTi50JCFMTw/b67hByjMVXZRwGha6wxVGkeihY=
 github.com/gorilla/css v1.0.0/go.mod h1:Dn721qIggHpt4+EFCcTLTU/vk5ySda2ReITrtgBl60c=
 github.com/gorilla/websocket v1.5.0 h1:PPwGk2jz7EePpoHN/+ClbZu8SPxiqlu12wZP/3sWmnc=
 github.com/gorilla/websocket v1.5.0/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
-github.com/gravityblast/fresh v0.0.0-20240621171608-8d1fef547a99 h1:A6qlLfihaWef15viqtecCz4XknZcgjgD7mEuhu7bHEc=
-github.com/gravityblast/fresh v0.0.0-20240621171608-8d1fef547a99/go.mod h1:ukFDwXV66bGV7JnfyxFKuKiVp4zH4orBKXML+VCSrhI=
 github.com/hashicorp/errwrap v1.0.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4=
 github.com/hashicorp/errwrap v1.1.0 h1:OxrOeh75EUXMY8TBjag2fzXGZ40LB6IKw45YeGUDY2I=
 github.com/hashicorp/errwrap v1.1.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4=
 github.com/hashicorp/go-multierror v1.1.1 h1:H5DkEtf6CXdFp0N0Em5UCwQpXMWke8IA0+lD48awMYo=
 github.com/hashicorp/go-multierror v1.1.1/go.mod h1:iw975J/qwKPdAO1clOe2L8331t/9/fmwbPZ6JB6eMoM=
-github.com/howeyc/fsnotify v0.9.0 h1:0gtV5JmOKH4A8SsFxG2BczSeXWWPvcMT0euZt5gDAxY=
-github.com/howeyc/fsnotify v0.9.0/go.mod h1:41HzSPxBGeFRQKEEwgh49TRw/nKBsYZ2cF1OzPjSJsA=
 github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpOxQnU=
 github.com/imroc/req/v3 v3.37.2 h1:vEemuA0cq9zJ6lhe+mSRhsZm951bT0CdiSH47+KTn6I=
 github.com/imroc/req/v3 v3.37.2/go.mod h1:DECzjVIrj6jcUr5n6e+z0ygmCO93rx4Jy0RjOEe1YCI=
@@ -141,9 +139,6 @@ 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-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA=
-github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=
-github.com/mattn/go-isatty v0.0.16/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/microcosm-cc/bluemonday v1.0.26 h1:xbqSvqzQMeEHCqMi64VAs4d8uy6Mequs3rQ0k/Khz58=
@@ -177,10 +172,6 @@ github.com/opentracing/opentracing-go v1.2.1-0.20220228012449-10b1cf09e00b h1:Ff
 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.8 h1:0ctb6s9mE31h0/lhu+J6OPmVeDxJn+kYnJc2jZR9tGQ=
 github.com/pelletier/go-toml/v2 v2.0.8/go.mod h1:vuYfssBdrU2XDZ9bYydBu6t+6a6PYNcZljzZR9VXg+4=
-github.com/pilu/config v0.0.0-20131214182432-3eb99e6c0b9a h1:Tg4E4cXPZSZyd3H1tJlYo6ZreXV0ZJvE/lorNqyw1AU=
-github.com/pilu/config v0.0.0-20131214182432-3eb99e6c0b9a/go.mod h1:9Or9aIl95Kp43zONcHd5tLZGKXb9iLx0pZjau0uJ5zg=
-github.com/pilu/fresh v0.0.0-20240621171608-8d1fef547a99 h1:+X7Gb40b5Bl3v5+3MiGK8Jhemjp65MHc+nkVCfq1Yfc=
-github.com/pilu/fresh v0.0.0-20240621171608-8d1fef547a99/go.mod h1:2LLTtftTZSdAPR/iVyennXZDLZOYzyDn+T0qEKJ8eSw=
 github.com/pkg/diff v0.0.0-20210226163009-20ebb0f2a09e/go.mod h1:pJLUxLENpZxwdsKMEsNbx1VGcRFpLqf3715MtcvvzbA=
 github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
 github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
@@ -203,6 +194,8 @@ github.com/rogpeppe/go-internal v1.8.0 h1:FCbCCtXNOY3UtUuHUYaghJg4y7Fd14rXifAYUA
 github.com/rogpeppe/go-internal v1.8.0/go.mod h1:WmiCO8CzOY8rg0OYDC4/i/2WRWAB6poM+XZ2dLUbcbE=
 github.com/rs/xid v1.5.0 h1:mKX4bl4iPYJtEIxp6CYiUuLQ/8DYMoz0PUdtGgMFRVc=
 github.com/rs/xid v1.5.0/go.mod h1:trrq9SKmegXys3aeAKXMUTdJsYXVwGY3RLcfgqegfbg=
+github.com/sashabaranov/go-openai v1.38.1 h1:TtZabbFQZa1nEni/IhVtDF/WQjVqDgd+cWR5OeddzF8=
+github.com/sashabaranov/go-openai v1.38.1/go.mod h1:lj5b/K+zjTSFxVLijLSTDZuP7adOgerWeFyZLUhAKRg=
 github.com/shirou/gopsutil v3.21.11+incompatible h1:+1+c1VGhc88SSonWP6foOcLhvnKlUeu/erjjvaPEYiI=
 github.com/shirou/gopsutil v3.21.11+incompatible/go.mod h1:5b4v6he4MtMOwMlS0TUMTu2PcXUg8+E1lC7eC3UO/RA=
 github.com/shopspring/decimal v1.3.1 h1:2Usl1nmF/WZucqkFZhnfFYxxxu8LG21F6nPQBE5gKV8=
@@ -239,6 +232,20 @@ github.com/ugorji/go/codec v1.2.11 h1:BMaWp1Bb6fHwEtbplGBGJ498wD+LKlNSl25MjdZY4d
 github.com/ugorji/go/codec v1.2.11/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg=
 github.com/xxl-job/xxl-job-executor-go v1.2.0 h1:MTl2DpwrK2+hNjRRks2k7vB3oy+3onqm9OaSarneeLQ=
 github.com/xxl-job/xxl-job-executor-go v1.2.0/go.mod h1:bUFhz/5Irp9zkdYk5MxhQcDDT6LlZrI8+rv5mHtQ1mo=
+github.com/ysmood/fetchup v0.3.0 h1:UhYz9xnLEVn2ukSuK3KCgcznWpHMdrmbsPpllcylyu8=
+github.com/ysmood/fetchup v0.3.0/go.mod h1:hbysoq65PXL0NQeNzUczNYIKpwpkwFL4LXMDEvIQq9A=
+github.com/ysmood/goob v0.4.0 h1:HsxXhyLBeGzWXnqVKtmT9qM7EuVs/XOgkX7T6r1o1AQ=
+github.com/ysmood/goob v0.4.0/go.mod h1:u6yx7ZhS4Exf2MwciFr6nIM8knHQIE22lFpWHnfql18=
+github.com/ysmood/gop v0.2.0 h1:+tFrG0TWPxT6p9ZaZs+VY+opCvHU8/3Fk6BaNv6kqKg=
+github.com/ysmood/gop v0.2.0/go.mod h1:rr5z2z27oGEbyB787hpEcx4ab8cCiPnKxn0SUHt6xzk=
+github.com/ysmood/got v0.40.0 h1:ZQk1B55zIvS7zflRrkGfPDrPG3d7+JOza1ZkNxcc74Q=
+github.com/ysmood/got v0.40.0/go.mod h1:W7DdpuX6skL3NszLmAsC5hT7JAhuLZhByVzHTq874Qg=
+github.com/ysmood/gotrace v0.6.0 h1:SyI1d4jclswLhg7SWTL6os3L1WOKeNn/ZtzVQF8QmdY=
+github.com/ysmood/gotrace v0.6.0/go.mod h1:TzhIG7nHDry5//eYZDYcTzuJLYQIkykJzCRIo4/dzQM=
+github.com/ysmood/gson v0.7.3 h1:QFkWbTH8MxyUTKPkVWAENJhxqdBa4lYTQWqZCiLG6kE=
+github.com/ysmood/gson v0.7.3/go.mod h1:3Kzs5zDl21g5F/BlLTNcuAGAYLKt2lV5G8D1zF3RNmg=
+github.com/ysmood/leakless v0.9.0 h1:qxCG5VirSBvmi3uynXFkcnLMzkphdh3xx5FtrORwDCU=
+github.com/ysmood/leakless v0.9.0/go.mod h1:R8iAXPRaG97QJwqxs74RdwzcRHT1SWCGTNqY8q0JvMQ=
 github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
 github.com/yusufpapurcu/wmi v1.2.4 h1:zFUKzehAFReQwLys1b/iSMl+JQGSCSjtVqQn9bBrPo0=
 github.com/yusufpapurcu/wmi v1.2.4/go.mod h1:SBZ9tNy3G9/m5Oi98Zks0QjeHVDvuK0qfxQmPyzfmi0=
@@ -302,7 +309,6 @@ golang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c/go.mod h1:oPkhp1MJrh7nUepCBc
 golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
-golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.1.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
--- a/api/handler/admin/chat_model_handler.go
+++ b/api/handler/admin/chat_model_handler.go
@@ -30,20 +30,21 @@ func NewChatModelHandler(app *core.AppServer, db *gorm.DB) *ChatModelHandler {

 func (h *ChatModelHandler) Save(c *gin.Context) {
 	var data struct {
-		Id          uint    `json:"id"`
-		Name        string  `json:"name"`
-		Value       string  `json:"value"`
-		Enabled     bool    `json:"enabled"`
-		SortNum     int     `json:"sort_num"`
-		Open        bool    `json:"open"`
-		Platform    string  `json:"platform"`
-		Power       int     `json:"power"`
-		MaxTokens   int     `json:"max_tokens"`  // 最大响应长度
-		MaxContext  int     `json:"max_context"` // 最大上下文长度
-		Temperature float32 `json:"temperature"` // 模型温度
-		KeyId       int     `json:"key_id,omitempty"`
-		CreatedAt   int64   `json:"created_at"`
-		Type        string  `json:"type"`
+		Id          uint              `json:"id"`
+		Name        string            `json:"name"`
+		Value       string            `json:"value"`
+		Enabled     bool              `json:"enabled"`
+		SortNum     int               `json:"sort_num"`
+		Open        bool              `json:"open"`
+		Platform    string            `json:"platform"`
+		Power       int               `json:"power"`
+		MaxTokens   int               `json:"max_tokens"`  // 最大响应长度
+		MaxContext  int               `json:"max_context"` // 最大上下文长度
+		Temperature float32           `json:"temperature"` // 模型温度
+		KeyId       int               `json:"key_id,omitempty"`
+		CreatedAt   int64             `json:"created_at"`
+		Type        string            `json:"type"`
+		Options     map[string]string `json:"options"`
 	}
 	if err := c.ShouldBindJSON(&data); err != nil {
 		resp.ERROR(c, types.InvalidArgs)
@@ -59,7 +60,6 @@ func (h *ChatModelHandler) Save(c *gin.Context) {
 	item.Name = data.Name
 	item.Value = data.Value
 	item.Enabled = data.Enabled
-	item.SortNum = data.SortNum
 	item.Open = data.Open
 	item.Power = data.Power
 	item.MaxTokens = data.MaxTokens
@@ -67,6 +67,7 @@ func (h *ChatModelHandler) Save(c *gin.Context) {
 	item.Temperature = data.Temperature
 	item.KeyId = data.KeyId
 	item.Type = data.Type
+	item.Options = utils.JsonEncode(data.Options)
 	var res *gorm.DB
 	if data.Id > 0 {
 		res = h.DB.Save(&item)
--- a/api/handler/admin/config_handler.go
+++ b/api/handler/admin/config_handler.go
@@ -59,6 +59,12 @@ func (h *ConfigHandler) Update(c *gin.Context) {
 		return
 	}

+	// 如果要启用图形验证码功能，则检查是否配置了 API 服务
+	if data.Config.EnabledVerify && h.App.Config.ApiConfig.AppId == "" {
+		resp.ERROR(c, "启用验证码服务需要先配置 GeekAI 官方 API 服务 AppId 和 Token")
+		return
+	}
+
 	value := utils.JsonEncode(&data.Config)
 	config := model.Config{Key: data.Key, Config: value}
 	res := h.DB.FirstOrCreate(&config, model.Config{Key: data.Key})
--- a/api/handler/admin/user_handler.go
+++ b/api/handler/admin/user_handler.go
@@ -20,6 +20,7 @@ import (
 	"time"

 	"github.com/go-redis/redis/v8"
+	"github.com/golang-jwt/jwt/v5"

 	"github.com/gin-gonic/gin"
 	"gorm.io/gorm"
@@ -320,3 +321,36 @@ func (h *UserHandler) LoginLog(c *gin.Context) {

 	resp.SUCCESS(c, vo.NewPage(total, page, pageSize, logs))
 }
+
+// GenLoginLink 生成登录链接
+func (h *UserHandler) GenLoginLink(c *gin.Context) {
+	id := c.Query("id")
+	if id == "" {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+	var user model.User
+	if err := h.DB.Where("id = ?", id).First(&user).Error; err != nil {
+		resp.ERROR(c, "用户不存在")
+		return
+	}
+
+	// 创建 token
+	token := jwt.NewWithClaims(jwt.SigningMethodHS256, jwt.MapClaims{
+		"user_id": user.Id,
+		"expired": time.Now().Add(time.Second * time.Duration(h.App.Config.Session.MaxAge)).Unix(),
+	})
+	tokenString, err := token.SignedString([]byte(h.App.Config.Session.SecretKey))
+	if err != nil {
+		resp.ERROR(c, "Failed to generate token, "+err.Error())
+		return
+	}
+	// 保存到 redis
+	sessionKey := fmt.Sprintf("users/%d", user.Id)
+	if _, err = h.redis.Set(c, sessionKey, tokenString, 0).Result(); err != nil {
+		resp.ERROR(c, "error with save token: "+err.Error())
+		return
+	}
+
+	resp.SUCCESS(c, tokenString)
+}
--- a/api/handler/chat_handler.go
+++ b/api/handler/chat_handler.go
@@ -22,15 +22,17 @@ import (
 	"geekai/utils"
 	"geekai/utils/resp"
 	"html/template"
+	"io"
 	"net/http"
 	"net/url"
-	"regexp"
+	"os"
 	"strings"
 	"time"
 	"unicode/utf8"

 	"github.com/gin-gonic/gin"
 	"github.com/go-redis/redis/v8"
+	"github.com/sashabaranov/go-openai"
 	"gorm.io/gorm"
 )

@@ -40,7 +42,7 @@ type ChatHandler struct {
 	uploadManager  *oss.UploaderManager
 	licenseService *service.LicenseService
 	ReqCancelFunc  *types.LMap[string, context.CancelFunc] // HttpClient 请求取消 handle function
-	ChatContexts   *types.LMap[string, []interface{}]      // 聊天上下文 Map [chatId] => []Message
+	ChatContexts   *types.LMap[string, []any]              // 聊天上下文 Map [chatId] => []Message
 	userService    *service.UserService
 }

@@ -51,7 +53,7 @@ func NewChatHandler(app *core.AppServer, db *gorm.DB, redis *redis.Client, manag
 		uploadManager:  manager,
 		licenseService: licenseService,
 		ReqCancelFunc:  types.NewLMap[string, context.CancelFunc](),
-		ChatContexts:   types.NewLMap[string, []interface{}](),
+		ChatContexts:   types.NewLMap[string, []any](),
 		userService:    userService,
 	}
 }
@@ -348,8 +350,14 @@ func (h *ChatHandler) doRequest(ctx context.Context, req types.ApiRequest, sessi
 		return nil, err
 	}
 	logger.Debugf("对话请求消息体：%+v", req)
-
-	apiURL := fmt.Sprintf("%s/v1/chat/completions", apiKey.ApiURL)
+	var apiURL string
+	p, _ := url.Parse(apiKey.ApiURL)
+	// 如果设置的是 BASE_URL 没有路径，则添加 /v1/chat/completions
+	if p.Path == "" {
+		apiURL = fmt.Sprintf("%s/v1/chat/completions", apiKey.ApiURL)
+	} else {
+		apiURL = apiKey.ApiURL
+	}
 	// 创建 HttpClient 请求对象
 	var client *http.Client
 	requestBody, err := json.Marshal(req)
@@ -495,28 +503,93 @@ func (h *ChatHandler) saveChatHistory(
 	}
 }

-// 将AI回复消息中生成的图片链接下载到本地
-func (h *ChatHandler) extractImgUrl(text string) string {
-	pattern := `!\[([^\]]*)]\(([^)]+)\)`
-	re := regexp.MustCompile(pattern)
-	matches := re.FindAllStringSubmatch(text, -1)
-
-	// 下载图片并替换链接地址
-	for _, match := range matches {
-		imageURL := match[2]
-		logger.Debug(imageURL)
-		// 对于相同地址的图片，已经被替换了，就不再重复下载了
-		if !strings.Contains(text, imageURL) {
-			continue
-		}
-
-		newImgURL, err := h.uploadManager.GetUploadHandler().PutUrlFile(imageURL, false)
-		if err != nil {
-			logger.Error("error with download image: ", err)
-			continue
-		}
-
-		text = strings.ReplaceAll(text, imageURL, newImgURL)
+// 文本生成语音
+func (h *ChatHandler) TextToSpeech(c *gin.Context) {
+	var data struct {
+		ModelId int    `json:"model_id"`
+		Text    string `json:"text"`
 	}
-	return text
+	if err := c.ShouldBindJSON(&data); err != nil {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+
+	textHash := utils.Sha256(fmt.Sprintf("%d/%s", data.ModelId, data.Text))
+	audioFile := fmt.Sprintf("%s/audio", h.App.Config.StaticDir)
+	if _, err := os.Stat(audioFile); err != nil {
+		os.MkdirAll(audioFile, 0755)
+	}
+	audioFile = fmt.Sprintf("%s/%s.mp3", audioFile, textHash)
+	if _, err := os.Stat(audioFile); err == nil {
+		// 设置响应头
+		c.Header("Content-Type", "audio/mpeg")
+		c.Header("Content-Disposition", "attachment; filename=speech.mp3")
+		c.File(audioFile)
+		return
+	}
+
+	// 查询模型
+	var chatModel model.ChatModel
+	err := h.DB.Where("id", data.ModelId).First(&chatModel).Error
+	if err != nil {
+		resp.ERROR(c, "找不到语音模型")
+		return
+	}
+
+	// 调用 DeepSeek 的 API 接口
+	var apiKey model.ApiKey
+	if chatModel.KeyId > 0 {
+		h.DB.Where("id", chatModel.KeyId).First(&apiKey)
+	}
+	if apiKey.Id == 0 {
+		h.DB.Where("type", "tts").Where("enabled", true).First(&apiKey)
+	}
+	if apiKey.Id == 0 {
+		resp.ERROR(c, "no TTS API key, please import key")
+		return
+	}
+
+	logger.Debugf("chatModel: %+v, apiKey: %+v", chatModel, apiKey)
+
+	// 调用 openai tts api
+	config := openai.DefaultConfig(apiKey.Value)
+	config.BaseURL = apiKey.ApiURL + "/v1"
+	client := openai.NewClientWithConfig(config)
+	voice := openai.VoiceAlloy
+	var options map[string]string
+	err = utils.JsonDecode(chatModel.Options, &options)
+	if err == nil {
+		voice = openai.SpeechVoice(options["voice"])
+	}
+	req := openai.CreateSpeechRequest{
+		Model: openai.SpeechModel(chatModel.Value),
+		Input: data.Text,
+		Voice: voice,
+	}
+
+	audioData, err := client.CreateSpeech(context.Background(), req)
+	if err != nil {
+		resp.ERROR(c, err.Error())
+		return
+	}
+
+	// 先将音频数据读取到内存
+	audioBytes, err := io.ReadAll(audioData)
+	if err != nil {
+		resp.ERROR(c, err.Error())
+		return
+	}
+
+	// 保存到音频文件
+	err = os.WriteFile(audioFile, audioBytes, 0644)
+	if err != nil {
+		logger.Error("failed to save audio file: ", err)
+	}
+
+	// 设置响应头
+	c.Header("Content-Type", "audio/mpeg")
+	c.Header("Content-Disposition", "attachment; filename=speech.mp3")
+
+	// 直接写入完整的音频数据到响应
+	c.Writer.Write(audioBytes)
 }
--- a/api/handler/chat_model_handler.go
+++ b/api/handler/chat_model_handler.go
@@ -30,14 +30,16 @@ func NewChatModelHandler(app *core.AppServer, db *gorm.DB) *ChatModelHandler {
 func (h *ChatModelHandler) List(c *gin.Context) {
 	var items []model.ChatModel
 	var chatModels = make([]vo.ChatModel, 0)
-	session := h.DB.Session(&gorm.Session{}).Where("type", "chat").Where("enabled", true)
+	session := h.DB.Session(&gorm.Session{}).Where("enabled", true)
 	t := c.Query("type")
 	if t != "" {
 		session = session.Where("type", t)
+	} else {
+		session = session.Where("type", "chat")
 	}

 	session = session.Where("open", true)
-	if h.IsLogin(c) {
+	if h.IsLogin(c) && t == "chat" {
 		user, _ := h.GetLoginUser(c)
 		var models []int
 		err := utils.JsonDecode(user.ChatModels, &models)
--- a/api/handler/chat_openai_handler.go
+++ b/api/handler/chat_openai_handler.go
@@ -89,6 +89,7 @@ func (h *ChatHandler) sendOpenAiMessage(
 		var function model.Function
 		var toolCall = false
 		var arguments = make([]string, 0)
+		var reasoning = false

 		scanner := bufio.NewScanner(response.Body)
 		for scanner.Scan() {
@@ -104,7 +105,9 @@ func (h *ChatHandler) sendOpenAiMessage(
 			if len(responseBody.Choices) == 0 { // Fixed: 兼容 Azure API 第一个输出空行
 				continue
 			}
-			if responseBody.Choices[0].Delta.Content == nil && responseBody.Choices[0].Delta.ToolCalls == nil {
+			if responseBody.Choices[0].Delta.Content == nil &&
+				responseBody.Choices[0].Delta.ToolCalls == nil &&
+				responseBody.Choices[0].Delta.ReasoningContent == "" {
 				continue
 			}

@@ -152,9 +155,25 @@ func (h *ChatHandler) sendOpenAiMessage(
 			if responseBody.Choices[0].FinishReason != "" {
 				break // 输出完成或者输出中断了
 			} else { // 正常输出结果
-				content := responseBody.Choices[0].Delta.Content
-				contents = append(contents, utils.InterfaceToString(content))
-				utils.SendChunkMsg(ws, content)
+				// 兼容思考过程
+				if responseBody.Choices[0].Delta.ReasoningContent != "" {
+					reasoningContent := responseBody.Choices[0].Delta.ReasoningContent
+					if !reasoning {
+						reasoningContent = fmt.Sprintf("<think>%s", reasoningContent)
+						reasoning = true
+					}
+
+					utils.SendChunkMsg(ws, reasoningContent)
+					contents = append(contents, reasoningContent)
+				} else if responseBody.Choices[0].Delta.Content != "" {
+					finalContent := responseBody.Choices[0].Delta.Content
+					if reasoning {
+						finalContent = fmt.Sprintf("</think>%s", responseBody.Choices[0].Delta.Content)
+						reasoning = false
+					}
+					contents = append(contents, utils.InterfaceToString(finalContent))
+					utils.SendChunkMsg(ws, finalContent)
+				}
 			}
 		} // end for

@@ -167,7 +186,7 @@ func (h *ChatHandler) sendOpenAiMessage(
 		}

 		if toolCall { // 调用函数完成任务
-			params := make(map[string]interface{})
+			params := make(map[string]any)
 			_ = utils.JsonDecode(strings.Join(arguments, ""), &params)
 			logger.Debugf("函数名称: %s, 函数参数：%s", function.Name, params)
 			params["user_id"] = userVo.Id
--- a/api/handler/function_handler.go
+++ b/api/handler/function_handler.go
@@ -13,6 +13,7 @@ import (
 	"geekai/core"
 	"geekai/core/types"
 	"geekai/service"
+	"geekai/service/crawler"
 	"geekai/service/dalle"
 	"geekai/service/oss"
 	"geekai/store/model"
@@ -252,6 +253,76 @@ func (h *FunctionHandler) Dall3(c *gin.Context) {
 	resp.SUCCESS(c, content)
 }

+// 实现一个联网搜索的函数工具，采用爬虫实现
+func (h *FunctionHandler) WebSearch(c *gin.Context) {
+	if err := h.checkAuth(c); err != nil {
+		resp.ERROR(c, err.Error())
+		return
+	}
+
+	var params map[string]interface{}
+	if err := c.ShouldBindJSON(&params); err != nil {
+		resp.ERROR(c, types.InvalidArgs)
+		return
+	}
+	
+	// 从参数中获取搜索关键词
+	keyword, ok := params["keyword"].(string)
+	if !ok || keyword == "" {
+		resp.ERROR(c, "搜索关键词不能为空")
+		return
+	}
+	
+	// 从参数中获取最大页数，默认为1页
+	maxPages := 1
+	if pages, ok := params["max_pages"].(float64); ok {
+		maxPages = int(pages)
+	}
+	
+	// 获取用户ID
+	userID, ok := params["user_id"].(float64)
+	if !ok {
+		resp.ERROR(c, "用户ID不能为空")
+		return
+	}
+	
+	// 查询用户信息
+	var user model.User
+	res := h.DB.Where("id = ?", int(userID)).First(&user)
+	if res.Error != nil {
+		resp.ERROR(c, "用户不存在")
+		return
+	}
+	
+	// 检查用户算力是否足够
+	searchPower := 1 // 每次搜索消耗1点算力
+	if user.Power < searchPower {
+		resp.ERROR(c, "算力不足，无法执行网络搜索")
+		return
+	}
+	
+	// 执行网络搜索
+	searchResults, err := crawler.SearchWeb(keyword, maxPages)
+	if err != nil {
+		resp.ERROR(c, fmt.Sprintf("搜索失败: %v", err))
+		return
+	}
+	
+	// 扣减用户算力
+	err = h.userService.DecreasePower(int(user.Id), searchPower, model.PowerLog{
+		Type:   types.PowerConsume,
+		Model:  "web_search",
+		Remark: fmt.Sprintf("网络搜索：%s", utils.CutWords(keyword, 10)),
+	})
+	if err != nil {
+		resp.ERROR(c, "扣减算力失败："+err.Error())
+		return
+	}
+	
+	// 返回搜索结果
+	resp.SUCCESS(c, searchResults)
+}
+
 // List 获取所有的工具函数列表
 func (h *FunctionHandler) List(c *gin.Context) {
 	var items []model.Function
--- a/api/main.go
+++ b/api/main.go
@@ -256,6 +256,7 @@ func main() {
 			group.GET("clear", h.Clear)
 			group.POST("tokens", h.Tokens)
 			group.GET("stop", h.StopGenerate)
+			group.POST("tts", h.TextToSpeech)
 		}),
 		fx.Invoke(func(s *core.AppServer, h *handler.NetHandler) {
 			s.Engine.POST("/api/upload", h.Upload)
@@ -335,6 +336,7 @@ func main() {
 			group.POST("save", h.Save)
 			group.GET("remove", h.Remove)
 			group.GET("loginLog", h.LoginLog)
+			group.GET("genLoginLink", h.GenLoginLink)
 			group.POST("resetPass", h.ResetPass)
 		}),
 		fx.Invoke(func(s *core.AppServer, h *admin.ChatAppHandler) {
@@ -431,6 +433,7 @@ func main() {
 			group.POST("weibo", h.WeiBo)
 			group.POST("zaobao", h.ZaoBao)
 			group.POST("dalle3", h.Dall3)
+			group.POST("websearch", h.WebSearch)
 			group.GET("list", h.List)
 		}),
 		fx.Invoke(func(s *core.AppServer, h *admin.ChatHandler) {
--- a/api/service/crawler/service.go
+++ b/api/service/crawler/service.go
@@ -0,0 +1,333 @@
+package crawler
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"geekai/logger"
+	"net/url"
+	"strings"
+	"time"
+
+	"github.com/go-rod/rod"
+	"github.com/go-rod/rod/lib/launcher"
+	"github.com/go-rod/rod/lib/proto"
+)
+
+// Service 网络爬虫服务
+type Service struct {
+	browser *rod.Browser
+}
+
+// NewService 创建一个新的爬虫服务
+func NewService() (*Service, error) {
+	// 启动浏览器
+	path, _ := launcher.LookPath()
+	u := launcher.New().Bin(path).
+		Headless(true).                   // 无头模式
+		Set("disable-web-security", "").  // 禁用网络安全限制
+		Set("disable-gpu", "").           // 禁用 GPU 加速
+		Set("no-sandbox", "").            // 禁用沙箱模式
+		Set("disable-setuid-sandbox", "").// 禁用 setuid 沙箱
+		MustLaunch()
+
+	browser := rod.New().ControlURL(u).MustConnect()
+	
+	return &Service{
+		browser: browser,
+	}, nil
+}
+
+// SearchResult 搜索结果
+type SearchResult struct {
+	Title   string `json:"title"`   // 标题
+	URL     string `json:"url"`     // 链接
+	Content string `json:"content"` // 内容摘要
+}
+
+// WebSearch 网络搜索
+func (s *Service) WebSearch(keyword string, maxPages int) ([]SearchResult, error) {
+	if keyword == "" {
+		return nil, errors.New("搜索关键词不能为空")
+	}
+	
+	if maxPages <= 0 {
+		maxPages = 1
+	}
+	if maxPages > 10 {
+		maxPages = 10 // 最多搜索 10 页
+	}
+
+	results := make([]SearchResult, 0)
+	
+	// 使用百度搜索
+	searchURL := fmt.Sprintf("https://www.baidu.com/s?wd=%s", url.QueryEscape(keyword))
+	
+	// 设置页面超时
+	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
+	defer cancel()
+	
+	// 创建页面
+	page := s.browser.MustPage()
+	defer page.MustClose()
+	
+	// 设置视口大小
+	err := page.SetViewport(&proto.EmulationSetDeviceMetricsOverride{
+		Width:  1280,
+		Height: 800,
+	})
+	if err != nil {
+		return nil, fmt.Errorf("设置视口失败: %v", err)
+	}
+	
+	// 导航到搜索页面
+	err = page.Context(ctx).Navigate(searchURL)
+	if err != nil {
+		return nil, fmt.Errorf("导航到搜索页面失败: %v", err)
+	}
+	
+	// 等待搜索结果加载完成
+	err = page.WaitLoad()
+	if err != nil {
+		return nil, fmt.Errorf("等待页面加载完成失败: %v", err)
+	}
+	
+	// 分析当前页面的搜索结果
+	for i := 0; i < maxPages; i++ {
+		if i > 0 {
+			// 点击下一页按钮
+			nextPage, err := page.Element("a.n")
+			if err != nil || nextPage == nil {
+				break // 没有下一页
+			}
+			
+			err = nextPage.Click(proto.InputMouseButtonLeft, 1)
+			if err != nil {
+				break // 点击下一页失败
+			}
+			
+			// 等待新页面加载
+			err = page.WaitLoad()
+			if err != nil {
+				break
+			}
+		}
+		
+		// 提取搜索结果
+		resultElements, err := page.Elements(".result, .c-container")
+		if err != nil || resultElements == nil {
+			continue
+		}
+		
+		for _, result := range resultElements {
+			// 获取标题
+			titleElement, err := result.Element("h3, .t")
+			if err != nil || titleElement == nil {
+				continue
+			}
+			
+			title, err := titleElement.Text()
+			if err != nil {
+				continue
+			}
+			
+			// 获取 URL
+			linkElement, err := titleElement.Element("a")
+			if err != nil || linkElement == nil {
+				continue
+			}
+			
+			href, err := linkElement.Attribute("href")
+			if err != nil || href == nil {
+				continue
+			}
+			
+			// 获取内容摘要 - 尝试多个可能的选择器
+			var contentElement *rod.Element
+			var content string
+			
+			// 尝试多个可能的选择器来适应不同版本的百度搜索结果
+			selectors := []string{".content-right_8Zs40", ".c-abstract", ".content_LJ0WN", ".content"}
+			for _, selector := range selectors {
+				contentElement, err = result.Element(selector)
+				if err == nil && contentElement != nil {
+					content, _ = contentElement.Text()
+					if content != "" {
+						break
+					}
+				}
+			}
+			
+			// 如果所有选择器都失败，尝试直接从结果块中提取文本
+			if content == "" {
+				// 获取结果元素的所有文本
+				fullText, err := result.Text()
+				if err == nil && fullText != "" {
+					// 简单处理：从全文中移除标题，剩下的可能是摘要
+					fullText = strings.Replace(fullText, title, "", 1)
+					// 清理文本
+					content = strings.TrimSpace(fullText)
+					// 限制内容长度
+					if len(content) > 200 {
+						content = content[:200] + "..."
+					}
+				}
+			}
+			
+			// 添加到结果集
+			results = append(results, SearchResult{
+				Title:   title,
+				URL:     *href,
+				Content: content,
+			})
+			
+			// 限制结果数量，每页最多 10 条
+			if len(results) >= 10*maxPages {
+				break
+			}
+		}
+	}
+	
+	// 获取真实 URL（百度搜索结果中的 URL 是短链接，需要跳转获取真实 URL）
+	for i, result := range results {
+		realURL, err := s.getRedirectURL(result.URL)
+		if err == nil && realURL != "" {
+			results[i].URL = realURL
+		}
+	}
+	
+	return results, nil
+}
+
+// 获取真实 URL
+func (s *Service) getRedirectURL(shortURL string) (string, error) {
+	// 创建页面
+	page, err := s.browser.Page(proto.TargetCreateTarget{URL: ""})
+	if err != nil {
+		return shortURL, err // 返回原始URL
+	}
+	defer func() {
+		_ = page.Close()
+	}()
+	
+	// 导航到短链接
+	err = page.Navigate(shortURL)
+	if err != nil {
+		return shortURL, err // 返回原始URL
+	}
+	
+	// 等待重定向完成
+	time.Sleep(2 * time.Second)
+	
+	// 获取当前 URL
+	info, err := page.Info()
+	if err != nil {
+		return shortURL, err // 返回原始URL
+	}
+	
+	return info.URL, nil
+}
+
+// Close 关闭浏览器
+func (s *Service) Close() error {
+	if s.browser != nil {
+		err := s.browser.Close()
+		s.browser = nil
+		return err
+	}
+	return nil
+}
+
+// SearchWeb 封装的搜索方法
+func SearchWeb(keyword string, maxPages int) (string, error) {
+	// 添加panic恢复机制
+	defer func() {
+		if r := recover(); r != nil {
+			log := logger.GetLogger()
+			log.Errorf("爬虫服务崩溃: %v", r)
+		}
+	}()
+	
+	service, err := NewService()
+	if err != nil {
+		return "", fmt.Errorf("创建爬虫服务失败: %v", err)
+	}
+	defer service.Close()
+	
+	// 设置超时上下文
+	ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
+	defer cancel()
+	
+	// 使用goroutine和通道来处理超时
+	resultChan := make(chan []SearchResult, 1)
+	errChan := make(chan error, 1)
+	
+	go func() {
+		results, err := service.WebSearch(keyword, maxPages)
+		if err != nil {
+			errChan <- err
+			return
+		}
+		resultChan <- results
+	}()
+	
+	// 等待结果或超时
+	select {
+	case <-ctx.Done():
+		return "", fmt.Errorf("搜索超时: %v", ctx.Err())
+	case err := <-errChan:
+		return "", fmt.Errorf("搜索失败: %v", err)
+	case results := <-resultChan:
+		if len(results) == 0 {
+			return "未找到关于 \"" + keyword + "\" 的相关搜索结果", nil
+		}
+		
+		// 格式化结果
+		var builder strings.Builder
+		builder.WriteString(fmt.Sprintf("为您找到关于 \"%s\" 的 %d 条搜索结果：\n\n", keyword, len(results)))
+		
+		for i, result := range results {
+			// // 尝试打开链接获取实际内容
+			// page := service.browser.MustPage()
+			// defer page.MustClose()
+			
+			// // 设置页面超时
+			// pageCtx, pageCancel := context.WithTimeout(context.Background(), 10*time.Second)
+			// defer pageCancel()
+			
+			// // 导航到目标页面
+			// err := page.Context(pageCtx).Navigate(result.URL)
+			// if err == nil {
+			// 	// 等待页面加载
+			// 	_ = page.WaitLoad()
+				
+			// 	// 获取页面标题
+			// 	title, err := page.Eval("() => document.title")
+			// 	if err == nil && title.Value.String() != "" {
+			// 		result.Title = title.Value.String()
+			// 	}
+				
+			// 	// 获取页面主要内容
+			// 	if content, err := page.Element("body"); err == nil {
+			// 		if text, err := content.Text(); err == nil {
+			// 			// 清理并截取内容
+			// 			text = strings.TrimSpace(text)
+			// 			if len(text) > 200 {
+			// 				text = text[:200] + "..."
+			// 			}
+			// 			result.Content = text
+			// 		}
+			// 	}
+			// }
+			
+			builder.WriteString(fmt.Sprintf("%d. **%s**\n", i+1, result.Title))
+			builder.WriteString(fmt.Sprintf("   链接: %s\n", result.URL))
+			if result.Content != "" {
+				builder.WriteString(fmt.Sprintf("   摘要: %s\n", result.Content))
+			}
+			builder.WriteString("\n")
+		}
+		
+		return builder.String(), nil
+	}
+} 
--- a/api/service/dalle/service.go
+++ b/api/service/dalle/service.go
@@ -35,15 +35,13 @@ type Service struct {
 	uploadManager *oss.UploaderManager
 	taskQueue     *store.RedisQueue
 	userService   *service.UserService
-	wsService     *service.WebsocketService
 }

-func NewService(db *gorm.DB, manager *oss.UploaderManager, redisCli *redis.Client, userService *service.UserService, wsService *service.WebsocketService) *Service {
+func NewService(db *gorm.DB, manager *oss.UploaderManager, redisCli *redis.Client, userService *service.UserService) *Service {
 	return &Service{
 		httpClient:    req.C().SetTimeout(time.Minute * 3),
 		db:            db,
 		taskQueue:     store.NewRedisQueue("DallE_Task_Queue", redisCli),
-		wsService:     wsService,
 		uploadManager: manager,
 		userService:   userService,
 	}
--- a/api/service/mj/service.go
+++ b/api/service/mj/service.go
@@ -28,17 +28,15 @@ type Service struct {
 	client          *Client // MJ Client
 	taskQueue       *store.RedisQueue
 	db              *gorm.DB
-	wsService       *service.WebsocketService
 	uploaderManager *oss.UploaderManager
 	userService     *service.UserService
 }

-func NewService(redisCli *redis.Client, db *gorm.DB, client *Client, manager *oss.UploaderManager, wsService *service.WebsocketService, userService *service.UserService) *Service {
+func NewService(redisCli *redis.Client, db *gorm.DB, client *Client, manager *oss.UploaderManager, userService *service.UserService) *Service {
 	return &Service{
 		db:              db,
 		taskQueue:       store.NewRedisQueue("MidJourney_Task_Queue", redisCli),
 		client:          client,
-		wsService:       wsService,
 		uploaderManager: manager,
 		userService:     userService,
 	}
--- a/api/service/sd/service.go
+++ b/api/service/sd/service.go
@@ -33,16 +33,14 @@ type Service struct {
 	taskQueue     *store.RedisQueue
 	db            *gorm.DB
 	uploadManager *oss.UploaderManager
-	wsService     *service.WebsocketService
 	userService   *service.UserService
 }

-func NewService(db *gorm.DB, manager *oss.UploaderManager, levelDB *store.LevelDB, redisCli *redis.Client, wsService *service.WebsocketService, userService *service.UserService) *Service {
+func NewService(db *gorm.DB, manager *oss.UploaderManager, redisCli *redis.Client, userService *service.UserService) *Service {
 	return &Service{
 		httpClient:    req.C(),
 		taskQueue:     store.NewRedisQueue("StableDiffusion_Task_Queue", redisCli),
 		db:            db,
-		wsService:     wsService,
 		uploadManager: manager,
 		userService:   userService,
 	}
--- a/api/service/suno/service.go
+++ b/api/service/suno/service.go
@@ -35,18 +35,16 @@ type Service struct {
 	uploadManager *oss.UploaderManager
 	taskQueue     *store.RedisQueue
 	notifyQueue   *store.RedisQueue
-	wsService     *service.WebsocketService
 	userService   *service.UserService
 }

-func NewService(db *gorm.DB, manager *oss.UploaderManager, redisCli *redis.Client, wsService *service.WebsocketService, userService *service.UserService) *Service {
+func NewService(db *gorm.DB, manager *oss.UploaderManager, redisCli *redis.Client, userService *service.UserService) *Service {
 	return &Service{
 		httpClient:    req.C().SetTimeout(time.Minute * 3),
 		db:            db,
 		taskQueue:     store.NewRedisQueue("Suno_Task_Queue", redisCli),
 		notifyQueue:   store.NewRedisQueue("Suno_Notify_Queue", redisCli),
 		uploadManager: manager,
-		wsService:     wsService,
 		userService:   userService,
 	}
 }
--- a/api/service/video/video.go
+++ b/api/service/video/video.go
@@ -36,16 +36,14 @@ type Service struct {
 	db            *gorm.DB
 	uploadManager *oss.UploaderManager
 	taskQueue     *store.RedisQueue
-	wsService     *service.WebsocketService
 	userService   *service.UserService
 }

-func NewService(db *gorm.DB, manager *oss.UploaderManager, redisCli *redis.Client, wsService *service.WebsocketService, userService *service.UserService) *Service {
+func NewService(db *gorm.DB, manager *oss.UploaderManager, redisCli *redis.Client, userService *service.UserService) *Service {
 	return &Service{
 		httpClient:    req.C().SetTimeout(time.Minute * 3),
 		db:            db,
 		taskQueue:     store.NewRedisQueue("Video_Task_Queue", redisCli),
-		wsService:     wsService,
 		uploadManager: manager,
 		userService:   userService,
 	}
--- a/api/static/audio/2a22a7d227c444f7b69710bd9327acee43457b460cf362331a06474a1b9cc82e.mp3
+++ b/api/static/audio/2a22a7d227c444f7b69710bd9327acee43457b460cf362331a06474a1b9cc82e.mp3
--- a/api/static/audio/5b17e861d45e57d6bb49a7268115e20b07d35c527d6db45b911707a4d985251e.mp3
+++ b/api/static/audio/5b17e861d45e57d6bb49a7268115e20b07d35c527d6db45b911707a4d985251e.mp3
--- a/api/static/audio/66f89a741101a0a523ca6c6eb596c3ded47c723be27ba6acba076449fd4203ed.mp3
+++ b/api/static/audio/66f89a741101a0a523ca6c6eb596c3ded47c723be27ba6acba076449fd4203ed.mp3
--- a/api/static/audio/b2ffb7b1f51c48cb5744554ece5be09049b05f2ac8f23a7c25e1d2ddb337ea07.mp3
+++ b/api/static/audio/b2ffb7b1f51c48cb5744554ece5be09049b05f2ac8f23a7c25e1d2ddb337ea07.mp3
--- a/api/static/audio/c423597a2cb4ca7faa2962c10c547a8f07c200f8282229c1506c96e922504e91.mp3
+++ b/api/static/audio/c423597a2cb4ca7faa2962c10c547a8f07c200f8282229c1506c96e922504e91.mp3
--- a/api/static/audio/ea3a312e1b03d3833aa631af8b4b5c74e24e8ee5b38f0ae946e73e9125a905a0.mp3
+++ b/api/static/audio/ea3a312e1b03d3833aa631af8b4b5c74e24e8ee5b38f0ae946e73e9125a905a0.mp3
--- a/api/static/audio/eee981117cf345a51f08049d6976b46c04442045c785fea6ede43e8c5fde0733.mp3
+++ b/api/static/audio/eee981117cf345a51f08049d6976b46c04442045c785fea6ede43e8c5fde0733.mp3
--- a/api/store/model/chat_model.go
+++ b/api/store/model/chat_model.go
@@ -13,4 +13,5 @@ type ChatModel struct {
 	Temperature float32 // 模型温度
 	KeyId       int     // 绑定 API KEY ID
 	Type        string  // 模型类型
+	Options     string  // 模型选项
 }
--- a/api/store/vo/chat_model.go
+++ b/api/store/vo/chat_model.go
@@ -2,16 +2,17 @@ package vo

 type ChatModel struct {
 	BaseVo
-	Name        string  `json:"name"`
-	Value       string  `json:"value"`
-	Enabled     bool    `json:"enabled"`
-	SortNum     int     `json:"sort_num"`
-	Power       int     `json:"power"`
-	Open        bool    `json:"open"`
-	MaxTokens   int     `json:"max_tokens"`  // 最大响应长度
-	MaxContext  int     `json:"max_context"` // 最大上下文长度
-	Temperature float32 `json:"temperature"` // 模型温度
-	KeyId       int     `json:"key_id,omitempty"`
-	KeyName     string  `json:"key_name"`
-	Type        string  `json:"type"`
+	Name        string            `json:"name"`
+	Value       string            `json:"value"`
+	Enabled     bool              `json:"enabled"`
+	SortNum     int               `json:"sort_num"`
+	Power       int               `json:"power"`
+	Open        bool              `json:"open"`
+	MaxTokens   int               `json:"max_tokens"`  // 最大响应长度
+	MaxContext  int               `json:"max_context"` // 最大上下文长度
+	Temperature float32           `json:"temperature"` // 模型温度
+	KeyId       int               `json:"key_id,omitempty"`
+	KeyName     string            `json:"key_name"`
+	Options     map[string]string `json:"options"`
+	Type        string            `json:"type"`
 }
--- a/api/test/crawler_test.go
+++ b/api/test/crawler_test.go
@@ -0,0 +1,214 @@
+package test
+
+import (
+	"geekai/service/crawler"
+	"strings"
+	"testing"
+	"time"
+)
+
+// TestNewService 测试创建爬虫服务
+func TestNewService(t *testing.T) {
+	defer func() {
+		if r := recover(); r != nil {
+			t.Fatalf("测试过程中发生崩溃: %v", r)
+		}
+	}()
+	
+	service, err := crawler.NewService()
+	if err != nil {
+		t.Logf("注意: 创建爬虫服务失败，可能是因为Chrome浏览器未安装: %v", err)
+		t.Skip("跳过测试 - 浏览器问题")
+		return
+	}
+	defer service.Close()
+
+	// 创建服务成功则测试通过
+	if service == nil {
+		t.Fatal("创建的爬虫服务为空")
+	}
+}
+
+// TestSearchWeb 测试网络搜索功能
+func TestSearchWeb(t *testing.T) {
+	defer func() {
+		if r := recover(); r != nil {
+			t.Fatalf("测试过程中发生崩溃: %v", r)
+		}
+	}()
+	
+	// 设置测试超时时间
+	timeout := time.After(600 * time.Second)
+	done := make(chan bool)
+
+	go func() {
+		defer func() {
+			if r := recover(); r != nil {
+				t.Logf("搜索过程中发生崩溃: %v", r)
+				done <- false
+				return
+			}
+		}()
+		
+		keyword := "Golang编程"
+		maxPages := 1
+
+		// 执行搜索
+		result, err := crawler.SearchWeb(keyword, maxPages)
+		if err != nil {
+			t.Logf("搜索失败，可能是网络问题或浏览器未安装: %v", err)
+			done <- false
+			return
+		}
+
+		// 验证结果不为空
+		if result == "" {
+			t.Log("搜索结果为空")
+			done <- false
+			return
+		}
+
+		// 验证结果包含关键字或部分关键字
+		if !strings.Contains(result, "Golang") && !strings.Contains(result, "golang") {
+			t.Logf("搜索结果中未包含关键字或部分关键字，获取到的结果: %s", result)
+			done <- false
+			return
+		}
+
+		// 验证结果格式，至少应包含"链接:"
+		if !strings.Contains(result, "链接:") {
+			t.Log("搜索结果格式不正确，没有找到'链接:'部分")
+			done <- false
+			return
+		}
+
+		done <- true
+		t.Logf("搜索结果: %s", result)
+	}()
+
+	select {
+	case <-timeout:
+		t.Log("测试超时 - 这可能是正常的，特别是在网络较慢或资源有限的环境中")
+		t.Skip("跳过测试 - 超时")
+	case success := <-done:
+		if !success {
+			t.Skip("跳过测试 - 搜索失败")
+		}
+	}
+}
+
+// 减少测试用例数量，只保留基本测试
+// 这样可以减少测试时间和资源消耗
+// 以下测试用例被注释掉，可以根据需要启用
+
+/*
+// TestSearchWebNoResults 测试搜索无结果的情况
+func TestSearchWebNoResults(t *testing.T) {
+	// 设置测试超时时间
+	timeout := time.After(60 * time.Second)
+	done := make(chan bool)
+
+	go func() {
+		// 使用一个极不可能有搜索结果的随机字符串
+		keyword := "askdjfhalskjdfhas98y234hlakjsdhflakjshdflakjshdfl"
+		maxPages := 1
+
+		// 执行搜索
+		result, err := crawler.SearchWeb(keyword, maxPages)
+		if err != nil {
+			t.Errorf("搜索失败: %v", err)
+			done <- false
+			return
+		}
+
+		// 验证结果为"未找到相关搜索结果"
+		if !strings.Contains(result, "未找到") && !strings.Contains(result, "0 条搜索结果") {
+			t.Errorf("对于无结果的搜索，预期返回包含'未找到'的信息，实际返回: %s", result)
+			done <- false
+			return
+		}
+
+		done <- true
+	}()
+
+	select {
+	case <-timeout:
+		t.Fatal("测试超时")
+	case success := <-done:
+		if !success {
+			t.Fatal("测试失败")
+		}
+	}
+}
+
+// TestSearchWebMultiplePages 测试多页搜索
+func TestSearchWebMultiplePages(t *testing.T) {
+	// 设置测试超时时间
+	timeout := time.After(120 * time.Second)
+	done := make(chan bool)
+
+	go func() {
+		keyword := "golang programming"
+		maxPages := 2
+
+		// 执行搜索
+		result, err := crawler.SearchWeb(keyword, maxPages)
+		if err != nil {
+			t.Errorf("搜索失败: %v", err)
+			done <- false
+			return
+		}
+
+		// 验证结果不为空
+		if result == "" {
+			t.Error("搜索结果为空")
+			done <- false
+			return
+		}
+
+		// 计算结果中的条目数
+		resultCount := strings.Count(result, "链接:")
+		if resultCount < 10 {
+			t.Errorf("多页搜索应返回至少10条结果，实际返回: %d", resultCount)
+			done <- false
+			return
+		}
+
+		done <- true
+	}()
+
+	select {
+	case <-timeout:
+		t.Fatal("测试超时")
+	case success := <-done:
+		if !success {
+			t.Fatal("测试失败")
+		}
+	}
+}
+
+// TestSearchWebWithMaxPageLimit 测试页数限制
+func TestSearchWebWithMaxPageLimit(t *testing.T) {
+	service, err := crawler.NewService()
+	if err != nil {
+		t.Fatalf("创建爬虫服务失败: %v", err)
+	}
+	defer service.Close()
+
+	// 传入一个超过限制的页数
+	results, err := service.WebSearch("golang", 15)
+	if err != nil {
+		t.Fatalf("搜索失败: %v", err)
+	}
+
+	// 验证结果不为空
+	if len(results) == 0 {
+		t.Fatal("搜索结果为空")
+	}
+
+	// 因为最大页数限制为10，所以结果数量应该小于等于10*10=100
+	if len(results) > 100 {
+		t.Errorf("搜索结果超过最大限制，预期最多100条，实际: %d", len(results))
+	}
+}
+*/ 
--- a/api/test/run_crawler_test.sh
+++ b/api/test/run_crawler_test.sh
@@ -0,0 +1,41 @@
+#!/bin/bash
+
+# 显示执行的命令
+set -x
+
+# 检查Chrome/Chromium浏览器是否已安装
+check_chrome() {
+  echo "检查Chrome/Chromium浏览器是否安装..."
+  which chromium-browser || which google-chrome || which chromium
+  if [ $? -ne 0 ]; then
+    echo "警告: 未找到Chrome或Chromium浏览器，测试可能会失败"
+    echo "尝试安装必要的依赖..."
+    sudo apt-get update && sudo apt-get install -y libnss3 libgbm1 libasound2 libatk1.0-0 libatk-bridge2.0-0 libcups2 libxkbcommon0 libxdamage1 libxfixes3 libxrandr2 libxcomposite1 libxcursor1 libxi6 libxtst6 libnss3 libnspr4 libpango1.0-0
+    echo "已安装依赖，但仍需安装Chrome/Chromium浏览器以完全支持测试"
+  else
+    echo "已找到Chrome/Chromium浏览器"
+  fi
+}
+
+# 切换到项目根目录
+cd ..
+
+# 检查环境
+check_chrome
+
+# 运行爬虫测试，使用超时限制
+echo "开始运行爬虫测试..."
+timeout 180s go test -v ./test/crawler_test.go -run "TestNewService|TestSearchWeb"
+TEST_RESULT=$?
+
+if [ $TEST_RESULT -eq 124 ]; then
+  echo "测试超时终止"
+  exit 1
+elif [ $TEST_RESULT -ne 0 ]; then
+  echo "测试失败，退出码: $TEST_RESULT"
+  exit $TEST_RESULT
+else
+  echo "测试成功完成"
+fi
+
+echo "测试完成" 
--- a/api/utils/openai.go
+++ b/api/utils/openai.go
@@ -13,6 +13,7 @@ import (
 	"geekai/core/types"
 	"geekai/store/model"
 	"io"
+	"net/url"
 	"time"

 	"github.com/imroc/req/v3"
@@ -47,7 +48,7 @@ type OpenAIResponse struct {
 }

 func OpenAIRequest(db *gorm.DB, prompt string, modelId int) (string, error) {
-	messages := make([]interface{}, 1)
+	messages := make([]any, 1)
 	messages[0] = types.Message{
 		Role:    "user",
 		Content: prompt,
@@ -55,7 +56,7 @@ func OpenAIRequest(db *gorm.DB, prompt string, modelId int) (string, error) {
 	return SendOpenAIMessage(db, messages, modelId)
 }

-func SendOpenAIMessage(db *gorm.DB, messages []interface{}, modelId int) (string, error) {
+func SendOpenAIMessage(db *gorm.DB, messages []any, modelId int) (string, error) {
 	var chatModel model.ChatModel
 	db.Where("id", modelId).First(&chatModel)
 	if chatModel.Value == "" {
@@ -74,10 +75,17 @@ func SendOpenAIMessage(db *gorm.DB, messages []interface{}, modelId int) (string
 	var response OpenAIResponse
 	client := req.C()
 	if len(apiKey.ProxyURL) > 5 {
-		client.SetProxyURL(apiKey.ApiURL)
+		client.SetProxyURL(apiKey.ProxyURL)
 	}
-	apiURL := fmt.Sprintf("%s/v1/chat/completions", apiKey.ApiURL)
-	logger.Infof("Sending %s request, API KEY:%s, PROXY: %s, Model: %s", apiKey.ApiURL, apiURL, apiKey.ProxyURL, chatModel.Name)
+	var apiURL string
+	p, _ := url.Parse(apiKey.ApiURL)
+	// 如果设置的是 BASE_URL 没有路径，则添加 /v1/chat/completions
+	if p.Path == "" {
+		apiURL = fmt.Sprintf("%s/v1/chat/completions", apiKey.ApiURL)
+	} else {
+		apiURL = apiKey.ApiURL
+	}
+	logger.Infof("Sending %s request, API KEY:%s, PROXY: %s, Model: %s", apiURL, apiKey.ApiURL, apiKey.ProxyURL, chatModel.Name)
 	r, err := client.R().SetHeader("Body-Type", "application/json").
 		SetHeader("Authorization", "Bearer "+apiKey.Value).
 		SetBody(types.ApiRequest{