From 2bcd2adc1cd104eb94cfde6d551a6ef4a78b7f05 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=85=AC=E6=98=8E?=
 <83812544+Ed1s0nZ@users.noreply.github.com>
Date: Fri, 17 Apr 2026 15:14:04 +0800
Subject: [PATCH] Add files via upload

---
 internal/handler/config.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/internal/handler/config.go b/internal/handler/config.go
index fffc33b1..766099ea 100644
--- a/internal/handler/config.go
+++ b/internal/handler/config.go
@@ -1255,6 +1255,9 @@ func updateOpenAIConfig(doc *yaml.Node, cfg config.OpenAIConfig) {
 	setStringInMap(openaiNode, "api_key", cfg.APIKey)
 	setStringInMap(openaiNode, "base_url", cfg.BaseURL)
 	setStringInMap(openaiNode, "model", cfg.Model)
+	if cfg.MaxTotalTokens > 0 {
+		setIntInMap(openaiNode, "max_total_tokens", cfg.MaxTotalTokens)
+	}
 }
 
 func updateFOFAConfig(doc *yaml.Node, cfg config.FofaConfig) {