[fix] (#34) token 消耗阈值自定义

This commit is contained in:
SXP-Simon
2025-11-04 18:45:46 +08:00
parent 7250293f7c
commit 036a5c3581
5 changed files with 33 additions and 3 deletions
+18
View File
@@ -117,6 +117,24 @@
"default": 2,
"hint": "重试之间的基准等待时间(秒),实际等待时间为基值乘以尝试次数。"
},
"topic_max_tokens": {
"type": "int",
"description": "话题分析最大 Token 数",
"default": 12288,
"hint": "话题分析时 LLM 能生成的最大 token 数量。当分析内容较多或者分析提示词复杂时,建议适当调大此值以保证输出质量。"
},
"golden_quote_max_tokens": {
"type": "int",
"description": "金句分析最大 Token 数",
"default": 4096,
"hint": "金句分析时 LLM 能生成的最大 token 数量。当分析内容较多或者分析提示词复杂时,建议适当调大此值以保证输出质量。"
},
"user_title_max_tokens": {
"type": "int",
"description": "用户称号分析最大 Token 数",
"default": 4096,
"hint": "用户称号分析时 LLM 能生成的最大 token 数量。当分析内容较多或者分析提示词复杂时,建议适当调大此值以保证输出质量。"
},
"custom_api_key": {
"type": "string",
"description": "自定义 LLM 服务 API Key (选填)",
@@ -29,7 +29,7 @@ class GoldenQuoteAnalyzer(BaseAnalyzer):
def get_max_tokens(self) -> int:
"""获取最大token数"""
return 1500
return self.config_manager.get_golden_quote_max_tokens()
def get_temperature(self) -> float:
"""获取温度参数"""
+1 -1
View File
@@ -29,7 +29,7 @@ class TopicAnalyzer(BaseAnalyzer):
def get_max_tokens(self) -> int:
"""获取最大token数"""
return 10000
return self.config_manager.get_topic_max_tokens()
def get_temperature(self) -> float:
"""获取温度参数"""
@@ -26,7 +26,7 @@ class UserTitleAnalyzer(BaseAnalyzer):
def get_max_tokens(self) -> int:
"""获取最大token数"""
return 1500
return self.config_manager.get_user_title_max_tokens()
def get_temperature(self) -> float:
"""获取温度参数"""
+12
View File
@@ -87,6 +87,18 @@ class ConfigManager:
"""获取LLM请求重试退避基值(秒),实际退避会乘以尝试次数"""
return self.config.get("llm_backoff", 2)
def get_topic_max_tokens(self) -> int:
"""获取话题分析最大token数"""
return self.config.get("topic_max_tokens", 12288)
def get_golden_quote_max_tokens(self) -> int:
"""获取金句分析最大token数"""
return self.config.get("golden_quote_max_tokens", 4096)
def get_user_title_max_tokens(self) -> int:
"""获取用户称号分析最大token数"""
return self.config.get("user_title_max_tokens", 4096)
def get_custom_api_key(self) -> str:
"""获取自定义 LLM 服务的 API Key"""
return self.config.get("custom_api_key", "")