fix: SESSION_IMAGE_CHAT_MAP

Author: deanxv

README.md

@@ -32,7 +32,7 @@ _觉得有点意思的话 别忘了点个 ⭐_
     - **deep-seek-r1**
 - [x] 支持**联网搜索**,在模型名后添加`-search`即可(如:`gpt-4o-search`)
 - [x] 支持识别**图片**/**文件**多轮对话
-- [x] 支持文生图接口(`/images/generations`)
+- [x] 支持文生图接口(`/images/generations`),详细请看[生图模型配置](#生图模型配置)
     - **flux**
     - **flux-speed**
     - **flux-pro/ultra**
@@ -173,10 +173,10 @@ Render 可以直接部署 docker 镜像,不需要 fork 仓库：[Render](https:/
 10. `ROUTE_PREFIX=hf`  [可选]路由前缀,默认为空,添加该变量后的接口示例:`/hf/v1/chat/completions`
 11. `RATE_LIMIT_COOKIE_LOCK_DURATION=600`  [可选]到达速率限制的cookie禁用时间,默认为600s
 12. `REASONING_HIDE=0`  [可选]**隐藏**推理过程(默认:0)[0:关闭,1:开启]
+13. `SESSION_IMAGE_CHAT_MAP=aed9196b-********-4ed6e32f7e4d=0c6785e6-********-7ff6e5a2a29c,aefwer6b-********-casds22=fda234-********-sfaw123`  [可选]Session绑定Image-Chat(多个请以,分隔),详细请看[进阶配置](#生图模型配置)
 
-~~11. `YES_CAPTCHA_CLIENT_KEY=******`  [可选]YesCaptcha Client Key 过谷歌验证,详细请看[使用YesCaptcha过谷歌验证](#使用YesCaptcha过谷歌验证)~~
+~~14. `YES_CAPTCHA_CLIENT_KEY=******`  [可选]YesCaptcha Client Key 过谷歌验证,详细请看[使用YesCaptcha过谷歌验证](#使用YesCaptcha过谷歌验证)~~
 
-~~12. `SESSION_IMAGE_CHAT_MAP=aed9196b-********-4ed6e32f7e4d=0c6785e6-********-7ff6e5a2a29c,aefwer6b-********-casds22=fda234-********-sfaw123`  [可选]Session绑定Image-Chat(多个请以,分隔),详细请看[进阶配置](#生图模型配置)~~
 
 ### cookie获取方式
 
@@ -211,15 +211,15 @@ Render 可以直接部署 docker 镜像,不需要 fork 仓库：[Render](https:/
    ![img.png](docs/img4.png)
 4. 配置环境变量 `MODEL_CHAT_MAP=claude-3-5-sonnet=3cdcc******474c5` (多个请以,分隔)
 
-### 生图模型配置[**暂不需要**]
+### 生图模型配置
 
 > 配置环境变量 SESSION_IMAGE_CHAT_MAP
 
-~~1. 打开**F12**开发者工具。~~
-~~2. 选择生成图像,选择任一生图模型,发起对话。~~
-~~3. 点击ask请求,此时最上方url中的`id`(或响应中的`id`)即为此对话唯一id,然后在请求头中获取`session_id`的值。
-   ![img.png](docs/img7.png)~~
-~~4. 配置环境变量 `SESSION_IMAGE_CHAT_MAP=aed9196b-********-4ed6e32f7e4d=0c6785e6-********-7ff6e5a2a29c` (即session=chatId的格式,多个请以,分隔)~~
+1. 打开**F12**开发者工具。
+2. 选择生成图像,选择任一生图模型,发起对话。
+3. 点击ask请求,此时最上方url中的`id`(或响应中的`id`)即为此对话唯一id,然后在请求头中获取`session_id`的值。
+   ![img.png](docs/img7.png)
+4. 配置环境变量 `SESSION_IMAGE_CHAT_MAP=aed9196b-********-4ed6e32f7e4d=0c6785e6-********-7ff6e5a2a29c` (即session=chatId的格式,多个请以,分隔)
 
 ### 使用YesCaptcha过谷歌验证[**暂不需要**]
 

check/check.go

@@ -44,25 +44,25 @@ func CheckEnvVariable() {
 		}
 	}
 
-	//if config.SessionImageChatMapStr != "" {
-	//	pattern := `^([a-zA-Z0-9\-\/]+=([a-zA-Z0-9\-\.]+))(,[a-zA-Z0-9\-\/]+=([a-zA-Z0-9\-\.]+))*`
-	//	match, _ := regexp.MatchString(pattern, config.SessionImageChatMapStr)
-	//	if !match {
-	//		logger.FatalLog("环境变量 SESSION_IMAGE_CHAT_MAP 设置有误")
-	//	} else {
-	//		sessionImageChatMap := make(map[string]string)
-	//		pairs := strings.Split(config.SessionImageChatMapStr, ",")
-	//
-	//		for _, pair := range pairs {
-	//			kv := strings.Split(pair, "=")
-	//			sessionImageChatMap["session_id="+kv[0]] = kv[1]
-	//		}
-	//
-	//		config.SessionImageChatMap = sessionImageChatMap
-	//	}
-	//} else {
-	//	logger.SysLog("环境变量 SESSION_IMAGE_CHAT_MAP 未设置，生图可能会异常")
-	//}
+	if config.SessionImageChatMapStr != "" {
+		pattern := `^([a-zA-Z0-9\-\/]+=([a-zA-Z0-9\-\.]+))(,[a-zA-Z0-9\-\/]+=([a-zA-Z0-9\-\.]+))*`
+		match, _ := regexp.MatchString(pattern, config.SessionImageChatMapStr)
+		if !match {
+			logger.FatalLog("环境变量 SESSION_IMAGE_CHAT_MAP 设置有误")
+		} else {
+			sessionImageChatMap := make(map[string]string)
+			pairs := strings.Split(config.SessionImageChatMapStr, ",")
+
+			for _, pair := range pairs {
+				kv := strings.Split(pair, "=")
+				sessionImageChatMap["session_id="+kv[0]] = kv[1]
+			}
+
+			config.SessionImageChatMap = sessionImageChatMap
+		}
+	} else {
+		logger.SysLog("环境变量 SESSION_IMAGE_CHAT_MAP 未设置，生图可能会异常")
+	}
 
 	logger.SysLog("environment variable check passed.")
 }

common/constants.go

@@ -3,7 +3,7 @@ package common
 import "time"
 
 var StartTime = time.Now().Unix() // unit: second
-var Version = "v1.10.12"          // this hard coding will be replaced automatically when building, no need to manually change
+var Version = "v1.10.13"          // this hard coding will be replaced automatically when building, no need to manually change
 
 var DefaultOpenaiModelList = []string{
 	"gpt-4o",

controller/chat.go

@@ -492,7 +492,7 @@ func createImageRequestBody(c *gin.Context, cookie string, openAIReq *model.Open
 	}
 	var currentQueryString string
 	if len(chatId) != 0 {
-		currentQueryString = fmt.Sprintf("id=%s&type=%s", chatId, imageType)
+		currentQueryString = fmt.Sprintf("id=%s&type=%s", "90280b44-9ddf-4f4f-ba62-d72d38a82f99", imageType)
 	} else {
 		currentQueryString = fmt.Sprintf("type=%s", imageType)
 	}
@@ -1377,31 +1377,31 @@ func ImageProcess(c *gin.Context, client cycletls.CycleTLS, openAIReq model.Open
 	)
 
 	var (
-		//sessionImageChatManager *config.SessionMapManager
-		maxRetries int
-		cookie     string
-		chatId     string
+		sessionImageChatManager *config.SessionMapManager
+		maxRetries              int
+		cookie                  string
+		chatId                  string
 	)
 
 	cookieManager := config.NewCookieManager()
-	//sessionImageChatManager = config.NewSessionMapManager()
+	sessionImageChatManager = config.NewSessionMapManager()
 	ctx := c.Request.Context()
 
 	// Initialize session manager and get initial cookie
-	//if len(config.SessionImageChatMap) == 0 {
-	//logger.Warnf(ctx, "未配置环境变量 SESSION_IMAGE_CHAT_MAP, 可能会生图失败!")
-	maxRetries = len(cookieManager.Cookies)
+	if len(config.SessionImageChatMap) == 0 {
+		logger.Warnf(ctx, "未配置环境变量 SESSION_IMAGE_CHAT_MAP, 可能会生图失败!")
+		maxRetries = len(cookieManager.Cookies)
 
-	var err error
-	cookie, err = cookieManager.GetRandomCookie()
-	if err != nil {
-		logger.Errorf(ctx, "Failed to get initial cookie: %v", err)
-		return nil, fmt.Errorf(errNoValidCookies)
+		var err error
+		cookie, err = cookieManager.GetRandomCookie()
+		if err != nil {
+			logger.Errorf(ctx, "Failed to get initial cookie: %v", err)
+			return nil, fmt.Errorf(errNoValidCookies)
+		}
+	} else {
+		maxRetries = sessionImageChatManager.GetSize()
+		cookie, chatId, _ = sessionImageChatManager.GetRandomKeyValue()
 	}
-	//} else {
-	//	maxRetries = sessionImageChatManager.GetSize()
-	//	cookie, chatId, _ = sessionImageChatManager.GetRandomKeyValue()
-	//}
 
 	for attempt := 0; attempt < maxRetries; attempt++ {
 		// Create request body